From a503b05813672b9c3e96fdb95eac306733466f55 Mon Sep 17 00:00:00 2001
From: Dmitry Vagner <typecat@Dmitrys-MacBook-Pro.local>
Date: Mon, 19 Dec 2022 14:39:23 -0800
Subject: [PATCH] fix

---
 .../curve/bn254/field_arithmetic/power.asm    |  6 +++++
 evm/src/cpu/kernel/tests/bn254_field.rs       | 27 ++++++++++++++-----
 2 files changed, 26 insertions(+), 7 deletions(-)

diff --git a/evm/src/cpu/kernel/asm/curve/bn254/field_arithmetic/power.asm b/evm/src/cpu/kernel/asm/curve/bn254/field_arithmetic/power.asm
index c49faab0..fc85cb44 100644
--- a/evm/src/cpu/kernel/asm/curve/bn254/field_arithmetic/power.asm
+++ b/evm/src/cpu/kernel/asm/curve/bn254/field_arithmetic/power.asm
@@ -1,3 +1,9 @@
+global test_power:
+    // stack: sqr, f, sqr, out, ret_stack, out
+    %store_fp12
+    // stack:         sqr, out, ret_stack, out
+    %jump(power)
+
 /// def power(square):
 ///     power_init()
 ///     power_loop_0()
diff --git a/evm/src/cpu/kernel/tests/bn254_field.rs b/evm/src/cpu/kernel/tests/bn254_field.rs
index 4de7c889..be3c8a0e 100644
--- a/evm/src/cpu/kernel/tests/bn254_field.rs
+++ b/evm/src/cpu/kernel/tests/bn254_field.rs
@@ -12,14 +12,14 @@ type Fp2 = [U256; 2];
 type Fp6 = [Fp2; 3];
 type Fp12 = [Fp6; 2];
 
-const zero: Fp = U256::from(0);
+const ZERO: Fp = U256([0, 0, 0, 0]);
 
 fn embed_fp2(x: Fp) -> Fp2 {
-    [x, zero]
+    [x, ZERO]
 }
 
 fn embed_fp2_fp6(a: Fp2) -> Fp6 {
-    [a, embed_fp2(zero), embed_fp2(zero)]
+    [a, embed_fp2(ZERO), embed_fp2(ZERO)]
 }
 
 fn embed_fp6(x: Fp) -> Fp6 {
@@ -27,11 +27,11 @@ fn embed_fp6(x: Fp) -> Fp6 {
 }
 
 fn embed_fp12(x: Fp) -> Fp12 {
-    [embed_fp6(x), embed_fp6(zero)]
+    [embed_fp6(x), embed_fp6(ZERO)]
 }
 
 fn gen_fp() -> Fp {
-    let rng = thread_rng();
+    let mut rng = thread_rng();
     let x64 = rng.gen::<u64>();
     U256([x64, x64, x64, x64]) % BN_BASE
 }
@@ -161,8 +161,8 @@ fn sh(c: Fp6) -> Fp6 {
 fn sparse_embed(x: [U256; 5]) -> Fp12 {
     let [g0, g1, g1_, g2, g2_] = x;
     [
-        [embed_fp2(g0), [g1, g1_], embed_fp2(zero)],
-        [embed_fp2(zero), [g2, g2_], embed_fp2(zero)],
+        [embed_fp2(g0), [g1, g1_], embed_fp2(ZERO)],
+        [embed_fp2(ZERO), [g2, g2_], embed_fp2(ZERO)],
     ]
 }
 
@@ -651,3 +651,16 @@ fn test_frob_fp12() -> Result<()> {
 
     Ok(())
 }
+
+fn make_power_stack(f: Fp12) -> Vec<U256> {
+    let sqr = U256::from(100);
+    let out = U256::from(300);
+    let f: Vec<U256> = f.into_iter().flatten().flatten().collect();
+    let ret_stack = U256::from(KERNEL.global_labels["ret_stack"]);
+
+    let mut input = vec![sqr];
+    input.extend(f);
+    input.extend(vec![sqr, out, ret_stack, out]);
+    input.reverse();
+    input
+}