86 lines
3.1 KiB
Solidity
86 lines
3.1 KiB
Solidity
pragma solidity ^0.4.18;
|
|
|
|
/*
|
|
Copyright 2017, Jordi Baylina, RJ Ewing
|
|
Contributors: Adrià Massanet <adria@codecontext.io>, Griff Green,
|
|
Arthur Lunn
|
|
|
|
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/>.
|
|
*/
|
|
|
|
import "@aragon/os/contracts/apps/AragonApp.sol";
|
|
import "./LiquidPledgingStorage.sol";
|
|
import "./LiquidPledgingACLHelpers.sol";
|
|
|
|
contract LiquidPledgingPlugins is AragonApp, LiquidPledgingStorage, LiquidPledgingACLHelpers {
|
|
|
|
bytes32 constant public PLUGIN_MANAGER_ROLE = keccak256("PLUGIN_MANAGER_ROLE");
|
|
|
|
function addValidPluginInstance(address addr) auth(PLUGIN_MANAGER_ROLE) public {
|
|
pluginInstanceWhitelist[addr] = true;
|
|
}
|
|
|
|
function addValidPluginContract(bytes32 contractHash) auth(PLUGIN_MANAGER_ROLE) public {
|
|
pluginContractWhitelist[contractHash] = true;
|
|
}
|
|
|
|
function addValidPluginContracts(bytes32[] contractHashes) external auth(PLUGIN_MANAGER_ROLE) {
|
|
for (uint8 i = 0; i < contractHashes.length; i++) {
|
|
addValidPluginContract(contractHashes[i]);
|
|
}
|
|
}
|
|
|
|
function removeValidPluginContract(bytes32 contractHash) external authP(PLUGIN_MANAGER_ROLE, arr(contractHash)) {
|
|
pluginContractWhitelist[contractHash] = false;
|
|
}
|
|
|
|
function removeValidPluginInstance(address addr) external auth(PLUGIN_MANAGER_ROLE) {
|
|
pluginInstanceWhitelist[addr] = false;
|
|
}
|
|
|
|
function useWhitelist(bool useWhitelist) external auth(PLUGIN_MANAGER_ROLE) {
|
|
whitelistDisabled = !useWhitelist;
|
|
}
|
|
|
|
function isValidPlugin(address addr) public view returns(bool) {
|
|
if (whitelistDisabled || addr == 0x0) {
|
|
return true;
|
|
}
|
|
|
|
// first check pluginInstances
|
|
if (pluginInstanceWhitelist[addr]) {
|
|
return true;
|
|
}
|
|
|
|
// if the addr isn't a valid instance, check the contract code
|
|
bytes32 contractHash = getCodeHash(addr);
|
|
|
|
return pluginContractWhitelist[contractHash];
|
|
}
|
|
|
|
function getCodeHash(address addr) public view returns(bytes32) {
|
|
bytes memory o_code;
|
|
assembly {
|
|
// retrieve the size of the code, this needs assembly
|
|
let size := extcodesize(addr)
|
|
// allocate output byte array - this could also be done without assembly
|
|
// by using o_code = new bytes(size)
|
|
o_code := mload(0x40)
|
|
mstore(o_code, size) // store length in memory
|
|
// actually retrieve the code, this needs assembly
|
|
extcodecopy(addr, add(o_code, 0x20), 0, size)
|
|
}
|
|
return keccak256(o_code);
|
|
}
|
|
} |