X-Git-Url: https://git.novaco.in/?p=electrum-server.git;a=blobdiff_plain;f=backends%2Firc%2F__init__.py;h=80e4a1d2e60581e1f91e867b9de85f4bd2c569cf;hp=0e7b7570e3bb262c82e6690fc490ada2a330fc09;hb=fc59efd91121cc8dc2e38d3dd0f6ae56b42f5749;hpb=c9c337a89a3f04d7a985a521ca552d421fd94fda diff --git a/backends/irc/__init__.py b/backends/irc/__init__.py index 0e7b757..80e4a1d 100644 --- a/backends/irc/__init__.py +++ b/backends/irc/__init__.py @@ -39,11 +39,13 @@ class IrcThread(threading.Thread): if self.report_host: self.host = self.report_host if not self.nick: - self.nick = Hash(self.report_host)[:10] + self.nick = Hash(self.host)[:5].encode("hex") self.prepend = 'E_' if config.get('server', 'coin') == 'litecoin': self.prepend = 'EL_' self.pruning = config.get('server', 'backend') == 'leveldb' + if self.pruning: + self.pruning_limit = config.get('leveldb', 'pruning_limit') self.nick = self.prepend + self.nick def get_peers(self): @@ -52,15 +54,20 @@ class IrcThread(threading.Thread): def getname(self): 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 + ' ' + s += 'p' + self.pruning_limit + ' ' + + def add_port(letter, number): + DEFAULT_PORTS = {'t':'50001', 's':'50002', 'h':'8081', 'g':'8082'} + if not number: return '' + if DEFAULT_PORTS[letter] == number: + return letter + ' ' + else: + return letter + number + ' ' + + s += add_port('t',self.stratum_tcp_port) + s += add_port('h',self.stratum_http_port) + s += add_port('s',self.stratum_tcp_ssl_port) + s += add_port('g',self.stratum_http_ssl_port) return s def run(self): @@ -76,40 +83,58 @@ class IrcThread(threading.Thread): time.sleep(10) continue + self.message = '' try: s.send('USER electrum 0 * :' + self.host + ' ' + ircname + '\n') s.send('NICK ' + self.nick + '\n') s.send('JOIN #electrum\n') - sf = s.makefile('r', 0) t = 0 while not self.processor.shared.stopped(): - line = sf.readline().rstrip('\r\n').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.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 - k = line.index('352') - ip = socket.gethostbyname(line[k+4]) - name = line[k+6] - host = line[k+9] - ports = line[k+10:] - self.peers[name] = (ip, host, ports) + try: + data = s.recv(2048) + except: + print_log( "irc: socket error" ) + time.sleep(1) + break + + self.message += data + + while self.message.find('\n') != -1: + pos = self.message.find('\n') + line = self.message[0:pos] + self.message = self.message[pos+1:] + line = line.strip('\r') + if not line: + continue + line = line.split() + 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.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 + k = line.index('352') + try: + ip = socket.gethostbyname(line[k+4]) + except: + print_log("gethostbyname error", line[k+4]) + continue + name = line[k+6] + host = line[k+9] + ports = line[k+10:] + self.peers[name] = (ip, host, ports) + if time.time() - t > 5*60: - self.processor.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 = {} except: traceback.print_exc(file=sys.stdout) finally: - sf.close() s.close() print_log("quitting IRC") @@ -139,19 +164,20 @@ class ServerProcessor(Processor): self.irc.start() Processor.run(self) - def process(self, request): + def process(self, session, request): method = request['method'] params = request['params'] result = None - if method in ['server.stop', 'server.info']: + if method in ['server.stop', 'server.info', 'server.debug']: try: password = request['params'][0] except: password = None if password != self.password: - self.push_response({'id': request['id'], + self.push_response(session, + {'id': request['id'], 'result': None, 'error': 'incorrect password'}) return @@ -177,16 +203,22 @@ class ServerProcessor(Processor): "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() - + elif method == 'server.debug': + try: + s = request['params'][1] + except: + s = None + + if s: + from guppy import hpy + h = hpy() + bp = self.dispatcher.request_dispatcher.processors['blockchain'] + try: + result = str(eval(s)) + except: + result = "error" else: - print_log("unknown method", request) + print_log("unknown method", method) if result != '': - self.push_response({'id': request['id'], 'result': result}) + self.push_response(session, {'id': request['id'], 'result': result})