fix: account for BEACON_ROOTS_ADDRESS access (#2382)

* fix: account for BEACON_ROOTS_ADDRESS access

It seems that, post CANCUN, both SSTORE and SLOAD invoke the system
transaction and that this results in the BEACON_ROOTS_ADDRESS being
accessed.  The existing tests did not account for this.

* support Prague for addresses seen by system tx

This adds support for the number of addresses seen in Prague by the
system transaction.

Author: DavePearce

arithmetization/src/main/java/net/consensys/linea/zktracer/Fork.java

@@ -187,6 +187,22 @@ public static GasCalculator getGasCalculatorFromFork(Fork fork) {
     };
   }
 
+  /**
+   * Return the number of contract addresses seen by the system transaction during execution. This
+   * is primary to testing purposes to ensure the right number were seen.
+   *
+   * @param fork
+   * @return
+   */
+  public static int numberOfAddressesSeenBySystemTransaction(Fork fork) {
+    return switch (fork) {
+      case LONDON, PARIS, SHANGHAI -> 0;
+      case CANCUN -> 1;
+      case PRAGUE -> 2;
+      default -> throw new IllegalArgumentException("Unknown fork: " + fork);
+    };
+  }
+
   // Used for blockchain ref tests with the Paris exception of "Merge"
   public static String toPascalCase(Fork fork) {
     return switch (fork) {

arithmetization/src/test/java/net/consensys/linea/zktracer/statemanager/HubShomeiTests.java

@@ -28,6 +28,7 @@
 import net.consensys.linea.testing.ToyExecutionEnvironmentV2;
 import net.consensys.linea.testing.ToyTransaction;
 import net.consensys.linea.zktracer.ChainConfig;
+import net.consensys.linea.zktracer.Fork;
 import net.consensys.linea.zktracer.ZkTracer;
 import net.consensys.linea.zktracer.opcode.OpCode;
 import org.apache.tuweni.bytes.Bytes;
@@ -127,7 +128,7 @@ void sandwichPrewarming(OpCode opcode, TestInfo testInfo) {
     final Set<Address> addressSeen = tracer.getAddressesSeenByHubForRelativeBlock(1);
     final Map<Address, Set<Bytes32>> storageSeen = tracer.getStoragesSeenByHubForRelativeBlock(1);
 
-    assert (addressSeen.size() == 4);
+    assert (addressSeen.size() == 4 + Fork.numberOfAddressesSeenBySystemTransaction(fork));
     assert (addressSeen.contains(senderAddress));
     assert (addressSeen.contains(recipientAccount.getAddress()));
     assert (addressSeen.contains(DEFAULT_COINBASE_ADDRESS));
@@ -182,7 +183,7 @@ void uselessPrewarming(OpCode opcode, TestInfo testInfo) {
     final Set<Address> addressSeen = tracer.getAddressesSeenByHubForRelativeBlock(1);
     final Map<Address, Set<Bytes32>> storageSeen = tracer.getStoragesSeenByHubForRelativeBlock(1);
 
-    assert (addressSeen.size() == 3);
+    assert (addressSeen.size() == 3 + Fork.numberOfAddressesSeenBySystemTransaction(fork));
     assert (addressSeen.contains(senderAddress));
     assert (addressSeen.contains(recipientAccount.getAddress()));
     assert (addressSeen.contains(DEFAULT_COINBASE_ADDRESS));