logos-blockchain/lssa-zkvm-testing
1
0
Fork 0
mirror of https://github.com/logos-blockchain/lssa-zkvm-testing.git synced 2026-01-04 14:23:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

add comments

Browse Source
This commit is contained in:
Sergio Chouhy 2025-07-11 19:35:54 -03:00
parent be7da3410b
commit 296fc2c5ba
2 changed files with 55 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
risc0-selective-privacy-poc/core/src/lib.rs
View File

@ -17,6 +17,10 @@ impl Account {
Self { address, balance: 0, nonce }
}
pub fn new(address: [u32; 8], nonce: [u32; 8]) -> Self {
Self { address, balance: 0, nonce }
}
/// Returns Hash(Account)
pub fn commitment(&self) -> [u32; 8] {
hash(&to_vec(&self).unwrap())

93
risc0-selective-privacy-poc/src/private_execution.rs
View File

@ -7,6 +7,57 @@ use outer_methods::{
use risc0_zkvm::{default_prover, ExecutorEnv, Receipt};
use toy_example_core::Account;
/// A private execution of the transfer function.
/// This actually "burns" a sender private account and "mints" two new private accounts:
/// one for the recepient with the transfered balance, and another owned by the sender with the remaining balance.
fn run_private_execution_of_transfer_program() {
let commitment_tree_root = [0xdd, 0xee, 0xaa, 0xdd, 0xbb, 0xee, 0xee, 0xff];
// This is supposed to be an existing private account (UTXO) with balance equal to 150.
// And it is supposed to be a private account of the user running this private execution (hence the access to the private key)
let sender_private_key = [0; 8];
let mut sender = Account::new_from_private_key(sender_private_key, [1; 8]);
sender.balance = 150;
let balance_to_move: u128 = 3;
// This is the new private account (UTXO) being minted by this private execution.
// (The `receiver_address` would be <Npk> in UTXO's terminology)
let receiver_address = [99; 8];
let receiver = Account::new(receiver_address, [1; 8]);
// Prove inner program and get post state of the accounts
let (inner_receipt, sender_post, receiver_post) = prove_inner(sender.clone(), receiver.clone(), balance_to_move);
// Prove outer program.
// This computes the nullifier for the input account
// and commitments for the accounts post states.
let mut env_builder = ExecutorEnv::builder();
env_builder.add_assumption(inner_receipt);
env_builder.write(&sender_private_key).unwrap();
env_builder.write(&sender).unwrap();
env_builder.write(&receiver) .unwrap();
env_builder.write(&balance_to_move).unwrap();
env_builder.write(&sender_post).unwrap();
env_builder.write(&receiver_post).unwrap();
env_builder.write(&commitment_tree_root).unwrap();
env_builder.write(&TRANSFER_ID).unwrap();
let env = env_builder.build().unwrap();
let prover = default_prover();
let prove_info = prover
.prove(env, OUTER_ELF)
.unwrap();
let receipt = prove_info.receipt;
// Sanity check
receipt.verify(OUTER_ID).unwrap();
let output: [[u32; 8]; 3] = receipt.journal.decode().unwrap();
println!("nullifier: {:?}", output[0]);
println!("commitment_1: {:?}", output[1]);
println!("commitment_2: {:?}", output[2]);
}
fn prove_inner(sender: Account, receiver: Account, balance_to_move: u128) -> (Receipt, Account, Account) {
let mut env_builder = ExecutorEnv::builder();
env_builder.write(&sender).unwrap();
@ -35,48 +86,6 @@ fn prove_inner(sender: Account, receiver: Account, balance_to_move: u128) -> (Re
(receipt, sender_post, receiver_post)
}
fn run_private_execution_of_transfer_program() {
let commitment_tree_root = [0xdd, 0xee, 0xaa, 0xdd, 0xbb, 0xee, 0xee, 0xff];
let sender_private_key = [0; 8];
let mut sender = Account::new_from_private_key(sender_private_key, [1; 8]);
sender.balance = 150;
let receiver_private_key = [99; 8];
let receiver = Account::new_from_private_key(receiver_private_key, [1; 8]);
let balance_to_move: u128 = 3;
// Prove inner
let (inner_receipt, sender_post, receiver_post) = prove_inner(sender.clone(), receiver.clone(), balance_to_move);
// Prover outer
let mut env_builder = ExecutorEnv::builder();
env_builder.add_assumption(inner_receipt);
env_builder.write(&sender_private_key).unwrap();
env_builder.write(&sender).unwrap();
env_builder.write(&receiver) .unwrap();
env_builder.write(&balance_to_move).unwrap();
env_builder.write(&sender_post).unwrap();
env_builder.write(&receiver_post).unwrap();
env_builder.write(&commitment_tree_root).unwrap();
env_builder.write(&TRANSFER_ID).unwrap();
let env = env_builder.build().unwrap();
let prover = default_prover();
let prove_info = prover
.prove(env, OUTER_ELF)
.unwrap();
let receipt = prove_info.receipt;
// Sanity check
receipt.verify(OUTER_ID).unwrap();
let output: [[u32; 8]; 3] = receipt.journal.decode().unwrap();
println!("nullifier: {:?}", output[0]);
println!("commitment_1: {:?}", output[1]);
println!("commitment_2: {:?}", output[2]);
}
#[cfg(test)]
mod tests {
use super::*;