From 2f8f078f0edaea8491e4daf9d4dce055bfaa35d2 Mon Sep 17 00:00:00 2001
From: Dave Collins <davec@conformal.com>
Date: Wed, 13 Mar 2019 01:11:02 -0500
Subject: [PATCH] txscript: Add benchmark for script parsing.

---
 txscript/bench_test.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/txscript/bench_test.go b/txscript/bench_test.go
index 2fab049a..15e6143f 100644
--- a/txscript/bench_test.go
+++ b/txscript/bench_test.go
@@ -49,3 +49,42 @@ func BenchmarkCalcSigHash(b *testing.B) {
 		}
 	}
 }
+
+// genComplexScript returns a script comprised of half as many opcodes as the
+// maximum allowed followed by as many max size data pushes fit without
+// exceeding the max allowed script size.
+func genComplexScript() ([]byte, error) {
+	var scriptLen int
+	builder := NewScriptBuilder()
+	for i := 0; i < MaxOpsPerScript/2; i++ {
+		builder.AddOp(OP_TRUE)
+		scriptLen++
+	}
+	maxData := bytes.Repeat([]byte{0x02}, MaxScriptElementSize)
+	for i := 0; i < (MaxScriptSize-scriptLen)/MaxScriptElementSize; i++ {
+		builder.AddData(maxData)
+	}
+	return builder.Script()
+}
+
+// BenchmarkScriptParsing benchmarks how long it takes to parse a very large
+// script.
+func BenchmarkScriptParsing(b *testing.B) {
+	script, err := genComplexScript()
+	if err != nil {
+		b.Fatalf("failed to create benchmark script: %v", err)
+	}
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		pops, err := parseScript(script)
+		if err != nil {
+			b.Fatalf("failed to parse script: %v", err)
+		}
+
+		for _, pop := range pops {
+			_ = pop.opcode
+			_ = pop.data
+		}
+	}
+}