Address.sol 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  1. // SPDX-License-Identifier: MIT
  2. // OpenZeppelin Contracts (last updated v5.0.0-rc.0) (utils/Address.sol)
  3. pragma solidity ^0.8.20;
  4. /**
  5. * @dev Collection of functions related to the address type
  6. */
  7. library Address {
  8. /**
  9. * @dev The ETH balance of the account is not enough to perform the operation.
  10. */
  11. error AddressInsufficientBalance(address account);
  12. /**
  13. * @dev There's no code at `target` (it is not a contract).
  14. */
  15. error AddressEmptyCode(address target);
  16. /**
  17. * @dev A call to an address target failed. The target may have reverted.
  18. */
  19. error FailedInnerCall();
  20. /**
  21. * @dev Replacement for Solidity's `transfer`: sends `amount` wei to
  22. * `recipient`, forwarding all available gas and reverting on errors.
  23. *
  24. * https://eips.ethereum.org/EIPS/eip-1884[EIP1884] increases the gas cost
  25. * of certain opcodes, possibly making contracts go over the 2300 gas limit
  26. * imposed by `transfer`, making them unable to receive funds via
  27. * `transfer`. {sendValue} removes this limitation.
  28. *
  29. * https://consensys.net/diligence/blog/2019/09/stop-using-soliditys-transfer-now/[Learn more].
  30. *
  31. * IMPORTANT: because control is transferred to `recipient`, care must be
  32. * taken to not create reentrancy vulnerabilities. Consider using
  33. * {ReentrancyGuard} or the
  34. * https://solidity.readthedocs.io/en/v0.8.20/security-considerations.html#use-the-checks-effects-interactions-pattern[checks-effects-interactions pattern].
  35. */
  36. function sendValue(address payable recipient, uint256 amount) internal {
  37. if (address(this).balance < amount) {
  38. revert AddressInsufficientBalance(address(this));
  39. }
  40. (bool success, ) = recipient.call{value: amount}("");
  41. if (!success) {
  42. revert FailedInnerCall();
  43. }
  44. }
  45. /**
  46. * @dev Performs a Solidity function call using a low level `call`. A
  47. * plain `call` is an unsafe replacement for a function call: use this
  48. * function instead.
  49. *
  50. * If `target` reverts with a revert reason or custom error, it is bubbled
  51. * up by this function (like regular Solidity function calls). However, if
  52. * the call reverted with no returned reason, this function reverts with a
  53. * {FailedInnerCall} error.
  54. *
  55. * Returns the raw returned data. To convert to the expected return value,
  56. * use https://solidity.readthedocs.io/en/latest/units-and-global-variables.html?highlight=abi.decode#abi-encoding-and-decoding-functions[`abi.decode`].
  57. *
  58. * Requirements:
  59. *
  60. * - `target` must be a contract.
  61. * - calling `target` with `data` must not revert.
  62. */
  63. function functionCall(address target, bytes memory data) internal returns (bytes memory) {
  64. return functionCallWithValue(target, data, 0);
  65. }
  66. /**
  67. * @dev Same as {xref-Address-functionCall-address-bytes-}[`functionCall`],
  68. * but also transferring `value` wei to `target`.
  69. *
  70. * Requirements:
  71. *
  72. * - the calling contract must have an ETH balance of at least `value`.
  73. * - the called Solidity function must be `payable`.
  74. */
  75. function functionCallWithValue(address target, bytes memory data, uint256 value) internal returns (bytes memory) {
  76. if (address(this).balance < value) {
  77. revert AddressInsufficientBalance(address(this));
  78. }
  79. (bool success, bytes memory returndata) = target.call{value: value}(data);
  80. return verifyCallResultFromTarget(target, success, returndata);
  81. }
  82. /**
  83. * @dev Same as {xref-Address-functionCall-address-bytes-}[`functionCall`],
  84. * but performing a static call.
  85. */
  86. function functionStaticCall(address target, bytes memory data) internal view returns (bytes memory) {
  87. (bool success, bytes memory returndata) = target.staticcall(data);
  88. return verifyCallResultFromTarget(target, success, returndata);
  89. }
  90. /**
  91. * @dev Same as {xref-Address-functionCall-address-bytes-}[`functionCall`],
  92. * but performing a delegate call.
  93. */
  94. function functionDelegateCall(address target, bytes memory data) internal returns (bytes memory) {
  95. (bool success, bytes memory returndata) = target.delegatecall(data);
  96. return verifyCallResultFromTarget(target, success, returndata);
  97. }
  98. /**
  99. * @dev Tool to verify that a low level call to smart-contract was successful, and reverts if the target
  100. * was not a contract or bubbling up the revert reason (falling back to {FailedInnerCall}) in case of an
  101. * unsuccessful call.
  102. */
  103. function verifyCallResultFromTarget(
  104. address target,
  105. bool success,
  106. bytes memory returndata
  107. ) internal view returns (bytes memory) {
  108. if (!success) {
  109. _revert(returndata);
  110. } else {
  111. // only check if target is a contract if the call was successful and the return data is empty
  112. // otherwise we already know that it was a contract
  113. if (returndata.length == 0 && target.code.length == 0) {
  114. revert AddressEmptyCode(target);
  115. }
  116. return returndata;
  117. }
  118. }
  119. /**
  120. * @dev Tool to verify that a low level call was successful, and reverts if it wasn't, either by bubbling the
  121. * revert reason or with a default {FailedInnerCall} error.
  122. */
  123. function verifyCallResult(bool success, bytes memory returndata) internal pure returns (bytes memory) {
  124. if (!success) {
  125. _revert(returndata);
  126. } else {
  127. return returndata;
  128. }
  129. }
  130. /**
  131. * @dev Reverts with returndata if present. Otherwise reverts with {FailedInnerCall}.
  132. */
  133. function _revert(bytes memory returndata) private pure {
  134. // Look for revert reason and bubble it up if present
  135. if (returndata.length > 0) {
  136. // The easiest way to bubble the revert reason is using memory via assembly
  137. /// @solidity memory-safe-assembly
  138. assembly {
  139. let returndata_size := mload(returndata)
  140. revert(add(32, returndata), returndata_size)
  141. }
  142. } else {
  143. revert FailedInnerCall();
  144. }
  145. }
  146. }