From: DeKaiju <longjinglv@gmail.com>
Date: Tue, 6 Apr 2021 09:27:57 +0000 (+0800)
Subject: feat(bcrp): implement IsBCRPScript func and unit test (#1857)
X-Git-Tag: 2.0.0-alpha~56
X-Git-Url: http://git.osdn.net/view?a=commitdiff_plain;ds=sidebyside;h=efab3273f42eb4be1df53e9b3fcbff77c20d3eca;p=bytom%2Fbytom.git

feat(bcrp): implement IsBCRPScript func and unit test (#1857)
---

diff --git a/consensus/segwit/segwit.go b/consensus/segwit/segwit.go
index b85c4e81..6f70d2cf 100644
--- a/consensus/segwit/segwit.go
+++ b/consensus/segwit/segwit.go
@@ -1,6 +1,7 @@
 package segwit
 
 import (
+	"bytes"
 	"errors"
 
 	"github.com/bytom/bytom/consensus"
@@ -8,6 +9,9 @@ import (
 	"github.com/bytom/bytom/protocol/vm/vmutil"
 )
 
+//bcrp bytom contract register protocol
+const bcrp = "bcrp"
+
 func IsP2WScript(prog []byte) bool {
 	return IsP2WPKHScript(prog) || IsP2WSHScript(prog) || IsStraightforward(prog)
 }
@@ -51,6 +55,46 @@ func IsP2WSHScript(prog []byte) bool {
 	return insts[1].Op == vm.OP_DATA_32 && len(insts[1].Data) == consensus.PayToWitnessScriptHashDataSize
 }
 
+// IsBCRPScript checks if a control program is bytom contract register protocol
+// BCRP script format: OP_FAIL + OP_PUSHDATA1 + "04" + hex("bcrp") + OP_PUSHDATA1 + "01" + "01" + OP_PUSHDATA1 + len(contract) + contract
+func IsBCRPScript(prog []byte) bool {
+	inst, err := vm.ParseProgram(prog)
+	if err != nil {
+		return false
+	}
+
+	if len(inst) != 4 {
+		return false
+	}
+
+	if inst[0].Op != vm.OP_FAIL {
+		return false
+	}
+
+	if inst[1].Op != vm.OP_PUSHDATA1 {
+		return false
+	}
+
+	if !bytes.Equal(inst[1].Data, []byte(bcrp)) {
+		return false
+	}
+
+	if inst[2].Op != vm.OP_PUSHDATA1 {
+		return false
+	}
+
+	// version 1
+	if !bytes.Equal(inst[2].Data, []byte{byte(1)}) {
+		return false
+	}
+
+	if inst[3].Op != vm.OP_PUSHDATA1 {
+		return false
+	}
+
+	return true
+}
+
 func ConvertP2PKHSigProgram(prog []byte) ([]byte, error) {
 	insts, err := vm.ParseProgram(prog)
 	if err != nil {
diff --git a/consensus/segwit/segwit_test.go b/consensus/segwit/segwit_test.go
index 14ec13b6..dfc7cd82 100644
--- a/consensus/segwit/segwit_test.go
+++ b/consensus/segwit/segwit_test.go
@@ -119,3 +119,47 @@ func TestProgramType(t *testing.T) {
 		}
 	}
 }
+
+func TestIsBCRPScript(t *testing.T) {
+	tests := []struct {
+		program  string
+		expected bool
+	}{
+		{
+			program:  "",
+			expected: false,
+		},
+		{
+			program:  "ae20ac20f5cdb9ada2ae9836bcfff32126d6b885aa3f73ee111a95d1bf37f3904aca5151ad",
+			expected: false,
+		},
+		{
+			program:  "694c04626372704c01014c2820e9108d3ca8049800727f6a3505b3a2710dc579405dde03c250f16d9a7e1e6e787403ae7cac00c0",
+			expected: false,
+		},
+		{
+			program:  "6a4c04424352504c01014c2820e9108d3ca8049800727f6a3505b3a2710dc579405dde03c250f16d9a7e1e6e787403ae7cac00c0",
+			expected: false,
+		},
+		{
+			program:  "6a4c04626372704c01024c2820e9108d3ca8049800727f6a3505b3a2710dc579405dde03c250f16d9a7e1e6e787403ae7cac00c0",
+			expected: false,
+		},
+		{
+			program:  "6a4c04626372704c01014c2820e9108d3ca8049800727f6a3505b3a2710dc579405dde03c250f16d9a7e1e6e787403ae7cac00c0",
+			expected: true,
+		},
+	}
+
+	for i, test := range tests {
+		program, err := hex.DecodeString(test.program)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		expected := IsBCRPScript(program)
+		if expected != test.expected {
+			t.Errorf("TestIsTemplateRegister #%d failed: got %v want %v", i, expected, test.expected)
+		}
+	}
+}