Add ReentrancyGuard

This commit is contained in:
Remco Bloemen
2017-03-24 11:01:06 +00:00
parent ffce7e3b08
commit a2bd1bb7f6
5 changed files with 136 additions and 0 deletions

31
test/ReentrancyGuard.js Normal file
View File

@ -0,0 +1,31 @@
'use strict';
import expectThrow from './helpers/expectThrow';
const ReentrancyMock = artifacts.require('./helper/ReentrancyMock.sol');
const ReentrancyAttack = artifacts.require('./helper/ReentrancyAttack.sol');
contract('ReentrancyGuard', function(accounts) {
let reentrancyMock;
beforeEach(async function() {
reentrancyMock = await ReentrancyMock.new();
let initialCounter = await reentrancyMock.counter();
assert.equal(initialCounter, 0);
});
it('should not allow remote callback', async function() {
let attacker = await ReentrancyAttack.new();
await expectThrow(reentrancyMock.countAndCall(attacker.address));
});
// The following are more side-effects that intended behaviour:
// I put them here as documentation, and to monitor any changes
// in the side-effects.
it('should not allow local recursion', async function() {
await expectThrow(reentrancyMock.countLocalRecursive(10));
});
it('should not allow indirect local recursion', async function() {
await expectThrow(reentrancyMock.countThisRecursive(10));
});
});