505 lines
18 KiB
Python
505 lines
18 KiB
Python
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 as count
|
|
from modules import monitor
|
|
|
|
from core.relay import sendRelayNotification
|
|
|
|
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):
|
|
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(main.relay[relay]["host"],
|
|
int(main.relay[relay]["port"]),
|
|
bot, contextFactory)
|
|
|
|
class IRCRelay(IRCClient):
|
|
def __init__(self, relay, relayCommands, user, stage2):
|
|
self.connected = False
|
|
self.buffer = ""
|
|
if user == None:
|
|
self.user = main.relay[relay]["user"]
|
|
else:
|
|
self.user = user
|
|
password = main.relay[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):
|
|
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
|
|
inst = main.pool[name]
|
|
alias = main.alias[inst["alias"]]
|
|
relay = main.relay[inst["relay"]]
|
|
network = main.network[inst["network"]]
|
|
|
|
self.nickname = alias["nick"]
|
|
self.realname = alias["realname"]
|
|
self.username = inst["alias"]+"/"+inst["network"]
|
|
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"])
|
|
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():
|
|
if cast["target"] == self.nickname:
|
|
#castDup = deepcopy(cast)
|
|
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"] == self.nickname:
|
|
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"] == self.nickname:
|
|
castDup = deepcopy(cast)
|
|
castDup["mtype"] = cast["type"]
|
|
castDup["type"] = "self"
|
|
castDup["name"] = self.name
|
|
self.event(**castDup)
|
|
if "message" in cast.keys() and not cast["type"] == "query": # Don't highlight queries
|
|
if self.nickname in cast["message"]:
|
|
castDup = deepcopy(cast)
|
|
castDup["mtype"] = cast["type"]
|
|
castDup["type"] = "highlight"
|
|
castDup["name"] = self.name
|
|
self.event(**castDup)
|
|
|
|
if "name" not in cast.keys():
|
|
cast["name"] = self.net
|
|
if cast["type"] in ["msg", "notice", "action"]:
|
|
userinfo.editUser(self.net, cast["nick"]+"!"+cast["ident"]+"@"+cast["host"])
|
|
count.event(self.net, cast["type"])
|
|
monitor.event(self.net, 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):
|
|
oldnick, ident, host = self.parsen(olduser)
|
|
userinfo.renameUser(self.net, oldnick, olduser, newnick, newnick+"!"+ident+"@"+host)
|
|
self.nickname = newnick
|
|
self.event(type="self", mtype="nick", nick=oldnick, ident=ident, host=host, 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)
|
|
self.who(channel).addCallback(self.got_who)
|
|
if main.config["Toggles"]["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)
|
|
log("Can no longer cover %s, removing records" % channel)
|
|
|
|
def left(self, user, channel, message):
|
|
self.event(type="part", muser=user, target=channel, message=message)
|
|
self.botLeft(channel)
|
|
|
|
def userJoined(self, user, channel):
|
|
nick, ident, host = self.parsen(user)
|
|
userinfo.addUser(self.net, channel, nick, user)
|
|
self.event(type="join", nick=nick, ident=ident, host=host, target=channel)
|
|
|
|
def userLeft(self, user, channel, message):
|
|
nick, ident, host = self.parsen(user)
|
|
userinfo.delUser(self.net, channel, nick, user)
|
|
self.event(type="part", nick=nick, ident=ident, host=host, target=channel, message=message)
|
|
|
|
def userQuit(self, user, quitMessage):
|
|
nick, ident, host = self.parsen(user)
|
|
userinfo.delUserByNetwork(self.net, nick, user)
|
|
self.event(type="quit", nick=nick, ident=ident, host=host, message=quitMessage)
|
|
|
|
def botQuit(self, user, quitMessage):
|
|
nick, ident, host = self.parsen(user)
|
|
userinfo.delUserByNetwork(self.net, nick, user)
|
|
self.event(type="quit", nick=nick, ident=ident, host=host, message=quitMessage)
|
|
|
|
def userKicked(self, kickee, channel, kicker, message):
|
|
nick, ident, host = self.parsen(kicker)
|
|
userinfo.editUser(self.net, kicker)
|
|
userinfo.delUserByNick(self.net, channel, kickee)
|
|
if kickee.lower == self.nickname.lower:
|
|
self.botLeft(channel)
|
|
self.event(type="kick", nick=nick, ident=ident, host=host, target=channel, message=message, user=kickee)
|
|
|
|
def userRenamed(self, oldname, newname):
|
|
nick, ident, host = self.parsen(oldname)
|
|
userinfo.renameUser(self.net, nick, oldname, newname, newname+"!"+ident+"@"+host)
|
|
self.event(type="nick", nick=nick, ident=ident, host=host, user=newname)
|
|
|
|
def topicUpdated(self, user, channel, newTopic):
|
|
nick, ident, host = self.parsen(user)
|
|
userinfo.editUser(self.net, user)
|
|
|
|
self.event(type="topic", nick=nick, ident=ident, host=host, target=channel, message= newTopic)
|
|
|
|
def modeChanged(self, user, channel, toset, modes, args):
|
|
nick, ident, host = self.parsen(user)
|
|
userinfo.editUser(self.net, user)
|
|
argList = list(args)
|
|
modeList = [i for i in modes]
|
|
for a, m in zip(argList, modeList):
|
|
self.event(type="mode", nick=nick, ident=ident, host=host, target=channel, modes=m, 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
|
|
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)
|
|
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.name, 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(self.name, {"type": "conn", "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(self.name, {"type": "conn", "status": "failed", "message": error})
|
|
self.retry(connector)
|
|
#ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)
|
|
|