X-Git-Url: https://git.novaco.in/?a=blobdiff_plain;f=server.py;h=260ad81b498db78895da49ff01b0624da6ae340b;hb=d1e80a452945a425297dd8ac8373924d627da7aa;hp=3271a896556bfe7dfbd73f0acfba4da28cb8dea9;hpb=3e4049f8b010a7d43f052fff3d0d6fb5fefd2538;p=electrum-server.git diff --git a/server.py b/server.py old mode 100644 new mode 100755 index 3271a89..260ad81 --- a/server.py +++ b/server.py @@ -21,21 +21,8 @@ Todo: * improve txpoint sorting """ -SERVER_MESSAGE = """ -Welcome to ecdsa.org. - -This service is free. Support this node: -19mP9FKrXqL46Si58pHdhGKow88SUPy1V8 - -The server code is free software; you may -download it and operate your own Electrum -node. See http://ecdsa.org/electrum -""" - - - -import time, socket, operator, thread, ast, sys +import time, socket, operator, thread, ast, sys,re import psycopg2, binascii import bitcoinrpc @@ -43,17 +30,41 @@ 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 + +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', 'ecdsa.org') +config.set('server', 'port', 50000) +config.set('server', 'password', '') +config.set('server', 'irc', 'yes') +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 dafault values." + +stopping = False +block_number = -1 sessions = {} +sessions_last_time = {} dblock = thread.allocate_lock() +peer_list = {} class MyStore(Datastore_class): - def safe_sql(self,sql): + def safe_sql(self,sql, params=()): try: dblock.acquire() - ret = self.selectall(sql) + ret = self.selectall(sql,params) dblock.release() return ret except: @@ -113,8 +124,8 @@ class MyStore(Datastore_class): JOIN txin ON (txin.tx_id = tx.tx_id) JOIN txout prevout ON (txin.txout_id = prevout.txout_id) JOIN pubkey ON (pubkey.pubkey_id = prevout.pubkey_id) - WHERE pubkey.pubkey_hash = '%s' - AND cc.in_longest = 1"""%dbhash) + WHERE pubkey.pubkey_hash = ? + AND cc.in_longest = 1""", (dbhash,)) def get_address_out_rows_memorypool(self, dbhash): return self.safe_sql(""" SELECT @@ -127,7 +138,7 @@ class MyStore(Datastore_class): JOIN txin ON (txin.tx_id = tx.tx_id) JOIN txout prevout ON (txin.txout_id = prevout.txout_id) JOIN pubkey ON (pubkey.pubkey_id = prevout.pubkey_id) - WHERE pubkey.pubkey_hash ='%s' """%(dbhash)) + WHERE pubkey.pubkey_hash = ? """, (dbhash,)) def get_address_in_rows(self, dbhash): return self.safe_sql(""" SELECT @@ -146,8 +157,8 @@ class MyStore(Datastore_class): JOIN tx ON (tx.tx_id = block_tx.tx_id) JOIN txout ON (txout.tx_id = tx.tx_id) JOIN pubkey ON (pubkey.pubkey_id = txout.pubkey_id) - WHERE pubkey.pubkey_hash = '%s' - AND cc.in_longest = 1"""%(dbhash)) + WHERE pubkey.pubkey_hash = ? + AND cc.in_longest = 1""", (dbhash,)) def get_address_in_rows_memorypool(self, dbhash): return self.safe_sql( """ SELECT @@ -159,7 +170,7 @@ class MyStore(Datastore_class): FROM tx JOIN txout ON (txout.tx_id = tx.tx_id) JOIN pubkey ON (pubkey.pubkey_id = txout.pubkey_id) - WHERE pubkey.pubkey_hash = '%s' """%(dbhash)) + WHERE pubkey.pubkey_hash = ? """, (dbhash,)) def get_txpoints(self, addr): version, binaddr = decode_check_address(addr) @@ -278,18 +289,20 @@ def send_tx(tx): def listen_thread(store): - s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) - s.bind(('ecdsa.org', 50000)) + s.bind((config.get('server','host'), config.getint('server','port'))) s.listen(1) - while True: + while not stopping: 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 + #print "client thread", ipaddr try: ipaddr = ipaddr[0] msg = '' @@ -299,7 +312,7 @@ def client_thread(ipaddr,conn): if d[-1]=='#': break - print msg + #print msg try: cmd, data = ast.literal_eval(msg[:-1]) @@ -309,18 +322,23 @@ def client_thread(ipaddr,conn): return if cmd=='b': - out = "%d"%store.get_block_number(1) - + out = "%d"%block_number elif cmd=='session': - import random, string - session_id = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(10)) - print "new session", ipaddr, session_id + session_id = random_string(10) + try: + addresses = ast.literal_eval(data) + except: + print "error" + conn.close() + return + + print time.asctime(), "session", ipaddr, session_id, addresses[0], len(addresses) - addresses = ast.literal_eval(data) sessions[session_id] = {} for a in addresses: sessions[session_id][a] = '' - out = repr( (session_id, SERVER_MESSAGE) ) + out = repr( (session_id, config.get('server','banner').replace('\\n','\n') ) ) + sessions_last_time[session_id] = time.time() elif cmd=='poll': session_id = data @@ -329,6 +347,7 @@ def client_thread(ipaddr,conn): print "session not found", ipaddr out = repr( (-1, {})) else: + sessions_last_time[session_id] = time.time() ret = {} for addr in addresses: status = store.get_status( addr ) @@ -336,7 +355,7 @@ def client_thread(ipaddr,conn): if last_status != status: sessions[session_id][addr] = status ret[addr] = status - out = repr( (store.get_block_number(1), ret ) ) + out = repr( (block_number, ret ) ) elif cmd == 'h': # history @@ -344,9 +363,26 @@ def client_thread(ipaddr,conn): h = store.get_txpoints( addr ) out = repr(h) + elif cmd == 'load': + if config.get('server','password') == data: + out = repr( len(sessions) ) + else: + out = 'wrong password' + elif cmd =='tx': - # transaction out = send_tx(data) + + elif cmd == 'stop': + global stopping + if config.get('server','password') == data: + stopping = True + out = 'ok' + else: + out = 'wrong password' + + elif cmd == 'peers': + out = repr(peer_list.values()) + else: out = None @@ -383,29 +419,120 @@ def memorypool_update(store): pass else: store.import_tx(tx, False) - print tx['hash'][::-1].encode('hex') + #print tx['hash'][::-1].encode('hex') store.commit() + + + +def clean_session_thread(): + while not stopping: + time.sleep(30) + t = time.time() + for k,t0 in sessions_last_time.items(): + if t - t0 > 60: + print "lost session",k + sessions.pop(k) + sessions_last_time.pop(k) + + +def irc_thread(): + global peer_list + NICK = 'E_'+random_string(10) + while not stopping: + try: + s = socket.socket() + s.connect(('irc.freenode.net', 6667)) + s.send('USER '+config.get('server','host')+' '+NICK+' bla :'+NICK+'\n') + s.send('NICK '+NICK+'\n') + s.send('JOIN #electrum\n') + t = 0 + while not stopping: + line = s.recv(2048) + line = line.rstrip('\r\n') + line = line.split() + if line[0]=='PING': + s.send('PONG '+line[1]+'\n') + elif '353' in line: # answer to /names + k = line.index('353') + try: + k2 = line.index('366') + except: + continue + for item in line[k+1:k2]: + if item[0:2] == 'E_': + s.send('USERHOST %s\n'%item) + elif '302' in line: # answer to /userhost + k = line.index('302') + m = re.match( "^:(.*?)=\+~(.*?)@(.*?)$", line[k+2] ) + if m: + name = m.group(1) + host = m.group(2) + ip = m.group(3) + peer_list[name] = (ip,host) + elif time.time() - t > 5*60: + s.send('NAMES #electrum\n') + t = time.time() + except: + traceback.print_exc(file=sys.stdout) + finally: + s.close() + + import traceback if __name__ == '__main__': + if len(sys.argv)>1: + cmd = sys.argv[1] + if cmd == 'load': + request = "('load','%s')#"%config.get('server','password') + elif cmd == 'peers': + request = "('peers','')#" + elif cmd == 'stop': + request = "('stop','%s')#"%config.get('server','password') + + 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() + print out + sys.exit(0) + + + print "starting Electrum server" conf = DataStore.CONFIG_DEFAULTS args, argv = readconf.parse_argv( [], conf) - args.dbtype='psycopg2' - args.connect_args = {"database":"abe"} + args.dbtype= config.get('database','type') + if args.dbtype == 'sqlite3': + args.connect_args = { 'database' : config.get('database','database') } + elif args.dbtype == 'MySQLdb': + args.connect_args = { 'db' : config.get('database','database'), 'user' : config.get('database','user'), 'passwd' : config.get('database','pass') } + elif args.dbtype == 'psycopg2': + args.connect_args = { 'database' : config.get('database','database') } store = MyStore(args) thread.start_new_thread(listen_thread, (store,)) + thread.start_new_thread(clean_session_thread, ()) + if (config.get('server','irc') == 'yes' ): + thread.start_new_thread(irc_thread, ()) - while True: + while not stopping: try: dblock.acquire() store.catch_up() memorypool_update(store) + block_number = store.get_block_number(1) dblock.release() except: traceback.print_exc(file=sys.stdout) time.sleep(10) + print "server stopped" +