Merge rust-bitcoin/rust-bitcoin#710: Refactor bitcoin_merkle_root functions
b454cf8e15
Return None from merkle_root functions (Tobin Harding)7a8b017ea3
Use correct spelling of merkle (Tobin Harding) Pull request description: ~Do two minor refactorings to the `bitcoin_merkle_root[_inline] functions.~ This PR has grown, is no longer a refactoring because the two functions have been changed to return an `Option`. First patch is cleanup. Here is the commit message for the second patch ``` The merkle_root of an empty tree is undefined, this is the only error case we have for the two `bitcoin_merkle_root*` functions. We can fully describe this error case by returning an `Option` if args are found to be empty. While we are at it, refactor out a recursive helper function to make reading the code between the two functions easier. ``` ACKs for top commit: Kixunil: ACKb454cf8e15
dr-orlovsky: ACKb454cf8e15
Tree-SHA512: 961714a8b0eb0dad493a1548317d875d64ca22d2d584c905c502369b5f6e5a9f8be1edd7345136b44964dc0bde7a4c43bfaff4287d1dbf7fd736da79818074e3
This commit is contained in:
commit
fe43e3c9d7
|
@ -176,7 +176,10 @@ impl Block {
|
||||||
|
|
||||||
/// check if merkle root of header matches merkle root of the transaction list
|
/// check if merkle root of header matches merkle root of the transaction list
|
||||||
pub fn check_merkle_root (&self) -> bool {
|
pub fn check_merkle_root (&self) -> bool {
|
||||||
self.header.merkle_root == self.merkle_root()
|
match self.merkle_root() {
|
||||||
|
Some(merkle_root) => self.header.merkle_root == merkle_root,
|
||||||
|
None => false,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// check if witness commitment in coinbase is matching the transaction list
|
/// check if witness commitment in coinbase is matching the transaction list
|
||||||
|
@ -197,8 +200,10 @@ impl Block {
|
||||||
let commitment = WitnessCommitment::from_slice(&coinbase.output[pos].script_pubkey.as_bytes()[6..38]).unwrap();
|
let commitment = WitnessCommitment::from_slice(&coinbase.output[pos].script_pubkey.as_bytes()[6..38]).unwrap();
|
||||||
// witness reserved value is in coinbase input witness
|
// witness reserved value is in coinbase input witness
|
||||||
if coinbase.input[0].witness.len() == 1 && coinbase.input[0].witness[0].len() == 32 {
|
if coinbase.input[0].witness.len() == 1 && coinbase.input[0].witness[0].len() == 32 {
|
||||||
let witness_root = self.witness_root();
|
match self.witness_root() {
|
||||||
return commitment == Self::compute_witness_commitment(&witness_root, coinbase.input[0].witness[0].as_slice())
|
Some(witness_root) => return commitment == Self::compute_witness_commitment(&witness_root, coinbase.input[0].witness[0].as_slice()),
|
||||||
|
None => return false,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -207,9 +212,9 @@ impl Block {
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Calculate the transaction merkle root.
|
/// Calculate the transaction merkle root.
|
||||||
pub fn merkle_root(&self) -> TxMerkleNode {
|
pub fn merkle_root(&self) -> Option<TxMerkleNode> {
|
||||||
let hashes = self.txdata.iter().map(|obj| obj.txid().as_hash());
|
let hashes = self.txdata.iter().map(|obj| obj.txid().as_hash());
|
||||||
bitcoin_merkle_root(hashes).into()
|
bitcoin_merkle_root(hashes).map(|h| h.into())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// compute witness commitment for the transaction list
|
/// compute witness commitment for the transaction list
|
||||||
|
@ -221,7 +226,7 @@ impl Block {
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Merkle root of transactions hashed for witness
|
/// Merkle root of transactions hashed for witness
|
||||||
pub fn witness_root(&self) -> WitnessMerkleNode {
|
pub fn witness_root(&self) -> Option<WitnessMerkleNode> {
|
||||||
let hashes = self.txdata.iter().enumerate().map(|(i, t)|
|
let hashes = self.txdata.iter().enumerate().map(|(i, t)|
|
||||||
if i == 0 {
|
if i == 0 {
|
||||||
// Replace the first hash with zeroes.
|
// Replace the first hash with zeroes.
|
||||||
|
@ -230,7 +235,7 @@ impl Block {
|
||||||
t.wtxid().as_hash()
|
t.wtxid().as_hash()
|
||||||
}
|
}
|
||||||
);
|
);
|
||||||
bitcoin_merkle_root(hashes).into()
|
bitcoin_merkle_root(hashes).map(|h| h.into())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The size of the header + the size of the varint with the tx count + the txs themselves
|
/// The size of the header + the size of the varint with the tx count + the txs themselves
|
||||||
|
@ -371,7 +376,7 @@ mod tests {
|
||||||
let real_decode = decode.unwrap();
|
let real_decode = decode.unwrap();
|
||||||
assert_eq!(real_decode.header.version, 1);
|
assert_eq!(real_decode.header.version, 1);
|
||||||
assert_eq!(serialize(&real_decode.header.prev_blockhash), prevhash);
|
assert_eq!(serialize(&real_decode.header.prev_blockhash), prevhash);
|
||||||
assert_eq!(real_decode.header.merkle_root, real_decode.merkle_root());
|
assert_eq!(real_decode.header.merkle_root, real_decode.merkle_root().unwrap());
|
||||||
assert_eq!(serialize(&real_decode.header.merkle_root), merkle);
|
assert_eq!(serialize(&real_decode.header.merkle_root), merkle);
|
||||||
assert_eq!(real_decode.header.time, 1231965655);
|
assert_eq!(real_decode.header.time, 1231965655);
|
||||||
assert_eq!(real_decode.header.bits, 486604799);
|
assert_eq!(real_decode.header.bits, 486604799);
|
||||||
|
@ -407,7 +412,7 @@ mod tests {
|
||||||
assert_eq!(real_decode.header.version, 0x20000000); // VERSIONBITS but no bits set
|
assert_eq!(real_decode.header.version, 0x20000000); // VERSIONBITS but no bits set
|
||||||
assert_eq!(serialize(&real_decode.header.prev_blockhash), prevhash);
|
assert_eq!(serialize(&real_decode.header.prev_blockhash), prevhash);
|
||||||
assert_eq!(serialize(&real_decode.header.merkle_root), merkle);
|
assert_eq!(serialize(&real_decode.header.merkle_root), merkle);
|
||||||
assert_eq!(real_decode.header.merkle_root, real_decode.merkle_root());
|
assert_eq!(real_decode.header.merkle_root, real_decode.merkle_root().unwrap());
|
||||||
assert_eq!(real_decode.header.time, 1472004949);
|
assert_eq!(real_decode.header.time, 1472004949);
|
||||||
assert_eq!(real_decode.header.bits, 0x1a06d450);
|
assert_eq!(real_decode.header.bits, 0x1a06d450);
|
||||||
assert_eq!(real_decode.header.nonce, 1879759182);
|
assert_eq!(real_decode.header.nonce, 1879759182);
|
||||||
|
|
File diff suppressed because one or more lines are too long
|
@ -543,7 +543,7 @@ mod tests {
|
||||||
|
|
||||||
// Calculate the merkle root and height
|
// Calculate the merkle root and height
|
||||||
let hashes = txids.iter().map(|t| t.as_hash());
|
let hashes = txids.iter().map(|t| t.as_hash());
|
||||||
let merkle_root_1: TxMerkleNode = bitcoin_merkle_root(hashes).into();
|
let merkle_root_1: TxMerkleNode = bitcoin_merkle_root(hashes).expect("hashes is not empty").into();
|
||||||
let mut height = 1;
|
let mut height = 1;
|
||||||
let mut ntx = num_tx;
|
let mut ntx = num_tx;
|
||||||
while ntx > 1 {
|
while ntx > 1 {
|
||||||
|
|
Loading…
Reference in New Issue