X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/4477123de1254afa80243f7f9eab62c5cbf7b786..6ad6917641c9e48128ec0abfe7d43e37c009efc1:/bot.py?ds=sidebyside diff --git a/bot.py b/bot.py index 610c6cd..4b8c3e8 100644 --- a/bot.py +++ b/bot.py @@ -4,20 +4,25 @@ # Erebus IRC bot - Author: John Runyon # "Bot" and "BotConnection" classes (handling a specific "arm") -import socket, sys, time, threading, os, random +import socket, sys, time, threading, os, random, struct from collections import deque -MAXLEN = 400 # arbitrary max length of a command generated by Bot.msg functions - if sys.version_info.major < 3: timerbase = threading._Timer + stringbase = basestring else: timerbase = threading.Timer + stringbase = str + class MyTimer(timerbase): def __init__(self, *args, **kwargs): timerbase.__init__(self, *args, **kwargs) self.daemon = True +if sys.version_info.major < 3: + stringbase = basestring +else: + stringbase = str #bots = {'erebus': bot.Bot(nick='Erebus', user='erebus', bind='', server='irc.quakenet.org', port=6667, realname='Erebus')} class Bot(object): @@ -31,7 +36,12 @@ class Bot(object): self.authname = authname self.authpass = authpass - curs = self.parent.query("SELECT chname FROM chans WHERE bot = %s AND active = 1", (self.nick,)) + self.connecttime = 0 # time at which we received numeric 001 + self.server = server # the address we try to (re-)connect to + self.port = port + self.servername = server # the name of the server we got connected to + + curs = self.parent.query("SELECT chname FROM chans WHERE bot = %s AND active = 1", (self.permnick,)) if curs: chansres = curs.fetchall() curs.close() @@ -46,12 +56,12 @@ class Bot(object): self.msgqueue = deque() self.slowmsgqueue = deque() - self.makemsgtimer() - self.msgtimer.start() + self._makemsgtimer() + self._msgtimer.start() def __del__(self): try: - curs = self.parent.query("UPDATE bots SET connected = 0 WHERE nick = %s", (self.nick,)) + curs = self.parent.query("UPDATE bots SET connected = 0 WHERE nick = %s", (self.permnick,)) curs.close() except: pass @@ -64,6 +74,7 @@ class Bot(object): self.parent.log(self.nick, *args, **kwargs) def connect(self): + self.log('!', "Connecting") if self.conn.connect(): self.parent.newfd(self, self.conn.socket.fileno()) @@ -87,12 +98,15 @@ class Bot(object): 'ERROR': self._goterror, } one = { #things to look for after source + 'NOTICE': self._gotconnected, '001': self._got001, + '004': self._got004, '376': self._gotRegistered, '422': self._gotRegistered, 'PRIVMSG': self._gotprivmsg, '353': self._got353, #NAMES '354': self._got354, #WHO + '396': self._gotHiddenHost, # hidden host has been set '433': self._got433, #nick in use 'JOIN': self._gotjoin, 'PART': self._gotpart, @@ -122,26 +136,37 @@ class Bot(object): self.conn.send("PONG %s" % (pieces[1])) self._checknick() def _goterror(self, pieces): + # TODO: better handling, just reconnect that single bot try: self.quit("Error detected: %s" % ' '.join(pieces)) + except: pass + try: curs = self.parent.query("UPDATE bots SET connected = 0") curs.close() except: pass sys.exit(2) os._exit(2) def _got001(self, pieces): - pass # wait until the end of MOTD instead + # We wait until the end of MOTD instead to consider ourselves registered, but consider uptime as of 001 + self.connecttime = time.time() + def _got004(self, pieces): + self.servername = pieces[3] def _gotRegistered(self, pieces): self.conn.registered(True) - curs = self.parent.query("UPDATE bots SET connected = 1 WHERE nick = %s", (self.nick,)) + curs = self.parent.query("UPDATE bots SET connected = 1 WHERE nick = %s", (self.permnick,)) if curs: curs.close() self.conn.send("MODE %s +x" % (pieces[2])) if self.authname is not None and self.authpass is not None: self.conn.send("AUTH %s %s" % (self.authname, self.authpass)) - for c in self.chans: - self.join(c.name) + if not self.parent.cfg.getboolean('erebus', 'wait_for_hidden_host'): + for c in self.chans: + self.join(c.name) + def _gotHiddenHost(self, pieces): + if self.parent.cfg.getboolean('erebus', 'wait_for_hidden_host'): + for c in self.chans: + self.join(c.name) def _gotprivmsg(self, pieces): nick = pieces[0].split('!')[0][1:] user = self.parent.user(nick) @@ -224,10 +249,13 @@ class Bot(object): def _gotnick(self, pieces): oldnick = pieces[0].split('!')[0][1:] newnick = pieces[2][1:] - if newnick.lower() != oldnick.lower(): - self.parent.users[newnick.lower()] = self.parent.users[oldnick.lower()] - del self.parent.users[oldnick.lower()] - self.parent.users[newnick.lower()].nickchange(newnick) + if oldnick == self.nick: + self.nick = newnick + else: + if newnick.lower() != oldnick.lower(): + self.parent.users[newnick.lower()] = self.parent.users[oldnick.lower()] + del self.parent.users[oldnick.lower()] + self.parent.users[newnick.lower()].nickchange(newnick) def _gotmode(self, pieces): source = pieces[0].split('!')[0][1:] chan = pieces[2] @@ -281,19 +309,20 @@ class Bot(object): if triggerused: msg = msg[len(self.parent.trigger):] pieces = msg.split() + if len(pieces) == 0: + return + if target != self.nick: # message was sent to a channel try: - if msg.startswith('*'): # message may be addressed to bot by "*BOTNICK" trigger? - if pieces[0][1:].lower() == self.nick.lower(): - pieces.pop(0) # command actually starts with next word - msg = ' '.join(pieces) # command actually starts with next word - triggerused = True + if pieces[0][:-1].lower() == self.nick.lower() and (pieces[0][-1] == ":" or pieces[0][-1] == ","): + pieces.pop(0) # command actually starts with next word + if len(pieces) == 0: # is there still anything left? + return + msg = ' '.join(pieces) + triggerused = True except IndexError: return # "message" is empty - if len(pieces) == 0: - return - if len(pieces) > 1: chanword = pieces[1] if chanword.startswith('#'): @@ -306,8 +335,8 @@ class Bot(object): for callback in self.parent.getchanhook(target.lower()): try: cbret = callback(self, user, chan, *pieces) - except NotImplementedError: - self.msg(user, "Command not implemented.") + if isinstance(cbret, stringbase): + self.reply(target, user, cbret) except: self.msg(user, "Command failed. Code: CBEXC%09.3f" % (time.time() % 100000)) self.__debug_cbexception("chanhook", user=user, target=target, msg=msg) @@ -327,10 +356,8 @@ class Bot(object): rancmd = True try: cbret = callback(self, user, chan, target, *pieces[1:]) - if cbret is NotImplemented: - raise NotImplementedError - except NotImplementedError: - self.msg(user, "Command not implemented.") + if isinstance(cbret, stringbase): + self.reply(target, user, cbret) except Exception: self.msg(user, "Command failed. Code: CBEXC%09.3f" % (time.time() % 100000)) self.__debug_cbexception("hook", user=user, target=target, msg=msg) @@ -353,35 +380,54 @@ class Bot(object): # print "%09.3f %s [!] %s" % (time.time() % 100000, self.nick, "!!! NOMSG") __import__('traceback').print_stack() - def msg(self, target, msg): - if self.parent.cfg.getboolean('erebus', 'nofakelag'): return self.fastmsg(target, msg) - cmd = self._formatmsg(target, msg) - if len(cmd) > MAXLEN: return False - if self.conn.exceeded or self.conn.bytessent+len(cmd) >= self.conn.recvq: - self.msgqueue.append(cmd) + + def reply(self, chan, user, msg): + if chan is not None and (isinstance(chan, self.parent.Channel) or (isinstance(chan, stringbase) and chan[0] == "#")): + self.msg(chan, "%s: %s" % (user, msg)) else: - self.conn.send(cmd) - self.conn.exceeded = True - return True + self.msg(user, msg) + + """ + Does the work for msg/slowmsg/fastmsg. Uses the append_callback to append to the correct queue. - def slowmsg(self, target, msg): + In the case of fastmsg, self.conn.exceeded may be True, however, in this case append_callback=self.conn.send, so it will still be sent immediately. + """ + def _msg(self, target, msg, truncate, append_callback): if self.parent.cfg.getboolean('erebus', 'nofakelag'): return self.fastmsg(target, msg) + cmd = self._formatmsg(target, msg) - if len(cmd) > MAXLEN: return False + # The max length is much shorter than recvq (510) because of the length the server adds on about the source (us). + # If you know your hostmask, you can of course figure the exact length, but it's very difficult to reliably know your hostmask. + maxlen = ( + self.conn.recvq + - 63 # max hostname len + - 11 # max ident len + - 3 # the symbols in :nick!user@host + - len(self.nick) + ) + if len(cmd) > maxlen: + if not truncate: + return False + else: + cmd = cmd[:maxlen] + if self.conn.exceeded or self.conn.bytessent+len(cmd) >= self.conn.recvq: - self.slowmsgqueue.append(cmd) + append_callback(cmd) else: self.conn.send(cmd) - self.conn.exceeded = True - return True - def fastmsg(self, target, msg): - cmd = self._formatmsg(target, msg) - if len(cmd) > MAXLEN: return False - self.conn.send(cmd) self.conn.exceeded = True return True + def msg(self, target, msg, truncate=False): + return self._msg(target, msg, truncate, self.msgqueue.append) + + def slowmsg(self, target, msg, truncate=False): + return self._msg(target, msg, truncate, self.slowmsgqueue.append) + + def fastmsg(self, target, msg, truncate=False): + return self._msg(target, msg, truncate, self.conn.send) + def _formatmsg(self, target, msg): if target is None or msg is None: return self.__debug_nomsg(target, msg) @@ -394,30 +440,34 @@ class Bot(object): return command def _popmsg(self): - self.makemsgtimer() + self._makemsgtimer() self.conn.bytessent -= self.conn.recvq/3 if self.conn.bytessent < 0: self.conn.bytessent = 0 - self.conn.exceeded = False + self.conn.exceeded = True + cmd = None try: cmd = self.msgqueue.popleft() - if not self.conn.exceeded and self.conn.bytessent+len(cmd) < self.conn.recvq: - self.conn.send(cmd) - self.conn.exceeded = True - else: raise IndexError except IndexError: try: cmd = self.slowmsgqueue.popleft() - if not self.conn.exceeded and self.conn.bytessent+len(cmd) < self.conn.recvq: - self.conn.send(cmd) - self.conn.exceeded = True except IndexError: pass - self.msgtimer.start() - def makemsgtimer(self): - self.msgtimer = threading.Timer(3, self._popmsg) - self.msgtimer.daemon = True + if cmd is not None: + if self.conn.bytessent+len(cmd) > self.conn.recvq: # If it's too long + if len(cmd) > self.conn.recvq: # Is the command itself somehow over max length??? + self._msgtimer.start() + raise ValueError('Somehow a command that was too long made it into the message queue. Uhoh!', cmd) + # Discard the message. + self.msgqueue.appendleft(cmd) # Phew, we've just sent too much recently. Put it (back) on the (primary) queue. + else: + self.conn.send(cmd) + + self._msgtimer.start() + + def _makemsgtimer(self): + self._msgtimer = MyTimer(3, self._popmsg) def join(self, chan): self.conn.send("JOIN %s" % (chan)) @@ -434,7 +484,7 @@ class Bot(object): class BotConnection(object): def __init__(self, parent, bind, server, port): self.parent = parent - self.buffer = bytearray(8192) + self.buffer = bytearray() self.socket = None self.bind = bind @@ -444,11 +494,21 @@ class BotConnection(object): self.state = 0 # 0=disconnected, 1=registering, 2=connected self.bytessent = 0 - self.recvq = 500 + self.recvq = 510 self.exceeded = False + self._nowrite = False def connect(self): - self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + if self.parent.parent.cfg.getboolean('erebus', 'tls'): + import ssl + undersocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + context = ssl.create_default_context() + self.socket = context.wrap_socket(undersocket, server_hostname=self.server) + else: + self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + self.socket.setsockopt(socket.SOL_TCP, socket.TCP_NODELAY, 1) # Does Python make SOL_TCP portable? Who knows, it's not documented, and it appears to come from the _socket C lib. + self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_LINGER, struct.pack('ii', 0, 0)) + self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1) self.socket.bind((self.bind, 0)) self.socket.connect((self.server, self.port)) return True @@ -464,10 +524,18 @@ class BotConnection(object): return self.state == 2 def send(self, line): - if self.parent.parent.cfg.getboolean('debug', 'io'): - self.parent.log('O', line) - self.bytessent += len(line) - self._write(line) + if not self._nowrite: + if self.parent.parent.cfg.getboolean('debug', 'io'): + self.parent.log('O', line) + self.bytessent += len(line) + try: + self._write(line) + except socket.error as e: + self._nowrite = True + self.parent._goterror(repr(e)) + else: + if self.parent.parent.cfg.getboolean('debug', 'io'): + self.parent.log('X', line) def _write(self, line): self.socket.sendall(line.encode('utf-8', 'backslashreplace')+b"\r\n")