tests: Add unix timestamp tests for OP_CLTV

test/functional/feature_cltv.py

@@ -7,6 +7,7 @@
 Test that the CHECKLOCKTIMEVERIFY soft-fork activates.
 """
 
+import time
 from test_framework.blocktools import (
     TIME_GENESIS_BLOCK,
     create_block,
@@ -26,7 +27,7 @@ from test_framework.script import (
     OP_DROP,
 )
 from test_framework.test_framework import BitcoinTestFramework
-from test_framework.util import assert_equal
+from test_framework.util import assert_equal, assert_not_equal
 from test_framework.wallet import (
     MiniWallet,
     MiniWalletMode,
@@ -72,12 +73,12 @@ def cltv_invalidate(tx, failure_reason):
 
 def cltv_validate(tx, height):
     # Modify the signature in vin 0 and nSequence/nLockTime of the tx to pass CLTV
-    scheme = [[CScriptNum(height), OP_CHECKLOCKTIMEVERIFY, OP_DROP], 0, height]
+    scheme = [[CScriptNum.encode(CScriptNum(height))[1:], OP_CHECKLOCKTIMEVERIFY, OP_DROP], 0, height]
 
     cltv_modify_tx(tx, prepend_scriptsig=scheme[0], nsequence=scheme[1], nlocktime=scheme[2])
 
 
-CLTV_HEIGHT = 111
+CLTV_HEIGHT = 112
 
 
 class BIP65Test(BitcoinTestFramework):
@@ -107,8 +108,9 @@ class BIP65Test(BitcoinTestFramework):
         self.test_cltv_info(is_active=False)
 
         self.log.info("Mining %d blocks", CLTV_HEIGHT - 2)
-        self.generate(wallet, 10)
-        self.generate(self.nodes[0], CLTV_HEIGHT - 2 - 10)
+        num_utxos = 13
+        self.generate(wallet, num_utxos)
+        self.generate(self.nodes[0], CLTV_HEIGHT - 2 - num_utxos)
         assert_equal(self.nodes[0].getblockcount(), CLTV_HEIGHT - 2)
 
         self.log.info("Test that invalid-according-to-CLTV transactions can still appear in a block")
@@ -199,6 +201,67 @@ class BIP65Test(BitcoinTestFramework):
         self.test_cltv_info(is_active=True)  # Active as of current tip
         assert_equal(int(self.nodes[0].getbestblockhash(), 16), block.sha256)
 
+        # test wall clock time
+        tip = block.sha256
+
+        current_time = int(time.time())
+        locktime = current_time + (60 * 60) # 1 hour
+
+        self.log.info("Test that one block mined with correct wall clock time does not satisfy median-time-past")
+        cltv_validate(spendtx,locktime)
+        block2 = create_block(tip, create_coinbase(CLTV_HEIGHT + 1), ntime=locktime, version=4)
+        block2.vtx.append(spendtx)
+        block2.hashMerkleRoot = block2.calc_merkle_root()
+        block2.solve()
+        peer.send_and_ping(msg_block(block2))
+        # expect failure as we haven't satisfied wall clock time yet
+        assert_not_equal(int(self.nodes[0].getbestblockhash(), 16), block2.sha256)
+
+        self.log.info("Test 7 blocks mined with current wall clock time can satisfy the median-time-past")
+        num = 7
+        for i in range(num):
+            b = create_block(tip, create_coinbase(CLTV_HEIGHT + i + 1), ntime=locktime + i, version=4)
+            b.solve()
+            peer.send_and_ping(msg_block(b))
+            tip = b.sha256
+
+        # now our median-past-time should allow for a transaction with
+        # a locktime of +1 hour to be confirmed
+        spendtx = wallet.create_self_transfer()['tx']
+        cltv_validate(spendtx,locktime)
+        block2.vtx.append(spendtx)
+        block2 = create_block(tip, create_coinbase(CLTV_HEIGHT + num + 1), ntime=locktime + num, version=4)
+        block2.vtx.append(spendtx)
+        block2.vtx.append(wallet.create_self_transfer()['tx'])
+        block2.hashMerkleRoot = block2.calc_merkle_root()
+        block2.solve()
+        peer.send_and_ping(msg_block(block2))
+        assert_equal(int(self.nodes[0].getbestblockhash(), 16), block2.sha256)
+
+        self.log.info("Test that we cannot mine locktimes that are UINT32_MAX")
+        UINT32_MAX = 2**32 - 1
+        self.nodes[0].setmocktime(UINT32_MAX)
+        # need to re-add connection as setting mocktime times out the p2p connection
+        peer = self.nodes[0].add_p2p_connection(P2PInterface())
+        tip = block2.sha256
+        # 6 blocks are allowed to have UINT32_MAX block time
+        for i in range(6):
+            b = create_block(tip, create_coinbase(CLTV_HEIGHT + num + i + 2), ntime=UINT32_MAX, version=4)
+            b.solve()
+            peer.send_and_ping(msg_block(b))
+            tip = b.sha256
+            assert_equal(int(self.nodes[0].getbestblockhash(), 16), tip)
+
+        spendtx = wallet.create_self_transfer()['tx']
+        # the 7th block means this block is invalid according to BIP113 (median-time-past)
+        block3 = create_block(tip, create_coinbase(CLTV_HEIGHT + num + 6 + 2), ntime=UINT32_MAX, version=4)
+        # cannot satisfy this locktime, need to be strictly greater than the median-past-time
+        cltv_validate(spendtx,UINT32_MAX)
+        block3.vtx.append(spendtx)
+        block3.hashMerkleRoot = block3.calc_merkle_root()
+        block3.solve()
+        peer.send_and_ping(msg_block(block3))
+        self.nodes[0].assert_debug_log("time-too-old, block's timestamp is too early")
 
 if __name__ == '__main__':
     BIP65Test(__file__).main()