1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
|
import twisted, sys, codecs, traceback
from twisted.words.protocols import irc
from twisted.internet import reactor, protocol, abstract, ssl
from twisted.web import resource, server
from twisted.protocols import basic
from twisted.names.client import Resolver
import hmac, time, config, random, qwebirc.config_options as config_options
from config import HMACTEMPORAL
if config.get("CONNECTION_RESOLVER"):
CONNECTION_RESOLVER = Resolver(servers=config.get("CONNECTION_RESOLVER"))
else:
CONNECTION_RESOLVER = None
if hasattr(config, "WEBIRC_MODE") and config.WEBIRC_MODE == "hmac":
HMACKEY = hmac.HMAC(key=config.HMACKEY)
def hmacfn(*args):
h = HMACKEY.copy()
h.update("%d %s" % (int(time.time() / HMACTEMPORAL), " ".join(args)))
return h.hexdigest()
def utf8_iso8859_1(data, table=dict((x, x.decode("iso-8859-1")) for x in map(chr, range(0, 256)))):
return (table.get(data.object[data.start]), data.start+1)
codecs.register_error("mixed-iso-8859-1", utf8_iso8859_1)
def irc_decode(x):
try:
return x.decode("utf-8", "mixed-iso-8859-1")
except UnicodeDecodeError:
return x.decode("iso-8859-1", "ignore")
class QWebIRCClient(basic.LineReceiver):
delimiter = "\n"
def __init__(self, *args, **kwargs):
self.__nickname = "(unregistered)"
def dataReceived(self, data):
basic.LineReceiver.dataReceived(self, data.replace("\r", ""))
def lineReceived(self, line):
line = irc_decode(irc.lowDequote(line))
try:
prefix, command, params = irc.parsemsg(line)
self.handleCommand(command, prefix, params)
except irc.IRCBadMessage:
# emit and ignore
traceback.print_exc()
return
if command == "001":
self.__nickname = params[0]
if self.__perform is not None:
for x in self.__perform:
self.write(x)
self.__perform = None
elif command == "NICK":
nick = prefix.split("!", 1)[0]
if nick == self.__nickname:
self.__nickname = params[0]
def handleCommand(self, command, prefix, params):
self("c", command, prefix, params)
def __call__(self, *args):
self.factory.publisher.event(args)
def write(self, data):
self.transport.write("%s\r\n" % irc.lowQuote(data.encode("utf-8")))
def connectionMade(self):
basic.LineReceiver.connectionMade(self)
self.lastError = None
f = self.factory.ircinit
nick, ident, ip, realname, hostname, pass_ = f["nick"], f["ident"], f["ip"], f["realname"], f["hostname"], f.get("password")
self.__nickname = nick
self.__perform = f.get("perform")
if not hasattr(config, "WEBIRC_MODE"):
self.write("USER %s bleh bleh %s :%s" % (ident, ip, realname))
elif config.WEBIRC_MODE == "hmac":
hmac = hmacfn(ident, ip)
self.write("USER %s bleh bleh %s %s :%s" % (ident, ip, hmac, realname))
elif config.WEBIRC_MODE == "webirc":
self.write("WEBIRC %s qwebirc %s %s" % (config.WEBIRC_PASSWORD, hostname, ip))
self.write("USER %s bleh %s :%s" % (ident, ip, realname))
elif config.WEBIRC_MODE == "cgiirc":
self.write("PASS %s_%s_%s" % (config.CGIIRC_STRING, ip, hostname))
self.write("USER %s bleh %s :%s" % (ident, ip, realname))
elif config.WEBIRC_MODE == config_options.WEBIRC_REALNAME or config.WEBIRC_MODE is None: # last bit is legacy
if ip == hostname:
dispip = ip
else:
dispip = "%s/%s" % (hostname, ip)
self.write("USER %s bleh bleh :%s - %s" % (ident, dispip, realname))
if pass_ is not None:
self.write("PASS :%s" % pass_)
self.write("NICK %s" % nick)
self.factory.client = self
self("connect")
def __str__(self):
return "<QWebIRCClient: %s!%s@%s>" % (self.__nickname, self.factory.ircinit["ident"], self.factory.ircinit["ip"])
def connectionLost(self, reason):
if self.lastError:
self.disconnect("Connection to IRC server lost: %s" % self.lastError)
else:
self.disconnect("Connection to IRC server lost.")
self.factory.client = None
basic.LineReceiver.connectionLost(self, reason)
def error(self, message):
self.lastError = message
self.write("QUIT :qwebirc exception: %s" % message)
self.transport.loseConnection()
def disconnect(self, reason):
self("disconnect", reason)
self.factory.publisher.disconnect()
class QWebIRCFactory(protocol.ClientFactory):
protocol = QWebIRCClient
def __init__(self, publisher, **kwargs):
self.client = None
self.publisher = publisher
self.ircinit = kwargs
def write(self, data):
self.client.write(data)
def error(self, reason):
self.client.error(reason)
def clientConnectionFailed(self, connector, reason):
protocol.ClientFactory.clientConnectionFailed(self, connector, reason)
self.publisher.event(["disconnect", "Connection to IRC server failed."])
self.publisher.disconnect()
def createIRC(*args, **kwargs):
f = QWebIRCFactory(*args, **kwargs)
tcpkwargs = {}
if hasattr(config, "OUTGOING_IP"):
tcpkwargs["bindAddress"] = (config.OUTGOING_IP, 0)
if CONNECTION_RESOLVER is None:
if hasattr(config, "SSLPORT"):
reactor.connectSSL(config.IRCSERVER, config.SSLPORT, f, ssl.ClientContextFactory(), **tcpkwargs)
else:
reactor.connectTCP(config.IRCSERVER, config.IRCPORT, f, **tcpkwargs)
return f
def callback(result):
name, port = random.choice(sorted((str(x.payload.target), x.payload.port) for x in result[0]))
reactor.connectTCP(name, port, f, **tcpkwargs)
def errback(err):
f.clientConnectionFailed(None, err) # None?!
d = CONNECTION_RESOLVER.lookupService(config.IRCSERVER, (1, 3, 11))
d.addCallbacks(callback, errback)
return f
if __name__ == "__main__":
e = createIRC(lambda x: 2, nick="slug__moo", ident="mooslug", ip="1.2.3.6", realname="mooooo", hostname="1.2.3.4")
reactor.run()
|
