123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- // SPDX-License-Identifier: MIT
- pragma solidity ^0.6.0;
- import "./Proxy.sol";
- import "../utils/Address.sol";
- /**
- * @title UpgradeableProxy
- * @dev This contract implements a proxy that allows to change the
- * implementation address to which it will delegate.
- * Such a change is called an implementation upgrade.
- */
- contract UpgradeableProxy is Proxy {
- /**
- * @dev Contract constructor.
- * @param _logic Address of the initial implementation.
- * @param _data Data to send as msg.data to the implementation to initialize the proxied contract.
- * It should include the signature and the parameters of the function to be called, as described in
- * https://solidity.readthedocs.io/en/v0.4.24/abi-spec.html#function-selector-and-argument-encoding.
- * This parameter is optional, if no data is given the initialization call to proxied contract will be skipped.
- */
- constructor(address _logic, bytes memory _data) public payable {
- assert(_IMPLEMENTATION_SLOT == bytes32(uint256(keccak256("eip1967.proxy.implementation")) - 1));
- _setImplementation(_logic);
- if(_data.length > 0) {
- // solhint-disable-next-line avoid-low-level-calls
- (bool success,) = _logic.delegatecall(_data);
- require(success);
- }
- }
- /**
- * @dev Emitted when the implementation is upgraded.
- * @param implementation Address of the new implementation.
- */
- event Upgraded(address indexed implementation);
- /**
- * @dev Storage slot with the address of the current implementation.
- * This is the keccak-256 hash of "eip1967.proxy.implementation" subtracted by 1, and is
- * validated in the constructor.
- */
- bytes32 private constant _IMPLEMENTATION_SLOT = 0x360894a13ba1a3210667c828492db98dca3e2076cc3735a920a3ca505d382bbc;
- /**
- * @dev Returns the current implementation.
- * @return impl Address of the current implementation
- */
- function _implementation() internal override view returns (address impl) {
- bytes32 slot = _IMPLEMENTATION_SLOT;
- // solhint-disable-next-line no-inline-assembly
- assembly {
- impl := sload(slot)
- }
- }
- /**
- * @dev Upgrades the proxy to a new implementation.
- * @param newImplementation Address of the new implementation.
- */
- function _upgradeTo(address newImplementation) internal {
- _setImplementation(newImplementation);
- emit Upgraded(newImplementation);
- }
- /**
- * @dev Sets the implementation address of the proxy.
- * @param newImplementation Address of the new implementation.
- */
- function _setImplementation(address newImplementation) internal {
- require(Address.isContract(newImplementation), "UpgradeableProxy: new implementation is not a contract");
- bytes32 slot = _IMPLEMENTATION_SLOT;
- // solhint-disable-next-line no-inline-assembly
- assembly {
- sstore(slot, newImplementation)
- }
- }
- }
|