net/quic/test_tools/crypto_test_utils_nss.cc

   1 // Copyright 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/test_tools/crypto_test_utils.h"
   6
   7 #include "base/stl_util.h"
   8 #include "base/strings/string_util.h"
   9 #include "crypto/ec_private_key.h"
  10 #include "crypto/ec_signature_creator.h"
  11 #include "net/quic/crypto/channel_id.h"
  12 #include "net/quic/crypto/channel_id_chromium.h"
  13
  14 using base::StringPiece;
  15 using std::string;
  16
  17 namespace net {
  18
  19 namespace test {
  20
  21 class TestChannelIDSource : public ChannelIDSource {
  22  public:
  23   virtual ~TestChannelIDSource() {
  24     STLDeleteValues(&hostname_to_key_);
  25   }
  26
  27   // ChannelIDSource implementation.
  28
  29   virtual QuicAsyncStatus GetChannelIDKey(
  30       const string& hostname,
  31       scoped_ptr<ChannelIDKey>* channel_id_key,
  32       ChannelIDSourceCallback* /*callback*/) OVERRIDE {
  33     channel_id_key->reset(new ChannelIDKeyChromium(HostnameToKey(hostname)));
  34     return QUIC_SUCCESS;
  35   }
  36
  37  private:
  38   typedef std::map<string, crypto::ECPrivateKey*> HostnameToKeyMap;
  39
  40   crypto::ECPrivateKey* HostnameToKey(const string& hostname) {
  41     HostnameToKeyMap::const_iterator it = hostname_to_key_.find(hostname);
  42     if (it != hostname_to_key_.end()) {
  43       return it->second->Copy();
  44     }
  45
  46     crypto::ECPrivateKey* keypair = crypto::ECPrivateKey::Create();
  47     if (!keypair) {
  48       return NULL;
  49     }
  50     hostname_to_key_[hostname] = keypair;
  51     return keypair->Copy();
  52   }
  53
  54   HostnameToKeyMap hostname_to_key_;
  55 };
  56
  57 // static
  58 ChannelIDSource* CryptoTestUtils::ChannelIDSourceForTesting() {
  59   return new TestChannelIDSource();
  60 }
  61
  62 }  // namespace test
  63
  64 }  // namespace net