OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/quic/crypto/strike_register.h" | 5 #include "net/quic/crypto/strike_register.h" |
6 | 6 |
7 #include <set> | 7 #include <set> |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
11 #include "base/rand_util.h" | 11 #include "base/rand_util.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
13 | 13 |
14 namespace { | 14 namespace { |
15 | 15 |
16 using net::InsertStatus; | 16 using net::InsertStatus; |
17 using net::StrikeRegister; | 17 using net::StrikeRegister; |
18 using std::make_pair; | |
19 using std::min; | 18 using std::min; |
20 using std::pair; | 19 using std::pair; |
21 using std::set; | 20 using std::set; |
22 using std::string; | 21 using std::string; |
23 | 22 |
24 const uint8 kOrbit[8] = { 1, 2, 3, 4, 5, 6, 7, 8 }; | 23 const uint8 kOrbit[8] = { 1, 2, 3, 4, 5, 6, 7, 8 }; |
25 | 24 |
26 // StrikeRegisterTests don't look at the random bytes so this function can | 25 // StrikeRegisterTests don't look at the random bytes so this function can |
27 // simply set the random bytes to 0. | 26 // simply set the random bytes to 0. |
28 void SetNonce(uint8 nonce[32], unsigned time, const uint8 orbit[8]) { | 27 void SetNonce(uint8 nonce[32], unsigned time, const uint8 orbit[8]) { |
(...skipping 196 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
225 return net::NONCE_INVALID_TIME_FAILURE; | 224 return net::NONCE_INVALID_TIME_FAILURE; |
226 } | 225 } |
227 | 226 |
228 // Check that the timestamp is in the current window. | 227 // Check that the timestamp is in the current window. |
229 if ((current_time > window_secs_ && | 228 if ((current_time > window_secs_ && |
230 nonce_time < (current_time - window_secs_)) || | 229 nonce_time < (current_time - window_secs_)) || |
231 nonce_time > (current_time + window_secs_)) { | 230 nonce_time > (current_time + window_secs_)) { |
232 return net::NONCE_INVALID_TIME_FAILURE; | 231 return net::NONCE_INVALID_TIME_FAILURE; |
233 } | 232 } |
234 | 233 |
235 pair<uint32, string> nonce = make_pair( | 234 pair<uint32, string> nonce = std::make_pair( |
236 nonce_time, | 235 nonce_time, string(reinterpret_cast<const char*>(nonce_bytes), 32)); |
237 string(reinterpret_cast<const char*>(nonce_bytes), 32)); | |
238 | 236 |
239 set<pair<uint32, string> >::const_iterator it = nonces_.find(nonce); | 237 set<pair<uint32, string> >::const_iterator it = nonces_.find(nonce); |
240 if (it != nonces_.end()) { | 238 if (it != nonces_.end()) { |
241 return net::NONCE_NOT_UNIQUE_FAILURE; | 239 return net::NONCE_NOT_UNIQUE_FAILURE; |
242 } | 240 } |
243 | 241 |
244 nonces_.insert(nonce); | 242 nonces_.insert(nonce); |
245 return net::NONCE_OK; | 243 return net::NONCE_OK; |
246 } | 244 } |
247 | 245 |
(...skipping 151 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
399 break; | 397 break; |
400 } | 398 } |
401 } | 399 } |
402 | 400 |
403 if (i != kMaxIterations) { | 401 if (i != kMaxIterations) { |
404 FAIL() << "Failed after " << i << " iterations"; | 402 FAIL() << "Failed after " << i << " iterations"; |
405 } | 403 } |
406 } | 404 } |
407 | 405 |
408 } // anonymous namespace | 406 } // anonymous namespace |
OLD | NEW |