forked from morpho-org/morpho-blue
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Reentrancy.spec
61 lines (52 loc) · 2.15 KB
/
Reentrancy.spec
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
// SPDX-License-Identifier: GPL-2.0-or-later
methods {
function extSloads(bytes32[]) external returns bytes32[] => NONDET DELETE;
function _.borrowRate(MorphoHarness.MarketParams marketParams, MorphoHarness.Market) external => summaryBorrowRate() expect uint256;
}
persistent ghost bool delegateCall;
persistent ghost bool callIsBorrowRate;
// True when storage has been accessed with either a SSTORE or a SLOAD.
persistent ghost bool hasAccessedStorage;
// True when a CALL has been done after storage has been accessed.
persistent ghost bool hasCallAfterAccessingStorage;
// True when storage has been accessed, after which an external call is made, followed by accessing storage again.
persistent ghost bool hasReentrancyUnsafeCall;
function summaryBorrowRate() returns uint256 {
uint256 result;
callIsBorrowRate = true;
return result;
}
hook ALL_SSTORE(uint loc, uint v) {
hasAccessedStorage = true;
hasReentrancyUnsafeCall = hasCallAfterAccessingStorage;
}
hook ALL_SLOAD(uint loc) uint v {
hasAccessedStorage = true;
hasReentrancyUnsafeCall = hasCallAfterAccessingStorage;
}
hook CALL(uint g, address addr, uint value, uint argsOffset, uint argsLength, uint retOffset, uint retLength) uint rc {
if (callIsBorrowRate) {
// The calls to borrow rate are trusted and don't count.
callIsBorrowRate = false;
} else {
hasCallAfterAccessingStorage = hasAccessedStorage;
}
}
hook DELEGATECALL(uint g, address addr, uint argsOffset, uint argsLength, uint retOffset, uint retLength) uint rc {
delegateCall = true;
}
// Check that no function is accessing storage, then making an external CALL other than to the IRM, and accessing storage again.
rule reentrancySafe(method f, env e, calldataarg data) {
// Set up the initial state.
require !callIsBorrowRate;
require !hasAccessedStorage && !hasCallAfterAccessingStorage && !hasReentrancyUnsafeCall;
f(e,data);
assert !hasReentrancyUnsafeCall;
}
// Check that the contract is truly immutable.
rule noDelegateCalls(method f, env e, calldataarg data) {
// Set up the initial state.
require !delegateCall;
f(e,data);
assert !delegateCall;
}