Merge pull request #3 from fsb4000/master
[electrum-server.git] / server.py
index 675ef5f..2a504a1 100755 (executable)
--- a/server.py
+++ b/server.py
 # License along with this program.  If not, see
 # <http://www.gnu.org/licenses/agpl.html>.
 
-import time, json, socket, operator, thread, ast, sys, re, traceback
 import ConfigParser
-from json import dumps, loads
-import urllib
+import logging
+import socket
+import sys
+import time
 import threading
+import traceback
+
+import json
+import os
+
+logging.basicConfig()
+
+if sys.maxsize <= 2**32:
+    print "Warning: it looks like you are using a 32bit system. You may experience crashes caused by mmap"
+
+
+def attempt_read_config(config, filename):
+    try:
+        with open(filename, 'r') as f:
+            config.readfp(f)
+    except IOError:
+        pass
+
+
+def create_config():
+    config = ConfigParser.ConfigParser()
+    # set some defaults, which will be overwritten by the config file
+    config.add_section('server')
+    config.set('server', 'banner', 'Welcome to Electrum!')
+    config.set('server', 'host', 'localhost')
+    config.set('server', 'report_host', '')
+    config.set('server', 'stratum_tcp_port', '40001')
+    config.set('server', 'stratum_http_port', '7081')
+    config.set('server', 'stratum_tcp_ssl_port', '40002')
+    config.set('server', 'stratum_http_ssl_port', '7082')
+    config.set('server', 'report_stratum_tcp_port', '')
+    config.set('server', 'report_stratum_http_port', '')
+    config.set('server', 'report_stratum_tcp_ssl_port', '')
+    config.set('server', 'report_stratum_http_ssl_port', '')
+    config.set('server', 'ssl_certfile', '')
+    config.set('server', 'ssl_keyfile', '')
+    config.set('server', 'password', '')
+    config.set('server', 'irc', 'no')
+    config.set('server', 'irc_nick', '')
+    config.set('server', 'novacoin', '')
+    config.set('server', 'datadir', '')
+
+    # use leveldb as default
+    config.set('server', 'backend', 'leveldb')
+    config.add_section('leveldb')
+    config.set('leveldb', 'path_fulltree', '/dev/shm/electrum_db')
+    config.set('leveldb', 'pruning_limit', '100')
+
+    for path in ('/etc/', ''):
+        filename = path + 'electrum.conf'
+        attempt_read_config(config, filename)
+
+    try:
+        with open('/etc/electrum.banner', 'r') as f:
+            config.set('server', 'banner', f.read())
+    except IOError:
+        pass
+
+    return config
+
+
+def run_rpc_command(params):
+    cmd = params[0]
+    import xmlrpclib
+    server = xmlrpclib.ServerProxy('http://localhost:7000')
+    func = getattr(server, cmd)
+    r = func(*params[1:])
+
+    if cmd == 'info':
+        now = time.time()
+        print 'type           address         sub  version  time'
+        for item in r:
+            print '%4s   %21s   %3s  %7s  %.2f' % (item.get('name'),
+                                                   item.get('address'),
+                                                   item.get('subscriptions'),
+                                                   item.get('version'),
+                                                   (now - item.get('time')),
+                                                   )
+    else:
+        print r
+
+
+def cmd_info():
+    return map(lambda s: {"time": s.time,
+                          "name": s.name,
+                          "address": s.address,
+                          "version": s.version,
+                          "subscriptions": len(s.subscriptions)},
+               dispatcher.request_dispatcher.get_sessions())
+
+def cmd_debug(s):
+    if s:
+        from guppy import hpy
+        h = hpy()
+        bp = dispatcher.request_dispatcher.processors['blockchain']
+        try:
+            result = str(eval(s))
+        except:
+            result = "error"
+        return result
+
+
+def get_port(config, name):
+    try:
+        return config.getint('server', name)
+    except:
+        return None
 
-config = ConfigParser.ConfigParser()
-# set some defaults, which will be overwritten by the config file
-config.add_section('server')
-config.set('server','banner', 'Welcome to Electrum!')
-config.set('server', 'host', 'localhost')
-config.set('server', 'port', '50000')
-config.set('server', 'password', '')
-config.set('server', 'irc', 'yes')
-config.set('server', 'ircname', 'Electrum server')
-config.add_section('database')
-config.set('database', 'type', 'psycopg2')
-config.set('database', 'database', 'abe')
-
-try:
-    f = open('/etc/electrum.conf','r')
-    config.readfp(f)
-    f.close()
-except:
-    print "Could not read electrum.conf. I will use the default values."
-
-try:
-    f = open('/etc/electrum.banner','r')
-    config.set('server','banner', f.read())
-    f.close()
-except:
-    pass
-
-
-password = config.get('server','password')
-
-
-from processor import Shared, Processor, Dispatcher
-from stratum_http import HttpServer
-from stratum import TcpServer
-from native import NativeServer
-from irc import Irc
-from abe_backend import AbeStore
-
-class AbeProcessor(Processor):
-    def process(self,request):
-        message_id = request['id']
-        method = request['method']
-        params = request.get('params',[])
-        #print request
-
-        result = ''
-        if method == 'numblocks.subscribe':
-            result = store.block_number
-        elif method == 'address.subscribe':
-            address = params[0]
-            store.watch_address(address)
-            status = store.get_status(address)
-            result = status
-        elif method == 'client.version':
-            #session.version = params[0]
-            pass
-        elif method == 'server.banner':
-            result = config.get('server','banner').replace('\\n','\n')
-        elif method == 'server.peers':
-            result = irc.get_peers()
-        elif method == 'address.get_history':
-            address = params[0]
-            result = store.get_history( address ) 
-        elif method == 'transaction.broadcast':
-            txo = store.send_tx(params[0])
-            print "sent tx:", txo
-            result = txo 
-        else:
-            print "unknown method", request
-
-        if result!='':
-            response = { 'id':message_id, 'method':method, 'params':params, 'result':result }
-            self.push_response(response)
-
-    def get_status(self,addr):
-        return store.get_status(addr)
+if __name__ == '__main__':
+    config = create_config()
+    password = config.get('server', 'password')
+    host = config.get('server', 'host')
+    stratum_tcp_port = get_port(config, 'stratum_tcp_port')
+    stratum_http_port = get_port(config, 'stratum_http_port')
+    stratum_tcp_ssl_port = get_port(config, 'stratum_tcp_ssl_port')
+    stratum_http_ssl_port = get_port(config, 'stratum_http_ssl_port')
+    ssl_certfile = config.get('server', 'ssl_certfile')
+    ssl_keyfile = config.get('server', 'ssl_keyfile')
+
+    if stratum_tcp_ssl_port or stratum_http_ssl_port:
+        assert ssl_certfile and ssl_keyfile
+
+    if len(sys.argv) > 1:
+        try:
+            run_rpc_command(sys.argv[1:])
+        except socket.error:
+            print "server not running"
+            sys.exit(1)
+        sys.exit(0)
 
+    try:
+        run_rpc_command(['getpid'])
+        is_running = True
+    except socket.error:
+        is_running = False
 
+    if is_running:
+        print "server already running"
+        sys.exit(1)
 
 
+    from processor import Dispatcher, print_log
+    from backends.irc import ServerProcessor
+    from transports.stratum_tcp import TcpServer
+    from transports.stratum_http import HttpServer
 
+    backend_name = config.get('server', 'backend')
+    if backend_name == 'libbitcoin':
+        from backends.libbitcoin import BlockchainProcessor
+    elif backend_name == 'leveldb':
+        from backends.bitcoind import BlockchainProcessor
+    else:
+        print "Unknown backend '%s' specified\n" % backend_name
+        sys.exit(1)
 
-if __name__ == '__main__':
+    print "\n\n\n\n\n"
+    print_log("Starting Electrum server on", host)
 
-    if len(sys.argv)>1:
-        import jsonrpclib
-        server = jsonrpclib.Server('http://%s:8081'%config.get('server','host'))
-        cmd = sys.argv[1]
-        if cmd == 'stop':
-            out = server.stop(password)
-        else:
-            out = "Unknown command: '%s'" % cmd
-        print out
-        sys.exit(0)
+    # Create hub
+    dispatcher = Dispatcher(config)
+    shared = dispatcher.shared
 
-    processor = AbeProcessor()
-    shared = Shared()
-    # Bind shared to processor since constructor is user defined
-    processor.shared = shared
-    processor.start()
+    # handle termination signals
+    import signal
+    def handler(signum = None, frame = None):
+        print_log('Signal handler called with signal', signum)
+        shared.stop()
+    for sig in [signal.SIGTERM, signal.SIGHUP, signal.SIGQUIT]:
+        signal.signal(sig, handler)
 
-    irc = Irc(processor, config.get('server','host'), config.get('server','ircname'))
-    if (config.get('server','irc') == 'yes' ): irc.start()
 
-    # backend
-    store = AbeStore(config)
+    # Create and register processors
+    chain_proc = BlockchainProcessor(config, shared)
+    dispatcher.register('blockchain', chain_proc)
 
-    # dispatcher
-    dispatcher = Dispatcher(shared, processor)
-    dispatcher.start()
+    server_proc = ServerProcessor(config)
+    dispatcher.register('server', server_proc)
 
-    host = config.get('server','host')
+    transports = []
     # Create various transports we need
-    transports = [ NativeServer(shared, store, irc, config.get('server','banner'), host, 50000),
-                   TcpServer(shared, processor, host, 50001),
-                   HttpServer(shared, processor, host, 8081),
-                   ]
+    if stratum_tcp_port:
+        tcp_server = TcpServer(dispatcher, host, stratum_tcp_port, False, None, None)
+        transports.append(tcp_server)
+
+    if stratum_tcp_ssl_port:
+        tcp_server = TcpServer(dispatcher, host, stratum_tcp_ssl_port, True, ssl_certfile, ssl_keyfile)
+        transports.append(tcp_server)
+
+    if stratum_http_port:
+        http_server = HttpServer(dispatcher, host, stratum_http_port, False, None, None)
+        transports.append(http_server)
+
+    if stratum_http_ssl_port:
+        http_server = HttpServer(dispatcher, host, stratum_http_ssl_port, True, ssl_certfile, ssl_keyfile)
+        transports.append(http_server)
+
     for server in transports:
         server.start()
 
-    print "starting Electrum server on", host
-    store.run(processor)
-    print "server stopped"
-
+    
+
+    from SimpleXMLRPCServer import SimpleXMLRPCServer
+    server = SimpleXMLRPCServer(('localhost',7000), allow_none=True, logRequests=False)
+    server.register_function(lambda: os.getpid(), 'getpid')
+    server.register_function(shared.stop, 'stop')
+    server.register_function(cmd_info, 'info')
+    server.register_function(cmd_debug, 'debug')
+    server.socket.settimeout(1)
+    while not shared.stopped():
+        try:
+            server.handle_request()
+        except socket.timeout:
+            continue
+        except:
+            shared.stop()
+
+    server_proc.join()
+    chain_proc.join()
+    print_log("Electrum Server stopped")