Drain the entire compute budget (#12478)

2020-09-25 11:08:10 -07:00
parent baeba41cfd
commit d00453f747
4 changed files with 7 additions and 4 deletions
--- a/programs/bpf/benches/bpf_loader.rs
+++ b/programs/bpf/benches/bpf_loader.rs
@@ -286,8 +286,9 @@ pub struct MockComputeMeter {
 }
 impl ComputeMeter for MockComputeMeter {
    fn consume(&mut self, amount: u64) -> Result<(), InstructionError> {
+        let exceeded = self.remaining < amount;
        self.remaining = self.remaining.saturating_sub(amount);
-        if self.remaining == 0 {
+        if exceeded {
            return Err(InstructionError::ComputationalBudgetExceeded);
        }
        Ok(())