internetrelaychat/IRCP
internetrelaychat/IRCP
1
Fork 0
mirror of git://git.acid.vegas/IRCP.git synced 2024-11-23 00:16:41 +00:00
Code Issues Activity

Color support implemented in the console out

Browse Source
This commit is contained in:
Dionysus 2023-05-30 15:08:52 -04:00
parent 2ed28b5abb
commit 79c91ba6d1
Signed by: acidvegas
GPG Key ID: EF4B922DB85DC9DE
1 changed files with 39 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
ircp.py
View File

@ -53,9 +53,10 @@ snapshot = {
'host' : None, 'host' : None,
'services' : False, 'services' : False,
'ssl' : False, 'ssl' : False,
#'proxy' : False,
'raw' : [], # all other data goes in here 'raw' : [], # all other data goes in here
'CAP' : None, 'CAP' : None,
'KILL', : None, # TODO: currently does not verify it was us being killed 'KILL' : None, # TODO: currently does not verify it was us being killed
'NOTICE' : None, 'NOTICE' : None,
# server information # server information
@ -123,11 +124,11 @@ snapshot = {
} }
def debug(data): def debug(data):
print('{0} | [~] - {1}'.format(time.strftime('%I:%M:%S'), data)) print('{0} \033[30m|\033[0m [\033[35m~\033[0m] {1}'.format(time.strftime('%I:%M:%S'), data))
def error(data, reason=None): def error(data, reason=None):
if settings.errors: if settings.errors:
print('{0} | [!] - {1} ({2})'.format(time.strftime('%I:%M:%S'), data, str(reason))) if reason else print('{0} | [!] - {1}'.format(time.strftime('%I:%M:%S'), data)) print('{0} \033[30m|\033[0m [\033[31m!\033[0m] {1} \033[30m({2})\033[0m'.format(time.strftime('%I:%M:%S'), data, str(reason))) if reason else print('{0} \033[30m|\033[0m [\033[31m!\033[0m] - {1}'.format(time.strftime('%I:%M:%S'), data))
def rndnick(): def rndnick():
prefix = random.choice(['st','sn','cr','pl','pr','fr','fl','qu','br','gr','sh','sk','tr','kl','wr','bl']+list('bcdfgklmnprstvwz')) prefix = random.choice(['st','sn','cr','pl','pr','fr','fl','qu','br','gr','sh','sk','tr','kl','wr','bl']+list('bcdfgklmnprstvwz'))
@ -144,14 +145,14 @@ def ssl_ctx():
class probe: class probe:
def __init__(self, server, semaphore): def __init__(self, server, semaphore):
self.server = server self.server = server
self.display = server.ljust(18)+' | ' self.display = server.ljust(18)+'\033[30m|\033[0m'
self.semaphore = semaphore self.semaphore = semaphore
self.nickname = None self.nickname = None
self.snapshot = {'raw':list()} self.snapshot = {'raw':list()}
self.multi = '' self.multi = ''
self.channels = {'all':list(), 'current':list(), 'users':dict()} self.channels = {'all':list(), 'current':list(), 'users':dict()}
self.nicks = {'all':list(), 'check':list()} self.nicks = {'all':list(), 'check':list()}
self.loops = {'init':None,'chan':None,'nick':None,'whois':None} self.loops = {'init':None, 'chan':None, 'nick':None, 'whois':None}
self.jthrottle = throttle.join self.jthrottle = throttle.join
self.reader = None self.reader = None
self.write = None self.write = None
@ -162,12 +163,12 @@ class probe:
await self.connect() await self.connect()
except Exception as ex: except Exception as ex:
if settings.errors_conn: if settings.errors_conn:
error(self.display + 'failed to connect using SSL/TLS', ex) error(self.display + '\033[1;31mdisconnected\033[0m - failed to connect using SSL/TLS', ex)
try: try:
await self.connect(True) await self.connect(True)
except Exception as ex: except Exception as ex:
if settings.errors_conn: if settings.errors_conn:
error(self.display + 'failed to connect', ex) error(self.display + '\033[1;31mdisconnected\033[0m - failed to connect', ex)
async def raw(self, data): async def raw(self, data):
self.writer.write(data[:510].encode('utf-8') + b'\r\n') self.writer.write(data[:510].encode('utf-8') + b'\r\n')
@ -210,9 +211,9 @@ class probe:
'pass': settings.ns_pass if settings.ns_pass else rndnick(), 'pass': settings.ns_pass if settings.ns_pass else rndnick(),
'mail': settings.ns_mail if settings.ns_mail else f'{rndnick()}@{rndnick()}.'+random.choice(('com','net','org')) 'mail': settings.ns_mail if settings.ns_mail else f'{rndnick()}@{rndnick()}.'+random.choice(('com','net','org'))
} }
cmds = ('ADMIN', 'CAP LS', 'INFO', 'IRCOPS', 'LINKS', 'MAP', 'MODULES -all', 'STATS p', 'VERSION') cmds = ['ADMIN', 'CAP LS', 'INFO', 'IRCOPS', 'LINKS', 'MAP', 'MODULES -all', 'STATS p', 'VERSION']
random.shuffle(cmds) random.shuffle(cmds)
cmds += ('PRIVMSG NickServ :REGISTER {0} {1}'.format(login['pass'], login['mail']), 'LIST') cmds += ['PRIVMSG NickServ :REGISTER {0} {1}'.format(login['pass'], login['mail']), 'LIST']
for command in cmds: for command in cmds:
try: try:
await self.raw(command) await self.raw(command)
@ -222,12 +223,12 @@ class probe:
await asyncio.sleep(1.5) await asyncio.sleep(1.5)
del login del login
if not self.channels['all']: if not self.channels['all']:
error(self.display + 'no channels found') error(self.display + '\033[31merror\033[0m - no channels found')
await self.raw('QUIT') await self.raw('QUIT')
except asyncio.CancelledError: except asyncio.CancelledError:
pass pass
except Exception as ex: except Exception as ex:
error(self.display + 'error in loop_initial', ex) error(self.display + '\033[31merror\033[0m - loop_initial', ex)
async def loop_channels(self): async def loop_channels(self):
try: try:
@ -250,7 +251,7 @@ class probe:
except asyncio.CancelledError: except asyncio.CancelledError:
pass pass
except Exception as ex: except Exception as ex:
error(self.display + 'error in loop_channels', ex) error(self.display + '\033[31merror\033[0m - loop_channels', ex)
async def loop_nick(self): async def loop_nick(self):
try: try:
@ -261,7 +262,7 @@ class probe:
except asyncio.CancelledError: except asyncio.CancelledError:
pass pass
except Exception as ex: except Exception as ex:
error(self.display + 'error in loop_nick', ex) error(self.display + '\033[31merror\033[0m - loop_nick', ex)
async def loop_whois(self): async def loop_whois(self):
try: try:
@ -269,7 +270,6 @@ class probe:
if self.nicks['check']: if self.nicks['check']:
nick = random.choice(self.nicks['check']) nick = random.choice(self.nicks['check'])
self.nicks['check'].remove(nick) self.nicks['check'].remove(nick)
debug(self.display + 'WHOIS ' + nick)
try: try:
await self.raw('WHOIS ' + nick) await self.raw('WHOIS ' + nick)
except: except:
@ -282,7 +282,7 @@ class probe:
except asyncio.CancelledError: except asyncio.CancelledError:
pass pass
except Exception as ex: except Exception as ex:
error(self.display + 'error in loop_whois', ex) error(self.display + '\033[31merror\033[0m - loop_whois', ex)
async def listen(self): async def listen(self):
while True: while True:
@ -313,7 +313,7 @@ class probe:
chan = args[3] chan = args[3]
if chan in self.channels['users']: if chan in self.channels['users']:
del self.channels['users'][chan] del self.channels['users'][chan]
error(f'{self.display}scanning {chan} failed', line) error(f'{self.display}\033[31merror\033[0m - {chan}', line)
elif line.startswith('ERROR :Closing Link'): elif line.startswith('ERROR :Closing Link'):
raise Exception('DroneBL') if 'dronebl' in line.lower() else Exception('Banned') raise Exception('DroneBL') if 'dronebl' in line.lower() else Exception('Banned')
elif line.startswith('ERROR :Trying to reconnect too fast') or line.startswith('ERROR :Your host is trying to (re)connect too fast') or line.startswith('ERROR :Reconnecting too fast'): elif line.startswith('ERROR :Trying to reconnect too fast') or line.startswith('ERROR :Your host is trying to (re)connect too fast') or line.startswith('ERROR :Reconnecting too fast'):
@ -327,11 +327,14 @@ class probe:
self.snapshot['server'] = self.server self.snapshot['server'] = self.server
self.snapshot['host'] = host self.snapshot['host'] = host
if len(host) > 25: if len(host) > 25:
self.display = f'{self.server.ljust(18)} | {host[:22]}... | ' self.display = f'{self.server.ljust(18)} \033[30m|\033[0m {host[:22]}... \033[30m|\033[0m '
else: else:
self.display = f'{self.server.ljust(18)} | {host.ljust(25)} | ' self.display = f'{self.server.ljust(18)} \033[30m|\033[0m {host.ljust(25)} \033[30m|\033[0m '
debug(self.display + 'connected') debug(self.display + '\033[1;32mconnected\033[0m')
self.loops['init'] = asyncio.create_task(self.loop_initial()) self.loops['init'] = asyncio.create_task(self.loop_initial())
elif numeric == '311' and len(args) >= 4: # RPL_WHOISUSER
nick = args[3]
debug(f'{self.display}\033[34mWHOIS\033[0m {nick}')
elif numeric == '322' and len(args) >= 4: # RPL_LIST elif numeric == '322' and len(args) >= 4: # RPL_LIST
chan = args[3] chan = args[3]
self.channels['all'].append(chan) self.channels['all'].append(chan)
@ -341,7 +344,7 @@ class probe:
elif numeric == '323': # RPL_LISTEND elif numeric == '323': # RPL_LISTEND
if self.channels['all']: if self.channels['all']:
del self.loops['init'] del self.loops['init']
debug(self.display + 'found {0} channel(s)'.format(str(len(self.channels['all'])))) debug(self.display + 'found \033[93m{0}\033[0m channel(s)'.format(str(len(self.channels['all']))))
self.loops['chan'] = asyncio.create_task(self.loop_channels()) self.loops['chan'] = asyncio.create_task(self.loop_channels())
self.loops['nick'] = asyncio.create_task(self.loop_nick()) self.loops['nick'] = asyncio.create_task(self.loop_nick())
self.loops['whois'] = asyncio.create_task(self.loop_whois()) self.loops['whois'] = asyncio.create_task(self.loop_whois())
@ -354,10 +357,10 @@ class probe:
chan = args[3] chan = args[3]
self.channels['current'].append(chan) self.channels['current'].append(chan)
if chan in self.channels['users']: if chan in self.channels['users']:
debug('{0}scanning {1} users in {2}'.format(self.display, self.channels['users'][chan].ljust(4), chan)) debug('{0}\033[32mJOIN\033[0m {1} \033[30m(found \033[93m{2}\033[30m users)\033[0m'.format(self.display, chan, self.channels['users'][chan]))
del self.channels['users'][chan] del self.channels['users'][chan]
else: else:
debug(f'{self.display}scanning users in {chan}') debug(f'{self.display}\033[32mJOIN\033[0m {chan}')
await self.raw('WHO ' + chan) await self.raw('WHO ' + chan)
await asyncio.sleep(throttle.part) await asyncio.sleep(throttle.part)
await self.raw('PART ' + chan) await self.raw('PART ' + chan)
@ -365,7 +368,7 @@ class probe:
elif numeric == '421' and len(args) >= 3: # ERR_UNKNOWNCOMMAND elif numeric == '421' and len(args) >= 3: # ERR_UNKNOWNCOMMAND
msg = ' '.join(args[2:]) msg = ' '.join(args[2:])
if 'You must be connected for' in msg: if 'You must be connected for' in msg:
error(self.display + 'delay found', msg) error(self.display + '\033[31merror\033[0m - delay found', msg)
elif numeric == '433': # ERR_NICKINUSE elif numeric == '433': # ERR_NICKINUSE
if not settings.nickname: if not settings.nickname:
await self.raw('NICK ' + rndnick()) await self.raw('NICK ' + rndnick())
@ -380,11 +383,18 @@ class probe:
if seconds.isdigit(): if seconds.isdigit():
seconds = int(seconds) seconds = int(seconds)
self.jthrottle = throttle.seconds if seconds > throttle.seconds else seconds self.jthrottle = throttle.seconds if seconds > throttle.seconds else seconds
error(self.display + 'delay found', msg) error(self.display + '\033[31merror\033[0m - delay found', msg)
elif numeric == '465': # ERR_YOUREBANNEDCREEP elif numeric == '465': # ERR_YOUREBANNEDCREEP
raise Exception('K-Lined') raise Exception('K-Lined')
elif numeric == '464': # ERR_PASSWDMISMATCH elif numeric == '464': # ERR_PASSWDMISMATCH
raise Exception('Network has a password') raise Exception('Network has a password')
elif numeric == 'KILL':
nick = args[2]
if nick == self.nickname:
raise Exception('KILL')
else:
if 'KILL' in self.snapshot:
del self.snapshot['KILL']
elif numeric in ('NOTICE','PRIVMSG') and len(args) >= 4: elif numeric in ('NOTICE','PRIVMSG') and len(args) >= 4:
nick = args[0].split('!')[1:] nick = args[0].split('!')[1:]
target = args[2] target = args[2]
@ -392,7 +402,7 @@ class probe:
if target == self.nickname: if target == self.nickname:
for i in ('You must have been using this nick for','You must be connected for','not connected long enough','Please wait', 'You cannot list within the first'): for i in ('You must have been using this nick for','You must be connected for','not connected long enough','Please wait', 'You cannot list within the first'):
if i in msg: if i in msg:
error(self.display + 'delay found', msg) error(self.display + '\033[31merror\033[0m - delay found', msg)
break break
if msg[:8] == '\001VERSION': if msg[:8] == '\001VERSION':
version = random.choice(('http://www.mibbit.com ajax IRC Client','mIRC v6.35 Khaled Mardam-Bey','xchat 0.24.1 Linux 2.6.27-8-eeepc i686','rZNC Version 1.0 [02/01/11] - Built from ZNC','thelounge v3.0.0 -- https://thelounge.chat/')) version = random.choice(('http://www.mibbit.com ajax IRC Client','mIRC v6.35 Khaled Mardam-Bey','xchat 0.24.1 Linux 2.6.27-8-eeepc i686','rZNC Version 1.0 [02/01/11] - Built from ZNC','thelounge v3.0.0 -- https://thelounge.chat/'))
@ -408,7 +418,7 @@ class probe:
except (UnicodeDecodeError, UnicodeEncodeError): except (UnicodeDecodeError, UnicodeEncodeError):
pass pass
except Exception as ex: except Exception as ex:
error(self.display + 'fatal error occured', ex) error(self.display + '\033[1;31mdisconnected\033[0m', ex)
break break
async def main(targets): async def main(targets):