OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "crypto/p224_spake.h" | 5 #include "crypto/p224_spake.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/strings/string_number_conversions.h" | 10 #include "base/strings/string_number_conversions.h" |
11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
12 | 12 |
13 namespace crypto { | 13 namespace crypto { |
14 | 14 |
15 namespace { | 15 namespace { |
16 | 16 |
17 std::string HexEncodeString(const std::string& binary_data) { | 17 std::string HexEncodeString(const std::string& binary_data) { |
18 return base::HexEncode(binary_data.c_str(), binary_data.size()); | 18 return base::HexEncode(binary_data.c_str(), binary_data.size()); |
19 } | 19 } |
20 | 20 |
21 bool RunExchange(P224EncryptedKeyExchange* client, | 21 bool RunExchange(P224EncryptedKeyExchange* client, |
22 P224EncryptedKeyExchange* server, | 22 P224EncryptedKeyExchange* server, |
23 bool is_password_same) { | 23 bool is_password_same) { |
24 for (;;) { | 24 for (;;) { |
25 std::string client_message, server_message; | 25 std::string client_message, server_message; |
26 client_message = client->GetMessage(); | 26 client_message = client->GetNextMessage(); |
27 server_message = server->GetMessage(); | 27 server_message = server->GetNextMessage(); |
28 | 28 |
29 P224EncryptedKeyExchange::Result client_result, server_result; | 29 P224EncryptedKeyExchange::Result client_result, server_result; |
30 client_result = client->ProcessMessage(server_message); | 30 client_result = client->ProcessMessage(server_message); |
31 server_result = server->ProcessMessage(client_message); | 31 server_result = server->ProcessMessage(client_message); |
32 | 32 |
33 // Check that we never hit the case where only one succeeds. | 33 // Check that we never hit the case where only one succeeds. |
34 EXPECT_EQ(client_result == P224EncryptedKeyExchange::kResultSuccess, | 34 EXPECT_EQ(client_result == P224EncryptedKeyExchange::kResultSuccess, |
35 server_result == P224EncryptedKeyExchange::kResultSuccess); | 35 server_result == P224EncryptedKeyExchange::kResultSuccess); |
36 | 36 |
37 if (client_result == P224EncryptedKeyExchange::kResultFailed || | 37 if (client_result == P224EncryptedKeyExchange::kResultFailed || |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
78 } | 78 } |
79 | 79 |
80 TEST(MutualAuth, ExpectedValues) { | 80 TEST(MutualAuth, ExpectedValues) { |
81 P224EncryptedKeyExchange client(P224EncryptedKeyExchange::kPeerTypeClient, | 81 P224EncryptedKeyExchange client(P224EncryptedKeyExchange::kPeerTypeClient, |
82 kPassword); | 82 kPassword); |
83 client.SetXForTesting("Client x"); | 83 client.SetXForTesting("Client x"); |
84 P224EncryptedKeyExchange server(P224EncryptedKeyExchange::kPeerTypeServer, | 84 P224EncryptedKeyExchange server(P224EncryptedKeyExchange::kPeerTypeServer, |
85 kPassword); | 85 kPassword); |
86 server.SetXForTesting("Server x"); | 86 server.SetXForTesting("Server x"); |
87 | 87 |
88 std::string client_message = client.GetMessage(); | 88 std::string client_message = client.GetNextMessage(); |
89 EXPECT_EQ( | 89 EXPECT_EQ( |
90 "3508EF7DECC8AB9F9C439FBB0154288BBECC0A82E8448F4CF29554EB" | 90 "3508EF7DECC8AB9F9C439FBB0154288BBECC0A82E8448F4CF29554EB" |
91 "BE9D486686226255EAD1D077C635B1A41F46AC91D7F7F32CED9EC3E0", | 91 "BE9D486686226255EAD1D077C635B1A41F46AC91D7F7F32CED9EC3E0", |
92 HexEncodeString(client_message)); | 92 HexEncodeString(client_message)); |
93 | 93 |
94 std::string server_message = server.GetMessage(); | 94 std::string server_message = server.GetNextMessage(); |
95 EXPECT_EQ( | 95 EXPECT_EQ( |
96 "A3088C18B75D2C2B107105661AEC85424777475EB29F1DDFB8C14AFB" | 96 "A3088C18B75D2C2B107105661AEC85424777475EB29F1DDFB8C14AFB" |
97 "F1603D0DF38413A00F420ACF2059E7997C935F5A957A193D09A2B584", | 97 "F1603D0DF38413A00F420ACF2059E7997C935F5A957A193D09A2B584", |
98 HexEncodeString(server_message)); | 98 HexEncodeString(server_message)); |
99 | 99 |
100 EXPECT_EQ(P224EncryptedKeyExchange::kResultPending, | 100 EXPECT_EQ(P224EncryptedKeyExchange::kResultPending, |
101 client.ProcessMessage(server_message)); | 101 client.ProcessMessage(server_message)); |
102 EXPECT_EQ(P224EncryptedKeyExchange::kResultPending, | 102 EXPECT_EQ(P224EncryptedKeyExchange::kResultPending, |
103 server.ProcessMessage(client_message)); | 103 server.ProcessMessage(client_message)); |
104 | 104 |
(...skipping 17 matching lines...) Expand all Loading... |
122 P224EncryptedKeyExchange server( | 122 P224EncryptedKeyExchange server( |
123 P224EncryptedKeyExchange::kPeerTypeServer, kPassword); | 123 P224EncryptedKeyExchange::kPeerTypeServer, kPassword); |
124 | 124 |
125 // We'll only be testing small values of i, but we don't want that to bias | 125 // We'll only be testing small values of i, but we don't want that to bias |
126 // the test coverage. So we disperse the value of i by multiplying by the | 126 // the test coverage. So we disperse the value of i by multiplying by the |
127 // FNV, 32-bit prime, producing a poor-man's PRNG. | 127 // FNV, 32-bit prime, producing a poor-man's PRNG. |
128 const uint32 rand = i * 16777619; | 128 const uint32 rand = i * 16777619; |
129 | 129 |
130 for (unsigned round = 0;; round++) { | 130 for (unsigned round = 0;; round++) { |
131 std::string client_message, server_message; | 131 std::string client_message, server_message; |
132 client_message = client.GetMessage(); | 132 client_message = client.GetNextMessage(); |
133 server_message = server.GetMessage(); | 133 server_message = server.GetNextMessage(); |
134 | 134 |
135 if ((rand & 1) == round) { | 135 if ((rand & 1) == round) { |
136 const bool server_or_client = rand & 2; | 136 const bool server_or_client = rand & 2; |
137 std::string* m = server_or_client ? &server_message : &client_message; | 137 std::string* m = server_or_client ? &server_message : &client_message; |
138 if (rand & 4) { | 138 if (rand & 4) { |
139 // Truncate | 139 // Truncate |
140 *m = m->substr(0, (i >> 3) % m->size()); | 140 *m = m->substr(0, (i >> 3) % m->size()); |
141 } else { | 141 } else { |
142 // Corrupt | 142 // Corrupt |
143 const size_t bits = m->size() * 8; | 143 const size_t bits = m->size() * 8; |
(...skipping 21 matching lines...) Expand all Loading... |
165 | 165 |
166 ASSERT_EQ(P224EncryptedKeyExchange::kResultPending, | 166 ASSERT_EQ(P224EncryptedKeyExchange::kResultPending, |
167 client_result); | 167 client_result); |
168 ASSERT_EQ(P224EncryptedKeyExchange::kResultPending, | 168 ASSERT_EQ(P224EncryptedKeyExchange::kResultPending, |
169 server_result); | 169 server_result); |
170 } | 170 } |
171 } | 171 } |
172 } | 172 } |
173 | 173 |
174 } // namespace crypto | 174 } // namespace crypto |
OLD | NEW |