backends/irc/__init__.py

   1 import socket
   2 import sys
   3 import threading
   4 import time
   5 import traceback
   6
   7 from processor import Processor
   8 from utils import Hash, print_log
   9 from version import VERSION
  10
  11
  12 class IrcThread(threading.Thread):
  13
  14     def __init__(self, processor, config):
  15         threading.Thread.__init__(self)
  16
  17         self.processor = processor
  18         self.daemon = True
  19         self.stratum_tcp_port = config.get('server', 'stratum_tcp_port')
  20         self.stratum_http_port = config.get('server', 'stratum_http_port')
  21         self.stratum_tcp_ssl_port = config.get('server', 'stratum_tcp_ssl_port')
  22         self.stratum_http_ssl_port = config.get('server', 'stratum_http_ssl_port')
  23         self.report_stratum_tcp_port = config.get('server', 'report_stratum_tcp_port')
  24         self.report_stratum_http_port = config.get('server', 'report_stratum_http_port')
  25         self.report_stratum_tcp_ssl_port = config.get('server', 'report_stratum_tcp_ssl_port')
  26         self.report_stratum_http_ssl_port = config.get('server', 'report_stratum_http_ssl_port')
  27         self.peers = {}
  28         self.host = config.get('server', 'host')
  29         self.report_host = config.get('server', 'report_host')
  30         self.nick = config.get('server', 'irc_nick')
  31         if self.report_stratum_tcp_port:
  32             self.stratum_tcp_port = self.report_stratum_tcp_port
  33         if self.report_stratum_http_port:
  34             self.stratum_http_port = self.report_stratum_http_port
  35         if self.report_stratum_tcp_ssl_port:
  36             self.stratum_tcp_ssl_port = self.report_stratum_tcp_ssl_port
  37         if self.report_stratum_http_ssl_port:
  38             self.stratum_http_ssl_port = self.report_stratum_http_ssl_port
  39         if self.report_host:
  40             self.host = self.report_host
  41         if not self.nick:
  42             self.nick = Hash(self.report_host)[:10]
  43         self.prepend = 'E_'
  44         if config.get('server', 'coin') == 'litecoin':
  45             self.prepend = 'EL_'
  46         self.pruning = config.get('server', 'backend') == 'leveldb'
  47         if self.pruning:
  48             self.pruning_limit = config.get('leveldb', 'pruning_limit')
  49         self.nick = self.prepend + self.nick
  50
  51     def get_peers(self):
  52         return self.peers.values()
  53
  54     def getname(self):
  55         s = 'v' + VERSION + ' '
  56         if self.pruning:
  57             s += 'p' + self.pruning_limit + ' '
  58         if self.stratum_tcp_port:
  59             s += 't' + self.stratum_tcp_port + ' '
  60         if self.stratum_http_port:
  61             s += 'h' + self.stratum_http_port + ' '
  62         if self.stratum_tcp_port:
  63             s += 's' + self.stratum_tcp_ssl_port + ' '
  64         if self.stratum_http_port:
  65             s += 'g' + self.stratum_http_ssl_port + ' '
  66         return s
  67
  68     def run(self):
  69         ircname = self.getname()
  70
  71         while not self.processor.shared.stopped():
  72             try:
  73                 s = socket.socket()
  74                 s.connect(('irc.freenode.net', 6667))
  75                 s.settimeout(300)
  76             except:
  77                 s.close()
  78                 time.sleep(10)
  79                 continue
  80
  81             try:
  82                 s.send('USER electrum 0 * :' + self.host + ' ' + ircname + '\n')
  83                 s.send('NICK ' + self.nick + '\n')
  84                 s.send('JOIN #electrum\n')
  85                 sf = s.makefile('r', 0)
  86                 t = 0
  87                 while not self.processor.shared.stopped():
  88                     line = sf.readline().rstrip('\r\n').split()
  89                     if not line:
  90                         continue
  91                     if line[0] == 'PING':
  92                         s.send('PONG ' + line[1] + '\n')
  93                     elif '353' in line:  # answer to /names
  94                         k = line.index('353')
  95                         for item in line[k+1:]:
  96                             if item.startswith(self.prepend):
  97                                 s.send('WHO %s\n' % item)
  98                     elif '352' in line:  # answer to /who
  99                         # warning: this is a horrible hack which apparently works
 100                         k = line.index('352')
 101                         try:
 102                             ip = socket.gethostbyname(line[k+4])
 103                         except:
 104                             print_log("gethostbyname error", line[k+4])
 105                             continue
 106                         name = line[k+6]
 107                         host = line[k+9]
 108                         ports = line[k+10:]
 109                         self.peers[name] = (ip, host, ports)
 110                     if time.time() - t > 5*60:
 111                         self.processor.push_response({'method': 'server.peers', 'params': [self.get_peers()]})
 112                         s.send('NAMES #electrum\n')
 113                         t = time.time()
 114                         self.peers = {}
 115             except:
 116                 traceback.print_exc(file=sys.stdout)
 117             finally:
 118                 sf.close()
 119                 s.close()
 120
 121         print_log("quitting IRC")
 122
 123
 124 class ServerProcessor(Processor):
 125
 126     def __init__(self, config):
 127         Processor.__init__(self)
 128         self.daemon = True
 129         self.banner = config.get('server', 'banner')
 130         self.password = config.get('server', 'password')
 131
 132         if config.get('server', 'irc') == 'yes':
 133             self.irc = IrcThread(self, config)
 134         else:
 135             self.irc = None
 136
 137     def get_peers(self):
 138         if self.irc:
 139             return self.irc.get_peers()
 140         else:
 141             return []
 142
 143     def run(self):
 144         if self.irc:
 145             self.irc.start()
 146         Processor.run(self)
 147
 148     def process(self, request):
 149         method = request['method']
 150         params = request['params']
 151         result = None
 152
 153         if method in ['server.stop', 'server.info']:
 154             try:
 155                 password = request['params'][0]
 156             except:
 157                 password = None
 158
 159             if password != self.password:
 160                 self.push_response({'id': request['id'],
 161                                     'result': None,
 162                                     'error': 'incorrect password'})
 163                 return
 164
 165         if method == 'server.banner':
 166             result = self.banner.replace('\\n', '\n')
 167
 168         elif method == 'server.peers.subscribe':
 169             result = self.get_peers()
 170
 171         elif method == 'server.version':
 172             result = VERSION
 173
 174         elif method == 'server.stop':
 175             self.shared.stop()
 176             result = 'stopping, please wait until all threads terminate.'
 177
 178         elif method == 'server.info':
 179             result = map(lambda s: {"time": s.time,
 180                                     "name": s.name,
 181                                     "address": s.address,
 182                                     "version": s.version,
 183                                     "subscriptions": len(s.subscriptions)},
 184                          self.dispatcher.request_dispatcher.get_sessions())
 185
 186         elif method == 'server.cache':
 187             p = self.dispatcher.request_dispatcher.processors['blockchain']
 188             result = len(repr(p.store.tx_cache))
 189
 190         elif method == 'server.load':
 191             p = self.dispatcher.request_dispatcher.processors['blockchain']
 192             result = p.queue.qsize()
 193
 194         else:
 195             print_log("unknown method", request)
 196
 197         if result != '':
 198             self.push_response({'id': request['id'], 'result': result})