Browse code
Reformat
Ed Langley authored on 14/06/2017 18:01:32
Showing 6 changed files
Showing 6 changed files
... | ... |
@@ -43,7 +43,7 @@ installer init_installer() |
43 | 43 |
sys_time time (sys_time::get()); |
44 | 44 |
int code_digits = 6; |
45 | 45 |
totp_generator generator = totp_generator (time, code_digits); |
46 |
- random_source rand (random_source::create(fstreams)); |
|
46 |
+ random_source rand (random_source::create (fstreams)); |
|
47 | 47 |
directory directory (directory::create (unistd, pwd)); |
48 | 48 |
tokens tokens (tokens::create (fstreams, generator, rand)); |
49 | 49 |
installer installer (installer::create (tokens, unistd, directory, |
... | ... |
@@ -65,3 +65,4 @@ int main (int argc, char *argv[]) |
65 | 65 |
<< "Run again to get another token from the same key." |
66 | 66 |
<< std::endl; |
67 | 67 |
} |
68 |
+ |
... | ... |
@@ -30,7 +30,8 @@ public: |
30 | 30 |
{ |
31 | 31 |
return ""; |
32 | 32 |
} |
33 |
- virtual std::string generate_key() const { |
|
33 |
+ virtual std::string generate_key() const |
|
34 |
+ { |
|
34 | 35 |
return ""; |
35 | 36 |
} |
36 | 37 |
virtual std::string ensure_key (const user &user) const |
... | ... |
@@ -73,3 +74,4 @@ public: |
73 | 74 |
}; |
74 | 75 |
|
75 | 76 |
#endif |
77 |
+ |
... | ... |
@@ -48,7 +48,7 @@ private: |
48 | 48 |
public: |
49 | 49 |
pstream open_fstream (const std::string &file_path) const override |
50 | 50 |
{ |
51 |
- pstream result = std::make_shared<std::istringstream>(""); |
|
51 |
+ pstream result = std::make_shared<std::istringstream> (""); |
|
52 | 52 |
std::string badinator (1, '\0'); |
53 | 53 |
result->read (&badinator[0], badinator.size()); |
54 | 54 |
return result; |
... | ... |
@@ -146,7 +146,7 @@ int reads_from_the_right_file () |
146 | 146 |
token))); |
147 | 147 |
totp_generator generator (totp_generator::delegate (new |
148 | 148 |
fake_totp_generator (token))); |
149 |
- random_source fake_rand(random_source::delegate (new class fake_rand)); |
|
149 |
+ random_source fake_rand (random_source::delegate (new class fake_rand)); |
|
150 | 150 |
|
151 | 151 |
//file_reader test_file_reader (file_reader::delegate (new fake_file_reader)); |
152 | 152 |
user test_user (user::delegate (new fake_user (home_directory))); |
... | ... |
@@ -170,7 +170,7 @@ int returns_empty_string_if_file_open_fail() |
170 | 170 |
"654321"))); |
171 | 171 |
totp_generator generator (totp_generator::delegate (new |
172 | 172 |
fake_totp_generator ())); |
173 |
- random_source fake_rand(random_source::delegate (new class fake_rand)); |
|
173 |
+ random_source fake_rand (random_source::delegate (new class fake_rand)); |
|
174 | 174 |
|
175 | 175 |
user test_user (user::delegate (new fake_user (home_directory))); |
176 | 176 |
tokens supplier (tokens::create (test_streams, generator, fake_rand)); |
... | ... |
@@ -196,7 +196,7 @@ int returns_empty_string_if_file_too_short() |
196 | 196 |
token))); |
197 | 197 |
totp_generator generator (totp_generator::delegate (new |
198 | 198 |
fake_totp_generator (token))); |
199 |
- random_source fake_rand(random_source::delegate (new class fake_rand)); |
|
199 |
+ random_source fake_rand (random_source::delegate (new class fake_rand)); |
|
200 | 200 |
|
201 | 201 |
//file_reader test_file_reader (file_reader::delegate (new fake_file_reader)); |
202 | 202 |
user test_user (user::delegate (new fake_user (home_directory))); |
... | ... |
@@ -220,7 +220,7 @@ int writes_the_key () |
220 | 220 |
totp_generator generator (totp_generator::delegate (new |
221 | 221 |
fake_totp_generator ())); |
222 | 222 |
std::string token ("token"); |
223 |
- random_source fake_rand(random_source::delegate (new class fake_rand)); |
|
223 |
+ random_source fake_rand (random_source::delegate (new class fake_rand)); |
|
224 | 224 |
tokens tokens (tokens::create (test_streams, generator, fake_rand)); |
225 | 225 |
|
226 | 226 |
//when |
... | ... |
@@ -247,17 +247,19 @@ int ensure_key_creates_key_file_if_not_exists () |
247 | 247 |
|
248 | 248 |
fstreams test_streams{fstreams::delegate (mockfs)}; |
249 | 249 |
|
250 |
- totp_generator generator (totp_generator::delegate (new fake_totp_generator ())); |
|
250 |
+ totp_generator generator (totp_generator::delegate (new |
|
251 |
+ fake_totp_generator ())); |
|
251 | 252 |
|
252 | 253 |
std::vector<uint8_t> random_bytes {4,2,4, 2,4, 2,4,2, 4,2}; |
253 |
- random_source fake_rand(random_source::delegate (new fake_rand_with_specified_result(random_bytes))); |
|
254 |
+ random_source fake_rand (random_source::delegate (new |
|
255 |
+ fake_rand_with_specified_result (random_bytes))); |
|
254 | 256 |
tokens tokens (tokens::create (test_streams, generator, fake_rand)); |
255 | 257 |
|
256 | 258 |
//when |
257 | 259 |
std::string actual = tokens.ensure_key (test_user); |
258 | 260 |
|
259 | 261 |
base32 codec; |
260 |
- octet_vector actual_decoded = codec.decode(actual); |
|
262 |
+ octet_vector actual_decoded = codec.decode (actual); |
|
261 | 263 |
// then |
262 | 264 |
std::string expected_filename (home_directory + "/.dual_control"); |
263 | 265 |
check (mockfs->captured_filename() == expected_filename, |
... | ... |
@@ -276,14 +278,15 @@ int ensure_key_reads_key_file_if_exists () |
276 | 278 |
user test_user (user::delegate (new fake_user (home_directory))); |
277 | 279 |
fstreams test_streams (fstreams::delegate (new fake_fstreams (key_file, |
278 | 280 |
key))); |
279 |
- totp_generator generator (totp_generator::delegate (new fake_totp_generator ())); |
|
280 |
- random_source fake_rand(random_source::delegate (new class fake_rand)); |
|
281 |
+ totp_generator generator (totp_generator::delegate (new |
|
282 |
+ fake_totp_generator ())); |
|
283 |
+ random_source fake_rand (random_source::delegate (new class fake_rand)); |
|
281 | 284 |
tokens tokens (tokens::create (test_streams, generator, fake_rand)); |
282 | 285 |
|
283 | 286 |
//when |
284 | 287 |
std::string actual = tokens.ensure_key (test_user); |
285 | 288 |
//then |
286 |
- check(actual == key, "read key does not match given key"); |
|
289 |
+ check (actual == key, "read key does not match given key"); |
|
287 | 290 |
succeed(); |
288 | 291 |
} |
289 | 292 |
|
... | ... |
@@ -292,14 +295,17 @@ int generate_key_uses_random_source () |
292 | 295 |
// given |
293 | 296 |
user test_user (user::delegate (new fake_user ("/nowhere"))); |
294 | 297 |
fstreams test_streams{fstreams::delegate (new fake_fstreams ("<>", "<>"))}; |
295 |
- totp_generator generator (totp_generator::delegate (new fake_totp_generator ())); |
|
298 |
+ totp_generator generator (totp_generator::delegate (new |
|
299 |
+ fake_totp_generator ())); |
|
296 | 300 |
|
297 | 301 |
std::vector<uint8_t> random_bytes1 {4,2,4, 2,4, 2,4,2, 4,2}; |
298 |
- random_source fake_rand1(random_source::delegate (new fake_rand_with_specified_result(random_bytes1))); |
|
302 |
+ random_source fake_rand1 (random_source::delegate (new |
|
303 |
+ fake_rand_with_specified_result (random_bytes1))); |
|
299 | 304 |
tokens tokens1 (tokens::create (test_streams, generator, fake_rand1)); |
300 | 305 |
|
301 | 306 |
std::vector<uint8_t> random_bytes2 {1,2,1, 2,1, 1,2,1, 2,1}; |
302 |
- random_source fake_rand2(random_source::delegate (new fake_rand_with_specified_result(random_bytes2))); |
|
307 |
+ random_source fake_rand2 (random_source::delegate (new |
|
308 |
+ fake_rand_with_specified_result (random_bytes2))); |
|
303 | 309 |
tokens tokens2 (tokens::create (test_streams, generator, fake_rand2)); |
304 | 310 |
|
305 | 311 |
//when |
... | ... |
@@ -307,7 +313,8 @@ int generate_key_uses_random_source () |
307 | 313 |
std::string second_key = tokens2.generate_key(); |
308 | 314 |
|
309 | 315 |
// then |
310 |
- check (first_key != second_key, "keys generated from differing random data should not match"); |
|
316 |
+ check (first_key != second_key, |
|
317 |
+ "keys generated from differing random data should not match"); |
|
311 | 318 |
succeed(); |
312 | 319 |
} |
313 | 320 |
|
... | ... |
@@ -326,3 +333,4 @@ int main (int argc, char *argv[]) |
326 | 333 |
{ |
327 | 334 |
return !run_tests(); |
328 | 335 |
} |
336 |
+ |