Fix: smart rollup address validation (#1012)

baking-bad · Feb 7, 2024 · fd5d5c2 · fd5d5c2
1 parent 18a5847
commit fd5d5c2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/internal/bcd/ast/validators_test.go b/internal/bcd/ast/validators_test.go
@@ -32,6 +32,10 @@ func TestAddressValidator(t *testing.T) {
 			name:    "test 5",
 			value:   "0x6e4943f7a23ab9cbe56f48ff72f6c27e8956762400",
 			wantErr: true,
+		}, {
+			name:    "test 6",
+			value:   "sr18wx6ezkeRjt1SZSeZ2UQzQN3Uc3YLMLqg",
+			wantErr: false,
 		},
 	}
 	for _, tt := range tests {

diff --git a/internal/bcd/literal.go b/internal/bcd/literal.go
@@ -28,7 +28,9 @@ func IsContractLazy(str string) bool {
 
 // IsAddressLazy -
 func IsAddressLazy(address string) bool {
-	return (len(address) == 36 && (strings.HasPrefix(address, "KT") || strings.HasPrefix(address, "tz"))) ||
+	return (len(address) == 36 && (strings.HasPrefix(address, "KT") ||
+		strings.HasPrefix(address, "tz") ||
+		strings.HasPrefix(address, "sr1"))) ||
 		(len(address) == 37 && strings.HasPrefix(address, "txr"))
 }