BeaconProxy.sol 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. // SPDX-License-Identifier: MIT
  2. // OpenZeppelin Contracts (last updated v4.7.0) (proxy/beacon/BeaconProxy.sol)
  3. pragma solidity ^0.8.0;
  4. import "./IBeacon.sol";
  5. import "../Proxy.sol";
  6. import "../ERC1967/ERC1967Upgrade.sol";
  7. /**
  8. * @dev This contract implements a proxy that gets the implementation address for each call from an {UpgradeableBeacon}.
  9. *
  10. * The beacon address is stored in storage slot `uint256(keccak256('eip1967.proxy.beacon')) - 1`, so that it doesn't
  11. * conflict with the storage layout of the implementation behind the proxy.
  12. *
  13. * _Available since v3.4._
  14. */
  15. contract BeaconProxy is Proxy, ERC1967Upgrade {
  16. /**
  17. * @dev Initializes the proxy with `beacon`.
  18. *
  19. * If `data` is nonempty, it's used as data in a delegate call to the implementation returned by the beacon. This
  20. * will typically be an encoded function call, and allows initializing the storage of the proxy like a Solidity
  21. * constructor.
  22. *
  23. * Requirements:
  24. *
  25. * - `beacon` must be a contract with the interface {IBeacon}.
  26. */
  27. constructor(address beacon, bytes memory data) payable {
  28. _upgradeBeaconToAndCall(beacon, data, false);
  29. }
  30. /**
  31. * @dev Returns the current beacon address.
  32. */
  33. function _beacon() internal view virtual returns (address) {
  34. return _getBeacon();
  35. }
  36. /**
  37. * @dev Returns the current implementation address of the associated beacon.
  38. */
  39. function _implementation() internal view virtual override returns (address) {
  40. return IBeacon(_getBeacon()).implementation();
  41. }
  42. /**
  43. * @dev Changes the proxy to use a new beacon. Deprecated: see {_upgradeBeaconToAndCall}.
  44. *
  45. * If `data` is nonempty, it's used as data in a delegate call to the implementation returned by the beacon.
  46. *
  47. * Requirements:
  48. *
  49. * - `beacon` must be a contract.
  50. * - The implementation returned by `beacon` must be a contract.
  51. */
  52. function _setBeacon(address beacon, bytes memory data) internal virtual {
  53. _upgradeBeaconToAndCall(beacon, data, false);
  54. }
  55. }