X-Git-Url: https://git.novaco.in/?p=electrum-server.git;a=blobdiff_plain;f=backends%2Firc%2F__init__.py;h=46ec92146f9bf427fae491bf320f6e02f9333270;hp=81b223fbf7225dfb72a2032e46e4bf742bdbe192;hb=3989b283341f5d456a5c6ffdd4e7fd00f57d0fcd;hpb=cf5c787b78aa20556950b3f40cafe72cd6ff2b1f diff --git a/backends/irc/__init__.py b/backends/irc/__init__.py index 81b223f..46ec921 100644 --- a/backends/irc/__init__.py +++ b/backends/irc/__init__.py @@ -5,66 +5,84 @@ def random_string(N): return ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(N)) from processor import Processor +from version import VERSION -class ServerProcessor(Processor): +class IrcThread(threading.Thread): - def __init__(self, config): - Processor.__init__(self) + def __init__(self, processor, config): + threading.Thread.__init__(self) + self.processor = processor self.daemon = True - self.peers = {} - self.banner = config.get('server','banner') - self.host = config.get('server','host') - self.password = config.get('server','password') - - self.native_port = config.get('server','native_port') self.stratum_tcp_port = config.get('server','stratum_tcp_port') self.stratum_http_port = config.get('server','stratum_http_port') - - self.irc = config.get('server', 'irc') == 'yes' - self.nick = config.get('server', 'irc_nick') + self.stratum_tcp_ssl_port = config.get('server','stratum_tcp_ssl_port') + self.stratum_http_ssl_port = config.get('server','stratum_http_ssl_port') + self.report_stratum_tcp_port = config.get('server','report_stratum_tcp_port') + self.report_stratum_http_port = config.get('server','report_stratum_http_port') + self.report_stratum_tcp_ssl_port = config.get('server','report_stratum_tcp_ssl_port') + self.report_stratum_http_ssl_port = config.get('server','report_stratum_http_ssl_port') + self.peers = {} + self.host = config.get('server','host') + self.report_host = config.get('server','report_host') + self.nick = config.get('server', 'irc_nick') + if self.report_stratum_tcp_port: self.stratum_tcp_port = self.report_stratum_tcp_port + if self.report_stratum_http_port: self.stratum_http_port = self.report_stratum_http_port + if self.report_stratum_tcp_ssl_port: self.stratum_tcp_ssl_port = self.report_stratum_tcp_ssl_port + if self.report_stratum_http_ssl_port: self.stratum_http_ssl_port = self.report_stratum_http_ssl_port + if self.report_host: self.host = self.report_host if not self.nick: self.nick = random_string(10) - + self.prepend = 'E_' + if config.get('server', 'coin') == 'litecoin': + self.prepend = 'EL_' + self.pruning = config.get('server', 'backend') == 'leveldb' + self.nick = self.prepend + self.nick def get_peers(self): return self.peers.values() def getname(self): - s = '' - if self.native_port: - s+= 'n' + self.native_port + ' ' + s = 'v' + VERSION + ' ' + if self.pruning: s += 'p ' if self.stratum_tcp_port: s += 't' + self.stratum_tcp_port + ' ' if self.stratum_http_port: s += 'h' + self.stratum_http_port + ' ' + if self.stratum_tcp_port: + s += 's' + self.stratum_tcp_ssl_port + ' ' + if self.stratum_http_port: + s += 'g' + self.stratum_http_ssl_port + ' ' return s def run(self): - if not self.irc: - return - ircname = self.getname() - while not self.shared.stopped(): + while not self.processor.shared.stopped(): try: s = socket.socket() s.connect(('irc.freenode.net', 6667)) + except: + time.sleep(10) + continue + + try: s.send('USER electrum 0 * :' + self.host + ' ' + ircname + '\n') - s.send('NICK E_' + self.nick + '\n') + s.send('NICK ' + self.nick + '\n') s.send('JOIN #electrum\n') sf = s.makefile('r', 0) t = 0 - while not self.shared.stopped(): + while not self.processor.shared.stopped(): line = sf.readline() line = line.rstrip('\r\n') line = line.split() + if not line: continue if line[0]=='PING': s.send('PONG '+line[1]+'\n') elif '353' in line: # answer to /names k = line.index('353') for item in line[k+1:]: - if item[0:2] == 'E_': + if item.startswith(self.prepend): s.send('WHO %s\n'%item) elif '352' in line: # answer to /who # warning: this is a horrible hack which apparently works @@ -76,7 +94,7 @@ class ServerProcessor(Processor): ports = line[k+10:] self.peers[name] = (ip, host, ports) if time.time() - t > 5*60: - self.push_response({'method':'server.peers', 'params':[self.get_peers()]}) + self.processor.push_response({'method':'server.peers', 'params':[self.get_peers()]}) s.send('NAMES #electrum\n') t = time.time() self.peers = {} @@ -86,13 +104,52 @@ class ServerProcessor(Processor): sf.close() s.close() + print "quitting IRC" + +class ServerProcessor(Processor): + + def __init__(self, config): + Processor.__init__(self) + self.daemon = True + self.banner = config.get('server','banner') + self.password = config.get('server','password') + + if config.get('server', 'irc') == 'yes': + self.irc = IrcThread(self, config) + else: + self.irc = None + + + def get_peers(self): + if self.irc: + return self.irc.get_peers() + else: + return [] + + + def run(self): + if self.irc: + self.irc.start() + Processor.run(self) + def process(self, request): method = request['method'] params = request['params'] result = None + if method in ['server.stop', 'server.info']: + try: + password = request['params'][0] + except: + password = None + + if password != self.password: + response = { 'id':request['id'], 'result':None, 'error':'incorrect password'} + self.push_response(response) + return + if method == 'server.banner': result = self.banner.replace('\\n','\n') @@ -100,21 +157,32 @@ class ServerProcessor(Processor): result = self.get_peers() elif method == 'server.version': - pass + result = VERSION elif method == 'server.stop': - print "stopping..." - try: - password = request['params'][0] - except: - password = None - if password == self.password: - self.shared.stop() - result = 'ok' + self.shared.stop() + result = 'stopping, please wait until all threads terminate.' + + elif method == 'server.info': + result = map(lambda s: { "time":s.time, + "name":s.name, + "address":s.address, + "version":s.version, + "subscriptions":len(s.subscriptions)}, + self.dispatcher.request_dispatcher.get_sessions()) + + elif method == 'server.cache': + p = self.dispatcher.request_dispatcher.processors['blockchain'] + result = len(repr(p.store.tx_cache)) + + elif method == 'server.load': + p = self.dispatcher.request_dispatcher.processors['blockchain'] + result = p.queue.qsize() + else: print "unknown method", request if result!='': - response = { 'id':request['id'], 'method':method, 'params':params, 'result':result } + response = { 'id':request['id'], 'result':result } self.push_response(response)