feat: Include zkvm and prooftype in ProofReceipt

Author: prajwolrg

adapters/native/src/host.rs

@@ -1,8 +1,8 @@
 use std::{fmt, sync::Arc};
 
 use zkaleido::{
-    Proof, ProofReceipt, ProofType, PublicValues, VerifyingKey, VerifyingKeyCommitment, ZkVmError,
-    ZkVmExecutor, ZkVmHost, ZkVmProver, ZkVmResult, ZkVmVerifier,
+    Proof, ProofReceipt, ProofType, PublicValues, VerifyingKey, VerifyingKeyCommitment, ZkVm,
+    ZkVmError, ZkVmExecutor, ZkVmHost, ZkVmProver, ZkVmResult, ZkVmVerifier,
 };
 
 use crate::{env::NativeMachine, input::NativeMachineInputBuilder, proof::NativeProofReceipt};
@@ -51,7 +51,7 @@ impl ZkVmProver for NativeHost {
     ) -> ZkVmResult<NativeProofReceipt> {
         let public_values = self.execute(native_machine)?;
         let proof = Proof::default();
-        Ok(ProofReceipt::new(proof, public_values).try_into()?)
+        Ok(ProofReceipt::new(proof, public_values, ZkVm::Native).try_into()?)
     }
 }
 

adapters/risc0/host/src/proof.rs

@@ -1,5 +1,5 @@
 use risc0_zkvm::{InnerReceipt, Receipt};
-use zkaleido::{Proof, ProofReceipt, PublicValues, ZkVmProofError};
+use zkaleido::{Proof, ProofReceipt, PublicValues, ZkVm, ZkVmProofError};
 
 #[derive(Debug, Clone)]
 pub struct Risc0ProofReceipt(Receipt);
@@ -51,6 +51,6 @@ impl TryFrom<Risc0ProofReceipt> for ProofReceipt {
         };
         let proof = Proof::new(proof_bytes);
         let public_values = PublicValues::new(value.0.journal.bytes.to_vec());
-        Ok(ProofReceipt::new(proof, public_values))
+        Ok(ProofReceipt::new(proof, public_values, ZkVm::Risc0))
     }
 }

adapters/sp1/host/src/proof.rs

@@ -1,5 +1,5 @@
 use sp1_sdk::{SP1Proof, SP1ProofWithPublicValues, SP1PublicValues};
-use zkaleido::{Proof, ProofReceipt, PublicValues, ZkVmProofError};
+use zkaleido::{Proof, ProofReceipt, PublicValues, ZkVm, ZkVmProofError};
 
 #[derive(Debug, Clone)]
 pub struct SP1ProofReceipt(SP1ProofWithPublicValues);
@@ -62,6 +62,6 @@ impl TryFrom<SP1ProofReceipt> for ProofReceipt {
         let proof = Proof::new(proof_bytes);
         let public_values = PublicValues::new(sp1_receipt.public_values.to_vec());
 
-        Ok(ProofReceipt::new(proof, public_values))
+        Ok(ProofReceipt::new(proof, public_values, ZkVm::SP1))
     }
 }

zkaleido/src/lib.rs

@@ -15,6 +15,7 @@
 //! - **Error Handling**: A set of error enums (e.g., `ZkVmError`) provides comprehensive error
 //!   reporting and integration with Rust's `thiserror` crate for detailed diagnostics.
 
+use arbitrary::Arbitrary;
 use borsh::{BorshDeserialize, BorshSerialize};
 use serde::{Deserialize, Serialize};
 
@@ -48,18 +49,21 @@ pub use verifier::*;
     Clone,
     Copy,
     PartialEq,
+    Default,
     Eq,
     Hash,
     BorshSerialize,
     BorshDeserialize,
     Serialize,
     Deserialize,
+    Arbitrary,
 )]
 pub enum ZkVm {
     /// SP1 ZKVM
     SP1,
     /// Risc0 ZKVM
     Risc0,
     /// Native ZKVM
+    #[default]
     Native,
 }

zkaleido/src/proof.rs

@@ -4,7 +4,7 @@ use arbitrary::Arbitrary;
 use borsh::{BorshDeserialize, BorshSerialize};
 use serde::{Deserialize, Serialize};
 
-use crate::{ZkVmError, ZkVmResult};
+use crate::{ZkVm, ZkVmError, ZkVmResult};
 
 /// Macro to define a newtype wrapper around `Vec<u8>` with common implementations.
 macro_rules! define_byte_wrapper {
@@ -90,14 +90,17 @@ pub struct ProofReceipt {
     proof: Proof,
     /// The public values associated with the proof.
     public_values: PublicValues,
+    /// ZKVM used to generate this proof
+    zkvm: ZkVm,
 }
 
 impl ProofReceipt {
     /// Creates a new `ProofReceipt` from proof and it's associated public values
-    pub fn new(proof: Proof, public_values: PublicValues) -> Self {
+    pub fn new(proof: Proof, public_values: PublicValues, zkvm: ZkVm) -> Self {
         Self {
             proof,
             public_values,
+            zkvm,
         }
     }
 
@@ -111,6 +114,11 @@ impl ProofReceipt {
         &self.public_values
     }
 
+    /// Returns the ZKVM used to generate the proof.
+    pub fn zkvm(&self) -> &ZkVm {
+        &self.zkvm
+    }
+
     /// Saves the proof to a path.
     pub fn save(&self, path: impl AsRef<Path>) -> ZkVmResult<()> {
         bincode::serialize_into(File::create(path).expect("failed to open file"), self)
@@ -181,9 +189,22 @@ impl VerifyingKeyCommitment {
 }
 
 /// Enumeration of proof types supported by the system.
-#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+#[derive(
+    Debug,
+    Clone,
+    Copy,
+    Default,
+    PartialEq,
+    Eq,
+    Serialize,
+    Deserialize,
+    BorshSerialize,
+    BorshDeserialize,
+    Arbitrary,
+)]
 pub enum ProofType {
     /// Represents a Groth16 proof.
+    #[default]
     Groth16,
     /// Represents a core proof.
     Core,