Update precompiles to London

aurora-is-near · Oct 27, 2021 · bfd0b37 · bfd0b37
1 parent 44f2944
commit bfd0b37
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 4 deletions.
diff --git a/engine-precompiles/src/lib.rs b/engine-precompiles/src/lib.rs
@@ -208,9 +208,44 @@ impl Precompiles {
         Precompiles(map)
     }
 
-    #[allow(dead_code)]
-    fn new_berlin() -> Self {
-        Self::new_istanbul()
+    pub fn new_berlin() -> Self {
+        let addresses = prelude::vec![
+            ECRecover::ADDRESS,
+            SHA256::ADDRESS,
+            RIPEMD160::ADDRESS,
+            Identity::ADDRESS,
+            ModExp::<Berlin>::ADDRESS,
+            Bn128Add::<Istanbul>::ADDRESS,
+            Bn128Mul::<Istanbul>::ADDRESS,
+            Bn128Pair::<Istanbul>::ADDRESS,
+            Blake2F::ADDRESS,
+            ExitToNear::ADDRESS,
+            ExitToEthereum::ADDRESS,
+        ];
+        let fun: prelude::Vec<PrecompileFn> = prelude::vec![
+            ECRecover::run,
+            SHA256::run,
+            RIPEMD160::run,
+            Identity::run,
+            ModExp::<Berlin>::run,
+            Bn128Add::<Istanbul>::run,
+            Bn128Mul::<Istanbul>::run,
+            Bn128Pair::<Istanbul>::run,
+            Blake2F::run,
+            ExitToNear::run,
+            ExitToEthereum::run,
+        ];
+        let mut map = prelude::BTreeMap::new();
+        for (address, fun) in addresses.into_iter().zip(fun) {
+            map.insert(address, fun);
+        }
+
+        Precompiles(map)
+    }
+
+    pub fn new_london() -> Self {
+        // no precompile changes in London HF
+        Self::new_berlin()
     }
 }
 

diff --git a/engine/src/engine.rs b/engine/src/engine.rs
@@ -265,7 +265,7 @@ struct StackExecutorParams {
 impl StackExecutorParams {
     fn new(gas_limit: u64) -> Self {
         Self {
-            precompiles: Precompiles::new_istanbul(),
+            precompiles: Precompiles::new_london(),
             gas_limit,
         }
     }