X-Git-Url: https://git.novaco.in/?a=blobdiff_plain;f=server.py;h=973f932f8edca0f92548ff6975da24843491c1ea;hb=2b521da6a9f9fb386706d86a3d412986c58eba18;hp=9e5a29bf447d26aa0594db1cbfac93b818f5d3b0;hpb=1b56844f8ca7a7b2758e74452ee3d13636813d96;p=electrum-server.git diff --git a/server.py b/server.py index 9e5a29b..973f932 100755 --- a/server.py +++ b/server.py @@ -20,18 +20,27 @@ Todo: * server should check and return bitcoind status.. * improve txpoint sorting * command to check cache + + mempool transactions do not need to be added to the database; it slows it down """ import time, socket, operator, thread, ast, sys,re import psycopg2, binascii -import bitcoinrpc from Abe.abe import hash_to_address, decode_check_address from Abe.DataStore import DataStore as Datastore_class from Abe import DataStore, readconf, BCDataStream, deserialize, util, base58 import ConfigParser +from json import dumps, loads +import urllib + +# we need to import electrum +sys.path.append('../client/') +from wallet import Wallet +from interface import Interface + config = ConfigParser.ConfigParser() # set some defaults, which will be overwritten by the config file @@ -54,13 +63,16 @@ try: except: print "Could not read electrum.conf. I will use the default values." +password = config.get('server','password') +bitcoind_url = 'http://%s:%s@%s:%s/' % ( config.get('bitcoind','user'), config.get('bitcoind','password'), config.get('bitcoind','host'), config.get('bitcoind','port')) + stopping = False block_number = -1 sessions = {} -sessions_last_time = {} dblock = thread.allocate_lock() peer_list = {} +wallets = {} # for ultra-light clients such as bccapi class MyStore(Datastore_class): @@ -74,14 +86,14 @@ class MyStore(Datastore_class): _hash = store.binout(row[6]) address = hash_to_address(chr(0), _hash) if self.tx_cache.has_key(address): - print "cache: invalidating", address, self.ismempool + print "cache: invalidating", address self.tx_cache.pop(address) outrows = self.get_tx_outputs(txid, False) for row in outrows: _hash = store.binout(row[6]) address = hash_to_address(chr(0), _hash) if self.tx_cache.has_key(address): - print "cache: invalidating", address, self.ismempool + print "cache: invalidating", address self.tx_cache.pop(address) def safe_sql(self,sql, params=(), lock=True): @@ -195,7 +207,7 @@ class MyStore(Datastore_class): JOIN pubkey ON (pubkey.pubkey_id = txout.pubkey_id) WHERE pubkey.pubkey_hash = ? """, (dbhash,)) - def get_txpoints(self, addr): + def get_history(self, addr): if config.get('server','cache') == 'yes': cached_version = self.tx_cache.get( addr ) @@ -204,7 +216,8 @@ class MyStore(Datastore_class): version, binaddr = decode_check_address(addr) if binaddr is None: - return "err" + return None + dbhash = self.binin(binaddr) rows = [] rows += self.get_address_out_rows( dbhash ) @@ -251,14 +264,20 @@ class MyStore(Datastore_class): if tx_hash in known_tx: continue + # this means that pending transactions were added to the db, even if they are not returned by getmemorypool address_has_mempool = True + + # this means pending transactions are returned by getmemorypool + if tx_hash not in self.mempool_keys: + continue + #print "mempool", tx_hash txpoint = { "nTime": 0, #"chain_id": 1, "height": 0, "is_in": int(is_in), - "blk_hash": 'mempool', + "blk_hash": 'mempool', "tx_hash": tx_hash, "tx_id": int(tx_id), "pos": int(pos), @@ -305,25 +324,125 @@ class MyStore(Datastore_class): return txpoints - def get_status(self, addr): - # last block for an address. - tx_points = self.get_txpoints(addr) - if not tx_points: - return None - else: - return tx_points[-1]['blk_hash'] + +class Direct_Interface(Interface): + def __init__(self): + pass + + def handler(self, method, params = ''): + cmds = {'session.new':new_session, + 'session.poll':poll_session, + 'session.update':update_session, + 'blockchain.transaction.broadcast':send_tx, + 'blockchain.address.get_history':store.get_history + } + func = cmds[method] + return func( params ) + def send_tx(tx): - import bitcoinrpc - conn = bitcoinrpc.connect_to_local() - try: - v = conn.importtransaction(tx) - except: - v = "error: transaction rejected by memorypool" - return v + postdata = dumps({"method": 'importtransaction', 'params': [tx], 'id':'jsonrpc'}) + respdata = urllib.urlopen(bitcoind_url, postdata).read() + r = loads(respdata) + if r['error'] != None: + out = "error: transaction rejected by memorypool" + else: + out = r['result'] + return out + +def random_string(N): + import random, string + return ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(N)) + + + +def cmd_stop(data): + global stopping + if password == data: + stopping = True + return 'ok' + else: + return 'wrong password' + +def cmd_load(pw): + if password == pw: + return repr( len(sessions) ) + else: + return 'wrong password' + + +def clear_cache(pw): + if password == pw: + store.tx_cache = {} + return 'ok' + else: + return 'wrong password' + +def get_cache(pw,addr): + if password == pw: + return store.tx_cache.get(addr) + else: + return 'wrong password' + + +def poll_session(session_id): + session = sessions.get(session_id) + if session is None: + print time.asctime(), "session not found", session_id + out = repr( (-1, {})) + else: + t1 = time.time() + addresses = session['addresses'] + session['last_time'] = time.time() + ret = {} + k = 0 + for addr in addresses: + if store.tx_cache.get( addr ) is not None: k += 1 + + # get addtess status, i.e. the last block for that address. + tx_points = store.get_history(addr) + if not tx_points: + status = None + else: + lastpoint = tx_points[-1] + status = lastpoint['blk_hash'] + # this is a temporary hack; move it up once old clients have disappeared + if status == 'mempool' and session['version'] != "old": + status = status + ':%d'% len(tx_points) + + last_status = addresses.get( addr ) + if last_status != status: + addresses[addr] = status + ret[addr] = status + if ret: + sessions[session_id]['addresses'] = addresses + out = repr( (block_number, ret ) ) + t2 = time.time() - t1 + if t2 > 10: + print "high load:", session_id, "%d/%d"%(k,len(addresses)), t2 + + return out + + +def new_session(version, addresses): + session_id = random_string(10) + sessions[session_id] = { 'addresses':{}, 'version':version } + for a in addresses: + sessions[session_id]['addresses'][a] = '' + out = repr( (session_id, config.get('server','banner').replace('\\n','\n') ) ) + sessions[session_id]['last_time'] = time.time() + return out + +def update_session(session_id,addresses): + sessions[session_id]['addresses'] = {} + for a in addresses: + sessions[session_id]['addresses'][a] = '' + sessions[session_id]['last_time'] = time.time() + return 'ok' + def listen_thread(store): s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) @@ -333,9 +452,7 @@ def listen_thread(store): conn, addr = s.accept() thread.start_new_thread(client_thread, (addr, conn,)) -def random_string(N): - import random, string - return ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(N)) + def client_thread(ipaddr,conn): #print "client thread", ipaddr @@ -347,143 +464,152 @@ def client_thread(ipaddr,conn): msg += d if not d: break - if d[-1]=='#': + if '#' in msg: + msg = msg.split('#', 1)[0] break - - #print msg - try: - cmd, data = ast.literal_eval(msg[:-1]) + cmd, data = ast.literal_eval(msg) except: - print "syntax error", repr(msg) + print "syntax error", repr(msg), ipaddr conn.close() return - if cmd=='b': - out = "%d"%block_number - elif cmd=='session': - session_id = random_string(10) + out = do_command(cmd, data, ipaddr) + if out: + #print ipaddr, cmd, len(out) try: - addresses = ast.literal_eval(data) + conn.send(out) except: - print "error" - conn.close() - return - - print time.asctime(), "session", ipaddr, session_id, addresses[0] if addresses else addresses, len(addresses) - - sessions[session_id] = {} - for a in addresses: - sessions[session_id][a] = '' - out = repr( (session_id, config.get('server','banner').replace('\\n','\n') ) ) - sessions_last_time[session_id] = time.time() - - elif cmd=='poll': - session_id = data - addresses = sessions.get(session_id) - if addresses is None: - print time.asctime(), " Session not found", session_id, ipaddr - out = repr( (-1, {})) - else: - t1 = time.time() - sessions_last_time[session_id] = time.time() - ret = {} - k = 0 - for addr in addresses: - if store.tx_cache.get( addr ) is not None: k += 1 - status = store.get_status( addr ) - last_status = addresses.get( addr ) - if last_status != status: - addresses[addr] = status - ret[addr] = status - if ret: - sessions[session_id] = addresses - out = repr( (block_number, ret ) ) - t2 = time.time() - t1 - if t2 > 10: - print "high load:", session_id, "%d/%d"%(k,len(addresses)), t2 - - elif cmd == 'h': - # history - address = data - out = repr( store.get_txpoints( address ) ) - - elif cmd == 'load': - if config.get('server','password') == data: - out = repr( len(sessions) ) - else: - out = 'wrong password' + print "error, could not send" - elif cmd =='tx': - out = send_tx(data) + finally: + conn.close() - elif cmd =='clear_cache': - if config.get('server','password') == data: - self.tx_cache = {} - out = 'ok' - else: - out = 'wrong password' - elif cmd =='get_cache': - try: - pw, addr = data - except: - addr = None - if addr: - if config.get('server','password') == pw: - out = store.tx_cache.get(addr) - out = repr(out) - else: - out = 'wrong password' - else: - out = "error: "+ repr(data) +def do_command(cmd, data, ipaddr): - elif cmd == 'stop': - global stopping - if config.get('server','password') == data: - stopping = True - out = 'ok' - else: - out = 'wrong password' + timestr = time.strftime("[%d/%m/%Y-%H:%M:%S]") - elif cmd == 'peers': - out = repr(peer_list.values()) + if cmd=='b': + out = "%d"%block_number + + elif cmd in ['session','new_session']: + try: + if cmd == 'session': + addresses = ast.literal_eval(data) + version = "old" + else: + version, addresses = ast.literal_eval(data) + if version[0]=="0": version = "v" + version + except: + print "error", data + return None + print timestr, "new session", ipaddr, addresses[0] if addresses else addresses, len(addresses), version + out = new_session(version, addresses) + elif cmd=='update_session': + try: + session_id, addresses = ast.literal_eval(data) + except: + print "error" + return None + print timestr, "update session", ipaddr, addresses[0] if addresses else addresses, len(addresses) + out = update_session(session_id,addresses) + + elif cmd == 'bccapi_login': + import electrum + print "data",data + v, k = ast.literal_eval(data) + master_public_key = k.decode('hex') # todo: sanitize. no need to decode twice... + print master_public_key + wallet_id = random_string(10) + w = Wallet( Direct_Interface() ) + w.master_public_key = master_public_key.decode('hex') + w.synchronize() + wallets[wallet_id] = w + out = wallet_id + print "wallets", wallets + + elif cmd == 'bccapi_getAccountInfo': + from wallet import int_to_hex + v, wallet_id = ast.literal_eval(data) + w = wallets.get(wallet_id) + if w is not None: + num = len(w.addresses) + c, u = w.get_balance() + out = int_to_hex(num,4) + int_to_hex(c,8) + int_to_hex( c+u, 8 ) + out = out.decode('hex') + else: + print "error",data + out = "error" + + elif cmd == 'bccapi_getAccountStatement': + from wallet import int_to_hex + v, wallet_id = ast.literal_eval(data) + w = wallets.get(wallet_id) + if w is not None: + num = len(w.addresses) + c, u = w.get_balance() + total_records = num_records = 0 + out = int_to_hex(num,4) + int_to_hex(c,8) + int_to_hex( c+u, 8 ) + int_to_hex( total_records ) + int_to_hex( num_records ) + out = out.decode('hex') else: - out = None + print "error",data + out = "error" - if out: - #print ipaddr, cmd, len(out) - try: - conn.send(out) - except: - print "error, could not send" + elif cmd == 'bccapi_getSendCoinForm': + out = '' - finally: - conn.close() - + elif cmd == 'bccapi_submitTransaction': + out = '' + + elif cmd=='poll': + out = poll_session(data) + + elif cmd == 'h': + # history + address = data + out = repr( store.get_history( address ) ) + + elif cmd == 'load': + out = cmd_load(data) + + elif cmd =='tx': + out = send_tx(data) + print timestr, "sent tx:", ipaddr, out + + elif cmd == 'stop': + out = cmd_stop(data) + + elif cmd == 'peers': + out = repr(peer_list.values()) -ds = BCDataStream.BCDataStream() + else: + out = None + + return out def memorypool_update(store): - conn = bitcoinrpc.connect_to_local() - try: - v = conn.getmemorypool() - except: - print "cannot contact bitcoin daemon" + ds = BCDataStream.BCDataStream() + store.mempool_keys = [] + + postdata = dumps({"method": 'getmemorypool', 'params': [], 'id':'jsonrpc'}) + respdata = urllib.urlopen(bitcoind_url, postdata).read() + r = loads(respdata) + if r['error'] != None: return - v = v['transactions'] + + v = r['result'].get('transactions') for hextx in v: ds.clear() ds.write(hextx.decode('hex')) tx = deserialize.parse_Transaction(ds) - #print "new tx",tx - tx['hash'] = util.double_sha256(tx['tx']) - + tx_hash = tx['hash'][::-1].encode('hex') + store.mempool_keys.append(tx_hash) if store.tx_find_id_and_value(tx): pass else: @@ -493,16 +619,14 @@ def memorypool_update(store): - def clean_session_thread(): while not stopping: time.sleep(30) t = time.time() - for k,t0 in sessions_last_time.items(): + for k,s in sessions.items(): + t0 = s['last_time'] if t - t0 > 5*60: - print time.asctime(), "lost session",k sessions.pop(k) - sessions_last_time.pop(k) def irc_thread(): @@ -547,37 +671,52 @@ def irc_thread(): s.close() + +def jsonrpc_thread(store): + # see http://code.google.com/p/jsonrpclib/ + from SocketServer import ThreadingMixIn + from jsonrpclib.SimpleJSONRPCServer import SimpleJSONRPCServer + class SimpleThreadedJSONRPCServer(ThreadingMixIn, SimpleJSONRPCServer): pass + server = SimpleThreadedJSONRPCServer(( config.get('server','host'), 8081)) + server.register_function(lambda : peer_list.values(), 'peers') + server.register_function(cmd_stop, 'stop') + server.register_function(cmd_load, 'load') + server.register_function(lambda : block_number, 'blocks') + server.register_function(clear_cache, 'clear_cache') + server.register_function(get_cache, 'get_cache') + server.register_function(send_tx, 'blockchain.transaction.broadcast') + server.register_function(store.get_history, 'blockchain.address.get_history') + server.register_function(new_session, 'session.new') + server.register_function(update_session, 'session.update') + server.register_function(poll_session, 'session.poll') + server.serve_forever() + + import traceback 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 == 'load': - request = "('load','%s')#"%config.get('server','password') + out = server.load(password) elif cmd == 'peers': - request = "('peers','')#" + out = server.peers() elif cmd == 'stop': - request = "('stop','%s')#"%config.get('server','password') + out = server.stop(password) elif cmd == 'clear_cache': - request = "('clear_cache','%s')#"%config.get('server','password') + out = server.clear_cache(password) elif cmd == 'get_cache': - request = "('get_cache',('%s','%s'))#"%(config.get('server','password'),sys.argv[2]) + out = server.get_cache(password,sys.argv[2]) elif cmd == 'h': - request = "('h','%s')#"%sys.argv[2] + out = server.blockchain.address.get_history(sys.argv[2]) + elif cmd == 'tx': + out = server.blockchain.transaction.broadcast(sys.argv[2]) elif cmd == 'b': - request = "('b','')#" - - s = socket.socket( socket.AF_INET, socket.SOCK_STREAM) - s.connect((config.get('server','host'), config.getint('server','port'))) - s.send( request ) - out = '' - while 1: - msg = s.recv(1024) - if msg: out += msg - else: break - s.close() + out = server.blocks() print out sys.exit(0) @@ -596,9 +735,10 @@ if __name__ == '__main__': args.connect_args = { 'database' : config.get('database','database') } store = MyStore(args) store.tx_cache = {} - store.ismempool = False + store.mempool_keys = {} thread.start_new_thread(listen_thread, (store,)) + thread.start_new_thread(jsonrpc_thread, (store,)) thread.start_new_thread(clean_session_thread, ()) if (config.get('server','irc') == 'yes' ): thread.start_new_thread(irc_thread, ()) @@ -607,13 +747,16 @@ if __name__ == '__main__': try: dblock.acquire() store.catch_up() - store.ismempool = True memorypool_update(store) - store.ismempool = False block_number = store.get_block_number(1) - dblock.release() + except IOError: + print "IOError: cannot reach bitcoind" + block_number = -1 except: traceback.print_exc(file=sys.stdout) + block_number = -1 + finally: + dblock.release() time.sleep(10) print "server stopped"