-
Notifications
You must be signed in to change notification settings - Fork 214
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
30 changed files
with
975 additions
and
254 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,133 @@ | ||
| // Copyright (C) 2018 Argent Labs Ltd. <https://argent.xyz> | ||
|
|
||
| // This program is free software: you can redistribute it and/or modify | ||
| // it under the terms of the GNU General Public License as published by | ||
| // the Free Software Foundation, either version 3 of the License, or | ||
| // (at your option) any later version. | ||
|
|
||
| // This program is distributed in the hope that it will be useful, | ||
| // but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| // GNU General Public License for more details. | ||
|
|
||
| // You should have received a copy of the GNU General Public License | ||
| // along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
|
|
||
| // SPDX-License-Identifier: GPL-3.0-only | ||
| pragma solidity ^0.6.12; | ||
| import "./base/Owned.sol"; | ||
|
|
||
| interface IArgentWallet { | ||
| /** | ||
| * @notice Returns the implementation of the wallet. | ||
| * @return The wallet implementation. | ||
| */ | ||
| function implementation() external view returns (address); | ||
| } | ||
|
|
||
| /** | ||
| * @title ArgentWalletDetector | ||
| * @notice Simple contract to detect if a given address represents an Argent wallet. | ||
| * The `isArgentWallet` method returns true if the codehash matches one of the deployed Proxy | ||
| * and if the target implementation matches one of the deployed BaseWallet. | ||
| * Only the owner of the contract can add code hash and implementations. | ||
| * @author Julien Niset - <[email protected]> | ||
| */ | ||
| contract ArgentWalletDetector is Owned { | ||
|
|
||
| // The accepted code hashes | ||
| bytes32[] private codes; | ||
| // The accepted implementations | ||
| address[] private implementations; | ||
| // mapping to efficiently check if a code is accepted | ||
| mapping (bytes32 => Info) public acceptedCodes; | ||
| // mapping to efficiently check is an implementation is accepted | ||
| mapping (address => Info) public acceptedImplementations; | ||
|
|
||
| struct Info { | ||
| bool exists; | ||
| uint128 index; | ||
| } | ||
|
|
||
| // emits when a new accepted code is added | ||
| event CodeAdded(bytes32 indexed code); | ||
| // emits when a new accepted implementation is added | ||
| event ImplementationAdded(address indexed implementation); | ||
|
|
||
| constructor(bytes32[] memory _codes, address[] memory _implementations) public { | ||
| for(uint i = 0; i < _codes.length; i++) { | ||
| addCode(_codes[i]); | ||
| } | ||
| for(uint j = 0; j < _implementations.length; j++) { | ||
| addImplementation(_implementations[j]); | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * @notice Adds a new accepted code hash. | ||
| * @param _code The new code hash. | ||
| */ | ||
| function addCode(bytes32 _code) public onlyOwner { | ||
| require(_code != bytes32(0), "AWR: empty _code"); | ||
| Info storage code = acceptedCodes[_code]; | ||
| if(!code.exists) { | ||
| codes.push(_code); | ||
| code.exists = true; | ||
| code.index = uint128(codes.length - 1); | ||
| emit CodeAdded(_code); | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * @notice Adds a new accepted implementation. | ||
| * @param _impl The new implementation. | ||
| */ | ||
| function addImplementation(address _impl) public onlyOwner { | ||
| require(_impl != address(0), "AWR: empty _impl"); | ||
| Info storage impl = acceptedImplementations[_impl]; | ||
| if(!impl.exists) { | ||
| implementations.push(_impl); | ||
| impl.exists = true; | ||
| impl.index = uint128(implementations.length - 1); | ||
| emit ImplementationAdded(_impl); | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * @notice Adds a new accepted code hash and implementation from a deployed Argent wallet. | ||
| * @param _argentWallet The deployed Argent wallet. | ||
| */ | ||
| function addCodeAndImplementationFromWallet(address _argentWallet) external onlyOwner { | ||
| bytes32 codeHash; | ||
| // solhint-disable-next-line no-inline-assembly | ||
| assembly { codeHash := extcodehash(_argentWallet) } | ||
| addCode(codeHash); | ||
| address implementation = IArgentWallet(_argentWallet).implementation(); | ||
| addImplementation(implementation); | ||
| } | ||
|
|
||
| /** | ||
| * @notice Gets the list of accepted implementations. | ||
| */ | ||
| function getImplementations() public view returns (address[] memory) { | ||
| return implementations; | ||
| } | ||
|
|
||
| /** | ||
| * @notice Gets the list of accepted code hash. | ||
| */ | ||
| function getCodes() public view returns (bytes32[] memory) { | ||
| return codes; | ||
| } | ||
|
|
||
| /** | ||
| * @notice Checks if an address is an Argent wallet | ||
| * @param _wallet The target wallet | ||
| */ | ||
| function isArgentWallet(address _wallet) external view returns (bool) { | ||
| bytes32 codeHash; | ||
| // solhint-disable-next-line no-inline-assembly | ||
| assembly { codeHash := extcodehash(_wallet) } | ||
| return acceptedCodes[codeHash].exists && acceptedImplementations[IArgentWallet(_wallet).implementation()].exists; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.