absolutebi/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-12-09 04:39:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Transaction::sign.

Browse Source 
While I don't love the introduction of empty_signed, it's practically fine.
This commit is contained in:
Luke Parker
2023-04-23 01:25:45 -04:00
parent 3f6565588f
commit 710e6e5217
6 changed files with 67 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
coordinator/src/tests/tributary/dkg.rs
View File

@@ -1,11 +1,9 @@
use core::time::Duration;
use zeroize::Zeroizing;
use rand_core::{RngCore, OsRng};
use ciphersuite::{group::ff::Field, Ciphersuite, Ristretto};
use schnorr::SchnorrSignature;
use ciphersuite::{Ciphersuite, Ristretto};
use frost::Participant;
use tokio::time::sleep;
@@ -17,7 +15,7 @@ use processor_messages::{
CoordinatorMessage,
};
use tributary::{Signed, Transaction as TransactionTrait, Tributary};
use tributary::Tributary;
use crate::{
processor::MemProcessor,
@@ -39,41 +37,14 @@ async fn dkg_commitments_test() {
let mut txs = vec![];
// Create DKG commitments for each key
for key in &keys {
let pub_key = Ristretto::generator() * **key;
let attempt = 0;
let mut commitments = vec![0; 256];
OsRng.fill_bytes(&mut commitments);
// Create the TX with a null signature so we can get its sig hash
let tx = Transaction::DkgCommitments(
attempt,
commitments.clone(),
Signed {
signer: pub_key,
nonce: 0,
signature: SchnorrSignature::<Ristretto> {
R: Ristretto::generator(),
s: <Ristretto as Ciphersuite>::F::ZERO,
},
},
);
// Re-create it with the actual signature
// We could mutate the existing one, we'd just have to match to the DkgCommitments enum variant
txs.push(Transaction::DkgCommitments(
attempt,
commitments,
Signed {
signer: pub_key,
nonce: 0,
signature: SchnorrSignature::<Ristretto>::sign(
key,
Zeroizing::new(<Ristretto as Ciphersuite>::F::random(&mut OsRng)),
tx.sig_hash(spec.genesis()),
),
},
));
let mut tx =
Transaction::DkgCommitments(attempt, commitments.clone(), Transaction::empty_signed());
tx.sign(&mut OsRng, spec.genesis(), key, 0);
txs.push(tx);
}
let mut last_block = tributaries[0].1.tip();

39
coordinator/src/tests/tributary/tx.rs
View File

@@ -1,17 +1,12 @@
use core::time::Duration;
use zeroize::Zeroizing;
use rand_core::{RngCore, OsRng};
use ciphersuite::{group::ff::Field, Ciphersuite, Ristretto};
use schnorr::SchnorrSignature;
use tokio::time::sleep;
use serai_db::MemDb;
use tributary::{Signed, Transaction as TransactionTrait, Tributary};
use tributary::Tributary;
use crate::{
LocalP2p,
@@ -33,41 +28,15 @@ async fn tx_test() {
let sender =
usize::try_from(OsRng.next_u64() % u64::try_from(tributaries.len()).unwrap()).unwrap();
let key = keys[sender].clone();
let pub_key = Ristretto::generator() * *key;
let attempt = 0;
let mut commitments = vec![0; 256];
OsRng.fill_bytes(&mut commitments);
// Create the TX with a null signature so we can get its sig hash
let tx = Transaction::DkgCommitments(
attempt,
commitments.clone(),
Signed {
signer: pub_key,
nonce: 0,
signature: SchnorrSignature::<Ristretto> {
R: Ristretto::generator(),
s: <Ristretto as Ciphersuite>::F::ZERO,
},
},
);
// Re-create it with the actual signature
// We could mutate the existing one, we'd just have to match to the DkgCommitments enum variant
let tx = Transaction::DkgCommitments(
attempt,
commitments,
Signed {
signer: pub_key,
nonce: 0,
signature: SchnorrSignature::<Ristretto>::sign(
&key,
Zeroizing::new(<Ristretto as Ciphersuite>::F::random(&mut OsRng)),
tx.sig_hash(spec.genesis()),
),
},
);
let mut tx =
Transaction::DkgCommitments(attempt, commitments.clone(), Transaction::empty_signed());
tx.sign(&mut OsRng, spec.genesis(), &key, 0);
assert!(tributaries[sender].1.add_transaction(tx.clone()).await);
// Sleep for two blocks