work on stufF
Signed-off-by: AlienTornadosaurusHex <>
This commit is contained in:
parent
735546619e
commit
cd2b889c2f
@ -2,7 +2,7 @@
|
||||
|
||||
This is my personal reconstruction of the Tornado `relayer-registry` repository which, although there is one being hosted with the same name on Github, is not actually the most up-to-date version of the repository, meaning that the contracts in that one are not the deployed ones.
|
||||
|
||||
The contract information here was taken directly off of Etherscan and the necessary dependencies added, everything organized. It is advisable to use MacOS or Linux for this repo, and the setup goes as follows:
|
||||
The contract information here was taken directly off of Etherscan and the necessary dependencies added, everything organized. It is advisable to use a UNIX compatible system for this repo, and the setup goes as follows:
|
||||
|
||||
```bash
|
||||
git clone https://development.tornadocash.community/thcydra/relayer-registry
|
||||
@ -10,7 +10,7 @@ cd relayer-registry
|
||||
yarn
|
||||
forge install
|
||||
yarn build
|
||||
forge -b # or forge -b --use 0.6.12
|
||||
forge build
|
||||
```
|
||||
|
||||
`yarn build` has to execute a script to modify a single Uniswap dependency because of a stupid version pragma, if you check it out (in `script/setup.sh`) you will just notice that it goes into the file mentioned and changes the version pragma from 0.7.0 to 0.6.12. That is all of the changes needed.
|
||||
|
33
foundry.toml
33
foundry.toml
@ -1,4 +1,23 @@
|
||||
[profile.default]
|
||||
# General
|
||||
src = 'src'
|
||||
out = 'out'
|
||||
libs = ["node_modules", "lib"]
|
||||
|
||||
# Compiler
|
||||
auto_detect_solc = true
|
||||
via_ir = true
|
||||
optimizer = true
|
||||
optimizer-runs = 1
|
||||
|
||||
# Network
|
||||
chain_id = 1
|
||||
rpc_endpoints = { mainnet = "${MAINNET_RPC_URL}" }
|
||||
|
||||
# Tests
|
||||
verbosity = 2
|
||||
|
||||
# Remappings
|
||||
remappings = [
|
||||
'ds-test/=lib/ds-test/src/',
|
||||
'solmate/=lib/solmate/src/',
|
||||
@ -10,3 +29,17 @@ remappings = [
|
||||
'@uniswap/v3-periphery/=lib/v3-periphery/',
|
||||
'@uniswap/v3-core/=lib/v3-core/',
|
||||
]
|
||||
|
||||
[fmt]
|
||||
line_length = 110
|
||||
number_underscore = 'thousands'
|
||||
bracket_spacing = true
|
||||
wrap_comments = true
|
||||
ignore = [
|
||||
"./src/v1/*",
|
||||
"./src/v1/interfaces/*",
|
||||
"./src/v1/libraries/*",
|
||||
"./src/v1/staking/*",
|
||||
"./src/v1/tornado-proxy/*",
|
||||
"./src/v1/utils/*"
|
||||
]
|
10
package.json
10
package.json
@ -1,8 +1,12 @@
|
||||
{
|
||||
"name": "@T-Hax/relayer-registry",
|
||||
"author": "T-Hax",
|
||||
"name": "@AlienTornadosaurusHex/relayer-registry",
|
||||
"author": "AlienTornadosaurusHex",
|
||||
"version": "1.0.0",
|
||||
"description": "Foundry reconstruction of the \"relayer-registry\" project.",
|
||||
"repository": {
|
||||
"type": "git",
|
||||
"url": "https://git.tornado.ws/AlienTornadosaurusHex/relayer-registry"
|
||||
},
|
||||
"scripts": {
|
||||
"postinstall": "chmod +x script/* && ./script/setup.sh",
|
||||
"build": "chmod +x script/* && ./script/setup.sh",
|
||||
@ -20,4 +24,4 @@
|
||||
"prettier-plugin-solidity": "^1.0.0-beta.19",
|
||||
"solhint": "^3.3.6"
|
||||
}
|
||||
}
|
||||
}
|
236
src/v2/FeeOracleManager.sol
Normal file
236
src/v2/FeeOracleManager.sol
Normal file
@ -0,0 +1,236 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.6.12;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
// OZ Imports
|
||||
|
||||
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
|
||||
import { Initializable } from "@openzeppelin/contracts/proxy/Initializable.sol";
|
||||
|
||||
// Tornado Imports
|
||||
|
||||
import { ITornadoInstance } from "tornado-anonymity-mining/contracts/interfaces/ITornadoInstance.sol";
|
||||
|
||||
// Local Imports
|
||||
|
||||
import { InstanceRegistry } from "../v1/tornado-proxy/InstanceRegistry.sol";
|
||||
|
||||
import { IFeeOracle } from "./interfaces/IFeeOracle.sol";
|
||||
|
||||
struct FeeData {
|
||||
uint160 feeAmount;
|
||||
uint64 feePercent;
|
||||
uint32 lastUpdated;
|
||||
}
|
||||
|
||||
/**
|
||||
* @title FeeManagerLegacyStorage
|
||||
* @author AlienTornadosaurusHex
|
||||
* @dev This is contract will help us layout storage properly for a proxy upgrade for the impl
|
||||
* FeeOracleManager (formerly FeeManager).
|
||||
*/
|
||||
contract FeeManagerLegacyStorage {
|
||||
uint24 public deprecatedUniswapTornPoolSwappingFee;
|
||||
uint32 public deprecatedUniswapTimePeriod;
|
||||
|
||||
uint24 public feeUpdateInterval;
|
||||
|
||||
mapping(ITornadoInstance => uint160) public oldFeesForInstance;
|
||||
mapping(ITornadoInstance => uint256) public oldFeesForInstanceUpdateTime;
|
||||
}
|
||||
|
||||
/**
|
||||
* @title FeeOracleManager
|
||||
* @author AlienTornadosaurusHex
|
||||
* @notice A contract which manages fee oracles and received data for other contracts to consume.
|
||||
* @dev This is an improved version of the FeeManager with a modified design from the original contract.
|
||||
*/
|
||||
contract FeeOracleManager is FeeManagerLegacyStorage, Initializable {
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ VARIABLES ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
/* @dev Divide protocol fee by this to get the percent value */
|
||||
uint64 public constant PROTOCOL_FEE_DIVISOR = 1 ether;
|
||||
|
||||
/* @dev The address of the TORN token */
|
||||
address public immutable tornTokenAddress;
|
||||
|
||||
/* @dev The Governance Proxy address */
|
||||
address public immutable governanceProxyAddress;
|
||||
|
||||
/* @dev The InstanceRegistry contract */
|
||||
InstanceRegistry public instanceRegistry;
|
||||
|
||||
/* @dev Each instance has a dedicated fee oracle, these only compute the values */
|
||||
mapping(ITornadoInstance => IFeeOracle) public instanceFeeOracles;
|
||||
|
||||
/* @dev The data for each instance will be stored in this contract */
|
||||
mapping(ITornadoInstance => FeeData) public feeDataForInstance;
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ EVENTS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
event FeeUpdated(address indexed instance, uint256 newFee);
|
||||
event NewOracleSet(address indexed instance, address oracle);
|
||||
event NewFeeUpdateIntervalSet(uint24 limit);
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ LOGIC ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
constructor(address _tornTokenAddress, address _governanceProxyAddress) public {
|
||||
tornTokenAddress = _tornTokenAddress;
|
||||
governanceProxyAddress = _governanceProxyAddress;
|
||||
}
|
||||
|
||||
modifier onlyGovernance() {
|
||||
require(msg.sender == governanceProxyAddress, "FeeOracleManager: onlyGovernance");
|
||||
_;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev If there will be a need to initialize the proxy again, simply pad storage and inherit again,
|
||||
* making sure to not reference old data anywhere.
|
||||
*/
|
||||
function initialize(
|
||||
address _uniswapPoolFeeOracle,
|
||||
address _instanceRegistryAddress,
|
||||
address[] calldata _instanceAddresses,
|
||||
uint256[] calldata _feePercents
|
||||
) external onlyGovernance initializer {
|
||||
uint256 numInstances = _instanceAddresses.length;
|
||||
|
||||
for (uint256 i = 0; i < numInstances; i++) {
|
||||
ITornadoInstance instance = ITornadoInstance(_instanceAddresses[i]);
|
||||
|
||||
feeDataForInstance[instance] = FeeData({
|
||||
feeAmount: oldFeesForInstance[instance],
|
||||
feePercent: uint64(_feePercents[i]),
|
||||
lastUpdated: uint32(oldFeesForInstanceUpdateTime[instance])
|
||||
});
|
||||
|
||||
instanceFeeOracles[instance] = IFeeOracle(_uniswapPoolFeeOracle);
|
||||
}
|
||||
|
||||
instanceRegistry = InstanceRegistry(_instanceRegistryAddress);
|
||||
}
|
||||
|
||||
function instanceFeeWithUpdate(ITornadoInstance _instance) public virtual returns (uint160) {
|
||||
FeeData memory feeData = feeDataForInstance[_instance];
|
||||
|
||||
if (feeUpdateInterval < now - feeData.lastUpdated) {
|
||||
feeData.feeAmount = updateFee(_instance);
|
||||
}
|
||||
|
||||
return feeData.feeAmount;
|
||||
}
|
||||
|
||||
function updateAllFees() public virtual returns (uint160[] memory newFees) {
|
||||
return updateFees(instanceRegistry.getAllInstanceAddresses());
|
||||
}
|
||||
|
||||
function updateFees(ITornadoInstance[] memory _instances)
|
||||
public
|
||||
virtual
|
||||
returns (uint160[] memory newFees)
|
||||
{
|
||||
uint256 numInstances = _instances.length;
|
||||
|
||||
newFees = new uint160[](numInstances);
|
||||
|
||||
for (uint256 i = 0; i < numInstances; i++) {
|
||||
newFees[i] = updateFee(_instances[i]);
|
||||
}
|
||||
}
|
||||
|
||||
function updateFee(ITornadoInstance _instance) public virtual returns (uint160 newFee) {
|
||||
// This will revert if no contract is set
|
||||
newFee = instanceFeeOracles[_instance].getFee();
|
||||
|
||||
feeDataForInstance[_instance] = FeeData({
|
||||
feeAmount: newFee,
|
||||
feePercent: feeDataForInstance[_instance].feePercent,
|
||||
lastUpdated: uint32(now)
|
||||
});
|
||||
|
||||
emit FeeUpdated(address(_instance), newFee);
|
||||
|
||||
return newFee;
|
||||
}
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ SETTERS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
function setInstanceRegistry(address _newInstanceRegistryAddress) external onlyGovernance {
|
||||
instanceRegistry = InstanceRegistry(_newInstanceRegistryAddress);
|
||||
}
|
||||
|
||||
function setFeeOracle(address _instanceAddress, address _oracleAddress) external onlyGovernance {
|
||||
ITornadoInstance instance = ITornadoInstance(_instanceAddress);
|
||||
IFeeOracle oracle = IFeeOracle(_oracleAddress);
|
||||
|
||||
// Reverts if no oracle or does not conform to interface
|
||||
uint160 fee = oracle.getFee();
|
||||
|
||||
// Ok, set the oracle
|
||||
instanceFeeOracles[instance] = oracle;
|
||||
|
||||
// Note down updated fee
|
||||
feeDataForInstance[instance] = FeeData({
|
||||
feeAmount: fee,
|
||||
feePercent: feeDataForInstance[instance].feePercent,
|
||||
lastUpdated: uint32(now)
|
||||
});
|
||||
|
||||
emit NewOracleSet(_instanceAddress, _oracleAddress);
|
||||
emit FeeUpdated(_instanceAddress, fee);
|
||||
}
|
||||
|
||||
function setFeePercentForInstance(ITornadoInstance _instance, uint64 _newFeePercent)
|
||||
external
|
||||
onlyGovernance
|
||||
{
|
||||
feeDataForInstance[_instance].feePercent = _newFeePercent;
|
||||
}
|
||||
|
||||
function setFeeUpdateInterval(uint24 newLimit) external onlyGovernance {
|
||||
feeUpdateInterval = newLimit;
|
||||
emit NewFeeUpdateIntervalSet(newLimit);
|
||||
}
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ GETTERS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
function version() public pure virtual returns (string memory) {
|
||||
return "v2-oracle-manager";
|
||||
}
|
||||
|
||||
function getUpdatedFeeForInstance(ITornadoInstance instance) public view virtual returns (uint160) {
|
||||
return instanceFeeOracles[instance].getFee();
|
||||
}
|
||||
|
||||
function getLastFeeForInstance(ITornadoInstance instance) public view virtual returns (uint160) {
|
||||
return uint160(feeDataForInstance[instance].feeAmount);
|
||||
}
|
||||
|
||||
function getLastUpdatedTimeForInstance(ITornadoInstance instance) public view virtual returns (uint32) {
|
||||
return feeDataForInstance[instance].lastUpdated;
|
||||
}
|
||||
|
||||
function getFeeDeviations() public view virtual returns (int256[] memory results) {
|
||||
ITornadoInstance[] memory instances = instanceRegistry.getAllInstanceAddresses();
|
||||
|
||||
uint256 numInstances = instances.length;
|
||||
|
||||
results = new int256[](numInstances);
|
||||
|
||||
for (uint256 i = 0; i < numInstances; i++) {
|
||||
ITornadoInstance instance = instances[i];
|
||||
|
||||
uint256 marketFee = instanceFeeOracles[instance].getFee();
|
||||
|
||||
int256 deviation;
|
||||
|
||||
if (marketFee != 0) {
|
||||
deviation = int256((feeDataForInstance[instance].feeAmount * 1000) / marketFee) - 1000;
|
||||
}
|
||||
|
||||
results[i] = deviation;
|
||||
}
|
||||
}
|
||||
}
|
221
src/v2/InstanceRegistry.sol
Normal file
221
src/v2/InstanceRegistry.sol
Normal file
@ -0,0 +1,221 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.6.12;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
// OZ Imports
|
||||
|
||||
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
|
||||
import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
|
||||
import { Initializable } from "@openzeppelin/contracts/proxy/Initializable.sol";
|
||||
|
||||
// Tornado Imports
|
||||
|
||||
import { EnsResolve } from "torn-token/contracts/ENS.sol";
|
||||
import { ITornadoInstance } from "tornado-anonymity-mining/contracts/interfaces/ITornadoInstance.sol";
|
||||
|
||||
// Local imports
|
||||
|
||||
import { TornadoRouter } from "./TornadoRouter.sol";
|
||||
|
||||
/**
|
||||
* @title InstanceRegistry
|
||||
* @author AlienTornadosaurusHex
|
||||
* @notice A contract which enumerates Tornado Cash pool instances, and also stores essential data regarding
|
||||
* them.
|
||||
* @dev This is contract will help us layout storage properly for a proxy upgrade for the impl
|
||||
* InstanceRegistry.
|
||||
*/
|
||||
contract InstanceRegistryLegacyStorage {
|
||||
/* From first contract */
|
||||
enum LegacyStatePlaceholder {
|
||||
DISABLED,
|
||||
ENABLED
|
||||
}
|
||||
|
||||
/* From first contract */
|
||||
struct LegacyInstanceStructPlaceholder {
|
||||
bool deprecatedIsERC20;
|
||||
address deprecatedToken;
|
||||
LegacyStatePlaceholder deprecatedState;
|
||||
uint24 deprecatedUniswapPoolSwappingFee;
|
||||
uint32 deprecatedProtocolFeePercentage;
|
||||
}
|
||||
|
||||
/* From Initializable.sol of first contract */
|
||||
bool private _deprecatedInitialized;
|
||||
|
||||
/* From Initializable.sol of first contract */
|
||||
bool private _deprecatedInitializing;
|
||||
|
||||
/* From first contract */
|
||||
address private _deprecatedRouterAddress;
|
||||
|
||||
/* From first contract */
|
||||
mapping(address => LegacyInstanceStructPlaceholder) private _deprecatedInstances;
|
||||
|
||||
/* From first contract */
|
||||
ITornadoInstance[] private _deprecatedInstanceIds;
|
||||
}
|
||||
|
||||
struct InstanceData {
|
||||
IERC20 token;
|
||||
uint80 index;
|
||||
bool isERC20;
|
||||
bool isEnabled;
|
||||
}
|
||||
|
||||
/**
|
||||
* @title InstanceRegistry
|
||||
* @author AlienTornadosaurusHex
|
||||
* @notice A contract which enumerates Tornado Cash pool instances, and also stores essential data regarding
|
||||
* them.
|
||||
* @dev This is an improved version of the InstanceRegistry with a modified design from the original contract.
|
||||
*/
|
||||
contract InstanceRegistry is InstanceRegistryLegacyStorage, EnsResolve, Initializable {
|
||||
using SafeERC20 for IERC20;
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ VARIABLES ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
/* @dev The address of the Governance proxy */
|
||||
address public immutable governanceProxyAddress;
|
||||
|
||||
/* @dev Essential data regarding instances, see struct above */
|
||||
mapping(ITornadoInstance => InstanceData) public instanceData;
|
||||
|
||||
/* @dev All instances enumerable */
|
||||
ITornadoInstance[] public instances;
|
||||
|
||||
/* @dev The router which processes txs which we must command to approve tokens */
|
||||
TornadoRouter public router;
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ EVENTS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
event NewRouterRegistered(address newRouterAddress);
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ LOGIC ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
constructor(address _governanceProxyAddress) public {
|
||||
governanceProxyAddress = _governanceProxyAddress;
|
||||
}
|
||||
|
||||
modifier onlyGovernance() {
|
||||
require(msg.sender == governanceProxyAddress, "InstanceRegistry: onlyGovernance");
|
||||
_;
|
||||
}
|
||||
|
||||
function initialize(address[] memory _instanceAddresses, address _router)
|
||||
external
|
||||
onlyGovernance
|
||||
initializer
|
||||
{
|
||||
uint256 numInstances = _instanceAddresses.length;
|
||||
|
||||
router = TornadoRouter(_router);
|
||||
|
||||
for (uint256 i = 0; i < numInstances; i++) {
|
||||
addInstance(ITornadoInstance(_instanceAddresses[i]));
|
||||
}
|
||||
}
|
||||
|
||||
function addInstance(ITornadoInstance _instance) public virtual onlyGovernance {
|
||||
bool isEnabled = instanceData[_instance].isEnabled;
|
||||
bool isERC20 = false;
|
||||
|
||||
IERC20 token;
|
||||
|
||||
// ETH instances do not know of a `token()` call
|
||||
try _instance.token() returns (address _tokenAddress) {
|
||||
token = IERC20(_tokenAddress);
|
||||
isERC20 = true;
|
||||
} catch {
|
||||
/* It's an ETH instance, do nothing */
|
||||
}
|
||||
|
||||
require(!isEnabled, "InstanceRegistry: can't add the same instance.");
|
||||
|
||||
if (isERC20) {
|
||||
uint256 routerAllowanceForInstance = token.allowance(address(router), address(_instance));
|
||||
|
||||
if (routerAllowanceForInstance == 0) {
|
||||
router.approveTokenForInstance(token, address(_instance), type(uint256).max);
|
||||
}
|
||||
}
|
||||
|
||||
instances.push(_instance);
|
||||
|
||||
instanceData[_instance] = InstanceData({
|
||||
token: token,
|
||||
index: uint64(instances.length - 1),
|
||||
isERC20: token != IERC20(address(0)), // TODO: Is this true?
|
||||
isEnabled: isEnabled
|
||||
});
|
||||
}
|
||||
|
||||
function removeInstanceByIndex(uint256 _instanceIndex) public virtual {
|
||||
_removeInstanceByAddress(address(instances[_instanceIndex]));
|
||||
}
|
||||
|
||||
function removeInstanceByAddress(address _instanceAddress) public virtual {
|
||||
_removeInstanceByAddress(_instanceAddress);
|
||||
}
|
||||
|
||||
function _removeInstanceByAddress(address _instanceAddress) internal virtual onlyGovernance {
|
||||
ITornadoInstance instance = ITornadoInstance(_instanceAddress);
|
||||
InstanceData memory data = instanceData[instance];
|
||||
|
||||
if (data.isERC20) {
|
||||
uint256 routerAllowanceForInstance = data.token.allowance(address(router), _instanceAddress);
|
||||
|
||||
if (routerAllowanceForInstance != 0) {
|
||||
router.approveTokenForInstance(data.token, _instanceAddress, 0);
|
||||
}
|
||||
}
|
||||
|
||||
instances[data.index] = instances[instances.length - 1];
|
||||
instances.pop();
|
||||
|
||||
delete instanceData[instance];
|
||||
}
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ SETTERS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
function setTornadoRouter(address _newRouterAddress) external onlyGovernance {
|
||||
router = TornadoRouter(_newRouterAddress);
|
||||
emit NewRouterRegistered(_newRouterAddress);
|
||||
}
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ GETTERS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
function version() public pure virtual returns (string memory) {
|
||||
return "v2-oracle-manager";
|
||||
}
|
||||
|
||||
function getAllInstances() public view virtual returns (ITornadoInstance[] memory allInstances) {
|
||||
uint256 numInstances = instances.length;
|
||||
|
||||
allInstances = new ITornadoInstance[](instances.length);
|
||||
|
||||
for (uint256 i = 0; i < numInstances; i++) {
|
||||
allInstances[i] = instances[i];
|
||||
}
|
||||
}
|
||||
|
||||
function getAllInstanceData() public view virtual returns (InstanceData[] memory data) {
|
||||
uint256 numInstances = instances.length;
|
||||
|
||||
data = new InstanceData[](numInstances);
|
||||
|
||||
for (uint256 i = 0; i < numInstances; i++) {
|
||||
data[i] = instanceData[instances[i]];
|
||||
}
|
||||
}
|
||||
|
||||
function getInstanceToken(ITornadoInstance _instance) public view virtual returns (IERC20) {
|
||||
return instanceData[_instance].token;
|
||||
}
|
||||
|
||||
function getInstanceIndex(ITornadoInstance _instance) public view virtual returns (uint80) {
|
||||
return instanceData[_instance].index;
|
||||
}
|
||||
}
|
142
src/v2/TornadoRouter.sol
Normal file
142
src/v2/TornadoRouter.sol
Normal file
@ -0,0 +1,142 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.6.12;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
// OZ Imports
|
||||
|
||||
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
|
||||
import { Initializable } from "@openzeppelin/contracts/proxy/Initializable.sol";
|
||||
import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
|
||||
import { Math } from "@openzeppelin/contracts/math/Math.sol";
|
||||
|
||||
// Tornado Imports
|
||||
|
||||
import { ITornadoInstance } from "tornado-anonymity-mining/contracts/interfaces/ITornadoInstance.sol";
|
||||
|
||||
// Local imports
|
||||
|
||||
import { RelayerRegistry } from "../v1/RelayerRegistry.sol";
|
||||
import { InstanceRegistry, InstanceData } from "./InstanceRegistry.sol";
|
||||
|
||||
contract TornadoRouter is Initializable {
|
||||
using SafeERC20 for IERC20;
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ VARIABLES ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
address public immutable governanceProxyAddress;
|
||||
|
||||
InstanceRegistry public instanceRegistry;
|
||||
|
||||
RelayerRegistry public relayerRegistry;
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ EVENTS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
event EncryptedNote(address indexed sender, bytes encryptedNote);
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ LOGIC ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
constructor(address _governanceProxyAddress) public {
|
||||
governanceProxyAddress = _governanceProxyAddress;
|
||||
}
|
||||
|
||||
modifier onlyGovernance() {
|
||||
require(msg.sender == governanceProxyAddress, "TornadoRouter: onlyGovernance");
|
||||
_;
|
||||
}
|
||||
|
||||
modifier onlyInstanceRegistry() {
|
||||
require(msg.sender == address(instanceRegistry), "TornadoRouter: onlyInstanceRegistry");
|
||||
_;
|
||||
}
|
||||
|
||||
function initalize(address _instanceRegistryAddress, address _relayerRegistryAddress)
|
||||
external
|
||||
onlyGovernance
|
||||
initializer
|
||||
{
|
||||
instanceRegistry = InstanceRegistry(_instanceRegistryAddress);
|
||||
relayerRegistry = RelayerRegistry(_relayerRegistryAddress);
|
||||
}
|
||||
|
||||
function deposit(ITornadoInstance _tornado, bytes32 _commitment, bytes calldata _encryptedNote)
|
||||
public
|
||||
payable
|
||||
virtual
|
||||
{
|
||||
(IERC20 token,, bool isERC20, bool isEnabled) = instanceRegistry.instanceData(_tornado);
|
||||
|
||||
require(isEnabled, "TornadoRouter: instance not enabled");
|
||||
|
||||
if (isERC20) {
|
||||
token.safeTransferFrom(msg.sender, address(this), _tornado.denomination());
|
||||
}
|
||||
|
||||
_tornado.deposit{ value: msg.value }(_commitment);
|
||||
|
||||
emit EncryptedNote(msg.sender, _encryptedNote);
|
||||
}
|
||||
|
||||
function withdraw(
|
||||
ITornadoInstance _tornado,
|
||||
bytes calldata _proof,
|
||||
bytes32 _root,
|
||||
bytes32 _nullifierHash,
|
||||
address payable _recipient,
|
||||
address payable _relayer,
|
||||
uint256 _fee,
|
||||
uint256 _refund
|
||||
) public payable virtual {
|
||||
(,,, bool isEnabled) = instanceRegistry.instanceData(_tornado);
|
||||
|
||||
require(isEnabled, "TornadoRouter: instance not enabled");
|
||||
|
||||
relayerRegistry.burn(msg.sender, _relayer, _tornado);
|
||||
|
||||
_tornado.withdraw{ value: msg.value }(
|
||||
_proof, _root, _nullifierHash, _recipient, _relayer, _fee, _refund
|
||||
);
|
||||
}
|
||||
|
||||
function backupNotes(bytes[] calldata _encryptedNotes) public virtual {
|
||||
for (uint256 i = 0; i < _encryptedNotes.length; i++) {
|
||||
emit EncryptedNote(msg.sender, _encryptedNotes[i]);
|
||||
}
|
||||
}
|
||||
|
||||
function rescueTokens(IERC20 _token, address payable _to, uint256 _amount)
|
||||
public
|
||||
virtual
|
||||
onlyGovernance
|
||||
{
|
||||
require(_to != address(0), "TORN: can not send to zero address");
|
||||
|
||||
if (_token == IERC20(0)) {
|
||||
// for Ether
|
||||
uint256 totalBalance = address(this).balance;
|
||||
uint256 balance = Math.min(totalBalance, _amount);
|
||||
_to.transfer(balance);
|
||||
} else {
|
||||
// any other erc20
|
||||
uint256 totalBalance = _token.balanceOf(address(this));
|
||||
uint256 balance = Math.min(totalBalance, _amount);
|
||||
require(balance > 0, "TORN: trying to send 0 balance");
|
||||
_token.safeTransfer(_to, balance);
|
||||
}
|
||||
}
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ SETTERS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
function approveTokenForInstance(IERC20 _token, address _spender, uint256 _amount)
|
||||
external
|
||||
onlyInstanceRegistry
|
||||
{
|
||||
_token.safeApprove(_spender, _amount);
|
||||
}
|
||||
|
||||
/* ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ GETTERS ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ */
|
||||
|
||||
function version() public pure virtual returns (string memory) {
|
||||
return "v2-oracle-manager";
|
||||
}
|
||||
}
|
17
src/v2/UniswapV3FeeOracle.sol
Normal file
17
src/v2/UniswapV3FeeOracle.sol
Normal file
@ -0,0 +1,17 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.6.12;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
// OZ imports
|
||||
|
||||
import { SafeMath } from "@openzeppelin/contracts/math/SafeMath.sol";
|
||||
|
||||
// Local imports
|
||||
|
||||
import { UniswapV3OracleHelper } from "./libraries/UniswapV3OracleHelper.sol";
|
||||
import { IFeeOracle } from "./interfaces/IFeeOracle.sol";
|
||||
|
||||
contract UniswapV3FeeOracle is IFeeOracle {
|
||||
function getFee() public view virtual override returns (uint160) { }
|
||||
}
|
8
src/v2/interfaces/IFeeOracle.sol
Normal file
8
src/v2/interfaces/IFeeOracle.sol
Normal file
@ -0,0 +1,8 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.6.12;
|
||||
pragma experimental ABIEncoderV2;
|
||||
|
||||
interface IFeeOracle {
|
||||
function getFee() external view returns (uint160);
|
||||
}
|
103
src/v2/libraries/UniswapV3OracleHelper.sol
Normal file
103
src/v2/libraries/UniswapV3OracleHelper.sol
Normal file
@ -0,0 +1,103 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.6.12;
|
||||
|
||||
import { OracleLibrary } from "@uniswap/v3-periphery/contracts/libraries/OracleLibrary.sol";
|
||||
import { IUniswapV3Factory } from "@uniswap/v3-core/contracts/interfaces/IUniswapV3Factory.sol";
|
||||
import { LowGasSafeMath } from "@uniswap/v3-core/contracts/libraries/LowGasSafeMath.sol";
|
||||
|
||||
interface IERC20Decimals {
|
||||
function decimals() external view returns (uint8);
|
||||
}
|
||||
|
||||
library UniswapV3OracleHelper {
|
||||
using LowGasSafeMath for uint256;
|
||||
|
||||
IUniswapV3Factory internal constant UniswapV3Factory =
|
||||
IUniswapV3Factory(0x1F98431c8aD98523631AE4a59f267346ea31F984);
|
||||
address internal constant WETH = 0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2;
|
||||
uint256 internal constant RATIO_DIVIDER = 1e18;
|
||||
|
||||
/**
|
||||
* @notice This function should return the price of baseToken in quoteToken, as in: quote/base (WETH/TORN)
|
||||
* @dev uses the Uniswap written OracleLibrary "getQuoteAtTick", does not call external libraries,
|
||||
* uses decimals() for the correct power of 10
|
||||
* @param baseToken token which will be denominated in quote token
|
||||
* @param quoteToken token in which price will be denominated
|
||||
* @param fee the uniswap pool fee, pools have different fees so this is a pool selector for our usecase
|
||||
* @param period the amount of seconds we are going to look into the past for the new token price
|
||||
* @return returns the price of baseToken in quoteToken
|
||||
*
|
||||
*/
|
||||
function getPriceOfTokenInToken(address baseToken, address quoteToken, uint24 fee, uint32 period)
|
||||
internal
|
||||
view
|
||||
returns (uint256)
|
||||
{
|
||||
uint128 base = uint128(10) ** uint128(IERC20Decimals(quoteToken).decimals());
|
||||
if (baseToken == quoteToken) {
|
||||
return base;
|
||||
} else {
|
||||
return OracleLibrary.getQuoteAtTick(
|
||||
OracleLibrary.consult(UniswapV3Factory.getPool(baseToken, quoteToken, fee), period),
|
||||
base,
|
||||
baseToken,
|
||||
quoteToken
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @notice This function should return the price of token in WETH
|
||||
* @dev simply feeds WETH in to the above function
|
||||
* @param token token which will be denominated in WETH
|
||||
* @param fee the uniswap pool fee, pools have different fees so this is a pool selector for our usecase
|
||||
* @param period the amount of seconds we are going to look into the past for the new token price
|
||||
* @return returns the price of token in WETH
|
||||
*
|
||||
*/
|
||||
function getPriceOfTokenInWETH(address token, uint24 fee, uint32 period)
|
||||
internal
|
||||
view
|
||||
returns (uint256)
|
||||
{
|
||||
return getPriceOfTokenInToken(token, WETH, fee, period);
|
||||
}
|
||||
|
||||
/**
|
||||
* @notice This function should return the price of WETH in token
|
||||
* @dev simply feeds WETH into getPriceOfTokenInToken
|
||||
* @param token token which WETH will be denominated in
|
||||
* @param fee the uniswap pool fee, pools have different fees so this is a pool selector for our usecase
|
||||
* @param period the amount of seconds we are going to look into the past for the new token price
|
||||
* @return returns the price of token in WETH
|
||||
*
|
||||
*/
|
||||
function getPriceOfWETHInToken(address token, uint24 fee, uint32 period)
|
||||
internal
|
||||
view
|
||||
returns (uint256)
|
||||
{
|
||||
return getPriceOfTokenInToken(WETH, token, fee, period);
|
||||
}
|
||||
|
||||
/**
|
||||
* @notice This function returns the price of token[0] in token[1], but more precisely and importantly the
|
||||
* price ratio of the tokens in WETH
|
||||
* @dev this is done as to always have good prices due to WETH-token pools mostly always having the most
|
||||
* liquidity
|
||||
* @param tokens array of tokens to get ratio for
|
||||
* @param fees the uniswap pool FEES, since these are two independent tokens
|
||||
* @param period the amount of seconds we are going to look into the past for the new token price
|
||||
* @return returns the price of token[0] in token[1]
|
||||
*
|
||||
*/
|
||||
function getPriceRatioOfTokens(address[2] memory tokens, uint24[2] memory fees, uint32 period)
|
||||
internal
|
||||
view
|
||||
returns (uint256)
|
||||
{
|
||||
return getPriceOfTokenInWETH(tokens[0], fees[0], period).mul(RATIO_DIVIDER)
|
||||
/ getPriceOfTokenInWETH(tokens[1], fees[1], period);
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user