espresso_types/v0/
nsproof.rs

1use hotshot_query_service::VidCommon;
2use hotshot_types::{data::VidCommitment, vid::avidm::AvidMShare};
3use serde::{Deserialize, Serialize};
4
5use crate::{
6    v0::{NamespaceId, NsIndex, NsPayload, NsTable, Payload, Transaction},
7    v0_1::ADVZNsProof,
8    v0_3::{AvidMIncorrectEncodingNsProof, AvidMNsProof},
9};
10
11#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
12pub struct NamespaceProofQueryData {
13    pub proof: Option<NsProof>,
14    pub transactions: Vec<Transaction>,
15}
16
17#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
18pub struct ADVZNamespaceProofQueryData {
19    pub proof: Option<ADVZNsProof>,
20    pub transactions: Vec<Transaction>,
21}
22
23/// Each variant represents a specific version of a namespace proof.
24#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
25pub enum NsProof {
26    /// V0 proof for ADVZ
27    V0(ADVZNsProof),
28    /// V1 proof for AvidM, contains only correct encoding proof
29    V1(AvidMNsProof),
30    /// Incorrect encoding proof for AvidM (only supported after API version 1.1)
31    V1IncorrectEncoding(AvidMIncorrectEncodingNsProof),
32}
33
34impl NsProof {
35    pub fn new(payload: &Payload, index: &NsIndex, common: &VidCommon) -> Option<NsProof> {
36        match common {
37            VidCommon::V0(common) => Some(NsProof::V0(ADVZNsProof::new(payload, index, common)?)),
38            VidCommon::V1(common) => Some(NsProof::V1(AvidMNsProof::new(payload, index, common)?)),
39        }
40    }
41
42    pub fn v1_1_new_with_incorrect_encoding(
43        shares: &[AvidMShare],
44        ns_table: &NsTable,
45        index: &NsIndex,
46        commit: &VidCommitment,
47        common: &VidCommon,
48    ) -> Option<NsProof> {
49        match common {
50            VidCommon::V1(common) => Some(NsProof::V1IncorrectEncoding(
51                AvidMIncorrectEncodingNsProof::new(shares, ns_table, index, commit, common)?,
52            )),
53            _ => None,
54        }
55    }
56
57    pub fn verify(
58        &self,
59        ns_table: &NsTable,
60        commit: &VidCommitment,
61        common: &VidCommon,
62    ) -> Option<(Vec<Transaction>, NamespaceId)> {
63        match (self, common) {
64            (Self::V0(proof), VidCommon::V0(common)) => proof.verify(ns_table, commit, common),
65            (Self::V1(proof), VidCommon::V1(common)) => proof.verify(ns_table, commit, common),
66            (Self::V1IncorrectEncoding(proof), VidCommon::V1(common)) => {
67                proof.verify(ns_table, commit, common)
68            },
69            _ => {
70                tracing::error!("Incompatible version of VidCommon and NsProof.");
71                None
72            },
73        }
74    }
75
76    pub fn export_all_txs(&self, ns_id: &NamespaceId) -> Vec<Transaction> {
77        match self {
78            Self::V0(proof) => proof.export_all_txs(ns_id),
79            Self::V1(AvidMNsProof(proof)) => {
80                NsPayload::from_bytes_slice(&proof.ns_payload).export_all_txs(ns_id)
81            },
82            Self::V1IncorrectEncoding(_) => vec![],
83        }
84    }
85}
espresso_types/v0/nsproof.rs

espresso_types/v0/
nsproof.rs
