txscript: Optimize ExtractWitnessProgramInfo

This commit is contained in:
Conner Fromknecht 2019-04-19 19:43:43 -07:00 committed by Roy Lee
parent ad01d080d9
commit a3166c8d9b
2 changed files with 4 additions and 11 deletions

View file

@ -131,23 +131,16 @@ func IsNullData(script []byte) bool {
// ExtractWitnessProgramInfo attempts to extract the witness program version,
// as well as the witness program itself from the passed script.
func ExtractWitnessProgramInfo(script []byte) (int, []byte, error) {
pops, err := parseScript(script)
if err != nil {
return 0, nil, err
}
// If at this point, the scripts doesn't resemble a witness program,
// then we'll exit early as there isn't a valid version or program to
// extract.
if !isWitnessProgram(pops) {
version, program, valid := extractWitnessProgramInfo(script)
if !valid {
return 0, nil, fmt.Errorf("script is not a witness program, " +
"unable to extract version or witness program")
}
witnessVersion := asSmallInt(pops[0].opcode.value)
witnessProgram := pops[1].data
return witnessVersion, witnessProgram, nil
return version, program, nil
}
// IsPushOnlyScript returns whether or not the passed script only pushes data

View file

@ -388,7 +388,7 @@ func isWitnessScriptHashScript(script []byte) bool {
}
// extractWitnessProgramInfo returns the version and program if the passed
// script constitutes a valid witness program. The alst return value indicates
// script constitutes a valid witness program. The last return value indicates
// whether or not the script is a valid witness program.
func extractWitnessProgramInfo(script []byte) (int, []byte, bool) {
// Skip parsing if we know the program is invalid based on size.