d-bis/dodo-contractV2
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use safetransfer

Browse Source
This commit is contained in:
Attens1423
2023-11-28 19:17:20 +08:00
parent 806cfb6a62
commit 81c0ce41ef
3 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
archive/DODOV2Proxy01.sol
View File

@@ -335,8 +335,7 @@ contract DODOV2Proxy01 is IDODOV2Proxy01, ReentrancyGuard, InitializableOwnable
uint256 originToTokenBalance = IERC20(toToken).balanceOf(msg.sender); uint256 originToTokenBalance = IERC20(toToken).balanceOf(msg.sender);
IWETH(_WETH_).deposit{value: msg.value}(); IWETH(_WETH_).deposit{value: msg.value}();
bool success = IWETH(_WETH_).transfer(dodoPairs[0], msg.value); SafeERC20.safeTransfer(IERC20(_WETH_), dodoPairs[0], msg.value);
require(success, "WETH_TRANSFER_FAIL");
for (uint256 i = 0; i < dodoPairs.length; i++) { for (uint256 i = 0; i < dodoPairs.length; i++) {
if (i == dodoPairs.length - 1) { if (i == dodoPairs.length - 1) {

3
contracts/SmartRoute/DODOV2Proxy02.sol
View File

@@ -226,8 +226,7 @@ contract DODOV2Proxy02 is IDODOV2Proxy01, ReentrancyGuard, InitializableOwnable
uint256 originToTokenBalance = IERC20(toToken).balanceOf(msg.sender); uint256 originToTokenBalance = IERC20(toToken).balanceOf(msg.sender);
IWETH(_WETH_).deposit{value: msg.value}(); IWETH(_WETH_).deposit{value: msg.value}();
bool success = IWETH(_WETH_).transfer(dodoPairs[0], msg.value); SafeERC20.safeTransfer(IERC20(_WETH_), dodoPairs[0], msg.value);
require(success, "WETH_TRANSFER_FAIL");
for (uint256 i = 0; i < dodoPairs.length; i++) { for (uint256 i = 0; i < dodoPairs.length; i++) {
if (i == dodoPairs.length - 1) { if (i == dodoPairs.length - 1) {

3
contracts/SmartRoute/adapter/UniV3Adapter.sol
View File

@@ -103,8 +103,7 @@ contract UniV3Adapter is IDODOAdapter, IUniswapV3SwapCallback, InitializableOwna
if (token == _WETH_ && address(this).balance >= value) { if (token == _WETH_ && address(this).balance >= value) {
// pay with WETH9 // pay with WETH9
IWETH(_WETH_).deposit{value: value}(); // wrap only what is needed to pay IWETH(_WETH_).deposit{value: value}(); // wrap only what is needed to pay
bool success = IWETH(_WETH_).transfer(recipient, value); SafeERC20.safeTransfer(IERC20(_WETH_), recipient, value);
require(success, "WETH_TRANSFER_FAIL");
} else if (payer == address(this)) { } else if (payer == address(this)) {
// pay with tokens already in the contract (for the exact input multihop case) // pay with tokens already in the contract (for the exact input multihop case)
SafeERC20.safeTransfer(IERC20(token), recipient, value); SafeERC20.safeTransfer(IERC20(token), recipient, value);