mirror of
https://github.com/logos-storage/logos-storage-proofs-circuits.git
synced 2026-01-14 03:13:08 +00:00
78 lines
2.6 KiB
Haskell
78 lines
2.6 KiB
Haskell
|
|
-- | Generate test vectors to compare with other implementations
|
|
|
|
module TestVectors where
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
import Control.Monad
|
|
|
|
import Data.Word
|
|
import qualified Data.ByteString as B
|
|
|
|
import Poseidon2.Merkle
|
|
import Poseidon2.Sponge
|
|
import Slot
|
|
|
|
import ZK.Algebra.Curves.BN128.Fr.Mont (Fr)
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
allTestVectors = do
|
|
testVectorsSponge
|
|
testVectorsHash
|
|
testVectorsMerkle
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
testVectorsSponge :: IO ()
|
|
testVectorsSponge = do
|
|
putStrLn ""
|
|
putStrLn "test vectors for sponge of field elements with rate=1"
|
|
putStrLn "-----------------------------------------------------"
|
|
forM_ [0..8] $ \n -> do
|
|
let input = map fromIntegral [1..n] :: [Fr]
|
|
putStrLn $ "hash of [1.." ++ show n ++ "] :: [Fr] = " ++ show (sponge1 input)
|
|
|
|
putStrLn ""
|
|
putStrLn "test vectors for sponge of field elements with rate=2"
|
|
putStrLn "-----------------------------------------------------"
|
|
forM_ [0..8] $ \n -> do
|
|
let input = map fromIntegral [1..n] :: [Fr]
|
|
putStrLn $ "hash of [1.." ++ show n ++ "] :: [Fr] = " ++ show (sponge2 input)
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
testVectorsHash :: IO ()
|
|
testVectorsHash = do
|
|
|
|
putStrLn ""
|
|
putStrLn "test vectors for hash (padded sponge with rate=2) of bytes"
|
|
putStrLn "----------------------------------------------------------"
|
|
forM_ [0..80] $ \n -> do
|
|
let input = map fromIntegral [1..n] :: [Word8]
|
|
let bs = B.pack input
|
|
putStrLn $ "hash of [1.." ++ show n ++ "] :: [Byte] = " ++ show (hashCell_ bs)
|
|
|
|
--------------------------------------------------------------------------------
|
|
|
|
testVectorsMerkle :: IO ()
|
|
testVectorsMerkle = do
|
|
putStrLn ""
|
|
putStrLn "test vectors for Merkle roots of field elements"
|
|
putStrLn "-----------------------------------------------"
|
|
forM_ [1..40] $ \n -> do
|
|
let input = map fromIntegral [1..n] :: [Fr]
|
|
putStrLn $ "Merkle root of [1.." ++ show n ++ "] :: [Fr] = " ++ show (calcMerkleRoot input)
|
|
|
|
putStrLn ""
|
|
putStrLn "test vectors for Merkle roots of sequence of bytes"
|
|
putStrLn "--------------------------------------------------"
|
|
forM_ [0..80] $ \n -> do
|
|
let input = map fromIntegral [1..n] :: [Word8]
|
|
let bs = B.pack input
|
|
let flds = cellDataToFieldElements (CellData bs)
|
|
putStrLn $ "Merkle root of [1.." ++ show n ++ "] :: [Byte] = " ++ show (calcMerkleRoot flds)
|
|
|
|
--------------------------------------------------------------------------------
|