mirror of
https://github.com/logos-storage/plonky2.git
synced 2026-01-03 14:23:07 +00:00
fix
This commit is contained in:
Nicholas Ward
parent
050c2e657e
commit
cae5a2cf21
@ -209,6 +209,9 @@ l_E_prime_return:
|
||||
DUP1
|
||||
%ceil_div_const(16)
|
||||
// stack: num_limbs, num_bytes, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
ISZERO
|
||||
%jumpi(copy_b_end)
|
||||
SWAP1
|
||||
// stack: num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
%stack () -> (@SEGMENT_CALLDATA, 96)
|
||||
@ -223,6 +226,7 @@ l_E_prime_return:
|
||||
// stack: b_loc=0, num_limbs, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
%store_limbs
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
copy_b_end:
|
||||
|
||||
// Copy E to kernel general memory.
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
@ -233,6 +237,9 @@ l_E_prime_return:
|
||||
DUP1
|
||||
%ceil_div_const(16)
|
||||
// stack: num_limbs, num_bytes, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
ISZERO
|
||||
%jumpi(copy_e_end)
|
||||
SWAP1
|
||||
// stack: num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP7
|
||||
@ -247,6 +254,7 @@ l_E_prime_return:
|
||||
// stack: e_loc=len, num_limbs, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
%store_limbs
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
copy_e_end:
|
||||
|
||||
// Copy M to kernel general memory.
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
@ -257,6 +265,9 @@ l_E_prime_return:
|
||||
DUP1
|
||||
%ceil_div_const(16)
|
||||
// stack: num_limbs, num_bytes, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
ISZERO
|
||||
%jumpi(copy_m_end)
|
||||
SWAP1
|
||||
// stack: num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP7
|
||||
@ -274,6 +285,7 @@ l_E_prime_return:
|
||||
// stack: m_loc=2*len, num_limbs, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
%store_limbs
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
copy_m_end:
|
||||
|
||||
%stack (len, l_M, ls: 2) -> (len, l_M)
|
||||
// stack: len, l_M, kexit_info
|
||||
|
||||
388
evm/src/cpu/kernel/asm/core/precompiles/expmod_test.asm
Normal file
388
evm/src/cpu/kernel/asm/core/precompiles/expmod_test.asm
Normal file
@ -0,0 +1,388 @@
|
||||
// Load bytes, packing 16 bytes into each limb, and store limbs on the stack.
|
||||
// We pass around total_num_limbs and len for conveience, because we can't access them from the stack
|
||||
// if they're hidden behind the variable number of limbs.
|
||||
mload_bytes_as_limbs:
|
||||
// stack: ctx, segment, offset, num_bytes, retdest, total_num_limbs, len, ..limbs
|
||||
DUP4
|
||||
// stack: num_bytes, ctx, segment, offset, num_bytes, retdest, total_num_limbs, len, ..limbs
|
||||
%min_const(16)
|
||||
// stack: min(16, num_bytes), ctx, segment, offset, num_bytes, retdest, total_num_limbs, len, ..limbs
|
||||
%stack (len, addr: 3) -> (addr, len, addr)
|
||||
// stack: ctx, segment, offset, min(16, num_bytes), ctx, segment, offset, num_bytes, retdest, total_num_limbs, len, ..limbs
|
||||
%mload_packing
|
||||
// stack: new_limb, ctx, segment, offset, num_bytes, retdest, total_num_limbs, len, ..limbs
|
||||
%stack (new, addr: 3, numb, ret, tot, len) -> (numb, addr, ret, tot, len, new)
|
||||
// stack: num_bytes, ctx, segment, offset, retdest, total_num_limbs, len, new_limb, ..limbs
|
||||
DUP1
|
||||
%min_const(16)
|
||||
SWAP1
|
||||
SUB
|
||||
// stack: num_bytes_new = num_bytes - min(16, num_bytes), ctx, segment, offset, retdest, total_num_limbs, len, ..limbs
|
||||
DUP1
|
||||
ISZERO
|
||||
%jumpi(mload_bytes_return)
|
||||
// stack: num_bytes_new, ctx, segment, offset, retdest, total_num_limbs, len, ..limbs
|
||||
SWAP3
|
||||
%add_const(16)
|
||||
SWAP3
|
||||
// stack: num_bytes_new, ctx, segment, offset + 16, retdest, total_num_limbs, len, ..limbs
|
||||
%stack (num, addr: 3) -> (addr, num)
|
||||
%jump(mload_bytes_as_limbs)
|
||||
mload_bytes_return:
|
||||
// stack: num_bytes_new, ctx, segment, offset, retdest, total_num_limbs, len, ..limbs
|
||||
%pop4
|
||||
// stack: retdest, total_num_limbs, len, ..limbs
|
||||
JUMP
|
||||
|
||||
store_limbs:
|
||||
// stack: offset, retdest, num_limbs, limb[num_limbs - 1], ..limb[0]
|
||||
DUP3
|
||||
// stack: num_limbs, offset, retdest, num_limbs, limb[num_limbs - 1], ..limb[0]
|
||||
ISZERO
|
||||
%jumpi(store_limbs_return)
|
||||
// stack: offset, retdest, num_limbs, limb[num_limbs - 1], ..limb[0]
|
||||
%stack (offset, ret, num, limb) -> (offset, limb, offset, ret, num)
|
||||
// stack: offset, limb[num_limbs - 1], offset, retdest, num_limbs, limb[num_limbs - 2], ..limb[0]
|
||||
%mstore_kernel_general
|
||||
// stack: offset, retdest, num_limbs, limb[num_limbs - 2], ..limb[0]
|
||||
%increment
|
||||
SWAP2
|
||||
%decrement
|
||||
SWAP2
|
||||
// stack: offset + 1, retdest, num_limbs - 1, limb[num_limbs - 2], ..limb[0]
|
||||
%jump(store_limbs)
|
||||
store_limbs_return:
|
||||
// stack: offset, retdest, num_limbs=0
|
||||
POP
|
||||
SWAP1
|
||||
POP
|
||||
JUMP
|
||||
|
||||
calculate_l_E_prime:
|
||||
// stack: l_E, l_B, retdest
|
||||
DUP1
|
||||
// stack: l_E, l_E, l_B, retdest
|
||||
%le_const(32)
|
||||
// stack: l_E <= 32, l_E, l_B, retdest
|
||||
%jumpi(case_le_32)
|
||||
// stack: l_E, l_B, retdest
|
||||
PUSH 32
|
||||
// stack: 32, l_E, l_B, retdest
|
||||
DUP3
|
||||
// stack: l_B, 32, l_E, l_B, retdest
|
||||
%add_const(96)
|
||||
// stack: 96 + l_B, 32, l_E, l_B, retdest
|
||||
PUSH @SEGMENT_CALLDATA
|
||||
GET_CONTEXT
|
||||
%mload_packing
|
||||
// stack: i[96 + l_B..128 + l_B], l_E, l_B, retdest
|
||||
%log2_floor
|
||||
// stack: log2(i[96 + l_B..128 + l_B]), l_E, l_B, retdest
|
||||
SWAP1
|
||||
// stack: l_E, log2(i[96 + l_B..128 + l_B]), l_B, retdest
|
||||
%sub_const(32)
|
||||
%mul_const(8)
|
||||
// stack: 8 * (l_E - 32), log2(i[96 + l_B..128 + l_B]), l_B, retdest
|
||||
ADD
|
||||
// stack: 8 * (l_E - 32) + log2(i[96 + l_B..128 + l_B]), l_B, retdest
|
||||
SWAP2
|
||||
%pop2
|
||||
// stack: 8 * (l_E - 32) + log2(i[96 + l_B..128 + l_B]), retdest
|
||||
SWAP1
|
||||
// stack: retdest, 8 * (l_E - 32) + log2(i[96 + l_B..128 + l_B])
|
||||
JUMP
|
||||
case_le_32:
|
||||
// stack: l_E, l_B, retdest
|
||||
SWAP1
|
||||
// stack: l_B, l_E, retdest
|
||||
%add_const(96)
|
||||
// stack: 96 + l_B, l_E, retdest
|
||||
PUSH @SEGMENT_CALLDATA
|
||||
GET_CONTEXT
|
||||
%mload_packing
|
||||
// stack: E, retdest
|
||||
%log2_floor
|
||||
// stack: log2(E), retdest
|
||||
SWAP1
|
||||
// stack: retdest, log2(E)
|
||||
JUMP
|
||||
|
||||
global expmod_test:
|
||||
// stack: address, retdest, new_ctx, (old stack)
|
||||
%pop2
|
||||
// stack: new_ctx, (old stack)
|
||||
DUP1
|
||||
SET_CONTEXT
|
||||
// stack: (empty)
|
||||
PUSH 0x100000000 // = 2^32 (is_kernel = true)
|
||||
// stack: kexit_info
|
||||
|
||||
// Load l_B from i[0..32].
|
||||
%stack () -> (@SEGMENT_CALLDATA, 0, 32)
|
||||
// stack: @SEGMENT_CALLDATA, 0, 32, kexit_info
|
||||
GET_CONTEXT
|
||||
// stack: ctx, @SEGMENT_CALLDATA, 0, 32, kexit_info
|
||||
%mload_packing
|
||||
// stack: l_B, kexit_info
|
||||
|
||||
// Load l_E from i[32..64].
|
||||
%stack () -> (@SEGMENT_CALLDATA, 32, 32)
|
||||
GET_CONTEXT
|
||||
%mload_packing
|
||||
// stack: l_E, l_B, kexit_info
|
||||
|
||||
// Load l_M from i[64..96].
|
||||
%stack () -> (@SEGMENT_CALLDATA, 64, 32)
|
||||
GET_CONTEXT
|
||||
%mload_packing
|
||||
// stack: l_M, l_E, l_B, kexit_info
|
||||
|
||||
%stack (l: 3) -> (l, l)
|
||||
// stack: l_M, l_E, l_B, l_M, l_E, l_B, kexit_info
|
||||
%max_3
|
||||
// stack: max_len, l_M, l_E, l_B, kexit_info
|
||||
|
||||
%ceil_div_const(16)
|
||||
// stack: len=ceil(max_len/16), l_M, l_E, l_B, kexit_info
|
||||
|
||||
// Calculate gas costs.
|
||||
|
||||
PUSH l_E_prime_return
|
||||
// stack: l_E_prime_return, len, l_M, l_E, l_B, kexit_info
|
||||
DUP5
|
||||
DUP5
|
||||
// stack: l_E, l_B, l_E_prime_return, len, l_M, l_E, l_B, kexit_info
|
||||
%jump(calculate_l_E_prime)
|
||||
l_E_prime_return:
|
||||
// stack: l_E_prime, len, l_M, l_E, l_B, kexit_info
|
||||
DUP5
|
||||
// stack: l_B, l_E_prime, len, l_M, l_E, l_B, kexit_info
|
||||
DUP4
|
||||
// stack: l_M, l_B, l_E_prime, len, l_M, l_E, l_B, kexit_info
|
||||
%max
|
||||
// stack: max(l_M, l_B), l_E_prime, len, l_M, l_E, l_B, kexit_info
|
||||
%expmod_gas_f
|
||||
// stack: f(max(l_M, l_B)), l_E_prime, len, l_M, l_E, l_B, kexit_info
|
||||
SWAP1
|
||||
// stack: l_E_prime, f(max(l_M, l_B)), len, l_M, l_E, l_B, kexit_info
|
||||
%max_const(1)
|
||||
// stack: max(1, l_E_prime), f(max(l_M, l_B)), len, l_M, l_E, l_B, kexit_info
|
||||
MUL
|
||||
// stack: max(1, l_E_prime) * f(max(l_M, l_B)), len, l_M, l_E, l_B, kexit_info
|
||||
%div_const(3) // G_quaddivisor
|
||||
// stack: (max(1, l_E_prime) * f(max(l_M, l_B))) / G_quaddivisor, len, l_M, l_E, l_B, kexit_info
|
||||
%max_const(200)
|
||||
// stack: g_r, len, l_M, l_E, l_B, kexit_info
|
||||
%stack (g_r, l: 4, kexit_info) -> (g_r, kexit_info, l)
|
||||
// stack: g_r, kexit_info, len, l_M, l_E, l_B
|
||||
POP // %charge_gas
|
||||
// stack: kexit_info, len, l_M, l_E, l_B
|
||||
%stack (kexit_info, l: 4) -> (l, kexit_info)
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
|
||||
// Copy B to kernel general memory.
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
// stack: len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP5
|
||||
// stack: num_bytes=l_B, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
%ceil_div_const(16)
|
||||
// stack: num_limbs, num_bytes, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
ISZERO
|
||||
%jumpi(copy_b_end)
|
||||
SWAP1
|
||||
// stack: num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
%stack () -> (@SEGMENT_CALLDATA, 96)
|
||||
GET_CONTEXT
|
||||
// stack: ctx, @SEGMENT_CALLDATA, 96, num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
%mload_bytes_as_limbs
|
||||
// stack: num_limbs, len, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
SWAP1
|
||||
POP
|
||||
// stack: num_limbs, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
PUSH 0
|
||||
// stack: b_loc=0, num_limbs, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
%store_limbs
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
copy_b_end:
|
||||
|
||||
// Copy E to kernel general memory.
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
// stack: len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP4
|
||||
// stack: num_bytes=l_E, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
%ceil_div_const(16)
|
||||
// stack: num_limbs, num_bytes, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
ISZERO
|
||||
%jumpi(copy_e_end)
|
||||
SWAP1
|
||||
// stack: num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP7
|
||||
%add_const(96)
|
||||
// stack: 96 + l_B, num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
PUSH @SEGMENT_CALLDATA
|
||||
GET_CONTEXT
|
||||
// stack: ctx, @SEGMENT_CALLDATA, 96 + l_B, num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
%mload_bytes_as_limbs
|
||||
// stack: num_limbs, len, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
SWAP1
|
||||
// stack: e_loc=len, num_limbs, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
%store_limbs
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
copy_e_end:
|
||||
|
||||
// Copy M to kernel general memory.
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
// stack: len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP3
|
||||
// stack: num_bytes=l_M, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
%ceil_div_const(16)
|
||||
// stack: num_limbs, num_bytes, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP1
|
||||
ISZERO
|
||||
%jumpi(copy_m_end)
|
||||
SWAP1
|
||||
// stack: num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
DUP7
|
||||
DUP7
|
||||
ADD
|
||||
%add_const(96)
|
||||
// stack: 96 + l_B + l_E, num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
PUSH @SEGMENT_CALLDATA
|
||||
GET_CONTEXT
|
||||
// stack: ctx, @SEGMENT_CALLDATA, 96 + l_B + l_E, num_bytes, num_limbs, len, len, l_M, l_E, l_B, kexit_info
|
||||
%mload_bytes_as_limbs
|
||||
// stack: num_limbs, len, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
SWAP1
|
||||
%mul_const(2)
|
||||
// stack: m_loc=2*len, num_limbs, limbs[num_limbs-1], .., limbs[0], len, l_M, l_E, l_B, kexit_info
|
||||
%store_limbs
|
||||
// stack: len, l_M, l_E, l_B, kexit_info
|
||||
copy_m_end:
|
||||
|
||||
%stack (len, l_M, ls: 2) -> (len, l_M)
|
||||
// stack: len, l_M, kexit_info
|
||||
|
||||
PUSH expmod_contd
|
||||
// stack: expmod_contd, len, l_M, kexit_info
|
||||
DUP2
|
||||
// stack: len, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
DUP1
|
||||
%mul_const(11)
|
||||
// stack: s5=11*len, len, expmod_contd, len, l_M, kexit_info
|
||||
SWAP1
|
||||
// stack: len, s5, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
DUP1
|
||||
%mul_const(9)
|
||||
// stack: s4=9*len, len, s5, expmod_contd, len, l_M, kexit_info
|
||||
SWAP1
|
||||
// stack: len, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
DUP1
|
||||
%mul_const(7)
|
||||
// stack: s3=7*len, len, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
SWAP1
|
||||
// stack: len, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
DUP1
|
||||
%mul_const(5)
|
||||
// stack: s2=5*len, len, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
SWAP1
|
||||
// stack: len, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
DUP1
|
||||
%mul_const(4)
|
||||
// stack: s1=4*len, len, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
SWAP1
|
||||
// stack: len, s1, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
DUP1
|
||||
%mul_const(3)
|
||||
// stack: out=3*len, len, s1, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
SWAP1
|
||||
// stack: len, out, s1, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
DUP1
|
||||
%mul_const(2)
|
||||
// stack: m_loc=2*len, len, out, s1, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
SWAP1
|
||||
// stack: len, m_loc, out, s1, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
PUSH 0
|
||||
// stack: b_loc=0, e_loc=len, m_loc, out, s1, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
DUP2
|
||||
// stack: len, b_loc, e_loc, m_loc, out, s1, s2, s3, s4, s5, expmod_contd, len, l_M, kexit_info
|
||||
|
||||
%jump(modexp_bignum)
|
||||
|
||||
expmod_contd:
|
||||
// stack: len, l_M, kexit_info
|
||||
|
||||
// Copy the result value from kernel general memory to the parent's return data.
|
||||
|
||||
// Store return data size: l_M (number of bytes).
|
||||
SWAP1
|
||||
// stack: l_M, len, kexit_info
|
||||
%mstore_parent_context_metadata(@CTX_METADATA_RETURNDATA_SIZE)
|
||||
// stack: len, kexit_info
|
||||
DUP1
|
||||
// stack: len, len, kexit_info
|
||||
%mul_const(3)
|
||||
// stack: out=3*len, len, kexit_info
|
||||
DUP2
|
||||
DUP2
|
||||
// stack: out, len, out, len, kexit_info
|
||||
ADD
|
||||
%decrement
|
||||
SWAP1
|
||||
%decrement
|
||||
SWAP1
|
||||
// stack: cur_address=out+len-1, end_address=out-1, len, kexit_info
|
||||
PUSH 0
|
||||
// stack: i=0, cur_address, end_address, len, kexit_info
|
||||
|
||||
// Store in big-endian format.
|
||||
expmod_store_loop:
|
||||
// stack: i, cur_address, end_address, len, kexit_info
|
||||
DUP2
|
||||
// stack: cur_address, i, cur_address, end_address, len, kexit_info
|
||||
%mload_kernel_general
|
||||
// stack: cur_limb, i, cur_address, end_address, len, kexit_info
|
||||
DUP2
|
||||
// stack: i, cur_limb, i, cur_address, end_address, len, kexit_info
|
||||
%mul_const(16)
|
||||
// stack: offset=16*i, cur_limb, i, cur_address, end_address, len, kexit_info
|
||||
%stack (offset, cur_limb) -> (@SEGMENT_RETURNDATA, offset, cur_limb, 16)
|
||||
// stack: @SEGMENT_RETURNDATA, offset, cur_limb, 16, i, cur_address, end_address, len, kexit_info
|
||||
%mload_context_metadata(@CTX_METADATA_PARENT_CONTEXT)
|
||||
// stack: parent_ctx, @SEGMENT_RETURNDATA, offset, cur_limb, 16, i, cur_address, end_address, len, kexit_info
|
||||
%mstore_unpacking
|
||||
// stack: offset', i, cur_address, end_address, len, kexit_info
|
||||
POP
|
||||
// stack: i, cur_address, end_address, len, kexit_info
|
||||
%increment
|
||||
SWAP1
|
||||
%decrement
|
||||
SWAP1
|
||||
// stack: i+1, cur_address-1, end_address, len, kexit_info
|
||||
DUP3
|
||||
DUP2
|
||||
EQ
|
||||
ISZERO
|
||||
%jumpi(expmod_store_loop)
|
||||
expmod_store_end:
|
||||
// stack: i, cur_address, len, kexit_info
|
||||
%pop3
|
||||
// stack: kexit_info
|
||||
PUSH 0
|
||||
// stack: dummy=0, kexit_info
|
||||
STOP
|
||||
Loading…
x
Reference in New Issue
Block a user