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.host)[:5].encode("hex")
  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
  59         def add_port(letter, number):
  60             DEFAULT_PORTS = {'t':'40001', 's':'40002', 'h':'7081', 'g':'7082'}
  61             if not number: return ''
  62             if DEFAULT_PORTS[letter] == number:
  63                 return letter + ' '
  64             else:
  65                 return letter + number + ' '
  66
  67         s += add_port('t',self.stratum_tcp_port)
  68         s += add_port('h',self.stratum_http_port)
  69         s += add_port('s',self.stratum_tcp_ssl_port)
  70         s += add_port('g',self.stratum_http_ssl_port)
  71         return s
  72
  73     def run(self):
  74         ircname = self.getname()
  75         print_log("joining IRC")
  76
  77         while not self.processor.shared.stopped():
  78             try:
  79                 s = socket.socket()
  80                 s.connect(('irc.freenode.net', 6667))
  81                 s.settimeout(0.3)
  82             except:
  83                 s.close()
  84                 print_log("IRC: reconnect in 10 s")
  85                 time.sleep(10)
  86                 continue
  87
  88             self.message = ''
  89             out_msg = []
  90
  91             try:
  92                 s.send('USER electrum 0 * :' + self.host + ' ' + ircname + '\n')
  93                 s.send('NICK ' + self.nick + '\n')
  94                 s.send('JOIN #electrum\n')
  95                 t = 0
  96
  97                 while not self.processor.shared.stopped():
  98                     try:
  99                         data = s.recv(2048)
 100                     except socket.timeout, e:
 101                         if out_msg:
 102                             m = out_msg.pop(0)
 103                             s.send(m)
 104                         continue
 105                     except:
 106                         print_log( "irc: socket error" )
 107                         time.sleep(1)
 108                         break
 109
 110                     self.message += data
 111
 112                     while self.message.find('\n') != -1:
 113                         pos = self.message.find('\n')
 114                         line = self.message[0:pos]
 115                         self.message = self.message[pos+1:]
 116                         line = line.strip('\r')
 117                         if not line:
 118                             continue
 119                         # print_log("<--", line)
 120                         line = line.split()
 121                         if line[0] == 'PING':
 122                             out_msg.append('PONG ' + line[1] + '\n')
 123                         elif line[1] == '353':  # answer to /names
 124                             for item in line[2:]:
 125                                 if item.startswith(self.prepend):
 126                                     out_msg.append('WHO %s\n' % item)
 127                         elif line[1] == '352':  # answer to /who
 128                             try:
 129                                 ip = socket.gethostbyname(line[5])
 130                             except:
 131                                 print_log("gethostbyname error", line[5])
 132                                 continue
 133                             name = line[7]
 134                             host = line[10]
 135                             ports = line[11:]
 136                             self.peers[name] = (ip, host, ports)
 137                         elif line[1] == 'KICK':
 138                             try:
 139                                 print_log("KICK", line[3] + line[4])
 140                             except:
 141                                 print_log("KICK", "error")
 142
 143                     if time.time() - t > 5*60:
 144                         #self.processor.push_response({'method': 'server.peers', 'params': [self.get_peers()]})
 145                         s.send('NAMES #electrum\n')
 146                         t = time.time()
 147                         self.peers = {}
 148             except:
 149                 traceback.print_exc(file=sys.stdout)
 150                 time.sleep(1)
 151             finally:
 152                 s.close()
 153
 154         print_log("quitting IRC")
 155
 156
 157 class ServerProcessor(Processor):
 158
 159     def __init__(self, config):
 160         Processor.__init__(self)
 161         self.daemon = True
 162         self.banner = config.get('server', 'banner')
 163
 164         if config.get('server', 'irc') == 'yes':
 165             self.irc = IrcThread(self, config)
 166         else:
 167             self.irc = None
 168
 169     def get_peers(self):
 170         if self.irc:
 171             return self.irc.get_peers()
 172         else:
 173             return []
 174
 175     def run(self):
 176         if self.irc:
 177             self.irc.start()
 178         Processor.run(self)
 179
 180     def process(self, session, request):
 181         method = request['method']
 182         params = request['params']
 183         result = None
 184
 185         if method == 'server.banner':
 186             result = self.banner.replace('\\n', '\n')
 187
 188         elif method == 'server.peers.subscribe':
 189             result = self.get_peers()
 190
 191         elif method == 'server.version':
 192             result = VERSION
 193
 194         else:
 195             print_log("unknown method", method)
 196
 197         if result != '':
 198             self.push_response(session, {'id': request['id'], 'result': result})