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/core/crypto/p256_key_exchange.h" | 5 #include "net/quic/core/crypto/p256_key_exchange.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 | 8 |
9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "net/quic/platform/api/quic_test.h" |
10 | 10 |
11 using std::string; | 11 using std::string; |
12 | 12 |
13 namespace net { | 13 namespace net { |
14 namespace test { | 14 namespace test { |
15 | 15 |
| 16 class P256KeyExchangeTest : public QuicTest {}; |
| 17 |
16 // SharedKey just tests that the basic key exchange identity holds: that both | 18 // SharedKey just tests that the basic key exchange identity holds: that both |
17 // parties end up with the same key. | 19 // parties end up with the same key. |
18 TEST(P256KeyExchange, SharedKey) { | 20 TEST_F(P256KeyExchangeTest, SharedKey) { |
19 for (int i = 0; i < 5; i++) { | 21 for (int i = 0; i < 5; i++) { |
20 string alice_private(P256KeyExchange::NewPrivateKey()); | 22 string alice_private(P256KeyExchange::NewPrivateKey()); |
21 string bob_private(P256KeyExchange::NewPrivateKey()); | 23 string bob_private(P256KeyExchange::NewPrivateKey()); |
22 | 24 |
23 ASSERT_FALSE(alice_private.empty()); | 25 ASSERT_FALSE(alice_private.empty()); |
24 ASSERT_FALSE(bob_private.empty()); | 26 ASSERT_FALSE(bob_private.empty()); |
25 ASSERT_NE(alice_private, bob_private); | 27 ASSERT_NE(alice_private, bob_private); |
26 | 28 |
27 std::unique_ptr<P256KeyExchange> alice(P256KeyExchange::New(alice_private)); | 29 std::unique_ptr<P256KeyExchange> alice(P256KeyExchange::New(alice_private)); |
28 std::unique_ptr<P256KeyExchange> bob(P256KeyExchange::New(bob_private)); | 30 std::unique_ptr<P256KeyExchange> bob(P256KeyExchange::New(bob_private)); |
29 | 31 |
30 ASSERT_TRUE(alice.get() != nullptr); | 32 ASSERT_TRUE(alice.get() != nullptr); |
31 ASSERT_TRUE(bob.get() != nullptr); | 33 ASSERT_TRUE(bob.get() != nullptr); |
32 | 34 |
33 const QuicStringPiece alice_public(alice->public_value()); | 35 const QuicStringPiece alice_public(alice->public_value()); |
34 const QuicStringPiece bob_public(bob->public_value()); | 36 const QuicStringPiece bob_public(bob->public_value()); |
35 | 37 |
36 std::string alice_shared, bob_shared; | 38 std::string alice_shared, bob_shared; |
37 ASSERT_TRUE(alice->CalculateSharedKey(bob_public, &alice_shared)); | 39 ASSERT_TRUE(alice->CalculateSharedKey(bob_public, &alice_shared)); |
38 ASSERT_TRUE(bob->CalculateSharedKey(alice_public, &bob_shared)); | 40 ASSERT_TRUE(bob->CalculateSharedKey(alice_public, &bob_shared)); |
39 ASSERT_EQ(alice_shared, bob_shared); | 41 ASSERT_EQ(alice_shared, bob_shared); |
40 } | 42 } |
41 } | 43 } |
42 | 44 |
43 } // namespace test | 45 } // namespace test |
44 } // namespace net | 46 } // namespace net |
OLD | NEW |