net/quic/crypto/p256_key_exchange_test.cc

   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
  10 using std::string;
  11
  12 namespace net {
  13 namespace test {
  14
  15 // SharedKey just tests that the basic key exchange identity holds: that both
  16 // parties end up with the same key.
  17 TEST(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