text: #!/usr/bin/env python3
  # Copyright (c) 2014-2018 The Bitcoin Core developers
  # Distributed under the MIT software license, see the accompanying
  # file COPYING or http://www.opensource.org/licenses/mit-license.php.
  """Test mining RPCs
  
  - getmininginfo
  - getblocktemplate proposal mode
  - submitblock"""
  
  import copy
  from decimal import Decimal
  
  from test_framework.blocktools import create_coinbase
  from test_framework.messages import (
      CBlock,
      CBlockHeader,
  )
  from test_framework.mininode import (
      P2PDataStore,
  )
  from test_framework.test_framework import BitcoinTestFramework
  from test_framework.util import (
      assert_equal,
      assert_raises_rpc_error,
      bytes_to_hex_str as b2x,
  )
  
  
  def assert_template(node, block, expect, rehash=True):
      if rehash:
          block.hashMerkleRoot = block.calc_merkle_root()
      rsp = node.getblocktemplate({'data': b2x(block.serialize()), 'mode': 'proposal'})
      assert_equal(rsp, expect)
  
  class MiningTest(BitcoinTestFramework):
      def set_test_params(self):
          self.num_nodes = 2
          self.setup_clean_chain = False
  
      def run_test(self):
          node = self.nodes[0]
  
          self.log.info('getmininginfo')
          mining_info = node.getmininginfo()
          assert_equal(mining_info['blocks'], 200)
          assert_equal(mining_info['chain'], 'regtest')
          assert_equal(mining_info['currentblocktx'], 0)
          assert_equal(mining_info['currentblockweight'], 0)
          assert_equal(mining_info['difficulty'], {
              'sha256d': Decimal('4.656542373906925E-10'),
              'neoscrypt': Decimal('4.656542373906925E-10'),
          })
          assert_equal(mining_info['networkhashps']['neoscrypt'], Decimal('0.003333333333333334'))
          assert_equal(mining_info['networkhashps']['sha256d'], Decimal('0'))
          assert_equal(mining_info['pooledtx'], 0)
  
          # Mine a block to leave initial block download
          node.generate(1)
          tmpl = node.getblocktemplate()
          self.log.info("getblocktemplate: Test capability advertised")
          assert 'proposal' in tmpl['capabilities']
          assert 'coinbasetxn' not in tmpl
  
          coinbase_tx = create_coinbase(height=int(tmpl["height"]) + 1)
          # sequence numbers must not be max for nLockTime to have effect
          coinbase_tx.vin[0].nSequence = 2 ** 32 - 2
          coinbase_tx.rehash()
  
          block = CBlock()
          block.nVersion = tmpl["version"]
          block.hashPrevBlock = int(tmpl["previousblockhash"], 16)
          block.nTime = tmpl["curtime"]
          block.nBits = 0
          block.nNonce = 0
          block.powData.nBits = int(tmpl["bits"], 16)
          assert_equal(tmpl["algo"], "neoscrypt")
          block.vtx = [coinbase_tx]
  
          self.log.info("getblocktemplate: Test valid block")
          assert_template(node, block, None)
  
          self.log.info("submitblock: Test block decode failure")
          assert_raises_rpc_error(-22, "Block decode failed", node.submitblock, b2x(block.serialize()[:-15]))
  
          self.log.info("getblocktemplate: Test bad input hash for coinbase transaction")
          bad_block = copy.deepcopy(block)
          bad_block.vtx[0].vin[0].prevout.hash += 1
          bad_block.vtx[0].rehash()
          assert_template(node, bad_block, 'bad-cb-missing')
  
          self.log.info("submitblock: Test invalid coinbase transaction")
          assert_raises_rpc_error(-22, "Block does not start with a coinbase", node.submitblock, b2x(bad_block.serialize()))
  
          self.log.info("getblocktemplate: Test truncated final transaction")
          assert_raises_rpc_error(-22, "Block decode failed", node.getblocktemplate, {'data': b2x(block.serialize()[:-1]), 'mode': 'proposal'})
  
          self.log.info("getblocktemplate: Test duplicate transaction")
          bad_block = copy.deepcopy(block)
          bad_block.vtx.append(bad_block.vtx[0])
          assert_template(node, bad_block, 'bad-txns-duplicate')
  
          self.log.info("getblocktemplate: Test invalid transaction")
          bad_block = copy.deepcopy(block)
          bad_tx = copy.deepcopy(bad_block.vtx[0])
          bad_tx.vin[0].prevout.hash = 255
          bad_tx.rehash()
          bad_block.vtx.append(bad_tx)
          assert_template(node, bad_block, 'bad-txns-inputs-missingorspent')
  
          self.log.info("getblocktemplate: Test nonfinal transaction")
          bad_block = copy.deepcopy(block)
          bad_block.vtx[0].nLockTime = 2 ** 32 - 1
          bad_block.vtx[0].rehash()
          assert_template(node, bad_block, 'bad-txns-nonfinal')
  
          self.log.info("getblocktemplate: Test bad tx count")
          # The tx count is immediately after the block header
          TX_COUNT_OFFSET = 80 + 1 + 4 + 80
          bad_block_sn = bytearray(block.serialize())
          assert_equal(bad_block_sn[TX_COUNT_OFFSET], 1)
          bad_block_sn[TX_COUNT_OFFSET] += 1
          assert_raises_rpc_error(-22, "Block decode failed", node.getblocktemplate, {'data': b2x(bad_block_sn), 'mode': 'proposal'})
  
          self.log.info("getblocktemplate: Test bad bits")
          bad_block = copy.deepcopy(block)
          bad_block.powData.nBits = 469762303  # impossible in the real world
          assert_template(node, bad_block, 'bad-diffbits')
  
          self.log.info("getblocktemplate: Test bad merkle root")
          bad_block = copy.deepcopy(block)
          bad_block.hashMerkleRoot += 1
          assert_template(node, bad_block, 'bad-txnmrklroot', False)
  
          self.log.info("getblocktemplate: Test bad timestamps")
          bad_block = copy.deepcopy(block)
          bad_block.nTime = 2 ** 31 - 1
          assert_template(node, bad_block, 'time-too-new')
          bad_block.nTime = 0
          assert_template(node, bad_block, 'time-too-old')
  
          self.log.info("getblocktemplate: Test not best block")
          bad_block = copy.deepcopy(block)
          bad_block.hashPrevBlock = 123
          assert_template(node, bad_block, 'inconclusive-not-best-prevblk')
  
          self.log.info('submitheader tests')
          assert_raises_rpc_error(-22, 'Block header decode failed', lambda: node.submitheader(hexdata='xx' * 80))
          assert_raises_rpc_error(-22, 'Block header decode failed', lambda: node.submitheader(hexdata='ff' * 78))
          assert_raises_rpc_error(-25, 'Must submit previous header', lambda: node.submitheader(hexdata='00' * (80 + 5 + 80)))
  
          block.solve()
  
          def chain_tip(b_hash, *, status='headers-only', branchlen=1):
              return {'hash': b_hash, 'height': 202, 'branchlen': branchlen, 'status': status}
  
          assert chain_tip(block.hash) not in node.getchaintips()
          node.submitheader(hexdata=b2x(block.serialize()))
          assert chain_tip(block.hash) in node.getchaintips()
          node.submitheader(hexdata=b2x(CBlockHeader(block).serialize()))  # Noop
          assert chain_tip(block.hash) in node.getchaintips()
  
          bad_block_root = copy.deepcopy(block)
          bad_block_root.hashMerkleRoot += 2
          bad_block_root.solve()
          assert chain_tip(bad_block_root.hash) not in node.getchaintips()
          node.submitheader(hexdata=b2x(CBlockHeader(bad_block_root).serialize()))
          assert chain_tip(bad_block_root.hash) in node.getchaintips()
          # Should still reject invalid blocks, even if we have the header:
          assert_equal(node.submitblock(hexdata=b2x(bad_block_root.serialize())), 'invalid')
          assert chain_tip(bad_block_root.hash) in node.getchaintips()
          # We know the header for this invalid block, so should just return early without error:
          node.submitheader(hexdata=b2x(CBlockHeader(bad_block_root).serialize()))
          assert chain_tip(bad_block_root.hash) in node.getchaintips()
  
          bad_block_lock = copy.deepcopy(block)
          bad_block_lock.vtx[0].nLockTime = 2**32 - 1
          bad_block_lock.vtx[0].rehash()
          bad_block_lock.hashMerkleRoot = bad_block_lock.calc_merkle_root()
          bad_block_lock.solve()
          assert_equal(node.submitblock(hexdata=b2x(bad_block_lock.serialize())), 'invalid')
          # Build a "good" block on top of the submitted bad block
          bad_block2 = copy.deepcopy(block)
          bad_block2.hashPrevBlock = bad_block_lock.sha256
          bad_block2.solve()
          assert_raises_rpc_error(-25, 'bad-prevblk', lambda: node.submitheader(hexdata=b2x(CBlockHeader(bad_block2).serialize())))
  
          # Should reject invalid header right away
          bad_block_time = copy.deepcopy(block)
          bad_block_time.nTime = 1
          bad_block_time.solve()
          assert_raises_rpc_error(-25, 'time-too-old', lambda: node.submitheader(hexdata=b2x(CBlockHeader(bad_block_time).serialize())))
  
          # Should ask for the block from a p2p node, if they announce the header as well:
          node.add_p2p_connection(P2PDataStore())
          node.p2p.wait_for_getheaders(timeout=5)  # Drop the first getheaders
          node.p2p.send_blocks_and_test(blocks=[block], rpc=node)
          # Must be active now:
          assert chain_tip(block.hash, status='active', branchlen=0) in node.getchaintips()
  
          # Building a few blocks should give the same results
          node.generate(10)
          assert_raises_rpc_error(-25, 'time-too-old', lambda: node.submitheader(hexdata=b2x(CBlockHeader(bad_block_time).serialize())))
          assert_raises_rpc_error(-25, 'bad-prevblk', lambda: node.submitheader(hexdata=b2x(CBlockHeader(bad_block2).serialize())))
          node.submitheader(hexdata=b2x(CBlockHeader(block).serialize()))
          node.submitheader(hexdata=b2x(CBlockHeader(bad_block_root).serialize()))
  
  
  if __name__ == '__main__':
      MiningTest().main()
  