X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/c22ee2bac0dbdaad33034ef008e75a3f5fc73c62..6ad6917641c9e48128ec0abfe7d43e37c009efc1:/bot.py?ds=sidebyside diff --git a/bot.py b/bot.py index 61b5216..4b8c3e8 100644 --- a/bot.py +++ b/bot.py @@ -4,17 +4,16 @@ # 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) @@ -37,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() @@ -52,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 @@ -70,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()) @@ -95,11 +100,13 @@ class Bot(object): 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, @@ -129,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) @@ -231,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] @@ -314,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) @@ -335,12 +356,8 @@ class Bot(object): rancmd = True try: cbret = callback(self, user, chan, target, *pieces[1:]) - if cbret is NotImplemented: - raise NotImplementedError - elif isinstance(cbret, stringbase): - self.reply(chan, user, cbret) - 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) @@ -365,51 +382,51 @@ class Bot(object): def reply(self, chan, user, msg): - if chan is not None: + 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.msg(user, msg) - def msg(self, target, msg, truncate=False): + """ + Does the work for msg/slowmsg/fastmsg. Uses the append_callback to append to the correct queue. + + 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: + # 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] + cmd = cmd[:maxlen] + if self.conn.exceeded or self.conn.bytessent+len(cmd) >= self.conn.recvq: - self.msgqueue.append(cmd) + append_callback(cmd) else: 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): - if self.parent.cfg.getboolean('erebus', 'nofakelag'): return self.fastmsg(target, msg) - cmd = self._formatmsg(target, msg) - 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) - else: - self.conn.send(cmd) - self.conn.exceeded = True - return True + return self._msg(target, msg, truncate, self.slowmsgqueue.append) def fastmsg(self, target, msg, truncate=False): - cmd = self._formatmsg(target, msg) - if len(cmd) > MAXLEN: - if not truncate: - return False - else: - cmd = cmd[:MAXLEN] - self.conn.send(cmd) - self.conn.exceeded = True - return True + return self._msg(target, msg, truncate, self.conn.send) def _formatmsg(self, target, msg): if target is None or msg is None: @@ -423,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)) @@ -463,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 @@ -473,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 @@ -493,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")