mirror of
https://github.com/logos-storage/plonky2.git
synced 2026-01-21 07:03:07 +00:00
finished pad
This commit is contained in:
Nicholas Ward
parent
0d22b3f8da
commit
aba45e2d2c
@ -34,7 +34,6 @@ global sha2_store:
|
|||||||
// stack: num_u256s, x[0], x[1], x[2], ... , x[num_u256s-1]
|
// stack: num_u256s, x[0], x[1], x[2], ... , x[num_u256s-1]
|
||||||
push 1
|
push 1
|
||||||
// stack: addr=1, counter=num_u256s, x[0], x[1], x[2], ... , x[num_u256s-1]
|
// stack: addr=1, counter=num_u256s, x[0], x[1], x[2], ... , x[num_u256s-1]
|
||||||
|
|
||||||
sha2_store_loop:
|
sha2_store_loop:
|
||||||
JUMPDEST
|
JUMPDEST
|
||||||
// stack: addr, counter, x[num_u256s-counter], ... , x[num_u256s-1]
|
// stack: addr, counter, x[num_u256s-counter], ... , x[num_u256s-1]
|
||||||
@ -58,56 +57,335 @@ sha2_store_loop:
|
|||||||
sha2_store_end:
|
sha2_store_end:
|
||||||
JUMPDEST
|
JUMPDEST
|
||||||
|
|
||||||
|
// Precodition: input is in memory, starting at [TODO: fix] 0, of the form
|
||||||
|
// num_u256s, x[0], x[1], ..., x[num_u256s-1]
|
||||||
global sha2_append1:
|
// Postcodition: input is in memory, starting at [TODO: fix] 0, of the form
|
||||||
|
// num_blocks, block0[0], block0[1], block1[0], ..., blocklast[1]
|
||||||
|
global sha2_pad:
|
||||||
// TODO: use kernel memory, and start address not at 0
|
// TODO: use kernel memory, and start address not at 0
|
||||||
push 0
|
push 0
|
||||||
mload
|
mload
|
||||||
// stack: num_u256s
|
// stack: num_u256s
|
||||||
mload
|
|
||||||
// stack: x[num_u256s-1]
|
|
||||||
dup1
|
dup1
|
||||||
// stack: x[num_u256s-1], x[num_u256s-1]
|
// stack: num_u256s, num_u256s
|
||||||
|
%iseven
|
||||||
|
// stack: is_even, num_u256s
|
||||||
|
swap1
|
||||||
|
// stack: num_u256s, is_even
|
||||||
|
dup1
|
||||||
|
// stack: num_u256s, num_u256s, is_even
|
||||||
|
mload
|
||||||
|
// stack: x[num_u256s-1], num_u256s, is_even
|
||||||
|
dup1
|
||||||
|
// stack: x[num_u256s-1], x[num_u256s-1], num_u256s, is_even
|
||||||
%count_bits
|
%count_bits
|
||||||
// stack: num_bits, x[num_u256s-1]
|
// stack: num_bits, x[num_u256s-1], num_u256s, is_even
|
||||||
|
dup1
|
||||||
|
// stack: num_bits, num_bits, x[num_u256s-1], num_u256s, is_even
|
||||||
|
swap3
|
||||||
|
// stack: num_u256s, num_bits, x[num_u256s-1], num_bits, is_even
|
||||||
|
%decrement
|
||||||
|
// stack: num_u256s-1, num_bits, x[num_u256s-1], num_bits, is_even
|
||||||
|
push 256
|
||||||
|
mul
|
||||||
|
// stack: (num_u256s-1)*256, num_bits, x[num_u256s-1], num_bits, is_even
|
||||||
|
add
|
||||||
|
// stack: message_bits, x[num_u256s-1], num_bits, is_even
|
||||||
|
swap2
|
||||||
|
// stack: num_bits, x[num_u256s-1], message_bits, is_even
|
||||||
|
dup1
|
||||||
|
// stack: num_bits, num_bits, x[num_u256s-1], message_bits, is_even
|
||||||
|
dup1
|
||||||
|
// stack: num_bits, num_bits, num_bits, x[num_u256s-1], message_bits, is_even
|
||||||
|
dup1
|
||||||
|
%lt(191)
|
||||||
|
// stack: num_bits<191, num_bits, num_bits,x[num_u256s-1], message_bits, is_even
|
||||||
|
swap1
|
||||||
|
// stack: num_bits, num_bits<191, num_bits, x[num_u256s-1], message_bits, is_even
|
||||||
|
dup1
|
||||||
%eq(256)
|
%eq(256)
|
||||||
%jumpi(append_if256)
|
// stack: num_bits==256, num_bits<191, num_bits, x[num_u256s-1], message_bits, is_even
|
||||||
%jump(append_else)
|
push 0
|
||||||
append_if256:
|
// stack: 0, num_bits==256, num_bits<191, num_bits, x[num_u256s-1], message_bits, is_even
|
||||||
|
swap6
|
||||||
|
// stack: is_even, num_bits==256, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
dup2
|
||||||
|
dup2
|
||||||
|
and
|
||||||
|
%jumpi(pad_case1)
|
||||||
|
not
|
||||||
|
// stack: is_odd, num_bits==256, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
dup2
|
||||||
|
dup2
|
||||||
|
and
|
||||||
|
%jumpi(pad_case2)
|
||||||
|
swap1
|
||||||
|
// stack: num_bits==256, is_odd, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
pop
|
||||||
|
// stack: is_odd, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
not
|
||||||
|
// stack: is_even, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
dup2
|
||||||
|
dup2
|
||||||
|
and
|
||||||
|
%jumpi(pad_case3)
|
||||||
|
not
|
||||||
|
// stack: is_odd, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
dup2
|
||||||
|
dup2
|
||||||
|
and
|
||||||
|
%jumpi(pad_case4)
|
||||||
|
swap1
|
||||||
|
// stack: num_bits<191, is_odd, num_bits, x[num_u256s-1], message_bits
|
||||||
|
pop
|
||||||
|
// stack: is_odd, num_bits, x[num_u256s-1], message_bits
|
||||||
|
not
|
||||||
|
// stack: is_even, num_bits, x[num_u256s-1], message_bits
|
||||||
|
%jumpi(pad_case5)
|
||||||
|
%jump(pad_case6)
|
||||||
|
pad_case1:
|
||||||
|
// CASE 1: num_u256s is even; num_bits == 256
|
||||||
JUMPDEST
|
JUMPDEST
|
||||||
// stack: num_bits, x[num_u256s-1]
|
// stack: is_odd, num_bits==256, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
%pop5
|
||||||
|
// stack: message_bits
|
||||||
|
push 0
|
||||||
|
mload
|
||||||
|
// stack: num_u256s, message_bits
|
||||||
|
%increment
|
||||||
|
// stack: num_u256s+1, message_bits
|
||||||
|
dup1
|
||||||
|
// stack: num_u256s+1, num_u256s+1, message_bits
|
||||||
|
push 2
|
||||||
|
push 255
|
||||||
|
%jump(exp)
|
||||||
|
// stack: 2^255, num_u256s+1, num_u256s+1, message_bits
|
||||||
|
swap
|
||||||
|
// stack: num_u256s+1, 2^255, num_u256s+1, message_bits
|
||||||
|
mstore
|
||||||
|
// stack: num_u256s+1, message_bits
|
||||||
|
%increment
|
||||||
|
// stack: num_u256s+2, message_bits
|
||||||
|
dup1
|
||||||
|
// stack: num_u256s+2, num_u256s+2, message_bits
|
||||||
|
swap2
|
||||||
|
// stack: message_bits, num_u256s+2, num_u256s+2
|
||||||
|
swap1
|
||||||
|
// stack: num_u256s+2, message_bits, num_u256s+2
|
||||||
|
mstore
|
||||||
|
// stack: num_u256s+2
|
||||||
|
%div2
|
||||||
|
// stack: num_blocks=(num_u256s+2)//2
|
||||||
|
push 0
|
||||||
|
mstore
|
||||||
|
%jump(pad_end)
|
||||||
|
pad_case2:
|
||||||
|
// CASE 2: num_u256s is odd; num_bits == 256
|
||||||
|
JUMPDEST
|
||||||
|
// stack: is_even, num_bits==256, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
%pop5
|
||||||
|
// stack: message_bits
|
||||||
|
push 0
|
||||||
|
mload
|
||||||
|
// stack: num_u256s, message_bits
|
||||||
|
%increment
|
||||||
|
// stack: num_u256s+1, message_bits
|
||||||
|
swap
|
||||||
|
// stack: message_bits, num_u256s+1
|
||||||
|
push 2
|
||||||
|
push 255
|
||||||
|
%jump(exp)
|
||||||
|
add
|
||||||
|
// stack: 2^255 + message_bits, num_u256s+1
|
||||||
|
swap1
|
||||||
|
// stack: num_u256s+1, 2^255 + message_bits
|
||||||
|
dup1
|
||||||
|
// stack: num_u256s+1, num_u256s+1, 2^255 + message_bits
|
||||||
|
swap2
|
||||||
|
// stack: 2^255 + message_bits, num_u256s+1, num_u256s+1
|
||||||
|
swap1
|
||||||
|
// stack: num_u256s+1, 2^255 + message_bits, num_u256s+1
|
||||||
|
mstore
|
||||||
|
// stack: num_u256s+1
|
||||||
|
div2
|
||||||
|
// stack: num_blocks=(num_u256s+1)//2
|
||||||
|
push 0
|
||||||
|
mstore
|
||||||
|
%jump(pad_end)
|
||||||
|
pad_case3:
|
||||||
|
// CASE 3: num_u256s is even; num_bits < 191
|
||||||
|
JUMPDEST
|
||||||
|
// stack: is_even, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
%pop2
|
%pop2
|
||||||
|
// stack: num_bits, x[num_u256s-1], message_bits
|
||||||
|
swap1
|
||||||
|
// stack: x[num_u256s-1], num_bits, message_bits
|
||||||
|
push 2
|
||||||
|
mul
|
||||||
|
%increment
|
||||||
|
// stack: 2*x[num_u256s-1]+1, num_bits, message_bits
|
||||||
|
swap1
|
||||||
|
// stack: num_bits, 2*x[num_u256s-1]+1, message_bits
|
||||||
|
push 255
|
||||||
|
sub
|
||||||
|
// stack: 256 - (num_bits + 1), 2*x[num_u256s-1]+1, message_bits
|
||||||
|
push 2
|
||||||
|
%jump(exp)
|
||||||
|
// stack: 2^(256 - (num_bits + 1)), 2*x[num_u256s-1]+1, message_bits
|
||||||
|
mul
|
||||||
|
// stack: [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
|
add
|
||||||
|
// stack: [x[num_u256s-1] || 1 || 0s | message_bits]
|
||||||
|
push 0
|
||||||
|
mload
|
||||||
|
// stack: num_u256s, [x[num_u256s-1] || 1 || 0s | message_bits]
|
||||||
|
mstore
|
||||||
|
push 0
|
||||||
|
mload
|
||||||
|
// stack: num_u256s
|
||||||
|
%div2
|
||||||
|
// stack: num_blocks=num_u256s//2
|
||||||
|
push 0
|
||||||
|
mstore
|
||||||
|
%jump(pad_end)
|
||||||
|
pad_case4:
|
||||||
|
// CASE 4: num_u256s is odd; num_bits < 191
|
||||||
|
JUMPDEST
|
||||||
|
// stack: is_odd, num_bits<191, num_bits, x[num_u256s-1], message_bits
|
||||||
|
%pop2
|
||||||
|
// stack: num_bits, x[num_u256s-1], message_bits
|
||||||
|
swap1
|
||||||
|
// stack: x[num_u256s-1], num_bits, message_bits
|
||||||
|
push 2
|
||||||
|
mul
|
||||||
|
%increment
|
||||||
|
// stack: 2*x[num_u256s-1]+1, num_bits, message_bits
|
||||||
|
swap1
|
||||||
|
// stack: num_bits, 2*x[num_u256s-1]+1, message_bits
|
||||||
|
push 255
|
||||||
|
sub
|
||||||
|
// stack: 256 - (num_bits + 1), 2*x[num_u256s-1]+1, message_bits
|
||||||
|
push 2
|
||||||
|
%jump(exp)
|
||||||
|
// stack: 2^(256 - (num_bits + 1)), 2*x[num_u256s-1]+1, message_bits
|
||||||
|
mul
|
||||||
|
// stack: [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
|
push 0
|
||||||
|
mload
|
||||||
|
// stack: num_u256s, [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
|
mstore
|
||||||
|
// stack: message_bits
|
||||||
|
push 0
|
||||||
|
mload
|
||||||
|
// stack: num_u256s, message_bits
|
||||||
|
%increment
|
||||||
|
// stack: num_u256s+1, message_bits
|
||||||
|
mstore
|
||||||
push 0
|
push 0
|
||||||
mload
|
mload
|
||||||
// stack: num_u256s
|
// stack: num_u256s
|
||||||
%increment
|
%increment
|
||||||
// stack: num_u256s+1
|
// stack: num_u256s+1
|
||||||
dup1
|
%div2
|
||||||
// stack: num_u256s+1, num_u256s+1
|
// stack: num_blocks=(num_u256s+1)//2
|
||||||
push 0
|
push 0
|
||||||
mstore
|
mstore
|
||||||
// stack: num_u256s+1
|
%jump(pad_end)
|
||||||
push 1
|
pad_case5:
|
||||||
// stack: 1, num_u256s+1
|
// CASE 5: num_u256s is even; 191 <= num_bits < 256
|
||||||
swap1
|
|
||||||
// stack: num_u256s+1, 1
|
|
||||||
mstore
|
|
||||||
%jump(append_end)
|
|
||||||
append_else:
|
|
||||||
JUMPDEST
|
JUMPDEST
|
||||||
// stack: num_bits, x[num_u256s-1]
|
// stack: is_even, num_bits, x[num_u256s-1], message_bits
|
||||||
pop
|
pop
|
||||||
// stack: x[num_u256s-1]
|
// stack: num_bits, x[num_u256s-1], message_bits
|
||||||
|
swap1
|
||||||
|
// stack: x[num_u256s-1], num_bits, message_bits
|
||||||
push 2
|
push 2
|
||||||
mul
|
mul
|
||||||
// stack: 2*x[num_u256s-1]
|
|
||||||
%increment
|
%increment
|
||||||
// stack: 2*x[num_u256s-1]+1
|
// stack: 2*x[num_u256s-1]+1, num_bits, message_bits
|
||||||
|
swap1
|
||||||
|
// stack: num_bits, 2*x[num_u256s-1]+1, message_bits
|
||||||
|
push 255
|
||||||
|
sub
|
||||||
|
// stack: 256 - (num_bits + 1), 2*x[num_u256s-1]+1, message_bits
|
||||||
|
push 2
|
||||||
|
%jump(exp)
|
||||||
|
// stack: 2^(256 - (num_bits + 1)), 2*x[num_u256s-1]+1, message_bits
|
||||||
|
mul
|
||||||
|
// stack: [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
push 0
|
push 0
|
||||||
mload
|
mload
|
||||||
// stack: num_u256s, 2*x[num_u256s-1]+1
|
// stack: num_u256s, [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
|
dup1
|
||||||
|
// stack: num_u256s, num_u256s, [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
|
swap2
|
||||||
|
// stack: [x[num_u256s-1] || 1 || 0s], num_u256s, num_u256s, message_bits
|
||||||
|
swap1
|
||||||
|
// stack: num_u256s, [x[num_u256s-1] || 1 || 0s], num_u256s, message_bits
|
||||||
mstore
|
mstore
|
||||||
append_end:
|
// stack: num_u256s, message_bits
|
||||||
|
push 2
|
||||||
|
add
|
||||||
|
// stack: num_u256s+2, message_bits
|
||||||
|
dup1
|
||||||
|
// stack: num_u256s+2, num_u256s+2, message_bits
|
||||||
|
swap2
|
||||||
|
// stack: message_bits, num_u256s+2, num_u256s+2
|
||||||
|
swap1
|
||||||
|
// stack: num_u256s+2, message_bits, num_u256s+2
|
||||||
|
mstore
|
||||||
|
// stack: num_u256s+2
|
||||||
|
div2
|
||||||
|
// stack: num_blocks=(num_u256s+2)//2
|
||||||
|
push 0
|
||||||
|
mstore
|
||||||
|
%jump(pad_end)
|
||||||
|
pad_case6:
|
||||||
|
// CASE 6: num_u256s is odd; 191 <= num_bits < 256
|
||||||
|
JUMPDEST
|
||||||
|
// stack: is_even, num_bits, x[num_u256s-1], message_bits
|
||||||
|
pop
|
||||||
|
// stack: num_bits, x[num_u256s-1], message_bits
|
||||||
|
swap1
|
||||||
|
// stack: x[num_u256s-1], num_bits, message_bits
|
||||||
|
push 2
|
||||||
|
mul
|
||||||
|
%increment
|
||||||
|
// stack: 2*x[num_u256s-1]+1, num_bits, message_bits
|
||||||
|
swap1
|
||||||
|
// stack: num_bits, 2*x[num_u256s-1]+1, message_bits
|
||||||
|
push 255
|
||||||
|
sub
|
||||||
|
// stack: 256 - (num_bits + 1), 2*x[num_u256s-1]+1, message_bits
|
||||||
|
push 2
|
||||||
|
%jump(exp)
|
||||||
|
// stack: 2^(256 - (num_bits + 1)), 2*x[num_u256s-1]+1, message_bits
|
||||||
|
mul
|
||||||
|
// stack: [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
|
push 0
|
||||||
|
mload
|
||||||
|
// stack: num_u256s, [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
|
dup1
|
||||||
|
// stack: num_u256s, num_u256s, [x[num_u256s-1] || 1 || 0s], message_bits
|
||||||
|
swap2
|
||||||
|
// stack: [x[num_u256s-1] || 1 || 0s], num_u256s, num_u256s, message_bits
|
||||||
|
swap1
|
||||||
|
// stack: num_u256s, [x[num_u256s-1] || 1 || 0s], num_u256s, message_bits
|
||||||
|
mstore
|
||||||
|
// stack: num_u256s, message_bits
|
||||||
|
%increment
|
||||||
|
// stack: num_u256s+1, message_bits
|
||||||
|
dup1
|
||||||
|
// stack: num_u256s+1, num_u256s+1, message_bits
|
||||||
|
swap2
|
||||||
|
// stack: message_bits, num_u256s+1, num_u256s+1
|
||||||
|
swap1
|
||||||
|
// stack: num_u256s+1, message_bits, num_u256s+1
|
||||||
|
mstore
|
||||||
|
// stack: num_u256s+1
|
||||||
|
div2
|
||||||
|
// stack: num_blocks=(num_u256s+1)//2
|
||||||
|
push 0
|
||||||
|
mstore
|
||||||
|
pad_end:
|
||||||
JUMPDEST
|
JUMPDEST
|
||||||
|
|
||||||
|
|||||||
@ -159,6 +159,16 @@
|
|||||||
CONSUME_GAS
|
CONSUME_GAS
|
||||||
%endmacro
|
%endmacro
|
||||||
|
|
||||||
|
%macro pop5
|
||||||
|
%pop3
|
||||||
|
%pop2
|
||||||
|
%endmacro
|
||||||
|
|
||||||
|
%macro pop6
|
||||||
|
%pop4
|
||||||
|
%pop2
|
||||||
|
%endmacro
|
||||||
|
|
||||||
// If pred is zero, yields z; otherwise, yields nz
|
// If pred is zero, yields z; otherwise, yields nz
|
||||||
%macro select
|
%macro select
|
||||||
// stack: pred, nz, z
|
// stack: pred, nz, z
|
||||||
@ -249,7 +259,19 @@
|
|||||||
%endmacro
|
%endmacro
|
||||||
|
|
||||||
%macro eq(x)
|
%macro eq(x)
|
||||||
dup1
|
|
||||||
push $x
|
push $x
|
||||||
eq
|
eq
|
||||||
%endmacro
|
%endmacro
|
||||||
|
|
||||||
|
%macro lt(x)
|
||||||
|
push $x
|
||||||
|
swap1
|
||||||
|
lt
|
||||||
|
%endmacro
|
||||||
|
|
||||||
|
%macro iseven
|
||||||
|
push 2
|
||||||
|
swap1
|
||||||
|
mod
|
||||||
|
iszero
|
||||||
|
%endmacro
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user