iio/DEFENDER
1
0
Fork 0
mirror of https://github.com/iio612/DEFENDER.git synced 2026-02-13 19:24:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update parse_uid, now it returns MUser object.

Browse Source
This commit is contained in:
adator
2025-11-02 21:17:15 +01:00
parent 34b5b4204e
commit 5a1432c1e6
4 changed files with 15 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
core/classes/protocols/inspircd.py
View File

@@ -1186,35 +1186,15 @@ class Inspircd(IProtocol):
# COMMON IRC PARSER
# ------------------------------------------------------------------------
def parse_uid(self, server_msg: list[str]) -> dict[str, str]:
def parse_uid(self, server_msg: list[str]) -> Optional['MUser']:
"""Parse UID and return dictionary.
>>> [':97K', 'UID', '97KAAAAAC', '1762113659', 'adator_', '172.18.128.1', '172.18.128.1', '...', '...', '172.18.128.1', '1762113659', '+', ':...']
Args:
server_msg (list[str]): _description_
"""
umodes = str(server_msg[11])
remote_ip = server_msg[9] if 'S' not in umodes else '127.0.0.1'
# Extract Geoip information
pattern = r'^.*geoip=cc=(\S{2}).*$'
geoip_match = match(pattern, server_msg[0])
geoip = geoip_match.group(1) if geoip_match else None
response = {
'uid': str(server_msg[2]),
'nickname': str(server_msg[4]),
'username': str(server_msg[7]),
'hostname': str(server_msg[5]),
'umodes': umodes,
'vhost': str(server_msg[6]),
'ip': remote_ip,
'realname': ' '.join(server_msg[12:]).lstrip(':'),
'geoip': geoip,
'reputation_score': 0,
'iswebirc': True if 'webirc' in server_msg[0] else False,
'iswebsocket': True if 'websocket' in server_msg[0] else False
}
return response
scopy = server_msg.copy()
uid = scopy[2]
return self._User.get_user(uid)
def parse_quit(self, server_msg: list[str]) -> dict[str, str]:
"""Parse quit and return dictionary.

4
core/classes/protocols/interface.py
View File

@@ -316,14 +316,14 @@ class IProtocol(ABC):
# ------------------------------------------------------------------------
@abstractmethod
def parse_uid(self, serverMsg: list[str]) -> dict[str, str]:
def parse_uid(self, serverMsg: list[str]) -> Optional['MUser']:
"""Parse UID and return dictionary.
Args:
serverMsg (list[str]): The UID IRCD message
Returns:
dict[str, str]: The response as dictionary.
Optional[MUser]: The MUser object or None
"""
@abstractmethod

29
core/classes/protocols/unreal6.py
View File

@@ -630,35 +630,18 @@ class Unrealircd6(IProtocol):
# COMMON IRC PARSER
# ------------------------------------------------------------------------
def parse_uid(self, serverMsg: list[str]) -> dict[str, str]:
def parse_uid(self, serverMsg: list[str]) -> Optional['MUser']:
"""Parse UID and return dictionary.
>>> ['@s2s-md/geoip=cc=GBtag...', ':001', 'UID', 'albatros', '0', '1721564597', 'albatros', 'hostname...', '001HB8G04', '0', '+iwxz', 'hostname-vhost', 'hostname-vhost', 'MyZBwg==', ':...']
Args:
serverMsg (list[str]): The UID ircd response
"""
umodes = str(serverMsg[10])
remote_ip = self._Base.decode_ip(str(serverMsg[13])) if 'S' not in umodes else '127.0.0.1'
scopy = serverMsg.copy()
if '@' in scopy[0]:
scopy.pop(0)
# Extract Geoip information
pattern = r'^.*geoip=cc=(\S{2}).*$'
geoip_match = match(pattern, serverMsg[0])
geoip = geoip_match.group(1) if geoip_match else None
response = {
'uid': str(serverMsg[8]),
'nickname': str(serverMsg[3]),
'username': str(serverMsg[6]),
'hostname': str(serverMsg[7]),
'umodes': umodes,
'vhost': str(serverMsg[11]),
'ip': remote_ip,
'realname': ' '.join(serverMsg[12:]).lstrip(':'),
'geoip': geoip,
'reputation_score': 0,
'iswebirc': True if 'webirc' in serverMsg[0] else False,
'iswebsocket': True if 'websocket' in serverMsg[0] else False
}
return response
uid = scopy[7]
return self._User.get_user(uid)
def parse_quit(self, serverMsg: list[str]) -> dict[str, str]:
"""Parse quit and return dictionary.

6
mods/defender/utils.py
View File

@@ -212,7 +212,7 @@ def handle_on_uid(uplink: 'Defender', srvmsg: list[str]):
uplink (Defender): The Defender instance
srvmsg (list[str]): The Server MSG
"""
parser_uid = uplink.Protocol.parse_uid(srvmsg)
_User = uplink.Protocol.parse_uid(srvmsg)
gconfig = uplink.Config
irc = uplink.Irc
confmodel = uplink.ModConfig
@@ -222,10 +222,8 @@ def handle_on_uid(uplink: 'Defender', srvmsg: list[str]):
return None
# Get User information
_User = irc.User.get_user(parser_uid.get('uid', None))
if _User is None:
irc.Logs.warning(f'This UID: [{parser_uid.get("uid", None)}] is not available please check why')
irc.Logs.warning(f'Error when parsing UID', exc_info=True)
return
# If user is not service or IrcOp then scan them