txscript: Optimize IsPayToWitnessPubKeyHash

This converts the IsPayToWitnessPubKeyHash function to analyze the raw
script instead of the far less efficient parseScript, thereby
significantly optimizing the function.

In order to accomplish this, it introduces two new functions. The first
one is named extractWitnessPubKeyHash and works with the raw script
bytes to simultaneously deteremine if the script is a p2wkh, and in case
it is, extract and return the hash. The second new function is name
isWitnessPubKeyHashScript which is defined in terms of the former.

The extract function is approach was chosen because it is common for
callers to want to only extract relevant details from the script if the
script is of the specific type. Extracting those details requires the
exact same checks to ensure the script is of the correct type, so it is
more efficient to combine the two and define the type determination in
terms of the result so long as the extraction does not require
allocations.

Finally, this deprecates the isWitnessPubKeyHash function that requires
opcodes in favor of the new functions and modifies the comment on
IsPayToWitnessPubKeyHash to explicitly call out the script version
semantics.

The following is a before and after comparison of executing
IsPayToWitnessPubKeyHash on a large script:

benchmark                          old ns/op     new ns/op     delta
BenchmarkIsWitnessPubKeyHash-8     68927         0.53          -100.00%

benchmark                          old allocs     new allocs     delta
BenchmarkIsWitnessPubKeyHash-8     1              0              -100.00%

benchmark                          old bytes     new bytes     delta
BenchmarkIsWitnessPubKeyHash-8     311299        0             -100.00%
This commit is contained in:
Conner Fromknecht 2021-02-04 13:22:40 -08:00 committed by Olaoluwa Osuntokun
parent e422d42d7f
commit 54d08ebd5f
No known key found for this signature in database
GPG key ID: 3BBD59E99B280306
2 changed files with 22 additions and 5 deletions

View file

@ -110,11 +110,7 @@ func IsPayToWitnessScriptHash(script []byte) bool {
// IsPayToWitnessPubKeyHash returns true if the is in the standard // IsPayToWitnessPubKeyHash returns true if the is in the standard
// pay-to-witness-pubkey-hash (P2WKH) format, false otherwise. // pay-to-witness-pubkey-hash (P2WKH) format, false otherwise.
func IsPayToWitnessPubKeyHash(script []byte) bool { func IsPayToWitnessPubKeyHash(script []byte) bool {
pops, err := parseScript(script) return isWitnessPubKeyHashScript(script)
if err != nil {
return false
}
return isWitnessPubKeyHash(pops)
} }
// isWitnessPubKeyHash returns true if the passed script is a // isWitnessPubKeyHash returns true if the passed script is a

View file

@ -400,6 +400,27 @@ func IsMultisigSigScript(script []byte) bool {
return isMultisigScript(scriptVersion, possibleRedeemScript) return isMultisigScript(scriptVersion, possibleRedeemScript)
} }
// extractWitnessPubKeyHash extracts the witness public key hash from the passed
// script if it is a standard pay-to-witness-pubkey-hash script. It will return
// nil otherwise.
func extractWitnessPubKeyHash(script []byte) []byte {
// A pay-to-witness-pubkey-hash script is of the form:
// OP_0 OP_DATA_20 <20-byte-hash>
if len(script) == 22 &&
script[0] == OP_0 &&
script[1] == OP_DATA_20 {
return script[2:22]
}
return nil
}
// isWitnessPubKeyHashScript returns whether or not the passed script is a
// standard pay-to-witness-pubkey-hash script.
func isWitnessPubKeyHashScript(script []byte) bool {
return extractWitnessPubKeyHash(script) != nil
}
// isNullData returns true if the passed script is a null data transaction, // isNullData returns true if the passed script is a null data transaction,
// false otherwise. // false otherwise.
func isNullData(pops []parsedOpcode) bool { func isNullData(pops []parsedOpcode) bool {