* 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
+import bitcoin
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.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 send_tx(tx):
- import bitcoinrpc
- conn = bitcoinrpc.connect_to_local()
+ import bitcoin
+ conn = bitcoin.connect_to_local()
try:
v = conn.importtransaction(tx)
except:
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
try:
ipaddr = ipaddr[0]
msg = ''
while 1:
d = conn.recv(1024)
- if d: msg+=d
- try:
- cmd, data = ast.literal_eval(msg)
+ msg += d
+ if d[-1]=='#':
break
- except:
- continue
-
+
+ #print msg
+
+ try:
+ cmd, data = ast.literal_eval(msg[:-1])
+ except:
+ print "syntax error", repr(msg)
+ conn.close()
+ return
if cmd=='b':
- out = "%d"%store.get_block_number(1)
+ out = "%d"%block_number
+ elif cmd=='session':
+ session_id = random_string(10)
+ try:
+ addresses = ast.literal_eval(data)
+ except:
+ print "error"
+ conn.close()
+ return
- elif cmd=='watch':
- addresses = ast.literal_eval(data)
- sessions[ipaddr] = {}
+ print time.asctime(), "session", ipaddr, session_id, addresses[0], len(addresses)
+
+ sessions[session_id] = {}
for a in addresses:
- sessions[ipaddr][a] = ''
- out = SERVER_MESSAGE
+ sessions[session_id][a] = ''
+ out = repr( (session_id, config.get('server','banner')) )
+ sessions_last_time[session_id] = time.time()
elif cmd=='poll':
- addresses = sessions.get(ipaddr)
+ session_id = data
+ addresses = sessions.get(session_id)
if not addresses:
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 )
- last_status = sessions[ipaddr].get( addr )
+ last_status = sessions[session_id].get( addr )
if last_status != status:
- sessions[ipaddr][addr] = 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
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
def memorypool_update(store):
- conn = bitcoinrpc.connect_to_local()
+ conn = bitcoin.connect_to_local()
try:
v = conn.getmemorypool()
except:
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')
+ 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, ())
+ 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"
+