modules/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
   9 class ServerProcessor(Processor):
  10
  11     def __init__(self, config):
  12         Processor.__init__(self)
  13         self.daemon = True
  14         self.peers = {}
  15         self.banner = config.get('server','banner')
  16         self.host = config.get('server','host')
  17
  18         self.native_port = config.get('server','native_port')
  19         self.stratum_tcp_port = config.get('server','stratum_tcp_port')
  20         self.stratum_http_port = config.get('server','stratum_http_port')
  21
  22         self.irc = config.get('server', 'irc') == 'yes'
  23         self.nick = config.get('server', 'irc_nick')
  24         if not self.nick: self.nick = random_string(10)
  25
  26
  27     def get_peers(self):
  28         return self.peers.values()
  29
  30
  31     def getname(self):
  32         s = ''
  33         if self.native_port:
  34             s+= 'n' + self.native_port + ' '
  35         if self.stratum_tcp_port:
  36             s += 's' + self.stratum_tcp_port + ' '
  37         if self.stratum_http_port:
  38             s += 'h' + self.stratum_http_port + ' '
  39         return s
  40
  41
  42     def run(self):
  43         if not self.irc:
  44             return
  45
  46         ircname = self.getname()
  47
  48         while not self.shared.stopped():
  49             try:
  50                 s = socket.socket()
  51                 s.connect(('irc.freenode.net', 6667))
  52                 s.send('USER electrum 0 * :' + self.host + ' ' + ircname + '\n')
  53                 s.send('NICK E_' + self.nick + '\n')
  54                 s.send('JOIN #electrum\n')
  55                 sf = s.makefile('r', 0)
  56                 t = 0
  57                 while not self.shared.stopped():
  58                     line = sf.readline()
  59                     line = line.rstrip('\r\n')
  60                     line = line.split()
  61                     if line[0]=='PING':
  62                         s.send('PONG '+line[1]+'\n')
  63                     elif '353' in line: # answer to /names
  64                         k = line.index('353')
  65                         for item in line[k+1:]:
  66                             if item[0:2] == 'E_':
  67                                 s.send('WHO %s\n'%item)
  68                     elif '352' in line: # answer to /who
  69                         # warning: this is a horrible hack which apparently works
  70                         k = line.index('352')
  71                         ip = line[k+4]
  72                         ip = socket.gethostbyname(ip)
  73                         name = line[k+6]
  74                         host = line[k+9]
  75                         ports  = line[k+10:]
  76                         self.peers[name] = (ip, host, ports)
  77                     if time.time() - t > 5*60:
  78                         self.push_response({'method':'server.peers', 'result':[self.get_peers()]})
  79                         s.send('NAMES #electrum\n')
  80                         t = time.time()
  81                         self.peers = {}
  82             except:
  83                 traceback.print_exc(file=sys.stdout)
  84             finally:
  85                 sf.close()
  86                 s.close()
  87
  88
  89
  90     def process(self, request):
  91         method = request['method']
  92         params = request['params']
  93         result = None
  94         if method == 'server.banner':
  95             result = self.banner.replace('\\n','\n')
  96         elif method == 'server.peers.subscribe':
  97             result = self.get_peers()
  98         elif method == 'server.version':
  99             print "version", params
 100         else:
 101             print "unknown method", request
 102
 103         if result!='':
 104             response = { 'id':request['id'], 'method':method, 'params':params, 'result':result }
 105             self.push_response(response)
 106
 107
 108