server.py

   1 #!/usr/bin/env python
   2 # Copyright(C) 2012 thomasv@gitorious
   3
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU Affero General Public License as
   6 # published by the Free Software Foundation, either version 3 of the
   7 # License, or (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful, but
  10 # WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public
  15 # License along with this program.  If not, see
  16 # <http://www.gnu.org/licenses/agpl.html>.
  17
  18 import time, sys, traceback, threading
  19 import ConfigParser
  20
  21 import logging
  22 logging.basicConfig()
  23
  24 if sys.maxsize <= 2**32:
  25     print "Warning: it looks like you are using a 32bit system. You may experience crashes caused by mmap"
  26
  27
  28 def attempt_read_config(config, filename):
  29     try:
  30         with open(filename, 'r') as f:
  31             config.readfp(f)
  32     except IOError:
  33         pass
  34
  35 def create_config():
  36     config = ConfigParser.ConfigParser()
  37     # set some defaults, which will be overwritten by the config file
  38     config.add_section('server')
  39     config.set('server', 'banner', 'Welcome to Electrum!')
  40     config.set('server', 'host', 'localhost')
  41     config.set('server', 'report_host', '')
  42     config.set('server', 'stratum_tcp_port', '50001')
  43     config.set('server', 'stratum_http_port', '8081')
  44     config.set('server', 'stratum_tcp_ssl_port', '50002')
  45     config.set('server', 'stratum_http_ssl_port', '8082')
  46     config.set('server', 'report_stratum_tcp_port', '')
  47     config.set('server', 'report_stratum_http_port', '')
  48     config.set('server', 'report_stratum_tcp_ssl_port', '')
  49     config.set('server', 'report_stratum_http_ssl_port', '')
  50     config.set('server', 'ssl_certfile', '')
  51     config.set('server', 'ssl_keyfile', '')
  52     config.set('server', 'password', '')
  53     config.set('server', 'irc', 'yes')
  54     config.set('server', 'irc_nick', '')
  55     config.set('server', 'coin', '')
  56     config.set('server', 'datadir', '')
  57     config.add_section('database')
  58     config.set('database', 'type', 'psycopg2')
  59     config.set('database', 'database', 'abe')
  60     config.set('database', 'limit', '1000')
  61     config.set('server', 'backend', 'abe')
  62
  63     for path in ('/etc/', ''):
  64         filename = path + 'electrum.conf'
  65         attempt_read_config(config, filename)
  66
  67     try:
  68         with open('/etc/electrum.banner', 'r') as f:
  69             config.set('server','banner', f.read())
  70     except IOError:
  71         pass
  72
  73     return config
  74
  75 def run_rpc_command(command, stratum_tcp_port):
  76     import socket, json
  77     try:
  78         s = socket.socket( socket.AF_INET, socket.SOCK_STREAM )
  79         s.connect(( host, int(stratum_tcp_port )))
  80     except:
  81         print "cannot connect to server."
  82         return
  83
  84     method = 'server.' + command
  85     request = json.dumps( { 'id':0, 'method':method, 'params':[password] } )
  86     s.send(request + '\n')
  87     msg = ''
  88     while True:
  89         o = s.recv(1024)
  90         msg += o
  91         if msg.find('\n') != -1: break
  92     s.close()
  93     r = json.loads(msg).get('result')
  94
  95     if command == 'info':
  96         now = time.time()
  97         print 'type           address   sub  version  time'
  98         for item in r:
  99             print '%4s   %15s   %3s  %7s  %.2f'%( item.get('name'),
 100                                                   item.get('address'),
 101                                                   item.get('subscriptions'),
 102                                                   item.get('version'),
 103                                                   (now - item.get('time')) )
 104     else:
 105         print r
 106
 107 if __name__ == '__main__':
 108     config = create_config()
 109     password = config.get('server', 'password')
 110     host = config.get('server', 'host')
 111     stratum_tcp_port = config.get('server', 'stratum_tcp_port')
 112     stratum_http_port = config.get('server', 'stratum_http_port')
 113     stratum_tcp_ssl_port = config.get('server', 'stratum_tcp_ssl_port')
 114     stratum_http_ssl_port = config.get('server', 'stratum_http_ssl_port')
 115     ssl_certfile = config.get('server', 'ssl_certfile')
 116     ssl_keyfile = config.get('server', 'ssl_keyfile')
 117
 118     if stratum_tcp_ssl_port or stratum_http_ssl_port: assert ssl_certfile and ssl_keyfile
 119
 120     if len(sys.argv) > 1:
 121         run_rpc_command(sys.argv[1], stratum_tcp_port)
 122         sys.exit(0)
 123
 124     from processor import Dispatcher, print_log
 125     from backends.irc import ServerProcessor
 126
 127     backend_name = config.get('server', 'backend')
 128     if backend_name == 'abe':
 129         from backends.abe import BlockchainProcessor
 130     elif backend_name == 'libbitcoin':
 131         from backends.libbitcoin import BlockchainProcessor
 132     elif backend_name == 'leveldb':
 133         from backends.bitcoind import BlockchainProcessor
 134     else:
 135         print "Unknown backend '%s' specified\n" % backend_name
 136         sys.exit(1)
 137
 138     for i in range(5): print ""
 139     print_log( "Starting Electrum server on", host)
 140
 141     # Create hub
 142     dispatcher = Dispatcher(config)
 143     shared = dispatcher.shared
 144
 145     # Create and register processors
 146     chain_proc = BlockchainProcessor(config, shared)
 147     dispatcher.register('blockchain', chain_proc)
 148
 149     server_proc = ServerProcessor(config)
 150     dispatcher.register('server', server_proc)
 151
 152     transports = []
 153     # Create various transports we need
 154     if stratum_tcp_port:
 155         from transports.stratum_tcp import TcpServer
 156         tcp_server = TcpServer(dispatcher, host, int(stratum_tcp_port), False, None, None)
 157         transports.append(tcp_server)
 158
 159     if stratum_tcp_ssl_port:
 160         from transports.stratum_tcp import TcpServer
 161         tcp_server = TcpServer(dispatcher, host, int(stratum_tcp_ssl_port), True, ssl_certfile, ssl_keyfile)
 162         transports.append(tcp_server)
 163
 164     if stratum_http_port:
 165         from transports.stratum_http import HttpServer
 166         http_server = HttpServer(dispatcher, host, int(stratum_http_port), False, None, None)
 167         transports.append(http_server)
 168
 169     if stratum_http_ssl_port:
 170         from transports.stratum_http import HttpServer
 171         http_server = HttpServer(dispatcher, host, int(stratum_http_ssl_port), True, ssl_certfile, ssl_keyfile)
 172         transports.append(http_server)
 173
 174     for server in transports:
 175         server.start()
 176
 177     while not shared.stopped():
 178         try:
 179             time.sleep(1)
 180         except:
 181             shared.stop()
 182
 183     print_log( "Electrum Server stopped")
 184