1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
use solana_program::{
account_info::AccountInfo,
bpf_loader_upgradeable::{self, UpgradeableLoaderState},
program::invoke,
program_error::ProgramError,
pubkey::Pubkey,
};
use bincode::deserialize;
use crate::error::GovernanceError;
pub fn get_program_data_address(program: &Pubkey) -> Pubkey {
Pubkey::find_program_address(&[program.as_ref()], &bpf_loader_upgradeable::id()).0
}
pub fn get_program_upgrade_authority(
upgradable_loader_state: &UpgradeableLoaderState,
) -> Result<Option<Pubkey>, ProgramError> {
let upgrade_authority = match upgradable_loader_state {
UpgradeableLoaderState::ProgramData {
slot: _,
upgrade_authority_address,
} => *upgrade_authority_address,
_ => return Err(ProgramError::InvalidAccountData),
};
Ok(upgrade_authority)
}
pub fn set_program_upgrade_authority<'a>(
program_address: &Pubkey,
program_data_info: &AccountInfo<'a>,
program_upgrade_authority_info: &AccountInfo<'a>,
new_authority_info: &AccountInfo<'a>,
bpf_upgrade_loader_info: &AccountInfo<'a>,
) -> Result<(), ProgramError> {
let set_upgrade_authority_instruction = bpf_loader_upgradeable::set_upgrade_authority(
program_address,
program_upgrade_authority_info.key,
Some(new_authority_info.key),
);
invoke(
&set_upgrade_authority_instruction,
&[
program_data_info.clone(),
program_upgrade_authority_info.clone(),
bpf_upgrade_loader_info.clone(),
new_authority_info.clone(),
],
)
}
pub fn assert_program_upgrade_authority_is_signer(
program_address: &Pubkey,
program_data_info: &AccountInfo,
program_upgrade_authority_info: &AccountInfo,
) -> Result<(), ProgramError> {
if program_data_info.owner != &bpf_loader_upgradeable::id() {
return Err(ProgramError::IncorrectProgramId);
}
let program_data_address = get_program_data_address(program_address);
if program_data_address != *program_data_info.key {
return Err(GovernanceError::InvalidProgramDataAccountAddress.into());
}
let upgrade_authority = if let UpgradeableLoaderState::ProgramData {
slot: _,
upgrade_authority_address,
} = deserialize(&program_data_info.data.borrow())
.map_err(|_| GovernanceError::InvalidProgramDataAccountData)?
{
upgrade_authority_address
} else {
None
};
let upgrade_authority = upgrade_authority.ok_or(GovernanceError::ProgramNotUpgradable)?;
if upgrade_authority != *program_upgrade_authority_info.key {
return Err(GovernanceError::InvalidUpgradeAuthority.into());
}
if !program_upgrade_authority_info.is_signer {
return Err(GovernanceError::UpgradeAuthorityMustSign.into());
}
Ok(())
}