# Erebus IRC bot - Author: John Runyon
# "Bot" and "BotConnection" classes (handling a specific "arm")
-import socket, sys, time, threading, os, random, struct
+import os, random, socket, struct, sys, threading, time, traceback
from collections import deque
if sys.version_info.major < 3:
def watchdog(self):
if time.time() > int(self.parent.cfg.get('watchdog', 'maxtime', default=300))+self.lastreceived:
self.parse("ERROR :Fake-error from watchdog timer.")
- self.watchdogtimer = MyTimer(int(self.parent.cfg.get('watchdog', 'interval', default=30)), self.watchdog)
+ if self.conn.registered():
+ self.conn.send("PING :%s" % (time.time()))
+ self._checknick()
+ watchdogtimer = MyTimer(int(self.parent.cfg.get('watchdog', 'interval', default=30)), self.watchdog)
+ watchdogtimer.start()
def log(self, *args, **kwargs):
self.parent.log(self.nick, *args, **kwargs)
def connect(self):
self.log('!', "Connecting")
if self.conn.connect():
+ self.log('!', "Connected")
self.parent.newfd(self, self.conn.socket.fileno())
def getdata(self):
- self.lastreceived = time.time()
- return self.conn.read()
+ try:
+ recvd = self.conn.read()
+ self.lastreceived = time.time()
+ return recvd
+ except EOFError as e:
+ return [":%s ERROR :%s%r" % (self.nick, e.__class__.__name__, e.args)]
def _checknick(self): # check if we're using the right nick, try changing
if self.nick != self.permnick and self.conn.registered():
numeric = pieces[0]
# dispatch dict
- dispatch = { #things to look for after source
+ dispatch = {
'NOTICE': self._gotconnected,
'001': self._got001,
'004': self._got004,
try:
callback(self, line)
except Exception:
- self.__debug_cbexception("numhook", line)
+ self._cbexception("numhook", line)
if numeric in dispatch:
dispatch[numeric](pieces)
self._checknick()
def _goterror(self, pieces):
# TODO: better handling, just reconnect that single bot
+ error = ' '.join(pieces)
+ try:
+ raise Exception(error)
+ except Exception as e:
+ self.parent.mustquit = e
try:
- self.quit("Error detected: %s" % ' '.join(pieces))
+ self.quit("Error detected: %s" % (error))
except: pass
try:
curs = self.parent.query("UPDATE bots SET connected = 0")
curs.close()
except: pass
- os._exit(2) # can't use sys.exit since we might be in a sub-thread
+ self.log('!', 'Bot exiting due to: %s' % (error))
def _got001(self, pieces):
# We wait until the end of MOTD instead to consider ourselves registered, but consider uptime as of 001
self.connecttime = time.time()
else:
pass # don't care about other modes
- def __debug_cbexception(self, source, *args, **kwargs):
+ def _cbexception(self, source, *args, chained=False, **kwargs):
+ if not chained: # skip hooks if we were caused by a hook
+ exc = sys.exception()
+ if self.parent.hasexceptionhook(exc):
+ for callback in self.parent.getexceptionhook(exc):
+ try:
+ callback(self, exc, source, *args, **kwargs)
+ except Exception:
+ self._cbexception('exceptionhook', chained=True, module=callback.__module__, function=callback.__name__, underlying=(source, args, kwargs))
if self.parent.cfg.getboolean('debug', 'cbexc'):
self.conn.send("PRIVMSG %s :%09.3f \ 34\1f!!! CBEXC\1f\ 3 %s" % (self.parent.cfg.get('debug', 'owner'), time.time() % 100000, source))
- __import__('traceback').print_exc()
+ traceback.print_exc(chain=not chained)
self.log('!', "CBEXC %s %r %r" % (source, args, kwargs))
-# print "%09.3f %s [!] CBEXC %s %r %r" % (time.time() % 100000, self.nick, source, args, kwargs)
def parsemsg(self, user, target, msg):
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)
+ self._cbexception("chanhook", user=user, target=target, msg=msg)
return # not to bot, don't process!
cmd = pieces[0].lower()
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)
+ self._cbexception("hook", user=user, target=target, msg=msg)
except SystemExit as e:
+ self.parent.mustquit = e
try:
curs = self.parent.query("UPDATE bots SET connected = 0")
curs.close()
self.conn.send("PRIVMSG %s :%09.3f \ 34\1f!!! NOMSG\1f\ 3 %r, %r" % (self.parent.cfg.get('debug', 'owner'), time.time() % 100000, target, msg))
self.log('!', "!!! NOMSG")
# print "%09.3f %s [!] %s" % (time.time() % 100000, self.nick, "!!! NOMSG")
- __import__('traceback').print_stack()
+ traceback.print_stack()
def reply(self, chan, user, msg):
def _write(self, line):
self.socket.sendall(line.encode('utf-8', 'backslashreplace')+b"\r\n")
+ def _getsockerr(self):
+ try: # SO_ERROR might not exist on all platforms
+ return self.socket.getsockopt(socket.SOL_SOCKET, socket.SO_ERROR)
+ except:
+ return None
+
def read(self):
- self.buffer += self.socket.recv(8192)
+ recvd = self.socket.recv(8192)
+ if recvd == b"":
+ raise EOFError("socket.recv returned empty", self.parent.nick, self._getsockerr())
+ self.buffer += recvd
lines = []
while b"\r\n" in self.buffer: