sol_parser_sdk/accounts/
utils.rs

1//! 账户解析工具函数
2//!
3//! 提供账户数据解析的通用工具函数
4
5use solana_sdk::pubkey::Pubkey;
6use solana_system_interface::program as system_program;
7use spl_token::solana_program::program_pack::Pack;
8use spl_token::state::Account as SplTokenStateAccount;
9
10/// 从字节数组中读取 Pubkey
11#[inline]
12pub fn read_pubkey(data: &[u8], offset: usize) -> Option<Pubkey> {
13    if data.len() < offset + 32 {
14        return None;
15    }
16    let bytes: [u8; 32] = data[offset..offset + 32].try_into().ok()?;
17    Some(Pubkey::new_from_array(bytes))
18}
19
20/// 从字节数组中读取 u64（小端序）
21#[inline]
22pub fn read_u64_le(data: &[u8], offset: usize) -> Option<u64> {
23    if data.len() < offset + 8 {
24        return None;
25    }
26    Some(u64::from_le_bytes(data[offset..offset + 8].try_into().ok()?))
27}
28
29/// 从字节数组中读取 u16（小端序）
30#[inline]
31pub fn read_u16_le(data: &[u8], offset: usize) -> Option<u16> {
32    if data.len() < offset + 2 {
33        return None;
34    }
35    Some(u16::from_le_bytes(data[offset..offset + 2].try_into().ok()?))
36}
37
38/// 从字节数组中读取 u8
39#[inline]
40pub fn read_u8(data: &[u8], offset: usize) -> Option<u8> {
41    data.get(offset).copied()
42}
43
44/// 检查账户是否是 Nonce Account
45///
46/// Nonce accounts 有一个 discriminator: [1, 0, 0, 0, 1, 0, 0, 0]
47#[inline]
48pub fn is_nonce_account(data: &[u8]) -> bool {
49    data.len() >= 8 && &data[0..8] == &[1, 0, 0, 0, 1, 0, 0, 0]
50}
51
52/// 检查账户所有者是否是 Token Program
53#[inline]
54pub fn is_token_program_account(owner: &Pubkey) -> bool {
55    owner.to_bytes() == spl_token::ID.to_bytes()
56        || owner.to_bytes() == spl_token_2022::ID.to_bytes()
57}
58
59/// 由 `get_account` 得到的 owner / data / executable，判断「该地址」是否对应普通用户侧钱包语义，并返回应作为 **用户公钥** 使用的 [`Pubkey`]。
60///
61/// - **系统程序** + 空 data：返回 `Some(address)`（原生 SOL 钱包）。
62/// - **SPL Token / Token-2022** 账户：返回 `Some(token_owner)`。
63/// - 可执行程序账户或其它：返回 `None`。
64#[inline]
65pub fn user_wallet_pubkey_for_onchain_account(
66    address: &Pubkey,
67    owner: &Pubkey,
68    data: &[u8],
69    executable: bool,
70) -> Option<Pubkey> {
71    if executable {
72        return None;
73    }
74    if owner == &system_program::id() {
75        return if data.is_empty() { Some(*address) } else { None };
76    }
77    if is_token_program_account(owner) && data.len() == SplTokenStateAccount::LEN {
78        let ta = SplTokenStateAccount::unpack(data).ok()?;
79        return Some(Pubkey::new_from_array(ta.owner.to_bytes()));
80    }
81    None
82}
83
84/// 检查账户是否匹配指定的 discriminator
85#[inline]
86pub fn has_discriminator(data: &[u8], discriminator: &[u8]) -> bool {
87    if data.len() < discriminator.len() {
88        return false;
89    }
90    &data[0..discriminator.len()] == discriminator
91}
sol_parser_sdk/accounts/utils.rs

sol_parser_sdk/accounts/
utils.rs
