mirror of
https://github.com/logos-blockchain/lssa.git
synced 2026-01-02 13:23:10 +00:00
1758 lines
59 KiB
Rust
1758 lines
59 KiB
Rust
use std::sync::{
|
|
atomic::{AtomicU64, Ordering},
|
|
Arc,
|
|
};
|
|
|
|
use common::{public_transfer_receipts::PublicNativeTokenSend, ExecutionFailureKind};
|
|
|
|
use accounts::{
|
|
account_core::{Account, AccountAddress},
|
|
key_management::ephemeral_key_holder::EphemeralKeyHolder,
|
|
};
|
|
use anyhow::Result;
|
|
use chain_storage::NodeChainStore;
|
|
use common::transaction::{Transaction, TransactionPayload, TxKind};
|
|
use config::NodeConfig;
|
|
use log::info;
|
|
use sc_core::proofs_circuits::{
|
|
generate_commitments, generate_nullifiers, generate_nullifiers_se, pedersen_commitment_vec,
|
|
};
|
|
use sequencer_client::{json::SendTxResponse, SequencerClient};
|
|
use serde::{Deserialize, Serialize};
|
|
use storage::sc_db_utils::DataBlobChangeVariant;
|
|
use tokio::{sync::RwLock, task::JoinHandle};
|
|
use utxo::utxo_core::UTXO;
|
|
use zkvm::{
|
|
gas_calculator::GasCalculator, prove_mint_utxo, prove_mint_utxo_multiple_assets,
|
|
prove_send_utxo, prove_send_utxo_deshielded, prove_send_utxo_multiple_assets_one_receiver,
|
|
prove_send_utxo_shielded,
|
|
};
|
|
|
|
pub const BLOCK_GEN_DELAY_SECS: u64 = 20;
|
|
|
|
pub mod chain_storage;
|
|
pub mod config;
|
|
///Module, which includes pre start setup helperfunctions
|
|
pub mod pre_start;
|
|
pub mod sequencer_client;
|
|
|
|
fn vec_u8_to_vec_u64(bytes: Vec<u8>) -> Vec<u64> {
|
|
// Pad with zeros to make sure it's a multiple of 8
|
|
let mut padded = bytes.clone();
|
|
while padded.len() % 8 != 0 {
|
|
padded.push(0);
|
|
}
|
|
|
|
padded
|
|
.chunks(8)
|
|
.map(|chunk| {
|
|
let mut array = [0u8; 8];
|
|
array.copy_from_slice(chunk);
|
|
u64::from_le_bytes(array)
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct MintMoneyPublicTx {
|
|
pub acc: AccountAddress,
|
|
pub amount: u128,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct SendMoneyShieldedTx {
|
|
pub acc_sender: AccountAddress,
|
|
pub amount: u128,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct SendMoneyDeshieldedTx {
|
|
pub receiver_data: Vec<(u128, AccountAddress)>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct UTXOPublication {
|
|
pub utxos: Vec<UTXO>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub enum ActionData {
|
|
MintMoneyPublicTx(MintMoneyPublicTx),
|
|
SendMoneyShieldedTx(SendMoneyShieldedTx),
|
|
SendMoneyDeshieldedTx(SendMoneyDeshieldedTx),
|
|
UTXOPublication(UTXOPublication),
|
|
}
|
|
|
|
pub struct NodeCore {
|
|
pub storage: Arc<RwLock<NodeChainStore>>,
|
|
pub curr_height: Arc<AtomicU64>,
|
|
pub node_config: NodeConfig,
|
|
pub db_updater_handle: JoinHandle<Result<()>>,
|
|
pub sequencer_client: Arc<SequencerClient>,
|
|
pub gas_calculator: GasCalculator,
|
|
}
|
|
|
|
impl NodeCore {
|
|
pub async fn start_from_config_update_chain(config: NodeConfig) -> Result<Self> {
|
|
let client = Arc::new(SequencerClient::new(config.clone())?);
|
|
|
|
let genesis_id = client.get_genesis_id().await?;
|
|
info!("Genesis id is {genesis_id:?}");
|
|
|
|
let genesis_block = client.get_block(genesis_id.genesis_id).await?.block;
|
|
|
|
let (mut storage, mut chain_height) = NodeChainStore::new(config.clone(), genesis_block)?;
|
|
|
|
pre_start::setup_empty_sc_states(&storage).await?;
|
|
|
|
//Chain update loop
|
|
loop {
|
|
let next_block = chain_height + 1;
|
|
|
|
if let Ok(block) = client.get_block(next_block).await {
|
|
storage.dissect_insert_block(block.block)?;
|
|
info!("Preprocessed block with id {next_block:?}");
|
|
} else {
|
|
break;
|
|
}
|
|
|
|
chain_height += 1;
|
|
}
|
|
|
|
let wrapped_storage = Arc::new(RwLock::new(storage));
|
|
let chain_height_wrapped = Arc::new(AtomicU64::new(chain_height));
|
|
|
|
let wrapped_storage_thread = wrapped_storage.clone();
|
|
let wrapped_chain_height_thread = chain_height_wrapped.clone();
|
|
let client_thread = client.clone();
|
|
|
|
let updater_handle = tokio::spawn(async move {
|
|
loop {
|
|
let next_block = wrapped_chain_height_thread.load(Ordering::Relaxed) + 1;
|
|
|
|
if let Ok(block) = client_thread.get_block(next_block).await {
|
|
{
|
|
let mut storage_guard = wrapped_storage_thread.write().await;
|
|
|
|
let block_insertion_result =
|
|
storage_guard.dissect_insert_block(block.block);
|
|
|
|
if block_insertion_result.is_err() {
|
|
info!("Block insertion failed due to {block_insertion_result:?}");
|
|
|
|
block_insertion_result?;
|
|
}
|
|
info!("Processed block with id {next_block:?}");
|
|
}
|
|
|
|
wrapped_chain_height_thread.store(next_block, Ordering::Relaxed);
|
|
} else {
|
|
tokio::time::sleep(std::time::Duration::from_secs(
|
|
config.seq_poll_timeout_secs,
|
|
))
|
|
.await;
|
|
}
|
|
}
|
|
});
|
|
|
|
Ok(Self {
|
|
storage: wrapped_storage,
|
|
curr_height: chain_height_wrapped,
|
|
node_config: config.clone(),
|
|
db_updater_handle: updater_handle,
|
|
sequencer_client: client.clone(),
|
|
gas_calculator: GasCalculator::from(config.gas_config),
|
|
})
|
|
}
|
|
|
|
pub async fn get_roots(&self) -> [[u8; 32]; 2] {
|
|
let storage = self.storage.read().await;
|
|
[
|
|
storage.utxo_commitments_store.get_root().unwrap_or([0; 32]),
|
|
storage.pub_tx_store.get_root().unwrap_or([0; 32]),
|
|
]
|
|
}
|
|
|
|
pub async fn create_new_account(&mut self) -> AccountAddress {
|
|
let account = Account::new();
|
|
account.log();
|
|
|
|
let addr = account.address;
|
|
|
|
{
|
|
let mut write_guard = self.storage.write().await;
|
|
|
|
write_guard.acc_map.insert(account.address, account);
|
|
}
|
|
|
|
addr
|
|
}
|
|
|
|
pub async fn mint_utxo_private(
|
|
&self,
|
|
acc: AccountAddress,
|
|
amount: u128,
|
|
) -> Result<(Transaction, [u8; 32]), ExecutionFailureKind> {
|
|
let (utxo, receipt) = prove_mint_utxo(amount, acc)?;
|
|
let result_hash = utxo.hash;
|
|
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let account = acc_map_read_guard.acc_map.get(&acc).unwrap();
|
|
|
|
let ephm_key_holder = EphemeralKeyHolder::new_os_random();
|
|
ephm_key_holder.log();
|
|
|
|
let eph_pub_key =
|
|
serde_json::to_vec(&ephm_key_holder.generate_ephemeral_public_key()).unwrap();
|
|
|
|
let encoded_data = Account::encrypt_data(
|
|
&ephm_key_holder,
|
|
account.key_holder.viewing_public_key,
|
|
&serde_json::to_vec(&utxo).unwrap(),
|
|
);
|
|
|
|
let tag = account.make_tag();
|
|
|
|
let comm = generate_commitments(&vec![utxo]);
|
|
|
|
let mint_utxo_addr_bytes: Vec<u8> = zkvm::test_methods::MINT_UTXO_ID
|
|
.iter()
|
|
.map(|num| num.to_le_bytes())
|
|
.flatten()
|
|
.collect();
|
|
let sc_addr = hex::encode(mint_utxo_addr_bytes);
|
|
|
|
//Sc does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let sc_state = acc_map_read_guard
|
|
.block_store
|
|
.get_sc_sc_state(&sc_addr)
|
|
.map_err(ExecutionFailureKind::db_error)?;
|
|
|
|
let mut vec_values_u64: Vec<Vec<u64>> = sc_state
|
|
.into_iter()
|
|
.map(|slice| vec_u8_to_vec_u64(slice.to_vec()))
|
|
.collect();
|
|
|
|
let context = acc_map_read_guard.produce_context(account.address);
|
|
|
|
//Will not panic, as PublicScContext is serializable
|
|
let context_public_info: Vec<u64> = context.produce_u64_list_from_context().unwrap();
|
|
vec_values_u64.push(context_public_info);
|
|
|
|
let vec_public_info: Vec<u64> = vec_values_u64.into_iter().flatten().collect();
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(vec_public_info);
|
|
|
|
Ok((
|
|
TransactionPayload {
|
|
tx_kind: TxKind::Private,
|
|
execution_input: vec![],
|
|
execution_output: vec![],
|
|
utxo_commitments_spent_hashes: vec![],
|
|
utxo_commitments_created_hashes: comm
|
|
.into_iter()
|
|
.map(|hash_data| hash_data.try_into().unwrap())
|
|
.collect(),
|
|
nullifier_created_hashes: vec![],
|
|
execution_proof_private: sc_core::transaction_payloads_tools::encode_receipt(
|
|
receipt,
|
|
)
|
|
.unwrap(),
|
|
encoded_data: vec![(encoded_data.0, encoded_data.1.to_vec(), tag)],
|
|
ephemeral_pub_key: eph_pub_key.to_vec(),
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
}
|
|
.into(),
|
|
result_hash,
|
|
))
|
|
}
|
|
|
|
pub async fn mint_utxo_multiple_assets_private(
|
|
&self,
|
|
acc: AccountAddress,
|
|
amount: u128,
|
|
number_of_assets: usize,
|
|
) -> Result<(Transaction, Vec<[u8; 32]>), ExecutionFailureKind> {
|
|
let (utxos, receipt) = prove_mint_utxo_multiple_assets(amount, number_of_assets, acc)?;
|
|
let result_hashes = utxos.iter().map(|utxo| utxo.hash).collect();
|
|
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let account = acc_map_read_guard.acc_map.get(&acc).unwrap();
|
|
|
|
let ephm_key_holder = EphemeralKeyHolder::new_os_random();
|
|
ephm_key_holder.log();
|
|
|
|
let eph_pub_key =
|
|
serde_json::to_vec(&ephm_key_holder.generate_ephemeral_public_key()).unwrap();
|
|
|
|
let encoded_data = utxos
|
|
.iter()
|
|
.map(|utxo| {
|
|
(
|
|
Account::encrypt_data(
|
|
&ephm_key_holder,
|
|
account.key_holder.viewing_public_key,
|
|
&serde_json::to_vec(&utxo).unwrap(),
|
|
),
|
|
account.make_tag(),
|
|
)
|
|
})
|
|
.map(|((ciphertext, nonce), tag)| (ciphertext, nonce.to_vec(), tag))
|
|
.collect();
|
|
|
|
let comm = generate_commitments(&utxos);
|
|
|
|
let mint_multiple_utxo_addr_bytes: Vec<u8> =
|
|
zkvm::test_methods::MINT_UTXO_MULTIPLE_ASSETS_ID
|
|
.iter()
|
|
.map(|num| num.to_le_bytes())
|
|
.flatten()
|
|
.collect();
|
|
let sc_addr = hex::encode(mint_multiple_utxo_addr_bytes);
|
|
|
|
//Sc does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let sc_state = acc_map_read_guard
|
|
.block_store
|
|
.get_sc_sc_state(&sc_addr)
|
|
.map_err(ExecutionFailureKind::db_error)?;
|
|
|
|
let mut vec_values_u64: Vec<Vec<u64>> = sc_state
|
|
.into_iter()
|
|
.map(|slice| vec_u8_to_vec_u64(slice.to_vec()))
|
|
.collect();
|
|
|
|
let context = acc_map_read_guard.produce_context(account.address);
|
|
|
|
//Will not panic, as PublicScContext is serializable
|
|
let context_public_info: Vec<u64> = context.produce_u64_list_from_context().unwrap();
|
|
vec_values_u64.push(context_public_info);
|
|
|
|
let vec_public_info: Vec<u64> = vec_values_u64.into_iter().flatten().collect();
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(vec_public_info);
|
|
|
|
Ok((
|
|
TransactionPayload {
|
|
tx_kind: TxKind::Private,
|
|
execution_input: vec![],
|
|
execution_output: vec![],
|
|
utxo_commitments_spent_hashes: vec![],
|
|
utxo_commitments_created_hashes: comm
|
|
.into_iter()
|
|
.map(|hash_data| hash_data.try_into().unwrap())
|
|
.collect(),
|
|
nullifier_created_hashes: vec![],
|
|
execution_proof_private: sc_core::transaction_payloads_tools::encode_receipt(
|
|
receipt,
|
|
)
|
|
.unwrap(),
|
|
encoded_data,
|
|
ephemeral_pub_key: eph_pub_key.to_vec(),
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
}
|
|
.into(),
|
|
result_hashes,
|
|
))
|
|
}
|
|
|
|
pub async fn transfer_utxo_private(
|
|
&self,
|
|
utxo: UTXO,
|
|
commitment_in: [u8; 32],
|
|
receivers: Vec<(u128, AccountAddress)>,
|
|
) -> Result<(Transaction, Vec<(AccountAddress, [u8; 32])>), ExecutionFailureKind> {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let account = acc_map_read_guard.acc_map.get(&utxo.owner).unwrap();
|
|
|
|
let nullifier = generate_nullifiers(
|
|
&utxo,
|
|
&account
|
|
.key_holder
|
|
.utxo_secret_key_holder
|
|
.nullifier_secret_key
|
|
.to_bytes()
|
|
.to_vec(),
|
|
);
|
|
|
|
let (resulting_utxos, receipt) = prove_send_utxo(utxo, receivers)?;
|
|
let utxo_hashes = resulting_utxos
|
|
.iter()
|
|
.map(|(utxo, addr)| (addr.clone(), utxo.hash))
|
|
.collect();
|
|
|
|
let utxos: Vec<UTXO> = resulting_utxos
|
|
.iter()
|
|
.map(|(utxo, _)| utxo.clone())
|
|
.collect();
|
|
|
|
let ephm_key_holder = EphemeralKeyHolder::new_os_random();
|
|
ephm_key_holder.log();
|
|
|
|
let eph_pub_key =
|
|
serde_json::to_vec(&ephm_key_holder.generate_ephemeral_public_key()).unwrap();
|
|
|
|
let encoded_data: Vec<(Vec<u8>, Vec<u8>, u8)> = utxos
|
|
.iter()
|
|
.map(|utxo_enc| {
|
|
let accout_enc = acc_map_read_guard.acc_map.get(&utxo_enc.owner).unwrap();
|
|
|
|
let (ciphertext, nonce) = Account::encrypt_data(
|
|
&ephm_key_holder,
|
|
accout_enc.key_holder.viewing_public_key,
|
|
&serde_json::to_vec(&utxo_enc).unwrap(),
|
|
);
|
|
|
|
let tag = accout_enc.make_tag();
|
|
|
|
(ciphertext, nonce.to_vec(), tag)
|
|
})
|
|
.collect();
|
|
|
|
let commitments = generate_commitments(&utxos);
|
|
|
|
let send_utxo_addr_bytes: Vec<u8> = zkvm::test_methods::SEND_UTXO_ID
|
|
.iter()
|
|
.map(|num| num.to_le_bytes())
|
|
.flatten()
|
|
.collect();
|
|
let sc_addr = hex::encode(send_utxo_addr_bytes);
|
|
|
|
//Sc does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let sc_state = acc_map_read_guard
|
|
.block_store
|
|
.get_sc_sc_state(&sc_addr)
|
|
.map_err(ExecutionFailureKind::db_error)?;
|
|
|
|
let mut vec_values_u64: Vec<Vec<u64>> = sc_state
|
|
.into_iter()
|
|
.map(|slice| vec_u8_to_vec_u64(slice.to_vec()))
|
|
.collect();
|
|
|
|
let context = acc_map_read_guard.produce_context(account.address);
|
|
|
|
//Will not panic, as PublicScContext is serializable
|
|
let context_public_info: Vec<u64> = context.produce_u64_list_from_context().unwrap();
|
|
vec_values_u64.push(context_public_info);
|
|
|
|
let vec_public_info: Vec<u64> = vec_values_u64.into_iter().flatten().collect();
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(vec_public_info);
|
|
|
|
Ok((
|
|
TransactionPayload {
|
|
tx_kind: TxKind::Private,
|
|
execution_input: vec![],
|
|
execution_output: vec![],
|
|
utxo_commitments_spent_hashes: vec![commitment_in],
|
|
utxo_commitments_created_hashes: commitments
|
|
.into_iter()
|
|
.map(|hash_data| hash_data.try_into().unwrap())
|
|
.collect(),
|
|
nullifier_created_hashes: vec![nullifier.try_into().unwrap()],
|
|
execution_proof_private: sc_core::transaction_payloads_tools::encode_receipt(
|
|
receipt,
|
|
)
|
|
.unwrap(),
|
|
encoded_data,
|
|
ephemeral_pub_key: eph_pub_key.to_vec(),
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
}
|
|
.into(),
|
|
utxo_hashes,
|
|
))
|
|
}
|
|
|
|
pub async fn transfer_utxo_multiple_assets_private(
|
|
&self,
|
|
utxos: Vec<UTXO>,
|
|
commitments_in: Vec<[u8; 32]>,
|
|
number_to_send: usize,
|
|
receiver: AccountAddress,
|
|
) -> Result<(Transaction, Vec<[u8; 32]>, Vec<[u8; 32]>), ExecutionFailureKind> {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let account = acc_map_read_guard.acc_map.get(&utxos[0].owner).unwrap();
|
|
|
|
let nsk = account
|
|
.key_holder
|
|
.utxo_secret_key_holder
|
|
.nullifier_secret_key
|
|
.to_bytes()
|
|
.to_vec();
|
|
|
|
let nullifiers = utxos
|
|
.iter()
|
|
.map(|utxo| generate_nullifiers(utxo, &nsk))
|
|
.map(|vecc| vecc.try_into().unwrap())
|
|
.collect();
|
|
|
|
let (resulting_utxos_receiver, resulting_utxos_not_spent, receipt) =
|
|
prove_send_utxo_multiple_assets_one_receiver(utxos, number_to_send, receiver)?;
|
|
|
|
let utxo_hashes_receiver = resulting_utxos_receiver
|
|
.iter()
|
|
.map(|utxo| utxo.hash)
|
|
.collect();
|
|
|
|
let utxo_hashes_not_spent = resulting_utxos_not_spent
|
|
.iter()
|
|
.map(|utxo| utxo.hash)
|
|
.collect();
|
|
|
|
let ephm_key_holder = EphemeralKeyHolder::new_os_random();
|
|
ephm_key_holder.log();
|
|
|
|
let eph_pub_key =
|
|
serde_json::to_vec(&ephm_key_holder.generate_ephemeral_public_key()).unwrap();
|
|
|
|
let mut encoded_data: Vec<(Vec<u8>, Vec<u8>, u8)> = resulting_utxos_receiver
|
|
.iter()
|
|
.map(|utxo_enc| {
|
|
let accout_enc = acc_map_read_guard.acc_map.get(&utxo_enc.owner).unwrap();
|
|
|
|
let (ciphertext, nonce) = Account::encrypt_data(
|
|
&ephm_key_holder,
|
|
accout_enc.key_holder.viewing_public_key,
|
|
&serde_json::to_vec(&utxo_enc).unwrap(),
|
|
);
|
|
|
|
let tag = accout_enc.make_tag();
|
|
|
|
(ciphertext, nonce.to_vec(), tag)
|
|
})
|
|
.collect();
|
|
|
|
let encoded_data_1: Vec<(Vec<u8>, Vec<u8>, u8)> = resulting_utxos_not_spent
|
|
.iter()
|
|
.map(|utxo_enc| {
|
|
let accout_enc = acc_map_read_guard.acc_map.get(&utxo_enc.owner).unwrap();
|
|
|
|
let (ciphertext, nonce) = Account::encrypt_data(
|
|
&ephm_key_holder,
|
|
accout_enc.key_holder.viewing_public_key,
|
|
&serde_json::to_vec(&utxo_enc).unwrap(),
|
|
);
|
|
|
|
let tag = accout_enc.make_tag();
|
|
|
|
(ciphertext, nonce.to_vec(), tag)
|
|
})
|
|
.collect();
|
|
|
|
encoded_data.extend(encoded_data_1);
|
|
|
|
let mut commitments = generate_commitments(&resulting_utxos_receiver);
|
|
let commitments_1 = generate_commitments(&resulting_utxos_not_spent);
|
|
|
|
commitments.extend(commitments_1);
|
|
|
|
let send_multiple_utxo_addr_bytes: Vec<u8> =
|
|
zkvm::test_methods::SEND_UTXO_MULTIPLE_ASSETS_ID
|
|
.iter()
|
|
.map(|num| num.to_le_bytes())
|
|
.flatten()
|
|
.collect();
|
|
let sc_addr = hex::encode(send_multiple_utxo_addr_bytes);
|
|
|
|
//Sc does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let sc_state = acc_map_read_guard
|
|
.block_store
|
|
.get_sc_sc_state(&sc_addr)
|
|
.map_err(ExecutionFailureKind::db_error)?;
|
|
|
|
let mut vec_values_u64: Vec<Vec<u64>> = sc_state
|
|
.into_iter()
|
|
.map(|slice| vec_u8_to_vec_u64(slice.to_vec()))
|
|
.collect();
|
|
|
|
let context = acc_map_read_guard.produce_context(account.address);
|
|
|
|
//Will not panic, as PublicScContext is serializable
|
|
let context_public_info: Vec<u64> = context.produce_u64_list_from_context().unwrap();
|
|
vec_values_u64.push(context_public_info);
|
|
|
|
let vec_public_info: Vec<u64> = vec_values_u64.into_iter().flatten().collect();
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(vec_public_info);
|
|
|
|
Ok((
|
|
TransactionPayload {
|
|
tx_kind: TxKind::Private,
|
|
execution_input: vec![],
|
|
execution_output: vec![],
|
|
utxo_commitments_spent_hashes: commitments_in,
|
|
utxo_commitments_created_hashes: commitments
|
|
.into_iter()
|
|
.map(|hash_data| hash_data.try_into().unwrap())
|
|
.collect(),
|
|
nullifier_created_hashes: nullifiers,
|
|
execution_proof_private: sc_core::transaction_payloads_tools::encode_receipt(
|
|
receipt,
|
|
)
|
|
.unwrap(),
|
|
encoded_data,
|
|
ephemeral_pub_key: eph_pub_key.to_vec(),
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
}
|
|
.into(),
|
|
utxo_hashes_receiver,
|
|
utxo_hashes_not_spent,
|
|
))
|
|
}
|
|
|
|
pub async fn transfer_balance_shielded(
|
|
&self,
|
|
acc: AccountAddress,
|
|
balance: u64,
|
|
receivers: Vec<(u128, AccountAddress)>,
|
|
) -> Result<(Transaction, Vec<(AccountAddress, [u8; 32])>), ExecutionFailureKind> {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let account = acc_map_read_guard.acc_map.get(&acc).unwrap();
|
|
|
|
// TODO: add to transaction structure and do the check. Research has to update the scheme as well.
|
|
let commitment = sc_core::transaction_payloads_tools::generate_secret_random_commitment(
|
|
balance, account,
|
|
)
|
|
.unwrap();
|
|
|
|
let nullifier = generate_nullifiers_se(
|
|
&commitment,
|
|
&account
|
|
.key_holder
|
|
.utxo_secret_key_holder
|
|
.nullifier_secret_key
|
|
.to_bytes()
|
|
.to_vec(),
|
|
);
|
|
|
|
let (resulting_utxos, receipt) = prove_send_utxo_shielded(acc, balance as u128, receivers)?;
|
|
let utxo_hashes = resulting_utxos
|
|
.iter()
|
|
.map(|(utxo, addr)| (addr.clone(), utxo.hash))
|
|
.collect();
|
|
|
|
let utxos: Vec<UTXO> = resulting_utxos
|
|
.iter()
|
|
.map(|(utxo, _)| utxo.clone())
|
|
.collect();
|
|
|
|
let ephm_key_holder = EphemeralKeyHolder::new_os_random();
|
|
ephm_key_holder.log();
|
|
|
|
let eph_pub_key =
|
|
serde_json::to_vec(&ephm_key_holder.generate_ephemeral_public_key()).unwrap();
|
|
|
|
let encoded_data: Vec<(Vec<u8>, Vec<u8>, u8)> = utxos
|
|
.iter()
|
|
.map(|utxo_enc| {
|
|
let accout_enc = acc_map_read_guard.acc_map.get(&utxo_enc.owner).unwrap();
|
|
|
|
let (ciphertext, nonce) = Account::encrypt_data(
|
|
&ephm_key_holder,
|
|
accout_enc.key_holder.viewing_public_key,
|
|
&serde_json::to_vec(&utxo_enc).unwrap(),
|
|
);
|
|
|
|
let tag = accout_enc.make_tag();
|
|
|
|
(ciphertext, nonce.to_vec(), tag)
|
|
})
|
|
.collect();
|
|
|
|
let commitments = generate_commitments(&utxos);
|
|
|
|
let mint_utxo_addr_bytes: Vec<u8> = zkvm::test_methods::SEND_UTXO_ID
|
|
.iter()
|
|
.map(|num| num.to_le_bytes())
|
|
.flatten()
|
|
.collect();
|
|
let sc_addr = hex::encode(mint_utxo_addr_bytes);
|
|
|
|
//Sc does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let sc_state = acc_map_read_guard
|
|
.block_store
|
|
.get_sc_sc_state(&sc_addr)
|
|
.map_err(ExecutionFailureKind::db_error)?;
|
|
|
|
let mut vec_values_u64: Vec<Vec<u64>> = sc_state
|
|
.into_iter()
|
|
.map(|slice| vec_u8_to_vec_u64(slice.to_vec()))
|
|
.collect();
|
|
|
|
let context = acc_map_read_guard.produce_context(account.address);
|
|
|
|
//Will not panic, as PublicScContext is serializable
|
|
let context_public_info: Vec<u64> = context.produce_u64_list_from_context().unwrap();
|
|
vec_values_u64.push(context_public_info);
|
|
|
|
let vec_public_info: Vec<u64> = vec_values_u64.into_iter().flatten().collect();
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(vec_public_info);
|
|
|
|
Ok((
|
|
TransactionPayload {
|
|
tx_kind: TxKind::Shielded,
|
|
execution_input: serde_json::to_vec(&ActionData::SendMoneyShieldedTx(
|
|
SendMoneyShieldedTx {
|
|
acc_sender: acc,
|
|
amount: balance as u128,
|
|
},
|
|
))
|
|
.unwrap(),
|
|
execution_output: vec![],
|
|
utxo_commitments_spent_hashes: vec![],
|
|
utxo_commitments_created_hashes: commitments
|
|
.into_iter()
|
|
.map(|hash_data| hash_data.try_into().unwrap())
|
|
.collect(),
|
|
nullifier_created_hashes: vec![nullifier.try_into().unwrap()],
|
|
execution_proof_private: sc_core::transaction_payloads_tools::encode_receipt(
|
|
receipt,
|
|
)
|
|
.unwrap(),
|
|
encoded_data,
|
|
ephemeral_pub_key: eph_pub_key.to_vec(),
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
}
|
|
.into(),
|
|
utxo_hashes,
|
|
))
|
|
}
|
|
|
|
pub async fn transfer_utxo_deshielded(
|
|
&self,
|
|
utxo: UTXO,
|
|
comm_gen_hash: [u8; 32],
|
|
receivers: Vec<(u128, AccountAddress)>,
|
|
) -> Result<Transaction, ExecutionFailureKind> {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let commitment_in = acc_map_read_guard
|
|
.utxo_commitments_store
|
|
.get_tx(comm_gen_hash)
|
|
.unwrap()
|
|
.hash;
|
|
|
|
let account = acc_map_read_guard.acc_map.get(&utxo.owner).unwrap();
|
|
|
|
let nullifier = generate_nullifiers(
|
|
&utxo,
|
|
&account
|
|
.key_holder
|
|
.utxo_secret_key_holder
|
|
.nullifier_secret_key
|
|
.to_bytes()
|
|
.to_vec(),
|
|
);
|
|
|
|
let (resulting_balances, receipt) = prove_send_utxo_deshielded(utxo, receivers)?;
|
|
|
|
let send_utxo_addr_bytes: Vec<u8> = zkvm::test_methods::SEND_UTXO_ID
|
|
.iter()
|
|
.map(|num| num.to_le_bytes())
|
|
.flatten()
|
|
.collect();
|
|
let sc_addr = hex::encode(send_utxo_addr_bytes);
|
|
|
|
//Sc does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let sc_state = acc_map_read_guard
|
|
.block_store
|
|
.get_sc_sc_state(&sc_addr)
|
|
.map_err(ExecutionFailureKind::db_error)?;
|
|
|
|
let mut vec_values_u64: Vec<Vec<u64>> = sc_state
|
|
.into_iter()
|
|
.map(|slice| vec_u8_to_vec_u64(slice.to_vec()))
|
|
.collect();
|
|
|
|
let context = acc_map_read_guard.produce_context(account.address);
|
|
|
|
//Will not panic, as PublicScContext is serializable
|
|
let context_public_info: Vec<u64> = context.produce_u64_list_from_context().unwrap();
|
|
vec_values_u64.push(context_public_info);
|
|
|
|
let vec_public_info: Vec<u64> = vec_values_u64.into_iter().flatten().collect();
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(vec_public_info);
|
|
|
|
Ok(TransactionPayload {
|
|
tx_kind: TxKind::Deshielded,
|
|
execution_input: serde_json::to_vec(&ActionData::SendMoneyDeshieldedTx(
|
|
SendMoneyDeshieldedTx {
|
|
receiver_data: resulting_balances,
|
|
},
|
|
))
|
|
.unwrap(),
|
|
execution_output: vec![],
|
|
utxo_commitments_spent_hashes: vec![commitment_in],
|
|
utxo_commitments_created_hashes: vec![],
|
|
nullifier_created_hashes: vec![nullifier.try_into().unwrap()],
|
|
execution_proof_private: sc_core::transaction_payloads_tools::encode_receipt(receipt)
|
|
.unwrap(),
|
|
encoded_data: vec![],
|
|
ephemeral_pub_key: vec![],
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
}
|
|
.into())
|
|
}
|
|
|
|
pub async fn send_private_mint_tx(
|
|
&self,
|
|
acc: AccountAddress,
|
|
amount: u128,
|
|
) -> Result<(SendTxResponse, [u8; 32], [u8; 32]), ExecutionFailureKind> {
|
|
//Considering proof time, needs to be done before proof
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let point_before_prove = std::time::Instant::now();
|
|
let (tx, utxo_hash) = self.mint_utxo_private(acc, amount).await?;
|
|
tx.log();
|
|
let point_after_prove = std::time::Instant::now();
|
|
|
|
let commitment_generated_hash = tx.utxo_commitments_created_hashes[0];
|
|
|
|
let timedelta = (point_after_prove - point_before_prove).as_millis();
|
|
info!("Mint utxo proof spent {timedelta:?} milliseconds");
|
|
|
|
Ok((
|
|
self.sequencer_client.send_tx(tx, tx_roots).await?,
|
|
utxo_hash,
|
|
commitment_generated_hash,
|
|
))
|
|
}
|
|
|
|
pub async fn send_private_mint_multiple_assets_tx(
|
|
&self,
|
|
acc: AccountAddress,
|
|
amount: u128,
|
|
number_of_assets: usize,
|
|
) -> Result<(SendTxResponse, Vec<[u8; 32]>, Vec<[u8; 32]>), ExecutionFailureKind> {
|
|
//Considering proof time, needs to be done before proof
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let point_before_prove = std::time::Instant::now();
|
|
let (tx, utxo_hashes) = self
|
|
.mint_utxo_multiple_assets_private(acc, amount, number_of_assets)
|
|
.await?;
|
|
tx.log();
|
|
let point_after_prove = std::time::Instant::now();
|
|
|
|
let commitment_generated_hashes = tx.utxo_commitments_created_hashes.clone();
|
|
|
|
let timedelta = (point_after_prove - point_before_prove).as_millis();
|
|
info!("Mint utxo proof spent {timedelta:?} milliseconds");
|
|
|
|
Ok((
|
|
self.sequencer_client.send_tx(tx, tx_roots).await?,
|
|
utxo_hashes,
|
|
commitment_generated_hashes,
|
|
))
|
|
}
|
|
|
|
pub async fn send_public_deposit(
|
|
&self,
|
|
acc: AccountAddress,
|
|
amount: u128,
|
|
) -> Result<SendTxResponse, ExecutionFailureKind> {
|
|
//Considering proof time, needs to be done before proof
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let public_context = {
|
|
let read_guard = self.storage.read().await;
|
|
|
|
read_guard.produce_context(acc)
|
|
};
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(
|
|
//Will not panic, as public context is serializable
|
|
public_context.produce_u64_list_from_context().unwrap(),
|
|
);
|
|
|
|
let sc_addr = hex::encode([0; 32]);
|
|
|
|
//Sc does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let tx: Transaction =
|
|
sc_core::transaction_payloads_tools::create_public_transaction_payload(
|
|
serde_json::to_vec(&ActionData::MintMoneyPublicTx(MintMoneyPublicTx {
|
|
acc,
|
|
amount,
|
|
}))
|
|
.unwrap(),
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
)
|
|
.into();
|
|
tx.log();
|
|
|
|
Ok(self.sequencer_client.send_tx(tx, tx_roots).await?)
|
|
}
|
|
|
|
pub async fn send_public_native_token_transfer(
|
|
&self,
|
|
from: AccountAddress,
|
|
to: AccountAddress,
|
|
moved_balance: u64,
|
|
) -> Result<SendTxResponse, ExecutionFailureKind> {
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let public_context = {
|
|
let read_guard = self.storage.read().await;
|
|
|
|
read_guard.produce_context(from)
|
|
};
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(
|
|
//Will not panic, as public context is serializable
|
|
public_context.produce_u64_list_from_context().unwrap(),
|
|
);
|
|
|
|
let sc_addr = hex::encode([0; 32]);
|
|
|
|
//Native does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let tx: Transaction =
|
|
sc_core::transaction_payloads_tools::create_public_transaction_payload(
|
|
serde_json::to_vec(&PublicNativeTokenSend {
|
|
from,
|
|
to,
|
|
moved_balance,
|
|
})
|
|
.unwrap(),
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
)
|
|
.into();
|
|
tx.log();
|
|
|
|
Ok(self.sequencer_client.send_tx(tx, tx_roots).await?)
|
|
}
|
|
|
|
pub async fn send_private_send_tx(
|
|
&self,
|
|
utxo: UTXO,
|
|
comm_hash: [u8; 32],
|
|
receivers: Vec<(u128, AccountAddress)>,
|
|
) -> Result<(SendTxResponse, Vec<([u8; 32], [u8; 32])>), ExecutionFailureKind> {
|
|
//Considering proof time, needs to be done before proof
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let point_before_prove = std::time::Instant::now();
|
|
let (tx, utxo_hashes) = self
|
|
.transfer_utxo_private(utxo, comm_hash, receivers)
|
|
.await?;
|
|
tx.log();
|
|
let point_after_prove = std::time::Instant::now();
|
|
|
|
let timedelta = (point_after_prove - point_before_prove).as_millis();
|
|
info!("Send private utxo proof spent {timedelta:?} milliseconds");
|
|
|
|
Ok((
|
|
self.sequencer_client.send_tx(tx, tx_roots).await?,
|
|
utxo_hashes,
|
|
))
|
|
}
|
|
|
|
pub async fn send_private_multiple_assets_send_tx(
|
|
&self,
|
|
utxos: Vec<UTXO>,
|
|
comm_hashes: Vec<[u8; 32]>,
|
|
number_to_send: usize,
|
|
receiver: AccountAddress,
|
|
) -> Result<(SendTxResponse, Vec<[u8; 32]>, Vec<[u8; 32]>), ExecutionFailureKind> {
|
|
//Considering proof time, needs to be done before proof
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let point_before_prove = std::time::Instant::now();
|
|
let (tx, utxo_hashes_received, utxo_hashes_not_spent) = self
|
|
.transfer_utxo_multiple_assets_private(utxos, comm_hashes, number_to_send, receiver)
|
|
.await?;
|
|
tx.log();
|
|
let point_after_prove = std::time::Instant::now();
|
|
|
|
let timedelta = (point_after_prove - point_before_prove).as_millis();
|
|
info!("Send private utxo proof spent {timedelta:?} milliseconds");
|
|
|
|
Ok((
|
|
self.sequencer_client.send_tx(tx, tx_roots).await?,
|
|
utxo_hashes_received,
|
|
utxo_hashes_not_spent,
|
|
))
|
|
}
|
|
|
|
pub async fn send_shielded_send_tx(
|
|
&self,
|
|
acc: AccountAddress,
|
|
amount: u64,
|
|
receivers: Vec<(u128, AccountAddress)>,
|
|
) -> Result<(SendTxResponse, Vec<([u8; 32], [u8; 32])>), ExecutionFailureKind> {
|
|
//Considering proof time, needs to be done before proof
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let point_before_prove = std::time::Instant::now();
|
|
let (tx, utxo_hashes) = self
|
|
.transfer_balance_shielded(acc, amount, receivers)
|
|
.await?;
|
|
tx.log();
|
|
let point_after_prove = std::time::Instant::now();
|
|
|
|
let timedelta = (point_after_prove - point_before_prove).as_millis();
|
|
info!("Send balance shielded proof spent {timedelta:?} milliseconds");
|
|
|
|
Ok((
|
|
self.sequencer_client.send_tx(tx, tx_roots).await?,
|
|
utxo_hashes,
|
|
))
|
|
}
|
|
|
|
pub async fn send_deshielded_send_tx(
|
|
&self,
|
|
utxo: UTXO,
|
|
comm_gen_hash: [u8; 32],
|
|
receivers: Vec<(u128, AccountAddress)>,
|
|
) -> Result<SendTxResponse, ExecutionFailureKind> {
|
|
//Considering proof time, needs to be done before proof
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let point_before_prove = std::time::Instant::now();
|
|
let tx = self
|
|
.transfer_utxo_deshielded(utxo, comm_gen_hash, receivers)
|
|
.await?;
|
|
tx.log();
|
|
let point_after_prove = std::time::Instant::now();
|
|
|
|
let timedelta = (point_after_prove - point_before_prove).as_millis();
|
|
info!("Send deshielded utxo proof spent {timedelta:?} milliseconds");
|
|
|
|
Ok(self.sequencer_client.send_tx(tx, tx_roots).await?)
|
|
}
|
|
|
|
pub async fn operate_account_mint_private(
|
|
&mut self,
|
|
acc_addr: AccountAddress,
|
|
amount: u128,
|
|
) -> Result<(UTXO, [u8; 32]), ExecutionFailureKind> {
|
|
let (resp, new_utxo_hash, comm_gen_hash) =
|
|
self.send_private_mint_tx(acc_addr, amount).await?;
|
|
info!("Response for mint private is {resp:?}");
|
|
|
|
info!("Awaiting new blocks");
|
|
tokio::time::sleep(std::time::Duration::from_secs(BLOCK_GEN_DELAY_SECS)).await;
|
|
|
|
let new_utxo = {
|
|
let mut write_guard = self.storage.write().await;
|
|
|
|
let acc = write_guard.acc_map.get_mut(&acc_addr).unwrap();
|
|
|
|
acc.utxos.get(&new_utxo_hash).unwrap().clone()
|
|
};
|
|
|
|
new_utxo.log();
|
|
info!(
|
|
"Account address is {:?} ,new utxo owner address is {:?}",
|
|
hex::encode(acc_addr),
|
|
hex::encode(new_utxo.owner)
|
|
);
|
|
info!(
|
|
"Account {:?} got new utxo with amount {amount:?}",
|
|
hex::encode(acc_addr)
|
|
);
|
|
|
|
Ok((new_utxo, comm_gen_hash))
|
|
}
|
|
|
|
pub async fn operate_account_mint_multiple_assets_private(
|
|
&mut self,
|
|
acc_addr: AccountAddress,
|
|
amount: u128,
|
|
number_of_assets: usize,
|
|
) -> Result<(Vec<UTXO>, Vec<[u8; 32]>), ExecutionFailureKind> {
|
|
let (resp, new_utxo_hashes, comm_gen_hashes) = self
|
|
.send_private_mint_multiple_assets_tx(acc_addr, amount, number_of_assets)
|
|
.await?;
|
|
info!("Response for mint multiple assets private is {resp:?}");
|
|
|
|
info!("Awaiting new blocks");
|
|
tokio::time::sleep(std::time::Duration::from_secs(BLOCK_GEN_DELAY_SECS)).await;
|
|
|
|
let new_utxos = {
|
|
let mut write_guard = self.storage.write().await;
|
|
|
|
new_utxo_hashes
|
|
.into_iter()
|
|
.map(|new_utxo_hash| {
|
|
let acc = write_guard.acc_map.get_mut(&acc_addr).unwrap();
|
|
|
|
let new_utxo = acc.utxos.get(&new_utxo_hash).unwrap().clone();
|
|
|
|
new_utxo.log();
|
|
info!(
|
|
"Account address is {:?} ,new utxo owner address is {:?}",
|
|
hex::encode(acc_addr),
|
|
hex::encode(new_utxo.owner)
|
|
);
|
|
info!(
|
|
"Account {:?} got new utxo with amount {amount:?} and asset {:?}",
|
|
hex::encode(acc_addr),
|
|
new_utxo.asset
|
|
);
|
|
|
|
new_utxo
|
|
})
|
|
.collect()
|
|
};
|
|
|
|
Ok((new_utxos, comm_gen_hashes))
|
|
}
|
|
|
|
pub async fn operate_account_send_deshielded_one_receiver(
|
|
&mut self,
|
|
acc_addr_rec: AccountAddress,
|
|
utxo: UTXO,
|
|
comm_gen_hash: [u8; 32],
|
|
) -> Result<(), ExecutionFailureKind> {
|
|
let amount = utxo.amount;
|
|
|
|
let old_balance = {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let acc = acc_map_read_guard.acc_map.get(&acc_addr_rec).unwrap();
|
|
|
|
acc.balance
|
|
};
|
|
|
|
info!(
|
|
"Balance of receiver {:?} now is {old_balance:?}",
|
|
hex::encode(acc_addr_rec)
|
|
);
|
|
|
|
let resp = self
|
|
.send_deshielded_send_tx(utxo, comm_gen_hash, vec![(amount, acc_addr_rec)])
|
|
.await?;
|
|
info!("Response for send deshielded is {resp:?}");
|
|
|
|
info!("Awaiting new blocks");
|
|
tokio::time::sleep(std::time::Duration::from_secs(BLOCK_GEN_DELAY_SECS)).await;
|
|
|
|
let new_balance = {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let acc = acc_map_read_guard.acc_map.get(&acc_addr_rec).unwrap();
|
|
|
|
acc.balance
|
|
};
|
|
|
|
info!(
|
|
"Balance of receiver {:?} now is {:?}, delta is {:?}",
|
|
hex::encode(acc_addr_rec),
|
|
new_balance,
|
|
new_balance - old_balance
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn operate_account_deposit_public(
|
|
&mut self,
|
|
acc_addr: AccountAddress,
|
|
amount: u128,
|
|
) -> Result<(), ExecutionFailureKind> {
|
|
let old_balance = {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let acc = acc_map_read_guard.acc_map.get(&acc_addr).unwrap();
|
|
|
|
acc.balance
|
|
};
|
|
|
|
info!(
|
|
"Balance of {:?} now is {old_balance:?}",
|
|
hex::encode(acc_addr)
|
|
);
|
|
|
|
let resp = self.send_public_deposit(acc_addr, amount).await?;
|
|
info!("Response for public deposit is {resp:?}");
|
|
|
|
info!("Awaiting new blocks");
|
|
tokio::time::sleep(std::time::Duration::from_secs(BLOCK_GEN_DELAY_SECS)).await;
|
|
|
|
let new_balance = {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let acc = acc_map_read_guard.acc_map.get(&acc_addr).unwrap();
|
|
|
|
acc.balance
|
|
};
|
|
|
|
info!(
|
|
"Balance of {:?} now is {new_balance:?}, delta is {:?}",
|
|
hex::encode(acc_addr),
|
|
new_balance - old_balance
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn operate_account_send_shielded_one_receiver(
|
|
&mut self,
|
|
acc_addr_sender: AccountAddress,
|
|
acc_addr_rec: AccountAddress,
|
|
amount: u128,
|
|
) -> Result<UTXO, ExecutionFailureKind> {
|
|
let (resp, new_utxo_hashes) = self
|
|
.send_shielded_send_tx(acc_addr_sender, amount as u64, vec![(amount, acc_addr_rec)])
|
|
.await?;
|
|
info!("Response for send shielded is {resp:?}");
|
|
|
|
let new_utxo_hash = new_utxo_hashes[0].1;
|
|
|
|
info!("Awaiting new blocks");
|
|
tokio::time::sleep(std::time::Duration::from_secs(BLOCK_GEN_DELAY_SECS)).await;
|
|
|
|
let new_utxo = {
|
|
let mut write_guard = self.storage.write().await;
|
|
|
|
let acc = write_guard.acc_map.get_mut(&acc_addr_rec).unwrap();
|
|
acc.log();
|
|
|
|
acc.utxos.get(&new_utxo_hash).unwrap().clone()
|
|
};
|
|
new_utxo.log();
|
|
info!(
|
|
"Account address is {:?} ,new utxo owner address is {:?}",
|
|
hex::encode(acc_addr_rec),
|
|
hex::encode(new_utxo.owner)
|
|
);
|
|
info!(
|
|
"Account {:?} got new utxo with amount {amount:?}",
|
|
hex::encode(acc_addr_rec)
|
|
);
|
|
|
|
Ok(new_utxo)
|
|
}
|
|
|
|
pub async fn operate_account_send_private_one_receiver(
|
|
&mut self,
|
|
acc_addr_rec: AccountAddress,
|
|
utxo: UTXO,
|
|
comm_gen_hash: [u8; 32],
|
|
) -> Result<UTXO, ExecutionFailureKind> {
|
|
let amount = utxo.amount;
|
|
|
|
let (resp, new_utxo_hashes) = self
|
|
.send_private_send_tx(utxo, comm_gen_hash, vec![(amount, acc_addr_rec)])
|
|
.await?;
|
|
info!("Response for send private is {resp:?}");
|
|
|
|
let new_utxo_hash = new_utxo_hashes[0].1;
|
|
|
|
info!("Awaiting new blocks");
|
|
tokio::time::sleep(std::time::Duration::from_secs(BLOCK_GEN_DELAY_SECS)).await;
|
|
|
|
let new_utxo = {
|
|
let mut write_guard = self.storage.write().await;
|
|
|
|
let acc = write_guard.acc_map.get_mut(&acc_addr_rec).unwrap();
|
|
acc.log();
|
|
|
|
acc.utxos.get(&new_utxo_hash).unwrap().clone()
|
|
};
|
|
new_utxo.log();
|
|
info!(
|
|
"Account address is {:?} ,new utxo owner address is {:?}",
|
|
hex::encode(acc_addr_rec),
|
|
hex::encode(new_utxo.owner)
|
|
);
|
|
info!(
|
|
"Account {:?} got new utxo with amount {:?}",
|
|
hex::encode(acc_addr_rec),
|
|
new_utxo.amount
|
|
);
|
|
|
|
Ok(new_utxo)
|
|
}
|
|
|
|
pub async fn operate_account_send_private_multiple_assets_one_receiver(
|
|
&mut self,
|
|
acc_addr: AccountAddress,
|
|
acc_addr_rec: AccountAddress,
|
|
utxos: Vec<UTXO>,
|
|
comm_gen_hashes: Vec<[u8; 32]>,
|
|
number_to_send: usize,
|
|
) -> Result<(), ExecutionFailureKind> {
|
|
let (resp, new_utxo_hashes_rec, new_utxo_hashes_not_sp) = self
|
|
.send_private_multiple_assets_send_tx(
|
|
utxos,
|
|
comm_gen_hashes,
|
|
number_to_send,
|
|
acc_addr_rec,
|
|
)
|
|
.await?;
|
|
info!("Response for send private multiple assets is {resp:?}");
|
|
|
|
info!("Awaiting new blocks");
|
|
tokio::time::sleep(std::time::Duration::from_secs(BLOCK_GEN_DELAY_SECS)).await;
|
|
|
|
{
|
|
let mut write_guard = self.storage.write().await;
|
|
|
|
for new_utxo_hash in new_utxo_hashes_rec {
|
|
let acc = write_guard.acc_map.get_mut(&acc_addr_rec).unwrap();
|
|
acc.log();
|
|
|
|
let new_utxo = acc.utxos.get(&new_utxo_hash).unwrap().clone();
|
|
|
|
new_utxo.log();
|
|
info!(
|
|
"Account address is {:?} ,new utxo owner address is {:?}",
|
|
hex::encode(acc_addr_rec),
|
|
hex::encode(new_utxo.owner)
|
|
);
|
|
info!(
|
|
"Account {:?} got new utxo with amount {:?} and asset {:?}",
|
|
hex::encode(acc_addr_rec),
|
|
new_utxo.amount,
|
|
new_utxo.asset,
|
|
);
|
|
}
|
|
|
|
for new_utxo_hash in new_utxo_hashes_not_sp {
|
|
let acc = write_guard.acc_map.get_mut(&acc_addr).unwrap();
|
|
acc.log();
|
|
|
|
let new_utxo = acc.utxos.get(&new_utxo_hash).unwrap().clone();
|
|
|
|
new_utxo.log();
|
|
info!(
|
|
"Account address is {:?} ,new utxo owner address is {:?}",
|
|
hex::encode(acc_addr),
|
|
hex::encode(new_utxo.owner)
|
|
);
|
|
info!(
|
|
"Account {:?} got new utxo with amount {:?} and asset {:?}",
|
|
hex::encode(acc_addr),
|
|
new_utxo.amount,
|
|
new_utxo.asset,
|
|
);
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn split_utxo(
|
|
&self,
|
|
utxo: UTXO,
|
|
commitment_in: [u8; 32],
|
|
receivers: Vec<(u128, AccountAddress)>,
|
|
visibility_list: [bool; 3],
|
|
) -> Result<(Transaction, Vec<(AccountAddress, [u8; 32])>), ExecutionFailureKind> {
|
|
let acc_map_read_guard = self.storage.read().await;
|
|
|
|
let account = acc_map_read_guard.acc_map.get(&utxo.owner).unwrap();
|
|
|
|
let nullifier = generate_nullifiers(
|
|
&utxo,
|
|
&account
|
|
.key_holder
|
|
.utxo_secret_key_holder
|
|
.nullifier_secret_key
|
|
.to_bytes()
|
|
.to_vec(),
|
|
);
|
|
|
|
let (resulting_utxos, receipt) = prove_send_utxo(utxo, receivers)?;
|
|
let utxo_hashes = resulting_utxos
|
|
.iter()
|
|
.map(|(utxo, addr)| (addr.clone(), utxo.hash))
|
|
.collect();
|
|
|
|
let utxos: Vec<UTXO> = resulting_utxos
|
|
.iter()
|
|
.map(|(utxo, _)| utxo.clone())
|
|
.collect();
|
|
|
|
let ephm_key_holder = EphemeralKeyHolder::new_os_random();
|
|
ephm_key_holder.log();
|
|
|
|
let eph_pub_key =
|
|
serde_json::to_vec(&ephm_key_holder.generate_ephemeral_public_key()).unwrap();
|
|
|
|
let encoded_data: Vec<(Vec<u8>, Vec<u8>, u8)> = utxos
|
|
.iter()
|
|
.map(|utxo_enc| {
|
|
let accout_enc = acc_map_read_guard.acc_map.get(&utxo_enc.owner).unwrap();
|
|
|
|
let (ciphertext, nonce) = Account::encrypt_data(
|
|
&ephm_key_holder,
|
|
accout_enc.key_holder.viewing_public_key,
|
|
&serde_json::to_vec(&utxo_enc).unwrap(),
|
|
);
|
|
|
|
let tag = accout_enc.make_tag();
|
|
|
|
(ciphertext, nonce.to_vec(), tag)
|
|
})
|
|
.collect();
|
|
|
|
let commitments = generate_commitments(&utxos);
|
|
|
|
let publication = ActionData::UTXOPublication(UTXOPublication {
|
|
utxos: utxos
|
|
.iter()
|
|
.enumerate()
|
|
.filter_map(|(id, item)| {
|
|
if visibility_list[id] {
|
|
Some(item.clone())
|
|
} else {
|
|
None
|
|
}
|
|
})
|
|
.collect(),
|
|
});
|
|
|
|
let send_utxo_addr_bytes: Vec<u8> = zkvm::test_methods::SEND_UTXO_ID
|
|
.iter()
|
|
.map(|num| num.to_le_bytes())
|
|
.flatten()
|
|
.collect();
|
|
let sc_addr = hex::encode(send_utxo_addr_bytes);
|
|
|
|
//Sc does not change its state
|
|
let state_changes: Vec<DataBlobChangeVariant> = vec![];
|
|
let new_len = 0;
|
|
let state_changes = (serde_json::to_value(state_changes).unwrap(), new_len);
|
|
|
|
let sc_state = acc_map_read_guard
|
|
.block_store
|
|
.get_sc_sc_state(&sc_addr)
|
|
.map_err(ExecutionFailureKind::db_error)?;
|
|
|
|
let mut vec_values_u64: Vec<Vec<u64>> = sc_state
|
|
.into_iter()
|
|
.map(|slice| vec_u8_to_vec_u64(slice.to_vec()))
|
|
.collect();
|
|
|
|
let context = acc_map_read_guard.produce_context(account.address);
|
|
|
|
//Will not panic, as PublicScContext is serializable
|
|
let context_public_info: Vec<u64> = context.produce_u64_list_from_context().unwrap();
|
|
vec_values_u64.push(context_public_info);
|
|
|
|
let vec_public_info: Vec<u64> = vec_values_u64.into_iter().flatten().collect();
|
|
|
|
let (tweak, secret_r, commitment) = pedersen_commitment_vec(vec_public_info);
|
|
|
|
Ok((
|
|
TransactionPayload {
|
|
tx_kind: TxKind::Shielded,
|
|
execution_input: vec![],
|
|
execution_output: serde_json::to_vec(&publication).unwrap(),
|
|
utxo_commitments_spent_hashes: vec![commitment_in],
|
|
utxo_commitments_created_hashes: commitments
|
|
.clone()
|
|
.into_iter()
|
|
.map(|hash_data| hash_data.try_into().unwrap())
|
|
.collect(),
|
|
nullifier_created_hashes: vec![nullifier.try_into().unwrap()],
|
|
execution_proof_private: sc_core::transaction_payloads_tools::encode_receipt(
|
|
receipt,
|
|
)
|
|
.unwrap(),
|
|
encoded_data,
|
|
ephemeral_pub_key: eph_pub_key.to_vec(),
|
|
commitment,
|
|
tweak,
|
|
secret_r,
|
|
sc_addr,
|
|
state_changes,
|
|
}
|
|
.into(),
|
|
utxo_hashes,
|
|
))
|
|
}
|
|
|
|
pub async fn send_split_tx(
|
|
&self,
|
|
utxo: UTXO,
|
|
comm_hash: [u8; 32],
|
|
receivers: Vec<(u128, AccountAddress)>,
|
|
visibility_list: [bool; 3],
|
|
) -> Result<(SendTxResponse, Vec<([u8; 32], [u8; 32])>, Vec<[u8; 32]>), ExecutionFailureKind>
|
|
{
|
|
//Considering proof time, needs to be done before proof
|
|
let tx_roots = self.get_roots().await;
|
|
|
|
let point_before_prove = std::time::Instant::now();
|
|
let (tx, utxo_hashes) = self
|
|
.split_utxo(utxo, comm_hash, receivers, visibility_list)
|
|
.await?;
|
|
tx.log();
|
|
let point_after_prove = std::time::Instant::now();
|
|
|
|
let timedelta = (point_after_prove - point_before_prove).as_millis();
|
|
info!("Send private utxo proof spent {timedelta:?} milliseconds");
|
|
|
|
let commitments = tx.utxo_commitments_created_hashes.clone();
|
|
|
|
Ok((
|
|
self.sequencer_client.send_tx(tx, tx_roots).await?,
|
|
utxo_hashes,
|
|
commitments,
|
|
))
|
|
}
|
|
|
|
pub async fn operate_account_send_split_utxo(
|
|
&mut self,
|
|
addrs_receivers: [AccountAddress; 3],
|
|
utxo: UTXO,
|
|
comm_gen_hash: [u8; 32],
|
|
visibility_list: [bool; 3],
|
|
) -> Result<(Vec<UTXO>, Vec<[u8; 32]>), ExecutionFailureKind> {
|
|
let (resp, new_utxo_hashes, commitments_hashes) = self
|
|
.send_split_tx(
|
|
utxo.clone(),
|
|
comm_gen_hash,
|
|
addrs_receivers
|
|
.clone()
|
|
.map(|addr| (utxo.amount / 3, addr))
|
|
.to_vec(),
|
|
visibility_list,
|
|
)
|
|
.await?;
|
|
info!("Response for send shielded is {resp:?}");
|
|
|
|
info!("Awaiting new blocks");
|
|
tokio::time::sleep(std::time::Duration::from_secs(BLOCK_GEN_DELAY_SECS)).await;
|
|
|
|
let new_utxos: Vec<UTXO> = {
|
|
let mut write_guard = self.storage.write().await;
|
|
|
|
new_utxo_hashes
|
|
.into_iter()
|
|
.map(|(acc_addr_rec, new_utxo_hash)| {
|
|
let acc = write_guard.acc_map.get_mut(&acc_addr_rec).unwrap();
|
|
|
|
let new_utxo = acc.utxos.get(&new_utxo_hash).unwrap().clone();
|
|
new_utxo.log();
|
|
|
|
info!(
|
|
"Account address is {:?} ,new utxo owner address is {:?}",
|
|
hex::encode(acc_addr_rec),
|
|
hex::encode(new_utxo.owner)
|
|
);
|
|
info!(
|
|
"Account {:?} got new utxo with amount {:?}",
|
|
hex::encode(acc_addr_rec),
|
|
new_utxo.amount
|
|
);
|
|
|
|
new_utxo
|
|
})
|
|
.collect()
|
|
};
|
|
|
|
Ok((new_utxos, commitments_hashes))
|
|
}
|
|
|
|
///Mint utxo, make it public
|
|
pub async fn subscenario_1(&mut self) -> Result<(), ExecutionFailureKind> {
|
|
let acc_addr = self.create_new_account().await;
|
|
|
|
let (new_utxo, comm_gen_hash) = self.operate_account_mint_private(acc_addr, 100).await?;
|
|
|
|
self.operate_account_send_deshielded_one_receiver(acc_addr, new_utxo, comm_gen_hash)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
///Deposit balance, make it private
|
|
pub async fn subscenario_2(&mut self) -> Result<(), ExecutionFailureKind> {
|
|
let acc_addr = self.create_new_account().await;
|
|
|
|
self.operate_account_deposit_public(acc_addr, 100).await?;
|
|
|
|
self.operate_account_send_shielded_one_receiver(acc_addr, acc_addr, 100)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
///Mint utxo, privately send it to another user
|
|
pub async fn subscenario_3(&mut self) -> Result<(), ExecutionFailureKind> {
|
|
let acc_addr = self.create_new_account().await;
|
|
let acc_addr_rec = self.create_new_account().await;
|
|
|
|
let (new_utxo, comm_gen_hash) = self.operate_account_mint_private(acc_addr, 100).await?;
|
|
|
|
self.operate_account_send_private_one_receiver(acc_addr_rec, new_utxo, comm_gen_hash)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
///Deposit balance, shielded send it to another user
|
|
pub async fn subscenario_4(&mut self) -> Result<(), ExecutionFailureKind> {
|
|
let acc_addr = self.create_new_account().await;
|
|
let acc_addr_rec = self.create_new_account().await;
|
|
|
|
self.operate_account_deposit_public(acc_addr, 100).await?;
|
|
|
|
self.operate_account_send_shielded_one_receiver(acc_addr, acc_addr_rec, 100)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
///Mint utxo, deshielded send it to another user
|
|
pub async fn subscenario_5(&mut self) -> Result<(), ExecutionFailureKind> {
|
|
let acc_addr = self.create_new_account().await;
|
|
let acc_addr_rec = self.create_new_account().await;
|
|
|
|
let (new_utxo, comm_gen_hash) = self.operate_account_mint_private(acc_addr, 100).await?;
|
|
|
|
self.operate_account_send_deshielded_one_receiver(acc_addr_rec, new_utxo, comm_gen_hash)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
///First complex scenario.
|
|
/// Creating accounts A, B, C, D.
|
|
/// Minting UTXO for A, splitting it between B, C, D.
|
|
/// Variable `visibility_list` decides, which of actions will be visible on blockchain.
|
|
/// Variable `publication index` decides, who of B, C or D moves its UTXO into public state.
|
|
pub async fn scenario_1(
|
|
&mut self,
|
|
visibility_list: [bool; 3],
|
|
publication_index: usize,
|
|
) -> Result<(), ExecutionFailureKind> {
|
|
let acc_addr_sender = self.create_new_account().await;
|
|
|
|
let acc_addr_rec_1 = self.create_new_account().await;
|
|
let acc_addr_rec_2 = self.create_new_account().await;
|
|
let acc_addr_rec_3 = self.create_new_account().await;
|
|
|
|
let addrs_receivers = [acc_addr_rec_1, acc_addr_rec_2, acc_addr_rec_3];
|
|
|
|
let (new_utxo, comm_gen_hash) = self
|
|
.operate_account_mint_private(acc_addr_sender, 99)
|
|
.await?;
|
|
|
|
let (new_utxos, comm_gen_hashes) = self
|
|
.operate_account_send_split_utxo(
|
|
addrs_receivers,
|
|
new_utxo,
|
|
comm_gen_hash,
|
|
visibility_list,
|
|
)
|
|
.await?;
|
|
|
|
self.operate_account_send_deshielded_one_receiver(
|
|
addrs_receivers[publication_index],
|
|
new_utxos[publication_index].clone(),
|
|
comm_gen_hashes[publication_index],
|
|
)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
///Mint number of different assets with same amount for account
|
|
pub async fn scenario_2(
|
|
&mut self,
|
|
number_of_assets: usize,
|
|
number_to_send: usize,
|
|
) -> Result<(), ExecutionFailureKind> {
|
|
let acc_addr_sender = self.create_new_account().await;
|
|
let acc_addr_receiver = self.create_new_account().await;
|
|
|
|
let (utxos, comm_gen_hashes) = self
|
|
.operate_account_mint_multiple_assets_private(acc_addr_sender, 100, number_of_assets)
|
|
.await?;
|
|
|
|
self.operate_account_send_private_multiple_assets_one_receiver(
|
|
acc_addr_sender,
|
|
acc_addr_receiver,
|
|
utxos,
|
|
comm_gen_hashes,
|
|
number_to_send,
|
|
)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
pub fn generate_commitments_helper(input_utxos: &[UTXO]) -> Vec<[u8; 32]> {
|
|
generate_commitments(input_utxos)
|
|
.into_iter()
|
|
.map(|comm_raw| comm_raw.try_into().unwrap())
|
|
.collect()
|
|
}
|