You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.9 KiB
81 lines
2.9 KiB
// 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)
|
|
}
|
|
}
|
|
}
|
|
|