criminal hacks

Signed-off-by: Ignacio Hagopian <[email protected]>
gballet · Oct 12, 2024 · 4d4eabc · 4d4eabc
1 parent 26452f4
commit 4d4eabc
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/core/vm/instructions.go b/core/vm/instructions.go
@@ -825,6 +825,9 @@ func opCall(pc *uint64, interpreter *EVMInterpreter, scope *ScopeContext) ([]byt
 	if err != nil || !scope.Contract.UseGas(dynamicCost) {
 		return nil, ErrOutOfGas
 	}
+	if memSize > 0 {
+		scope.Memory.Resize(memSize)
+	}
 
 	interpreter.evm.callGasTemp, err = callGas(interpreter.evm.chainRules.IsEIP150, scope.Contract.Gas, 0, scope.Stack.Back(0))
 	if err != nil {
@@ -886,6 +889,9 @@ func opCallCode(pc *uint64, interpreter *EVMInterpreter, scope *ScopeContext) ([
 	if err != nil || !scope.Contract.UseGas(dynamicCost) {
 		return nil, ErrOutOfGas
 	}
+	if memSize > 0 {
+		scope.Memory.Resize(memSize)
+	}
 	interpreter.evm.callGasTemp, err = callGas(interpreter.evm.chainRules.IsEIP150, scope.Contract.Gas, 0, scope.Stack.Back(0))
 	if err != nil {
 		return nil, err
@@ -940,6 +946,9 @@ func opDelegateCall(pc *uint64, interpreter *EVMInterpreter, scope *ScopeContext
 	if err != nil || !scope.Contract.UseGas(dynamicCost) {
 		return nil, ErrOutOfGas
 	}
+	if memSize > 0 {
+		scope.Memory.Resize(memSize)
+	}
 	interpreter.evm.callGasTemp, err = callGas(interpreter.evm.chainRules.IsEIP150, scope.Contract.Gas, 0, scope.Stack.Back(0))
 	if err != nil {
 		return nil, err
@@ -988,6 +997,9 @@ func opStaticCall(pc *uint64, interpreter *EVMInterpreter, scope *ScopeContext)
 	if err != nil || !scope.Contract.UseGas(dynamicCost) {
 		return nil, ErrOutOfGas
 	}
+	if memSize > 0 {
+		scope.Memory.Resize(memSize)
+	}
 	interpreter.evm.callGasTemp, err = callGas(interpreter.evm.chainRules.IsEIP150, scope.Contract.Gas, 0, scope.Stack.Back(0))
 	if err != nil {
 		return nil, err

diff --git a/core/vm/interpreter.go b/core/vm/interpreter.go
@@ -233,7 +233,8 @@ func (in *EVMInterpreter) Run(contract *Contract, input []byte, readOnly bool) (
 				in.evm.Config.Tracer.CaptureState(pc, op, gasCopy, cost, callContext, in.returnData, in.evm.depth, err)
 				logged = true
 			}
-			if memorySize > 0 {
+			// TODO(hack): remove if conditions
+			if op != CALL && op != CALLCODE && op != DELEGATECALL && op != STATICCALL && memorySize > 0 {
 				mem.Resize(memorySize)
 			}
 		} else if debug {