@ -8,151 +8,45 @@ use ff::*;
#[ PrimeFieldGenerator = " 7 " ]
pub struct Fr ( FrRepr ) ;
const NROUNDSF : usize = 8 ;
const NROUNDSP : usize = 57 ;
const T : usize = 6 ;
mod constants ;
#[ derive(Debug) ]
pub struct Constants {
c : Vec < Fr > ,
m : Vec < Vec < Fr > > ,
pub c : Vec < Vec < Fr > > ,
pub m : Vec < Vec < Vec < Fr > > > ,
pub n_rounds_f : usize ,
pub n_rounds_p : Vec < usize > ,
}
pub fn load_constants ( ) -> Constants {
let c_str : Vec < & str > = vec ! [
"14397397413755236225575615486459253198602422701513067526754101844196324375522" ,
"10405129301473404666785234951972711717481302463898292859783056520670200613128" ,
"5179144822360023508491245509308555580251733042407187134628755730783052214509" ,
"9132640374240188374542843306219594180154739721841249568925550236430986592615" ,
"20360807315276763881209958738450444293273549928693737723235350358403012458514" ,
"17933600965499023212689924809448543050840131883187652471064418452962948061619" ,
"3636213416533737411392076250708419981662897009810345015164671602334517041153" ,
"2008540005368330234524962342006691994500273283000229509835662097352946198608" ,
"16018407964853379535338740313053768402596521780991140819786560130595652651567" ,
"20653139667070586705378398435856186172195806027708437373983929336015162186471" ,
"17887713874711369695406927657694993484804203950786446055999405564652412116765" ,
"4852706232225925756777361208698488277369799648067343227630786518486608711772" ,
"8969172011633935669771678412400911310465619639756845342775631896478908389850" ,
"20570199545627577691240476121888846460936245025392381957866134167601058684375" ,
"16442329894745639881165035015179028112772410105963688121820543219662832524136" ,
"20060625627350485876280451423010593928172611031611836167979515653463693899374" ,
"16637282689940520290130302519163090147511023430395200895953984829546679599107" ,
"15599196921909732993082127725908821049411366914683565306060493533569088698214" ,
"16894591341213863947423904025624185991098788054337051624251730868231322135455" ,
"1197934381747032348421303489683932612752526046745577259575778515005162320212" ,
"6172482022646932735745595886795230725225293469762393889050804649558459236626" ,
"21004037394166516054140386756510609698837211370585899203851827276330669555417" ,
"15262034989144652068456967541137853724140836132717012646544737680069032573006" ,
"15017690682054366744270630371095785995296470601172793770224691982518041139766" ,
"15159744167842240513848638419303545693472533086570469712794583342699782519832" ,
"11178069035565459212220861899558526502477231302924961773582350246646450941231" ,
"21154888769130549957415912997229564077486639529994598560737238811887296922114" ,
"20162517328110570500010831422938033120419484532231241180224283481905744633719" ,
"2777362604871784250419758188173029886707024739806641263170345377816177052018" ,
"15732290486829619144634131656503993123618032247178179298922551820261215487562" ,
"6024433414579583476444635447152826813568595303270846875177844482142230009826" ,
"17677827682004946431939402157761289497221048154630238117709539216286149983245" ,
"10716307389353583413755237303156291454109852751296156900963208377067748518748" ,
"14925386988604173087143546225719076187055229908444910452781922028996524347508" ,
"8940878636401797005293482068100797531020505636124892198091491586778667442523" ,
"18911747154199663060505302806894425160044925686870165583944475880789706164410" ,
"8821532432394939099312235292271438180996556457308429936910969094255825456935" ,
"20632576502437623790366878538516326728436616723089049415538037018093616927643" ,
"71447649211767888770311304010816315780740050029903404046389165015534756512" ,
"2781996465394730190470582631099299305677291329609718650018200531245670229393" ,
"12441376330954323535872906380510501637773629931719508864016287320488688345525" ,
"2558302139544901035700544058046419714227464650146159803703499681139469546006" ,
"10087036781939179132584550273563255199577525914374285705149349445480649057058" ,
"4267692623754666261749551533667592242661271409704769363166965280715887854739" ,
"4945579503584457514844595640661884835097077318604083061152997449742124905548" ,
"17742335354489274412669987990603079185096280484072783973732137326144230832311" ,
"6266270088302506215402996795500854910256503071464802875821837403486057988208" ,
"2716062168542520412498610856550519519760063668165561277991771577403400784706" ,
"19118392018538203167410421493487769944462015419023083813301166096764262134232" ,
"9386595745626044000666050847309903206827901310677406022353307960932745699524" ,
"9121640807890366356465620448383131419933298563527245687958865317869840082266" ,
"3078975275808111706229899605611544294904276390490742680006005661017864583210" ,
"7157404299437167354719786626667769956233708887934477609633504801472827442743" ,
"14056248655941725362944552761799461694550787028230120190862133165195793034373" ,
"14124396743304355958915937804966111851843703158171757752158388556919187839849" ,
"11851254356749068692552943732920045260402277343008629727465773766468466181076" ,
"9799099446406796696742256539758943483211846559715874347178722060519817626047" ,
"10156146186214948683880719664738535455146137901666656566575307300522957959544" ,
"19908645952733301583346063785055921934459499091029406575311417879963332475861" ,
"11766105336238068471342414351862472329437473380853789942065610694000443387471" ,
"11002137593249972174092192767251572171769044073555430468487809799220351297047" ,
"284136377911685911941431040940403846843630064858778505937392780738953624163" ,
"19448733709802908339787967270452055364068697565906862913410983275341804035680" ,
"14423660424692802524250720264041003098290275890428483723270346403986712981505" ,
"10635360132728137321700090133109897687122647659471659996419791842933639708516" ,
] ;
let m_str : Vec < Vec < & str > > = vec ! [
vec ! [
"19167410339349846567561662441069598364702008768579734801591448511131028229281" ,
"14183033936038168803360723133013092560869148726790180682363054735190196956789" ,
"9067734253445064890734144122526450279189023719890032859456830213166173619761" ,
"16378664841697311562845443097199265623838619398287411428110917414833007677155" ,
"12968540216479938138647596899147650021419273189336843725176422194136033835172" ,
"3636162562566338420490575570584278737093584021456168183289112789616069756675" ,
] ,
vec ! [
"17034139127218860091985397764514160131253018178110701196935786874261236172431" ,
"2799255644797227968811798608332314218966179365168250111693473252876996230317" ,
"2482058150180648511543788012634934806465808146786082148795902594096349483974" ,
"16563522740626180338295201738437974404892092704059676533096069531044355099628" ,
"10468644849657689537028565510142839489302836569811003546969773105463051947124" ,
"3328913364598498171733622353010907641674136720305714432354138807013088636408" ,
] ,
vec ! [
"18985203040268814769637347880759846911264240088034262814847924884273017355969" ,
"8652975463545710606098548415650457376967119951977109072274595329619335974180" ,
"970943815872417895015626519859542525373809485973005165410533315057253476903" ,
"19406667490568134101658669326517700199745817783746545889094238643063688871948" ,
"17049854690034965250221386317058877242629221002521630573756355118745574274967" ,
"4964394613021008685803675656098849539153699842663541444414978877928878266244" ,
] ,
vec ! [
"19025623051770008118343718096455821045904242602531062247152770448380880817517" ,
"9077319817220936628089890431129759976815127354480867310384708941479362824016" ,
"4770370314098695913091200576539533727214143013236894216582648993741910829490" ,
"4298564056297802123194408918029088169104276109138370115401819933600955259473" ,
"6905514380186323693285869145872115273350947784558995755916362330070690839131" ,
"4783343257810358393326889022942241108539824540285247795235499223017138301952" ,
] ,
vec ! [
"16205238342129310687768799056463408647672389183328001070715567975181364448609" ,
"8303849270045876854140023508764676765932043944545416856530551331270859502246" ,
"20218246699596954048529384569730026273241102596326201163062133863539137060414" ,
"1712845821388089905746651754894206522004527237615042226559791118162382909269" ,
"13001155522144542028910638547179410124467185319212645031214919884423841839406" ,
"16037892369576300958623292723740289861626299352695838577330319504984091062115" ,
] ,
vec ! [
"15162889384227198851506890526431746552868519326873025085114621698588781611738" ,
"13272957914179340594010910867091459756043436017766464331915862093201960540910" ,
"9416416589114508529880440146952102328470363729880726115521103179442988482948" ,
"8035240799672199706102747147502951589635001418759394863664434079699838251138" ,
"21642389080762222565487157652540372010968704000567605990102641816691459811717" ,
"20261355950827657195644012399234591122288573679402601053407151083849785332516" ,
] ,
] ;
let mut c : Vec < Fr > = Vec ::new ( ) ;
let ( c_str , m_str ) = constants ::constants ( ) ;
let mut c : Vec < Vec < Fr > > = Vec ::new ( ) ;
for i in 0 . . c_str . len ( ) {
let n : Fr = Fr ::from_str ( c_str [ i ] ) . unwrap ( ) ;
c . push ( n ) ;
let mut cci : Vec < Fr > = Vec ::new ( ) ;
for j in 0 . . c_str [ i ] . len ( ) {
let b : Fr = Fr ::from_str ( c_str [ i ] [ j ] ) . unwrap ( ) ;
cci . push ( b ) ;
}
c . push ( cci ) ;
}
let mut m : Vec < Vec < Fr > > = Vec ::new ( ) ;
let mut m : Vec < Vec < Vec < Fr > > > = Vec ::new ( ) ;
for i in 0 . . m_str . len ( ) {
let mut mj : Vec < Fr > = Vec ::new ( ) ;
let mut mi : Vec < Vec < Fr > > = Vec ::new ( ) ;
for j in 0 . . m_str [ i ] . len ( ) {
let n : Fr = Fr ::from_str ( m_str [ i ] [ j ] ) . unwrap ( ) ;
mj . push ( n ) ;
let mut mij : Vec < Fr > = Vec ::new ( ) ;
for k in 0 . . m_str [ i ] [ j ] . len ( ) {
let b : Fr = Fr ::from_str ( m_str [ i ] [ j ] [ k ] ) . unwrap ( ) ;
mij . push ( b ) ;
}
mi . push ( mij ) ;
}
m . push ( mj ) ;
m . push ( mi ) ;
}
Constants {
c : c ,
m : m ,
n_rounds_f : 8 ,
n_rounds_p : vec ! [ 56 , 57 , 56 , 60 , 60 , 63 , 64 , 63 ] ,
}
Constants { c : c , m : m }
}
pub struct Poseidon {
@ -164,15 +58,15 @@ impl Poseidon {
constants : load_constants ( ) ,
}
}
pub fn ark ( & self , state : & mut Vec < Fr > , c : & Fr ) {
pub fn ark ( & self , state : & mut Vec < Fr > , c : & Ve c< Fr > , it : usize ) {
for i in 0 . . state . len ( ) {
state [ i ] . add_assign ( c ) ;
state [ i ] . add_assign ( & c [ it + i ] ) ;
}
}
pub fn sbox ( & self , state : & mut Vec < Fr > , i : usize ) {
if i < NROUNDSF / 2 | | i > = NROUNDSF / 2 + NROUNDSP {
for j in 0 . . T {
pub fn sbox ( & self , n_rounds_f : usize , n_rounds_p : usize , state : & mut Vec < Fr > , i : usize ) {
if i < n_rounds_f / 2 | | i > = n_rounds_f / 2 + n_rounds_p {
for j in 0 . . state . len ( ) {
let aux = state [ j ] ;
state [ j ] . square ( ) ;
state [ j ] . square ( ) ;
@ -191,7 +85,7 @@ impl Poseidon {
for i in 0 . . state . len ( ) {
new_state . push ( Fr ::zero ( ) ) ;
for j in 0 . . state . len ( ) {
let mut mij = m [ i ] [ j ] ;
let mut mij = m [ j ] [ i ] ;
mij . mul_assign ( & state [ j ] ) ;
new_state [ i ] . add_assign ( & mij ) ;
}
@ -200,19 +94,25 @@ impl Poseidon {
}
pub fn hash ( & self , inp : Vec < Fr > ) -> Result < Fr , String > {
if inp . len ( ) = = 0 | | inp . len ( ) > T {
let t = inp . len ( ) + 1 ;
if inp . len ( ) = = 0 | | inp . len ( ) > = self . constants . n_rounds_p . len ( ) - 1 {
return Err ( "Wrong inputs length" . to_string ( ) ) ;
}
let n_rounds_f = self . constants . n_rounds_f . clone ( ) ;
let n_rounds_p = self . constants . n_rounds_p [ t - 2 ] . clone ( ) ;
let mut state = inp . clone ( ) ;
for _ in inp . len ( ) . . T {
for _ in inp . len ( ) . . t {
state . push ( Fr ::zero ( ) ) ;
}
// state[state.len() - 1] = Fr::zero();
for i in 0 . . ( NROUNDSF + NROUNDSP ) {
self . ark ( & mut state , & self . constants . c [ i ] ) ;
self . sbox ( & mut state , i ) ;
state = self . mix ( & state , & self . constants . m ) ;
for i in 0 . . ( n_rounds_f + n_rounds_p ) {
self . ark ( & mut state , & self . constants . c [ t - 2 ] , i * t ) ;
self . sbox ( n_rounds_f , n_rounds_p , & mut state , i ) ;
if i < n_rounds_f + n_rounds_p - 1 {
state = self . mix ( & state , & self . constants . m [ t - 2 ] ) ;
}
}
Ok ( state [ 0 ] )
@ -244,29 +144,44 @@ mod tests {
#[ test ]
fn test_load_constants ( ) {
let constants = load_constants ( ) ;
let cons = load_constants ( ) ;
assert_eq ! (
constants . c [ 0 ] . to_string ( ) ,
"Fr(0x1fd4a35e68f0946f8f5dfd2ac9d7882ce2466ec1c9766f69b5a14c3f84a17be2 )"
cons . c [ 0 ] [ 0 ] . to_string ( ) ,
"Fr(0x09c46e9ec68e9bd4fe1faaba294cba38a71aa177534cdd1b6c7dc0dbd0abd7a7 )"
) ;
assert_eq ! (
constants . c [ constant s . c . len ( ) - 1 ] . to_string ( ) ,
"Fr(0x1783668830df79d79707da6a0e7c023c5f54a112be7ee3f7d49b420213ab7f64 )"
cons . c [ cons . c . len ( ) - 1 ] [ 0 ] . to_string ( ) ,
"Fr(0x2088ce9534577bf38be7bc457f2756d558d66e0c07b9cc001a580bd42cda0e77 )"
) ;
assert_eq ! (
constants . m [ 0 ] [ 0 ] . to_string ( ) ,
"Fr(0x2a605eab3c12c29701b9a8944a16ff3d64c199efa7c857c65e4c0560ab3b0ca1 )"
cons . m [ 0 ] [ 0 ] [ 0 ] . to_string ( ) ,
"Fr(0x066f6f85d6f68a85ec10345351a23a3aaf07f38af8c952a7bceca70bd2af7ad5 )"
) ;
assert_eq ! (
constants . m [ constant s . m . len ( ) - 1 ] [ 0 ] . to_string ( ) ,
"Fr(0x2185e429a96435c62e21e7091f4262685769e43fa49c9d40f88c41016c588ada )"
cons . m [ cons . m . len ( ) - 1 ] [ 0 ] [ 0 ] . to_string ( ) ,
"Fr(0x0190f922d97c8a7dcf0a142a3be27749d1c64bc22f1c556aaa24925d158cac56 )"
) ;
}
#[ test ]
fn test_hash ( ) {
let b0 : Fr = Fr ::from_str ( "0" ) . unwrap ( ) ;
let b1 : Fr = Fr ::from_str ( "1" ) . unwrap ( ) ;
let b2 : Fr = Fr ::from_str ( "2" ) . unwrap ( ) ;
let b3 : Fr = Fr ::from_str ( "3" ) . unwrap ( ) ;
let b4 : Fr = Fr ::from_str ( "4" ) . unwrap ( ) ;
let b5 : Fr = Fr ::from_str ( "5" ) . unwrap ( ) ;
let b6 : Fr = Fr ::from_str ( "6" ) . unwrap ( ) ;
let mut big_arr : Vec < Fr > = Vec ::new ( ) ;
big_arr . push ( b1 . clone ( ) ) ;
let poseidon = Poseidon ::new ( ) ;
let h = poseidon . hash ( big_arr . clone ( ) ) . unwrap ( ) ;
assert_eq ! (
h . to_string ( ) ,
"Fr(0x186a5454a7c47c73dfc74ac32ea40a57d27eeb4e2bfc6551dd7b66686d3fd1ab)" // "11043376183861534927536506085090418075369306574649619885724436265926427398571"
) ;
let mut big_arr : Vec < Fr > = Vec ::new ( ) ;
big_arr . push ( b1 . clone ( ) ) ;
big_arr . push ( b2 . clone ( ) ) ;
@ -274,18 +189,73 @@ mod tests {
let h = poseidon . hash ( big_arr . clone ( ) ) . unwrap ( ) ;
assert_eq ! (
h . to_string ( ) ,
"Fr(0x1b10d227ef9aa736dc05bba919a8462ce45bc5b7d632e8ff9bfe6874a002eaed)" // "12242166908188651009877250812424843524687801523336557272219921456462821518061 "
"Fr(0x25d86fb7c42fd70a7e800e871f22f2f03a282abb18f86c347a1078a92f713f60)" // "17117985411748610629288516079940078114952304104811071254131751175361957805920 "
) ;
let b3 : Fr = Fr ::from_str ( "3" ) . unwrap ( ) ;
let b4 : Fr = Fr ::from_str ( "4" ) . unwrap ( ) ;
let mut big_arr34 : Vec < Fr > = Vec ::new ( ) ;
big_arr34 . push ( b3 . clone ( ) ) ;
big_arr34 . push ( b4 . clone ( ) ) ;
let h34 = poseidon . hash ( big_arr34 . clone ( ) ) . unwrap ( ) ;
let mut big_arr : Vec < Fr > = Vec ::new ( ) ;
big_arr . push ( b1 . clone ( ) ) ;
big_arr . push ( b2 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
let poseidon = Poseidon ::new ( ) ;
let h = poseidon . hash ( big_arr . clone ( ) ) . unwrap ( ) ;
assert_eq ! (
h34 . to_string ( ) ,
"Fr(0x25fe79e16e9d530a400584ffe556b1fe69acda374065860c43698cab1c16b3a1)" // "17185195740979599334254027721507328033796809509313949281114643312710535000993"
h . to_string ( ) ,
"Fr(0x08ca0a9154fccd6426092b2404e1ceeb80a7849734f1d3fe7952c2075e489566)" // "3975478831357328722254985704342968745327876719981393787143845259590563829094"
) ;
let mut big_arr : Vec < Fr > = Vec ::new ( ) ;
big_arr . push ( b1 . clone ( ) ) ;
big_arr . push ( b2 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
let poseidon = Poseidon ::new ( ) ;
let h = poseidon . hash ( big_arr . clone ( ) ) . unwrap ( ) ;
assert_eq ! (
h . to_string ( ) ,
"Fr(0x2bb6c270db4ca49d129e315cdad9e0e678c1692c420dbf4667fdabc0f158e4ae)" // "19772360636270345724087386688434825760738403416279047262510528378903625000110"
) ;
let mut big_arr : Vec < Fr > = Vec ::new ( ) ;
big_arr . push ( b3 . clone ( ) ) ;
big_arr . push ( b4 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
let poseidon = Poseidon ::new ( ) ;
let h = poseidon . hash ( big_arr . clone ( ) ) . unwrap ( ) ;
assert_eq ! (
h . to_string ( ) ,
"Fr(0x07087ef123b0fc18a7487a9b3112aec23601e3d2b7ea27a85b35c7ecb595e6f6)" // "3181200837746671699652342497997860344148947482942465819251904554707352676086"
) ;
let mut big_arr : Vec < Fr > = Vec ::new ( ) ;
big_arr . push ( b3 . clone ( ) ) ;
big_arr . push ( b4 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
big_arr . push ( b0 . clone ( ) ) ;
let h = poseidon . hash ( big_arr . clone ( ) ) . unwrap ( ) ;
assert_eq ! (
h . to_string ( ) ,
"Fr(0x128a815839bb66db834533b9c837e5a09df55e90aa9aba7ad46782234e083c20)" // "8386348873272147968934270337233829407378789978142456170950021426339096575008"
) ;
let mut big_arr : Vec < Fr > = Vec ::new ( ) ;
big_arr . push ( b1 . clone ( ) ) ;
big_arr . push ( b2 . clone ( ) ) ;
big_arr . push ( b3 . clone ( ) ) ;
big_arr . push ( b4 . clone ( ) ) ;
big_arr . push ( b5 . clone ( ) ) ;
big_arr . push ( b6 . clone ( ) ) ;
let h = poseidon . hash ( big_arr . clone ( ) ) . unwrap ( ) ;
assert_eq ! (
h . to_string ( ) ,
"Fr(0x0b807dafd5ecc62acdf7ae48e3a1dfb14ccc1ce398f865ac85ff0b4afd90ea6c)" // "5202465217520500374834597824465244016759843635092906214933648999760272616044"
) ;
}
}