logos-blockchain/lssa
1
0
Fork 0
mirror of https://github.com/logos-blockchain/lssa.git synced 2026-01-09 00:33:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

add encode_utxo_for_owners implementation

Browse Source
This commit is contained in:
Rostyslav Tyshko 2025-06-06 16:53:20 -04:00
parent 036585d747
commit 5ccbb5ac21
2 changed files with 31 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
sc_core/src/public_context.rs
View File

@ -3,6 +3,7 @@ use std::collections::BTreeMap;
use accounts::account_core::{AccountAddress, AccountPublicMask};
use common::merkle_tree_public::TreeHashType;
use serde::{ser::SerializeStruct, Serialize};
use crate::traits::IPrivateOutput;
pub const PUBLIC_SC_CONTEXT: &str = "PublicSCContext";
pub const CALLER_ADDRESS: &str = "caller_address";
@ -87,6 +88,34 @@ impl PublicSCContext {
Ok(u64_list)
}
pub fn encode_utxo_for_owners<IPO: IPrivateOutput>(&self, private_outputs: IPO) -> Vec<(Vec<u8>, Vec<u8>, u8)> {
let utxos = private_outputs.make_utxo_list();
// ToDo: when errorhandling is implemented this `unwrap()` call has to be removed
let caller_account_mask = self.account_masks.get(&self.caller_address).unwrap();
let ephm_key_holder = caller_account_mask.produce_ephemeral_key_holder();
let encoded_data = utxos
.iter()
.map(|utxo| {
// ToDo: when errorhandling is implemented this `unwrap()` call has to be removed
let account_mask = self.account_masks.get(&utxo.owner).unwrap();
(
AccountPublicMask::encrypt_data(
&ephm_key_holder,
account_mask.viewing_public_key,
&serde_json::to_vec(&utxo).unwrap(),
),
account_mask.make_tag(),
)
})
.map(|((ciphertext, nonce), tag)| (ciphertext, nonce.to_vec(), tag))
.collect();
encoded_data
}
}
#[cfg(test)]

5
template/zk_test_template/test_methods/guest/src/bin/template_sc.rs
View File

@ -44,17 +44,16 @@ fn main() {
env::commit(&state_changes);
//Next, push one of possible variants depending on execution type
//ToDo: Make UTXO encoding for their owners available from PublicSCContext
{
public => {
env::commit(&public_outputs);
},
private => {
env::commit(&(public_context.encode_utxo_for_owners(private_outputs.make_utxo_list())));
env::commit(&(public_context.encode_utxo_for_owners(private_outputs)));
},
shielded | deshielded => {
env::commit(&public_outputs);
env::commit(&(public_context.encode_utxo_for_owners(private_outputs.make_utxo_list())));
env::commit(&(public_context.encode_utxo_for_owners(private_outputs)));
},
}
}