VoteProgram.safeWithdraw function to safeguard against accidental vot…

…e account closures (#26586) feat: safe withdraw function Co-authored-by: aschonfeld <[email protected]>
solana-labs · Aug 16, 2022 · 2cc39ab · 2cc39ab
1 parent 55652a0
commit 2cc39ab
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/src/programs/vote.ts b/src/programs/vote.ts
@@ -410,4 +410,25 @@ export class VoteProgram {
       data,
     });
   }
+
+  /**
+   * Generate a transaction to withdraw safely from a Vote account.
+   *
+   * This function was created as a safeguard for vote accounts running validators, `safeWithdraw`
+   * checks that the withdraw amount will not exceed the specified balance while leaving enough left
+   * to cover rent. If you wish to close the vote account by withdrawing the full amount, call the
+   * `withdraw` method directly.
+   */
+  static safeWithdraw(
+    params: WithdrawFromVoteAccountParams,
+    currentVoteAccountBalance: number,
+    rentExemptMinimum: number,
+  ): Transaction {
+    if (params.lamports > currentVoteAccountBalance - rentExemptMinimum) {
+      throw new Error(
+        'Withdraw will leave vote account with insuffcient funds.',
+      );
+    }
+    return VoteProgram.withdraw(params);
+  }
 }
diff --git a/test/program-tests/vote.test.ts b/test/program-tests/vote.test.ts
@@ -167,6 +167,21 @@ describe('VoteProgram', () => {
 
       // Withdraw from Vote account
       let recipient = Keypair.generate();
+      const voteBalance = await connection.getBalance(newVoteAccount.publicKey);
+
+      expect(() =>
+        VoteProgram.safeWithdraw(
+          {
+            votePubkey: newVoteAccount.publicKey,
+            authorizedWithdrawerPubkey: authorized.publicKey,
+            lamports: voteBalance - minimumAmount + 1,
+            toPubkey: recipient.publicKey,
+          },
+          voteBalance,
+          minimumAmount,
+        ),
+      ).to.throw('Withdraw will leave vote account with insuffcient funds.');
+
       let withdraw = VoteProgram.withdraw({
         votePubkey: newVoteAccount.publicKey,
         authorizedWithdrawerPubkey: authorized.publicKey,