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