OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 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/local_strike_register_client.h" | 5 #include "net/quic/crypto/local_strike_register_client.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 | 8 |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
10 #include "base/strings/string_piece.h" | 10 #include "base/strings/string_piece.h" |
11 #include "base/sys_byteorder.h" | 11 #include "base/sys_byteorder.h" |
12 #include "net/quic/crypto/crypto_protocol.h" | 12 #include "net/quic/crypto/crypto_protocol.h" |
13 #include "net/quic/quic_time.h" | 13 #include "net/quic/quic_time.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
15 | 15 |
16 using base::StringPiece; | 16 using base::StringPiece; |
17 using std::string; | 17 using std::string; |
18 | 18 |
19 namespace net { | 19 namespace net { |
20 namespace test { | 20 namespace test { |
21 namespace { | 21 namespace { |
22 | 22 |
23 class RecordResultCallback : public StrikeRegisterClient::ResultCallback { | 23 class RecordResultCallback : public StrikeRegisterClient::ResultCallback { |
24 public: | 24 public: |
25 // RecordResultCallback stores the argument to RunImpl in | 25 // RecordResultCallback stores the argument to RunImpl in |
26 // |*saved_value| and sets |*called| to true. The callback is self | 26 // |*saved_value| and sets |*called| to true. The callback is self |
27 // deleting. | 27 // deleting. |
28 RecordResultCallback(bool* called, bool* saved_value) | 28 RecordResultCallback(bool* called, bool* saved_value) |
29 : called_(called), | 29 : called_(called), saved_value_(saved_value) { |
30 saved_value_(saved_value) { | |
31 *called_ = false; | 30 *called_ = false; |
32 } | 31 } |
33 | 32 |
34 protected: | 33 protected: |
35 virtual void RunImpl(bool nonce_is_valid_and_unique) OVERRIDE { | 34 virtual void RunImpl(bool nonce_is_valid_and_unique) OVERRIDE { |
36 *called_ = true; | 35 *called_ = true; |
37 *saved_value_ = nonce_is_valid_and_unique; | 36 *saved_value_ = nonce_is_valid_and_unique; |
38 } | 37 } |
39 | 38 |
40 private: | 39 private: |
41 bool* called_; | 40 bool* called_; |
42 bool* saved_value_; | 41 bool* saved_value_; |
43 | 42 |
44 DISALLOW_COPY_AND_ASSIGN(RecordResultCallback); | 43 DISALLOW_COPY_AND_ASSIGN(RecordResultCallback); |
45 }; | 44 }; |
46 | 45 |
47 const uint8 kOrbit[] = "\x12\x34\x56\x78\x9A\xBC\xDE\xF0"; | 46 const uint8 kOrbit[] = "\x12\x34\x56\x78\x9A\xBC\xDE\xF0"; |
48 const uint32 kCurrentTimeExternalSecs = 12345678; | 47 const uint32 kCurrentTimeExternalSecs = 12345678; |
49 size_t kMaxEntries = 100; | 48 size_t kMaxEntries = 100; |
50 uint32 kWindowSecs = 60; | 49 uint32 kWindowSecs = 60; |
51 | 50 |
52 class LocalStrikeRegisterClientTest : public ::testing::Test { | 51 class LocalStrikeRegisterClientTest : public ::testing::Test { |
53 protected: | 52 protected: |
54 LocalStrikeRegisterClientTest() { | 53 LocalStrikeRegisterClientTest() {} |
55 } | |
56 | 54 |
57 virtual void SetUp() { | 55 virtual void SetUp() { |
58 strike_register_.reset(new LocalStrikeRegisterClient( | 56 strike_register_.reset(new LocalStrikeRegisterClient( |
59 kMaxEntries, kCurrentTimeExternalSecs, kWindowSecs, kOrbit, | 57 kMaxEntries, |
| 58 kCurrentTimeExternalSecs, |
| 59 kWindowSecs, |
| 60 kOrbit, |
60 net::StrikeRegister::NO_STARTUP_PERIOD_NEEDED)); | 61 net::StrikeRegister::NO_STARTUP_PERIOD_NEEDED)); |
61 } | 62 } |
62 | 63 |
63 scoped_ptr<LocalStrikeRegisterClient> strike_register_; | 64 scoped_ptr<LocalStrikeRegisterClient> strike_register_; |
64 }; | 65 }; |
65 | 66 |
66 TEST_F(LocalStrikeRegisterClientTest, CheckOrbit) { | 67 TEST_F(LocalStrikeRegisterClientTest, CheckOrbit) { |
67 EXPECT_TRUE(strike_register_->IsKnownOrbit( | 68 EXPECT_TRUE(strike_register_->IsKnownOrbit( |
68 StringPiece(reinterpret_cast<const char*>(kOrbit), kOrbitSize))); | 69 StringPiece(reinterpret_cast<const char*>(kOrbit), kOrbitSize))); |
69 EXPECT_FALSE(strike_register_->IsKnownOrbit( | 70 EXPECT_FALSE(strike_register_->IsKnownOrbit( |
(...skipping 47 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
117 QuicWallTime::FromUNIXSeconds(kCurrentTimeExternalSecs), | 118 QuicWallTime::FromUNIXSeconds(kCurrentTimeExternalSecs), |
118 new RecordResultCallback(&called, &is_valid)); | 119 new RecordResultCallback(&called, &is_valid)); |
119 EXPECT_TRUE(called); | 120 EXPECT_TRUE(called); |
120 EXPECT_TRUE(is_valid); | 121 EXPECT_TRUE(is_valid); |
121 } | 122 } |
122 } | 123 } |
123 | 124 |
124 } // namespace | 125 } // namespace |
125 } // namespace test | 126 } // namespace test |
126 } // namespace net | 127 } // namespace net |
OLD | NEW |