jsonrpc_getwork.py

   1 # Eloipool - Python Bitcoin pool server
   2 # Copyright (C) 2011-2012  Luke Dashjr <luke-jr+eloipool@utopios.org>
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU Affero General Public License as
   6 # published by the Free Software Foundation, either version 3 of the
   7 # License, or (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 from binascii import b2a_hex
  18 from jsonrpcserver import JSONRPCHandler
  19 import logging
  20 try:
  21         import midstate
  22         assert midstate.SHA256(b'This is just a test, ignore it. I am making it over 64-bytes long.')[:8] == (0x755f1a94, 0x999b270c, 0xf358c014, 0xfd39caeb, 0x0dcc9ebc, 0x4694cd1a, 0x8e95678e, 0x75fac450)
  23 except:
  24         logging.getLogger('jsonrpc_getwork').warning('Error importing \'midstate\' module; work will not provide midstates')
  25         midstate = None
  26 from struct import pack
  27 from util import RejectedShare, swap32
  28
  29 _CheckForDupesHACK = {}
  30 _RealDupes = {}
  31 class _getwork:
  32         getwork_rv_template = {
  33                 'data': '000000800000000000000000000000000000000000000000000000000000000000000000000000000000000080020000',
  34                 'target': 'ffffffffffffffffffffffffffffffffffffffffffffffffffffffff00000000',
  35                 'hash1': '00000000000000000000000000000000000000000000000000000000000000000000008000000000000000000000000000000000000000000000000000010000',
  36         }
  37         def doJSON_getwork(self, data=None):
  38                 if not data is None:
  39                         return self.doJSON_submitwork(data)
  40                 rv = dict(self.getwork_rv_template)
  41                 hdr = self.server.getBlockHeader(self.Username)
  42                 if isinstance(hdr, tuple):
  43                         (hdr, x) = hdr
  44                 else:
  45                         x = None
  46
  47                 # FIXME: this assumption breaks with internal rollntime
  48                 # NOTE: noncerange needs to set nonce to start value at least
  49                 global _CheckForDupesHACK
  50                 uhdr = hdr[:68] + hdr[72:]
  51                 if uhdr in _CheckForDupesHACK:
  52                         _RealDupes[uhdr] = (_CheckForDupesHACK[uhdr], (hdr, x))
  53                         raise self.server.RaiseRedFlags(RuntimeError('issuing duplicate work'))
  54                 _CheckForDupesHACK[uhdr] = (hdr, x)
  55
  56                 data = b2a_hex(swap32(hdr)).decode('utf8') + rv['data']
  57                 # TODO: endian shuffle etc
  58                 rv['data'] = data
  59                 if midstate and 'midstate' not in self.extensions and 'midstate' not in self.quirks:
  60                         h = midstate.SHA256(hdr)[:8]
  61                         rv['midstate'] = b2a_hex(pack('<LLLLLLLL', *h)).decode('ascii')
  62                 return rv
  63
  64         def doJSON_submitwork(self, datax):
  65                 data = swap32(bytes.fromhex(datax))[:80]
  66                 share = {
  67                         'data': data,
  68                         '_origdata' : datax,
  69                         'username': self.Username,
  70                         'remoteHost': self.remoteHost,
  71                 }
  72                 try:
  73                         self.server.receiveShare(share)
  74                 except RejectedShare as rej:
  75                         self._JSONHeaders['X-Reject-Reason'] = str(rej)
  76                         return False
  77                 return True
  78
  79 JSONRPCHandler._register(_getwork)