66 lines
2.5 KiB
Solidity
66 lines
2.5 KiB
Solidity
// SPDX-License-Identifier: MIT
|
|
|
|
pragma solidity ^0.8.0;
|
|
|
|
import "../token/ERC20/IERC20Upgradeable.sol";
|
|
import "../interfaces/IERC3156Upgradeable.sol";
|
|
import "../utils/AddressUpgradeable.sol";
|
|
import "../proxy/utils/Initializable.sol";
|
|
|
|
/**
|
|
* @dev WARNING: this IERC3156FlashBorrower mock implementation is for testing purposes ONLY.
|
|
* Writing a secure flash lock borrower is not an easy task, and should be done with the utmost care.
|
|
* This is not an example of how it should be done, and no pattern present in this mock should be considered secure.
|
|
* Following best practices, always have your contract properly audited before using them to manipulate important funds on
|
|
* live networks.
|
|
*/
|
|
contract ERC3156FlashBorrowerMockUpgradeable is Initializable, IERC3156FlashBorrowerUpgradeable {
|
|
bytes32 internal constant _RETURN_VALUE = keccak256("ERC3156FlashBorrower.onFlashLoan");
|
|
|
|
bool _enableApprove;
|
|
bool _enableReturn;
|
|
|
|
event BalanceOf(address token, address account, uint256 value);
|
|
event TotalSupply(address token, uint256 value);
|
|
|
|
function __ERC3156FlashBorrowerMock_init(bool enableReturn, bool enableApprove) internal onlyInitializing {
|
|
__ERC3156FlashBorrowerMock_init_unchained(enableReturn, enableApprove);
|
|
}
|
|
|
|
function __ERC3156FlashBorrowerMock_init_unchained(bool enableReturn, bool enableApprove) internal onlyInitializing {
|
|
_enableApprove = enableApprove;
|
|
_enableReturn = enableReturn;
|
|
}
|
|
|
|
function onFlashLoan(
|
|
address, /*initiator*/
|
|
address token,
|
|
uint256 amount,
|
|
uint256 fee,
|
|
bytes calldata data
|
|
) public override returns (bytes32) {
|
|
require(msg.sender == token);
|
|
|
|
emit BalanceOf(token, address(this), IERC20Upgradeable(token).balanceOf(address(this)));
|
|
emit TotalSupply(token, IERC20Upgradeable(token).totalSupply());
|
|
|
|
if (data.length > 0) {
|
|
// WARNING: This code is for testing purposes only! Do not use.
|
|
AddressUpgradeable.functionCall(token, data);
|
|
}
|
|
|
|
if (_enableApprove) {
|
|
IERC20Upgradeable(token).approve(token, amount + fee);
|
|
}
|
|
|
|
return _enableReturn ? _RETURN_VALUE : bytes32(0);
|
|
}
|
|
|
|
/**
|
|
* This empty reserved space is put in place to allow future versions to add new
|
|
* variables without shifting down storage in the inheritance chain.
|
|
* See https://docs.openzeppelin.com/contracts/4.x/upgradeable#storage_gaps
|
|
*/
|
|
uint256[50] private __gap;
|
|
}
|