from twisted.internet.protocol import ReconnectingClientFactory from twisted.words.protocols.irc import IRCClient from twisted.internet.defer import Deferred from twisted.internet.task import LoopingCall, deferLater from twisted.internet import reactor from string import digits from random import randint from copy import deepcopy from modules import userinfo from modules import counters from modules import monitor from core.relay import sendRelayNotification from utils.dedup import dedup import main from utils.logging.log import * from utils.logging.send import * from twisted.internet.ssl import DefaultOpenSSLContextFactory def deliverRelayCommands(relay, relayCommands, user=None, stage2=None): # where relay is a dictionary extracted from the Network object keyFN = main.certPath+main.config["Key"] certFN = main.certPath+main.config["Certificate"] contextFactory = DefaultOpenSSLContextFactory(keyFN.encode("utf-8", "replace"), certFN.encode("utf-8", "replace")) bot = IRCBotFactory(None, relay, relayCommands, user, stage2) rct = reactor.connectSSL(relay["host"], int(relay["port"]), bot, contextFactory) class IRCRelay(IRCClient): def __init__(self, relay, relayCommands, user, stage2): self.connected = False self.buffer = "" if user == None: self.user = relay["user"] else: self.user = user password = relay["password"] self.nickname = self.user self.realname = self.user self.username = self.user self.password = self.user+":"+password self.relayCommands = relayCommands self.relay = relay self.stage2 = stage2 def parsen(self, user): step = user.split("!") nick = step[0] if len(step) == 2: step2 = step[1].split("@") ident, host = step2 else: ident = nick host = nick return [nick, ident, host] def privmsg(self, user, channel, msg): nick, ident, host = self.parsen(user) if nick[0] == main.config["Tweaks"]["ZNC"]["Prefix"]: nick = nick[1:] if nick in self.relayCommands.keys(): sendAll("[%s] %s -> %s" % (self.relay, nick, msg)) def irc_ERR_PASSWDMISMATCH(self, prefix, params): log("%s: relay password mismatch" % self.relay) sendAll("%s: relay password mismatch" % self.relay) def signedOn(self): self.connected = True log("signed on as a relay: %s" % self.relay) #sendRelayNotification("Relay", {"type": "conn", "status": "connected"}) nobody actually cares for i in self.relayCommands.keys(): for x in self.relayCommands[i]: self.msg(main.config["Tweaks"]["ZNC"]["Prefix"]+i, x) if not self.stage2 == None: # [["user", {"sasl": ["message1", "message2"]}], []] if not len(self.stage2) == 0: user = self.stage2[0].pop(0) commands = self.stage2[0].pop(0) del self.stage2[0] deliverRelayCommands(self.relay, commands, user, self.stage2) deferLater(reactor, 1, self.transport.loseConnection) return class IRCBot(IRCClient): def __init__(self, name, relay): self.connected = False self.channels = [] self.net = "".join([x for x in name if not x in digits]) if self.net == "": error("Network with all numbers: %s" % name) self.buffer = "" self.name = name alias = relay["alias"] self.nickname = alias["nick"] self.realname = alias["realname"] self.username = alias["nick"]+"/"+relay["net"] self.password = relay["password"] self.userinfo = None self.fingerReply = None self.versionName = None self.versionNum = None self.versionEnv = None self.sourceURL = None self._who = {} self._getWho = {} self._names = {} def parsen(self, user): step = user.split("!") nick = step[0] if len(step) == 2: step2 = step[1].split("@") if len(step2) == 2: ident, host = step2 else: ident = nick host = nick else: ident = nick host = nick return [nick, ident, host] def event(self, **cast): for i in list(cast.keys()): # Make a copy of the .keys() as Python 3 cannot handle iterating over if cast[i] == "": # a dictionary that changes length with each iteration del cast[i] if "muser" in cast.keys(): cast["nick"], cast["ident"], cast["host"] = self.parsen(cast["muser"]) #if not cast["type"] in ["nick", "kick", "quit", "part", "join"]: # del cast["muser"] if set(["nick", "ident", "host", "message"]).issubset(set(cast)): if "message" in cast.keys(): if cast["ident"] == "znc" and cast["host"] == "znc.in": cast["type"] = "znc" cast["name"] = self.name del cast["nick"] del cast["ident"] del cast["host"] del cast["target"] if not cast["type"] in ["query", "self", "highlight", "znc", "who"]: if "target" in cast.keys() and not cast["type"] == "mode": # don't handle modes here if cast["target"].lower() == self.nickname.lower(): # as they are target == nickname #castDup = deepcopy(cast) # however modes are not queries! cast["mtype"] = cast["type"] cast["type"] = "query" cast["name"] = self.name #self.event(**castDup) # Don't call self.event for this one because queries are not events on a # channel, but we still want to see them if "user" in cast.keys(): if cast["user"].lower() == self.nickname.lower(): castDup = deepcopy(cast) castDup["mtype"] = cast["type"] castDup["type"] = "self" castDup["name"] = self.name self.event(**castDup) if "nick" in cast.keys(): if cast["nick"].lower() == self.nickname.lower(): castDup = deepcopy(cast) castDup["mtype"] = cast["type"] castDup["type"] = "self" castDup["name"] = self.name if not cast["target"].lower() == self.nickname.lower(): # modes has been set on us directly self.event(**castDup) # don't tell anyone else if "message" in cast.keys() and not cast["type"] == "query": # Don't highlight queries if not cast["message"] == None: if self.nickname.lower() in cast["message"].lower(): castDup = deepcopy(cast) castDup["mtype"] = cast["type"] castDup["type"] = "highlight" castDup["name"] = self.name self.event(**castDup) if not "name" in cast.keys(): cast["name"] = self.net counters.event(self.net, cast["type"]) monitor.event(self.name, cast) def privmsg(self, user, channel, msg): self.event(type="msg", muser=user, target=channel, message=msg) def noticed(self, user, channel, msg): self.event(type="notice", muser=user, target=channel, message=msg) def action(self, user, channel, msg): self.event(type="action", muser=user, target=channel, message=msg) def get(self, var): try: result = getattr(self, var) except AttributeError: result = None return result def setNick(self, nickname): self._attemptedNick = nickname self.sendLine("NICK %s" % nickname) self.nickname = nickname def alterCollidedNick(self, nickname): newnick = nickname + "_" return newnick def nickChanged(self, olduser, newnick): self.nickname = newnick self.event(type="self", mtype="nick", muser=olduser, user=newnick) def irc_ERR_NICKNAMEINUSE(self, prefix, params): self._attemptedNick = self.alterCollidedNick(self._attemptedNick) self.setNick(self._attemptedNick) def irc_ERR_PASSWDMISMATCH(self, prefix, params): log("%s: password mismatch" % self.name) sendAll("%s: password mismatch" % self.name) def who(self, channel): d = Deferred() if channel not in self._who: self._who[channel] = ([], []) self._who[channel][0].append(d) self.sendLine("WHO %s" % channel) return d def irc_RPL_WHOREPLY(self, prefix, params): channel = params[1] ident = params[2] host = params[3] server = params[4] nick = params[5] status = params[6] realname = params[7] if channel not in self._who: return n = self._who[channel][1] n.append([nick, nick, host, server, status, realname]) self.event(type="who", nick=nick, ident=ident, host=host, realname=realname, target=channel, server=server, status=status) def irc_RPL_ENDOFWHO(self, prefix, params): channel = params[1] if channel not in self._who: return callbacks, info = self._who[channel] for cb in callbacks: cb.callback((channel, info)) del self._who[channel] def got_who(self, whoinfo): userinfo.initialUsers(self.net, whoinfo[0], whoinfo[1]) def sanit(self, data): if len(data) >= 1: if data[0] in ["!", "~", "&", "@", "%", "+"]: data = data[1:] return data return data else: return False def names(self, channel): d = Deferred() if channel not in self._names: self._names[channel] = ([], []) self._names[channel][0].append(d) self.sendLine("NAMES %s" % channel) return d def irc_RPL_NAMREPLY(self, prefix, params): channel = params[2] nicklist = params[3].split(' ') if channel not in self._names: return n = self._names[channel][1] n.append(nicklist) def irc_RPL_ENDOFNAMES(self, prefix, params): channel = params[1] if channel not in self._names: return callbacks, namelist = self._names[channel] for cb in callbacks: cb.callback((channel, namelist)) del self._names[channel] def got_names(self, nicklist): newNicklist = [] for i in nicklist[1]: for x in i: f = self.sanit(x) # need to store this as well, or potentially just do not remove it... if f: newNicklist.append(f) userinfo.initialNames(self.net, nicklist[0], newNicklist) #twisted sucks so i have to do this to actually get the user info def irc_JOIN(self, prefix, params): """ Called when a user joins a channel. """ nick = prefix.split('!')[0] channel = params[-1] if nick == self.nickname: self.joined(channel) else: self.userJoined(prefix, channel) def irc_PART(self, prefix, params): """ Called when a user leaves a channel. """ nick = prefix.split('!')[0] channel = params[0] if len(params) >= 2: message = params[1] else: message = None if nick == self.nickname: self.left(prefix, channel, message) else: self.userLeft(prefix, channel, message) def irc_QUIT(self, prefix, params): """ Called when a user has quit. """ nick = prefix.split('!')[0] #if nick == self.nickname: #self.botQuit(prefix, params[0]) #else: self.userQuit(prefix, params[0]) def irc_NICK(self, prefix, params): """ Called when a user changes their nickname. """ nick = prefix.split('!', 1)[0] if nick == self.nickname: self.nickChanged(prefix, params[0]) else: self.userRenamed(prefix, params[0]) def irc_KICK(self, prefix, params): """ Called when a user is kicked from a channel. """ #kicker = prefix.split('!')[0] channel = params[0] kicked = params[1] message = params[-1] # Checks on whether it was us that was kicked are done in userKicked self.userKicked(kicked, channel, prefix, message) def irc_TOPIC(self, prefix, params): """ Someone in the channel set the topic. """ #user = prefix.split('!')[0] channel = params[0] newtopic = params[1] self.topicUpdated(prefix, channel, newtopic) #END hacks def signedOn(self): self.connected = True log("signed on: %s" % self.name) self.event(type="conn", status="connected") def joined(self, channel): if not channel in self.channels: self.channels.append(channel) self.names(channel).addCallback(self.got_names) if main.config["Toggles"]["Who"]: self.who(channel).addCallback(self.got_who) lc = LoopingCall(self.who, channel) self._getWho[channel] = lc intrange = main.config["Tweaks"]["Delays"]["WhoRange"] minint = main.config["Tweaks"]["Delays"]["WhoLoop"] interval = randint(minint, minint+intrange) lc.start(interval) def botLeft(self, channel): if channel in self.channels: self.channels.remove(channel) if channel in self._getWho.keys(): lc = self._getWho[channel] lc.stop() del self._getWho[channel] userinfo.delChannel(self.net, channel) # < we do not need to deduplicate this #log("Can no longer cover %s, removing records" % channel)# as it will only be matched once -- # other bots have different nicknames so def left(self, user, channel, message): # even if they saw it, they wouldn't react self.event(type="part", muser=user, target=channel, message=message) self.botLeft(channel) def userJoined(self, user, channel): self.event(type="join", muser=user, target=channel) def userLeft(self, user, channel, message): self.event(type="part", muser=user, target=channel, message=message) def userQuit(self, user, quitMessage): self.chanlessEvent({"type": "quit", "muser": user, "message": quitMessage}) def userKicked(self, kickee, channel, kicker, message): if kickee.lower() == self.nickname.lower(): self.botLeft(channel) self.event(type="kick", muser=kicker, target=channel, message=message, user=kickee) def chanlessEvent(self, cast): cast["nick"], cast["ident"], cast["host"] = self.parsen(cast["muser"]) if dedup(self.name, cast): return # stop right there sir! chans = userinfo.getChanList(self.net, cast["nick"]) if chans == None: error("No channels returned for chanless event: %s" % cast) # self.event(**cast) -- no, should NEVER happen return # getChansSingle returns all channels of the user, we only want to use # ones we have common with them realChans = set(chans).intersection(set(self.channels)) for i in realChans: cast["target"] = i self.event(**cast) def userRenamed(self, oldname, newname): self.chanlessEvent({"type": "nick", "muser": oldname, "user": newname}) def topicUpdated(self, user, channel, newTopic): self.event(type="topic", muser=user, target=channel, message= newTopic) def modeChanged(self, user, channel, toset, modes, args): argList = list(args) modeList = [i for i in modes] for a, m in zip(argList, modeList): self.event(type="mode", muser=user, target=channel, modes=m, status=toset, modeargs=a) class IRCBotFactory(ReconnectingClientFactory): def __init__(self, name, relay=None, relayCommands=None, user=None, stage2=None): if not name == None: self.name = name self.net = "".join([x for x in self.name if not x in digits]) else: self.name = "Relay to "+relay["net"]+relay["id"] self.client = None self.maxDelay = main.config["Tweaks"]["Delays"]["MaxDelay"] self.initialDelay = main.config["Tweaks"]["Delays"]["InitialDelay"] self.factor = main.config["Tweaks"]["Delays"]["Factor"] self.jitter = main.config["Tweaks"]["Delays"]["Jitter"] self.relay, self.relayCommands, self.user, self.stage2 = relay, relayCommands, user, stage2 def buildProtocol(self, addr): if self.relay == None: entry = IRCBot(self.name, self.relay) main.IRCPool[self.name] = entry else: entry = IRCRelay(self.relay, self.relayCommands, self.user, self.stage2) self.client = entry return entry def clientConnectionLost(self, connector, reason): if not self.relay: userinfo.delNetwork(self.net, self.client.channels) if not self.client == None: self.client.connected = False self.client.channels = [] error = reason.getErrorMessage() log("%s: connection lost: %s" % (self.name, error)) if not self.relay: sendAll("%s: connection lost: %s" % (self.name, error)) sendRelayNotification({"type": "conn", "name": self.name, "status": "lost", "message": error}) self.retry(connector) #ReconnectingClientFactory.clientConnectionLost(self, connector, reason) def clientConnectionFailed(self, connector, reason): if not self.client == None: self.client.connected = False self.client.channels = [] error = reason.getErrorMessage() log("%s: connection failed: %s" % (self.name, error)) if not self.relay: sendAll("%s: connection failed: %s" % (self.name, error)) sendRelayNotification({"type": "conn", "name": self.name, "status": "failed", "message": error}) self.retry(connector) #ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)