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