SafeERC20Helper.sol 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. pragma solidity ^0.5.0;
  2. import "../token/ERC20/IERC20.sol";
  3. import "../token/ERC20/SafeERC20.sol";
  4. contract ERC20ReturnFalseMock {
  5. uint256 private _allowance;
  6. // IERC20's functions are not pure, but these mock implementations are: to prevent Solidity from issuing warnings,
  7. // we write to a dummy state variable.
  8. uint256 private _dummy;
  9. function transfer(address, uint256) public returns (bool) {
  10. _dummy = 0;
  11. return false;
  12. }
  13. function transferFrom(address, address, uint256) public returns (bool) {
  14. _dummy = 0;
  15. return false;
  16. }
  17. function approve(address, uint256) public returns (bool) {
  18. _dummy = 0;
  19. return false;
  20. }
  21. function allowance(address, address) public view returns (uint256) {
  22. require(_dummy == 0); // Duummy read from a state variable so that the function is view
  23. return 0;
  24. }
  25. }
  26. contract ERC20ReturnTrueMock {
  27. mapping (address => uint256) private _allowances;
  28. // IERC20's functions are not pure, but these mock implementations are: to prevent Solidity from issuing warnings,
  29. // we write to a dummy state variable.
  30. uint256 private _dummy;
  31. function transfer(address, uint256) public returns (bool) {
  32. _dummy = 0;
  33. return true;
  34. }
  35. function transferFrom(address, address, uint256) public returns (bool) {
  36. _dummy = 0;
  37. return true;
  38. }
  39. function approve(address, uint256) public returns (bool) {
  40. _dummy = 0;
  41. return true;
  42. }
  43. function setAllowance(uint256 allowance_) public {
  44. _allowances[msg.sender] = allowance_;
  45. }
  46. function allowance(address owner, address) public view returns (uint256) {
  47. return _allowances[owner];
  48. }
  49. }
  50. contract ERC20NoReturnMock {
  51. mapping (address => uint256) private _allowances;
  52. // IERC20's functions are not pure, but these mock implementations are: to prevent Solidity from issuing warnings,
  53. // we write to a dummy state variable.
  54. uint256 private _dummy;
  55. function transfer(address, uint256) public {
  56. _dummy = 0;
  57. }
  58. function transferFrom(address, address, uint256) public {
  59. _dummy = 0;
  60. }
  61. function approve(address, uint256) public {
  62. _dummy = 0;
  63. }
  64. function setAllowance(uint256 allowance_) public {
  65. _allowances[msg.sender] = allowance_;
  66. }
  67. function allowance(address owner, address) public view returns (uint256) {
  68. return _allowances[owner];
  69. }
  70. }
  71. contract SafeERC20Wrapper {
  72. using SafeERC20 for IERC20;
  73. IERC20 private _token;
  74. constructor (IERC20 token) public {
  75. _token = token;
  76. }
  77. function transfer() public {
  78. _token.safeTransfer(address(0), 0);
  79. }
  80. function transferFrom() public {
  81. _token.safeTransferFrom(address(0), address(0), 0);
  82. }
  83. function approve(uint256 amount) public {
  84. _token.safeApprove(address(0), amount);
  85. }
  86. function increaseAllowance(uint256 amount) public {
  87. _token.safeIncreaseAllowance(address(0), amount);
  88. }
  89. function decreaseAllowance(uint256 amount) public {
  90. _token.safeDecreaseAllowance(address(0), amount);
  91. }
  92. function setAllowance(uint256 allowance_) public {
  93. ERC20ReturnTrueMock(address(_token)).setAllowance(allowance_);
  94. }
  95. function allowance() public view returns (uint256) {
  96. return _token.allowance(address(0), address(0));
  97. }
  98. }