Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Transfer pda authority #255

Closed
wants to merge 2 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 3 additions & 4 deletions staking/app/StakeConnection.ts
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
import {
Provider,
Program,
Wallet,
utils,
Expand Down Expand Up @@ -27,6 +26,7 @@ import {
import BN from "bn.js";
import * as idljs from "@project-serum/anchor/dist/cjs/coder/borsh/idl";
import { Staking } from "../target/types/staking";
import IDL from "../target/idl/staking.json";
import { batchInstructions } from "./transaction";
import { PythBalance } from "./pythBalance";
import {
Expand All @@ -36,7 +36,7 @@ import {
} from "@solana/spl-governance";
import { GOVERNANCE_ADDRESS } from "./constants";
import assert from "assert";
import { PositionAccountJs, Position } from "./PositionAccountJs";
import { PositionAccountJs } from "./PositionAccountJs";
let wasm = wasm2;
export { wasm };

Expand Down Expand Up @@ -83,9 +83,8 @@ export class StakeConnection {
stakingProgramAddress: PublicKey
): Promise<StakeConnection> {
const provider = new AnchorProvider(connection, wallet, {});
const idl = (await Program.fetchIdl(stakingProgramAddress, provider))!;
const program = new Program(
idl,
IDL as Idl,
stakingProgramAddress,
provider
) as unknown as Program<Staking>;
Expand Down
8 changes: 5 additions & 3 deletions staking/app/deploy/4_transfer_authorities_to_multisig.ts
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,9 @@ import {
} from "./mainnet_beta";

import { GOVERNANCE_ADDRESS, REALM_ID, STAKING_ADDRESS } from "../constants";
import { AnchorProvider, Program, Wallet } from "@project-serum/anchor";
import { AnchorProvider, Idl, Program, Wallet } from "@project-serum/anchor";
import IDL from "../../target/idl/staking.json";

// Actual transaction hash :
// mainnet-beta : 3FDjeBC946SZ6ZgSiDiNzFHKS5hs9bAXYrJKGZrGw1tuVcwi4BxXB1qvqsVmvtcnG5mzYvLM4hmPLjUTiCiY6Tfe
// devnet : 54WrJp6FDXvJCVzaGojUtWz4brm8wJHx3ZTYCpSTF2EwmeswySYsQY335XhJ1A7KL2N4mhYW7NtAGJpMA2fM9M6W
Expand All @@ -24,8 +26,8 @@ async function main() {
new Wallet(AUTHORITY_KEYPAIR),
{}
);
const idl = (await Program.fetchIdl(STAKING_ADDRESS, provider))!;
const program = new Program(idl, STAKING_ADDRESS, provider);

const program = new Program(IDL as Idl, STAKING_ADDRESS, provider);

const tx = new Transaction();
withSetRealmAuthority(
Expand Down
10 changes: 10 additions & 0 deletions staking/programs/staking/src/context.rs
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,16 @@ pub struct UpdateGovernanceAuthority<'info> {
pub config: Account<'info, global_config::GlobalConfig>,
}

#[derive(Accounts)]
#[instruction(new_authority : Pubkey)]
pub struct UpdatePdaAuthority<'info> {
#[account(address = config.pda_authority)]
pub governance_signer: Signer<'info>,
#[account(mut, seeds = [CONFIG_SEED.as_bytes()], bump = config.bump)]
pub config: Account<'info, global_config::GlobalConfig>,
}


#[derive(Accounts)]
#[instruction(freeze : bool)]
pub struct UpdateFreeze<'info> {
Expand Down
9 changes: 9 additions & 0 deletions staking/programs/staking/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -87,6 +87,15 @@ pub mod staking {
Ok(())
}

pub fn update_pda_authority(
ctx: Context<UpdatePdaAuthority>,
new_authority: Pubkey,
) -> Result<()> {
let config = &mut ctx.accounts.config;
config.pda_authority = new_authority;
Ok(())
}

pub fn update_freeze(ctx: Context<UpdateFreeze>, freeze: bool) -> Result<()> {
let config = &mut ctx.accounts.config;
config.freeze = freeze;
Expand Down
30 changes: 30 additions & 0 deletions staking/target/idl/staking.json
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,36 @@
}
]
},
{
"name": "updatePdaAuthority",
"accounts": [
{
"name": "governanceSigner",
"isMut": false,
"isSigner": true
},
{
"name": "config",
"isMut": true,
"isSigner": false,
"pda": {
"seeds": [
{
"kind": "const",
"type": "string",
"value": "config"
}
]
}
}
],
"args": [
{
"name": "newAuthority",
"type": "publicKey"
}
]
},
{
"name": "updateFreeze",
"accounts": [
Expand Down
60 changes: 60 additions & 0 deletions staking/target/types/staking.ts
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,36 @@ export type Staking = {
}
]
},
{
"name": "updatePdaAuthority",
"accounts": [
{
"name": "governanceSigner",
"isMut": false,
"isSigner": true
},
{
"name": "config",
"isMut": true,
"isSigner": false,
"pda": {
"seeds": [
{
"kind": "const",
"type": "string",
"value": "config"
}
]
}
}
],
"args": [
{
"name": "newAuthority",
"type": "publicKey"
}
]
},
{
"name": "updateFreeze",
"accounts": [
Expand Down Expand Up @@ -1981,6 +2011,36 @@ export const IDL: Staking = {
}
]
},
{
"name": "updatePdaAuthority",
"accounts": [
{
"name": "governanceSigner",
"isMut": false,
"isSigner": true
},
{
"name": "config",
"isMut": true,
"isSigner": false,
"pda": {
"seeds": [
{
"kind": "const",
"type": "string",
"value": "config"
}
]
}
}
],
"args": [
{
"name": "newAuthority",
"type": "publicKey"
}
]
},
{
"name": "updateFreeze",
"accounts": [
Expand Down
73 changes: 72 additions & 1 deletion staking/tests/config.ts
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,9 @@ describe("config", async () => {
const pythMintAuthority = new Keypair();
const zeroPubkey = new PublicKey(0);

const pdaAuthorityKeypair = new Keypair();
const config = readAnchorConfig(ANCHOR_CONFIG_PATH);
const pdaAuthority = PublicKey.unique();
const pdaAuthority = pdaAuthorityKeypair.publicKey;
const governanceProgram = new PublicKey(config.programs.localnet.governance);

let errMap: Map<number, string>;
Expand Down Expand Up @@ -416,4 +417,74 @@ describe("config", async () => {
errMap
);
});

it("updates pda authority", async () => {
await expectFail(
program.methods.updatePdaAuthority(program.provider.wallet.publicKey),
"An address constraint was violated",
errMap
);

const pdaConnection = await StakeConnection.createStakeConnection(
program.provider.connection,
new Wallet(pdaAuthorityKeypair),
program.programId
);

await pdaConnection.program.provider.connection.requestAirdrop(
pdaAuthorityKeypair.publicKey,
1_000_000_000_000
);

// Airdrops are not instant unfortunately, wait
await new Promise((resolve) => setTimeout(resolve, 2000));

await pdaConnection.program.methods
.updatePdaAuthority(program.provider.wallet.publicKey)
.rpc();

let configAccountData = await program.account.globalConfig.fetch(
configAccount
);

assert.equal(
JSON.stringify(configAccountData),
JSON.stringify({
bump,
governanceAuthority: program.provider.wallet.publicKey,
pythTokenMint: pythMintAccount.publicKey,
pythGovernanceRealm: zeroPubkey,
unlockingDuration: 2,
epochDuration: new BN(3600),
freeze: true,
pdaAuthority: program.provider.wallet.publicKey,
governanceProgram,
pythTokenListTime: null,
agreementHash: getDummyAgreementHash(),
mockClockTime: new BN(30),
})
);

await program.methods.updatePdaAuthority(pdaAuthority).rpc();

configAccountData = await program.account.globalConfig.fetch(configAccount);

assert.equal(
JSON.stringify(configAccountData),
JSON.stringify({
bump,
governanceAuthority: program.provider.wallet.publicKey,
pythTokenMint: pythMintAccount.publicKey,
pythGovernanceRealm: zeroPubkey,
unlockingDuration: 2,
epochDuration: new BN(3600),
freeze: true,
pdaAuthority: pdaAuthority,
governanceProgram,
pythTokenListTime: null,
agreementHash: getDummyAgreementHash(),
mockClockTime: new BN(30),
})
);
});
});
3 changes: 2 additions & 1 deletion staking/tsconfig.api.json
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,8 @@
"esModuleInterop": true,
"skipLibCheck": true,
"outDir": "lib",
"declaration": true
"declaration": true,
"resolveJsonModule": true
},
"include": ["./app/*.ts"],
"exclude": ["./tests", "./docker", "./migrations"]
Expand Down
3 changes: 2 additions & 1 deletion staking/tsconfig.json
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,8 @@
"module": "commonjs",
"target": "es6",
"esModuleInterop": true,
"skipLibCheck": true
"skipLibCheck": true,
"resolveJsonModule": true
},
"include": ["./app/*.ts", "./tests"],
"exclude": ["./docker", "./migrations"]
Expand Down