From c92ee806b3fdcde75965b21c43f906e9b813a7ed Mon Sep 17 00:00:00 2001
From: Manuel Bilbao <manuel.bilbao@lambdaclass.com>
Date: Fri, 12 Apr 2024 14:01:50 -0300
Subject: [PATCH] Add gas charge to remaining opcodes

---
 system-contracts/contracts/EvmInterpreter.yul | 22 ++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/system-contracts/contracts/EvmInterpreter.yul b/system-contracts/contracts/EvmInterpreter.yul
index 4e4071a9f..4600bb0f0 100644
--- a/system-contracts/contracts/EvmInterpreter.yul
+++ b/system-contracts/contracts/EvmInterpreter.yul
@@ -347,6 +347,8 @@ object "EVMInterpreter" {
                     b, sp := popStackItem(sp)
 
                     sp := pushStackItem(sp, or(a,b))
+
+                    evmGasLeft := chargeGas(evmGasLeft, 3)
                 }
                 case 0x0A { // OP_EXP
                     let a, exponent
@@ -355,6 +357,16 @@ object "EVMInterpreter" {
                     exponent, sp := popStackItem(sp)
 
                     sp := pushStackItem(sp, exp(a, exponent))
+
+                    let expSizeByte
+                    switch exponent {
+                        case 0 { expSizeByte := 0 }
+                        default {
+                            expSizeBytes := div(add(exponent, 256), 256)
+                        }
+                    }
+
+                    evmGasLeft := chargeGas(evmGasLeft, add(10, mul(50, expSizeBytes)))
                 }
                 case 0x0B { // OP_SIGNEXTEND
                     let b, x
@@ -363,6 +375,8 @@ object "EVMInterpreter" {
                     x, sp := popStackItem(sp)
 
                     sp := pushStackItem(sp, signextend(b, x))
+
+                    evmGasLeft := chargeGas(evmGasLeft, 5)
                 }
                 case 0x08 { // OP_ADDMOD
                     let a, b, N
@@ -372,6 +386,8 @@ object "EVMInterpreter" {
                     N, sp := popStackItem(sp)
 
                     sp := pushStackItem(sp, addmod(a, b, N))
+
+                    evmGasLeft := chargeGas(evmGasLeft, 8)
                 }
                 case 0x09 { // OP_MULMOD
                     let a, b, N
@@ -381,6 +397,8 @@ object "EVMInterpreter" {
                     N, sp := popStackItem(sp)
 
                     sp := pushStackItem(sp, mulmod(a, b, N))
+
+                    evmGasLeft := chargeGas(evmGasLeft, 8)
                 }
                 case 0x10 { // OP_LT
                     let a, b
@@ -517,7 +535,9 @@ object "EVMInterpreter" {
                     // TODO: Handle cold/warm slots and updates, etc for gas costs.
                     evmGasLeft := chargeGas(evmGasLeft, 100)
                 }
-                case 0x5B {} // OP_JUMPDEST
+                case 0x5B { // OP_JUMPDEST
+                    evmGasLeft := chargeGas(evmGasLeft, 1)
+                }
                 case 0x5F { // OP_PUSH0
                     let value := 0