Line data Source code
1 : // Copyright (c) 2015 The Bitcoin Core developers
2 : // Distributed under the MIT software license, see the accompanying
3 : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
4 :
5 : #include <vector>
6 : #include "prevector.h"
7 : #include "random.h"
8 :
9 : #include "serialize.h"
10 : #include "streams.h"
11 :
12 : #include "test/test_pivx.h"
13 :
14 : #include <boost/test/unit_test.hpp>
15 :
16 : BOOST_FIXTURE_TEST_SUITE(PrevectorTests, TestingSetup)
17 :
18 : template<unsigned int N, typename T>
19 : class prevector_tester {
20 : typedef std::vector<T> realtype;
21 : realtype real_vector;
22 : realtype real_vector_alt;
23 :
24 : typedef prevector<N, T> pretype;
25 : pretype pre_vector;
26 : pretype pre_vector_alt;
27 :
28 : typedef typename pretype::size_type Size;
29 : bool passed = true;
30 : FastRandomContext rand_cache;
31 : uint256 rand_seed;
32 :
33 : template <typename A, typename B>
34 11868500 : void local_check_equal(A a, B b)
35 : {
36 11868500 : local_check(a == b);
37 : }
38 26301000 : void local_check(bool b)
39 : {
40 5595390 : passed &= b;
41 : }
42 220352 : void test() {
43 220352 : const pretype& const_pre_vector = pre_vector;
44 423872 : local_check_equal(real_vector.size(), pre_vector.size());
45 220352 : local_check_equal(real_vector.empty(), pre_vector.empty());
46 3018050 : for (Size s = 0; s < real_vector.size(); s++) {
47 2797700 : local_check(real_vector[s] == pre_vector[s]);
48 5595390 : local_check(&(pre_vector[s]) == &(pre_vector.begin()[s]));
49 2797700 : local_check(&(pre_vector[s]) == &*(pre_vector.begin() + s));
50 5595390 : local_check(&(pre_vector[s]) == &*((pre_vector.end() + s) - real_vector.size()));
51 : }
52 : // local_check(realtype(pre_vector) == real_vector);
53 220352 : local_check(pretype(real_vector.begin(), real_vector.end()) == pre_vector);
54 661056 : local_check(pretype(pre_vector.begin(), pre_vector.end()) == pre_vector);
55 220352 : size_t pos = 0;
56 :
57 3458750 : for (const T& v : pre_vector) {
58 2797700 : local_check(v == real_vector[pos++]);
59 : }
60 220352 : pos = 0;
61 3018050 : for (const T& v : const_pre_vector) {
62 2797700 : local_check(v == real_vector[pos++]);
63 : }
64 :
65 440704 : CDataStream ss1(SER_DISK, 0);
66 220352 : CDataStream ss2(SER_DISK, 0);
67 220352 : ss1 << real_vector;
68 220352 : ss2 << pre_vector;
69 220352 : local_check_equal(ss1.size(), ss2.size());
70 11631500 : for (Size s = 0; s < ss1.size(); s++) {
71 11411100 : local_check_equal(ss1[s], ss2[s]);
72 : }
73 220352 : }
74 :
75 : public:
76 8320 : void resize(Size s) {
77 8320 : real_vector.resize(s);
78 8320 : local_check_equal(real_vector.size(), s);
79 8320 : pre_vector.resize(s);
80 8320 : local_check_equal(pre_vector.size(), s);
81 8320 : test();
82 8320 : }
83 :
84 1664 : void reserve(Size s) {
85 1664 : real_vector.reserve(s);
86 1664 : local_check(real_vector.capacity() >= s);
87 1664 : pre_vector.reserve(s);
88 1664 : local_check(pre_vector.capacity() >= s);
89 1664 : test();
90 1664 : }
91 :
92 34304 : void insert(Size position, const T& value) {
93 34304 : real_vector.insert(real_vector.begin() + position, value);
94 68608 : pre_vector.insert(pre_vector.begin() + position, value);
95 34304 : test();
96 34304 : }
97 :
98 7872 : void insert(Size position, Size count, const T& value) {
99 7872 : real_vector.insert(real_vector.begin() + position, count, value);
100 15744 : pre_vector.insert(pre_vector.begin() + position, count, value);
101 7872 : test();
102 7872 : }
103 :
104 : template<typename I>
105 1728 : void insert_range(Size position, I first, I last) {
106 1728 : real_vector.insert(real_vector.begin() + position, first, last);
107 3456 : pre_vector.insert(pre_vector.begin() + position, first, last);
108 1728 : test();
109 1728 : }
110 :
111 16704 : void erase(Size position) {
112 16704 : real_vector.erase(real_vector.begin() + position);
113 33408 : pre_vector.erase(pre_vector.begin() + position);
114 16704 : test();
115 16704 : }
116 :
117 10432 : void erase(Size first, Size last) {
118 10432 : real_vector.erase(real_vector.begin() + first, real_vector.begin() + last);
119 20864 : pre_vector.erase(pre_vector.begin() + first, pre_vector.begin() + last);
120 10432 : test();
121 10432 : }
122 :
123 119360 : void update(Size pos, const T& value) {
124 119360 : real_vector[pos] = value;
125 119360 : pre_vector[pos] = value;
126 119360 : test();
127 119360 : }
128 :
129 3584 : void push_back(const T& value) {
130 3584 : real_vector.push_back(value);
131 3584 : pre_vector.push_back(value);
132 3584 : test();
133 3584 : }
134 :
135 3648 : void pop_back() {
136 3648 : real_vector.pop_back();
137 3648 : pre_vector.pop_back();
138 3648 : test();
139 3648 : }
140 :
141 128 : void clear() {
142 256 : real_vector.clear();
143 128 : pre_vector.clear();
144 128 : }
145 :
146 64 : void assign(Size n, const T& value) {
147 128 : real_vector.assign(n, value);
148 64 : pre_vector.assign(n, value);
149 : }
150 :
151 602368 : Size size() {
152 602368 : return real_vector.size();
153 : }
154 :
155 : Size capacity() {
156 : return pre_vector.capacity();
157 : }
158 :
159 1024 : void shrink_to_fit() {
160 1024 : pre_vector.shrink_to_fit();
161 1024 : test();
162 1024 : }
163 :
164 8064 : void swap() {
165 8064 : real_vector.swap(real_vector_alt);
166 8064 : pre_vector.swap(pre_vector_alt);
167 8064 : test();
168 8064 : }
169 :
170 1728 : void move() {
171 1728 : real_vector = std::move(real_vector_alt);
172 1728 : real_vector_alt.clear();
173 1728 : pre_vector = std::move(pre_vector_alt);
174 1728 : pre_vector_alt.clear();
175 1728 : }
176 :
177 5056 : void copy() {
178 5056 : real_vector = real_vector_alt;
179 5056 : pre_vector = pre_vector_alt;
180 5056 : }
181 :
182 3648 : void resize_uninitialized(realtype values) {
183 3648 : size_t r = values.size();
184 3648 : size_t s = real_vector.size() / 2;
185 3648 : if (real_vector.capacity() < s + r) {
186 704 : real_vector.reserve(s + r);
187 : }
188 3648 : real_vector.resize(s);
189 3648 : pre_vector.resize_uninitialized(s);
190 36800 : for (auto v : values) {
191 33152 : real_vector.push_back(v);
192 : }
193 3648 : auto p = pre_vector.size();
194 3648 : pre_vector.resize_uninitialized(p + r);
195 36800 : for (auto v : values) {
196 33152 : pre_vector[p] = v;
197 33152 : ++p;
198 : }
199 3648 : test();
200 3648 : }
201 :
202 64 : ~prevector_tester() {
203 128 : BOOST_CHECK_MESSAGE(passed, "insecure_rand: " + rand_seed.ToString());
204 256 : }
205 128 : prevector_tester() {
206 64 : SeedInsecureRand();
207 64 : rand_seed = InsecureRand256();
208 64 : rand_cache = FastRandomContext(rand_seed);
209 64 : }
210 : };
211 :
212 2 : BOOST_AUTO_TEST_CASE(PrevectorTestInt)
213 : {
214 65 : for (int j = 0; j < 64; j++) {
215 128 : prevector_tester<8, int> test;
216 131136 : for (int i = 0; i < 2048; i++) {
217 131072 : int r = InsecureRand32();
218 131072 : if ((r % 4) == 0) {
219 34304 : test.insert(InsecureRand32() % (test.size() + 1), InsecureRand32());
220 : }
221 131072 : if (test.size() > 0 && ((r >> 2) % 4) == 1) {
222 16704 : test.erase(InsecureRand32() % test.size());
223 : }
224 131072 : if (((r >> 4) % 8) == 2) {
225 16000 : int new_size = std::max<int>(0, std::min<int>(30, test.size() + (InsecureRand32() % 5) - 2));
226 8320 : test.resize(new_size);
227 : }
228 131072 : if (((r >> 7) % 8) == 3) {
229 7872 : test.insert(InsecureRand32() % (test.size() + 1), 1 + (InsecureRand32() % 2), InsecureRand32());
230 : }
231 131072 : if (((r >> 10) % 8) == 4) {
232 8384 : int del = std::min<int>(test.size(), 1 + (InsecureRand32() % 2));
233 8384 : int beg = InsecureRand32() % (test.size() + 1 - del);
234 8384 : test.erase(beg, beg + del);
235 : }
236 131072 : if (((r >> 13) % 16) == 5) {
237 3584 : test.push_back(InsecureRand32());
238 : }
239 131072 : if (test.size() > 0 && ((r >> 17) % 16) == 6) {
240 3648 : test.pop_back();
241 : }
242 131072 : if (((r >> 21) % 32) == 7) {
243 1728 : int values[4];
244 1728 : int num = 1 + (InsecureRand32() % 4);
245 6016 : for (int i = 0; i < num; i++) {
246 4288 : values[i] = InsecureRand32();
247 : }
248 1728 : test.insert_range(InsecureRand32() % (test.size() + 1), values, values + num);
249 : }
250 131072 : if (((r >> 26) % 32) == 8) {
251 2048 : int del = std::min<int>(test.size(), 1 + (InsecureRand32() % 4));
252 2048 : int beg = InsecureRand32() % (test.size() + 1 - del);
253 2048 : test.erase(beg, beg + del);
254 : }
255 131072 : r = InsecureRand32();
256 131072 : if (r % 32 == 9) {
257 1664 : test.reserve(InsecureRand32() % 32);
258 : }
259 131072 : if ((r >> 5) % 64 == 10) {
260 1024 : test.shrink_to_fit();
261 : }
262 131072 : if (test.size() > 0) {
263 119360 : test.update(InsecureRand32() % test.size(), InsecureRand32());
264 : }
265 131072 : if (((r >> 11) % 1024) == 11) {
266 128 : test.clear();
267 : }
268 131072 : if (((r >> 21) % 512) == 12) {
269 128 : test.assign(InsecureRand32() % 32, InsecureRand32());
270 : }
271 131072 : if (((r >> 15) % 8) == 3) {
272 8064 : test.swap();
273 : }
274 131072 : if (((r >> 15) % 16) == 8) {
275 5056 : test.copy();
276 : }
277 131072 : if (((r >> 15) % 32) == 18) {
278 1728 : test.move();
279 : }
280 131072 : if (InsecureRandBits(5) == 19) {
281 3648 : unsigned int num = 1 + (InsecureRandBits(4));
282 7296 : std::vector<int> values(num);
283 36800 : for (auto &v : values) {
284 33152 : v = InsecureRand32();
285 : }
286 7296 : test.resize_uninitialized(values);
287 : }
288 : }
289 : }
290 1 : }
291 :
292 : BOOST_AUTO_TEST_SUITE_END()
|
