clone_solana_loader_v3_interface/
state.rs

1use clone_solana_pubkey::Pubkey;
2
3/// Upgradeable loader account states
4#[cfg_attr(
5    feature = "frozen-abi",
6    derive(clone_solana_frozen_abi_macro::AbiExample)
7)]
8#[cfg_attr(
9    feature = "serde",
10    derive(serde_derive::Deserialize, serde_derive::Serialize)
11)]
12#[derive(Debug, PartialEq, Eq, Clone, Copy)]
13pub enum UpgradeableLoaderState {
14    /// Account is not initialized.
15    Uninitialized,
16    /// A Buffer account.
17    Buffer {
18        /// Authority address
19        authority_address: Option<Pubkey>,
20        // The raw program data follows this serialized structure in the
21        // account's data.
22    },
23    /// An Program account.
24    Program {
25        /// Address of the ProgramData account.
26        programdata_address: Pubkey,
27    },
28    // A ProgramData account.
29    ProgramData {
30        /// Slot that the program was last modified.
31        slot: u64,
32        /// Address of the Program's upgrade authority.
33        upgrade_authority_address: Option<Pubkey>,
34        // The raw program data follows this serialized structure in the
35        // account's data.
36    },
37}
38impl UpgradeableLoaderState {
39    /// Size of a serialized program account.
40    pub const fn size_of_uninitialized() -> usize {
41        4 // see test_state_size_of_uninitialized
42    }
43
44    /// Size of a buffer account's serialized metadata.
45    pub const fn size_of_buffer_metadata() -> usize {
46        37 // see test_state_size_of_buffer_metadata
47    }
48
49    /// Size of a programdata account's serialized metadata.
50    pub const fn size_of_programdata_metadata() -> usize {
51        45 // see test_state_size_of_programdata_metadata
52    }
53
54    /// Size of a serialized program account.
55    pub const fn size_of_program() -> usize {
56        36 // see test_state_size_of_program
57    }
58
59    /// Size of a serialized buffer account.
60    pub const fn size_of_buffer(program_len: usize) -> usize {
61        Self::size_of_buffer_metadata().saturating_add(program_len)
62    }
63
64    /// Size of a serialized programdata account.
65    pub const fn size_of_programdata(program_len: usize) -> usize {
66        Self::size_of_programdata_metadata().saturating_add(program_len)
67    }
68}
69
70#[cfg(test)]
71mod tests {
72    use {super::*, bincode::serialized_size};
73
74    #[test]
75    fn test_state_size_of_uninitialized() {
76        let buffer_state = UpgradeableLoaderState::Uninitialized;
77        let size = serialized_size(&buffer_state).unwrap();
78        assert_eq!(UpgradeableLoaderState::size_of_uninitialized() as u64, size);
79    }
80
81    #[test]
82    fn test_state_size_of_buffer_metadata() {
83        let buffer_state = UpgradeableLoaderState::Buffer {
84            authority_address: Some(Pubkey::default()),
85        };
86        let size = serialized_size(&buffer_state).unwrap();
87        assert_eq!(
88            UpgradeableLoaderState::size_of_buffer_metadata() as u64,
89            size
90        );
91    }
92
93    #[test]
94    fn test_state_size_of_programdata_metadata() {
95        let programdata_state = UpgradeableLoaderState::ProgramData {
96            upgrade_authority_address: Some(Pubkey::default()),
97            slot: 0,
98        };
99        let size = serialized_size(&programdata_state).unwrap();
100        assert_eq!(
101            UpgradeableLoaderState::size_of_programdata_metadata() as u64,
102            size
103        );
104    }
105
106    #[test]
107    fn test_state_size_of_program() {
108        let program_state = UpgradeableLoaderState::Program {
109            programdata_address: Pubkey::default(),
110        };
111        let size = serialized_size(&program_state).unwrap();
112        assert_eq!(UpgradeableLoaderState::size_of_program() as u64, size);
113    }
114}
clone_solana_loader_v3_interface/state.rs

clone_solana_loader_v3_interface/
state.rs
