p2pool/bitcoin/worker_interface.py

   1 from __future__ import division
   2
   3 import base64
   4 import random
   5
   6 from twisted.internet import defer
   7
   8 import p2pool
   9 from p2pool import data as p2pool_data
  10 from p2pool.util import jsonrpc, variable
  11 from p2pool.bitcoin import getwork
  12
  13 def get_username(request):
  14     try:
  15         return base64.b64decode(request.getHeader('Authorization').split(' ', 1)[1]).split(':')[0]
  16     except: # XXX
  17         return None
  18
  19 def get_id(request):
  20     return request.getClientIP(), request.getHeader('Authorization')
  21
  22 class LongPollingWorkerInterface(jsonrpc.Server):
  23     def __init__(self, parent):
  24         jsonrpc.Server.__init__(self)
  25         self.parent = parent
  26
  27     def rpc_getwork(self, request, data=None):
  28         return self.parent.getwork(request, data, long_poll=True)
  29
  30 class WorkerInterface(jsonrpc.Server):
  31     def __init__(self, compute, response_callback, new_work_event=variable.Event()):
  32         jsonrpc.Server.__init__(self)
  33
  34         self.compute = compute
  35         self.response_callback = response_callback
  36         self.new_work_event = new_work_event
  37
  38         self.worker_views = {}
  39
  40         self.putChild('long-polling', LongPollingWorkerInterface(self))
  41         self.putChild('', self)
  42
  43     def rpc_getwork(self, request, data=None):
  44         return self.getwork(request, data, long_poll=False)
  45
  46     @defer.inlineCallbacks
  47     def getwork(self, request, data, long_poll):
  48         request.setHeader('X-Long-Polling', '/long-polling')
  49         request.setHeader('X-Roll-NTime', 'expire=60')
  50
  51         if data is not None:
  52             defer.returnValue(self.response_callback(getwork.decode_data(data), request))
  53
  54         request_id = get_id(request)
  55
  56         if p2pool.DEBUG:
  57             id = random.randrange(1000, 10000)
  58             print 'POLL %i START long_poll=%r user_agent=%r x-work-identifier=%r user=%r' % (id, long_poll, request.getHeader('User-Agent'), request.getHeader('X-Work-Identifier'), get_username(request))
  59
  60         if request_id not in self.worker_views:
  61             self.worker_views[request_id] = variable.Variable(None)
  62
  63         if long_poll and self.worker_views[request_id].value in [None, self.new_work_event.times]:
  64             if p2pool.DEBUG:
  65                 print 'POLL %i WAITING user=%r' % (id, get_username(request))
  66             yield self.new_work_event.get_deferred()
  67
  68         res, identifier = self.compute(request)
  69
  70         if long_poll:
  71             self.worker_views[request_id].set(self.new_work_event.times)
  72
  73         if p2pool.DEBUG:
  74             print 'POLL %i END %s user=%r' % (id, p2pool_data.format_hash(identifier), get_username(request)) # XXX identifier is hack
  75
  76         defer.returnValue(res.getwork(identifier=str(identifier)))