Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(303)

Side by Side Diff: net/quic/crypto/local_strike_register_client_test.cc

Issue 2193073003: Move shared files in net/quic/ into net/quic/core/ (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: io_thread_unittest.cc Created 4 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « net/quic/crypto/local_strike_register_client.cc ('k') | net/quic/crypto/null_decrypter.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
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
3 // found in the LICENSE file.
4
5 #include "net/quic/crypto/local_strike_register_client.h"
6
7 #include <memory>
8
9 #include "base/macros.h"
10 #include "base/strings/string_piece.h"
11 #include "base/sys_byteorder.h"
12 #include "net/quic/crypto/crypto_protocol.h"
13 #include "net/quic/quic_time.h"
14 #include "testing/gtest/include/gtest/gtest.h"
15
16 using base::StringPiece;
17 using std::string;
18
19 namespace net {
20 namespace test {
21 namespace {
22
23 class RecordResultCallback : public StrikeRegisterClient::ResultCallback {
24 public:
25 // RecordResultCallback stores the argument to RunImpl in
26 // |*saved_value| and sets |*called| to true. The callback is self
27 // deleting.
28 RecordResultCallback(bool* called,
29 bool* saved_value,
30 InsertStatus* saved_nonce_error)
31 : called_(called),
32 saved_value_(saved_value),
33 saved_nonce_error_(saved_nonce_error) {
34 *called_ = false;
35 }
36
37 protected:
38 void RunImpl(bool nonce_is_valid_and_unique,
39 InsertStatus nonce_error) override {
40 *called_ = true;
41 *saved_value_ = nonce_is_valid_and_unique;
42 *saved_nonce_error_ = nonce_error;
43 }
44
45 private:
46 bool* called_;
47 bool* saved_value_;
48 InsertStatus* saved_nonce_error_;
49
50 DISALLOW_COPY_AND_ASSIGN(RecordResultCallback);
51 };
52
53 const uint8_t kOrbit[] = "\x12\x34\x56\x78\x9A\xBC\xDE\xF0";
54 const uint32_t kCurrentTimeExternalSecs = 12345678;
55 size_t kMaxEntries = 100;
56 uint32_t kWindowSecs = 60;
57
58 class LocalStrikeRegisterClientTest : public ::testing::Test {
59 protected:
60 LocalStrikeRegisterClientTest() {}
61
62 void SetUp() override {
63 strike_register_.reset(new LocalStrikeRegisterClient(
64 kMaxEntries, kCurrentTimeExternalSecs, kWindowSecs, kOrbit,
65 StrikeRegister::NO_STARTUP_PERIOD_NEEDED));
66 }
67
68 std::unique_ptr<LocalStrikeRegisterClient> strike_register_;
69 };
70
71 TEST_F(LocalStrikeRegisterClientTest, CheckOrbit) {
72 EXPECT_TRUE(strike_register_->IsKnownOrbit(
73 StringPiece(reinterpret_cast<const char*>(kOrbit), kOrbitSize)));
74 EXPECT_FALSE(strike_register_->IsKnownOrbit(
75 StringPiece(reinterpret_cast<const char*>(kOrbit), kOrbitSize - 1)));
76 EXPECT_FALSE(strike_register_->IsKnownOrbit(
77 StringPiece(reinterpret_cast<const char*>(kOrbit), kOrbitSize + 1)));
78 EXPECT_FALSE(strike_register_->IsKnownOrbit(
79 StringPiece(reinterpret_cast<const char*>(kOrbit) + 1, kOrbitSize)));
80 }
81
82 TEST_F(LocalStrikeRegisterClientTest, IncorrectNonceLength) {
83 string valid_nonce;
84 uint32_t norder = htonl(kCurrentTimeExternalSecs);
85 valid_nonce.assign(reinterpret_cast<const char*>(&norder), sizeof(norder));
86 valid_nonce.append(string(reinterpret_cast<const char*>(kOrbit), kOrbitSize));
87 valid_nonce.append(string(20, '\x17')); // 20 'random' bytes.
88
89 {
90 // Validation fails if you remove a byte from the nonce.
91 bool called = false;
92 bool is_valid = false;
93 InsertStatus nonce_error = NONCE_UNKNOWN_FAILURE;
94 string short_nonce = valid_nonce.substr(0, valid_nonce.length() - 1);
95 strike_register_->VerifyNonceIsValidAndUnique(
96 short_nonce, QuicWallTime::FromUNIXSeconds(kCurrentTimeExternalSecs),
97 new RecordResultCallback(&called, &is_valid, &nonce_error));
98 EXPECT_TRUE(called);
99 EXPECT_FALSE(is_valid);
100 EXPECT_EQ(NONCE_INVALID_FAILURE, nonce_error);
101 }
102
103 {
104 // Validation fails if you add a byte to the nonce.
105 bool called = false;
106 bool is_valid = false;
107 InsertStatus nonce_error = NONCE_UNKNOWN_FAILURE;
108 string long_nonce(valid_nonce);
109 long_nonce.append("a");
110 strike_register_->VerifyNonceIsValidAndUnique(
111 long_nonce, QuicWallTime::FromUNIXSeconds(kCurrentTimeExternalSecs),
112 new RecordResultCallback(&called, &is_valid, &nonce_error));
113 EXPECT_TRUE(called);
114 EXPECT_FALSE(is_valid);
115 EXPECT_EQ(NONCE_INVALID_FAILURE, nonce_error);
116 }
117
118 {
119 // Verify that the base nonce validates was valid.
120 bool called = false;
121 bool is_valid = false;
122 InsertStatus nonce_error = NONCE_UNKNOWN_FAILURE;
123 strike_register_->VerifyNonceIsValidAndUnique(
124 valid_nonce, QuicWallTime::FromUNIXSeconds(kCurrentTimeExternalSecs),
125 new RecordResultCallback(&called, &is_valid, &nonce_error));
126 EXPECT_TRUE(called);
127 EXPECT_TRUE(is_valid);
128 EXPECT_EQ(NONCE_OK, nonce_error);
129 }
130 }
131
132 } // namespace
133 } // namespace test
134 } // namespace net
OLDNEW
« no previous file with comments | « net/quic/crypto/local_strike_register_client.cc ('k') | net/quic/crypto/null_decrypter.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698