ERC165Checker.sol 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. // SPDX-License-Identifier: MIT
  2. // OpenZeppelin Contracts (last updated v4.8.0-rc.0) (utils/introspection/ERC165Checker.sol)
  3. pragma solidity ^0.8.0;
  4. import "./IERC165.sol";
  5. /**
  6. * @dev Library used to query support of an interface declared via {IERC165}.
  7. *
  8. * Note that these functions return the actual result of the query: they do not
  9. * `revert` if an interface is not supported. It is up to the caller to decide
  10. * what to do in these cases.
  11. */
  12. library ERC165Checker {
  13. // As per the EIP-165 spec, no interface should ever match 0xffffffff
  14. bytes4 private constant _INTERFACE_ID_INVALID = 0xffffffff;
  15. /**
  16. * @dev Returns true if `account` supports the {IERC165} interface.
  17. */
  18. function supportsERC165(address account) internal view returns (bool) {
  19. // Any contract that implements ERC165 must explicitly indicate support of
  20. // InterfaceId_ERC165 and explicitly indicate non-support of InterfaceId_Invalid
  21. return
  22. supportsERC165InterfaceUnchecked(account, type(IERC165).interfaceId) &&
  23. !supportsERC165InterfaceUnchecked(account, _INTERFACE_ID_INVALID);
  24. }
  25. /**
  26. * @dev Returns true if `account` supports the interface defined by
  27. * `interfaceId`. Support for {IERC165} itself is queried automatically.
  28. *
  29. * See {IERC165-supportsInterface}.
  30. */
  31. function supportsInterface(address account, bytes4 interfaceId) internal view returns (bool) {
  32. // query support of both ERC165 as per the spec and support of _interfaceId
  33. return supportsERC165(account) && supportsERC165InterfaceUnchecked(account, interfaceId);
  34. }
  35. /**
  36. * @dev Returns a boolean array where each value corresponds to the
  37. * interfaces passed in and whether they're supported or not. This allows
  38. * you to batch check interfaces for a contract where your expectation
  39. * is that some interfaces may not be supported.
  40. *
  41. * See {IERC165-supportsInterface}.
  42. *
  43. * _Available since v3.4._
  44. */
  45. function getSupportedInterfaces(address account, bytes4[] memory interfaceIds)
  46. internal
  47. view
  48. returns (bool[] memory)
  49. {
  50. // an array of booleans corresponding to interfaceIds and whether they're supported or not
  51. bool[] memory interfaceIdsSupported = new bool[](interfaceIds.length);
  52. // query support of ERC165 itself
  53. if (supportsERC165(account)) {
  54. // query support of each interface in interfaceIds
  55. for (uint256 i = 0; i < interfaceIds.length; i++) {
  56. interfaceIdsSupported[i] = supportsERC165InterfaceUnchecked(account, interfaceIds[i]);
  57. }
  58. }
  59. return interfaceIdsSupported;
  60. }
  61. /**
  62. * @dev Returns true if `account` supports all the interfaces defined in
  63. * `interfaceIds`. Support for {IERC165} itself is queried automatically.
  64. *
  65. * Batch-querying can lead to gas savings by skipping repeated checks for
  66. * {IERC165} support.
  67. *
  68. * See {IERC165-supportsInterface}.
  69. */
  70. function supportsAllInterfaces(address account, bytes4[] memory interfaceIds) internal view returns (bool) {
  71. // query support of ERC165 itself
  72. if (!supportsERC165(account)) {
  73. return false;
  74. }
  75. // query support of each interface in interfaceIds
  76. for (uint256 i = 0; i < interfaceIds.length; i++) {
  77. if (!supportsERC165InterfaceUnchecked(account, interfaceIds[i])) {
  78. return false;
  79. }
  80. }
  81. // all interfaces supported
  82. return true;
  83. }
  84. /**
  85. * @notice Query if a contract implements an interface, does not check ERC165 support
  86. * @param account The address of the contract to query for support of an interface
  87. * @param interfaceId The interface identifier, as specified in ERC-165
  88. * @return true if the contract at account indicates support of the interface with
  89. * identifier interfaceId, false otherwise
  90. * @dev Assumes that account contains a contract that supports ERC165, otherwise
  91. * the behavior of this method is undefined. This precondition can be checked
  92. * with {supportsERC165}.
  93. * Interface identification is specified in ERC-165.
  94. */
  95. function supportsERC165InterfaceUnchecked(address account, bytes4 interfaceId) internal view returns (bool) {
  96. // prepare call
  97. bytes memory encodedParams = abi.encodeWithSelector(IERC165.supportsInterface.selector, interfaceId);
  98. // perform static call
  99. bool success;
  100. uint256 returnSize;
  101. uint256 returnValue;
  102. assembly {
  103. success := staticcall(30000, account, add(encodedParams, 0x20), mload(encodedParams), 0x00, 0x20)
  104. returnSize := returndatasize()
  105. returnValue := mload(0x00)
  106. }
  107. return success && returnSize >= 0x20 && returnValue > 0;
  108. }
  109. }