safe_token_2022/
generic_token_account.rs

1//! Generic Token Account, copied from safe_token::state
2// Remove all of this and use safe-token's version once token 3.4.0 is released
3use {
4    crate::state::AccountState,
5    solana_program::pubkey::{Pubkey, PUBKEY_BYTES},
6};
7
8const SAFE_TOKEN_ACCOUNT_MINT_OFFSET: usize = 0;
9const SAFE_TOKEN_ACCOUNT_OWNER_OFFSET: usize = 32;
10
11/// A trait for token Account structs to enable efficiently unpacking various fields
12/// without unpacking the complete state.
13pub trait GenericTokenAccount {
14    /// Check if the account data is a valid token account
15    fn valid_account_data(account_data: &[u8]) -> bool;
16
17    /// Call after account length has already been verified to unpack the account owner
18    fn unpack_account_owner_unchecked(account_data: &[u8]) -> &Pubkey {
19        Self::unpack_pubkey_unchecked(account_data, SAFE_TOKEN_ACCOUNT_OWNER_OFFSET)
20    }
21
22    /// Call after account length has already been verified to unpack the account mint
23    fn unpack_account_mint_unchecked(account_data: &[u8]) -> &Pubkey {
24        Self::unpack_pubkey_unchecked(account_data, SAFE_TOKEN_ACCOUNT_MINT_OFFSET)
25    }
26
27    /// Call after account length has already been verified to unpack a Pubkey at
28    /// the specified offset. Panics if `account_data.len()` is less than `PUBKEY_BYTES`
29    fn unpack_pubkey_unchecked(account_data: &[u8], offset: usize) -> &Pubkey {
30        bytemuck::from_bytes(&account_data[offset..offset + PUBKEY_BYTES])
31    }
32
33    /// Unpacks an account's owner from opaque account data.
34    fn unpack_account_owner(account_data: &[u8]) -> Option<&Pubkey> {
35        if Self::valid_account_data(account_data) {
36            Some(Self::unpack_account_owner_unchecked(account_data))
37        } else {
38            None
39        }
40    }
41
42    /// Unpacks an account's mint from opaque account data.
43    fn unpack_account_mint(account_data: &[u8]) -> Option<&Pubkey> {
44        if Self::valid_account_data(account_data) {
45            Some(Self::unpack_account_mint_unchecked(account_data))
46        } else {
47            None
48        }
49    }
50}
51
52/// The offset of state field in Account's C representation
53pub const ACCOUNT_INITIALIZED_INDEX: usize = 108;
54
55/// Check if the account data buffer represents an initialized account.
56/// This is checking the `state` (AccountState) field of an Account object.
57pub fn is_initialized_account(account_data: &[u8]) -> bool {
58    *account_data
59        .get(ACCOUNT_INITIALIZED_INDEX)
60        .unwrap_or(&(AccountState::Uninitialized as u8))
61        != AccountState::Uninitialized as u8
62}
safe_token_2022/generic_token_account.rs

safe_token_2022/
generic_token_account.rs
