milksad
/
rust-bitcoin-unsafe-fast
14
0
Fork 0
Code Issues Pull Requests Activity

Minor improvements to Witness::get_tapscript commit

This commit is contained in:
sanket1729 2022-11-12 02:02:13 -08:00
parent 1d0b0e6ed8
commit 865fd5ac90
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
bitcoin/src/blockdata/witness.rs
View File

@ -16,6 +16,7 @@ use crate::sighash::EcdsaSighashType;
use crate::io::{self, Read, Write}; use crate::io::{self, Read, Write};
use crate::prelude::*; use crate::prelude::*;
use crate::VarInt; use crate::VarInt;
use crate::util::taproot::TAPROOT_ANNEX_PREFIX;
const U32_SIZE: usize = core::mem::size_of::<u32>(); const U32_SIZE: usize = core::mem::size_of::<u32>();
@ -304,7 +305,7 @@ impl Witness {
/// This does not guarantee that this represents a P2TR [`Witness`]. /// This does not guarantee that this represents a P2TR [`Witness`].
/// It merely gets the second to last or third to last element depending /// It merely gets the second to last or third to last element depending
/// on the first byte of the last element being equal to 0x50. /// on the first byte of the last element being equal to 0x50.
pub fn get_tapscript(&self) -> Option<&[u8]> { pub fn tapscript(&self) -> Option<&[u8]> {
let len = self.len(); let len = self.len();
self self
.last() .last()
@ -313,7 +314,7 @@ impl Witness {
// If there are at least two witness elements, and the first byte of // If there are at least two witness elements, and the first byte of
// the last element is 0x50, this last element is called annex a // the last element is 0x50, this last element is called annex a
// and is removed from the witness stack. // and is removed from the witness stack.
if len >= 2 && last_elem.first().filter(|&&v| v == 0x50).is_some() { if len >= 2 && last_elem.first() == Some(&TAPROOT_ANNEX_PREFIX) {
// account for the extra item removed from the end // account for the extra item removed from the end
3 3
} else { } else {
@ -604,8 +605,8 @@ mod test {
}; };
// With or without annex, the tapscript should be returned. // With or without annex, the tapscript should be returned.
assert_eq!(witness.get_tapscript(), Some(&tapscript[..])); assert_eq!(witness.tapscript(), Some(&tapscript[..]));
assert_eq!(witness_annex.get_tapscript(), Some(&tapscript[..])); assert_eq!(witness_annex.tapscript(), Some(&tapscript[..]));
} }
#[test] #[test]