X-Git-Url: https://git.novaco.in/?a=blobdiff_plain;f=server.py;h=5846fba7748832ac8af900659e3017c0661a2b06;hb=3051f46df11fc6c9ebe321c9ef6d726e65310129;hp=675ef5f650c24b2a5ba434f89d1f0e8b4db622ad;hpb=23de59ae6d8da801586cb2992a76e651fa72c0c4;p=electrum-server.git diff --git a/server.py b/server.py index 675ef5f..5846fba 100755 --- a/server.py +++ b/server.py @@ -15,133 +15,117 @@ # License along with this program. If not, see # . -import time, json, socket, operator, thread, ast, sys, re, traceback +import time, sys, traceback import ConfigParser -from json import dumps, loads -import urllib -import threading - -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) - - - - +import logging +logging.basicConfig() + +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', 'stratum_tcp_port', '50001') + config.set('server', 'stratum_http_port', '8081') + config.set('server', 'password', '') + config.set('server', 'irc', 'yes') + config.set('server', 'irc_nick', '') + config.add_section('database') + config.set('database', 'type', 'psycopg2') + config.set('database', 'database', 'abe') + config.set('server', 'backend', 'abe') + + 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(command, stratum_tcp_port): + import socket, json + s = socket.socket( socket.AF_INET, socket.SOCK_STREAM ) + s.connect(( host, int(stratum_tcp_port ))) + method = 'server.' + command + request = json.dumps( { 'id':0, 'method':method, 'params':[password] } ) + s.send(request + '\n') + msg = '' + while True: + o = s.recv(1024) + msg += o + if msg.find('\n') != -1: break + s.close() + r = json.loads(msg).get('result') + if command == 'stop': print r + elif command == 'info': + now = time.time() + for item in r: + print '%15s %3s %7s %.2f'%( item.get('address'), item.get('subscriptions'), item.get('version'), (now - item.get('time')) ) if __name__ == '__main__': - - 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 + config = create_config() + password = config.get('server', 'password') + host = config.get('server', 'host') + stratum_tcp_port = config.get('server', 'stratum_tcp_port') + stratum_http_port = config.get('server', 'stratum_http_port') + + if len(sys.argv) > 1: + run_rpc_command(sys.argv[1], stratum_tcp_port) sys.exit(0) - processor = AbeProcessor() - shared = Shared() - # Bind shared to processor since constructor is user defined - processor.shared = shared - processor.start() + from processor import Dispatcher + + from backends.irc import ServerProcessor + backend_name = config.get('server', 'backend') + try: + backend = __import__("backends." + backend_name, + fromlist=["BlockchainProcessor"]) + except ImportError: + sys.stderr.write("Unknown backend '%s' specified\n" % backend_name) + raise - irc = Irc(processor, config.get('server','host'), config.get('server','ircname')) - if (config.get('server','irc') == 'yes' ): irc.start() + # Create hub + dispatcher = Dispatcher() + shared = dispatcher.shared - # backend - store = AbeStore(config) + # Create and register processors + chain_proc = backend.BlockchainProcessor(config) + 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: + from transports.stratum_tcp import TcpServer + tcp_server = TcpServer(dispatcher, host, int(stratum_tcp_port)) + transports.append(tcp_server) + + if stratum_http_port: + from transports.stratum_http import HttpServer + http_server = HttpServer(dispatcher, host, int(stratum_http_port)) + transports.append(http_server) + for server in transports: server.start() - print "starting Electrum server on", host - store.run(processor) - print "server stopped" + print "Starting Electrum server on", host + while not shared.stopped(): + time.sleep(1) + print "Server stopped"