From d533f7730176fe2ab26b25da7c4f611763d6ef0a Mon Sep 17 00:00:00 2001
From: "Jeff Washington (jwash)"
 <75863576+jeffwashington@users.noreply.github.com>
Date: Tue, 27 Apr 2021 09:58:41 -0500
Subject: [PATCH] bank.withdraw uses checked_sub_lamports (#16848)

* bank.withdraw uses checked_sub_lamports

* retain previous codepath for error

* map_err from clippy
---
 runtime/src/bank.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index 980f676145..6dda7d1216 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -4034,8 +4034,9 @@ impl Bank {
                     .checked_add(min_balance)
                     .filter(|required_balance| *required_balance <= account.lamports())
                     .ok_or(TransactionError::InsufficientFundsForFee)?;
-
-                account.lamports -= lamports;
+                account
+                    .checked_sub_lamports(lamports)
+                    .map_err(|_| TransactionError::InsufficientFundsForFee)?;
                 self.store_account(pubkey, &account);
 
                 Ok(())