update ix

Author: SC4RECOIN

programs/portal/src/instructions/admin.rs

@@ -1,5 +1,8 @@
 use anchor_lang::prelude::*;
-use anchor_spl::{token_2022::spl_token_2022::instruction::AuthorityType, token_interface};
+use anchor_spl::{
+    token_2022::{spl_token_2022::instruction::AuthorityType, Token2022},
+    token_interface,
+};
 
 use crate::{
     config::Config,
@@ -9,6 +12,7 @@ use crate::{
     pending_token_authority::PendingTokenAuthority,
     queue::{inbox::InboxRateLimit, outbox::OutboxRateLimit, rate_limit::RateLimitState},
     registered_transceiver::RegisteredTransceiver,
+    TOKEN_AUTHORITY_SEED,
 };
 
 // * Transfer ownership
@@ -639,9 +643,24 @@ pub struct SetMint<'info> {
     pub config: Account<'info, Config>,
 
     pub mint: InterfaceAccount<'info, token_interface::Mint>,
+
+    /// CHECK: This account is validated by its seeds
+    #[account(
+        seeds = [TOKEN_AUTHORITY_SEED],
+        bump,
+    )]
+    pub token_authority: UncheckedAccount<'info>,
+
+    #[account(
+        associated_token::mint = mint,
+        associated_token::authority = token_authority,
+        associated_token::token_program = Token2022::id(),
+    )]
+    pub custody: InterfaceAccount<'info, token_interface::TokenAccount>,
 }
 
 pub fn set_mint(ctx: Context<SetMint>) -> Result<()> {
     ctx.accounts.config.mint = ctx.accounts.mint.key();
+    ctx.accounts.config.custody = ctx.accounts.custody.key();
     Ok(())
 }

services/cli/utils.ts

@@ -3,6 +3,7 @@ import { SolanaNtt } from '@wormhole-foundation/sdk-solana-ntt';
 import { SolanaPlatform, SolanaSendSigner } from '@wormhole-foundation/sdk-solana';
 import { AccountAddress, sha256, Wormhole } from '@wormhole-foundation/sdk';
 import { AnchorProvider, Wallet } from '@coral-xyz/anchor';
+import { getAssociatedTokenAddressSync, TOKEN_2022_PROGRAM_ID } from '@solana/spl-token';
 
 const PORTAL = new PublicKey('mzp1q2j5Hr1QuLC3KFBCAUz5aUckT6qyuZKZ3WJnMmY');
 
@@ -65,6 +66,21 @@ export function updatePortalMint(owner: PublicKey, config: PublicKey, mMint: Pub
         isSigner: false,
         isWritable: false,
       },
+      {
+        pubkey: PublicKey.findProgramAddressSync([Buffer.from('token_authority')], PORTAL)[0],
+        isSigner: false,
+        isWritable: false,
+      },
+      {
+        pubkey: getAssociatedTokenAddressSync(
+          mMint,
+          PublicKey.findProgramAddressSync([Buffer.from('token_authority')], PORTAL)[0],
+          true,
+          TOKEN_2022_PROGRAM_ID,
+        ),
+        isSigner: false,
+        isWritable: false,
+      },
     ],
     data: Buffer.concat([Buffer.from(sha256('global:set_mint').subarray(0, 8))]),
   });