Add support for validation rule OP-080 (#394)

stackup-wallet · Apr 18, 2024 · f145dfd · f145dfd
1 parent 85e227c
commit f145dfd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/pkg/entrypoint/simulation/simulateutils.go b/pkg/entrypoint/simulation/simulateutils.go
@@ -25,15 +25,19 @@ var (
 		"BASEFEE",
 		"BLOCKHASH",
 		"NUMBER",
-		"SELFBALANCE",
-		"BALANCE",
 		"ORIGIN",
 		"GAS",
 		"CREATE",
 		"COINBASE",
 		"SELFDESTRUCT",
 	)
 
+	// List of opcodes not allowed during validation for unstaked entities.
+	bannedUnstakedOpCodes = mapset.NewSet(
+		"SELFBALANCE",
+		"BALANCE",
+	)
+
 	revertOpCode = "REVERT"
 	returnOpCode = "RETURN"
 

diff --git a/pkg/entrypoint/simulation/tracevalidation.go b/pkg/entrypoint/simulation/tracevalidation.go
@@ -93,6 +93,10 @@ func TraceSimulateValidation(in *TraceInput) (*TraceOutput, error) {
 			if bannedOpCodes.Contains(opcode) {
 				return nil, fmt.Errorf("%s uses banned opcode: %s", title, opcode)
 			}
+
+			if !entity.IsStaked && bannedUnstakedOpCodes.Contains(opcode) {
+				return nil, fmt.Errorf("unstaked %s uses banned opcode: %s", title, opcode)
+			}
 		}
 
 		ic.Add(entity.Address)