From 78161bda7f95cfd72b1386ab7bb93c4aad7b741c Mon Sep 17 00:00:00 2001 From: Georg Date: Thu, 26 Aug 2021 19:18:14 +0200 Subject: Added nickname and IP address counting Signed-off-by: Georg --- __pycache__/config.cpython-38.pyc | Bin 2453 -> 2704 bytes __pycache__/plugin.cpython-38.pyc | Bin 8505 -> 10414 bytes config.py | 66 +++++++++++++------ plugin.py | 135 ++++++++++++++++++++++++++++++++------ 4 files changed, 161 insertions(+), 40 deletions(-) diff --git a/__pycache__/config.cpython-38.pyc b/__pycache__/config.cpython-38.pyc index 172cadd..d7e3edd 100644 Binary files a/__pycache__/config.cpython-38.pyc and b/__pycache__/config.cpython-38.pyc differ diff --git a/__pycache__/plugin.cpython-38.pyc b/__pycache__/plugin.cpython-38.pyc index 7fa3b98..1a89f54 100644 Binary files a/__pycache__/plugin.cpython-38.pyc and b/__pycache__/plugin.cpython-38.pyc differ diff --git a/config.py b/config.py index 9617979..983cb6d 100644 --- a/config.py +++ b/config.py @@ -61,62 +61,90 @@ conf.registerGlobalValue(SnoParser, 'AutoVhost', conf.registerGlobalValue(SnoParser, 'preventHighlight', registry.Boolean(True, ("""Toggles in channel highlights with ZWSP"""))) +conf.registerGlobalValue(SnoParser, 'debug', + registry.Boolean('false', + """ + SnoParser: Verbose output. Note: there is a seperate debug option for the `whois` client. + """ + , private=True +)) + ### -## WHOIS related settings below: +# REDIS related settings below: ### -conf.registerGroup(SnoParser, 'whois') -conf.registerGlobalValue(SnoParser.whois, 'debug', - registry.Boolean('false', +conf.registerGroup(SnoParser, 'redis') +conf.registerGlobalValue(SnoParser.redis, 'db1', + registry.Integer('1', """ - SnoParser: True: Very verbose console output. False: Mostly silent operation. + Redis: Database number for counting of NICKNAMES. """ , private=True )) -conf.registerGlobalValue(SnoParser.whois, 'sample', - registry.String('', +conf.registerGlobalValue(SnoParser.redis, 'db2', + registry.Integer('2', """ - SnoParser: This allows to set a testing IP address, if the plugin shall be evaluated on i.e. a local network. This will override all IP addresses from SNOTICES! + Redis: Database number for counting of IP ADDRESSES. """ , private=True )) -conf.registerGroup(SnoParser.whois, 'redis') -conf.registerGlobalValue(SnoParser.whois.redis, 'host', +conf.registerGlobalValue(SnoParser.redis, 'host', registry.String('127.0.0.1', """ Redis: IP address or hostname. """ , private=True )) -conf.registerGlobalValue(SnoParser.whois.redis, 'port', +conf.registerGlobalValue(SnoParser.redis, 'port', registry.Integer('6379', """ Redis: Port. """ , private=True )) -conf.registerGlobalValue(SnoParser.whois.redis, 'username', +conf.registerGlobalValue(SnoParser.redis, 'username', registry.String('', """ Redis: Username. This is optional and has not been tested. It is recommended to perform initial tests on a local instance with no username and password. """ , private=True )) -conf.registerGlobalValue(SnoParser.whois.redis, 'password', +conf.registerGlobalValue(SnoParser.redis, 'password', registry.String('', """ Redis: Password. This is optional and has not been tested. It is recommended to perform initial tests on a local instance with no username and password. """ )) -conf.registerGlobalValue(SnoParser.whois.redis, 'db', - registry.Integer('0', +conf.registerGlobalValue(SnoParser.redis, 'timeout', + registry.Integer('5', """ - Redis: Database number. It is recommended to use a dedicated, isolated, Redis instance for SnoParser instead of using an existing one with a different database. + Redis: Socket Timeout. The developer does not know what to recommend here, but `5` seems to yield good results. """ )) -conf.registerGlobalValue(SnoParser.whois.redis, 'timeout', - registry.Integer('5', + + +### +## WHOIS related settings below: +### +conf.registerGroup(SnoParser, 'whois') +conf.registerGlobalValue(SnoParser.whois, 'debug', + registry.Boolean('false', """ - Redis: Socket Timeout. The developer does not know what to recommend here, but `5` seems to yield good results. + SnoParser: True: Very verbose console output. False: Mostly silent operation. + """ + , private=True +)) +conf.registerGlobalValue(SnoParser.whois, 'sample', + registry.String('', + """ + SnoParser: This allows to set a testing IP address, if the plugin shall be evaluated on i.e. a local network. This will override all IP addresses from SNOTICES! + """ + , private=True +)) +conf.registerGroup(SnoParser.whois, 'redis') +conf.registerGlobalValue(SnoParser.whois.redis, 'db', + registry.Integer('0', + """ + Redis: Database number for WHOIS query caching. """ )) diff --git a/plugin.py b/plugin.py index d47440f..4d9a797 100644 --- a/plugin.py +++ b/plugin.py @@ -53,25 +53,60 @@ class SnoParser(callbacks.Plugin): """Parses the Server Notices from ErgoIRCd""" threaded = True - def redis_connect(self) -> redis.client.Redis: + def redis_connect_whois(self) -> redis.client.Redis: try: - redis_client = redis.Redis( - host = self.registryValue('whois.redis.host'), - port = self.registryValue('whois.redis.port'), - password = self.registryValue('whois.redis.password'), - username = self.registryValue('whois.redis.username'), + redis_client_whois = redis.Redis( + host = self.registryValue('redis.host'), + port = self.registryValue('redis.port'), + password = self.registryValue('redis.password'), + username = self.registryValue('redis.username'), db = self.registryValue('whois.redis.db'), - socket_timeout = int(self.registryValue('whois.redis.timeout')) + socket_timeout = int(self.registryValue('redis.timeout')) ) - ping = redis_client.ping() + ping = redis_client_whois.ping() if ping is True: - return redis_client + return redis_client_whois except redis.AuthenticationError: print("Could not authenticate to Redis backend.") + def redis_connect_nicks(self) -> redis.client.Redis: + try: + redis_client_nicks = redis.Redis( + host = self.registryValue('redis.host'), + port = self.registryValue('redis.port'), + password = self.registryValue('redis.password'), + username = self.registryValue('redis.username'), + db = self.registryValue('redis.db1'), + socket_timeout = int(self.registryValue('redis.timeout')) + ) + ping = redis_client_nicks.ping() + if ping is True: + return redis_client_nicks + except redis.AuthenticationError: + print("Could not authenticate to Redis backend.") + + def redis_connect_ips(self) -> redis.client.Redis: + try: + redis_client_ips = redis.Redis( + host = self.registryValue('redis.host'), + port = self.registryValue('redis.port'), + password = self.registryValue('redis.password'), + username = self.registryValue('redis.username'), + db = self.registryValue('redis.db2'), + socket_timeout = int(self.registryValue('redis.timeout')) + ) + ping = redis_client_ips.ping() + if ping is True: + return redis_client_ips + except redis.AuthenticationError: + print("Could not authenticate to Redis backend.") + + def __init__(self, irc): super().__init__(irc) - self.redis_client = self.redis_connect() + self.redis_client_whois = self.redis_connect_whois() + self.redis_client_nicks = self.redis_connect_nicks() + self.redis_client_ips = self.redis_connect_ips() def whois_fresh(self, sourceip: str) -> dict: """Data from cache.""" @@ -96,17 +131,17 @@ class SnoParser(callbacks.Plugin): def whois_get_cache(self, key: str) -> str: """Data from Redis.""" - k = self.redis_client.get(key) + k = self.redis_client_whois.get(key) # self = SnoParser() -# val = self.redis_client.get(key) +# val = self.redis_client_whois.get(key) val = k return val def whois_set_cache(self, key: str, value: str) -> bool: """Data to Redis.""" - state = self.redis_client.setex(key, timedelta(seconds=3600), value=value,) + state = self.redis_client_whois.setex(key, timedelta(seconds=3600), value=value,) return state def whois_run(self, sourceip: str) -> dict: @@ -141,19 +176,76 @@ class SnoParser(callbacks.Plugin): print(data) return data - def query(self, irc, msg, args, ipaddress): + def nick_run(self, nickname: str) -> dict: + """Tracks nicknames""" + + data = self.redis_client_nicks.get(nickname) + + if data is not None: + if self.registryValue('debug'): + print("SNOPARSER DEBUG - NICK_RUN, SEEN: TRUE") + print(nickname) + print(data) + self.redis_client_nicks.incrby(nickname,amount=1) + if data: + decoded = data.decode('utf-8') + return decoded + else: + return 0 + else: + if self.registryValue('debug'): + print("SNOPARSER DEBUG _ NICK_RUN, SEEN: FALSE") + print(nickname) + print(data) + self.redis_client_nicks.set(nickname,value='1') + if data: + decoded = data.decode('utf-8') + return decoded + else: + return 0 + + def ip_run(self, ipaddress: str) -> dict: + """Tracks IP addresses""" + + data = self.redis_client_ips.get(ipaddress) + + if data is not None: + if self.registryValue('debug'): + print("SNOPARSER DEBUG - IP_RUN, SEEN: TRUE") + print(ipaddress) + print(data) + self.redis_client_ips.incrby(ipaddress,amount=1) + if data: + decoded = data.decode('utf-8') + return decoded + else: + return 0 + else: + if self.registryValue('debug'): + print("SNOPARSER DEBUG _ IP_RUN, SEEN: FALSE") + print(ipaddress) + print(data) + self.redis_client_ips.set(ipaddress,value='1') + if data: + decoded = data.decode('utf-8') + return decoded + else: + return 0 + + def ipquery(self, irc, msg, args, ipaddress): """ Queries the cache for an address. """ data = self.whois_get_cache(key=ipaddress) decoded = data.decode('utf-8') - ttl = self.redis_client.ttl(ipaddress) + ttl = self.redis_client_whois.ttl(ipaddress) + count = self.redis_client_ips.get(ipaddress) print('SnoParser manual query: ', data, ' ', ttl) - irc.reply(f'{decoded} - Remaining: {ttl}s') + irc.reply(f'{decoded} - Count: {count} - Remaining: {ttl}s') - query = wrap(query, ['anything']) + ipquery = wrap(ipquery, ['anything']) def doNotice(self, irc, msg): @@ -173,13 +265,14 @@ class SnoParser(callbacks.Plugin): else: ip = couple.group(4) realname = couple.group(5) - ip_seen = 0 - nick_seen = 0 - whoisout = self.whois_run(sourceip=ip) + ip_seen = self.ip_run(ipaddress=ip) + nick_seen = self.nick_run(nickname=nickname) + whois = self.whois_run(sourceip=ip) + DictFromSnotice = {'notice': 'connect', 'nickname': nickname, 'username': username, 'host': host, 'ip': ip, 'realname': realname, 'ipCount': ip_seen, 'nickCount': nick_seen} #repl = f"\x02\x1FNOTICE: {DictFromSnotice['notice']} \x0F\x11\x0303==>>\x0F \x02Nick:\x0F {DictFromSnotice['nickname']} \x02Username:\x0F {DictFromSnotice['username']} \x02Hostname:\x0F {DictFromSnotice['host']} \x02IP:\x0F {DictFromSnotice['ip']} \x02Realname:\x0F {DictFromSnotice['realname']} \x02IPcount:\x0F {DictFromSnotice['ipCount']} \x02NickCount:\x0F {DictFromSnotice['nickCount']}" - repl = f"\x02\x1F{DictFromSnotice['notice']} \x0F\x11\x0303==>>\x0F \x02Nick:\x0F {DictFromSnotice['nickname']} \x02Username:\x0F {DictFromSnotice['username']} \x02Hostname:\x0F {DictFromSnotice['host']} \x02IP:\x0F {DictFromSnotice['ip']} {whoisout} \x02Realname:\x0F {DictFromSnotice['realname']} \x02IPcount:\x0F {DictFromSnotice['ipCount']} \x02NickCount:\x0F {DictFromSnotice['nickCount']}" + repl = f"\x02\x1F{DictFromSnotice['notice']} \x0F\x11\x0303==>>\x0F \x02Nick:\x0F {DictFromSnotice['nickname']} \x02Username:\x0F {DictFromSnotice['username']} \x02Hostname:\x0F {DictFromSnotice['host']} \x02IP:\x0F {DictFromSnotice['ip']} {whois} \x02Realname:\x0F {DictFromSnotice['realname']} \x02IPcount:\x0F {DictFromSnotice['ipCount']} \x02NickCount:\x0F {DictFromSnotice['nickCount']}" self._sendSnotice(irc, msg, repl) if 'XLINE' in text and 'temporary' in text: -- cgit v1.2.3