fixes identity and not set CTX to 0

0xPolygonHermez · Feb 7, 2023 · 5d164c5 · 5d164c5
1 parent 9ea22fe
commit 5d164c5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/main/end.zkasm b/main/end.zkasm
@@ -1,4 +1,4 @@
 finalWait:
     ${beforeLast()}                                       :JMPN(finalWait)
-    ; Set all registers to 0 except inputs: B (oldstateRoot), C (oldAccInputHash), SP (oldNumBatch) & GAS (chainID)
-    0 => A, D, E, CTX, PC, MAXMEM, SR, HASHPOS, RR, RCX   :JMP(start)
+    ; Set all registers to 0 except inputs: B (oldstateRoot), C (oldAccInputHash), SP (oldNumBatch), GAS (chainID) & CTX (forkID)
+    0 => A, D, E, PC, MAXMEM, SR, HASHPOS, RR, RCX   :JMP(start)
diff --git a/main/precompiled/identity.zkasm b/main/precompiled/identity.zkasm
@@ -18,8 +18,9 @@ IDENTITY:
     GAS - %IDENTITY_WORD_GAS*A => GAS   :JMPN(outOfGas)
     CTX             :MSTORE(currentCTX)
     CTX => A
-    $ => CTX        :MLOAD(originCTX), JMPZ(handleGas)
-    A               :MSTORE(retDataCTX)
+    $ => B               :MLOAD(originCTX), JMPZ(handleGas)
+    B => CTX
+    A                    :MSTORE(retDataCTX)
     A => CTX
     0 => B
     0 => E