zkpo 0.1.7

Zero knowledge program operations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

//! A `ZKAgent` implementation for the risc0@2
//! prover implementation.
//!
//! By default produces recursively verified STARK proofs.

use std::sync::OnceLock;

use anyhow::Result;
use risc0_zkvm::Digest;
use risc0_zkvm::ExecutorEnv;
use risc0_zkvm::Receipt;
use risc0_zkvm::default_prover;

use crate::prelude::*;

static DEFAULT: OnceLock<ZKRiscZeroAgent> = OnceLock::new();

#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[derive(Clone, Debug)]
pub struct ZKRiscZeroArg {
    pub receipt_bytes: Vec<u8>,
    pub program_id: [u8; 32],
}

impl ZKExe for ZKRiscZeroArg {
    fn cipher_bytes(&self) -> &[u8] {
        &self.receipt_bytes
    }

    fn program_id(&self) -> &[u8; 32] {
        &self.program_id
    }

    fn agent(&self) -> &dyn ZKAgent {
        ZKRiscZeroAgent::singleton()
    }

    fn program(&self) -> Option<&dyn ZKProgram> {
        None
    }
}

/// A risc0@2 zk agent. Produces compressed STARKs. `ZKProgram`s compiled
/// from the risc0 toolchain are not yet functional.
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[derive(Clone, Default, Debug)]
pub struct ZKRiscZeroAgent;

impl ZKRiscZeroAgent {
    pub fn singleton() -> &'static Self {
        DEFAULT.get_or_init(|| ZKRiscZeroAgent::default())
    }
}

impl ZKAgent for ZKRiscZeroAgent {
    fn execute(&self, input: &[u8], program: &dyn ZKProgram) -> Result<Box<dyn ZKExe>> {
        // build an executor with the supplied input
        let env = ExecutorEnv::builder().write_slice(input).build()?;
        // use the default risc0 prover
        let prover = default_prover();
        // Produce a receipt by proving the specified ELF binary.
        let receipt = prover.prove(env, program.elf())?.receipt;
        Ok(Box::new(ZKRiscZeroArg {
            receipt_bytes: bincode::serialize(&receipt)?,
            program_id: program.id().clone(),
        }))
    }

    fn verify(&self, proof: &dyn ZKExe) -> Result<Vec<u8>> {
        let receipt = bincode::deserialize::<Receipt>(proof.cipher_bytes())?;
        let digest = Digest::from_bytes(*proof.program_id());
        receipt.verify(digest)?;
        Ok(receipt.journal.bytes)
    }
}