rusk: support stake multisig

dusk-network · Sep 9, 2024 · 69764d4 · 69764d4
1 parent 21cdbc7
commit 69764d4
Show file tree

Hide file tree

Showing 12 changed files with 60 additions and 18 deletions.
diff --git a/rusk/src/lib/http/rusk.rs b/rusk/src/lib/http/rusk.rs
@@ -137,7 +137,7 @@ impl Rusk {
             .provisioners(None)
             .expect("Cannot query state for provisioners")
             .map(|(key, stake)| {
-                let key = bs58::encode(key.to_bytes()).into_string();
+                let key = bs58::encode(key.account.to_bytes()).into_string();
                 let amount = stake.amount.unwrap_or_default();
 
                 Provisioner {

diff --git a/rusk/src/lib/node/rusk.rs b/rusk/src/lib/node/rusk.rs
@@ -9,6 +9,7 @@ use std::sync::{mpsc, Arc, LazyLock};
 use std::time::{Duration, Instant};
 use std::{fs, io};
 
+use execution_core::stake::StakeKeys;
 use execution_core::transfer::PANIC_NONCE_NOT_READY;
 use parking_lot::RwLock;
 use sha3::{Digest, Sha3_256};
@@ -396,12 +397,12 @@ impl Rusk {
     pub fn provisioners(
         &self,
         base_commit: Option<[u8; 32]>,
-    ) -> Result<impl Iterator<Item = (BlsPublicKey, StakeData)>> {
+    ) -> Result<impl Iterator<Item = (StakeKeys, StakeData)>> {
         let (sender, receiver) = mpsc::channel();
         self.feeder_query(STAKE_CONTRACT, "stakes", &(), sender, base_commit)?;
         Ok(receiver.into_iter().map(|bytes| {
-            rkyv::from_bytes::<(BlsPublicKey, StakeData)>(&bytes).expect(
-                "The contract should only return (pk, stake_data) tuples",
+            rkyv::from_bytes::<(StakeKeys, StakeData)>(&bytes).expect(
+                "The contract should only return (StakeKeys, StakeData) tuples",
             )
         }))
     }

diff --git a/rusk/src/lib/node/vm.rs b/rusk/src/lib/node/vm.rs
@@ -17,6 +17,7 @@ use execution_core::{
     transfer::Transaction as ProtocolTransaction, Event,
 };
 use node::vm::VMExecution;
+use node_data::bls::PublicKey;
 use node_data::ledger::{Block, Slash, SpentTransaction, Transaction};
 
 use super::Rusk;
@@ -189,7 +190,7 @@ impl VMExecution for Rusk {
     fn get_changed_provisioners(
         &self,
         base_commit: [u8; 32],
-    ) -> anyhow::Result<Vec<(node_data::bls::PublicKey, Option<Stake>)>> {
+    ) -> anyhow::Result<Vec<(PublicKey, Option<Stake>)>> {
         self.query_provisioners_change(Some(base_commit))
     }
 
@@ -243,7 +244,7 @@ impl Rusk {
             .provisioners(base_commit)
             .map_err(|e| anyhow::anyhow!("Cannot get provisioners {e}"))?
             .map(|(pk, stake)| {
-                (node_data::bls::PublicKey::new(pk), Self::to_stake(stake))
+                (PublicKey::new(pk.account), Self::to_stake(stake))
             });
         let mut ret = Provisioners::empty();
         for (pubkey_bls, stake) in provisioners {
@@ -256,20 +257,15 @@ impl Rusk {
     fn query_provisioners_change(
         &self,
         base_commit: Option<[u8; 32]>,
-    ) -> anyhow::Result<Vec<(node_data::bls::PublicKey, Option<Stake>)>> {
+    ) -> anyhow::Result<Vec<(PublicKey, Option<Stake>)>> {
         info!("Received get_provisioners_change request");
         Ok(self
             .last_provisioners_change(base_commit)
             .map_err(|e| {
                 anyhow::anyhow!("Cannot get provisioners change: {e}")
             })?
             .into_iter()
-            .map(|(pk, stake)| {
-                (
-                    node_data::bls::PublicKey::new(pk),
-                    stake.map(Self::to_stake),
-                )
-            })
+            .map(|(pk, stake)| (PublicKey::new(pk), stake.map(Self::to_stake)))
             .collect())
     }
 

diff --git a/rusk/tests/common/keys.rs b/rusk/tests/common/keys.rs
@@ -10,12 +10,21 @@ use rand::prelude::*;
 use rand::rngs::StdRng;
 use tracing::info;
 
-use execution_core::signatures::bls::SecretKey as BlsSecretKey;
+use dusk_bytes::Serializable;
+use execution_core::signatures::bls::{
+    PublicKey as BlsPublicKey, SecretKey as BlsSecretKey,
+};
 
 #[allow(dead_code)]
 pub static STAKE_SK: LazyLock<BlsSecretKey> = LazyLock::new(|| {
     info!("Generating BlsSecretKey");
     let mut rng = StdRng::seed_from_u64(0xdead);
 
-    BlsSecretKey::random(&mut rng)
+    let sk = BlsSecretKey::random(&mut rng);
+    let pk = BlsPublicKey::from(&sk);
+    info!(
+        "Generated BlsSecretKey for BlsPublicKey {}",
+        bs58::encode(pk.to_bytes()).into_string()
+    );
+    sk
 });
diff --git a/rusk/tests/config/bench.toml b/rusk/tests/config/bench.toml
@@ -1,6 +1,7 @@
-[acl.stake]
-owners = []
-allowlist = []
+## Block generator address
+[[stake]]
+address = "owFNnhnMuPHhyQcZ1dAqC5wuScqnXGkrEZWRQvYBU9wwNFHmG32gRTFdZhYSgw98KzNqm7rpCBkwoL4nzCE6gqGB9BrjsDJx3UgfTk78CM8BBWsehXamPSPYcC9ynq8Pi3V"
+amount = 0
 
 [[phoenix_balance]]
 address = "ivmscertKgRyX8wNMJJsQcSVEyPsfSMUQXSAgeAPQXsndqFq9Pmknzhm61QvcEEdxPaGgxDS4RHpb6KKccrnSKN"

diff --git a/rusk/tests/config/contract_deployment.toml b/rusk/tests/config/contract_deployment.toml
@@ -1,3 +1,8 @@
+## Block generator address
+[[stake]]
+address = "owFNnhnMuPHhyQcZ1dAqC5wuScqnXGkrEZWRQvYBU9wwNFHmG32gRTFdZhYSgw98KzNqm7rpCBkwoL4nzCE6gqGB9BrjsDJx3UgfTk78CM8BBWsehXamPSPYcC9ynq8Pi3V"
+amount = 0
+
 [[phoenix_balance]]
 address = "ivmscertKgRyX8wNMJJsQcSVEyPsfSMUQXSAgeAPQXsndqFq9Pmknzhm61QvcEEdxPaGgxDS4RHpb6KKccrnSKN"
 seed = 57005

diff --git a/rusk/tests/config/gas-behavior.toml b/rusk/tests/config/gas-behavior.toml
@@ -1,3 +1,8 @@
+## Block generator address
+[[stake]]
+address = "owFNnhnMuPHhyQcZ1dAqC5wuScqnXGkrEZWRQvYBU9wwNFHmG32gRTFdZhYSgw98KzNqm7rpCBkwoL4nzCE6gqGB9BrjsDJx3UgfTk78CM8BBWsehXamPSPYcC9ynq8Pi3V"
+amount = 0
+
 [[phoenix_balance]]
 address = "ivmscertKgRyX8wNMJJsQcSVEyPsfSMUQXSAgeAPQXsndqFq9Pmknzhm61QvcEEdxPaGgxDS4RHpb6KKccrnSKN"
 seed = 57005

diff --git a/rusk/tests/config/multi_transfer.toml b/rusk/tests/config/multi_transfer.toml
@@ -1,3 +1,8 @@
+## Block generator address
+[[stake]]
+address = "owFNnhnMuPHhyQcZ1dAqC5wuScqnXGkrEZWRQvYBU9wwNFHmG32gRTFdZhYSgw98KzNqm7rpCBkwoL4nzCE6gqGB9BrjsDJx3UgfTk78CM8BBWsehXamPSPYcC9ynq8Pi3V"
+amount = 0
+
 [[phoenix_balance]]
 address = "ivmscertKgRyX8wNMJJsQcSVEyPsfSMUQXSAgeAPQXsndqFq9Pmknzhm61QvcEEdxPaGgxDS4RHpb6KKccrnSKN"
 seed = 57005

diff --git a/rusk/tests/config/slash.toml b/rusk/tests/config/slash.toml
@@ -1,3 +1,8 @@
+## Block generator address
+[[stake]]
+address = "owFNnhnMuPHhyQcZ1dAqC5wuScqnXGkrEZWRQvYBU9wwNFHmG32gRTFdZhYSgw98KzNqm7rpCBkwoL4nzCE6gqGB9BrjsDJx3UgfTk78CM8BBWsehXamPSPYcC9ynq8Pi3V"
+amount = 0
+
 [[phoenix_balance]]
 address = "ivmscertKgRyX8wNMJJsQcSVEyPsfSMUQXSAgeAPQXsndqFq9Pmknzhm61QvcEEdxPaGgxDS4RHpb6KKccrnSKN"
 seed = 57005

diff --git a/rusk/tests/config/stake.toml b/rusk/tests/config/stake.toml
@@ -1,3 +1,8 @@
+## Block generator address
+[[stake]]
+address = "owFNnhnMuPHhyQcZ1dAqC5wuScqnXGkrEZWRQvYBU9wwNFHmG32gRTFdZhYSgw98KzNqm7rpCBkwoL4nzCE6gqGB9BrjsDJx3UgfTk78CM8BBWsehXamPSPYcC9ynq8Pi3V"
+amount = 0
+
 [[phoenix_balance]]
 address = "ivmscertKgRyX8wNMJJsQcSVEyPsfSMUQXSAgeAPQXsndqFq9Pmknzhm61QvcEEdxPaGgxDS4RHpb6KKccrnSKN"
 seed = 57005

diff --git a/rusk/tests/config/transfer.toml b/rusk/tests/config/transfer.toml
@@ -1,3 +1,8 @@
+## Block generator address
+[[stake]]
+address = "owFNnhnMuPHhyQcZ1dAqC5wuScqnXGkrEZWRQvYBU9wwNFHmG32gRTFdZhYSgw98KzNqm7rpCBkwoL4nzCE6gqGB9BrjsDJx3UgfTk78CM8BBWsehXamPSPYcC9ynq8Pi3V"
+amount = 0
+
 [[phoenix_balance]]
 address = "ivmscertKgRyX8wNMJJsQcSVEyPsfSMUQXSAgeAPQXsndqFq9Pmknzhm61QvcEEdxPaGgxDS4RHpb6KKccrnSKN"
 seed = 57005

diff --git a/rusk/tests/config/unspendable.toml b/rusk/tests/config/unspendable.toml
@@ -1,3 +1,8 @@
+## Block generator address
+[[stake]]
+address = "owFNnhnMuPHhyQcZ1dAqC5wuScqnXGkrEZWRQvYBU9wwNFHmG32gRTFdZhYSgw98KzNqm7rpCBkwoL4nzCE6gqGB9BrjsDJx3UgfTk78CM8BBWsehXamPSPYcC9ynq8Pi3V"
+amount = 0
+
 [[phoenix_balance]]
 address = "ivmscertKgRyX8wNMJJsQcSVEyPsfSMUQXSAgeAPQXsndqFq9Pmknzhm61QvcEEdxPaGgxDS4RHpb6KKccrnSKN"
 seed = 57005