Clean up code, correct a few bugs, add leader based one-time-key/BP gen

This commit is contained in:
Luke Parker
2022-04-28 20:09:31 -04:00
parent c4b7cb71d7
commit 1d0a0c7c16
6 changed files with 253 additions and 114 deletions

View File

@@ -1,8 +1,10 @@
use monero::{consensus::deserialize, util::ringct::Bulletproof};
use curve25519_dalek::{scalar::Scalar, edwards::EdwardsPoint};
use crate::{Commitment, transaction::TransactionError, free, c_gen_bp};
use monero::{consensus::{Encodable, deserialize}, util::ringct::Bulletproof};
pub fn generate(outputs: Vec<Commitment>) -> Result<Bulletproof, TransactionError> {
use crate::{Commitment, transaction::TransactionError, free, c_generate_bp, c_verify_bp};
pub fn generate(outputs: &[Commitment]) -> Result<Bulletproof, TransactionError> {
if outputs.len() > 16 {
return Err(TransactionError::TooManyOutputs)?;
}
@@ -11,7 +13,7 @@ pub fn generate(outputs: Vec<Commitment>) -> Result<Bulletproof, TransactionErro
let amounts: Vec<u64> = outputs.iter().map(|commitment| commitment.amount).collect();
let res;
unsafe {
let ptr = c_gen_bp(outputs.len() as u8, amounts.as_ptr(), masks.as_ptr());
let ptr = c_generate_bp(outputs.len() as u8, amounts.as_ptr(), masks.as_ptr());
let len = ((ptr.read() as usize) << 8) + (ptr.add(1).read() as usize);
res = deserialize(
std::slice::from_raw_parts(ptr.add(2), len)
@@ -21,3 +23,18 @@ pub fn generate(outputs: Vec<Commitment>) -> Result<Bulletproof, TransactionErro
Ok(res)
}
pub fn verify(bp: &Bulletproof, commitments: &[EdwardsPoint]) -> bool {
if commitments.len() > 16 {
return false;
}
let mut serialized = vec![];
bp.consensus_encode(&mut serialized).unwrap();
let commitments: Vec<[u8; 32]> = commitments.iter().map(
|commitment| (commitment * Scalar::from(8 as u8).invert()).compress().to_bytes()
).collect();
unsafe {
c_verify_bp(serialized.len(), serialized.as_ptr(), commitments.len() as u8, commitments.as_ptr())
}
}