From 45cfa5b574eab362113f41d2fc573b0d1cf1bf0f Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Sat, 20 Oct 2018 20:39:13 -0500
Subject: [PATCH] Add instruction to transfer account ownership

---
 src/token_program.rs | 39 +++++++++++++++++++++++++++++++++++++--
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/src/token_program.rs b/src/token_program.rs
index a84e7b3874..7391c7a753 100644
--- a/src/token_program.rs
+++ b/src/token_program.rs
@@ -86,9 +86,10 @@ pub struct TokenAccountInfo {
 #[derive(Debug, Serialize, Deserialize, PartialEq)]
 enum Command {
     NewToken(TokenInfo),
-    NewTokenAccount(),
+    NewTokenAccount,
     Transfer(u64),
     Approve(u64),
+    SetOwner,
 }
 
 #[derive(Debug, Serialize, Deserialize, PartialEq)]
@@ -369,6 +370,33 @@ impl TokenProgram {
         Ok(())
     }
 
+    pub fn process_command_setowner(
+        tx: &Transaction,
+        pix: usize,
+        input_program_accounts: &[TokenProgram],
+        output_program_accounts: &mut Vec<(usize, TokenProgram)>,
+    ) -> Result<()> {
+        if input_program_accounts.len() < 3 {
+            error!("Expected 3 accounts");
+            Err(Error::InvalidArgument)?;
+        }
+
+        if let TokenProgram::Account(source_account) = &input_program_accounts[1] {
+            if Some(&source_account.owner) != tx.key(pix, 0) {
+                info!("owner of account 1 not present");
+                Err(Error::InvalidArgument)?;
+            }
+
+            let mut output_source_account = source_account.clone();
+            output_source_account.owner = *tx.key(pix, 2).unwrap();
+            output_program_accounts.push((1, TokenProgram::Account(output_source_account)));
+        } else {
+            info!("account 1 is invalid");
+            Err(Error::InvalidArgument)?;
+        }
+        Ok(())
+    }
+
     pub fn process_transaction(
         tx: &Transaction,
         pix: usize,
@@ -402,7 +430,7 @@ impl TokenProgram {
                 &input_program_accounts,
                 &mut output_program_accounts,
             )?,
-            Command::NewTokenAccount() => Self::process_command_newaccount(
+            Command::NewTokenAccount => Self::process_command_newaccount(
                 tx,
                 pix,
                 &input_program_accounts,
@@ -424,6 +452,13 @@ impl TokenProgram {
                 &input_program_accounts,
                 &mut output_program_accounts,
             )?,
+
+            Command::SetOwner => Self::process_command_setowner(
+                tx,
+                pix,
+                &input_program_accounts,
+                &mut output_program_accounts,
+            )?,
         }
 
         for (index, program_account) in &output_program_accounts {