| Index: webrtc/base/sslstreamadapter_unittest.cc
|
| diff --git a/webrtc/base/sslstreamadapter_unittest.cc b/webrtc/base/sslstreamadapter_unittest.cc
|
| deleted file mode 100644
|
| index b5184e6d44e7179f5c317eeb52eb536e6697559e..0000000000000000000000000000000000000000
|
| --- a/webrtc/base/sslstreamadapter_unittest.cc
|
| +++ /dev/null
|
| @@ -1,1352 +0,0 @@
|
| -/*
|
| - * Copyright 2011 The WebRTC Project Authors. All rights reserved.
|
| - *
|
| - * Use of this source code is governed by a BSD-style license
|
| - * that can be found in the LICENSE file in the root of the source
|
| - * tree. An additional intellectual property rights grant can be found
|
| - * in the file PATENTS. All contributing project authors may
|
| - * be found in the AUTHORS file in the root of the source tree.
|
| - */
|
| -
|
| -
|
| -#include <algorithm>
|
| -#include <memory>
|
| -#include <set>
|
| -#include <string>
|
| -
|
| -#include "webrtc/base/bufferqueue.h"
|
| -#include "webrtc/base/checks.h"
|
| -#include "webrtc/base/gunit.h"
|
| -#include "webrtc/base/helpers.h"
|
| -#include "webrtc/base/ssladapter.h"
|
| -#include "webrtc/base/sslidentity.h"
|
| -#include "webrtc/base/sslstreamadapter.h"
|
| -#include "webrtc/base/stream.h"
|
| -
|
| -using ::testing::WithParamInterface;
|
| -using ::testing::Values;
|
| -using ::testing::Combine;
|
| -using ::testing::tuple;
|
| -
|
| -static const int kBlockSize = 4096;
|
| -static const char kExporterLabel[] = "label";
|
| -static const unsigned char kExporterContext[] = "context";
|
| -static int kExporterContextLen = sizeof(kExporterContext);
|
| -
|
| -static const char kRSA_PRIVATE_KEY_PEM[] =
|
| - "-----BEGIN RSA PRIVATE KEY-----\n"
|
| - "MIICdwIBADANBgkqhkiG9w0BAQEFAASCAmEwggJdAgEAAoGBAMYRkbhmI7kVA/rM\n"
|
| - "czsZ+6JDhDvnkF+vn6yCAGuRPV03zuRqZtDy4N4to7PZu9PjqrRl7nDMXrG3YG9y\n"
|
| - "rlIAZ72KjcKKFAJxQyAKLCIdawKRyp8RdK3LEySWEZb0AV58IadqPZDTNHHRX8dz\n"
|
| - "5aTSMsbbkZ+C/OzTnbiMqLL/vg6jAgMBAAECgYAvgOs4FJcgvp+TuREx7YtiYVsH\n"
|
| - "mwQPTum2z/8VzWGwR8BBHBvIpVe1MbD/Y4seyI2aco/7UaisatSgJhsU46/9Y4fq\n"
|
| - "2TwXH9QANf4at4d9n/R6rzwpAJOpgwZgKvdQjkfrKTtgLV+/dawvpxUYkRH4JZM1\n"
|
| - "CVGukMfKNrSVH4Ap4QJBAOJmGV1ASPnB4r4nc99at7JuIJmd7fmuVUwUgYi4XgaR\n"
|
| - "WhScBsgYwZ/JoywdyZJgnbcrTDuVcWG56B3vXbhdpMsCQQDf9zeJrjnPZ3Cqm79y\n"
|
| - "kdqANep0uwZciiNiWxsQrCHztywOvbFhdp8iYVFG9EK8DMY41Y5TxUwsHD+67zao\n"
|
| - "ZNqJAkEA1suLUP/GvL8IwuRneQd2tWDqqRQ/Td3qq03hP7e77XtF/buya3Ghclo5\n"
|
| - "54czUR89QyVfJEC6278nzA7n2h1uVQJAcG6mztNL6ja/dKZjYZye2CY44QjSlLo0\n"
|
| - "MTgTSjdfg/28fFn2Jjtqf9Pi/X+50LWI/RcYMC2no606wRk9kyOuIQJBAK6VSAim\n"
|
| - "1pOEjsYQn0X5KEIrz1G3bfCbB848Ime3U2/FWlCHMr6ch8kCZ5d1WUeJD3LbwMNG\n"
|
| - "UCXiYxSsu20QNVw=\n"
|
| - "-----END RSA PRIVATE KEY-----\n";
|
| -
|
| -static const char kCERT_PEM[] =
|
| - "-----BEGIN CERTIFICATE-----\n"
|
| - "MIIBmTCCAQKgAwIBAgIEbzBSAjANBgkqhkiG9w0BAQsFADARMQ8wDQYDVQQDEwZX\n"
|
| - "ZWJSVEMwHhcNMTQwMTAyMTgyNDQ3WhcNMTQwMjAxMTgyNDQ3WjARMQ8wDQYDVQQD\n"
|
| - "EwZXZWJSVEMwgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAMYRkbhmI7kVA/rM\n"
|
| - "czsZ+6JDhDvnkF+vn6yCAGuRPV03zuRqZtDy4N4to7PZu9PjqrRl7nDMXrG3YG9y\n"
|
| - "rlIAZ72KjcKKFAJxQyAKLCIdawKRyp8RdK3LEySWEZb0AV58IadqPZDTNHHRX8dz\n"
|
| - "5aTSMsbbkZ+C/OzTnbiMqLL/vg6jAgMBAAEwDQYJKoZIhvcNAQELBQADgYEAUflI\n"
|
| - "VUe5Krqf5RVa5C3u/UTAOAUJBiDS3VANTCLBxjuMsvqOG0WvaYWP3HYPgrz0jXK2\n"
|
| - "LJE/mGw3MyFHEqi81jh95J+ypl6xKW6Rm8jKLR87gUvCaVYn/Z4/P3AqcQTB7wOv\n"
|
| - "UD0A8qfhfDM+LK6rPAnCsVN0NRDY3jvd6rzix9M=\n"
|
| - "-----END CERTIFICATE-----\n";
|
| -
|
| -class SSLStreamAdapterTestBase;
|
| -
|
| -class SSLDummyStreamBase : public rtc::StreamInterface,
|
| - public sigslot::has_slots<> {
|
| - public:
|
| - SSLDummyStreamBase(SSLStreamAdapterTestBase* test,
|
| - const std::string &side,
|
| - rtc::StreamInterface* in,
|
| - rtc::StreamInterface* out) :
|
| - test_base_(test),
|
| - side_(side),
|
| - in_(in),
|
| - out_(out),
|
| - first_packet_(true) {
|
| - in_->SignalEvent.connect(this, &SSLDummyStreamBase::OnEventIn);
|
| - out_->SignalEvent.connect(this, &SSLDummyStreamBase::OnEventOut);
|
| - }
|
| -
|
| - rtc::StreamState GetState() const override { return rtc::SS_OPEN; }
|
| -
|
| - rtc::StreamResult Read(void* buffer, size_t buffer_len,
|
| - size_t* read, int* error) override {
|
| - rtc::StreamResult r;
|
| -
|
| - r = in_->Read(buffer, buffer_len, read, error);
|
| - if (r == rtc::SR_BLOCK)
|
| - return rtc::SR_BLOCK;
|
| - if (r == rtc::SR_EOS)
|
| - return rtc::SR_EOS;
|
| -
|
| - if (r != rtc::SR_SUCCESS) {
|
| - ADD_FAILURE();
|
| - return rtc::SR_ERROR;
|
| - }
|
| -
|
| - return rtc::SR_SUCCESS;
|
| - }
|
| -
|
| - // Catch readability events on in and pass them up.
|
| - void OnEventIn(rtc::StreamInterface* stream, int sig, int err) {
|
| - int mask = (rtc::SE_READ | rtc::SE_CLOSE);
|
| -
|
| - if (sig & mask) {
|
| - LOG(LS_VERBOSE) << "SSLDummyStreamBase::OnEvent side=" << side_
|
| - << " sig=" << sig << " forwarding upward";
|
| - PostEvent(sig & mask, 0);
|
| - }
|
| - }
|
| -
|
| - // Catch writeability events on out and pass them up.
|
| - void OnEventOut(rtc::StreamInterface* stream, int sig, int err) {
|
| - if (sig & rtc::SE_WRITE) {
|
| - LOG(LS_VERBOSE) << "SSLDummyStreamBase::OnEvent side=" << side_
|
| - << " sig=" << sig << " forwarding upward";
|
| -
|
| - PostEvent(sig & rtc::SE_WRITE, 0);
|
| - }
|
| - }
|
| -
|
| - // Write to the outgoing FifoBuffer
|
| - rtc::StreamResult WriteData(const void* data, size_t data_len,
|
| - size_t* written, int* error) {
|
| - return out_->Write(data, data_len, written, error);
|
| - }
|
| -
|
| - rtc::StreamResult Write(const void* data, size_t data_len,
|
| - size_t* written, int* error) override;
|
| -
|
| - void Close() override {
|
| - LOG(LS_INFO) << "Closing outbound stream";
|
| - out_->Close();
|
| - }
|
| -
|
| - protected:
|
| - SSLStreamAdapterTestBase* test_base_;
|
| - const std::string side_;
|
| - rtc::StreamInterface* in_;
|
| - rtc::StreamInterface* out_;
|
| - bool first_packet_;
|
| -};
|
| -
|
| -class SSLDummyStreamTLS : public SSLDummyStreamBase {
|
| - public:
|
| - SSLDummyStreamTLS(SSLStreamAdapterTestBase* test,
|
| - const std::string& side,
|
| - rtc::FifoBuffer* in,
|
| - rtc::FifoBuffer* out) :
|
| - SSLDummyStreamBase(test, side, in, out) {
|
| - }
|
| -};
|
| -
|
| -class BufferQueueStream : public rtc::BufferQueue,
|
| - public rtc::StreamInterface {
|
| - public:
|
| - BufferQueueStream(size_t capacity, size_t default_size)
|
| - : rtc::BufferQueue(capacity, default_size) {
|
| - }
|
| -
|
| - // Implementation of abstract StreamInterface methods.
|
| -
|
| - // A buffer queue stream is always "open".
|
| - rtc::StreamState GetState() const override { return rtc::SS_OPEN; }
|
| -
|
| - // Reading a buffer queue stream will either succeed or block.
|
| - rtc::StreamResult Read(void* buffer, size_t buffer_len,
|
| - size_t* read, int* error) override {
|
| - if (!ReadFront(buffer, buffer_len, read)) {
|
| - return rtc::SR_BLOCK;
|
| - }
|
| - return rtc::SR_SUCCESS;
|
| - }
|
| -
|
| - // Writing to a buffer queue stream will either succeed or block.
|
| - rtc::StreamResult Write(const void* data, size_t data_len,
|
| - size_t* written, int* error) override {
|
| - if (!WriteBack(data, data_len, written)) {
|
| - return rtc::SR_BLOCK;
|
| - }
|
| - return rtc::SR_SUCCESS;
|
| - }
|
| -
|
| - // A buffer queue stream can not be closed.
|
| - void Close() override {}
|
| -
|
| - protected:
|
| - void NotifyReadableForTest() override {
|
| - PostEvent(rtc::SE_READ, 0);
|
| - }
|
| -
|
| - void NotifyWritableForTest() override {
|
| - PostEvent(rtc::SE_WRITE, 0);
|
| - }
|
| -};
|
| -
|
| -class SSLDummyStreamDTLS : public SSLDummyStreamBase {
|
| - public:
|
| - SSLDummyStreamDTLS(SSLStreamAdapterTestBase* test,
|
| - const std::string& side,
|
| - BufferQueueStream* in,
|
| - BufferQueueStream* out) :
|
| - SSLDummyStreamBase(test, side, in, out) {
|
| - }
|
| -};
|
| -
|
| -static const int kFifoBufferSize = 4096;
|
| -static const int kBufferCapacity = 1;
|
| -static const size_t kDefaultBufferSize = 2048;
|
| -
|
| -class SSLStreamAdapterTestBase : public testing::Test,
|
| - public sigslot::has_slots<> {
|
| - public:
|
| - SSLStreamAdapterTestBase(
|
| - const std::string& client_cert_pem,
|
| - const std::string& client_private_key_pem,
|
| - bool dtls,
|
| - rtc::KeyParams client_key_type = rtc::KeyParams(rtc::KT_DEFAULT),
|
| - rtc::KeyParams server_key_type = rtc::KeyParams(rtc::KT_DEFAULT))
|
| - : client_cert_pem_(client_cert_pem),
|
| - client_private_key_pem_(client_private_key_pem),
|
| - client_key_type_(client_key_type),
|
| - server_key_type_(server_key_type),
|
| - client_stream_(nullptr),
|
| - server_stream_(nullptr),
|
| - client_identity_(nullptr),
|
| - server_identity_(nullptr),
|
| - delay_(0),
|
| - mtu_(1460),
|
| - loss_(0),
|
| - lose_first_packet_(false),
|
| - damage_(false),
|
| - dtls_(dtls),
|
| - handshake_wait_(5000),
|
| - identities_set_(false) {
|
| - // Set use of the test RNG to get predictable loss patterns.
|
| - rtc::SetRandomTestMode(true);
|
| - }
|
| -
|
| - ~SSLStreamAdapterTestBase() {
|
| - // Put it back for the next test.
|
| - rtc::SetRandomTestMode(false);
|
| - }
|
| -
|
| - void SetUp() override {
|
| - CreateStreams();
|
| -
|
| - client_ssl_.reset(rtc::SSLStreamAdapter::Create(client_stream_));
|
| - server_ssl_.reset(rtc::SSLStreamAdapter::Create(server_stream_));
|
| -
|
| - // Set up the slots
|
| - client_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
|
| - server_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
|
| -
|
| - if (!client_cert_pem_.empty() && !client_private_key_pem_.empty()) {
|
| - client_identity_ = rtc::SSLIdentity::FromPEMStrings(
|
| - client_private_key_pem_, client_cert_pem_);
|
| - } else {
|
| - client_identity_ = rtc::SSLIdentity::Generate("client", client_key_type_);
|
| - }
|
| - server_identity_ = rtc::SSLIdentity::Generate("server", server_key_type_);
|
| -
|
| - client_ssl_->SetIdentity(client_identity_);
|
| - server_ssl_->SetIdentity(server_identity_);
|
| - }
|
| -
|
| - void TearDown() override {
|
| - client_ssl_.reset(nullptr);
|
| - server_ssl_.reset(nullptr);
|
| - }
|
| -
|
| - virtual void CreateStreams() = 0;
|
| -
|
| - // Recreate the client/server identities with the specified validity period.
|
| - // |not_before| and |not_after| are offsets from the current time in number
|
| - // of seconds.
|
| - void ResetIdentitiesWithValidity(int not_before, int not_after) {
|
| - CreateStreams();
|
| -
|
| - client_ssl_.reset(rtc::SSLStreamAdapter::Create(client_stream_));
|
| - server_ssl_.reset(rtc::SSLStreamAdapter::Create(server_stream_));
|
| -
|
| - client_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
|
| - server_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
|
| -
|
| - time_t now = time(nullptr);
|
| -
|
| - rtc::SSLIdentityParams client_params;
|
| - client_params.key_params = rtc::KeyParams(rtc::KT_DEFAULT);
|
| - client_params.common_name = "client";
|
| - client_params.not_before = now + not_before;
|
| - client_params.not_after = now + not_after;
|
| - client_identity_ = rtc::SSLIdentity::GenerateForTest(client_params);
|
| -
|
| - rtc::SSLIdentityParams server_params;
|
| - server_params.key_params = rtc::KeyParams(rtc::KT_DEFAULT);
|
| - server_params.common_name = "server";
|
| - server_params.not_before = now + not_before;
|
| - server_params.not_after = now + not_after;
|
| - server_identity_ = rtc::SSLIdentity::GenerateForTest(server_params);
|
| -
|
| - client_ssl_->SetIdentity(client_identity_);
|
| - server_ssl_->SetIdentity(server_identity_);
|
| - }
|
| -
|
| - virtual void OnEvent(rtc::StreamInterface *stream, int sig, int err) {
|
| - LOG(LS_VERBOSE) << "SSLStreamAdapterTestBase::OnEvent sig=" << sig;
|
| -
|
| - if (sig & rtc::SE_READ) {
|
| - ReadData(stream);
|
| - }
|
| -
|
| - if ((stream == client_ssl_.get()) && (sig & rtc::SE_WRITE)) {
|
| - WriteData();
|
| - }
|
| - }
|
| -
|
| - void SetPeerIdentitiesByDigest(bool correct, bool expect_success) {
|
| - unsigned char server_digest[20];
|
| - size_t server_digest_len;
|
| - unsigned char client_digest[20];
|
| - size_t client_digest_len;
|
| - bool rv;
|
| - rtc::SSLPeerCertificateDigestError err;
|
| - rtc::SSLPeerCertificateDigestError expected_err =
|
| - expect_success
|
| - ? rtc::SSLPeerCertificateDigestError::NONE
|
| - : rtc::SSLPeerCertificateDigestError::VERIFICATION_FAILED;
|
| -
|
| - LOG(LS_INFO) << "Setting peer identities by digest";
|
| -
|
| - rv = server_identity_->certificate().ComputeDigest(
|
| - rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
|
| - ASSERT_TRUE(rv);
|
| - rv = client_identity_->certificate().ComputeDigest(
|
| - rtc::DIGEST_SHA_1, client_digest, 20, &client_digest_len);
|
| - ASSERT_TRUE(rv);
|
| -
|
| - if (!correct) {
|
| - LOG(LS_INFO) << "Setting bogus digest for server cert";
|
| - server_digest[0]++;
|
| - }
|
| - rv = client_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, server_digest,
|
| - server_digest_len, &err);
|
| - EXPECT_EQ(expected_err, err);
|
| - EXPECT_EQ(expect_success, rv);
|
| -
|
| - if (!correct) {
|
| - LOG(LS_INFO) << "Setting bogus digest for client cert";
|
| - client_digest[0]++;
|
| - }
|
| - rv = server_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, client_digest,
|
| - client_digest_len, &err);
|
| - EXPECT_EQ(expected_err, err);
|
| - EXPECT_EQ(expect_success, rv);
|
| -
|
| - identities_set_ = true;
|
| - }
|
| -
|
| - void SetupProtocolVersions(rtc::SSLProtocolVersion server_version,
|
| - rtc::SSLProtocolVersion client_version) {
|
| - server_ssl_->SetMaxProtocolVersion(server_version);
|
| - client_ssl_->SetMaxProtocolVersion(client_version);
|
| - }
|
| -
|
| - void TestHandshake(bool expect_success = true) {
|
| - server_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS :
|
| - rtc::SSL_MODE_TLS);
|
| - client_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS :
|
| - rtc::SSL_MODE_TLS);
|
| -
|
| - if (!dtls_) {
|
| - // Make sure we simulate a reliable network for TLS.
|
| - // This is just a check to make sure that people don't write wrong
|
| - // tests.
|
| - RTC_CHECK_EQ(1460, mtu_);
|
| - RTC_CHECK(!loss_);
|
| - RTC_CHECK(!lose_first_packet_);
|
| - }
|
| -
|
| - if (!identities_set_)
|
| - SetPeerIdentitiesByDigest(true, true);
|
| -
|
| - // Start the handshake
|
| - int rv;
|
| -
|
| - server_ssl_->SetServerRole();
|
| - rv = server_ssl_->StartSSL();
|
| - ASSERT_EQ(0, rv);
|
| -
|
| - rv = client_ssl_->StartSSL();
|
| - ASSERT_EQ(0, rv);
|
| -
|
| - // Now run the handshake
|
| - if (expect_success) {
|
| - EXPECT_TRUE_WAIT((client_ssl_->GetState() == rtc::SS_OPEN)
|
| - && (server_ssl_->GetState() == rtc::SS_OPEN),
|
| - handshake_wait_);
|
| - } else {
|
| - EXPECT_TRUE_WAIT(client_ssl_->GetState() == rtc::SS_CLOSED,
|
| - handshake_wait_);
|
| - }
|
| - }
|
| -
|
| - // This tests that the handshake can complete before the identity is
|
| - // verified, and the identity will be verified after the fact.
|
| - void TestHandshakeWithDelayedIdentity(bool valid_identity) {
|
| - server_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
|
| - client_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
|
| -
|
| - if (!dtls_) {
|
| - // Make sure we simulate a reliable network for TLS.
|
| - // This is just a check to make sure that people don't write wrong
|
| - // tests.
|
| - RTC_CHECK_EQ(1460, mtu_);
|
| - RTC_CHECK(!loss_);
|
| - RTC_CHECK(!lose_first_packet_);
|
| - }
|
| -
|
| - // Start the handshake
|
| - int rv;
|
| -
|
| - server_ssl_->SetServerRole();
|
| - rv = server_ssl_->StartSSL();
|
| - ASSERT_EQ(0, rv);
|
| -
|
| - rv = client_ssl_->StartSSL();
|
| - ASSERT_EQ(0, rv);
|
| -
|
| - // Now run the handshake.
|
| - EXPECT_TRUE_WAIT(
|
| - client_ssl_->IsTlsConnected() && server_ssl_->IsTlsConnected(),
|
| - handshake_wait_);
|
| -
|
| - // Until the identity has been verified, the state should still be
|
| - // SS_OPENING and writes should return SR_BLOCK.
|
| - EXPECT_EQ(rtc::SS_OPENING, client_ssl_->GetState());
|
| - EXPECT_EQ(rtc::SS_OPENING, server_ssl_->GetState());
|
| - unsigned char packet[1];
|
| - size_t sent;
|
| - EXPECT_EQ(rtc::SR_BLOCK, client_ssl_->Write(&packet, 1, &sent, 0));
|
| - EXPECT_EQ(rtc::SR_BLOCK, server_ssl_->Write(&packet, 1, &sent, 0));
|
| -
|
| - // If we set an invalid identity at this point, SetPeerCertificateDigest
|
| - // should return false.
|
| - SetPeerIdentitiesByDigest(valid_identity, valid_identity);
|
| - // State should then transition to SS_OPEN or SS_CLOSED based on validation
|
| - // of the identity.
|
| - if (valid_identity) {
|
| - EXPECT_EQ(rtc::SS_OPEN, client_ssl_->GetState());
|
| - EXPECT_EQ(rtc::SS_OPEN, server_ssl_->GetState());
|
| - } else {
|
| - EXPECT_EQ(rtc::SS_CLOSED, client_ssl_->GetState());
|
| - EXPECT_EQ(rtc::SS_CLOSED, server_ssl_->GetState());
|
| - }
|
| - }
|
| -
|
| - rtc::StreamResult DataWritten(SSLDummyStreamBase *from, const void *data,
|
| - size_t data_len, size_t *written,
|
| - int *error) {
|
| - // Randomly drop loss_ percent of packets
|
| - if (rtc::CreateRandomId() % 100 < static_cast<uint32_t>(loss_)) {
|
| - LOG(LS_VERBOSE) << "Randomly dropping packet, size=" << data_len;
|
| - *written = data_len;
|
| - return rtc::SR_SUCCESS;
|
| - }
|
| - if (dtls_ && (data_len > mtu_)) {
|
| - LOG(LS_VERBOSE) << "Dropping packet > mtu, size=" << data_len;
|
| - *written = data_len;
|
| - return rtc::SR_SUCCESS;
|
| - }
|
| -
|
| - // Optionally damage application data (type 23). Note that we don't damage
|
| - // handshake packets and we damage the last byte to keep the header
|
| - // intact but break the MAC.
|
| - if (damage_ && (*static_cast<const unsigned char *>(data) == 23)) {
|
| - std::vector<char> buf(data_len);
|
| -
|
| - LOG(LS_VERBOSE) << "Damaging packet";
|
| -
|
| - memcpy(&buf[0], data, data_len);
|
| - buf[data_len - 1]++;
|
| -
|
| - return from->WriteData(&buf[0], data_len, written, error);
|
| - }
|
| -
|
| - return from->WriteData(data, data_len, written, error);
|
| - }
|
| -
|
| - void SetDelay(int delay) {
|
| - delay_ = delay;
|
| - }
|
| - int GetDelay() { return delay_; }
|
| -
|
| - void SetLoseFirstPacket(bool lose) {
|
| - lose_first_packet_ = lose;
|
| - }
|
| - bool GetLoseFirstPacket() { return lose_first_packet_; }
|
| -
|
| - void SetLoss(int percent) {
|
| - loss_ = percent;
|
| - }
|
| -
|
| - void SetDamage() {
|
| - damage_ = true;
|
| - }
|
| -
|
| - void SetMtu(size_t mtu) {
|
| - mtu_ = mtu;
|
| - }
|
| -
|
| - void SetHandshakeWait(int wait) {
|
| - handshake_wait_ = wait;
|
| - }
|
| -
|
| - void SetDtlsSrtpCryptoSuites(const std::vector<int>& ciphers, bool client) {
|
| - if (client)
|
| - client_ssl_->SetDtlsSrtpCryptoSuites(ciphers);
|
| - else
|
| - server_ssl_->SetDtlsSrtpCryptoSuites(ciphers);
|
| - }
|
| -
|
| - bool GetDtlsSrtpCryptoSuite(bool client, int* retval) {
|
| - if (client)
|
| - return client_ssl_->GetDtlsSrtpCryptoSuite(retval);
|
| - else
|
| - return server_ssl_->GetDtlsSrtpCryptoSuite(retval);
|
| - }
|
| -
|
| - std::unique_ptr<rtc::SSLCertificate> GetPeerCertificate(bool client) {
|
| - if (client)
|
| - return client_ssl_->GetPeerCertificate();
|
| - else
|
| - return server_ssl_->GetPeerCertificate();
|
| - }
|
| -
|
| - bool GetSslCipherSuite(bool client, int* retval) {
|
| - if (client)
|
| - return client_ssl_->GetSslCipherSuite(retval);
|
| - else
|
| - return server_ssl_->GetSslCipherSuite(retval);
|
| - }
|
| -
|
| - int GetSslVersion(bool client) {
|
| - if (client)
|
| - return client_ssl_->GetSslVersion();
|
| - else
|
| - return server_ssl_->GetSslVersion();
|
| - }
|
| -
|
| - bool ExportKeyingMaterial(const char *label,
|
| - const unsigned char *context,
|
| - size_t context_len,
|
| - bool use_context,
|
| - bool client,
|
| - unsigned char *result,
|
| - size_t result_len) {
|
| - if (client)
|
| - return client_ssl_->ExportKeyingMaterial(label,
|
| - context, context_len,
|
| - use_context,
|
| - result, result_len);
|
| - else
|
| - return server_ssl_->ExportKeyingMaterial(label,
|
| - context, context_len,
|
| - use_context,
|
| - result, result_len);
|
| - }
|
| -
|
| - // To be implemented by subclasses.
|
| - virtual void WriteData() = 0;
|
| - virtual void ReadData(rtc::StreamInterface *stream) = 0;
|
| - virtual void TestTransfer(int size) = 0;
|
| -
|
| - protected:
|
| - std::string client_cert_pem_;
|
| - std::string client_private_key_pem_;
|
| - rtc::KeyParams client_key_type_;
|
| - rtc::KeyParams server_key_type_;
|
| - SSLDummyStreamBase *client_stream_; // freed by client_ssl_ destructor
|
| - SSLDummyStreamBase *server_stream_; // freed by server_ssl_ destructor
|
| - std::unique_ptr<rtc::SSLStreamAdapter> client_ssl_;
|
| - std::unique_ptr<rtc::SSLStreamAdapter> server_ssl_;
|
| - rtc::SSLIdentity *client_identity_; // freed by client_ssl_ destructor
|
| - rtc::SSLIdentity *server_identity_; // freed by server_ssl_ destructor
|
| - int delay_;
|
| - size_t mtu_;
|
| - int loss_;
|
| - bool lose_first_packet_;
|
| - bool damage_;
|
| - bool dtls_;
|
| - int handshake_wait_;
|
| - bool identities_set_;
|
| -};
|
| -
|
| -class SSLStreamAdapterTestTLS
|
| - : public SSLStreamAdapterTestBase,
|
| - public WithParamInterface<tuple<rtc::KeyParams, rtc::KeyParams>> {
|
| - public:
|
| - SSLStreamAdapterTestTLS()
|
| - : SSLStreamAdapterTestBase("",
|
| - "",
|
| - false,
|
| - ::testing::get<0>(GetParam()),
|
| - ::testing::get<1>(GetParam())),
|
| - client_buffer_(kFifoBufferSize),
|
| - server_buffer_(kFifoBufferSize) {
|
| - }
|
| -
|
| - void CreateStreams() override {
|
| - client_stream_ =
|
| - new SSLDummyStreamTLS(this, "c2s", &client_buffer_, &server_buffer_);
|
| - server_stream_ =
|
| - new SSLDummyStreamTLS(this, "s2c", &server_buffer_, &client_buffer_);
|
| - }
|
| -
|
| - // Test data transfer for TLS
|
| - void TestTransfer(int size) override {
|
| - LOG(LS_INFO) << "Starting transfer test with " << size << " bytes";
|
| - // Create some dummy data to send.
|
| - size_t received;
|
| -
|
| - send_stream_.ReserveSize(size);
|
| - for (int i = 0; i < size; ++i) {
|
| - char ch = static_cast<char>(i);
|
| - send_stream_.Write(&ch, 1, nullptr, nullptr);
|
| - }
|
| - send_stream_.Rewind();
|
| -
|
| - // Prepare the receive stream.
|
| - recv_stream_.ReserveSize(size);
|
| -
|
| - // Start sending
|
| - WriteData();
|
| -
|
| - // Wait for the client to close
|
| - EXPECT_TRUE_WAIT(server_ssl_->GetState() == rtc::SS_CLOSED, 10000);
|
| -
|
| - // Now check the data
|
| - recv_stream_.GetSize(&received);
|
| -
|
| - EXPECT_EQ(static_cast<size_t>(size), received);
|
| - EXPECT_EQ(0, memcmp(send_stream_.GetBuffer(),
|
| - recv_stream_.GetBuffer(), size));
|
| - }
|
| -
|
| - void WriteData() override {
|
| - size_t position, tosend, size;
|
| - rtc::StreamResult rv;
|
| - size_t sent;
|
| - char block[kBlockSize];
|
| -
|
| - send_stream_.GetSize(&size);
|
| - if (!size)
|
| - return;
|
| -
|
| - for (;;) {
|
| - send_stream_.GetPosition(&position);
|
| - if (send_stream_.Read(block, sizeof(block), &tosend, nullptr) !=
|
| - rtc::SR_EOS) {
|
| - rv = client_ssl_->Write(block, tosend, &sent, 0);
|
| -
|
| - if (rv == rtc::SR_SUCCESS) {
|
| - send_stream_.SetPosition(position + sent);
|
| - LOG(LS_VERBOSE) << "Sent: " << position + sent;
|
| - } else if (rv == rtc::SR_BLOCK) {
|
| - LOG(LS_VERBOSE) << "Blocked...";
|
| - send_stream_.SetPosition(position);
|
| - break;
|
| - } else {
|
| - ADD_FAILURE();
|
| - break;
|
| - }
|
| - } else {
|
| - // Now close
|
| - LOG(LS_INFO) << "Wrote " << position << " bytes. Closing";
|
| - client_ssl_->Close();
|
| - break;
|
| - }
|
| - }
|
| - };
|
| -
|
| - void ReadData(rtc::StreamInterface *stream) override {
|
| - char buffer[1600];
|
| - size_t bread;
|
| - int err2;
|
| - rtc::StreamResult r;
|
| -
|
| - for (;;) {
|
| - r = stream->Read(buffer, sizeof(buffer), &bread, &err2);
|
| -
|
| - if (r == rtc::SR_ERROR || r == rtc::SR_EOS) {
|
| - // Unfortunately, errors are the way that the stream adapter
|
| - // signals close in OpenSSL.
|
| - stream->Close();
|
| - return;
|
| - }
|
| -
|
| - if (r == rtc::SR_BLOCK)
|
| - break;
|
| -
|
| - ASSERT_EQ(rtc::SR_SUCCESS, r);
|
| - LOG(LS_VERBOSE) << "Read " << bread;
|
| -
|
| - recv_stream_.Write(buffer, bread, nullptr, nullptr);
|
| - }
|
| - }
|
| -
|
| - private:
|
| - rtc::FifoBuffer client_buffer_;
|
| - rtc::FifoBuffer server_buffer_;
|
| - rtc::MemoryStream send_stream_;
|
| - rtc::MemoryStream recv_stream_;
|
| -};
|
| -
|
| -class SSLStreamAdapterTestDTLS
|
| - : public SSLStreamAdapterTestBase,
|
| - public WithParamInterface<tuple<rtc::KeyParams, rtc::KeyParams>> {
|
| - public:
|
| - SSLStreamAdapterTestDTLS()
|
| - : SSLStreamAdapterTestBase("",
|
| - "",
|
| - true,
|
| - ::testing::get<0>(GetParam()),
|
| - ::testing::get<1>(GetParam())),
|
| - client_buffer_(kBufferCapacity, kDefaultBufferSize),
|
| - server_buffer_(kBufferCapacity, kDefaultBufferSize),
|
| - packet_size_(1000),
|
| - count_(0),
|
| - sent_(0) {}
|
| -
|
| - SSLStreamAdapterTestDTLS(const std::string& cert_pem,
|
| - const std::string& private_key_pem) :
|
| - SSLStreamAdapterTestBase(cert_pem, private_key_pem, true),
|
| - client_buffer_(kBufferCapacity, kDefaultBufferSize),
|
| - server_buffer_(kBufferCapacity, kDefaultBufferSize),
|
| - packet_size_(1000), count_(0), sent_(0) {
|
| - }
|
| -
|
| - void CreateStreams() override {
|
| - client_stream_ =
|
| - new SSLDummyStreamDTLS(this, "c2s", &client_buffer_, &server_buffer_);
|
| - server_stream_ =
|
| - new SSLDummyStreamDTLS(this, "s2c", &server_buffer_, &client_buffer_);
|
| - }
|
| -
|
| - void WriteData() override {
|
| - unsigned char *packet = new unsigned char[1600];
|
| -
|
| - while (sent_ < count_) {
|
| - unsigned int rand_state = sent_;
|
| - packet[0] = sent_;
|
| - for (size_t i = 1; i < packet_size_; i++) {
|
| - // This is a simple LC PRNG. Keep in synch with identical code below.
|
| - rand_state = (rand_state * 251 + 19937) >> 7;
|
| - packet[i] = rand_state & 0xff;
|
| - }
|
| -
|
| - size_t sent;
|
| - rtc::StreamResult rv = client_ssl_->Write(packet, packet_size_, &sent, 0);
|
| - if (rv == rtc::SR_SUCCESS) {
|
| - LOG(LS_VERBOSE) << "Sent: " << sent_;
|
| - sent_++;
|
| - } else if (rv == rtc::SR_BLOCK) {
|
| - LOG(LS_VERBOSE) << "Blocked...";
|
| - break;
|
| - } else {
|
| - ADD_FAILURE();
|
| - break;
|
| - }
|
| - }
|
| -
|
| - delete [] packet;
|
| - }
|
| -
|
| - void ReadData(rtc::StreamInterface *stream) override {
|
| - unsigned char buffer[2000];
|
| - size_t bread;
|
| - int err2;
|
| - rtc::StreamResult r;
|
| -
|
| - for (;;) {
|
| - r = stream->Read(buffer, 2000, &bread, &err2);
|
| -
|
| - if (r == rtc::SR_ERROR) {
|
| - // Unfortunately, errors are the way that the stream adapter
|
| - // signals close right now
|
| - stream->Close();
|
| - return;
|
| - }
|
| -
|
| - if (r == rtc::SR_BLOCK)
|
| - break;
|
| -
|
| - ASSERT_EQ(rtc::SR_SUCCESS, r);
|
| - LOG(LS_VERBOSE) << "Read " << bread;
|
| -
|
| - // Now parse the datagram
|
| - ASSERT_EQ(packet_size_, bread);
|
| - unsigned char packet_num = buffer[0];
|
| -
|
| - unsigned int rand_state = packet_num;
|
| - for (size_t i = 1; i < packet_size_; i++) {
|
| - // This is a simple LC PRNG. Keep in synch with identical code above.
|
| - rand_state = (rand_state * 251 + 19937) >> 7;
|
| - ASSERT_EQ(rand_state & 0xff, buffer[i]);
|
| - }
|
| - received_.insert(packet_num);
|
| - }
|
| - }
|
| -
|
| - void TestTransfer(int count) override {
|
| - count_ = count;
|
| -
|
| - WriteData();
|
| -
|
| - EXPECT_TRUE_WAIT(sent_ == count_, 10000);
|
| - LOG(LS_INFO) << "sent_ == " << sent_;
|
| -
|
| - if (damage_) {
|
| - WAIT(false, 2000);
|
| - EXPECT_EQ(0U, received_.size());
|
| - } else if (loss_ == 0) {
|
| - EXPECT_EQ_WAIT(static_cast<size_t>(sent_), received_.size(), 1000);
|
| - } else {
|
| - LOG(LS_INFO) << "Sent " << sent_ << " packets; received " <<
|
| - received_.size();
|
| - }
|
| - };
|
| -
|
| - private:
|
| - BufferQueueStream client_buffer_;
|
| - BufferQueueStream server_buffer_;
|
| - size_t packet_size_;
|
| - int count_;
|
| - int sent_;
|
| - std::set<int> received_;
|
| -};
|
| -
|
| -
|
| -rtc::StreamResult SSLDummyStreamBase::Write(const void* data, size_t data_len,
|
| - size_t* written, int* error) {
|
| - LOG(LS_VERBOSE) << "Writing to loopback " << data_len;
|
| -
|
| - if (first_packet_) {
|
| - first_packet_ = false;
|
| - if (test_base_->GetLoseFirstPacket()) {
|
| - LOG(LS_INFO) << "Losing initial packet of length " << data_len;
|
| - *written = data_len; // Fake successful writing also to writer.
|
| - return rtc::SR_SUCCESS;
|
| - }
|
| - }
|
| -
|
| - return test_base_->DataWritten(this, data, data_len, written, error);
|
| -};
|
| -
|
| -class SSLStreamAdapterTestDTLSFromPEMStrings : public SSLStreamAdapterTestDTLS {
|
| - public:
|
| - SSLStreamAdapterTestDTLSFromPEMStrings() :
|
| - SSLStreamAdapterTestDTLS(kCERT_PEM, kRSA_PRIVATE_KEY_PEM) {
|
| - }
|
| -};
|
| -
|
| -// Basic tests: TLS
|
| -
|
| -// Test that we can make a handshake work
|
| -TEST_P(SSLStreamAdapterTestTLS, TestTLSConnect) {
|
| - TestHandshake();
|
| -};
|
| -
|
| -// Test that closing the connection on one side updates the other side.
|
| -TEST_P(SSLStreamAdapterTestTLS, TestTLSClose) {
|
| - TestHandshake();
|
| - client_ssl_->Close();
|
| - EXPECT_EQ_WAIT(rtc::SS_CLOSED, server_ssl_->GetState(), handshake_wait_);
|
| -};
|
| -
|
| -// Test transfer -- trivial
|
| -TEST_P(SSLStreamAdapterTestTLS, TestTLSTransfer) {
|
| - TestHandshake();
|
| - TestTransfer(100000);
|
| -};
|
| -
|
| -// Test read-write after close.
|
| -TEST_P(SSLStreamAdapterTestTLS, ReadWriteAfterClose) {
|
| - TestHandshake();
|
| - TestTransfer(100000);
|
| - client_ssl_->Close();
|
| -
|
| - rtc::StreamResult rv;
|
| - char block[kBlockSize];
|
| - size_t dummy;
|
| -
|
| - // It's an error to write after closed.
|
| - rv = client_ssl_->Write(block, sizeof(block), &dummy, nullptr);
|
| - ASSERT_EQ(rtc::SR_ERROR, rv);
|
| -
|
| - // But after closed read gives you EOS.
|
| - rv = client_ssl_->Read(block, sizeof(block), &dummy, nullptr);
|
| - ASSERT_EQ(rtc::SR_EOS, rv);
|
| -};
|
| -
|
| -// Test a handshake with a bogus peer digest
|
| -TEST_P(SSLStreamAdapterTestTLS, TestTLSBogusDigest) {
|
| - SetPeerIdentitiesByDigest(false, true);
|
| - TestHandshake(false);
|
| -};
|
| -
|
| -TEST_P(SSLStreamAdapterTestTLS, TestTLSDelayedIdentity) {
|
| - TestHandshakeWithDelayedIdentity(true);
|
| -};
|
| -
|
| -TEST_P(SSLStreamAdapterTestTLS, TestTLSDelayedIdentityWithBogusDigest) {
|
| - TestHandshakeWithDelayedIdentity(false);
|
| -};
|
| -
|
| -// Test that the correct error is returned when SetPeerCertificateDigest is
|
| -// called with an unknown algorithm.
|
| -TEST_P(SSLStreamAdapterTestTLS,
|
| - TestSetPeerCertificateDigestWithUnknownAlgorithm) {
|
| - unsigned char server_digest[20];
|
| - size_t server_digest_len;
|
| - bool rv;
|
| - rtc::SSLPeerCertificateDigestError err;
|
| -
|
| - rv = server_identity_->certificate().ComputeDigest(
|
| - rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
|
| - ASSERT_TRUE(rv);
|
| -
|
| - rv = client_ssl_->SetPeerCertificateDigest("unknown algorithm", server_digest,
|
| - server_digest_len, &err);
|
| - EXPECT_EQ(rtc::SSLPeerCertificateDigestError::UNKNOWN_ALGORITHM, err);
|
| - EXPECT_FALSE(rv);
|
| -}
|
| -
|
| -// Test that the correct error is returned when SetPeerCertificateDigest is
|
| -// called with an invalid digest length.
|
| -TEST_P(SSLStreamAdapterTestTLS, TestSetPeerCertificateDigestWithInvalidLength) {
|
| - unsigned char server_digest[20];
|
| - size_t server_digest_len;
|
| - bool rv;
|
| - rtc::SSLPeerCertificateDigestError err;
|
| -
|
| - rv = server_identity_->certificate().ComputeDigest(
|
| - rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
|
| - ASSERT_TRUE(rv);
|
| -
|
| - rv = client_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, server_digest,
|
| - server_digest_len - 1, &err);
|
| - EXPECT_EQ(rtc::SSLPeerCertificateDigestError::INVALID_LENGTH, err);
|
| - EXPECT_FALSE(rv);
|
| -}
|
| -
|
| -// Test moving a bunch of data
|
| -
|
| -// Basic tests: DTLS
|
| -// Test that we can make a handshake work
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnect) {
|
| - TestHandshake();
|
| -};
|
| -
|
| -// Test that we can make a handshake work if the first packet in
|
| -// each direction is lost. This gives us predictable loss
|
| -// rather than having to tune random
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnectWithLostFirstPacket) {
|
| - SetLoseFirstPacket(true);
|
| - TestHandshake();
|
| -};
|
| -
|
| -// Test a handshake with loss and delay
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnectWithLostFirstPacketDelay2s) {
|
| - SetLoseFirstPacket(true);
|
| - SetDelay(2000);
|
| - SetHandshakeWait(20000);
|
| - TestHandshake();
|
| -};
|
| -
|
| -// Test a handshake with small MTU
|
| -// Disabled due to https://code.google.com/p/webrtc/issues/detail?id=3910
|
| -TEST_P(SSLStreamAdapterTestDTLS, DISABLED_TestDTLSConnectWithSmallMtu) {
|
| - SetMtu(700);
|
| - SetHandshakeWait(20000);
|
| - TestHandshake();
|
| -};
|
| -
|
| -// Test transfer -- trivial
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransfer) {
|
| - TestHandshake();
|
| - TestTransfer(100);
|
| -};
|
| -
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransferWithLoss) {
|
| - TestHandshake();
|
| - SetLoss(10);
|
| - TestTransfer(100);
|
| -};
|
| -
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransferWithDamage) {
|
| - SetDamage(); // Must be called first because first packet
|
| - // write happens at end of handshake.
|
| - TestHandshake();
|
| - TestTransfer(100);
|
| -};
|
| -
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSDelayedIdentity) {
|
| - TestHandshakeWithDelayedIdentity(true);
|
| -};
|
| -
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSDelayedIdentityWithBogusDigest) {
|
| - TestHandshakeWithDelayedIdentity(false);
|
| -};
|
| -
|
| -// Test DTLS-SRTP with all high ciphers
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpHigh) {
|
| - std::vector<int> high;
|
| - high.push_back(rtc::SRTP_AES128_CM_SHA1_80);
|
| - SetDtlsSrtpCryptoSuites(high, true);
|
| - SetDtlsSrtpCryptoSuites(high, false);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_80);
|
| -};
|
| -
|
| -// Test DTLS-SRTP with all low ciphers
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpLow) {
|
| - std::vector<int> low;
|
| - low.push_back(rtc::SRTP_AES128_CM_SHA1_32);
|
| - SetDtlsSrtpCryptoSuites(low, true);
|
| - SetDtlsSrtpCryptoSuites(low, false);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_32);
|
| -};
|
| -
|
| -// Test DTLS-SRTP with a mismatch -- should not converge
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpHighLow) {
|
| - std::vector<int> high;
|
| - high.push_back(rtc::SRTP_AES128_CM_SHA1_80);
|
| - std::vector<int> low;
|
| - low.push_back(rtc::SRTP_AES128_CM_SHA1_32);
|
| - SetDtlsSrtpCryptoSuites(high, true);
|
| - SetDtlsSrtpCryptoSuites(low, false);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_FALSE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_FALSE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
|
| -};
|
| -
|
| -// Test DTLS-SRTP with each side being mixed -- should select high
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpMixed) {
|
| - std::vector<int> mixed;
|
| - mixed.push_back(rtc::SRTP_AES128_CM_SHA1_80);
|
| - mixed.push_back(rtc::SRTP_AES128_CM_SHA1_32);
|
| - SetDtlsSrtpCryptoSuites(mixed, true);
|
| - SetDtlsSrtpCryptoSuites(mixed, false);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_80);
|
| -};
|
| -
|
| -// Test DTLS-SRTP with all GCM-128 ciphers.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCM128) {
|
| - std::vector<int> gcm128;
|
| - gcm128.push_back(rtc::SRTP_AEAD_AES_128_GCM);
|
| - SetDtlsSrtpCryptoSuites(gcm128, true);
|
| - SetDtlsSrtpCryptoSuites(gcm128, false);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_128_GCM);
|
| -};
|
| -
|
| -// Test DTLS-SRTP with all GCM-256 ciphers.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCM256) {
|
| - std::vector<int> gcm256;
|
| - gcm256.push_back(rtc::SRTP_AEAD_AES_256_GCM);
|
| - SetDtlsSrtpCryptoSuites(gcm256, true);
|
| - SetDtlsSrtpCryptoSuites(gcm256, false);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_256_GCM);
|
| -};
|
| -
|
| -// Test DTLS-SRTP with mixed GCM-128/-256 ciphers -- should not converge.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCMMismatch) {
|
| - std::vector<int> gcm128;
|
| - gcm128.push_back(rtc::SRTP_AEAD_AES_128_GCM);
|
| - std::vector<int> gcm256;
|
| - gcm256.push_back(rtc::SRTP_AEAD_AES_256_GCM);
|
| - SetDtlsSrtpCryptoSuites(gcm128, true);
|
| - SetDtlsSrtpCryptoSuites(gcm256, false);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_FALSE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_FALSE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
|
| -};
|
| -
|
| -// Test DTLS-SRTP with both GCM-128/-256 ciphers -- should select GCM-256.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCMMixed) {
|
| - std::vector<int> gcmBoth;
|
| - gcmBoth.push_back(rtc::SRTP_AEAD_AES_256_GCM);
|
| - gcmBoth.push_back(rtc::SRTP_AEAD_AES_128_GCM);
|
| - SetDtlsSrtpCryptoSuites(gcmBoth, true);
|
| - SetDtlsSrtpCryptoSuites(gcmBoth, false);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_256_GCM);
|
| -};
|
| -
|
| -// Test SRTP cipher suite lengths.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpKeyAndSaltLengths) {
|
| - int key_len;
|
| - int salt_len;
|
| -
|
| - ASSERT_FALSE(rtc::GetSrtpKeyAndSaltLengths(
|
| - rtc::SRTP_INVALID_CRYPTO_SUITE, &key_len, &salt_len));
|
| -
|
| - ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(
|
| - rtc::SRTP_AES128_CM_SHA1_32, &key_len, &salt_len));
|
| - ASSERT_EQ(128/8, key_len);
|
| - ASSERT_EQ(112/8, salt_len);
|
| -
|
| - ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(
|
| - rtc::SRTP_AES128_CM_SHA1_80, &key_len, &salt_len));
|
| - ASSERT_EQ(128/8, key_len);
|
| - ASSERT_EQ(112/8, salt_len);
|
| -
|
| - ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(
|
| - rtc::SRTP_AEAD_AES_128_GCM, &key_len, &salt_len));
|
| - ASSERT_EQ(128/8, key_len);
|
| - ASSERT_EQ(96/8, salt_len);
|
| -
|
| - ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(
|
| - rtc::SRTP_AEAD_AES_256_GCM, &key_len, &salt_len));
|
| - ASSERT_EQ(256/8, key_len);
|
| - ASSERT_EQ(96/8, salt_len);
|
| -};
|
| -
|
| -// Test an exporter
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestDTLSExporter) {
|
| - TestHandshake();
|
| - unsigned char client_out[20];
|
| - unsigned char server_out[20];
|
| -
|
| - bool result;
|
| - result = ExportKeyingMaterial(kExporterLabel,
|
| - kExporterContext, kExporterContextLen,
|
| - true, true,
|
| - client_out, sizeof(client_out));
|
| - ASSERT_TRUE(result);
|
| -
|
| - result = ExportKeyingMaterial(kExporterLabel,
|
| - kExporterContext, kExporterContextLen,
|
| - true, false,
|
| - server_out, sizeof(server_out));
|
| - ASSERT_TRUE(result);
|
| -
|
| - ASSERT_TRUE(!memcmp(client_out, server_out, sizeof(client_out)));
|
| -}
|
| -
|
| -// Test not yet valid certificates are not rejected.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestCertNotYetValid) {
|
| - long one_day = 60 * 60 * 24;
|
| - // Make the certificates not valid until one day later.
|
| - ResetIdentitiesWithValidity(one_day, one_day);
|
| - TestHandshake();
|
| -}
|
| -
|
| -// Test expired certificates are not rejected.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestCertExpired) {
|
| - long one_day = 60 * 60 * 24;
|
| - // Make the certificates already expired.
|
| - ResetIdentitiesWithValidity(-one_day, -one_day);
|
| - TestHandshake();
|
| -}
|
| -
|
| -// Test data transfer using certs created from strings.
|
| -TEST_F(SSLStreamAdapterTestDTLSFromPEMStrings, TestTransfer) {
|
| - TestHandshake();
|
| - TestTransfer(100);
|
| -}
|
| -
|
| -// Test getting the remote certificate.
|
| -TEST_F(SSLStreamAdapterTestDTLSFromPEMStrings, TestDTLSGetPeerCertificate) {
|
| - // Peer certificates haven't been received yet.
|
| - ASSERT_FALSE(GetPeerCertificate(true));
|
| - ASSERT_FALSE(GetPeerCertificate(false));
|
| -
|
| - TestHandshake();
|
| -
|
| - // The client should have a peer certificate after the handshake.
|
| - std::unique_ptr<rtc::SSLCertificate> client_peer_cert =
|
| - GetPeerCertificate(true);
|
| - ASSERT_TRUE(client_peer_cert);
|
| -
|
| - // It's not kCERT_PEM.
|
| - std::string client_peer_string = client_peer_cert->ToPEMString();
|
| - ASSERT_NE(kCERT_PEM, client_peer_string);
|
| -
|
| - // It must not have a chain, because the test certs are self-signed.
|
| - ASSERT_FALSE(client_peer_cert->GetChain());
|
| -
|
| - // The server should have a peer certificate after the handshake.
|
| - std::unique_ptr<rtc::SSLCertificate> server_peer_cert =
|
| - GetPeerCertificate(false);
|
| - ASSERT_TRUE(server_peer_cert);
|
| -
|
| - // It's kCERT_PEM
|
| - ASSERT_EQ(kCERT_PEM, server_peer_cert->ToPEMString());
|
| -
|
| - // It must not have a chain, because the test certs are self-signed.
|
| - ASSERT_FALSE(server_peer_cert->GetChain());
|
| -}
|
| -
|
| -// Test getting the used DTLS ciphers.
|
| -// DTLS 1.2 enabled for neither client nor server -> DTLS 1.0 will be used.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestGetSslCipherSuite) {
|
| - SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
|
| - ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_TRUE(rtc::SSLStreamAdapter::IsAcceptableCipher(
|
| - server_cipher, ::testing::get<1>(GetParam()).type()));
|
| -}
|
| -
|
| -// Test getting the used DTLS 1.2 ciphers.
|
| -// DTLS 1.2 enabled for client and server -> DTLS 1.2 will be used.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestGetSslCipherSuiteDtls12Both) {
|
| - SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_12);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(true));
|
| - ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(false));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_TRUE(rtc::SSLStreamAdapter::IsAcceptableCipher(
|
| - server_cipher, ::testing::get<1>(GetParam()).type()));
|
| -}
|
| -
|
| -// DTLS 1.2 enabled for client only -> DTLS 1.0 will be used.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestGetSslCipherSuiteDtls12Client) {
|
| - SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_12);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
|
| - ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_TRUE(rtc::SSLStreamAdapter::IsAcceptableCipher(
|
| - server_cipher, ::testing::get<1>(GetParam()).type()));
|
| -}
|
| -
|
| -// DTLS 1.2 enabled for server only -> DTLS 1.0 will be used.
|
| -TEST_P(SSLStreamAdapterTestDTLS, TestGetSslCipherSuiteDtls12Server) {
|
| - SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_10);
|
| - TestHandshake();
|
| -
|
| - int client_cipher;
|
| - ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
|
| - int server_cipher;
|
| - ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
|
| -
|
| - ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
|
| - ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
|
| -
|
| - ASSERT_EQ(client_cipher, server_cipher);
|
| - ASSERT_TRUE(rtc::SSLStreamAdapter::IsAcceptableCipher(
|
| - server_cipher, ::testing::get<1>(GetParam()).type()));
|
| -}
|
| -
|
| -// The RSA keysizes here might look strange, why not include the RFC's size
|
| -// 2048?. The reason is test case slowness; testing two sizes to exercise
|
| -// parametrization is sufficient.
|
| -INSTANTIATE_TEST_CASE_P(
|
| - SSLStreamAdapterTestsTLS,
|
| - SSLStreamAdapterTestTLS,
|
| - Combine(Values(rtc::KeyParams::RSA(1024, 65537),
|
| - rtc::KeyParams::RSA(1152, 65537),
|
| - rtc::KeyParams::ECDSA(rtc::EC_NIST_P256)),
|
| - Values(rtc::KeyParams::RSA(1024, 65537),
|
| - rtc::KeyParams::RSA(1152, 65537),
|
| - rtc::KeyParams::ECDSA(rtc::EC_NIST_P256))));
|
| -INSTANTIATE_TEST_CASE_P(
|
| - SSLStreamAdapterTestsDTLS,
|
| - SSLStreamAdapterTestDTLS,
|
| - Combine(Values(rtc::KeyParams::RSA(1024, 65537),
|
| - rtc::KeyParams::RSA(1152, 65537),
|
| - rtc::KeyParams::ECDSA(rtc::EC_NIST_P256)),
|
| - Values(rtc::KeyParams::RSA(1024, 65537),
|
| - rtc::KeyParams::RSA(1152, 65537),
|
| - rtc::KeyParams::ECDSA(rtc::EC_NIST_P256))));
|
|
|