mirror of
https://github.com/serai-dex/serai.git
synced 2024-12-26 05:29:55 +00:00
98190b7b83
* initial staking pallet * add staking pallet to runtime * support session rotation for serai * optimizations & cleaning * fix deny * add serai network to initial networks * a few tweaks & comments * fix some pr comments * Rewrite validator-sets with logarithmic algorithms Uses the fact the underlying DB is sorted to achieve sorting of potential validators by stake. Removes release of deallocated stake for now. --------- Co-authored-by: Luke Parker <lukeparker5132@gmail.com>
80 lines
2.2 KiB
Rust
80 lines
2.2 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use zeroize::Zeroizing;
|
|
use rand_core::OsRng;
|
|
|
|
use sp_core::{Pair, sr25519::Signature};
|
|
|
|
use ciphersuite::{group::GroupEncoding, 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, musig_key, set_keys_message},
|
|
ValidatorSetsEvent,
|
|
},
|
|
Serai,
|
|
};
|
|
|
|
use crate::common::{serai, tx::publish_tx};
|
|
|
|
#[allow(dead_code)]
|
|
pub async fn set_validator_set_keys(set: ValidatorSet, key_pair: KeyPair) -> [u8; 32] {
|
|
let pair = insecure_pair_from_name("Alice");
|
|
let public = pair.public();
|
|
|
|
let serai = serai().await;
|
|
let public_key = <Ristretto as Ciphersuite>::read_G::<&[u8]>(&mut public.0.as_ref()).unwrap();
|
|
assert_eq!(
|
|
serai
|
|
.get_validator_set_musig_key(set, serai.get_latest_block_hash().await.unwrap())
|
|
.await
|
|
.unwrap()
|
|
.unwrap(),
|
|
musig_key(set, &[public]).0
|
|
);
|
|
|
|
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();
|
|
assert_eq!(
|
|
serai
|
|
.get_validator_set_musig_key(set, serai.get_latest_block_hash().await.unwrap())
|
|
.await
|
|
.unwrap()
|
|
.unwrap(),
|
|
threshold_keys.group_key().to_bytes()
|
|
);
|
|
|
|
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),
|
|
);
|
|
|
|
// Vote in a key pair
|
|
let block = publish_tx(&Serai::set_validator_set_keys(
|
|
set.network,
|
|
key_pair.clone(),
|
|
Signature(sig.to_bytes()),
|
|
))
|
|
.await;
|
|
|
|
assert_eq!(
|
|
serai.get_key_gen_events(block).await.unwrap(),
|
|
vec![ValidatorSetsEvent::KeyGen { set, key_pair: key_pair.clone() }]
|
|
);
|
|
assert_eq!(serai.get_keys(set, block).await.unwrap(), Some(key_pair));
|
|
|
|
block
|
|
}
|