From 50810bd4918fb87490b216d6bc4c3e3ff3571921 Mon Sep 17 00:00:00 2001 From: mingda Date: Thu, 30 Jul 2020 13:58:20 +0800 Subject: [PATCH] add prevent retrance to donate --- contracts/impl/Settlement.sol | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/contracts/impl/Settlement.sol b/contracts/impl/Settlement.sol index d11eb29..bb133d1 100644 --- a/contracts/impl/Settlement.sol +++ b/contracts/impl/Settlement.sol @@ -65,12 +65,12 @@ contract Settlement is Storage { emit Donate(amount, false); } - function donateBaseToken(uint256 amount) external { + function donateBaseToken(uint256 amount) external preventReentrant { _baseTokenTransferIn(msg.sender, amount); _donateBaseToken(amount); } - function donateQuoteToken(uint256 amount) external { + function donateQuoteToken(uint256 amount) external preventReentrant { _quoteTokenTransferIn(msg.sender, amount); _donateQuoteToken(amount); }