From 60169ee5eea4e9407c1e4dd3c8e229d622acb60b Mon Sep 17 00:00:00 2001 From: pradyuman-verma Date: Mon, 27 Dec 2021 21:11:55 +0530 Subject: [PATCH] updated setUint --- contracts/mainnet/connectors/sushi-incentive/events.sol | 1 + contracts/mainnet/connectors/sushi-incentive/helpers.sol | 5 ++++- contracts/mainnet/connectors/sushi-incentive/interface.sol | 1 + contracts/mainnet/connectors/sushi-incentive/main.sol | 5 ++++- 4 files changed, 10 insertions(+), 2 deletions(-) diff --git a/contracts/mainnet/connectors/sushi-incentive/events.sol b/contracts/mainnet/connectors/sushi-incentive/events.sol index 155d1983..4ef2b9b3 100644 --- a/contracts/mainnet/connectors/sushi-incentive/events.sol +++ b/contracts/mainnet/connectors/sushi-incentive/events.sol @@ -1,3 +1,4 @@ +// SPDX-License-Identifier: MIT pragma solidity ^0.7.0; contract Events { diff --git a/contracts/mainnet/connectors/sushi-incentive/helpers.sol b/contracts/mainnet/connectors/sushi-incentive/helpers.sol index a02379db..c4b5db99 100644 --- a/contracts/mainnet/connectors/sushi-incentive/helpers.sol +++ b/contracts/mainnet/connectors/sushi-incentive/helpers.sol @@ -1,5 +1,4 @@ // SPDX-License-Identifier: MIT - pragma solidity ^0.7.0; pragma experimental ABIEncoderV2; @@ -100,4 +99,8 @@ contract Helpers is DSMath, Basic { address(this) ); } + + function _balance(address token1, address token2) internal view returns (uint balance) { + balance = IERC20(token1).balanceOf(address(this)) + IERC20(token2).balanceOf(address(this)); + } } diff --git a/contracts/mainnet/connectors/sushi-incentive/interface.sol b/contracts/mainnet/connectors/sushi-incentive/interface.sol index 777e4d77..d8b92e92 100644 --- a/contracts/mainnet/connectors/sushi-incentive/interface.sol +++ b/contracts/mainnet/connectors/sushi-incentive/interface.sol @@ -1,3 +1,4 @@ +// SPDX-License-Identifier: MIT pragma solidity ^0.7.0; pragma experimental ABIEncoderV2; diff --git a/contracts/mainnet/connectors/sushi-incentive/main.sol b/contracts/mainnet/connectors/sushi-incentive/main.sol index bfb56d79..87921e57 100644 --- a/contracts/mainnet/connectors/sushi-incentive/main.sol +++ b/contracts/mainnet/connectors/sushi-incentive/main.sol @@ -1,3 +1,4 @@ +// SPDX-License-Identifier: MIT pragma solidity ^0.7.0; pragma experimental ABIEncoderV2; @@ -198,10 +199,12 @@ abstract contract SushipswapIncentiveResolver is Helpers, Events { if (data.poolId == uint256(-1) || data.version == 0) { data = _getPoolId(token1, token2); } - setUint(setId, data.poolId); + uint256 currentBal = _balance(token1, token2); require(data.poolId != uint256(-1), "pool-does-not-exist"); (uint256 lpAmount, uint256 rewardsAmount) = _getUserInfo(data); _emergencyWithdraw(data); + uint256 finalBal = _balance(token1, token2); + setUint(setId, uint256(currentBal - finalBal)); _eventName = "LogEmergencyWithdraw(address,address,uint256,uint256,uint256,uint256)"; _eventParam = abi.encode( token1,