zebra.lucky
/
electrum
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

separate commands from transport

master
ThomasV 14 years ago
parent
c8ddf49833
commit
789639ff74
1 changed files with 171 additions and 146 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 317
      server/server.py

317
server/server.py
Unescape View File

@ -358,171 +358,196 @@ def client_thread(ipaddr,conn):
conn.close() conn.close()
return return
if cmd=='b': out = do_command(cmd, data, ipaddr)
out = "%d"%block_number if out:
#print ipaddr, cmd, len(out)
elif cmd in ['session','new_session']:
session_id = random_string(10)
try: try:
if cmd == 'session': conn.send(out)
addresses = ast.literal_eval(data)
version = "old"
else:
version, addresses = ast.literal_eval(data)
if version[0]=="0": version = "v" + version
except: except:
print "error", data print "error, could not send"
conn.close()
return
print time.strftime("[%d/%m/%Y-%H:%M:%S]"), "new session", ipaddr, addresses[0] if addresses else addresses, len(addresses), version finally:
conn.close()
sessions[session_id] = { 'addresses':{}, 'version':version, 'ip':ipaddr }
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()
elif cmd=='update_session': def do_command(cmd, data, ipaddr):
try:
session_id, addresses = ast.literal_eval(data)
except:
print "error"
conn.close()
return
print time.strftime("[%d/%m/%Y-%H:%M:%S]"), "update session", ipaddr, addresses[0] if addresses else addresses, len(addresses) if cmd=='b':
out = "%d"%block_number
sessions[session_id]['addresses'] = {} elif cmd in ['session','new_session']:
for a in addresses: session_id = random_string(10)
sessions[session_id]['addresses'][a] = '' try:
out = 'ok' if cmd == 'session':
sessions[session_id]['last_time'] = time.time() addresses = ast.literal_eval(data)
version = "old"
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 = electrum.Wallet()
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':
import electrum
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 = electrum.int_to_hex(num,4) + electrum.int_to_hex(c,8) + electrum.int_to_hex( c+u, 8 )
out = out.decode('hex')
else:
print "error",data
out = "error"
elif cmd=='poll':
session_id = data
session = sessions.get(session_id)
if session is None:
print time.asctime(), "session not found", session_id, ipaddr
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
elif cmd == 'h':
# history
address = data
out = repr( store.get_history( address ) )
elif cmd == 'load':
if config.get('server','password') == data:
out = repr( len(sessions) )
else: else:
out = 'wrong password' version, addresses = ast.literal_eval(data)
if version[0]=="0": version = "v" + version
except:
print "error", data
return None
elif cmd =='tx': print time.strftime("[%d/%m/%Y-%H:%M:%S]"), "new session", ipaddr, addresses[0] if addresses else addresses, len(addresses), version
out = send_tx(data)
print "sent tx:", out
elif cmd =='clear_cache': sessions[session_id] = { 'addresses':{}, 'version':version, 'ip':ipaddr }
if config.get('server','password') == data: for a in addresses:
store.tx_cache = {} sessions[session_id]['addresses'][a] = ''
out = 'ok' out = repr( (session_id, config.get('server','banner').replace('\\n','\n') ) )
else: sessions[session_id]['last_time'] = time.time()
out = 'wrong password'
elif cmd =='get_cache': elif cmd=='update_session':
try: try:
pw, addr = data session_id, addresses = ast.literal_eval(data)
except: except:
addr = None print "error"
if addr: return None
if config.get('server','password') == pw:
out = store.tx_cache.get(addr) print time.strftime("[%d/%m/%Y-%H:%M:%S]"), "update session", ipaddr, addresses[0] if addresses else addresses, len(addresses)
out = repr(out)
sessions[session_id]['addresses'] = {}
for a in addresses:
sessions[session_id]['addresses'][a] = ''
out = 'ok'
sessions[session_id]['last_time'] = time.time()
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 = electrum.Wallet()
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 electrum 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 electrum 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:
print "error",data
out = "error"
elif cmd == 'bccapi_getSendCoinForm':
out = ''
elif cmd == 'bccapi_submitTransaction':
out = ''
elif cmd=='poll':
session_id = data
session = sessions.get(session_id)
if session is None:
print time.asctime(), "session not found", session_id, ipaddr
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: else:
out = 'wrong password' lastpoint = tx_points[-1]
else: status = lastpoint['blk_hash']
out = "error: "+ repr(data) # 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
elif cmd == 'h':
# history
address = data
out = repr( store.get_history( address ) )
elif cmd == 'load':
if config.get('server','password') == data:
out = repr( len(sessions) )
else:
out = 'wrong password'
elif cmd == 'stop': elif cmd =='tx':
global stopping out = send_tx(data)
if config.get('server','password') == data: print "sent tx:", out
stopping = True
out = 'ok' elif cmd =='clear_cache':
if config.get('server','password') == data:
store.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: else:
out = 'wrong password' out = 'wrong password'
else:
out = "error: "+ repr(data)
elif cmd == 'peers': elif cmd == 'stop':
out = repr(peer_list.values()) global stopping
if config.get('server','password') == data:
stopping = True
out = 'ok'
else: else:
out = None out = 'wrong password'
if out: elif cmd == 'peers':
#print ipaddr, cmd, len(out) out = repr(peer_list.values())
try:
conn.send(out) else:
except: out = None
print "error, could not send"
return out
finally:
conn.close()

Write Preview
Loading…
Cancel
Save