password = config.get('server','password')
stopping = False
-block_number = -1
sessions = {}
m_sessions = [{}] # served by http
sessions[session_id]['last_time'] = time.time()
ret, addresses = modified_addresses(session)
if ret: sessions[session_id]['addresses'] = addresses
- return repr( (block_number,ret))
+ return repr( (store.block_number,ret))
def poll_session_json(session_id, message_id):
sessions[session_id]['version'] = version
m_sessions[0] = sessions
-def create_session_json(_, __):
- sessions = m_sessions[0]
- session_id = random_string(10)
- print "creating session", session_id
- sessions[session_id] = { 'addresses':{}, 'numblocks':('','') }
- sessions[session_id]['last_time'] = time.time()
- m_sessions[0] = sessions
- return session_id
-
def get_banner(_,__):
import stratum
class AbeProcessor(stratum.Processor):
- def process(self,session,request):
+ def process(self,request):
message_id = request['id']
method = request['method']
params = request.get('params',[])
result = ''
if method == 'numblocks.subscribe':
- session.subscribe_to_numblocks(message_id)
- result = block_number
+ result = store.block_number
elif method == 'address.subscribe':
address = params[0]
+ store.watch_address(address)
status = store.get_status(address)
- session.subscribe_to_address(address,message_id,status)
result = status
elif method == 'client.version':
- session.version = params[0]
+ #session.version = params[0]
+ pass
elif method == 'server.banner':
result = config.get('server','banner').replace('\\n','\n')
elif method == 'server.peers':
print "unknown method", request
if result!='':
- response = { 'id':message_id, 'result':result }
- self.push_response(session,response)
+ response = { 'id':message_id, 'method':method, 'params':params, 'result':result }
+ self.push_response(response)
def get_status(self,addr):
return store.get_status(addr)
# supported protocols
thread.start_new_thread(native_server_thread, ())
-
- thread.start_new_thread(http_server_thread, ())
thread.start_new_thread(clean_session_thread, ())
- #tcp stratum
- stratum_processor = AbeProcessor()
+ #thread.start_new_thread(http_server_thread, ())
+
+
+ processor = AbeProcessor()
shared = stratum.Shared()
# Bind shared to processor since constructor is user defined
- stratum_processor.shared = shared
- stratum_processor.start()
+ processor.shared = shared
+ processor.start()
+
# Create various transports we need
- server = stratum.TcpServer(shared, stratum_processor, "ecdsa.org",50001)
+
+ #tcp stratum
+ tcpserver = stratum.TcpServer(shared, processor, "ecdsa.org",50001)
+ tcpserver.start()
+
+ #http stratum
+ from StratumJSONRPCServer import HttpServer
+ server = HttpServer(shared, processor, "ecdsa.org",8081)
server.start()
+
if (config.get('server','irc') == 'yes' ):
thread.start_new_thread(irc_thread, ())
print "starting Electrum server"
-
- old_block_number = None
- while not stopping:
- block_number = store.main_iteration()
-
- if block_number != old_block_number:
- old_block_number = block_number
- stratum_processor.update_from_blocknum(block_number)
-
- while True:
- try:
- addr = store.address_queue.get(False)
- except:
- break
-
- stratum_processor.update_from_address(addr)
-
- time.sleep(10)
+ store.run(processor)
print "server stopped"
def __init__(self):
self.shared = None
- self.lock = threading.Lock()
- self.sessions = []
threading.Thread.__init__(self)
self.daemon = True
self.request_queue = queue.Queue()
self.response_queue = queue.Queue()
+ self.id_session = {}
- def add_session(self, session):
- with self.lock:
- self.sessions.append(session)
-
- def push_response(self, session, item):
- self.response_queue.put((session,item))
+ def push_response(self, item):
+ self.response_queue.put(item)
def pop_response(self):
return self.response_queue.get()
def pop_request(self):
return self.request_queue.get()
- def collect_garbage(self):
- # Deep copy entire sessions list and blank it
- # This is done to minimise lock contention
- with self.lock:
- sessions = self.sessions[:]
- self.sessions = []
- for session in sessions:
- if not session.stopped():
- # If session is still alive then re-add it back
- # to our internal register
- self.add_session(session)
-
def run(self):
if self.shared is None:
raise TypeError("self.shared not set in Processor")
while not self.shared.stopped():
- self.collect_garbage()
session, request = self.pop_request()
- self.process(session, request)
+
+ method = request['method']
+ params = request.get('params',[])
+
+ if method == 'numblocks.subscribe':
+ session.subscribe_to_numblocks()
+
+ elif method == 'address.subscribe':
+ address = params[0]
+ session.subscribe_to_address(address)
+
+ elif method == 'server.peers':
+ session.subscribe_to_peers()
+
+ message_id = request['id']
+ self.id_session[message_id] = session
+ self.process(request)
self.stop()
def stop(self):
pass
- def process(self, session, request):
+ def process(self, request):
print "New request", request
# Do stuff...
# response = request
# When ready, you call
- # self.push_response(session,response)
+ # self.push_response(response)
- def update_from_blocknum(self,block_number):
- for session in self.sessions:
- if not session.stopped():
- if session.numblocks_sub is not None:
- response = { 'id':session.numblocks_sub, 'result':block_number }
- self.push_response(session,response)
-
- def update_from_address(self,addr):
- for session in self.sessions:
- if not session.stopped():
- m = session.addresses_sub.get(addr)
- if m:
- status = self.get_status( addr )
- message_id, last_status = m
- if status != last_status:
- session.subscribe_to_address(addr,message_id, status)
- response = { 'id':message_id, 'result':status }
- self.push_response(session,response)
-
- def get_status(self,addr):
- # return status of an address
- # return store.get_status(addr)
- pass
class Session:
else:
return self._connection
- def subscribe_to_numblocks(self,message_id):
+ def subscribe_to_numblocks(self):
with self.lock:
- self.numblocks_sub = message_id
+ self.numblocks_sub = True
- def subscribe_to_address(self,address,message_id,status):
+ def subscribe_to_peers(self):
+ pass
+
+ def subscribe_to_address(self,address):
with self.lock:
- self.addresses_sub[address] = message_id,status
+ self.addresses_sub[address] = 'unknown'
class TcpResponder(threading.Thread):
- def __init__(self, shared, processor):
+ def __init__(self, shared, processor, server):
self.shared = shared
self.processor = processor
+ self.server = server
threading.Thread.__init__(self)
+
def run(self):
while not self.shared.stopped():
- session,response = self.processor.pop_response()
- raw_response = json.dumps(response)
- # Possible race condition here by having session
- # close connection?
- # I assume Python connections are thread safe interfaces
+ response = self.processor.pop_response()
+ # if it is a subscription, find the list of sessions that suuscribed
+
+ # if there is an id, there should be a session
+ # note: I must add and remove the session id to the message id..
+
+ message_id = response.get('id')
try:
- connection = session.connection()
- connection.send(raw_response + "\n")
+ method = response['method']
except:
- session.stop()
+ print "no method", response
+ continue
+
+ if message_id:
+ session = self.processor.id_session.pop(message_id)
+ self.send_response(response, session)
+
+ elif method == 'numblocks.subscribe':
+ for session in self.server.sessions:
+ if not session.stopped():
+ if session.numblocks_sub:
+ self.send_response(response, session)
+
+ elif method == 'address.subscribe':
+ for session in self.server.sessions:
+ if not session.stopped():
+ addr = response['params'][0]
+ last_status = session.addresses_sub.get(addr)
+ if last_status:
+ new_status = response.get('result')
+ if new_status != last_status:
+ session.addresses_sub[addr] = new_status
+ self.send_response(response, session)
+ else:
+ print "error", response
+
+
+ def send_response(self, response, session):
+ raw_response = json.dumps(response)
+ # Possible race condition here by having session
+ # close connection?
+ # I assume Python connections are thread safe interfaces
+ try:
+ connection = session.connection()
+ connection.send(raw_response + "\n")
+ except:
+ session.stop()
class TcpClientRequestor(threading.Thread):
try:
command = json.loads(raw_command)
except:
- self.processor.push_response(self.session,
- {"error": "bad JSON", "request": raw_command})
+ self.processor.push_response({"error": "bad JSON", "request": raw_command})
return True
try:
method = command['method']
except KeyError:
# Return an error JSON in response.
- self.processor.push_response(self.session,
- {"error": "syntax error", "request": raw_command})
+ self.processor.push_response({"error": "syntax error", "request": raw_command})
else:
self.processor.push_request(self.session,command)
def __init__(self, shared, processor, host, port):
self.shared = shared
self.processor = processor
- self.clients = []
+ self.sessions = []
threading.Thread.__init__(self)
self.daemon = True
self.host = host
self.port = port
+ self.lock = threading.Lock()
def run(self):
print "TCP server started."
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((self.host, self.port))
sock.listen(1)
- responder = TcpResponder(self.shared, self.processor)
+ responder = TcpResponder(self.shared, self.processor, self)
responder.start()
while not self.shared.stopped():
session = Session(*sock.accept())
client_req = TcpClientRequestor(self.shared, self.processor, session)
client_req.start()
- self.processor.add_session(session)
+ self.add_session(session)
+ self.collect_garbage()
+
+ def add_session(self, session):
+ with self.lock:
+ self.sessions.append(session)
+
+ def collect_garbage(self):
+ # Deep copy entire sessions list and blank it
+ # This is done to minimise lock contention
+ with self.lock:
+ sessions = self.sessions[:]
+ self.sessions = []
+ for session in sessions:
+ if not session.stopped():
+ # If session is still alive then re-add it back
+ # to our internal register
+ self.add_session(session)
+
class Shared:
git://git.novaco.in / electrum-server.git / commitdiff