From 2f59927b30b0291f3637aaa5d5264a75a68b4bee Mon Sep 17 00:00:00 2001
From: Mark Spanbroek <mark@spanbroek.net>
Date: Mon, 14 Feb 2022 17:12:05 +0100
Subject: [PATCH] Check result of ERC20 transferFrom call

---
 contracts/Collateral.sol | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/contracts/Collateral.sol b/contracts/Collateral.sol
index e1634d8..624313c 100644
--- a/contracts/Collateral.sol
+++ b/contracts/Collateral.sol
@@ -26,8 +26,13 @@ contract Collateral {
     totals.balance -= amount;
   }
 
+  function transferFrom(address sender, uint256 amount) private {
+    address receiver = address(this);
+    require(token.transferFrom(sender, receiver, amount), "Transfer failed");
+  }
+
   function deposit(uint256 amount) public invariant {
-    token.transferFrom(msg.sender, address(this), amount);
+    transferFrom(msg.sender, amount);
     totals.deposited += amount;
     add(msg.sender, amount);
   }