Neil Booth
7 years ago
14 changed files with 20 additions and 270 deletions
@ -1,126 +0,0 @@ |
|||
# Copyright (c) 2016-2017, Neil Booth |
|||
# |
|||
# All rights reserved. |
|||
# |
|||
# See the file "LICENCE" for information about the copyright |
|||
# and warranty status of this software. |
|||
|
|||
'''IRC connectivity to discover peers. |
|||
|
|||
Only calling start() requires the IRC Python module. |
|||
''' |
|||
|
|||
import asyncio |
|||
import re |
|||
|
|||
from collections import namedtuple |
|||
|
|||
from lib.hash import double_sha256 |
|||
from lib.util import LoggedClass |
|||
|
|||
|
|||
class IRC(LoggedClass): |
|||
|
|||
class DisconnectedError(Exception): |
|||
pass |
|||
|
|||
def __init__(self, env, peer_mgr): |
|||
super().__init__() |
|||
self.coin = env.coin |
|||
self.peer_mgr = peer_mgr |
|||
|
|||
# If this isn't something a peer or client expects |
|||
# then you won't appear in the client's network dialog box |
|||
self.channel = env.coin.IRC_CHANNEL |
|||
self.prefix = env.coin.IRC_PREFIX |
|||
self.nick = '{}{}'.format(self.prefix, |
|||
env.irc_nick if env.irc_nick else |
|||
double_sha256(env.host.encode()) |
|||
[:5].hex()) |
|||
self.peer_regexp = re.compile('({}[^!]*)!'.format(self.prefix)) |
|||
|
|||
async def start(self, name_pairs): |
|||
'''Start IRC connections if enabled in environment.''' |
|||
import irc.client as irc_client |
|||
from jaraco.stream import buffer |
|||
|
|||
# see https://pypi.python.org/pypi/irc under DecodingInput |
|||
irc_client.ServerConnection.buffer_class = \ |
|||
buffer.LenientDecodingLineBuffer |
|||
|
|||
# Register handlers for events we're interested in |
|||
reactor = irc_client.Reactor() |
|||
for event in 'welcome join whoreply disconnect'.split(): |
|||
reactor.add_global_handler(event, getattr(self, 'on_' + event)) |
|||
|
|||
# Note: Multiple nicks in same channel will trigger duplicate events |
|||
clients = [IrcClient(self.coin, real_name, self.nick + suffix, |
|||
reactor.server()) |
|||
for (real_name, suffix) in name_pairs] |
|||
|
|||
while True: |
|||
try: |
|||
for client in clients: |
|||
client.connect(self) |
|||
while True: |
|||
reactor.process_once() |
|||
await asyncio.sleep(2) |
|||
except irc_client.ServerConnectionError as e: |
|||
self.logger.error('connection error: {}'.format(e)) |
|||
except self.DisconnectedError: |
|||
self.logger.error('disconnected') |
|||
await asyncio.sleep(10) |
|||
|
|||
def log_event(self, event): |
|||
self.logger.info('IRC event type {} source {} args {}' |
|||
.format(event.type, event.source, event.arguments)) |
|||
|
|||
def on_welcome(self, connection, event): |
|||
'''Called when we connect to irc server.''' |
|||
connection.join(self.channel) |
|||
|
|||
def on_disconnect(self, connection, event): |
|||
'''Called if we are disconnected.''' |
|||
self.log_event(event) |
|||
raise self.DisconnectedError |
|||
|
|||
def on_join(self, connection, event): |
|||
'''Called when someone new connects to our channel, including us.''' |
|||
# /who the channel when we join. We used to /who on each |
|||
# namreply event, but the IRC server would frequently kick us |
|||
# for flooding. This requests only once including the tor case. |
|||
if event.source.startswith(self.nick + '!'): |
|||
connection.who(self.channel) |
|||
else: |
|||
match = self.peer_regexp.match(event.source) |
|||
if match: |
|||
connection.who(match.group(1)) |
|||
|
|||
def on_whoreply(self, connection, event): |
|||
'''Called when a response to our who requests arrives. |
|||
|
|||
The nick is the 4th argument, and real name is in the 6th |
|||
argument preceeded by '0 ' for some reason. |
|||
''' |
|||
nick = event.arguments[4] |
|||
if nick.startswith(self.prefix): |
|||
line = event.arguments[6].split() |
|||
hp_string = ' '.join(line[1:]) # hostname, ports, version etc. |
|||
self.peer_mgr.add_irc_peer(nick, hp_string) |
|||
|
|||
|
|||
class IrcClient(object): |
|||
|
|||
def __init__(self, coin, real_name, nick, server): |
|||
self.irc_host = coin.IRC_SERVER |
|||
self.irc_port = coin.IRC_PORT |
|||
self.nick = nick |
|||
self.real_name = real_name |
|||
self.server = server |
|||
|
|||
def connect(self, irc): |
|||
'''Connect this client to its IRC server''' |
|||
irc.logger.info('joining {} as "{}" with real name "{}"' |
|||
.format(irc.channel, self.nick, self.real_name)) |
|||
self.server.connect(self.irc_host, self.irc_port, self.nick, |
|||
ircname=self.real_name) |
|||
Loading…
Reference in new issue