backends/irc/__init__.py

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