Actually attempting to register dealer keys

This commit is contained in:
Jędrzej Stuczyński
2022-05-13 17:29:21 +01:00
parent 106e179f6c
commit 1165a61ca6
12 changed files with 157 additions and 15 deletions
@@ -1378,7 +1378,7 @@ impl<C> NymdClient<C> {
// }
}
fn cosmwasm_coin_to_cosmos_coin(coin: Coin) -> CosmosCoin {
pub(crate) fn cosmwasm_coin_to_cosmos_coin(coin: Coin) -> CosmosCoin {
CosmosCoin {
denom: coin.denom.parse().unwrap(),
// this might be a bit iffy, cosmwasm coin stores value as u128, while cosmos does it as u64
@@ -3,11 +3,14 @@
use crate::nymd::cosmwasm_client::types::ExecuteResult;
use crate::nymd::error::NymdError;
use crate::nymd::{Fee, NymdClient, SigningCosmWasmClient};
use crate::nymd::{cosmwasm_coin_to_cosmos_coin, Fee, NymdClient, SigningCosmWasmClient};
use async_trait::async_trait;
use coconut_dkg_common::msg::ExecuteMsg as DkgExecuteMsg;
use coconut_dkg_common::msg::QueryMsg as DkgQueryMsg;
use coconut_dkg_common::types::{EncodedBTEPublicKeyWithProof, EncodedEd25519PublicKey, Epoch, BlacklistingResponse, PagedBlacklistingResponse, PagedDealerResponse};
use coconut_dkg_common::types::{
BlacklistingResponse, EncodedBTEPublicKeyWithProof, EncodedEd25519PublicKey, Epoch,
MinimumDepositResponse, PagedBlacklistingResponse, PagedDealerResponse,
};
#[async_trait]
pub trait DkgClient {
@@ -30,6 +33,7 @@ pub trait DkgClient {
) -> Result<PagedBlacklistingResponse, NymdError>;
async fn get_blacklisting(&self, dealer: String) -> Result<BlacklistingResponse, NymdError>;
async fn get_deposit_amount(&self) -> Result<MinimumDepositResponse, NymdError>;
async fn register_dealer(
&self,
@@ -107,6 +111,13 @@ where
.await
}
async fn get_deposit_amount(&self) -> Result<MinimumDepositResponse, NymdError> {
let request = DkgQueryMsg::GetDepositAmount {};
self.client
.query_contract_smart(self.coconut_dkg_contract_address()?, &request)
.await
}
async fn register_dealer(
&self,
identity: EncodedEd25519PublicKey,
@@ -121,6 +132,7 @@ where
owner_signature,
host: listening_address,
};
let deposit = self.get_deposit_amount().await?;
self.client
.execute(
@@ -129,7 +141,7 @@ where
&req,
fee.unwrap_or_default(),
format!("registering {} as a dealer", self.address()),
Vec::new(),
vec![cosmwasm_coin_to_cosmos_coin(deposit.amount)],
)
.await
}
@@ -53,6 +53,7 @@ pub enum QueryMsg {
GetBlacklisting {
dealer: String,
},
GetDepositAmount {},
}
#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, JsonSchema)]
@@ -7,11 +7,13 @@ pub use crate::dealer::{
BlacklistedDealer, Blacklisting, BlacklistingReason, BlacklistingResponse, DealerDetails,
PagedBlacklistingResponse, PagedDealerResponse,
};
pub use cosmwasm_std::Addr;
pub use cosmwasm_std::{Addr, Coin};
pub type BlockHeight = u64;
pub type EncodedEd25519PublicKey = String;
pub type EncodedEd25519PublicKeyRef<'a> = &'a str;
pub type EncodedBTEPublicKeyWithProof = String;
pub type EncodedBTEPublicKeyWithProofRef<'a> = &'a str;
pub type NodeIndex = u64;
pub type EpochId = u32;
@@ -120,3 +122,15 @@ pub enum EpochState {
finish_by: Option<BlockHeight>,
},
}
#[derive(Clone, Debug, Deserialize, PartialEq, Serialize)]
#[serde(rename_all = "snake_case")]
pub struct MinimumDepositResponse {
pub amount: Coin,
}
impl MinimumDepositResponse {
pub fn new(amount: Coin) -> Self {
MinimumDepositResponse { amount }
}
}
@@ -6,7 +6,6 @@ use crate::dealers::storage::{IndexedDealersMap, BLACKLISTED_DEALERS};
use coconut_dkg_common::dealer::{
BlacklistedDealer, BlacklistingResponse, PagedBlacklistingResponse, PagedDealerResponse,
};
use coconut_dkg_common::types::Blacklisting;
use cosmwasm_std::{Deps, Order, StdResult};
use cw_storage_plus::Bound;
+8 -2
View File
@@ -1,6 +1,7 @@
// Copyright 2022 - Nym Technologies SA <contact@nymtech.net>
// SPDX-License-Identifier: Apache-2.0
use crate::constants::MINIMUM_DEPOSIT;
use crate::dealers::queries::{
query_blacklisted_dealers_paged, query_blacklisting, query_current_dealers_paged,
query_past_dealers_paged,
@@ -8,9 +9,10 @@ use crate::dealers::queries::{
use crate::epoch::queries::query_current_epoch;
use crate::error::ContractError;
use coconut_dkg_common::msg::{ExecuteMsg, InstantiateMsg, MigrateMsg, QueryMsg};
use coconut_dkg_common::types::{Epoch, EpochState};
use coconut_dkg_common::types::{Epoch, EpochState, MinimumDepositResponse};
use config::defaults::STAKE_DENOM;
use cosmwasm_std::{
entry_point, to_binary, Deps, DepsMut, Env, MessageInfo, QueryResponse, Response,
entry_point, to_binary, Coin, Deps, DepsMut, Env, MessageInfo, QueryResponse, Response,
};
use epoch::storage as epoch_storage;
@@ -139,6 +141,10 @@ pub fn query(deps: Deps<'_>, _env: Env, msg: QueryMsg) -> Result<QueryResponse,
to_binary(&query_blacklisted_dealers_paged(deps, start_after, limit)?)?
}
QueryMsg::GetBlacklisting { dealer } => to_binary(&query_blacklisting(deps, dealer)?)?,
QueryMsg::GetDepositAmount {} => to_binary(&MinimumDepositResponse::new(Coin::new(
MINIMUM_DEPOSIT.u128(),
STAKE_DENOM,
)))?,
};
Ok(response)
+31 -2
View File
@@ -10,6 +10,8 @@ use coconut_dkg_common::types::{Addr, BlockHeight, DealerDetails};
use futures::channel::mpsc;
use futures::StreamExt;
use log::{debug, info, trace};
use std::net::SocketAddr;
use validator_client::nymd::SigningCosmWasmClient;
// essentially events originating from the contract watcher could only drive our state forward
// (TODO: is it actually true? I guess we'll find out soon enough)
@@ -26,20 +28,28 @@ pub(crate) struct ProcessingLoop<C> {
contract_publisher: Publisher<C>,
// network_sender: Sender
// TODO: this HAS to go away from here as it doesn't fit. but I've put it here temporarily to validate dealer registration
network_host: SocketAddr,
}
impl<C> ProcessingLoop<C> {
impl<C> ProcessingLoop<C>
where
C: SigningCosmWasmClient + Send + Sync,
{
pub(crate) fn new(
dkg_state: DkgState,
dispatcher_sender: DispatcherSender,
contract_events_receiver: ContractEventsReceiver,
contract_publisher: Publisher<C>,
network_host: SocketAddr,
) -> Self {
ProcessingLoop {
dkg_state,
dispatcher_sender,
contract_events_receiver,
contract_publisher,
network_host,
}
}
@@ -106,7 +116,26 @@ impl<C> ProcessingLoop<C> {
async fn process_new_key_submission(&self, height: BlockHeight) {
debug!("attempting to register our own dealer keys for this round of dkg");
info!(".... but that's not implemented yet....");
let chain_address = self.contract_publisher.get_address().await;
let registration = self
.dkg_state
.prepare_dealer_registration(chain_address, self.network_host.to_string());
if let Err(err) = self
.contract_publisher
.register_dealer(
registration.identity,
registration.bte_key,
registration.owner_signature,
registration.listening_address,
)
.await
{
error!("failed to register our dealer - {}", err)
} else {
info!("registered our dealer for this DKG round")
}
}
async fn process_dealer_changes(&self, changes: Vec<DealerChange>, height: BlockHeight) {
+2 -1
View File
@@ -72,13 +72,14 @@ where
config.contract_polling_interval,
);
let publisher = smart_contract::Publisher::new(nyxd_client);
let mut net_listener = Listener::new(config.network_address, state_accessor);
let mut net_listener = Listener::new(config.network_address.clone(), state_accessor);
let mut processing_loop = ProcessingLoop::new(
state,
dispatcher_sender,
contracts_events_receiver,
publisher,
config.network_address,
);
tokio::spawn(async move { event_dispatcher.run().await });
@@ -1,8 +1,10 @@
// Copyright 2022 - Nym Technologies SA <contact@nymtech.net>
// SPDX-License-Identifier: Apache-2.0
use crate::dkg::error::DkgError;
use crate::Client;
use validator_client::nymd::SigningCosmWasmClient;
use coconut_dkg_common::types::{EncodedBTEPublicKeyWithProof, EncodedEd25519PublicKey};
use validator_client::nymd::{AccountId, SigningCosmWasmClient};
pub(crate) struct Publisher<C> {
client: Client<C>,
@@ -16,6 +18,23 @@ where
Publisher { client }
}
pub(crate) async fn get_address(&self) -> AccountId {
self.client.address().await
}
pub(crate) async fn register_dealer(
&self,
identity: EncodedEd25519PublicKey,
bte_key: EncodedBTEPublicKeyWithProof,
owner_signature: String,
listening_address: String,
) -> Result<(), DkgError> {
self.client
.register_dealer(identity, bte_key, owner_signature, listening_address)
.await?;
Ok(())
}
pub(crate) async fn submit_dealing_commitment(&self) {
// self.client.submit_dealing_commitment().await;
//
@@ -133,6 +133,9 @@ where
.await;
}
}
} else {
// TODO: change to trace
debug!("our dkg key is already registered in the dkg contract")
}
Ok(())
+34 -2
View File
@@ -2,7 +2,10 @@
// SPDX-License-Identifier: Apache-2.0
use crate::Client;
use coconut_dkg_common::types::{Addr, BlockHeight, DealerDetails, Epoch, NodeIndex};
use coconut_dkg_common::types::{
Addr, BlockHeight, DealerDetails, EncodedBTEPublicKeyWithProof, EncodedEd25519PublicKey, Epoch,
NodeIndex,
};
use crypto::asymmetric::identity;
use dkg::{bte, Dealing};
use futures::lock::Mutex;
@@ -17,7 +20,7 @@ mod accessor;
use crate::dkg::error::DkgError;
pub(crate) use accessor::StateAccessor;
use validator_client::nymd::SigningCosmWasmClient;
use validator_client::nymd::{AccountId, SigningCosmWasmClient};
type IdentityBytes = [u8; identity::PUBLIC_KEY_LENGTH];
@@ -99,6 +102,13 @@ pub struct ReceivedDealing {
signature: identity::Signature,
}
pub(crate) struct DealerRegistration {
pub(crate) identity: EncodedEd25519PublicKey,
pub(crate) bte_key: EncodedBTEPublicKeyWithProof,
pub(crate) owner_signature: String,
pub(crate) listening_address: String,
}
#[derive(Debug, Clone)]
pub(crate) struct DkgState {
inner_state: Arc<Mutex<DkgStateInner>>,
@@ -286,4 +296,26 @@ impl DkgState {
}
}
}
pub(crate) fn prepare_dealer_registration(
&self,
chain_address: AccountId,
listening_address: String,
) -> DealerRegistration {
let bte_key = bs58::encode(&self.keys.bte_public_key.to_bytes()).into_string();
// chain_address || host || bte_keys
let mut plaintext = chain_address.to_string();
plaintext.push_str(&listening_address);
plaintext.push_str(&bte_key);
let owner_signature = self.keys.identity.private_key().sign_text(&plaintext);
DealerRegistration {
identity: self.keys.identity.public_key().to_base58_string(),
bte_key,
owner_signature,
listening_address,
}
}
}
+27 -1
View File
@@ -6,7 +6,8 @@ use crate::rewarded_set_updater::error::RewardingError;
#[cfg(feature = "coconut")]
use async_trait::async_trait;
use coconut_dkg_common::types::{
BlacklistedDealer, BlacklistingResponse, DealerDetails, Epoch as DkgEpoch,
BlacklistedDealer, BlacklistingResponse, Coin, DealerDetails, EncodedBTEPublicKeyWithProof,
EncodedEd25519PublicKey, Epoch as DkgEpoch,
};
use config::defaults::{DEFAULT_NETWORK, DEFAULT_VALIDATOR_API_PORT};
use mixnet_contract_common::Interval;
@@ -479,6 +480,31 @@ where
self.0.read().await.nymd.get_blacklisting(address).await
}
pub(crate) async fn get_minimum_deposit(&self) -> Result<Coin, NymdError> {
self.0
.read()
.await
.nymd
.get_deposit_amount()
.await
.map(|res| res.amount)
}
pub(crate) async fn register_dealer(
&self,
identity: EncodedEd25519PublicKey,
bte_key: EncodedBTEPublicKeyWithProof,
owner_signature: String,
listening_address: String,
) -> Result<ExecuteResult, NymdError> {
self.0
.write()
.await
.nymd
.register_dealer(identity, bte_key, owner_signature, listening_address, None)
.await
}
pub(crate) async fn submit_dealing_commitment(
&self,
epoch_id: u32,