txscript/bench_test.go

@@ -129,3 +129,18 @@ func BenchmarkDisasmString(b *testing.B) {
 		}
 	}
 }
+
+// BenchmarkIsPubKeyScript benchmarks how long it takes to analyze a very large
+// script to determine if it is a standard pay-to-pubkey script.
+func BenchmarkIsPubKeyScript(b *testing.B) {
+	script, err := genComplexScript()
+	if err != nil {
+		b.Fatalf("failed to create benchmark script: %v", err)
+	}
+
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		_ = IsPayToPubKey(script)
+	}
+}

txscript/script.go

@@ -63,6 +63,16 @@ func isScriptHash(pops []parsedOpcode) bool {
 		pops[2].opcode.value == OP_EQUAL
 }
 
+// IsPayToPubKey returns true if the script is in the standard pay-to-pubkey
+// (P2PK) format, false otherwise.
+func IsPayToPubKey(script []byte) bool {
+	pops, err := parseScript(script)
+	if err != nil {
+		return false
+	}
+	return isPubkey(pops)
+}
+
 // IsPayToScriptHash returns true if the script is in the standard
 // pay-to-script-hash (P2SH) format, false otherwise.
 func IsPayToScriptHash(script []byte) bool {