| OLD | NEW |
| (Empty) |
| 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 ~TestChannelIDSource() override { STLDeleteValues(&hostname_to_key_); } | |
| 24 | |
| 25 // ChannelIDSource implementation. | |
| 26 | |
| 27 QuicAsyncStatus GetChannelIDKey( | |
| 28 const string& hostname, | |
| 29 scoped_ptr<ChannelIDKey>* channel_id_key, | |
| 30 ChannelIDSourceCallback* /*callback*/) override { | |
| 31 channel_id_key->reset(new ChannelIDKeyChromium(HostnameToKey(hostname))); | |
| 32 return QUIC_SUCCESS; | |
| 33 } | |
| 34 | |
| 35 private: | |
| 36 typedef std::map<string, crypto::ECPrivateKey*> HostnameToKeyMap; | |
| 37 | |
| 38 crypto::ECPrivateKey* HostnameToKey(const string& hostname) { | |
| 39 HostnameToKeyMap::const_iterator it = hostname_to_key_.find(hostname); | |
| 40 if (it != hostname_to_key_.end()) { | |
| 41 return it->second->Copy(); | |
| 42 } | |
| 43 | |
| 44 crypto::ECPrivateKey* keypair = crypto::ECPrivateKey::Create(); | |
| 45 if (!keypair) { | |
| 46 return nullptr; | |
| 47 } | |
| 48 hostname_to_key_[hostname] = keypair; | |
| 49 return keypair->Copy(); | |
| 50 } | |
| 51 | |
| 52 HostnameToKeyMap hostname_to_key_; | |
| 53 }; | |
| 54 | |
| 55 // static | |
| 56 ChannelIDSource* CryptoTestUtils::ChannelIDSourceForTesting() { | |
| 57 return new TestChannelIDSource(); | |
| 58 } | |
| 59 | |
| 60 } // namespace test | |
| 61 | |
| 62 } // namespace net | |
| OLD | NEW |