-
Notifications
You must be signed in to change notification settings - Fork 185
/
arrays.fe
102 lines (85 loc) · 2.93 KB
/
arrays.fe
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
use std::evm
contract Foo {
my_array: Array<u256, 10>
pub fn __init__(mut self) {
self.my_array = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
}
pub fn get_from_storage(self, index: u256) -> u256 {
return self.my_array[index]
}
pub fn get_from_memory(index: u256) -> u256 {
let my_array: Array<u256, 10> = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
return my_array[index]
}
}
#test
unsafe fn test_foo() {
let mut ctx: Context = Context()
let foo: Foo = Foo.create(ctx, 0)
assert foo.get_from_storage(index: 3) == 3
assert Foo::get_from_memory(index: 3) == 4
}
contract ArrayIndexOobDyn {
pub unsafe fn __call__() {
let index: u256 = evm::call_data_load(offset: 0)
let mut my_array: Array<u256, 4> = [0, 1, 2, 3]
evm::sstore(offset: 0, value: my_array[index])
}
}
const FREE_MEM_PTR: u256 = 1024
const CALL_GAS: u256 = 100000
#test
unsafe fn test_array_oob_dyn() {
let mut ctx: Context = Context()
let array_index_oob: ArrayIndexOobDyn = ArrayIndexOobDyn.create(ctx, value: 0)
let expected_revert_data: Array<u8, 36> = [78, 72, 123, 113, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 50]
// store index value
evm::mstore(offset: FREE_MEM_PTR, value: 26)
// verify that the call reverts
assert evm::call(
gas: CALL_GAS,
addr: address(array_index_oob),
value: 0,
input_offset: FREE_MEM_PTR,
input_len: 32,
output_offset: FREE_MEM_PTR,
output_len: 36
) == 0
// check the revert data
let mut offset: u256 = FREE_MEM_PTR
for expected_byte in expected_revert_data {
assert evm::shr(248, evm::mload(offset)) == expected_byte
offset += 1
}
}
// The following tests does not pass due absent runtime check.
// Obviously this couild also be a compile-time check.
//
// contract ArrayIndexOobStatic {
// pub unsafe fn __call__() {
// let mut my_array: Array<u256, 4> = [0, 1, 2, 3]
// evm::sstore(offset: 0, value: my_array[26])
// }
// }
// #test
// unsafe fn test_array_oob_static() {
// let mut ctx: Context = Context()
// let array_index_oob: ArrayIndexOobStatic = ArrayIndexOobStatic.create(ctx, value: 0)
// let expected_revert_data: Array<u8, 36> = [78, 72, 123, 113, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 50]
// // verify that the call reverts
// assert evm::call(
// gas: CALL_GAS,
// addr: address(array_index_oob),
// value: 0,
// input_offset: 0,
// input_len: 0,
// output_offset: FREE_MEM_PTR,
// output_len: 36
// ) == 0
// // check the revert data
// let mut offset: u256 = FREE_MEM_PTR
// for expected_byte in expected_revert_data {
// assert evm::shr(248, evm::mload(offset)) == expected_byte
// offset += 1
// }
// }