| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * libjingle | |
| 3 * Copyright 2015 Google Inc. | |
| 4 * | |
| 5 * Redistribution and use in source and binary forms, with or without | |
| 6 * modification, are permitted provided that the following conditions are met: | |
| 7 * | |
| 8 * 1. Redistributions of source code must retain the above copyright notice, | |
| 9 * this list of conditions and the following disclaimer. | |
| 10 * 2. Redistributions in binary form must reproduce the above copyright notice, | |
| 11 * this list of conditions and the following disclaimer in the documentation | |
| 12 * and/or other materials provided with the distribution. | |
| 13 * 3. The name of the author may not be used to endorse or promote products | |
| 14 * derived from this software without specific prior written permission. | |
| 15 * | |
| 16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED | |
| 17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF | |
| 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO | |
| 19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
| 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
| 21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; | |
| 22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, | |
| 23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR | |
| 24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF | |
| 25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| 26 */ | |
| 27 | |
| 28 #include "talk/app/webrtc/dtlsidentitystore.h" | |
| 29 | |
| 30 #include "talk/app/webrtc/webrtcsessiondescriptionfactory.h" | |
| 31 #include "webrtc/base/gunit.h" | |
| 32 #include "webrtc/base/logging.h" | |
| 33 #include "webrtc/base/ssladapter.h" | |
| 34 | |
| 35 using webrtc::DtlsIdentityStoreImpl; | |
| 36 | |
| 37 static const int kTimeoutMs = 10000; | |
| 38 | |
| 39 class MockDtlsIdentityRequestObserver : | |
| 40 public webrtc::DtlsIdentityRequestObserver { | |
| 41 public: | |
| 42 MockDtlsIdentityRequestObserver() | |
| 43 : call_back_called_(false), last_request_success_(false) {} | |
| 44 void OnFailure(int error) override { | |
| 45 EXPECT_FALSE(call_back_called_); | |
| 46 call_back_called_ = true; | |
| 47 last_request_success_ = false; | |
| 48 } | |
| 49 void OnSuccess(const std::string& der_cert, | |
| 50 const std::string& der_private_key) override { | |
| 51 LOG(LS_WARNING) << "The string version of OnSuccess is called unexpectedly"; | |
| 52 EXPECT_TRUE(false); | |
| 53 } | |
| 54 void OnSuccess(rtc::scoped_ptr<rtc::SSLIdentity> identity) override { | |
| 55 EXPECT_FALSE(call_back_called_); | |
| 56 call_back_called_ = true; | |
| 57 last_request_success_ = true; | |
| 58 } | |
| 59 | |
| 60 void Reset() { | |
| 61 call_back_called_ = false; | |
| 62 last_request_success_ = false; | |
| 63 } | |
| 64 | |
| 65 bool LastRequestSucceeded() const { | |
| 66 return call_back_called_ && last_request_success_; | |
| 67 } | |
| 68 | |
| 69 bool call_back_called() const { | |
| 70 return call_back_called_; | |
| 71 } | |
| 72 | |
| 73 private: | |
| 74 bool call_back_called_; | |
| 75 bool last_request_success_; | |
| 76 }; | |
| 77 | |
| 78 class DtlsIdentityStoreTest : public testing::Test { | |
| 79 protected: | |
| 80 DtlsIdentityStoreTest() | |
| 81 : worker_thread_(new rtc::Thread()), | |
| 82 store_(new DtlsIdentityStoreImpl(rtc::Thread::Current(), | |
| 83 worker_thread_.get())), | |
| 84 observer_( | |
| 85 new rtc::RefCountedObject<MockDtlsIdentityRequestObserver>()) { | |
| 86 RTC_CHECK(worker_thread_->Start()); | |
| 87 } | |
| 88 ~DtlsIdentityStoreTest() {} | |
| 89 | |
| 90 static void SetUpTestCase() { | |
| 91 rtc::InitializeSSL(); | |
| 92 } | |
| 93 static void TearDownTestCase() { | |
| 94 rtc::CleanupSSL(); | |
| 95 } | |
| 96 | |
| 97 rtc::scoped_ptr<rtc::Thread> worker_thread_; | |
| 98 rtc::scoped_ptr<DtlsIdentityStoreImpl> store_; | |
| 99 rtc::scoped_refptr<MockDtlsIdentityRequestObserver> observer_; | |
| 100 }; | |
| 101 | |
| 102 TEST_F(DtlsIdentityStoreTest, RequestIdentitySuccessRSA) { | |
| 103 EXPECT_TRUE_WAIT(store_->HasFreeIdentityForTesting(rtc::KT_RSA), kTimeoutMs); | |
| 104 | |
| 105 store_->RequestIdentity(rtc::KT_RSA, observer_.get()); | |
| 106 EXPECT_TRUE_WAIT(observer_->LastRequestSucceeded(), kTimeoutMs); | |
| 107 | |
| 108 EXPECT_TRUE_WAIT(store_->HasFreeIdentityForTesting(rtc::KT_RSA), kTimeoutMs); | |
| 109 | |
| 110 observer_->Reset(); | |
| 111 | |
| 112 // Verifies that the callback is async when a free identity is ready. | |
| 113 store_->RequestIdentity(rtc::KT_RSA, observer_.get()); | |
| 114 EXPECT_FALSE(observer_->call_back_called()); | |
| 115 EXPECT_TRUE_WAIT(observer_->LastRequestSucceeded(), kTimeoutMs); | |
| 116 } | |
| 117 | |
| 118 TEST_F(DtlsIdentityStoreTest, RequestIdentitySuccessECDSA) { | |
| 119 // Since store currently does not preemptively generate free ECDSA identities | |
| 120 // we do not invoke HasFreeIdentityForTesting between requests. | |
| 121 | |
| 122 store_->RequestIdentity(rtc::KT_ECDSA, observer_.get()); | |
| 123 EXPECT_TRUE_WAIT(observer_->LastRequestSucceeded(), kTimeoutMs); | |
| 124 | |
| 125 observer_->Reset(); | |
| 126 | |
| 127 // Verifies that the callback is async when a free identity is ready. | |
| 128 store_->RequestIdentity(rtc::KT_ECDSA, observer_.get()); | |
| 129 EXPECT_FALSE(observer_->call_back_called()); | |
| 130 EXPECT_TRUE_WAIT(observer_->LastRequestSucceeded(), kTimeoutMs); | |
| 131 } | |
| 132 | |
| 133 TEST_F(DtlsIdentityStoreTest, DeleteStoreEarlyNoCrashRSA) { | |
| 134 EXPECT_FALSE(store_->HasFreeIdentityForTesting(rtc::KT_RSA)); | |
| 135 | |
| 136 store_->RequestIdentity(rtc::KT_RSA, observer_.get()); | |
| 137 store_.reset(); | |
| 138 | |
| 139 worker_thread_->Stop(); | |
| 140 EXPECT_FALSE(observer_->call_back_called()); | |
| 141 } | |
| 142 | |
| 143 TEST_F(DtlsIdentityStoreTest, DeleteStoreEarlyNoCrashECDSA) { | |
| 144 EXPECT_FALSE(store_->HasFreeIdentityForTesting(rtc::KT_ECDSA)); | |
| 145 | |
| 146 store_->RequestIdentity(rtc::KT_ECDSA, observer_.get()); | |
| 147 store_.reset(); | |
| 148 | |
| 149 worker_thread_->Stop(); | |
| 150 EXPECT_FALSE(observer_->call_back_called()); | |
| 151 } | |
| 152 | |
| OLD | NEW |