Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

crypto: poseidon2: Clean the stack in Poseidon utils #9

Merged
merged 1 commit into from
Jan 27, 2025
Merged

crypto: poseidon2: Clean the stack in Poseidon utils #9

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 12 additions & 4 deletions src/crypto/poseidon2/roundUtils.huff
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,9 @@
ADD_RC(<RC1>) // [a + RC1, (b + RC2)^5, (c + RC3)^5, a, b, c]
PUSH_PRIME() swap1 // [a + RC1, PRIME, (b + RC2)^5, (c + RC3)^5, a, b, c]
SBOX() // [(a + RC1)^5, (b + RC2)^5, (c + RC3)^5, a, b, c]

// Pop the old state off the stack
swap3 pop swap3 pop swap3 pop // [a', b', c']

// Multiply the intermediate state by the external round MDS matrix
EXTERNAL_MDS()
Expand Down Expand Up @@ -104,10 +107,15 @@
PUSH_PRIME() dup2 PUSH_PRIME() // [PRIME, sum, PRIME, sum, state[0], state[1], state[2]]
dup7 dup1 addmod // [state[2] * 2, sum, PRIME, sum, state[0], state[1], state[2]]
addmod // [state'[2], sum, state[0], state[1], state[2]]
PUSH_PRIME() dup3 // [sum, PRIME, state'[2], state[0], state[1], state[2]]
dup6 addmod // [state'[1], state'[2], sum, state[0], state[1], state[2]]
PUSH_PRIME() dup4 // [sum, PRIME, state'[1], state'[2], sum, state[0], state[1], state[2]]
dup6 addmod // [state'[0], state'[1], state'[2], sum, state[0], state[1], state[2]]
swap4 pop // [sum, state[0], state[1], state'[2]]

PUSH_PRIME() dup2 // [sum, PRIME, sum, state[0], state[1], state'[2]]
dup5 addmod // [state'[1], sum, state[0], state[1], state'[2]]
swap3 pop // [sum, state[0], state'[1], state'[2]]

PUSH_PRIME() dup2 // [sum, PRIME, sum, state[0], state'[1], state'[2]]
dup4 addmod // [state'[0], sum, state[0], state'[1], state'[2]]
swap2 pop pop // [state'[0], state'[1], state'[2]]
}

/// @dev Apply the external MDS matrix to the sponge state
Expand Down
Loading