You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

41 lines
1.4 KiB

  1. use ark_ff::PrimeField;
  2. use ark_serialize::CanonicalSerialize;
  3. use merlin::Transcript as MerlinTranscript;
  4. use std::marker::PhantomData;
  5. // TODO poseidon transcript (not keccak)
  6. // This Transcript approach is a modified version from https://github.com/arkworks-rs/gemini ,
  7. // using Merlin transcript (https://merlin.cool).
  8. pub struct Transcript<F: PrimeField> {
  9. phantom: PhantomData<F>,
  10. transcript: MerlinTranscript,
  11. }
  12. impl<F: PrimeField> Transcript<F> {
  13. pub fn new() -> Self {
  14. Self {
  15. phantom: PhantomData::default(),
  16. transcript: MerlinTranscript::new(b"transcript"),
  17. }
  18. }
  19. pub fn add<S: CanonicalSerialize>(&mut self, label: &'static [u8], r: &S) {
  20. let mut msg = Vec::new();
  21. r.serialize_uncompressed(&mut msg).unwrap();
  22. self.transcript.append_message(label, &msg);
  23. }
  24. pub fn get_challenge(&mut self, label: &'static [u8]) -> F {
  25. let mut bytes = [0u8; 64];
  26. self.transcript.challenge_bytes(label, &mut bytes);
  27. let challenge = F::from_le_bytes_mod_order(bytes.as_ref());
  28. self.add(b"get challenge", &challenge);
  29. challenge
  30. }
  31. pub fn get_challenge_vec(&mut self, label: &'static [u8], n: usize) -> Vec<F> {
  32. let mut c: Vec<F> = vec![F::zero(); n];
  33. for i in 0..n {
  34. c[i] = self.get_challenge(label);
  35. }
  36. c
  37. }
  38. }