p2pool/bitcoin/worker_interface.py

   1 from __future__ import division
   2
   3 import StringIO
   4 import json
   5 import random
   6 import sys
   7
   8 from twisted.internet import defer
   9
  10 import p2pool
  11 from p2pool.bitcoin import data as bitcoin_data, getwork
  12 from p2pool.util import expiring_dict, jsonrpc, pack, variable
  13
  14 class _Provider(object):
  15     def __init__(self, parent, long_poll):
  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 class _GETableServer(jsonrpc.HTTPServer):
  23     def __init__(self, provider, render_get_func):
  24         jsonrpc.HTTPServer.__init__(self, provider)
  25         self.render_GET = render_get_func
  26
  27 class WorkerBridge(object):
  28     def __init__(self):
  29         self.new_work_event = variable.Event()
  30
  31     def preprocess_request(self, request):
  32         return request, # *args to self.compute
  33
  34     def get_work(self, request):
  35         raise NotImplementedError()
  36
  37 class WorkerInterface(object):
  38     def __init__(self, worker_bridge):
  39         self.worker_bridge = worker_bridge
  40
  41         self.worker_views = {}
  42
  43         self.work_cache = {}
  44         self.work_cache_times = self.worker_bridge.new_work_event.times
  45
  46         self.merkle_root_to_handler = expiring_dict.ExpiringDict(300)
  47
  48     def attach_to(self, res, get_handler=None):
  49         res.putChild('', _GETableServer(_Provider(self, long_poll=False), get_handler))
  50
  51         def repost(request):
  52             request.content = StringIO.StringIO(json.dumps(dict(id=0, method='getwork')))
  53             return s.render_POST(request)
  54         s = _GETableServer(_Provider(self, long_poll=True), repost)
  55         res.putChild('long-polling', s)
  56
  57     @defer.inlineCallbacks
  58     def _getwork(self, request, data, long_poll):
  59         request.setHeader('X-Long-Polling', '/long-polling')
  60         request.setHeader('X-Roll-NTime', 'expire=10')
  61         request.setHeader('X-Is-P2Pool', 'true')
  62         if request.getHeader('Host') is not None:
  63             request.setHeader('X-Stratum', 'stratum+tcp://' + request.getHeader('Host'))
  64
  65         if data is not None:
  66             header = getwork.decode_data(data)
  67             if header['merkle_root'] not in self.merkle_root_to_handler:
  68                 print >>sys.stderr, '''Couldn't link returned work's merkle root with its handler. This should only happen if this process was recently restarted!'''
  69                 defer.returnValue(False)
  70             defer.returnValue(self.merkle_root_to_handler[header['merkle_root']](header, request.getUser() if request.getUser() is not None else '', 0))
  71
  72         if p2pool.DEBUG:
  73             id = random.randrange(1000, 10000)
  74             print 'POLL %i START is_long_poll=%r user_agent=%r user=%r' % (id, long_poll, request.getHeader('User-Agent'), request.getUser())
  75
  76         if long_poll:
  77             request_id = request.getClientIP(), request.getHeader('Authorization')
  78             if self.worker_views.get(request_id, self.worker_bridge.new_work_event.times) != self.worker_bridge.new_work_event.times:
  79                 if p2pool.DEBUG:
  80                     print 'POLL %i PUSH' % (id,)
  81             else:
  82                 if p2pool.DEBUG:
  83                     print 'POLL %i WAITING' % (id,)
  84                 yield self.worker_bridge.new_work_event.get_deferred()
  85             self.worker_views[request_id] = self.worker_bridge.new_work_event.times
  86
  87         key = self.worker_bridge.preprocess_request(request.getUser() if request.getUser() is not None else '')
  88
  89         if self.work_cache_times != self.worker_bridge.new_work_event.times:
  90             self.work_cache = {}
  91             self.work_cache_times = self.worker_bridge.new_work_event.times
  92
  93         if key in self.work_cache:
  94             res, orig_timestamp, handler = self.work_cache.pop(key)
  95         else:
  96             x, handler = self.worker_bridge.get_work(*key)
  97             res = getwork.BlockAttempt(
  98                 version=x['version'],
  99                 previous_block=x['previous_block'],
 100                 merkle_root=bitcoin_data.check_merkle_link(bitcoin_data.hash256(x['coinb1'] + pack.IntType(32).pack(0) + x['coinb2']), x['merkle_link']),
 101                 timestamp=x['timestamp'],
 102                 bits=x['bits'],
 103                 share_target=x['share_target'],
 104             )
 105             assert res.merkle_root not in self.merkle_root_to_handler
 106             orig_timestamp = res.timestamp
 107
 108         self.merkle_root_to_handler[res.merkle_root] = handler
 109
 110         if res.timestamp + 120 < orig_timestamp + 1800:
 111             self.work_cache[key] = res.update(timestamp=res.timestamp + 120), orig_timestamp, handler
 112
 113         if p2pool.DEBUG:
 114             print 'POLL %i END identifier=%i' % (id, self.worker_bridge.new_work_event.times)
 115
 116         defer.returnValue(res.getwork(identifier=str(self.worker_bridge.new_work_event.times), submitold=True))