logos-blockchain/lez-programs
1
0
Fork 0
mirror of https://github.com/logos-blockchain/lez-programs.git synced 2026-06-30 20:19:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(pda): use descriptive string seeds for PDA derivation

Browse Source 
Replace the hardcoded numeric byte-stream seeds ([0; 32], [1; 32], ...)
used for domain separation in PDA derivation with descriptive byte-string
constants, mirroring the AMM config account's existing b"CONFIG" seed.

  amm:         [0; 32] -> b"LIQUIDITY_TOKEN"
               [1; 32] -> b"LP_LOCK_HOLDING"
  stablecoin:  [0; 32] -> b"POSITION"
               [1; 32] -> b"POSITION_VAULT"
  twap_oracle: [2; 32] -> b"PRICE_OBSERVATIONS"
               [3; 32] -> b"ORACLE_PRICE_ACCOUNT"
               [4; 32] -> b"CURRENT_TICK_ACCOUNT"

Since the seeds are now variable-length, each compute_*_pda_seed function
builds its hash input with a Vec and extend_from_slice instead of a
fixed-size buffer with offset writes.

Closes #146
This commit is contained in:
r4bbit 2026-06-19 14:38:01 +02:00
parent 0e2c5f9329
commit 03345db803
3 changed files with 38 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
programs/amm/core/src/lib.rs
View File

@ -8,10 +8,10 @@ use nssa_core::{
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use spel_framework_macros::account_type; use spel_framework_macros::account_type;
// These stable seed bytes are part of the PDA derivation scheme and must stay unchanged for // These stable domain-separation tags are part of the PDA derivation scheme and must stay
// compatibility. // unchanged for address compatibility.
const LIQUIDITY_TOKEN_PDA_SEED: [u8; 32] = [0; 32]; const LIQUIDITY_TOKEN_PDA_SEED: &[u8] = b"LIQUIDITY_TOKEN";
const LP_LOCK_HOLDING_PDA_SEED: [u8; 32] = [1; 32]; const LP_LOCK_HOLDING_PDA_SEED: &[u8] = b"LP_LOCK_HOLDING";
/// AMM Program Instruction. /// AMM Program Instruction.
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
@ -423,10 +423,9 @@ pub fn compute_liquidity_token_pda(amm_program_id: ProgramId, pool_id: AccountId
pub fn compute_liquidity_token_pda_seed(pool_id: AccountId) -> PdaSeed { pub fn compute_liquidity_token_pda_seed(pool_id: AccountId) -> PdaSeed {
use risc0_zkvm::sha::{Impl, Sha256}; use risc0_zkvm::sha::{Impl, Sha256};
let mut bytes = [0; 64]; let mut bytes = Vec::new();
let (pool_bytes, seed_bytes) = bytes.split_at_mut(32); bytes.extend_from_slice(&pool_id.to_bytes());
pool_bytes.copy_from_slice(&pool_id.to_bytes()); bytes.extend_from_slice(LIQUIDITY_TOKEN_PDA_SEED);
seed_bytes.copy_from_slice(&LIQUIDITY_TOKEN_PDA_SEED);
PdaSeed::new( PdaSeed::new(
Impl::hash_bytes(&bytes) Impl::hash_bytes(&bytes)
@ -443,10 +442,9 @@ pub fn compute_lp_lock_holding_pda(amm_program_id: ProgramId, pool_id: AccountId
pub fn compute_lp_lock_holding_pda_seed(pool_id: AccountId) -> PdaSeed { pub fn compute_lp_lock_holding_pda_seed(pool_id: AccountId) -> PdaSeed {
use risc0_zkvm::sha::{Impl, Sha256}; use risc0_zkvm::sha::{Impl, Sha256};
let mut bytes = [0; 64]; let mut bytes = Vec::new();
let (pool_bytes, seed_bytes) = bytes.split_at_mut(32); bytes.extend_from_slice(&pool_id.to_bytes());
pool_bytes.copy_from_slice(&pool_id.to_bytes()); bytes.extend_from_slice(LP_LOCK_HOLDING_PDA_SEED);
seed_bytes.copy_from_slice(&LP_LOCK_HOLDING_PDA_SEED);
PdaSeed::new( PdaSeed::new(
Impl::hash_bytes(&bytes) Impl::hash_bytes(&bytes)

20
programs/stablecoin/core/src/lib.rs
View File

@ -8,8 +8,10 @@ use nssa_core::{
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use spel_framework_macros::account_type; use spel_framework_macros::account_type;
const POSITION_PDA_DOMAIN: [u8; 32] = [0; 32]; // Stable domain-separation tags for the position PDAs; these must stay unchanged for address
const POSITION_VAULT_PDA_DOMAIN: [u8; 32] = [1; 32]; // compatibility.
const POSITION_PDA_DOMAIN: &[u8] = b"POSITION";
const POSITION_VAULT_PDA_DOMAIN: &[u8] = b"POSITION_VAULT";
/// Stablecoin Program Instruction. /// Stablecoin Program Instruction.
#[derive(Debug, Serialize, Deserialize)] #[derive(Debug, Serialize, Deserialize)]
@ -122,10 +124,10 @@ pub fn compute_position_pda_seed(
) -> PdaSeed { ) -> PdaSeed {
use risc0_zkvm::sha::{Impl, Sha256 as _}; use risc0_zkvm::sha::{Impl, Sha256 as _};
let mut bytes = [0u8; 96]; let mut bytes = Vec::new();
bytes[0..32].copy_from_slice(&owner_id.to_bytes()); bytes.extend_from_slice(&owner_id.to_bytes());
bytes[32..64].copy_from_slice(&collateral_definition_id.to_bytes()); bytes.extend_from_slice(&collateral_definition_id.to_bytes());
bytes[64..96].copy_from_slice(&POSITION_PDA_DOMAIN); bytes.extend_from_slice(POSITION_PDA_DOMAIN);
let mut out = [0u8; 32]; let mut out = [0u8; 32];
out.copy_from_slice(Impl::hash_bytes(&bytes).as_bytes()); out.copy_from_slice(Impl::hash_bytes(&bytes).as_bytes());
@ -151,9 +153,9 @@ pub fn compute_position_pda(
pub fn compute_position_vault_pda_seed(position_id: AccountId) -> PdaSeed { pub fn compute_position_vault_pda_seed(position_id: AccountId) -> PdaSeed {
use risc0_zkvm::sha::{Impl, Sha256 as _}; use risc0_zkvm::sha::{Impl, Sha256 as _};
let mut bytes = [0u8; 64]; let mut bytes = Vec::new();
bytes[0..32].copy_from_slice(&position_id.to_bytes()); bytes.extend_from_slice(&position_id.to_bytes());
bytes[32..64].copy_from_slice(&POSITION_VAULT_PDA_DOMAIN); bytes.extend_from_slice(POSITION_VAULT_PDA_DOMAIN);
let mut out = [0u8; 32]; let mut out = [0u8; 32];
out.copy_from_slice(Impl::hash_bytes(&bytes).as_bytes()); out.copy_from_slice(Impl::hash_bytes(&bytes).as_bytes());

34
programs/twap_oracle/core/src/lib.rs
View File

@ -211,7 +211,7 @@ impl From<&PriceObservations> for Data {
// PDA helpers // PDA helpers
// ────────────────────────────────────────────────────────────────────────────── // ──────────────────────────────────────────────────────────────────────────────
const PRICE_OBSERVATIONS_PDA_SEED: [u8; 32] = [2; 32]; const PRICE_OBSERVATIONS_PDA_SEED: &[u8] = b"PRICE_OBSERVATIONS";
/// Derives the [`AccountId`] for a price source's [`PriceObservations`] PDA. /// Derives the [`AccountId`] for a price source's [`PriceObservations`] PDA.
/// ///
@ -232,7 +232,7 @@ pub fn compute_price_observations_pda(
/// Derives the [`PdaSeed`] for a price source's [`PriceObservations`]. /// Derives the [`PdaSeed`] for a price source's [`PriceObservations`].
/// ///
/// Hash input: `price_source_id (32 bytes) || window_duration_le (8 bytes) || /// Hash input: `price_source_id (32 bytes) || window_duration_le (8 bytes) ||
/// PRICE_OBSERVATIONS_PDA_SEED (32 bytes)`. /// PRICE_OBSERVATIONS_PDA_SEED`.
#[must_use] #[must_use]
pub fn compute_price_observations_pda_seed( pub fn compute_price_observations_pda_seed(
price_source_id: AccountId, price_source_id: AccountId,
@ -240,10 +240,10 @@ pub fn compute_price_observations_pda_seed(
) -> PdaSeed { ) -> PdaSeed {
use risc0_zkvm::sha::{Impl, Sha256}; use risc0_zkvm::sha::{Impl, Sha256};
let mut bytes = [0u8; 72]; let mut bytes = Vec::new();
bytes[..32].copy_from_slice(&price_source_id.to_bytes()); bytes.extend_from_slice(&price_source_id.to_bytes());
bytes[32..40].copy_from_slice(&window_duration.to_le_bytes()); bytes.extend_from_slice(&window_duration.to_le_bytes());
bytes[40..72].copy_from_slice(&PRICE_OBSERVATIONS_PDA_SEED); bytes.extend_from_slice(PRICE_OBSERVATIONS_PDA_SEED);
PdaSeed::new( PdaSeed::new(
Impl::hash_bytes(&bytes) Impl::hash_bytes(&bytes)
@ -253,7 +253,7 @@ pub fn compute_price_observations_pda_seed(
) )
} }
const ORACLE_PRICE_ACCOUNT_PDA_SEED: [u8; 32] = [3; 32]; const ORACLE_PRICE_ACCOUNT_PDA_SEED: &[u8] = b"ORACLE_PRICE_ACCOUNT";
/// Derives the [`AccountId`] for a price source's [`OraclePriceAccount`] PDA. /// Derives the [`AccountId`] for a price source's [`OraclePriceAccount`] PDA.
/// ///
@ -274,7 +274,7 @@ pub fn compute_oracle_price_account_pda(
/// Derives the [`PdaSeed`] for a price source's [`OraclePriceAccount`]. /// Derives the [`PdaSeed`] for a price source's [`OraclePriceAccount`].
/// ///
/// Hash input: `price_source_id (32 bytes) || window_duration_le (8 bytes) || /// Hash input: `price_source_id (32 bytes) || window_duration_le (8 bytes) ||
/// ORACLE_PRICE_ACCOUNT_PDA_SEED (32 bytes)`. /// ORACLE_PRICE_ACCOUNT_PDA_SEED`.
#[must_use] #[must_use]
pub fn compute_oracle_price_account_pda_seed( pub fn compute_oracle_price_account_pda_seed(
price_source_id: AccountId, price_source_id: AccountId,
@ -282,10 +282,10 @@ pub fn compute_oracle_price_account_pda_seed(
) -> PdaSeed { ) -> PdaSeed {
use risc0_zkvm::sha::{Impl, Sha256}; use risc0_zkvm::sha::{Impl, Sha256};
let mut bytes = [0u8; 72]; let mut bytes = Vec::new();
bytes[..32].copy_from_slice(&price_source_id.to_bytes()); bytes.extend_from_slice(&price_source_id.to_bytes());
bytes[32..40].copy_from_slice(&window_duration.to_le_bytes()); bytes.extend_from_slice(&window_duration.to_le_bytes());
bytes[40..72].copy_from_slice(&ORACLE_PRICE_ACCOUNT_PDA_SEED); bytes.extend_from_slice(ORACLE_PRICE_ACCOUNT_PDA_SEED);
PdaSeed::new( PdaSeed::new(
Impl::hash_bytes(&bytes) Impl::hash_bytes(&bytes)
@ -459,7 +459,7 @@ impl From<&CurrentTickAccount> for Data {
} }
} }
const CURRENT_TICK_ACCOUNT_PDA_SEED: [u8; 32] = [4; 32]; const CURRENT_TICK_ACCOUNT_PDA_SEED: &[u8] = b"CURRENT_TICK_ACCOUNT";
/// Derives the [`AccountId`] for a price source's [`CurrentTickAccount`] PDA. /// Derives the [`AccountId`] for a price source's [`CurrentTickAccount`] PDA.
#[must_use] #[must_use]
@ -475,14 +475,14 @@ pub fn compute_current_tick_account_pda(
/// Derives the [`PdaSeed`] for a price source's [`CurrentTickAccount`]. /// Derives the [`PdaSeed`] for a price source's [`CurrentTickAccount`].
/// ///
/// Hash input: `price_source_id (32 bytes) || CURRENT_TICK_ACCOUNT_PDA_SEED (32 bytes)`. /// Hash input: `price_source_id (32 bytes) || CURRENT_TICK_ACCOUNT_PDA_SEED`.
#[must_use] #[must_use]
pub fn compute_current_tick_account_pda_seed(price_source_id: AccountId) -> PdaSeed { pub fn compute_current_tick_account_pda_seed(price_source_id: AccountId) -> PdaSeed {
use risc0_zkvm::sha::{Impl, Sha256}; use risc0_zkvm::sha::{Impl, Sha256};
let mut bytes = [0u8; 64]; let mut bytes = Vec::new();
bytes[..32].copy_from_slice(&price_source_id.to_bytes()); bytes.extend_from_slice(&price_source_id.to_bytes());
bytes[32..64].copy_from_slice(&CURRENT_TICK_ACCOUNT_PDA_SEED); bytes.extend_from_slice(CURRENT_TICK_ACCOUNT_PDA_SEED);
PdaSeed::new( PdaSeed::new(
Impl::hash_bytes(&bytes) Impl::hash_bytes(&bytes)