| 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/curve25519_key_exchange.h" | 5 #include "net/quic/core/crypto/curve25519_key_exchange.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "net/quic/core/crypto/quic_random.h" | 9 #include "net/quic/core/crypto/quic_random.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 | 11 |
| 12 using base::StringPiece; | |
| 13 using std::string; | 12 using std::string; |
| 14 | 13 |
| 15 namespace net { | 14 namespace net { |
| 16 namespace test { | 15 namespace test { |
| 17 | 16 |
| 18 // SharedKey just tests that the basic key exchange identity holds: that both | 17 // SharedKey just tests that the basic key exchange identity holds: that both |
| 19 // parties end up with the same key. | 18 // parties end up with the same key. |
| 20 TEST(Curve25519KeyExchange, SharedKey) { | 19 TEST(Curve25519KeyExchange, SharedKey) { |
| 21 QuicRandom* const rand = QuicRandom::GetInstance(); | 20 QuicRandom* const rand = QuicRandom::GetInstance(); |
| 22 | 21 |
| 23 for (int i = 0; i < 5; i++) { | 22 for (int i = 0; i < 5; i++) { |
| 24 const string alice_key(Curve25519KeyExchange::NewPrivateKey(rand)); | 23 const string alice_key(Curve25519KeyExchange::NewPrivateKey(rand)); |
| 25 const string bob_key(Curve25519KeyExchange::NewPrivateKey(rand)); | 24 const string bob_key(Curve25519KeyExchange::NewPrivateKey(rand)); |
| 26 | 25 |
| 27 std::unique_ptr<Curve25519KeyExchange> alice( | 26 std::unique_ptr<Curve25519KeyExchange> alice( |
| 28 Curve25519KeyExchange::New(alice_key)); | 27 Curve25519KeyExchange::New(alice_key)); |
| 29 std::unique_ptr<Curve25519KeyExchange> bob( | 28 std::unique_ptr<Curve25519KeyExchange> bob( |
| 30 Curve25519KeyExchange::New(bob_key)); | 29 Curve25519KeyExchange::New(bob_key)); |
| 31 | 30 |
| 32 const StringPiece alice_public(alice->public_value()); | 31 const QuicStringPiece alice_public(alice->public_value()); |
| 33 const StringPiece bob_public(bob->public_value()); | 32 const QuicStringPiece bob_public(bob->public_value()); |
| 34 | 33 |
| 35 string alice_shared, bob_shared; | 34 string alice_shared, bob_shared; |
| 36 ASSERT_TRUE(alice->CalculateSharedKey(bob_public, &alice_shared)); | 35 ASSERT_TRUE(alice->CalculateSharedKey(bob_public, &alice_shared)); |
| 37 ASSERT_TRUE(bob->CalculateSharedKey(alice_public, &bob_shared)); | 36 ASSERT_TRUE(bob->CalculateSharedKey(alice_public, &bob_shared)); |
| 38 ASSERT_EQ(alice_shared, bob_shared); | 37 ASSERT_EQ(alice_shared, bob_shared); |
| 39 } | 38 } |
| 40 } | 39 } |
| 41 | 40 |
| 42 } // namespace test | 41 } // namespace test |
| 43 } // namespace net | 42 } // namespace net |
| OLD | NEW |