p2pool/bitcoin/worker_interface.py

   1 from __future__ import division
   2
   3 import StringIO
   4 import json
   5 import random
   6
   7 from twisted.internet import defer
   8
   9 import p2pool
  10 from p2pool.bitcoin import getwork
  11 from p2pool.util import jsonrpc, variable
  12
  13 class _Page(jsonrpc.Server):
  14     def __init__(self, parent, long_poll):
  15         jsonrpc.Server.__init__(self)
  16         self.parent = parent
  17         self.long_poll = long_poll
  18
  19     def rpc_getwork(self, request, data=None):
  20         return self.parent._getwork(request, data, long_poll=self.long_poll)
  21
  22     def render_GET(self, request):
  23         request.content = StringIO.StringIO(json.dumps(dict(id=0, method='getwork')))
  24         return self.render_POST(request)
  25
  26 class WorkerBridge(object):
  27     def __init__(self):
  28         self.new_work_event = variable.Event()
  29
  30     def preprocess_request(self, request):
  31         return request, # *args to self.compute
  32
  33     def get_work(self, request):
  34         raise NotImplementedError()
  35
  36     def got_response(self, block_header):
  37         print self.got_response, "called with", block_header
  38
  39 class WorkerInterface(object):
  40     def __init__(self, worker_bridge):
  41         self.worker_bridge = worker_bridge
  42
  43         self.worker_views = {}
  44
  45         self.work_cache = {} # request_process_func(request) -> blockattempt
  46         self.work_cache_times = self.worker_bridge.new_work_event.times
  47
  48     def attach_to(self, res):
  49         res.putChild('', _Page(self, long_poll=False))
  50         res.putChild('long-polling', _Page(self, long_poll=True))
  51
  52     @defer.inlineCallbacks
  53     def _getwork(self, request, data, long_poll):
  54         request.setHeader('X-Long-Polling', '/long-polling')
  55         request.setHeader('X-Roll-NTime', 'expire=10')
  56         request.setHeader('X-Is-P2Pool', 'true')
  57
  58         if data is not None:
  59             defer.returnValue(self.worker_bridge.got_response(getwork.decode_data(data), request))
  60
  61         if p2pool.DEBUG:
  62             id = random.randrange(1000, 10000)
  63             print 'POLL %i START is_long_poll=%r user_agent=%r user=%r' % (id, long_poll, request.getHeader('User-Agent'), request.getUser())
  64
  65         if long_poll:
  66             request_id = request.getClientIP(), request.getHeader('Authorization')
  67             if self.worker_views.get(request_id, self.worker_bridge.new_work_event.times) != self.worker_bridge.new_work_event.times:
  68                 if p2pool.DEBUG:
  69                     print 'POLL %i PUSH' % (id,)
  70             else:
  71                 if p2pool.DEBUG:
  72                     print 'POLL %i WAITING' % (id,)
  73                 yield self.worker_bridge.new_work_event.get_deferred()
  74             self.worker_views[request_id] = self.worker_bridge.new_work_event.times
  75
  76         key = self.worker_bridge.preprocess_request(request)
  77
  78         if self.work_cache_times != self.worker_bridge.new_work_event.times:
  79             self.work_cache = {}
  80             self.work_cache_times = self.worker_bridge.new_work_event.times
  81
  82         if key in self.work_cache:
  83             res, orig_timestamp = self.work_cache.pop(key)
  84         else:
  85             res = self.worker_bridge.get_work(*key)
  86             orig_timestamp = res.timestamp
  87
  88         if res.timestamp + 12 < orig_timestamp + 600:
  89             self.work_cache[key] = res.update(timestamp=res.timestamp + 12), orig_timestamp
  90
  91         if p2pool.DEBUG:
  92             print 'POLL %i END identifier=%i' % (id, self.worker_bridge.new_work_event.times)
  93
  94         defer.returnValue(res.getwork(identifier=str(self.worker_bridge.new_work_event.times), submitold=True))