mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-01-10 20:03:34 -03:00
test: add compact block filter (BIP158) helper routines
By now, we add one helper for calculating ranged hashes and another one for finding relevant scriptPubKeys given a block.
This commit is contained in:
parent
25ee74dd11
commit
3bca6cd61a
1 changed files with 49 additions and 0 deletions
49
test/functional/test_framework/blockfilter.py
Normal file
49
test/functional/test_framework/blockfilter.py
Normal file
|
@ -0,0 +1,49 @@
|
|||
#!/usr/bin/env python3
|
||||
# Copyright (c) 2022 The Bitcoin Core developers
|
||||
# Distributed under the MIT software license, see the accompanying
|
||||
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
||||
"""Helper routines relevant for compact block filters (BIP158).
|
||||
"""
|
||||
from .siphash import siphash
|
||||
|
||||
|
||||
def bip158_basic_element_hash(script_pub_key, N, block_hash):
|
||||
""" Calculates the ranged hash of a filter element as defined in BIP158:
|
||||
|
||||
'The first step in the filter construction is hashing the variable-sized
|
||||
raw items in the set to the range [0, F), where F = N * M.'
|
||||
|
||||
'The items are first passed through the pseudorandom function SipHash, which takes a
|
||||
128-bit key k and a variable-sized byte vector and produces a uniformly random 64-bit
|
||||
output. Implementations of this BIP MUST use the SipHash parameters c = 2 and d = 4.'
|
||||
|
||||
'The parameter k MUST be set to the first 16 bytes of the hash (in standard
|
||||
little-endian representation) of the block for which the filter is constructed. This
|
||||
ensures the key is deterministic while still varying from block to block.'
|
||||
"""
|
||||
M = 784931
|
||||
block_hash_bytes = bytes.fromhex(block_hash)[::-1]
|
||||
k0 = int.from_bytes(block_hash_bytes[0:8], 'little')
|
||||
k1 = int.from_bytes(block_hash_bytes[8:16], 'little')
|
||||
return (siphash(k0, k1, script_pub_key) * (N * M)) >> 64
|
||||
|
||||
|
||||
def bip158_relevant_scriptpubkeys(node, block_hash):
|
||||
""" Determines the basic filter relvant scriptPubKeys as defined in BIP158:
|
||||
|
||||
'A basic filter MUST contain exactly the following items for each transaction in a block:
|
||||
- The previous output script (the script being spent) for each input, except for
|
||||
the coinbase transaction.
|
||||
- The scriptPubKey of each output, aside from all OP_RETURN output scripts.'
|
||||
"""
|
||||
spks = set()
|
||||
for tx in node.getblock(blockhash=block_hash, verbosity=3)['tx']:
|
||||
# gather prevout scripts
|
||||
for i in tx['vin']:
|
||||
if 'prevout' in i:
|
||||
spks.add(bytes.fromhex(i['prevout']['scriptPubKey']['hex']))
|
||||
# gather output scripts
|
||||
for o in tx['vout']:
|
||||
if o['scriptPubKey']['type'] != 'nulldata':
|
||||
spks.add(bytes.fromhex(o['scriptPubKey']['hex']))
|
||||
return spks
|
Loading…
Reference in a new issue