mirror of
https://github.com/logos-storage/nim-goldilocks-hash.git
synced 2026-02-18 12:33:06 +00:00
Poseidon2 permutation reference implementation (Haskell)
This commit is contained in:
Balazs Komuves
parent
5512db6242
commit
f60ca83c55
173
reference/Constants.hs
Normal file
173
reference/Constants.hs
Normal file
@ -0,0 +1,173 @@
|
||||
|
||||
module Constants where
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
import Data.Array
|
||||
import Data.Word
|
||||
|
||||
import Goldilocks
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
internalDiagElems :: Array Int F
|
||||
internalDiagElems = listArray (0,11) $ map toF
|
||||
[ 0xc3b6c08e23ba9300
|
||||
, 0xd84b5de94a324fb6
|
||||
, 0x0d0c371c5b35b84f
|
||||
, 0x7964f570e7188037
|
||||
, 0x5daf18bbd996604b
|
||||
, 0x6743bc47b9595257
|
||||
, 0x5528b9362c59bb70
|
||||
, 0xac45e25b7127b68b
|
||||
, 0xa2077d7dfbb606b5
|
||||
, 0xf3faac6faee378ae
|
||||
, 0x0c6388b51545e883
|
||||
, 0xd27dbb6944917b60
|
||||
]
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
initialRoundConsts :: Array Int [F]
|
||||
initialRoundConsts = listArray (0,3) $ map (map toF)
|
||||
[ [ 0x13dcf33aba214f46
|
||||
, 0x30b3b654a1da6d83
|
||||
, 0x1fc634ada6159b56
|
||||
, 0x937459964dc03466
|
||||
, 0xedd2ef2ca7949924
|
||||
, 0xede9affde0e22f68
|
||||
, 0x8515b9d6bac9282d
|
||||
, 0x6b5c07b4e9e900d8
|
||||
, 0x1ec66368838c8a08
|
||||
, 0x9042367d80d1fbab
|
||||
, 0x400283564a3c3799
|
||||
, 0x4a00be0466bca75e
|
||||
]
|
||||
, [ 0x7913beee58e3817f
|
||||
, 0xf545e88532237d90
|
||||
, 0x22f8cb8736042005
|
||||
, 0x6f04990e247a2623
|
||||
, 0xfe22e87ba37c38cd
|
||||
, 0xd20e32c85ffe2815
|
||||
, 0x117227674048fe73
|
||||
, 0x4e9fb7ea98a6b145
|
||||
, 0xe0866c232b8af08b
|
||||
, 0x00bbc77916884964
|
||||
, 0x7031c0fb990d7116
|
||||
, 0x240a9e87cf35108f
|
||||
]
|
||||
, [ 0x2e6363a5a12244b3
|
||||
, 0x5e1c3787d1b5011c
|
||||
, 0x4132660e2a196e8b
|
||||
, 0x3a013b648d3d4327
|
||||
, 0xf79839f49888ea43
|
||||
, 0xfe85658ebafe1439
|
||||
, 0xb6889825a14240bd
|
||||
, 0x578453605541382b
|
||||
, 0x4508cda8f6b63ce9
|
||||
, 0x9c3ef35848684c91
|
||||
, 0x0812bde23c87178c
|
||||
, 0xfe49638f7f722c14
|
||||
]
|
||||
, [ 0x8e3f688ce885cbf5
|
||||
, 0xb8e110acf746a87d
|
||||
, 0xb4b2e8973a6dabef
|
||||
, 0x9e714c5da3d462ec
|
||||
, 0x6438f9033d3d0c15
|
||||
, 0x24312f7cf1a27199
|
||||
, 0x23f843bb47acbf71
|
||||
, 0x9183f11a34be9f01
|
||||
, 0x839062fbb9d45dbf
|
||||
, 0x24b56e7e6c2e43fa
|
||||
, 0xe1683da61c962a72
|
||||
, 0xa95c63971a19bfa7
|
||||
]
|
||||
]
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
finalRoundConsts :: Array Int [F]
|
||||
finalRoundConsts = listArray (0,3) $ map (map toF)
|
||||
[ [ 0xc68be7c94882a24d
|
||||
, 0xaf996d5d5cdaedd9
|
||||
, 0x9717f025e7daf6a5
|
||||
, 0x6436679e6e7216f4
|
||||
, 0x8a223d99047af267
|
||||
, 0xbb512e35a133ba9a
|
||||
, 0xfbbf44097671aa03
|
||||
, 0xf04058ebf6811e61
|
||||
, 0x5cca84703fac7ffb
|
||||
, 0x9b55c7945de6469f
|
||||
, 0x8e05bf09808e934f
|
||||
, 0x2ea900de876307d7
|
||||
]
|
||||
, [ 0x7748fff2b38dfb89
|
||||
, 0x6b99a676dd3b5d81
|
||||
, 0xac4bb7c627cf7c13
|
||||
, 0xadb6ebe5e9e2f5ba
|
||||
, 0x2d33378cafa24ae3
|
||||
, 0x1e5b73807543f8c2
|
||||
, 0x09208814bfebb10f
|
||||
, 0x782e64b6bb5b93dd
|
||||
, 0xadd5a48eac90b50f
|
||||
, 0xadd4c54c736ea4b1
|
||||
, 0xd58dbb86ed817fd8
|
||||
, 0x6d5ed1a533f34ddd
|
||||
]
|
||||
, [ 0x28686aa3e36b7cb9
|
||||
, 0x591abd3476689f36
|
||||
, 0x047d766678f13875
|
||||
, 0xa2a11112625f5b49
|
||||
, 0x21fd10a3f8304958
|
||||
, 0xf9b40711443b0280
|
||||
, 0xd2697eb8b2bde88e
|
||||
, 0x3493790b51731b3f
|
||||
, 0x11caf9dd73764023
|
||||
, 0x7acfb8f72878164e
|
||||
, 0x744ec4db23cefc26
|
||||
, 0x1e00e58f422c6340
|
||||
]
|
||||
, [ 0x21dd28d906a62dda
|
||||
, 0xf32a46ab5f465b5f
|
||||
, 0xbfce13201f3f7e6b
|
||||
, 0xf30d2e7adb5304e2
|
||||
, 0xecdf4ee4abad48e9
|
||||
, 0xf94e82182d395019
|
||||
, 0x4ee52e3744d887c5
|
||||
, 0xa1341c7cac0083b2
|
||||
, 0x2302fb26c30c834a
|
||||
, 0xaea3c587273bf7d3
|
||||
, 0xf798e24961823ec7
|
||||
, 0x962deba3e9a2cd94
|
||||
]
|
||||
]
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
internalRoundConsts :: [F] -- :: Array Int F
|
||||
internalRoundConsts = map toF -- listArray (0,21) $ map toF
|
||||
[ 0x4adf842aa75d4316
|
||||
, 0xf8fbb871aa4ab4eb
|
||||
, 0x68e85b6eb2dd6aeb
|
||||
, 0x07a0b06b2d270380
|
||||
, 0xd94e0228bd282de4
|
||||
, 0x8bdd91d3250c5278
|
||||
, 0x209c68b88bba778f
|
||||
, 0xb5e18cdab77f3877
|
||||
, 0xb296a3e808da93fa
|
||||
, 0x8370ecbda11a327e
|
||||
, 0x3f9075283775dad8
|
||||
, 0xb78095bb23c6aa84
|
||||
, 0x3f36b9fe72ad4e5f
|
||||
, 0x69bc96780b10b553
|
||||
, 0x3f1d341f2eb7b881
|
||||
, 0x4e939e9815838818
|
||||
, 0xda366b3ae2a31604
|
||||
, 0xbc89db1e7287d509
|
||||
, 0x6102f411f9ef5659
|
||||
, 0x58725c5e7ac1f0ab
|
||||
, 0x0df5856c798883e7
|
||||
, 0xf7bb62a8da4c961b
|
||||
]
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
@ -10,6 +10,7 @@ import Prelude hiding ( div )
|
||||
import qualified Prelude
|
||||
|
||||
import Data.Bits
|
||||
import Data.Word
|
||||
import Data.Ratio
|
||||
|
||||
import Text.Printf
|
||||
@ -18,6 +19,14 @@ import Text.Printf
|
||||
|
||||
type F = Goldilocks
|
||||
|
||||
fromF :: F -> Word64
|
||||
fromF (Goldilocks x) = fromInteger x
|
||||
|
||||
toF :: Word64 -> F
|
||||
toF = mkGoldilocks . fromIntegral
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
newtype Goldilocks
|
||||
= Goldilocks Integer
|
||||
deriving Eq
|
||||
|
||||
108
reference/Poseidon2.hs
Normal file
108
reference/Poseidon2.hs
Normal file
@ -0,0 +1,108 @@
|
||||
|
||||
{-# LANGUAGE Strict #-}
|
||||
module Poseidon2 where
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
import Data.Array (Array)
|
||||
import Data.Array.IArray
|
||||
|
||||
import Goldilocks
|
||||
import Constants
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
type State = Array Int F
|
||||
|
||||
zeroState :: State
|
||||
zeroState = listArray (0,11) (replicate 12 0)
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
permutation :: State -> State
|
||||
permutation
|
||||
= finalRounds
|
||||
. internalRounds
|
||||
. initialRounds
|
||||
. externalDiffusion
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
initialRounds :: State -> State
|
||||
initialRounds
|
||||
= externalRound (initialRoundConsts ! 3)
|
||||
. externalRound (initialRoundConsts ! 2)
|
||||
. externalRound (initialRoundConsts ! 1)
|
||||
. externalRound (initialRoundConsts ! 0)
|
||||
|
||||
internalRounds :: State -> State
|
||||
internalRounds = foldr1 (.) (map (internalRound $) (reverse internalRoundConsts))
|
||||
|
||||
finalRounds :: State -> State
|
||||
finalRounds
|
||||
= externalRound (finalRoundConsts ! 3)
|
||||
. externalRound (finalRoundConsts ! 2)
|
||||
. externalRound (finalRoundConsts ! 1)
|
||||
. externalRound (finalRoundConsts ! 0)
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
externalRound :: [F] -> State -> State
|
||||
externalRound rcs = externalDiffusion . sboxExternal rcs
|
||||
|
||||
internalRound :: F -> State -> State
|
||||
internalRound rc = internalDiffusion . sboxInternal rc
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
sbox1 :: F -> F
|
||||
sbox1 x = pow x 7
|
||||
|
||||
sboxRC :: F -> F -> F
|
||||
sboxRC rc x = sbox1 (x+rc)
|
||||
|
||||
sboxInternal :: F -> State -> State
|
||||
sboxInternal rc s = s // [ (0, sboxRC rc (s!0)) ]
|
||||
|
||||
sboxExternal :: [F] -> State -> State
|
||||
sboxExternal rcs s = listArray (0,11) $ zipWith sboxRC rcs (elems s)
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
internalDiffusion :: State -> State
|
||||
internalDiffusion state = listArray (0,11) $ [ s + (state!i * internalDiagElems!i) | i<-[0..11] ] where
|
||||
s = sum (elems state)
|
||||
|
||||
{-
|
||||
matM4 :: Array (Int,Int) F
|
||||
matM4 = amap toF $ listArray ((0,0),(3,3))
|
||||
[ 5 , 7 , 1 , 3
|
||||
, 4 , 6 , 1 , 1
|
||||
, 1 , 3 , 5 , 7
|
||||
, 1 , 1 , 4 , 6
|
||||
]
|
||||
-}
|
||||
|
||||
matM12:: Array (Int,Int) F
|
||||
matM12 = amap toF $ listArray ((0,0),(11,11))
|
||||
[ 2*5 , 2*7 , 2*1 , 2*3 , 5 , 7 , 1 , 3 , 5 , 7 , 1 , 3
|
||||
, 2*4 , 2*6 , 2*1 , 2*1 , 4 , 6 , 1 , 1 , 4 , 6 , 1 , 1
|
||||
, 2*1 , 2*3 , 2*5 , 2*7 , 1 , 3 , 5 , 7 , 1 , 3 , 5 , 7
|
||||
, 2*1 , 2*1 , 2*4 , 2*6 , 1 , 1 , 4 , 6 , 1 , 1 , 4 , 6
|
||||
, 5 , 7 , 1 , 3 , 2*5 , 2*7 , 2*1 , 2*3 , 5 , 7 , 1 , 3
|
||||
, 4 , 6 , 1 , 1 , 2*4 , 2*6 , 2*1 , 2*1 , 4 , 6 , 1 , 1
|
||||
, 1 , 3 , 5 , 7 , 2*1 , 2*3 , 2*5 , 2*7 , 1 , 3 , 5 , 7
|
||||
, 1 , 1 , 4 , 6 , 2*1 , 2*1 , 2*4 , 2*6 , 1 , 1 , 4 , 6
|
||||
, 5 , 7 , 1 , 3 , 5 , 7 , 1 , 3 , 2*5 , 2*7 , 2*1 , 2*3
|
||||
, 4 , 6 , 1 , 1 , 4 , 6 , 1 , 1 , 2*4 , 2*6 , 2*1 , 2*1
|
||||
, 1 , 3 , 5 , 7 , 1 , 3 , 5 , 7 , 2*1 , 2*3 , 2*5 , 2*7
|
||||
, 1 , 1 , 4 , 6 , 1 , 1 , 4 , 6 , 2*1 , 2*1 , 2*4 , 2*6
|
||||
]
|
||||
|
||||
externalDiffusion :: State -> State
|
||||
externalDiffusion state = listArray (0,11)
|
||||
[ sum [ matM12!(i,j) * state!j | j<-[0..11] ]
|
||||
| i<-[0..11]
|
||||
]
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
Loading…
x
Reference in New Issue
Block a user