Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
O
openzeppelin-contracts-upgradeable
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
俞永鹏
openzeppelin-contracts-upgradeable
Commits
be866876
Commit
be866876
authored
Apr 21, 2021
by
Francisco Giordano
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
inline Address.functionDelegateCall and remove from Address library
parent
f0ec852a
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
71 additions
and
128 deletions
+71
-128
AddressImpl.sol
contracts/mocks/AddressImpl.sol
+0
-5
ERC1967Upgrade.sol
contracts/proxy/ERC1967/ERC1967Upgrade.sol
+37
-4
Address.sol
contracts/utils/Address.sol
+0
-24
Multicall.sol
contracts/utils/Multicall.sol
+34
-1
01-inline-address-functiondelegatecall.patch
...deable/patch/01-inline-address-functiondelegatecall.patch
+0
-94
No files found.
contracts/mocks/AddressImpl.sol
View file @
be866876
...
@@ -32,11 +32,6 @@ contract AddressImpl {
...
@@ -32,11 +32,6 @@ contract AddressImpl {
emit CallReturnValue(abi.decode(returnData, (string)));
emit CallReturnValue(abi.decode(returnData, (string)));
}
}
function functionDelegateCall(address target, bytes calldata data) external {
bytes memory returnData = Address.functionDelegateCall(target, data);
emit CallReturnValue(abi.decode(returnData, (string)));
}
// sendValue's tests require the contract to hold Ether
// sendValue's tests require the contract to hold Ether
receive () external payable { }
receive () external payable { }
}
}
contracts/proxy/ERC1967/ERC1967Upgrade.sol
View file @
be866876
...
@@ -50,7 +50,7 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
...
@@ -50,7 +50,7 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
_setImplementation(newImplementation);
_setImplementation(newImplementation);
emit Upgraded(newImplementation);
emit Upgraded(newImplementation);
if (data.length > 0 || forceCall) {
if (data.length > 0 || forceCall) {
Address.
functionDelegateCall(newImplementation, data);
_
functionDelegateCall(newImplementation, data);
}
}
}
}
...
@@ -65,7 +65,7 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
...
@@ -65,7 +65,7 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
// Initial upgrade and setup call
// Initial upgrade and setup call
_setImplementation(newImplementation);
_setImplementation(newImplementation);
if (data.length > 0 || forceCall) {
if (data.length > 0 || forceCall) {
Address.
functionDelegateCall(newImplementation, data);
_
functionDelegateCall(newImplementation, data);
}
}
// Perform rollback test if not already in progress
// Perform rollback test if not already in progress
...
@@ -73,7 +73,7 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
...
@@ -73,7 +73,7 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
if (!rollbackTesting.value) {
if (!rollbackTesting.value) {
// Trigger rollback using upgradeTo from the new implementation
// Trigger rollback using upgradeTo from the new implementation
rollbackTesting.value = true;
rollbackTesting.value = true;
Address.
functionDelegateCall(
_
functionDelegateCall(
newImplementation,
newImplementation,
abi.encodeWithSignature(
abi.encodeWithSignature(
"upgradeTo(address)",
"upgradeTo(address)",
...
@@ -99,7 +99,7 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
...
@@ -99,7 +99,7 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
_setBeacon(newBeacon);
_setBeacon(newBeacon);
emit BeaconUpgraded(newBeacon);
emit BeaconUpgraded(newBeacon);
if (data.length > 0 || forceCall) {
if (data.length > 0 || forceCall) {
Address.
functionDelegateCall(IBeacon(newBeacon).implementation(), data);
_
functionDelegateCall(IBeacon(newBeacon).implementation(), data);
}
}
}
}
...
@@ -112,4 +112,37 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
...
@@ -112,4 +112,37 @@ abstract contract ERC1967Upgrade is ERC1967Storage {
emit AdminChanged(_getAdmin(), newAdmin);
emit AdminChanged(_getAdmin(), newAdmin);
_setAdmin(newAdmin);
_setAdmin(newAdmin);
}
}
/**
* @dev Same as {xref-Address-functionCall-address-bytes-string-}[`functionCall`],
* but performing a delegate call.
*
* _Available since v3.4._
*/
function _functionDelegateCall(address target, bytes memory data) private returns (bytes memory) {
require(Address.isContract(target), "Address: delegate call to non-contract");
// solhint-disable-next-line avoid-low-level-calls
(bool success, bytes memory returndata) = target.delegatecall(data);
return _verifyCallResult(success, returndata, "Address: low-level delegate call failed");
}
function _verifyCallResult(bool success, bytes memory returndata, string memory errorMessage) private pure returns(bytes memory) {
if (success) {
return returndata;
} else {
// Look for revert reason and bubble it up if present
if (returndata.length > 0) {
// The easiest way to bubble the revert reason is using memory via assembly
// solhint-disable-next-line no-inline-assembly
assembly {
let returndata_size := mload(returndata)
revert(add(32, returndata), returndata_size)
}
} else {
revert(errorMessage);
}
}
}
}
}
contracts/utils/Address.sol
View file @
be866876
...
@@ -144,30 +144,6 @@ library Address {
...
@@ -144,30 +144,6 @@ library Address {
return _verifyCallResult(success, returndata, errorMessage);
return _verifyCallResult(success, returndata, errorMessage);
}
}
/**
* @dev Same as {xref-Address-functionCall-address-bytes-}[`functionCall`],
* but performing a delegate call.
*
* _Available since v3.4._
*/
function functionDelegateCall(address target, bytes memory data) internal returns (bytes memory) {
return functionDelegateCall(target, data, "Address: low-level delegate call failed");
}
/**
* @dev Same as {xref-Address-functionCall-address-bytes-string-}[`functionCall`],
* but performing a delegate call.
*
* _Available since v3.4._
*/
function functionDelegateCall(address target, bytes memory data, string memory errorMessage) internal returns (bytes memory) {
require(isContract(target), "Address: delegate call to non-contract");
// solhint-disable-next-line avoid-low-level-calls
(bool success, bytes memory returndata) = target.delegatecall(data);
return _verifyCallResult(success, returndata, errorMessage);
}
function _verifyCallResult(bool success, bytes memory returndata, string memory errorMessage) private pure returns(bytes memory) {
function _verifyCallResult(bool success, bytes memory returndata, string memory errorMessage) private pure returns(bytes memory) {
if (success) {
if (success) {
return returndata;
return returndata;
...
...
contracts/utils/Multicall.sol
View file @
be866876
...
@@ -16,8 +16,41 @@ abstract contract Multicall {
...
@@ -16,8 +16,41 @@ abstract contract Multicall {
function multicall(bytes[] calldata data) external returns (bytes[] memory results) {
function multicall(bytes[] calldata data) external returns (bytes[] memory results) {
results = new bytes[](data.length);
results = new bytes[](data.length);
for (uint i = 0; i < data.length; i++) {
for (uint i = 0; i < data.length; i++) {
results[i] =
Address.
functionDelegateCall(address(this), data[i]);
results[i] =
_
functionDelegateCall(address(this), data[i]);
}
}
return results;
return results;
}
}
/**
* @dev Same as {xref-Address-functionCall-address-bytes-string-}[`functionCall`],
* but performing a delegate call.
*
* _Available since v3.4._
*/
function _functionDelegateCall(address target, bytes memory data) private returns (bytes memory) {
require(Address.isContract(target), "Address: delegate call to non-contract");
// solhint-disable-next-line avoid-low-level-calls
(bool success, bytes memory returndata) = target.delegatecall(data);
return _verifyCallResult(success, returndata, "Address: low-level delegate call failed");
}
function _verifyCallResult(bool success, bytes memory returndata, string memory errorMessage) private pure returns(bytes memory) {
if (success) {
return returndata;
} else {
// Look for revert reason and bubble it up if present
if (returndata.length > 0) {
// The easiest way to bubble the revert reason is using memory via assembly
// solhint-disable-next-line no-inline-assembly
assembly {
let returndata_size := mload(returndata)
revert(add(32, returndata), returndata_size)
}
} else {
revert(errorMessage);
}
}
}
}
}
scripts/upgradeable/patch/01-inline-address-functiondelegatecall.patch
deleted
100644 → 0
View file @
f0ec852a
diff --git a/contracts/mocks/AddressImplUpgradeable.sol b/contracts/mocks/AddressImplUpgradeable.sol
index 4e92f588..4cbcabb8 100644
--- a/contracts/mocks/AddressImplUpgradeable.sol
+++ b/contracts/mocks/AddressImplUpgradeable.sol
@@ -39,11 +39,6 @@ contract AddressImplUpgradeable is Initializable {
emit CallReturnValue(abi.decode(returnData, (string)));
}
- function functionDelegateCall(address target, bytes calldata data) external {
- bytes memory returnData = AddressUpgradeable.functionDelegateCall(target, data);
- emit CallReturnValue(abi.decode(returnData, (string)));
- }
-
// sendValue's tests require the contract to hold Ether
receive () external payable { }
uint256[49] private __gap;
diff --git a/contracts/utils/AddressUpgradeable.sol b/contracts/utils/AddressUpgradeable.sol
index 89241895..99c1ce78 100644
--- a/contracts/utils/AddressUpgradeable.sol
+++ b/contracts/utils/AddressUpgradeable.sol
@@ -144,30 +144,6 @@ library AddressUpgradeable {
return _verifyCallResult(success, returndata, errorMessage);
}
- /**
- * @dev Same as {xref-Address-functionCall-address-bytes-}[`functionCall`],
- * but performing a delegate call.
- *
- * _Available since v3.4._
- */
- function functionDelegateCall(address target, bytes memory data) internal returns (bytes memory) {
- return functionDelegateCall(target, data, "Address: low-level delegate call failed");
- }
-
- /**
- * @dev Same as {xref-Address-functionCall-address-bytes-string-}[`functionCall`],
- * but performing a delegate call.
- *
- * _Available since v3.4._
- */
- function functionDelegateCall(address target, bytes memory data, string memory errorMessage) internal returns (bytes memory) {
- require(isContract(target), "Address: delegate call to non-contract");
-
- // solhint-disable-next-line avoid-low-level-calls
- (bool success, bytes memory returndata) = target.delegatecall(data);
- return _verifyCallResult(success, returndata, errorMessage);
- }
-
function _verifyCallResult(bool success, bytes memory returndata, string memory errorMessage) private pure returns(bytes memory) {
if (success) {
return returndata;
diff --git a/contracts/utils/MulticallUpgradeable.sol b/contracts/utils/MulticallUpgradeable.sol
index 62d5047c..a535dfc5 100644
--- a/contracts/utils/MulticallUpgradeable.sol
+++ b/contracts/utils/MulticallUpgradeable.sol
@@ -21,9 +21,37 @@ abstract contract MulticallUpgradeable is Initializable {
function multicall(bytes[] calldata data) external returns (bytes[] memory results) {
results = new bytes[](data.length);
for (uint i = 0; i < data.length; i++) {
- results[i] = AddressUpgradeable.functionDelegateCall(address(this), data[i]);
+ results[i] = _functionDelegateCall(address(this), data[i]);
}
return results;
}
+
+ function _functionDelegateCall(address target, bytes memory data) private returns (bytes memory) {
+ require(AddressUpgradeable.isContract(target), "Address: delegate call to non-contract");
+
+ // solhint-disable-next-line avoid-low-level-calls
+ (bool success, bytes memory returndata) = target.delegatecall(data);
+ return _verifyCallResult(success, returndata, "Address: low-level delegate call failed");
+ }
+
+ function _verifyCallResult(bool success, bytes memory returndata, string memory errorMessage) private pure returns(bytes memory) {
+ if (success) {
+ return returndata;
+ } else {
+ // Look for revert reason and bubble it up if present
+ if (returndata.length > 0) {
+ // The easiest way to bubble the revert reason is using memory via assembly
+
+ // solhint-disable-next-line no-inline-assembly
+ assembly {
+ let returndata_size := mload(returndata)
+ revert(add(32, returndata), returndata_size)
+ }
+ } else {
+ revert(errorMessage);
+ }
+ }
+ }
+
uint256[50] private __gap;
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment