pda

Author: billythedummy

Cargo.lock

@@ -23,5 +23,6 @@ serde_json = { version = "^1", default-features = false }
 # group them together to ensure same version throughout
 # and to deal with dependency hell more easily
 solana-account-decoder-client-types = { version = "^2", default-features = false }
+solana-pubkey = { version = "^2", default-features = false }
 
 # workspace members

Cargo.toml

@@ -12,3 +12,4 @@ bs58 = { workspace = true }
 serde = { workspace = true, features = ["derive"] }
 serde_json = { workspace = true }
 solana-account-decoder-client-types = { workspace = true }
+solana-pubkey = { workspace = true, features = ["curve25519"] }

core/Cargo.toml

@@ -2,9 +2,11 @@
 
 mod internal_utils;
 mod keys;
+mod pda;
 mod state;
 mod typedefs;
 
 pub use keys::*;
+pub use pda::*;
 pub use state::*;
 pub use typedefs::*;

core/src/lib.rs

@@ -0,0 +1,15 @@
+use crate::LIDO_STATE_ADDR;
+
+pub const VALIDATOR_STAKE_ACCOUNT_SEED: [u8; 23] = *b"validator_stake_account";
+
+pub type ValidatorStakeSeeds<'a> = (&'a [u8; 32], &'a [u8; 32], &'a [u8; 23], [u8; 8]);
+
+#[inline]
+pub const fn validator_stake_seeds(vote_acc_addr: &[u8; 32], seed: u64) -> ValidatorStakeSeeds<'_> {
+    (
+        &LIDO_STATE_ADDR,
+        vote_acc_addr,
+        &VALIDATOR_STAKE_ACCOUNT_SEED,
+        seed.to_le_bytes(),
+    )
+}

core/src/pda.rs

@@ -6,6 +6,8 @@ mod seed_range;
 
 pub use seed_range::*;
 
+use crate::{validator_stake_seeds, ValidatorStakeSeeds};
+
 #[repr(C)]
 #[derive(Debug, Default, Clone, Copy, PartialEq, Eq, Hash, BorshDeserialize, BorshSerialize)]
 pub struct Validator {
@@ -18,6 +20,16 @@ pub struct Validator {
     active: bool,
 }
 
+impl Validator {
+    /// Returns the seeds to find the PDA of the validator
+    /// stake account to service stsol withdrawals from
+    #[inline]
+    pub const fn withdraw_stake_acc_seeds(&self) -> ValidatorStakeSeeds<'_> {
+        validator_stake_seeds(&self.vote_account_address, self.stake_seeds.begin())
+    }
+}
+
+/// Getters and setters
 impl Validator {
     impl_set_with_get_ref!(
         set_vote_account_address,

core/src/typedefs/validator/mod.rs

@@ -1,5 +1,7 @@
 mod accounts;
 mod paths;
+mod pda;
 
 pub use accounts::*;
 pub use paths::*;
+pub use pda::*;

core/tests/common/mod.rs

@@ -0,0 +1,10 @@
+use solana_pubkey::Pubkey;
+use solido_legacy_core::{Validator, PROGRAM_ID};
+
+pub fn find_withdraw_stake_acc(validator: &Validator) -> (Pubkey, u8) {
+    let (s0, s1, s2, s3) = validator.withdraw_stake_acc_seeds();
+    Pubkey::find_program_address(
+        &[s0.as_slice(), s1.as_slice(), s2.as_slice(), s3.as_slice()],
+        &Pubkey::new_from_array(PROGRAM_ID),
+    )
+}

core/tests/common/pda.rs

@@ -1 +1,2 @@
+mod pda;
 mod state;

core/tests/tests/mod.rs

@@ -0,0 +1,27 @@
+use solana_pubkey::Pubkey;
+use solido_legacy_core::ValidatorList;
+
+use crate::common::{find_withdraw_stake_acc, KeyedUiAccount};
+
+const LARGEST_VAL_VOTE: [u8; 32] =
+    const_crypto::bs58::decode_pubkey("DdCNGDpP7qMgoAy6paFzhhak2EeyCZcgjH7ak5u5v28m");
+
+#[test]
+fn withdraw_stake_acc_sanity() {
+    let account = KeyedUiAccount::from_test_fixtures_file("validator-list");
+    let data = account.account_data();
+    let deser = ValidatorList::deserialize(data.as_slice()).unwrap();
+
+    let val = deser
+        .entries
+        .iter()
+        .find(|v| *v.vote_account_address() == LARGEST_VAL_VOTE)
+        .unwrap();
+
+    let (pda, _bump) = find_withdraw_stake_acc(val);
+
+    assert_eq!(
+        pda,
+        Pubkey::from_str_const("58EuDup5Gg9nEqDoBBGEDnZ5fMc1u7fQi6kHmEESSUkh")
+    );
+}