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 "base/memory/ptr_util.h" | |
6 #include "base/stl_util.h" | |
7 #include "base/strings/string_util.h" | |
8 #include "crypto/ec_private_key.h" | |
9 #include "crypto/ec_signature_creator.h" | |
10 #include "net/quic/crypto/channel_id.h" | |
11 #include "net/quic/crypto/channel_id_chromium.h" | |
12 #include "net/quic/test_tools/crypto_test_utils.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 std::unique_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 std::unique_ptr<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 base::WrapUnique(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 base::WrapUnique(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 |