poanetwork
/
hbbft
mirror of https://github.com/poanetwork/hbbft.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Assert matching keys in NetworkInfo.

This commit is contained in:
Andreas Fackler 2019-08-23 10:29:31 +02:00 committed by Andreas Fackler
parent d15467eb09
commit c146634686
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/network_info.rs
View File

@ -112,7 +112,8 @@ impl<N: NodeIdT> NetworkInfo<N> {
///
/// # Panics
///
/// Panics if `public_keys` is empty.
/// Panics if `public_keys` is empty or the requirement above is not satisfied, i.e. if the
/// secret key doesn't match the public key or is missing.
pub fn new<SKS: Into<Option<SecretKeyShare>>>(
our_id: N,
secret_key_share: SKS,
@ -122,15 +123,22 @@ impl<N: NodeIdT> NetworkInfo<N> {
) -> Self {
let val_set = Arc::new(ValidatorSet::from(public_keys.keys()));
let is_validator = val_set.contains(&our_id);
let public_key_shares = public_keys
.keys()
.enumerate()
.map(|(idx, id)| (id.clone(), public_key_set.public_key_share(idx)))
let secret_key_share = secret_key_share.into();
assert_eq!(is_validator, secret_key_share.is_some());
let public_key_shares: BTreeMap<N, PublicKeyShare> = val_set
.all_indices()
.map(|(id, idx)| (id.clone(), public_key_set.public_key_share(idx)))
.collect();
if let Some(sks) = &secret_key_share {
assert_eq!(
Some(&sks.public_key_share()),
public_key_shares.get(&our_id)
);
}
NetworkInfo {
our_id,
is_validator,
secret_key_share: secret_key_share.into(),
secret_key_share,
secret_key,
public_key_set,
public_key_shares,