mirror of
https://github.com/serai-dex/serai.git
synced 2024-12-23 12:09:37 +00:00
91 lines
2.4 KiB
Rust
91 lines
2.4 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use serai_abi::primitives::NetworkId;
|
|
use zeroize::Zeroizing;
|
|
use rand_core::OsRng;
|
|
|
|
use sp_core::{
|
|
sr25519::{Pair, Signature},
|
|
Pair as PairTrait,
|
|
};
|
|
|
|
use ciphersuite::{Ciphersuite, Ristretto};
|
|
use frost::dkg::musig::musig;
|
|
use schnorrkel::Schnorrkel;
|
|
|
|
use serai_client::{
|
|
primitives::insecure_pair_from_name,
|
|
validator_sets::{
|
|
primitives::{ValidatorSet, KeyPair, musig_context, set_keys_message},
|
|
ValidatorSetsEvent,
|
|
},
|
|
Amount, Serai, SeraiValidatorSets,
|
|
};
|
|
|
|
use crate::common::tx::publish_tx;
|
|
|
|
#[allow(dead_code)]
|
|
pub async fn set_keys(serai: &Serai, set: ValidatorSet, key_pair: KeyPair) -> [u8; 32] {
|
|
let pair = insecure_pair_from_name("Alice");
|
|
let public = pair.public();
|
|
|
|
let public_key = <Ristretto as Ciphersuite>::read_G::<&[u8]>(&mut public.0.as_ref()).unwrap();
|
|
let secret_key = <Ristretto as Ciphersuite>::read_F::<&[u8]>(
|
|
&mut pair.as_ref().secret.to_bytes()[.. 32].as_ref(),
|
|
)
|
|
.unwrap();
|
|
assert_eq!(Ristretto::generator() * secret_key, public_key);
|
|
let threshold_keys =
|
|
musig::<Ristretto>(&musig_context(set), &Zeroizing::new(secret_key), &[public_key]).unwrap();
|
|
|
|
let sig = frost::tests::sign_without_caching(
|
|
&mut OsRng,
|
|
frost::tests::algorithm_machines(
|
|
&mut OsRng,
|
|
&Schnorrkel::new(b"substrate"),
|
|
&HashMap::from([(threshold_keys.params().i(), threshold_keys.into())]),
|
|
),
|
|
&set_keys_message(&set, &[], &key_pair),
|
|
);
|
|
|
|
// Set the key pair
|
|
let block = publish_tx(
|
|
serai,
|
|
&SeraiValidatorSets::set_keys(set.network, vec![], key_pair.clone(), Signature(sig.to_bytes())),
|
|
)
|
|
.await;
|
|
|
|
assert_eq!(
|
|
serai.as_of(block).validator_sets().key_gen_events().await.unwrap(),
|
|
vec![ValidatorSetsEvent::KeyGen { set, key_pair: key_pair.clone() }]
|
|
);
|
|
assert_eq!(serai.as_of(block).validator_sets().keys(set).await.unwrap(), Some(key_pair));
|
|
|
|
block
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub async fn allocate_stake(
|
|
serai: &Serai,
|
|
network: NetworkId,
|
|
amount: Amount,
|
|
pair: &Pair,
|
|
nonce: u32,
|
|
) -> [u8; 32] {
|
|
// get the call
|
|
let tx = serai.sign(pair, SeraiValidatorSets::allocate(network, amount), nonce, 0);
|
|
publish_tx(serai, &tx).await
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub async fn deallocate_stake(
|
|
serai: &Serai,
|
|
network: NetworkId,
|
|
amount: Amount,
|
|
pair: &Pair,
|
|
nonce: u32,
|
|
) -> [u8; 32] {
|
|
// get the call
|
|
let tx = serai.sign(pair, SeraiValidatorSets::deallocate(network, amount), nonce, 0);
|
|
publish_tx(serai, &tx).await
|
|
}
|