token-cli: display permanent delegate warning

Author: 2501babe

token/cli/src/main.rs

@@ -1763,11 +1763,23 @@ async fn command_address(
 async fn command_display(config: &Config<'_>, address: Pubkey) -> CommandResult {
     let account_data = config.get_account_checked(&address).await?;
 
-    let decimals = if let Some(mint_address) = get_token_account_mint(&account_data.data) {
-        Some(config.get_mint_info(&mint_address, None).await?.decimals)
-    } else {
-        None
-    };
+    let (decimals, has_permanent_delegate) =
+        if let Some(mint_address) = get_token_account_mint(&account_data.data) {
+            let mint_account = config.get_account_checked(&mint_address).await?;
+            let mint_state = StateWithExtensionsOwned::<Mint>::unpack(mint_account.data)
+                .map_err(|_| format!("Could not deserialize token mint {}", mint_address))?;
+
+            let has_permanent_delegate =
+                if let Ok(permanent_delegate) = mint_state.get_extension::<PermanentDelegate>() {
+                    Option::<Pubkey>::from(permanent_delegate.delegate).is_some()
+                } else {
+                    false
+                };
+
+            (Some(mint_state.base.decimals), has_permanent_delegate)
+        } else {
+            (None, false)
+        };
 
     let token_data = parse_token(&account_data.data, decimals);
 
@@ -1786,6 +1798,7 @@ async fn command_display(config: &Config<'_>, address: Pubkey) -> CommandResult
                 program_id: config.program_id.to_string(),
                 is_associated: associated_address == address,
                 account,
+                has_permanent_delegate,
             };
 
             Ok(config.output_format.formatted_string(&cli_output))

token/cli/src/output.rs

@@ -191,6 +191,8 @@ pub(crate) struct CliTokenAccount {
     pub(crate) is_associated: bool,
     #[serde(flatten)]
     pub(crate) account: UiTokenAccount,
+    #[serde(skip_serializing)]
+    pub(crate) has_permanent_delegate: bool,
 }
 
 impl QuietDisplay for CliTokenAccount {}
@@ -257,6 +259,22 @@ impl fmt::Display for CliTokenAccount {
             writeln!(f, "* Please run `spl-token gc` to clean up Aux accounts")?;
         }
 
+        if self.has_permanent_delegate {
+            writeln!(f)?;
+            writeln!(
+                f,
+                "* {} ",
+                style("This token has a permanent delegate!").bold()
+            )?;
+            writeln!(
+                f,
+                "  This means the mint may withdraw {} funds from this account at {} time.",
+                style("all").bold(),
+                style("any").bold(),
+            )?;
+            writeln!(f, "  If this was not adequately disclosed to you, you may be dealing with a malicious mint.")?;
+        }
+
         Ok(())
     }
 }

token/cli/src/sort.rs

@@ -76,6 +76,7 @@ pub(crate) fn sort_and_parse_token_accounts(
                         program_id: program_id.to_string(),
                         account: ui_token_account,
                         is_associated,
+                        has_permanent_delegate: false,
                     };
 
                     let entry = cli_accounts.entry(btree_key);