Move contracts to subdirectories (#1253)

* Move contracts to subdirectories Fixes #1177. This Change also removes the LimitBalance contract. * fix import * move MerkleProof to cryptography * Fix import
OpenZeppelin · Sep 3, 2018 · 2441fd7 · 2441fd7
1 parent 964bc40
commit 2441fd7
Show file tree

Hide file tree

Showing 21 changed files with 64 additions and 161 deletions.
diff --git a/contracts/LimitBalance.sol b/contracts/LimitBalance.sol
diff --git a/contracts/access/SignatureBouncer.sol b/contracts/access/SignatureBouncer.sol
@@ -2,7 +2,7 @@ pragma solidity ^0.4.24;
 
 import "../ownership/Ownable.sol";
 import "../access/rbac/RBAC.sol";
-import "../ECRecovery.sol";
+import "../cryptography/ECDSA.sol";
 
 
 /**
@@ -30,7 +30,7 @@ import "../ECRecovery.sol";
  * much more complex. See https://ethereum.stackexchange.com/a/50616 for more details.
  */
 contract SignatureBouncer is Ownable, RBAC {
-  using ECRecovery for bytes32;
+  using ECDSA for bytes32;
 
   string public constant ROLE_BOUNCER = "bouncer";
   uint internal constant METHOD_ID_SIZE = 4;

diff --git a/contracts/Bounty.sol → contracts/bounties/BreakInvariantBounty.sol b/contracts/Bounty.sol → contracts/bounties/BreakInvariantBounty.sol
@@ -1,15 +1,15 @@
 pragma solidity ^0.4.24;
 
 
-import "./payment/PullPayment.sol";
-import "./lifecycle/Destructible.sol";
+import "../payment/PullPayment.sol";
+import "../lifecycle/Destructible.sol";
 
 
 /**
- * @title Bounty
+ * @title BreakInvariantBounty
  * @dev This bounty will pay out to a researcher if they break invariant logic of the contract.
  */
-contract Bounty is PullPayment, Destructible {
+contract BreakInvariantBounty is PullPayment, Destructible {
   bool public claimed;
   mapping(address => address) public researchers;
 

diff --git a/contracts/ECRecovery.sol → contracts/cryptography/ECDSA.sol b/contracts/ECRecovery.sol → contracts/cryptography/ECDSA.sol
@@ -8,7 +8,7 @@ pragma solidity ^0.4.24;
  * See https://github.com/ethereum/solidity/issues/864
  */
 
-library ECRecovery {
+library ECDSA {
 
   /**
    * @dev Recover signer address from a message by using their signature

diff --git a/contracts/MerkleProof.sol → contracts/cryptography/MerkleProof.sol b/contracts/MerkleProof.sol → contracts/cryptography/MerkleProof.sol
diff --git a/contracts/mocks/AutoIncrementingImpl.sol b/contracts/mocks/AutoIncrementingImpl.sol
@@ -1,6 +1,6 @@
 pragma solidity ^0.4.24;
 
-import "../AutoIncrementing.sol";
+import "../utils/AutoIncrementing.sol";
 
 
 contract AutoIncrementingImpl {

diff --git a/contracts/mocks/ECRecoveryMock.sol → contracts/mocks/ECDSAMock.sol b/contracts/mocks/ECRecoveryMock.sol → contracts/mocks/ECDSAMock.sol
@@ -1,11 +1,11 @@
 pragma solidity ^0.4.24;
 
 
-import "../ECRecovery.sol";
+import "../cryptography/ECDSA.sol";
 
 
-contract ECRecoveryMock {
-  using ECRecovery for bytes32;
+contract ECDSAMock {
+  using ECDSA for bytes32;
 
   function recover(bytes32 _hash, bytes _signature)
     public

diff --git a/contracts/mocks/InsecureInvariantTargetBounty.sol b/contracts/mocks/InsecureInvariantTargetBounty.sol
@@ -0,0 +1,20 @@
+pragma solidity ^0.4.24;
+
+// When this line is split, truffle parsing fails.
+// See: https://github.com/ethereum/solidity/issues/4871
+// solium-disable-next-line max-len
+import {BreakInvariantBounty, Target} from "../../contracts/bounties/BreakInvariantBounty.sol";
+
+
+contract InsecureInvariantTargetMock is Target {
+  function checkInvariant() public returns(bool) {
+    return false;
+  }
+}
+
+
+contract InsecureInvariantTargetBounty is BreakInvariantBounty {
+  function deployContract() internal returns (address) {
+    return new InsecureInvariantTargetMock();
+  }
+}
diff --git a/contracts/mocks/InsecureTargetBounty.sol b/contracts/mocks/InsecureTargetBounty.sol
diff --git a/contracts/mocks/LimitBalanceMock.sol b/contracts/mocks/LimitBalanceMock.sol
diff --git a/contracts/mocks/MerkleProofWrapper.sol b/contracts/mocks/MerkleProofWrapper.sol
@@ -1,6 +1,6 @@
 pragma solidity ^0.4.24;
 
-import { MerkleProof } from "../MerkleProof.sol";
+import { MerkleProof } from "../cryptography/MerkleProof.sol";
 
 
 contract MerkleProofWrapper {

diff --git a/contracts/mocks/ReentrancyMock.sol b/contracts/mocks/ReentrancyMock.sol
@@ -1,6 +1,6 @@
 pragma solidity ^0.4.24;
 
-import "../ReentrancyGuard.sol";
+import "../utils/ReentrancyGuard.sol";
 import "./ReentrancyAttack.sol";
 
 

diff --git a/contracts/mocks/SecureInvariantTargetBounty.sol b/contracts/mocks/SecureInvariantTargetBounty.sol
@@ -0,0 +1,20 @@
+pragma solidity ^0.4.24;
+
+// When this line is split, truffle parsing fails.
+// See: https://github.com/ethereum/solidity/issues/4871
+// solium-disable-next-line max-len
+import {BreakInvariantBounty, Target} from "../../contracts/bounties/BreakInvariantBounty.sol";
+
+
+contract SecureInvariantTargetMock is Target {
+  function checkInvariant() public returns(bool) {
+    return true;
+  }
+}
+
+
+contract SecureInvariantTargetBounty is BreakInvariantBounty {
+  function deployContract() internal returns (address) {
+    return new SecureInvariantTargetMock();
+  }
+}
diff --git a/contracts/mocks/SecureTargetBounty.sol b/contracts/mocks/SecureTargetBounty.sol
diff --git a/contracts/token/ERC721/ERC721Basic.sol b/contracts/token/ERC721/ERC721Basic.sol
@@ -3,7 +3,7 @@ pragma solidity ^0.4.24;
 import "./IERC721Basic.sol";
 import "./IERC721Receiver.sol";
 import "../../math/SafeMath.sol";
-import "../../AddressUtils.sol";
+import "../../utils/Address.sol";
 import "../../introspection/SupportsInterfaceWithLookup.sol";
 
 
@@ -14,7 +14,7 @@ import "../../introspection/SupportsInterfaceWithLookup.sol";
 contract ERC721Basic is SupportsInterfaceWithLookup, IERC721Basic {
 
   using SafeMath for uint256;
-  using AddressUtils for address;
+  using Address for address;
 
   // Equals to `bytes4(keccak256("onERC721Received(address,address,uint256,bytes)"))`
   // which can be also obtained as `IERC721Receiver(0).onERC721Received.selector`

diff --git a/contracts/AddressUtils.sol → contracts/utils/Address.sol b/contracts/AddressUtils.sol → contracts/utils/Address.sol
@@ -4,7 +4,7 @@ pragma solidity ^0.4.24;
 /**
  * Utility library of inline functions on addresses
  */
-library AddressUtils {
+library Address {
 
   /**
    * Returns whether the target address is a contract

diff --git a/contracts/AutoIncrementing.sol → contracts/utils/AutoIncrementing.sol b/contracts/AutoIncrementing.sol → contracts/utils/AutoIncrementing.sol
diff --git a/contracts/ReentrancyGuard.sol → contracts/utils/ReentrancyGuard.sol b/contracts/ReentrancyGuard.sol → contracts/utils/ReentrancyGuard.sol
diff --git a/test/Bounty.test.js → test/BreakInvariantBounty.test.js b/test/Bounty.test.js → test/BreakInvariantBounty.test.js
@@ -2,8 +2,8 @@ const { ethGetBalance, ethSendTransaction } = require('./helpers/web3');
 const expectEvent = require('./helpers/expectEvent');
 const { assertRevert } = require('./helpers/assertRevert');
 
-const SecureTargetBounty = artifacts.require('SecureTargetBounty');
-const InsecureTargetBounty = artifacts.require('InsecureTargetBounty');
+const SecureInvariantTargetBounty = artifacts.require('SecureInvariantTargetBounty');
+const InsecureInvariantTargetBounty = artifacts.require('InsecureInvariantTargetBounty');
 
 require('chai')
   .use(require('chai-bignumber')(web3.BigNumber))
@@ -17,10 +17,10 @@ const sendReward = async (from, to, value) => ethSendTransaction({
 
 const reward = new web3.BigNumber(web3.toWei(1, 'ether'));
 
-contract('Bounty', function ([_, owner, researcher, nonTarget]) {
+contract('BreakInvariantBounty', function ([_, owner, researcher, nonTarget]) {
   context('against secure contract', function () {
     beforeEach(async function () {
-      this.bounty = await SecureTargetBounty.new({ from: owner });
+      this.bounty = await SecureInvariantTargetBounty.new({ from: owner });
     });
 
     it('can set reward', async function () {
@@ -53,7 +53,7 @@ contract('Bounty', function ([_, owner, researcher, nonTarget]) {
 
   context('against broken contract', function () {
     beforeEach(async function () {
-      this.bounty = await InsecureTargetBounty.new();
+      this.bounty = await InsecureInvariantTargetBounty.new();
 
       const result = await this.bounty.createTarget({ from: researcher });
       const event = expectEvent.inLogs(result.logs, 'TargetCreated');

diff --git a/test/LimitBalance.test.js b/test/LimitBalance.test.js
diff --git a/test/library/ECRecovery.test.js → test/library/ECDSA.test.js b/test/library/ECRecovery.test.js → test/library/ECDSA.test.js
@@ -1,17 +1,17 @@
 const { signMessage, toEthSignedMessageHash } = require('../helpers/sign');
 const { expectThrow } = require('../helpers/expectThrow');
 
-const ECRecoveryMock = artifacts.require('ECRecoveryMock');
+const ECDSAMock = artifacts.require('ECDSAMock');
 
 require('chai')
   .should();
 
 const TEST_MESSAGE = web3.sha3('OpenZeppelin');
 const WRONG_MESSAGE = web3.sha3('Nope');
 
-contract('ECRecovery', function ([_, anyone]) {
+contract('ECDSA', function ([_, anyone]) {
   beforeEach(async function () {
-    this.mock = await ECRecoveryMock.new();
+    this.mock = await ECDSAMock.new();
   });
 
   it('recover v0', async function () {