Simplify math

Author: febo

p-token/src/processor/close_account.rs

@@ -44,14 +44,11 @@ pub fn process_close_account(accounts: &[AccountInfo]) -> ProgramResult {
         }
     }
 
-    let destination_starting_lamports = destination_account_info.lamports();
     // SAFETY: single mutable borrow to `destination_account_info` lamports and
     // there are no "active" borrows of `source_account_info` account data.
     unsafe {
         // Moves the lamports to the destination account.
-        *destination_account_info.borrow_mut_lamports_unchecked() = destination_starting_lamports
-            .checked_add(source_account_info.lamports())
-            .ok_or(TokenError::Overflow)?;
+        *destination_account_info.borrow_mut_lamports_unchecked() += source_account_info.lamports();
         // Closes the source account.
         source_account_info.close_unchecked();
     }

p-token/src/processor/shared/burn.rs

@@ -83,8 +83,7 @@ pub fn process_burn(
         source_account.set_amount(updated_source_amount);
         // Note: The amount of a token account is always within the range of the
         // mint supply (`u64`).
-        let mint_supply = mint.supply().checked_sub(amount).unwrap();
-        mint.set_supply(mint_supply);
+        mint.set_supply(mint.supply() - amount);
     }
 
     Ok(())

p-token/src/processor/withdraw_excess_lamports.rs

@@ -86,13 +86,12 @@ pub fn process_withdraw_excess_lamports(accounts: &[AccountInfo]) -> ProgramResu
             source_starting_lamports - transfer_amount;
     }
 
-    let destination_starting_lamports = destination_info.lamports();
     // SAFETY: single mutable borrow to `destination_info` lamports.
     unsafe {
         // Moves the lamports to the destination account.
-        *destination_info.borrow_mut_lamports_unchecked() = destination_starting_lamports
-            .checked_add(transfer_amount)
-            .ok_or(TokenError::Overflow)?;
+        //
+        // Note: The total lamports supply is bound to `u64::MAX`.
+        *destination_info.borrow_mut_lamports_unchecked() += transfer_amount;
     }
 
     Ok(())