From cab035d5eef415af7cb0c007bdd569e92aa4ffc5 Mon Sep 17 00:00:00 2001
From: Lorenzo Gentile <lorenzogentile404@gmail.com>
Date: Tue, 22 Oct 2024 22:43:53 +0200
Subject: [PATCH] test(mxp): add failing tests for debugging purposes

---
 .../hub/ZeroSizeCallDataOrReturnDataTest.java | 108 ++++++++++++++++++
 .../linea/zktracer/module/mxp/MxpTest.java    |   3 +-
 2 files changed, 110 insertions(+), 1 deletion(-)
 create mode 100644 arithmetization/src/test/java/net/consensys/linea/zktracer/module/hub/ZeroSizeCallDataOrReturnDataTest.java

diff --git a/arithmetization/src/test/java/net/consensys/linea/zktracer/module/hub/ZeroSizeCallDataOrReturnDataTest.java b/arithmetization/src/test/java/net/consensys/linea/zktracer/module/hub/ZeroSizeCallDataOrReturnDataTest.java
new file mode 100644
index 0000000000..87023b3d34
--- /dev/null
+++ b/arithmetization/src/test/java/net/consensys/linea/zktracer/module/hub/ZeroSizeCallDataOrReturnDataTest.java
@@ -0,0 +1,108 @@
+/*
+ * Copyright ConsenSys Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package net.consensys.linea.zktracer.module.hub;
+
+import java.util.List;
+
+import net.consensys.linea.testing.BytecodeCompiler;
+import net.consensys.linea.testing.BytecodeRunner;
+import net.consensys.linea.testing.ToyAccount;
+import net.consensys.linea.zktracer.opcode.OpCode;
+import org.hyperledger.besu.datatypes.Address;
+import org.hyperledger.besu.datatypes.Wei;
+import org.junit.jupiter.api.Test;
+
+public class ZeroSizeCallDataOrReturnDataTest {
+
+  @Test
+  void zeroSizeHugeReturnAtOffsetTest() {
+    BytecodeCompiler program = BytecodeCompiler.newProgram();
+    program
+        .push(0)
+        .push("ff".repeat(32)) // return data offset
+        .push(0)
+        .push(0)
+        .push("ca11ee") // address
+        .push(1000) // gas
+        .op(OpCode.STATICCALL);
+
+    BytecodeCompiler calleeProgram = BytecodeCompiler.newProgram();
+    calleeProgram.op(OpCode.CALLDATASIZE);
+
+    final ToyAccount calleeAccount =
+        ToyAccount.builder()
+            .balance(Wei.fromEth(1))
+            .nonce(10)
+            .address(Address.fromHexString("ca11ee"))
+            .code(calleeProgram.compile())
+            .build();
+
+    BytecodeRunner.of(program.compile()).run(Wei.fromEth(1), 30000L, List.of(calleeAccount));
+    // TODO: this test is supposed to fail as the ones below, but it does not. Understand why
+  }
+
+  @Test
+  void zeroSizeHugeCallDataOffsetTest() {
+    BytecodeCompiler program = BytecodeCompiler.newProgram();
+    program
+        .push(0)
+        .push(0)
+        .push(0)
+        .push("ff".repeat(32)) // call data offset
+        .push("ca11ee") // address
+        .push(1000) // gas
+        .op(OpCode.STATICCALL);
+
+    BytecodeCompiler calleeProgram = BytecodeCompiler.newProgram();
+    calleeProgram.op(OpCode.CALLDATASIZE);
+
+    final ToyAccount calleeAccount =
+        ToyAccount.builder()
+            .balance(Wei.fromEth(1))
+            .nonce(10)
+            .address(Address.fromHexString("ca11ee"))
+            .code(calleeProgram.compile())
+            .build();
+
+    BytecodeRunner.of(program.compile()).run(Wei.fromEth(1), 30000L, List.of(calleeAccount));
+  }
+
+  @Test
+  void zeroSizeHugeReturnDataOffsetTest() {
+    BytecodeCompiler program = BytecodeCompiler.newProgram();
+    program
+        .push(0)
+        .push(0)
+        .push(0)
+        .push(0)
+        .push("ca11ee") // address
+        .push(1000) // gas
+        .op(OpCode.STATICCALL);
+
+    BytecodeCompiler calleeProgram = BytecodeCompiler.newProgram();
+    calleeProgram.push(0).push("ff".repeat(32)).op(OpCode.RETURN);
+
+    final ToyAccount calleeAccount =
+        ToyAccount.builder()
+            .balance(Wei.fromEth(1))
+            .nonce(10)
+            .address(Address.fromHexString("ca11ee"))
+            .code(calleeProgram.compile())
+            .build();
+
+    BytecodeRunner.of(program.compile()).run(Wei.fromEth(1), 30000L, List.of(calleeAccount));
+  }
+}
diff --git a/arithmetization/src/test/java/net/consensys/linea/zktracer/module/mxp/MxpTest.java b/arithmetization/src/test/java/net/consensys/linea/zktracer/module/mxp/MxpTest.java
index a86e2a63ed..f08cb20808 100644
--- a/arithmetization/src/test/java/net/consensys/linea/zktracer/module/mxp/MxpTest.java
+++ b/arithmetization/src/test/java/net/consensys/linea/zktracer/module/mxp/MxpTest.java
@@ -419,7 +419,8 @@ private void triggerNonTrivialOrNoop(BytecodeCompiler program, boolean isHalting
         appendOpCodeCall(List.of(value, offset1), opCode, program);
         break;
       case LOG0, SHA3, RETURN, REVERT: // RETURN and REVERT are selected only when isHalting is true
-        System.out.println(opCode + " size: " + size1.toBigInteger() + " offset: " + offset1.toBigInteger());
+        System.out.println(
+            opCode + " size: " + size1.toBigInteger() + " offset: " + offset1.toBigInteger());
         appendOpCodeCall(List.of(size1, offset1), opCode, program);
         if (opCode == OpCode.SHA3) {
           program.op(OpCode.POP);