miden-objects 0.17.0-rc.5

Canonical Protobuf representations for Miden protocol objects
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

use alloc::collections::BTreeMap;
use alloc::sync::Arc;
use alloc::vec;
use alloc::vec::Vec;

use miden_protocol::Word;
use miden_protocol::account::{
    AccountId,
    AccountIdVersion,
    AccountType,
    AccountUpdateDetails,
    AssetCallbackFlag,
};
use miden_protocol::batch::{ProposedBatch, ProvenBatch};
use miden_protocol::block::BlockHeader;
use miden_protocol::crypto::merkle::mmr::{Mmr, PartialMmr};
use miden_protocol::testing::dummy_execution_proof;
use miden_protocol::transaction::{
    InputNoteCommitment,
    OutputNote,
    PartialBlockchain,
    ProvenTransaction,
    TxAccountUpdate,
};
use prost::Message;

use crate::test_utils::error_source;
use crate::{DecodeMessage, DecodeMessageExt, VerifyWith, proto};

fn proposal() -> ProposedBatch {
    let mut mmr = Mmr::default();
    for number in 0..3 {
        mmr.add(BlockHeader::mock(number, None, None, &[]).commitment()).unwrap();
    }
    let chain = PartialBlockchain::new(PartialMmr::from_peaks(mmr.peaks()), vec![]).unwrap();
    let header = BlockHeader::mock(3, Some(chain.peaks().hash_peaks()), None, &[]);
    let account = AccountId::dummy(
        [7; 15],
        AccountIdVersion::Version1,
        AccountType::Private,
        AssetCallbackFlag::Disabled,
    );
    let update = TxAccountUpdate::new(
        account,
        Word::from([1_u32; 4]),
        Word::from([2_u32; 4]),
        Word::empty(),
        AccountUpdateDetails::Private,
    )
    .unwrap();
    let transaction = ProvenTransaction::new(
        update,
        Vec::<InputNoteCommitment>::new(),
        Vec::<OutputNote>::new(),
        header.block_num(),
        header.commitment(),
        header.block_num() + 1,
        dummy_execution_proof(),
    )
    .unwrap();
    // A dummy proof is deliberate: structural decoding must not perform proof verification.
    ProposedBatch::new_unverified(vec![Arc::new(transaction)], header, chain, BTreeMap::new())
        .unwrap()
}

fn proven(proposal: &ProposedBatch) -> ProvenBatch {
    ProvenBatch::new(
        proposal.reference_block_header().commitment(),
        proposal.reference_block_header().block_num(),
        proposal.account_updates().values().cloned(),
        proposal.input_notes().clone(),
        proposal.output_notes().to_vec(),
        proposal.batch_expiration_block_num(),
        proposal.transaction_headers(),
        dummy_execution_proof(),
    )
    .unwrap()
}

#[test]
fn proposal_proofs_are_only_checked_by_explicit_verification() {
    let proposal = proposal();
    let wire: proto::transaction::ProposedBatch = (&proposal).into();
    let wire = proto::transaction::ProposedBatch::decode(wire.encode_to_vec().as_slice()).unwrap();
    let decoded = wire.decode_fields().unwrap();
    assert_eq!(decoded.transactions.as_slice().len(), 1);
    let error = decoded.verify_with(96).unwrap_err();
    assert!(
        matches!(
            error_source::<miden_protocol::errors::ProposedBatchError>(&error),
            Some(miden_protocol::errors::ProposedBatchError::TransactionVerificationFailed { .. })
        ),
        "{error}"
    );
}

#[test]
fn proven_batch_roundtrips_and_checks_proposal_agreement() {
    let proposal = proposal();
    let batch = proven(&proposal);
    let wire: proto::transaction::ProvenBatch = (&batch).into();
    let wire = proto::transaction::ProvenBatch::decode(wire.encode_to_vec().as_slice()).unwrap();
    assert_eq!(wire.clone().decode_and_build_unchecked().unwrap(), batch);
    assert_eq!(wire.decode_and_verify_with(&proposal).unwrap(), batch);
}

#[test]
fn proven_batch_rejects_changed_proposal_fields() {
    let proposal = proposal();
    let batch = proven(&proposal);
    for field in ["reference block number", "reference block commitment", "expiration block"] {
        let mut wire: proto::transaction::ProvenBatch = (&batch).into();
        match field {
            "reference block number" => wire.reference_block_num.as_mut().unwrap().block_num -= 1,
            "reference block commitment" => {
                wire.reference_block_commitment = Some(Word::empty().into())
            },
            "expiration block" => wire.expiration_block_num.as_mut().unwrap().block_num += 1,
            _ => unreachable!(),
        }
        let error = wire.decode_and_verify_with(&proposal).unwrap_err();
        assert!(
            matches!(
                error_source::<crate::decoded::transaction::ProvenBatchError>(&error),
                Some(crate::decoded::transaction::ProvenBatchError::ProposalMismatch(actual)) if *actual == field
            ),
            "{error}"
        );
    }
}