@@ -230,17 +230,14 @@ mod tests {
230230
231231 let pok = MessagesPoKGenerator :: init ( & mut rng, & messages, & params, & h) . unwrap ( ) ;
232232
233- let mut chal_bytes_prover = vec ! [ ] ;
234- pok. challenge_contribution ( & mut chal_bytes_prover , & params, & h)
233+ let mut chal_bytes = vec ! [ ] ;
234+ pok. challenge_contribution ( & mut chal_bytes , & params, & h)
235235 . unwrap ( ) ;
236- let challenge_prover =
237- compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes_prover) ;
236+ let challenge = compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes) ;
238237
239- let proof = pok. clone ( ) . gen_proof ( & challenge_prover ) . unwrap ( ) ;
238+ let proof = pok. clone ( ) . gen_proof ( & challenge ) . unwrap ( ) ;
240239
241- proof
242- . verify ( & challenge_prover, empty ( ) , & params, & h)
243- . unwrap ( ) ;
240+ proof. verify ( & challenge, empty ( ) , & params, & h) . unwrap ( ) ;
244241 }
245242 }
246243
@@ -269,21 +266,15 @@ mod tests {
269266
270267 assert_eq ! ( messages. len( ) / 2 , pok. com_j. len( ) ) ;
271268
272- let mut chal_bytes_prover = vec ! [ ] ;
273- pok. challenge_contribution ( & mut chal_bytes_prover , & params, & h)
269+ let mut chal_bytes = vec ! [ ] ;
270+ pok. challenge_contribution ( & mut chal_bytes , & params, & h)
274271 . unwrap ( ) ;
275- let challenge_prover =
276- compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes_prover) ;
272+ let challenge = compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes) ;
277273
278- let proof = pok. clone ( ) . gen_proof ( & challenge_prover ) . unwrap ( ) ;
274+ let proof = pok. clone ( ) . gen_proof ( & challenge ) . unwrap ( ) ;
279275
280276 proof
281- . verify (
282- & challenge_prover,
283- ( 0 ..messages. len ( ) ) . step_by ( 2 ) ,
284- & params,
285- & h,
286- )
277+ . verify ( & challenge, ( 0 ..messages. len ( ) ) . step_by ( 2 ) , & params, & h)
287278 . unwrap ( ) ;
288279 }
289280 }
@@ -385,17 +376,16 @@ mod tests {
385376
386377 assert_eq ! ( messages. len( ) / 2 , pok. com_j. len( ) ) ;
387378
388- let mut chal_bytes_prover = vec ! [ ] ;
389- pok. challenge_contribution ( & mut chal_bytes_prover , & params, & h)
379+ let mut chal_bytes = vec ! [ ] ;
380+ pok. challenge_contribution ( & mut chal_bytes , & params, & h)
390381 . unwrap ( ) ;
391- let challenge_prover =
392- compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes_prover) ;
382+ let challenge = compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes) ;
393383
394- let proof = pok. clone ( ) . gen_proof ( & challenge_prover ) . unwrap ( ) ;
384+ let proof = pok. clone ( ) . gen_proof ( & challenge ) . unwrap ( ) ;
395385 let mut indices = ( 0 ..messages. len ( ) ) . step_by ( 2 ) . rev ( ) ;
396386
397387 assert_eq ! (
398- proof. verify( & challenge_prover , indices. clone( ) , & params, & h, ) ,
388+ proof. verify( & challenge , indices. clone( ) , & params, & h, ) ,
399389 Err ( MessagesPoKError :: RevealedIndicesMustBeUniqueAndSorted {
400390 previous: indices. next( ) . unwrap( ) ,
401391 current: indices. next( ) . unwrap( )
@@ -426,17 +416,14 @@ mod tests {
426416 )
427417 . unwrap ( ) ;
428418
429- let mut chal_bytes_prover = vec ! [ ] ;
430- pok. challenge_contribution ( & mut chal_bytes_prover , & params, & h)
419+ let mut chal_bytes = vec ! [ ] ;
420+ pok. challenge_contribution ( & mut chal_bytes , & params, & h)
431421 . unwrap ( ) ;
432- let challenge_prover =
433- compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes_prover) ;
422+ let challenge = compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes) ;
434423
435- let proof = pok. clone ( ) . gen_proof ( & challenge_prover ) . unwrap ( ) ;
424+ let proof = pok. clone ( ) . gen_proof ( & challenge ) . unwrap ( ) ;
436425
437- proof
438- . verify ( & challenge_prover, empty ( ) , & params, & h)
439- . unwrap ( ) ;
426+ proof. verify ( & challenge, empty ( ) , & params, & h) . unwrap ( ) ;
440427 }
441428 }
442429
@@ -460,22 +447,17 @@ mod tests {
460447 )
461448 . unwrap ( ) ;
462449
463- let mut chal_bytes_prover = vec ! [ ] ;
464- pok. challenge_contribution ( & mut chal_bytes_prover , & params, & h)
450+ let mut chal_bytes = vec ! [ ] ;
451+ pok. challenge_contribution ( & mut chal_bytes , & params, & h)
465452 . unwrap ( ) ;
466- let challenge_prover =
467- compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes_prover) ;
453+ let challenge = compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes) ;
468454
469- let mut proof = pok. clone ( ) . gen_proof ( & challenge_prover) . unwrap ( ) ;
470- assert ! ( proof
471- . verify( & challenge_prover, empty( ) , & params, & h)
472- . is_ok( ) ) ;
455+ let mut proof = pok. clone ( ) . gen_proof ( & challenge) . unwrap ( ) ;
456+ assert ! ( proof. verify( & challenge, empty( ) , & params, & h) . is_ok( ) ) ;
473457
474458 proof. com_resp . response . 0 [ 0 ] = rand ( & mut rng) ;
475459
476- assert ! ( proof
477- . verify( & challenge_prover, empty( ) , & params, & h)
478- . is_err( ) ) ;
460+ assert ! ( proof. verify( & challenge, empty( ) , & params, & h) . is_err( ) ) ;
479461 }
480462
481463 #[ test]
@@ -498,22 +480,17 @@ mod tests {
498480 )
499481 . unwrap ( ) ;
500482
501- let mut chal_bytes_prover = vec ! [ ] ;
502- pok. challenge_contribution ( & mut chal_bytes_prover , & params, & h)
483+ let mut chal_bytes = vec ! [ ] ;
484+ pok. challenge_contribution ( & mut chal_bytes , & params, & h)
503485 . unwrap ( ) ;
504- let challenge_prover =
505- compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes_prover) ;
486+ let challenge = compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes) ;
506487
507- let mut proof = pok. clone ( ) . gen_proof ( & challenge_prover) . unwrap ( ) ;
508- assert ! ( proof
509- . verify( & challenge_prover, empty( ) , & params, & h)
510- . is_ok( ) ) ;
488+ let mut proof = pok. clone ( ) . gen_proof ( & challenge) . unwrap ( ) ;
489+ assert ! ( proof. verify( & challenge, empty( ) , & params, & h) . is_ok( ) ) ;
511490
512491 * proof. com_resp . value = G1 :: rand ( & mut rng) . into_affine ( ) ;
513492
514- assert ! ( proof
515- . verify( & challenge_prover, empty( ) , & params, & h)
516- . is_err( ) ) ;
493+ assert ! ( proof. verify( & challenge, empty( ) , & params, & h) . is_err( ) ) ;
517494 }
518495
519496 #[ test]
@@ -536,36 +513,44 @@ mod tests {
536513 )
537514 . unwrap ( ) ;
538515
539- let mut chal_bytes_prover = vec ! [ ] ;
540- pok. challenge_contribution ( & mut chal_bytes_prover , & params, & h)
516+ let mut chal_bytes = vec ! [ ] ;
517+ pok. challenge_contribution ( & mut chal_bytes , & params, & h)
541518 . unwrap ( ) ;
542- let challenge_prover =
543- compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes_prover) ;
519+ let challenge = compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes) ;
544520
545- let mut proof = pok. clone ( ) . gen_proof ( & challenge_prover ) . unwrap ( ) ;
521+ let mut proof = pok. clone ( ) . gen_proof ( & challenge ) . unwrap ( ) ;
546522
547- assert ! ( proof
548- . verify( & challenge_prover, empty( ) , & params, & h)
549- . is_ok( ) ) ;
523+ assert ! ( proof. verify( & challenge, empty( ) , & params, & h) . is_ok( ) ) ;
550524
551525 * proof. com_j_resp . first_mut ( ) . unwrap ( ) . value = G1 :: rand ( & mut rng) . into_affine ( ) ;
552526
553- assert ! ( proof
554- . verify( & challenge_prover, empty( ) , & params, & h)
555- . is_err( ) ) ;
527+ assert ! ( proof. verify( & challenge, empty( ) , & params, & h) . is_err( ) ) ;
556528 }
557529
558530 #[ test]
559531 fn empty_proof ( ) {
560532 let mut rng = StdRng :: seed_from_u64 ( 0u64 ) ;
561- let ( _sk , _pk , params, _messages ) = test_setup :: < Bls12_381 , Blake2b512 , _ > ( & mut rng, 1 ) ;
533+ let ( _ , _ , params, messages ) = test_setup :: < Bls12_381 , Blake2b512 , _ > ( & mut rng, 1 ) ;
562534 let h = G1 :: rand ( & mut rng) . into_affine ( ) ;
563535
564- assert_eq ! (
565- MessagesPoKGenerator :: init( & mut rng, & [ ] , & params, & h) ,
566- Err ( MessagesPoKError :: MessageInputError (
567- MessageUnpackingError :: NoMessagesProvided
568- ) )
569- ) ;
536+ let pok = MessagesPoKGenerator :: init (
537+ & mut rng,
538+ messages. iter ( ) . map ( |_| CommitMessage :: RevealMessage ) ,
539+ & params,
540+ & h,
541+ )
542+ . unwrap ( ) ;
543+
544+ let mut chal_bytes = vec ! [ ] ;
545+ pok. challenge_contribution ( & mut chal_bytes, & params, & h)
546+ . unwrap ( ) ;
547+ let challenge = compute_random_oracle_challenge :: < Fr , Blake2b512 > ( & chal_bytes) ;
548+
549+ let proof = pok. clone ( ) . gen_proof ( & challenge) . unwrap ( ) ;
550+ let indices = ( 0 ..messages. len ( ) ) . rev ( ) ;
551+
552+ assert ! ( proof
553+ . verify( & challenge, indices. clone( ) , & params, & h)
554+ . is_ok( ) ) ;
570555 }
571556}
0 commit comments