Skip to content
This repository was archived by the owner on Sep 20, 2019. It is now read-only.
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions contracts/Vault/Connectors/ConnectorInterface.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
pragma solidity ^0.4.23;

interface ConnectorInterface {

function register() external;
function deposit(address token, address user, uint amount) external;
function withdraw(address token, address user, uint amount) external;
function balanceOf(address token, address user) external returns (uint);
function receiver() external returns (bytes4);

}
60 changes: 60 additions & 0 deletions contracts/Vault/VaultV2.sol
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
pragma solidity ^0.4.23;

import "./Connectors/ConnectorInterface.sol";

contract VaultV2 {

event Deposited(address token, address user, uint amount);
event Withdrawn(address token, address user, uint amount);

mapping (bytes4 => ConnectorInterface) public receivers;
mapping (address => ConnectorInterface) public connectors;

function () public payable {
// @todo move deposit into here potentially

ConnectorInterface connector = receivers[msg.sig];
require(address(connector) != 0x0);

assembly {
calldatacopy(0x0, 0x0, calldatasize)

result := delegatecall(sub(gas, 10000), connector, 0x0, calldatasize, 0, 0)
size := returndatasize

returndatacopy(0, 0, size)

switch result case 0 { revert(0, size) }
default { return(0, size) }
}
}

function addConnector(ConnectorInterface connector) external {
connector.delegatecall(connector.register.selector);

bytes4 receiver = connector.receiver();
if (receiver != 0x0) {
receivers[receiver] = connector;
}
}

function deposit(address token, address user, uint amount) external payable {
ConnectorInterface connector = connectors[token];
connectors[token].delegatecall.value(msg.value)(
connector.deposit.selector,
token,
user,
amount
);

emit Deposited(token, user, amount);
}

function withdraw(address token, address user, uint amount) external {
ConnectorInterface connector = connectors[token];
require(connector.balanceOf(token, user) >= amount);
connector.delegatecall(connector.withdraw.selector, token, user, amount);

emit Withdrawn(token, user, amount);
}
}