update executor to support token2022

Cargo.lock

@@ -691,6 +691,7 @@ dependencies = [
  "solana-sdk",
  "solana-security-txt",
  "spl-token 3.5.0",
+ "spl-token-2022 1.0.0",
  "test-case",
  "tokio",
 ]

programs/autobahn-executor/Cargo.toml

@@ -10,6 +10,7 @@ test-bpf = ["no-entrypoint"]
 [dependencies]
 solana-program = "1.17"
 spl-token = { version = "3.5.0", features = ["no-entrypoint"] }
+spl-token-2022 = { version = "1.0.0", features = ["no-entrypoint"] }
 bytemuck = "1.16.1"
 solana-security-txt = "1.1.1"
 default-env = "0.1.1"

programs/autobahn-executor/src/lib.rs

@@ -75,11 +75,29 @@ pub fn process_instruction(
 }
 
 fn get_balance(account: &AccountInfo) -> Result<u64, ProgramError> {
-    let token = spl_token::state::Account::unpack(&account.try_borrow_data()?)?;
-    Ok(token.amount)
+    match account.owner {
+        &spl_token::ID => {
+            let token = spl_token::state::Account::unpack(&account.try_borrow_data()?)?;
+            Ok(token.amount)
+        }
+        &spl_token_2022::ID => {
+            let token = spl_token_2022::state::Account::unpack(&account.try_borrow_data()?)?;
+            Ok(token.amount)
+        }
+        _ => Err(ProgramError::IllegalOwner),
+    }
 }
 
 fn get_mint(account: &AccountInfo) -> Result<Pubkey, ProgramError> {
-    let token = spl_token::state::Account::unpack(&account.try_borrow_data()?)?;
-    Ok(token.mint)
+    match account.owner {
+        &spl_token::ID => {
+            let token = spl_token::state::Account::unpack(&account.try_borrow_data()?)?;
+            Ok(token.mint)
+        }
+        &spl_token_2022::ID => {
+            let token: spl_token_2022::state::Account = spl_token_2022::state::Account::unpack(&account.try_borrow_data()?)?;
+            Ok(token.mint)
+        }
+        _ => Err(ProgramError::IllegalOwner),
+    }
 }