token 2022: make IncorrectMintAuthority a TokenError

solana-labs · Oct 18, 2023 · 2776f2e · 2776f2e
1 parent 58ac7c2
commit 2776f2e
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 12 deletions.
diff --git a/token-metadata/example/src/processor.rs b/token-metadata/example/src/processor.rs
@@ -12,7 +12,7 @@ use {
         pubkey::Pubkey,
     },
     spl_pod::optional_keys::OptionalNonZeroPubkey,
-    spl_token_2022::{extension::StateWithExtensions, state::Mint},
+    spl_token_2022::{error::TokenError, extension::StateWithExtensions, state::Mint},
     spl_token_metadata_interface::{
         error::TokenMetadataError,
         instruction::{
@@ -65,7 +65,7 @@ pub fn process_initialize(
             return Err(ProgramError::MissingRequiredSignature);
         }
         if mint.base.mint_authority.as_ref() != COption::Some(mint_authority_info.key) {
-            return Err(TokenMetadataError::IncorrectMintAuthority.into());
+            return Err(TokenError::IncorrectMintAuthority.into());
         }
     }
 

diff --git a/token-metadata/example/tests/initialize.rs b/token-metadata/example/tests/initialize.rs
@@ -12,9 +12,8 @@ use {
         system_instruction,
         transaction::{Transaction, TransactionError},
     },
-    spl_token_metadata_interface::{
-        error::TokenMetadataError, instruction::initialize, state::TokenMetadata,
-    },
+    spl_token_2022::error::TokenError,
+    spl_token_metadata_interface::{instruction::initialize, state::TokenMetadata},
     spl_type_length_value::{
         error::TlvError,
         state::{TlvState, TlvStateBorrowed},
@@ -265,7 +264,7 @@ async fn fail_incorrect_authority() {
         error,
         TransactionError::InstructionError(
             1,
-            InstructionError::Custom(TokenMetadataError::IncorrectMintAuthority as u32)
+            InstructionError::Custom(TokenError::IncorrectMintAuthority as u32)
         )
     );
 }
diff --git a/token-metadata/interface/src/error.rs b/token-metadata/interface/src/error.rs
@@ -11,9 +11,6 @@ pub enum TokenMetadataError {
     /// Mint has no mint authority
     #[error("Mint has no mint authority")]
     MintHasNoMintAuthority,
-    /// Incorrect mint authority has signed the instruction
-    #[error("Incorrect mint authority has signed the instruction")]
-    IncorrectMintAuthority,
     /// Incorrect metadata update authority has signed the instruction
     #[error("Incorrect metadata update authority has signed the instruction")]
     IncorrectUpdateAuthority,

diff --git a/token/program-2022-test/tests/token_metadata_initialize.rs b/token/program-2022-test/tests/token_metadata_initialize.rs
@@ -11,7 +11,7 @@ use {
     },
     spl_token_2022::{error::TokenError, extension::BaseStateWithExtensions, processor::Processor},
     spl_token_client::token::{ExtensionInitializationParams, TokenError as TokenClientError},
-    spl_token_metadata_interface::{error::TokenMetadataError, state::TokenMetadata},
+    spl_token_metadata_interface::state::TokenMetadata,
     std::{convert::TryInto, sync::Arc},
 };
 
@@ -110,7 +110,7 @@ async fn success_initialize() {
         TokenClientError::Client(Box::new(TransportError::TransactionError(
             TransactionError::InstructionError(
                 1,
-                InstructionError::Custom(TokenMetadataError::IncorrectMintAuthority as u32)
+                InstructionError::Custom(TokenError::IncorrectMintAuthority as u32)
             )
         )))
     );

diff --git a/token/program-2022/src/error.rs b/token/program-2022/src/error.rs
@@ -232,6 +232,9 @@ pub enum TokenError {
     /// Ciphertext arithmetic failed
     #[error("Ciphertext arithmetic failed")]
     CiphertextArithmeticFailed,
+    /// Incorrect mint authority has signed the instruction
+    #[error("Incorrect mint authority has signed the instruction")]
+    IncorrectMintAuthority,
 }
 impl From<TokenError> for ProgramError {
     fn from(e: TokenError) -> Self {
@@ -407,6 +410,9 @@ impl PrintProgramError for TokenError {
             TokenError::CiphertextArithmeticFailed => {
                 msg!("Ciphertext arithmetic failed")
             }
+            TokenError::IncorrectMintAuthority => {
+                msg!("Incorrect mint authority has signed the instruction")
+            }
         }
     }
 }
diff --git a/token/program-2022/src/extension/token_metadata/processor.rs b/token/program-2022/src/extension/token_metadata/processor.rs
@@ -61,7 +61,7 @@ pub fn process_initialize(
             return Err(ProgramError::MissingRequiredSignature);
         }
         if mint.base.mint_authority.as_ref() != COption::Some(mint_authority_info.key) {
-            return Err(TokenMetadataError::IncorrectMintAuthority.into());
+            return Err(TokenError::IncorrectMintAuthority.into());
         }
 
         if mint.get_extension::<MetadataPointer>().is_err() {