blob: f6d20d1607ced98c582f44921825e9b0517d9cce [file] [log] [blame]
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001/*
2 * Copyright 2011 The WebRTC Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000011#include <algorithm>
jbauch555604a2016-04-26 03:13:22 -070012#include <memory>
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000013#include <set>
14#include <string>
15
Harald Alvestrand8515d5a2020-03-20 22:51:32 +010016#include "absl/memory/memory.h"
Steve Anton10542f22019-01-11 09:11:00 -080017#include "rtc_base/buffer_queue.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020018#include "rtc_base/checks.h"
19#include "rtc_base/gunit.h"
20#include "rtc_base/helpers.h"
Niels Möller13339482019-03-28 13:30:15 +010021#include "rtc_base/memory/fifo_buffer.h"
Niels Möllere7547d52018-11-01 09:33:08 +010022#include "rtc_base/memory_stream.h"
Steve Anton10542f22019-01-11 09:11:00 -080023#include "rtc_base/message_digest.h"
24#include "rtc_base/ssl_adapter.h"
25#include "rtc_base/ssl_identity.h"
26#include "rtc_base/ssl_stream_adapter.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020027#include "rtc_base/stream.h"
Harald Alvestrand13799132020-03-09 19:39:36 +010028#include "test/field_trial.h"
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000029
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +020030using ::testing::Combine;
31using ::testing::tuple;
Harald Alvestrand13799132020-03-09 19:39:36 +010032using ::testing::Values;
33using ::testing::WithParamInterface;
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +020034
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000035static const int kBlockSize = 4096;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000036static const char kExporterLabel[] = "label";
37static const unsigned char kExporterContext[] = "context";
38static int kExporterContextLen = sizeof(kExporterContext);
39
Harald Alvestrand13799132020-03-09 19:39:36 +010040// A private key used for testing, broken into pieces in order to avoid
41// issues with Git's checks for private keys in repos.
42#define RSA_PRIVATE_KEY_HEADER "-----BEGIN RSA PRIVATE KEY-----\n"
43
44static const char kRSA_PRIVATE_KEY_PEM[] = RSA_PRIVATE_KEY_HEADER
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000045 "MIICdwIBADANBgkqhkiG9w0BAQEFAASCAmEwggJdAgEAAoGBAMYRkbhmI7kVA/rM\n"
46 "czsZ+6JDhDvnkF+vn6yCAGuRPV03zuRqZtDy4N4to7PZu9PjqrRl7nDMXrG3YG9y\n"
47 "rlIAZ72KjcKKFAJxQyAKLCIdawKRyp8RdK3LEySWEZb0AV58IadqPZDTNHHRX8dz\n"
48 "5aTSMsbbkZ+C/OzTnbiMqLL/vg6jAgMBAAECgYAvgOs4FJcgvp+TuREx7YtiYVsH\n"
49 "mwQPTum2z/8VzWGwR8BBHBvIpVe1MbD/Y4seyI2aco/7UaisatSgJhsU46/9Y4fq\n"
50 "2TwXH9QANf4at4d9n/R6rzwpAJOpgwZgKvdQjkfrKTtgLV+/dawvpxUYkRH4JZM1\n"
51 "CVGukMfKNrSVH4Ap4QJBAOJmGV1ASPnB4r4nc99at7JuIJmd7fmuVUwUgYi4XgaR\n"
52 "WhScBsgYwZ/JoywdyZJgnbcrTDuVcWG56B3vXbhdpMsCQQDf9zeJrjnPZ3Cqm79y\n"
53 "kdqANep0uwZciiNiWxsQrCHztywOvbFhdp8iYVFG9EK8DMY41Y5TxUwsHD+67zao\n"
54 "ZNqJAkEA1suLUP/GvL8IwuRneQd2tWDqqRQ/Td3qq03hP7e77XtF/buya3Ghclo5\n"
55 "54czUR89QyVfJEC6278nzA7n2h1uVQJAcG6mztNL6ja/dKZjYZye2CY44QjSlLo0\n"
56 "MTgTSjdfg/28fFn2Jjtqf9Pi/X+50LWI/RcYMC2no606wRk9kyOuIQJBAK6VSAim\n"
57 "1pOEjsYQn0X5KEIrz1G3bfCbB848Ime3U2/FWlCHMr6ch8kCZ5d1WUeJD3LbwMNG\n"
58 "UCXiYxSsu20QNVw=\n"
59 "-----END RSA PRIVATE KEY-----\n";
60
Harald Alvestrand13799132020-03-09 19:39:36 +010061#undef RSA_PRIVATE_KEY_HEADER
62
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000063static const char kCERT_PEM[] =
64 "-----BEGIN CERTIFICATE-----\n"
65 "MIIBmTCCAQKgAwIBAgIEbzBSAjANBgkqhkiG9w0BAQsFADARMQ8wDQYDVQQDEwZX\n"
66 "ZWJSVEMwHhcNMTQwMTAyMTgyNDQ3WhcNMTQwMjAxMTgyNDQ3WjARMQ8wDQYDVQQD\n"
67 "EwZXZWJSVEMwgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAMYRkbhmI7kVA/rM\n"
68 "czsZ+6JDhDvnkF+vn6yCAGuRPV03zuRqZtDy4N4to7PZu9PjqrRl7nDMXrG3YG9y\n"
69 "rlIAZ72KjcKKFAJxQyAKLCIdawKRyp8RdK3LEySWEZb0AV58IadqPZDTNHHRX8dz\n"
70 "5aTSMsbbkZ+C/OzTnbiMqLL/vg6jAgMBAAEwDQYJKoZIhvcNAQELBQADgYEAUflI\n"
71 "VUe5Krqf5RVa5C3u/UTAOAUJBiDS3VANTCLBxjuMsvqOG0WvaYWP3HYPgrz0jXK2\n"
72 "LJE/mGw3MyFHEqi81jh95J+ypl6xKW6Rm8jKLR87gUvCaVYn/Z4/P3AqcQTB7wOv\n"
73 "UD0A8qfhfDM+LK6rPAnCsVN0NRDY3jvd6rzix9M=\n"
74 "-----END CERTIFICATE-----\n";
75
Jian Cui0a8798b2017-11-16 16:58:02 -080076static const char kIntCert1[] =
77 "-----BEGIN CERTIFICATE-----\n"
78 "MIIEUjCCAjqgAwIBAgIBAjANBgkqhkiG9w0BAQsFADCBljELMAkGA1UEBhMCVVMx\n"
79 "EzARBgNVBAgMCkNhbGlmb3JuaWExFjAUBgNVBAcMDU1vdW50YWluIFZpZXcxFDAS\n"
80 "BgNVBAoMC0dvb2dsZSwgSW5jMQwwCgYDVQQLDANHVFAxFzAVBgNVBAMMDnRlbGVw\n"
81 "aG9ueS5nb29nMR0wGwYJKoZIhvcNAQkBFg5ndHBAZ29vZ2xlLmNvbTAeFw0xNzA5\n"
82 "MjYwNDA5MDNaFw0yMDA2MjIwNDA5MDNaMGQxCzAJBgNVBAYTAlVTMQswCQYDVQQI\n"
83 "DAJDQTEWMBQGA1UEBwwNTW91bnRhaW4gVmlldzEXMBUGA1UECgwOdGVsZXBob255\n"
84 "Lmdvb2cxFzAVBgNVBAMMDnRlbGVwaG9ueS5nb29nMIGfMA0GCSqGSIb3DQEBAQUA\n"
85 "A4GNADCBiQKBgQDJXWeeU1v1+wlqkVobzI3aN7Uh2iVQA9YCdq5suuabtiD/qoOD\n"
86 "NKpmQqsx7WZGGWSZTDFEBaUpvIK7Hb+nzRqk6iioPCFOFuarm6GxO1xVneImMuE6\n"
87 "tuWb3YZPr+ikChJbl11y5UcSbg0QsbeUc+jHl5umNvrL85Y+z8SP0rxbBwIDAQAB\n"
88 "o2AwXjAdBgNVHQ4EFgQU7tdZobqlN8R8V72FQnRxmqq8tKswHwYDVR0jBBgwFoAU\n"
89 "5GgKMUtcxkQ2dJrtNR5YOlIAPDswDwYDVR0TAQH/BAUwAwEB/zALBgNVHQ8EBAMC\n"
90 "AQYwDQYJKoZIhvcNAQELBQADggIBADObh9Z+z14FmP9zSenhFtq7hFnmNrSkklk8\n"
91 "eyYWXKfOuIriEQQBZsz76ZcnzStih8Rj+yQ0AXydk4fJ5LOwC2cUqQBar17g6Pd2\n"
92 "8g4SIL4azR9WvtiSvpuGlwp25b+yunaacDne6ebnf/MUiiKT5w61Xo3cEPVfl38e\n"
93 "/Up2l0bioid5enUTmg6LY6RxDO6tnZQkz3XD+nNSwT4ehtkqFpHYWjErj0BbkDM2\n"
94 "hiVc/JsYOZn3DmuOlHVHU6sKwqh3JEyvHO/d7DGzMGWHpHwv2mCTJq6l/sR95Tc2\n"
95 "GaQZgGDVNs9pdEouJCDm9e/PbQWRYhnat82PTkXx/6mDAAwdZlIi/pACzq8K4p7e\n"
96 "6hF0t8uKGnXJubHPXxlnJU6yxZ0yWmivAGjwWK4ur832gKlho4jeMDhiI/T3QPpl\n"
97 "iMNsIvxRhdD+GxJkQP1ezayw8s+Uc9KwKglrkBSRRDLCJUfPOvMmXLUDSTMX7kp4\n"
98 "/Ak1CA8dVLJIlfEjLBUuvAttlP7+7lsKNgxAjCxZkWLXIyGULzNPQwVWkGfCbrQs\n"
99 "XyMvSbFsSIb7blV7eLlmf9a+2RprUUkc2ALXLLCI9YQXmxm2beBfMyNmmebwBJzT\n"
100 "B0OR+5pFFNTJPoNlqpdrDsGrDu7JlUtk0ZLZzYyKXbgy2qXxfd4OWzXXjxpLMszZ\n"
101 "LDIpOAkj\n"
102 "-----END CERTIFICATE-----\n";
103
104static const char kCACert[] =
105 "-----BEGIN CERTIFICATE-----\n"
106 "MIIGETCCA/mgAwIBAgIJAKN9r/BdbGUJMA0GCSqGSIb3DQEBCwUAMIGWMQswCQYD\n"
107 "VQQGEwJVUzETMBEGA1UECAwKQ2FsaWZvcm5pYTEWMBQGA1UEBwwNTW91bnRhaW4g\n"
108 "VmlldzEUMBIGA1UECgwLR29vZ2xlLCBJbmMxDDAKBgNVBAsMA0dUUDEXMBUGA1UE\n"
109 "AwwOdGVsZXBob255Lmdvb2cxHTAbBgkqhkiG9w0BCQEWDmd0cEBnb29nbGUuY29t\n"
110 "MB4XDTE3MDcyNzIzMDE0NVoXDTE3MDgyNjIzMDE0NVowgZYxCzAJBgNVBAYTAlVT\n"
111 "MRMwEQYDVQQIDApDYWxpZm9ybmlhMRYwFAYDVQQHDA1Nb3VudGFpbiBWaWV3MRQw\n"
112 "EgYDVQQKDAtHb29nbGUsIEluYzEMMAoGA1UECwwDR1RQMRcwFQYDVQQDDA50ZWxl\n"
113 "cGhvbnkuZ29vZzEdMBsGCSqGSIb3DQEJARYOZ3RwQGdvb2dsZS5jb20wggIiMA0G\n"
114 "CSqGSIb3DQEBAQUAA4ICDwAwggIKAoICAQCfvpF7aBV5Hp1EHsWoIlL3GeHwh8dS\n"
115 "lv9VQCegN9rD06Ny7MgcED5AiK2vqXmUmOVS+7NbATkdVYN/eozDhKtN3Q3n87kJ\n"
116 "Nt/TD/TcZZHOZIGsRPbrf2URK26E/5KzTzbzXVBOA1e+gSj+EBbltGqb01ZO5ErF\n"
117 "iPGViPM/HpYKdq6mfz2bS5PhU67XZMM2zvToyReQ/Fjm/6PJhwKSRXSgZF5djPhk\n"
118 "2LfOKMLS0AeZtd2C4DFsCU41lfLUkybioDgFuzTQ3TFi1K8A07KYTMmLY/yQppnf\n"
119 "SpNX58shlVhM+Ed37K1Z0rU0OfVCZ5P+KKaSSfMranjlU7zeUIhZYjqq/EYrEhbS\n"
120 "dLnNHwgJrqxzId3kq8uuLM6+VB7JZKnZLfT90GdAbX4+tutNe21smmogF9f80vEy\n"
121 "gM4tOp9rXrvz9vCwWHXVY9kdKemdLAsREoO6MS9k2ctK4jj80o2dROuFC6Q3e7mz\n"
122 "RjvZr5Tvi464c2o9o/jNlJ0O6q7V2eQzohD+7VnV5QPpRGXxlIeqpR2zoAg+WtRS\n"
123 "4OgHOVYiD3M6uAlggJA5pcDjMfkEZ+pkhtVcT4qMCEoruk6GbyPxS565oSHu16bH\n"
124 "EjeCqbZOVND5T3oA7nz6aQSs8sJabt0jmxUkGVnE+4ZDIuuRtkRma+0P/96Mtqor\n"
125 "OlpNWY1OBDY64QIDAQABo2AwXjAdBgNVHQ4EFgQU5GgKMUtcxkQ2dJrtNR5YOlIA\n"
126 "PDswHwYDVR0jBBgwFoAU5GgKMUtcxkQ2dJrtNR5YOlIAPDswDwYDVR0TAQH/BAUw\n"
127 "AwEB/zALBgNVHQ8EBAMCAQYwDQYJKoZIhvcNAQELBQADggIBAARQly5/bB6VUL2C\n"
128 "ykDYgWt48go407pAra6tL2kjpdfxV5PdL7iMZRkeht00vj+BVahIqZKrNOa/f5Fx\n"
129 "vlpahZFu0PDN436aQwRZ9qWut2qDOK0/z9Hhj6NWybquRFwMwqkPG/ivLMDU8Dmj\n"
130 "CIplpngPYNwXCs0KzdjSXYxqxJbwMjQXELD+/RcurY0oTtJMM1/2vKQMzw24UJqe\n"
131 "XLJAlsnd2AnWzWNUEviDZY89j9NdkHerBmV2gGzcU+X5lgOO5M8odBv0ZC9D+a6Z\n"
132 "QPZAOfdGVw60hhGvTW5s/s0dHwCpegRidhs0MD0fTmwwjYFBSmUx3Gztr4JTzOOr\n"
133 "7e5daJuak2ujQ5DqcGBvt1gePjSudb5brS7JQtN8tI/FyrnR4q/OuOwv1EvlC5RG\n"
134 "hLX+TXaWqFxB1Hd8ebKRR40mboFG6KcUI3lLBthDvQE7jnq48QfZMjlMQK0ZF1l7\n"
135 "SrlwRXWA74bU8CLJvnZKKo9p4TsTiDYGSYC6tNHKj5s3TGWL46oqGyZ0KdGNhrtC\n"
136 "rIGenMhth1vPYjyy0XuGBndXT85yi+IM2l8g8oU845+plxIhgpSI8bbC0oLwnhQ5\n"
137 "ARfsiYLkXDE7imSS0CSUmye76372mlzAIB1is4bBB/SzpPQtBuB9LDKtONgpSGHn\n"
138 "dGaXBy+qbVXVyGXaeEbIRjtJ6m92\n"
139 "-----END CERTIFICATE-----\n";
140
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000141class SSLStreamAdapterTestBase;
142
jbauche488a0d2015-11-19 05:17:58 -0800143class SSLDummyStreamBase : public rtc::StreamInterface,
144 public sigslot::has_slots<> {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000145 public:
jbauche488a0d2015-11-19 05:17:58 -0800146 SSLDummyStreamBase(SSLStreamAdapterTestBase* test,
Yves Gerey665174f2018-06-19 15:03:05 +0200147 const std::string& side,
jbauche488a0d2015-11-19 05:17:58 -0800148 rtc::StreamInterface* in,
Yves Gerey665174f2018-06-19 15:03:05 +0200149 rtc::StreamInterface* out)
150 : test_base_(test), side_(side), in_(in), out_(out), first_packet_(true) {
jbauche488a0d2015-11-19 05:17:58 -0800151 in_->SignalEvent.connect(this, &SSLDummyStreamBase::OnEventIn);
152 out_->SignalEvent.connect(this, &SSLDummyStreamBase::OnEventOut);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000153 }
154
jbauche488a0d2015-11-19 05:17:58 -0800155 rtc::StreamState GetState() const override { return rtc::SS_OPEN; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000156
Yves Gerey665174f2018-06-19 15:03:05 +0200157 rtc::StreamResult Read(void* buffer,
158 size_t buffer_len,
159 size_t* read,
160 int* error) override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000161 rtc::StreamResult r;
162
163 r = in_->Read(buffer, buffer_len, read, error);
164 if (r == rtc::SR_BLOCK)
165 return rtc::SR_BLOCK;
166 if (r == rtc::SR_EOS)
167 return rtc::SR_EOS;
168
169 if (r != rtc::SR_SUCCESS) {
170 ADD_FAILURE();
171 return rtc::SR_ERROR;
172 }
173
174 return rtc::SR_SUCCESS;
175 }
176
177 // Catch readability events on in and pass them up.
jbauche488a0d2015-11-19 05:17:58 -0800178 void OnEventIn(rtc::StreamInterface* stream, int sig, int err) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000179 int mask = (rtc::SE_READ | rtc::SE_CLOSE);
180
181 if (sig & mask) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100182 RTC_LOG(LS_VERBOSE) << "SSLDummyStreamBase::OnEvent side=" << side_
183 << " sig=" << sig << " forwarding upward";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000184 PostEvent(sig & mask, 0);
185 }
186 }
187
188 // Catch writeability events on out and pass them up.
jbauche488a0d2015-11-19 05:17:58 -0800189 void OnEventOut(rtc::StreamInterface* stream, int sig, int err) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000190 if (sig & rtc::SE_WRITE) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100191 RTC_LOG(LS_VERBOSE) << "SSLDummyStreamBase::OnEvent side=" << side_
192 << " sig=" << sig << " forwarding upward";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000193
194 PostEvent(sig & rtc::SE_WRITE, 0);
195 }
196 }
197
198 // Write to the outgoing FifoBuffer
Yves Gerey665174f2018-06-19 15:03:05 +0200199 rtc::StreamResult WriteData(const void* data,
200 size_t data_len,
201 size_t* written,
202 int* error) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000203 return out_->Write(data, data_len, written, error);
204 }
205
Yves Gerey665174f2018-06-19 15:03:05 +0200206 rtc::StreamResult Write(const void* data,
207 size_t data_len,
208 size_t* written,
209 int* error) override;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000210
jbauche488a0d2015-11-19 05:17:58 -0800211 void Close() override {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100212 RTC_LOG(LS_INFO) << "Closing outbound stream";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000213 out_->Close();
214 }
215
jbauche488a0d2015-11-19 05:17:58 -0800216 protected:
217 SSLStreamAdapterTestBase* test_base_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000218 const std::string side_;
jbauche488a0d2015-11-19 05:17:58 -0800219 rtc::StreamInterface* in_;
220 rtc::StreamInterface* out_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000221 bool first_packet_;
222};
223
jbauche488a0d2015-11-19 05:17:58 -0800224class SSLDummyStreamTLS : public SSLDummyStreamBase {
225 public:
226 SSLDummyStreamTLS(SSLStreamAdapterTestBase* test,
227 const std::string& side,
228 rtc::FifoBuffer* in,
Yves Gerey665174f2018-06-19 15:03:05 +0200229 rtc::FifoBuffer* out)
230 : SSLDummyStreamBase(test, side, in, out) {}
jbauche488a0d2015-11-19 05:17:58 -0800231};
232
Yves Gerey665174f2018-06-19 15:03:05 +0200233class BufferQueueStream : public rtc::BufferQueue, public rtc::StreamInterface {
jbauche488a0d2015-11-19 05:17:58 -0800234 public:
235 BufferQueueStream(size_t capacity, size_t default_size)
Yves Gerey665174f2018-06-19 15:03:05 +0200236 : rtc::BufferQueue(capacity, default_size) {}
jbauche488a0d2015-11-19 05:17:58 -0800237
238 // Implementation of abstract StreamInterface methods.
239
240 // A buffer queue stream is always "open".
241 rtc::StreamState GetState() const override { return rtc::SS_OPEN; }
242
243 // Reading a buffer queue stream will either succeed or block.
Yves Gerey665174f2018-06-19 15:03:05 +0200244 rtc::StreamResult Read(void* buffer,
245 size_t buffer_len,
246 size_t* read,
247 int* error) override {
jbauche488a0d2015-11-19 05:17:58 -0800248 if (!ReadFront(buffer, buffer_len, read)) {
249 return rtc::SR_BLOCK;
250 }
251 return rtc::SR_SUCCESS;
252 }
253
254 // Writing to a buffer queue stream will either succeed or block.
Yves Gerey665174f2018-06-19 15:03:05 +0200255 rtc::StreamResult Write(const void* data,
256 size_t data_len,
257 size_t* written,
258 int* error) override {
jbauche488a0d2015-11-19 05:17:58 -0800259 if (!WriteBack(data, data_len, written)) {
260 return rtc::SR_BLOCK;
261 }
262 return rtc::SR_SUCCESS;
263 }
264
265 // A buffer queue stream can not be closed.
266 void Close() override {}
267
268 protected:
Yves Gerey665174f2018-06-19 15:03:05 +0200269 void NotifyReadableForTest() override { PostEvent(rtc::SE_READ, 0); }
jbauche488a0d2015-11-19 05:17:58 -0800270
Yves Gerey665174f2018-06-19 15:03:05 +0200271 void NotifyWritableForTest() override { PostEvent(rtc::SE_WRITE, 0); }
jbauche488a0d2015-11-19 05:17:58 -0800272};
273
274class SSLDummyStreamDTLS : public SSLDummyStreamBase {
275 public:
276 SSLDummyStreamDTLS(SSLStreamAdapterTestBase* test,
277 const std::string& side,
278 BufferQueueStream* in,
Yves Gerey665174f2018-06-19 15:03:05 +0200279 BufferQueueStream* out)
280 : SSLDummyStreamBase(test, side, in, out) {}
jbauche488a0d2015-11-19 05:17:58 -0800281};
282
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000283static const int kFifoBufferSize = 4096;
jbauche488a0d2015-11-19 05:17:58 -0800284static const int kBufferCapacity = 1;
285static const size_t kDefaultBufferSize = 2048;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000286
Mirko Bonadei6a489f22019-04-09 15:11:12 +0200287class SSLStreamAdapterTestBase : public ::testing::Test,
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000288 public sigslot::has_slots<> {
289 public:
torbjorng4e572472015-10-08 09:42:49 -0700290 SSLStreamAdapterTestBase(
291 const std::string& client_cert_pem,
292 const std::string& client_private_key_pem,
293 bool dtls,
294 rtc::KeyParams client_key_type = rtc::KeyParams(rtc::KT_DEFAULT),
295 rtc::KeyParams server_key_type = rtc::KeyParams(rtc::KT_DEFAULT))
jbauche488a0d2015-11-19 05:17:58 -0800296 : client_cert_pem_(client_cert_pem),
297 client_private_key_pem_(client_private_key_pem),
298 client_key_type_(client_key_type),
299 server_key_type_(server_key_type),
deadbeef37f5ecf2017-02-27 14:06:41 -0800300 client_stream_(nullptr),
301 server_stream_(nullptr),
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200302 delay_(0),
303 mtu_(1460),
304 loss_(0),
305 lose_first_packet_(false),
306 damage_(false),
307 dtls_(dtls),
308 handshake_wait_(5000),
Benjamin Wrightaf1f8652019-04-01 18:25:23 +0000309 identities_set_(false) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000310 // Set use of the test RNG to get predictable loss patterns.
311 rtc::SetRandomTestMode(true);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000312 }
313
Steve Anton9de3aac2017-10-24 10:08:26 -0700314 ~SSLStreamAdapterTestBase() override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000315 // Put it back for the next test.
316 rtc::SetRandomTestMode(false);
317 }
318
torbjorng7593aad2015-11-19 12:20:51 -0800319 void SetUp() override {
jbauche488a0d2015-11-19 05:17:58 -0800320 CreateStreams();
321
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100322 client_ssl_ =
323 rtc::SSLStreamAdapter::Create(absl::WrapUnique(client_stream_));
324 server_ssl_ =
325 rtc::SSLStreamAdapter::Create(absl::WrapUnique(server_stream_));
jbauche488a0d2015-11-19 05:17:58 -0800326
327 // Set up the slots
328 client_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
329 server_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
330
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100331 std::unique_ptr<rtc::SSLIdentity> client_identity;
jbauche488a0d2015-11-19 05:17:58 -0800332 if (!client_cert_pem_.empty() && !client_private_key_pem_.empty()) {
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100333 client_identity = rtc::SSLIdentity::CreateFromPEMStrings(
jbauche488a0d2015-11-19 05:17:58 -0800334 client_private_key_pem_, client_cert_pem_);
335 } else {
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100336 client_identity = rtc::SSLIdentity::Create("client", client_key_type_);
jbauche488a0d2015-11-19 05:17:58 -0800337 }
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100338 auto server_identity = rtc::SSLIdentity::Create("server", server_key_type_);
jbauche488a0d2015-11-19 05:17:58 -0800339
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100340 client_ssl_->SetIdentity(std::move(client_identity));
341 server_ssl_->SetIdentity(std::move(server_identity));
jbauche488a0d2015-11-19 05:17:58 -0800342 }
343
torbjorng7593aad2015-11-19 12:20:51 -0800344 void TearDown() override {
jbauche488a0d2015-11-19 05:17:58 -0800345 client_ssl_.reset(nullptr);
346 server_ssl_.reset(nullptr);
347 }
348
349 virtual void CreateStreams() = 0;
350
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000351 // Recreate the client/server identities with the specified validity period.
352 // |not_before| and |not_after| are offsets from the current time in number
353 // of seconds.
354 void ResetIdentitiesWithValidity(int not_before, int not_after) {
jbauche488a0d2015-11-19 05:17:58 -0800355 CreateStreams();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000356
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100357 client_ssl_ =
358 rtc::SSLStreamAdapter::Create(absl::WrapUnique(client_stream_));
359 server_ssl_ =
360 rtc::SSLStreamAdapter::Create(absl::WrapUnique(server_stream_));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000361
362 client_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
363 server_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
364
Torbjorn Granlund46c9cc02015-12-01 13:06:34 +0100365 time_t now = time(nullptr);
366
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000367 rtc::SSLIdentityParams client_params;
torbjorng4e572472015-10-08 09:42:49 -0700368 client_params.key_params = rtc::KeyParams(rtc::KT_DEFAULT);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000369 client_params.common_name = "client";
Torbjorn Granlund46c9cc02015-12-01 13:06:34 +0100370 client_params.not_before = now + not_before;
371 client_params.not_after = now + not_after;
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100372 auto client_identity = rtc::SSLIdentity::CreateForTest(client_params);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000373
374 rtc::SSLIdentityParams server_params;
torbjorng4e572472015-10-08 09:42:49 -0700375 server_params.key_params = rtc::KeyParams(rtc::KT_DEFAULT);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000376 server_params.common_name = "server";
Torbjorn Granlund46c9cc02015-12-01 13:06:34 +0100377 server_params.not_before = now + not_before;
378 server_params.not_after = now + not_after;
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100379 auto server_identity = rtc::SSLIdentity::CreateForTest(server_params);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000380
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100381 client_ssl_->SetIdentity(std::move(client_identity));
382 server_ssl_->SetIdentity(std::move(server_identity));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000383 }
384
Yves Gerey665174f2018-06-19 15:03:05 +0200385 virtual void OnEvent(rtc::StreamInterface* stream, int sig, int err) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100386 RTC_LOG(LS_VERBOSE) << "SSLStreamAdapterTestBase::OnEvent sig=" << sig;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000387
388 if (sig & rtc::SE_READ) {
389 ReadData(stream);
390 }
391
392 if ((stream == client_ssl_.get()) && (sig & rtc::SE_WRITE)) {
393 WriteData();
394 }
395 }
396
deadbeef89824f62016-09-30 11:55:43 -0700397 void SetPeerIdentitiesByDigest(bool correct, bool expect_success) {
398 unsigned char server_digest[20];
399 size_t server_digest_len;
400 unsigned char client_digest[20];
401 size_t client_digest_len;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000402 bool rv;
deadbeef89824f62016-09-30 11:55:43 -0700403 rtc::SSLPeerCertificateDigestError err;
404 rtc::SSLPeerCertificateDigestError expected_err =
405 expect_success
406 ? rtc::SSLPeerCertificateDigestError::NONE
407 : rtc::SSLPeerCertificateDigestError::VERIFICATION_FAILED;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000408
Mirko Bonadei675513b2017-11-09 11:09:25 +0100409 RTC_LOG(LS_INFO) << "Setting peer identities by digest";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000410
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100411 rv = server_identity()->certificate().ComputeDigest(
deadbeef89824f62016-09-30 11:55:43 -0700412 rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000413 ASSERT_TRUE(rv);
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100414 rv = client_identity()->certificate().ComputeDigest(
deadbeef89824f62016-09-30 11:55:43 -0700415 rtc::DIGEST_SHA_1, client_digest, 20, &client_digest_len);
416 ASSERT_TRUE(rv);
417
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000418 if (!correct) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100419 RTC_LOG(LS_INFO) << "Setting bogus digest for server cert";
deadbeef89824f62016-09-30 11:55:43 -0700420 server_digest[0]++;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000421 }
deadbeef89824f62016-09-30 11:55:43 -0700422 rv = client_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, server_digest,
423 server_digest_len, &err);
424 EXPECT_EQ(expected_err, err);
425 EXPECT_EQ(expect_success, rv);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000426
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000427 if (!correct) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100428 RTC_LOG(LS_INFO) << "Setting bogus digest for client cert";
deadbeef89824f62016-09-30 11:55:43 -0700429 client_digest[0]++;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000430 }
deadbeef89824f62016-09-30 11:55:43 -0700431 rv = server_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, client_digest,
432 client_digest_len, &err);
433 EXPECT_EQ(expected_err, err);
434 EXPECT_EQ(expect_success, rv);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000435
436 identities_set_ = true;
437 }
438
Joachim Bauch831c5582015-05-20 12:48:41 +0200439 void SetupProtocolVersions(rtc::SSLProtocolVersion server_version,
440 rtc::SSLProtocolVersion client_version) {
441 server_ssl_->SetMaxProtocolVersion(server_version);
442 client_ssl_->SetMaxProtocolVersion(client_version);
443 }
444
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000445 void TestHandshake(bool expect_success = true) {
Yves Gerey665174f2018-06-19 15:03:05 +0200446 server_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
447 client_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000448
449 if (!dtls_) {
450 // Make sure we simulate a reliable network for TLS.
451 // This is just a check to make sure that people don't write wrong
452 // tests.
nissec16fa5e2017-02-07 07:18:43 -0800453 RTC_CHECK_EQ(1460, mtu_);
454 RTC_CHECK(!loss_);
455 RTC_CHECK(!lose_first_packet_);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000456 }
457
458 if (!identities_set_)
deadbeef89824f62016-09-30 11:55:43 -0700459 SetPeerIdentitiesByDigest(true, true);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000460
461 // Start the handshake
462 int rv;
463
464 server_ssl_->SetServerRole();
Taylor Brandstetterc8762a82016-08-11 12:01:49 -0700465 rv = server_ssl_->StartSSL();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000466 ASSERT_EQ(0, rv);
467
Taylor Brandstetterc8762a82016-08-11 12:01:49 -0700468 rv = client_ssl_->StartSSL();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000469 ASSERT_EQ(0, rv);
470
471 // Now run the handshake
472 if (expect_success) {
Yves Gerey665174f2018-06-19 15:03:05 +0200473 EXPECT_TRUE_WAIT((client_ssl_->GetState() == rtc::SS_OPEN) &&
474 (server_ssl_->GetState() == rtc::SS_OPEN),
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000475 handshake_wait_);
476 } else {
477 EXPECT_TRUE_WAIT(client_ssl_->GetState() == rtc::SS_CLOSED,
478 handshake_wait_);
479 }
480 }
481
deadbeef89824f62016-09-30 11:55:43 -0700482 // This tests that the handshake can complete before the identity is
483 // verified, and the identity will be verified after the fact.
484 void TestHandshakeWithDelayedIdentity(bool valid_identity) {
485 server_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
486 client_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
487
488 if (!dtls_) {
489 // Make sure we simulate a reliable network for TLS.
490 // This is just a check to make sure that people don't write wrong
491 // tests.
nissec16fa5e2017-02-07 07:18:43 -0800492 RTC_CHECK_EQ(1460, mtu_);
493 RTC_CHECK(!loss_);
494 RTC_CHECK(!lose_first_packet_);
deadbeef89824f62016-09-30 11:55:43 -0700495 }
496
497 // Start the handshake
498 int rv;
499
500 server_ssl_->SetServerRole();
501 rv = server_ssl_->StartSSL();
502 ASSERT_EQ(0, rv);
503
504 rv = client_ssl_->StartSSL();
505 ASSERT_EQ(0, rv);
506
507 // Now run the handshake.
508 EXPECT_TRUE_WAIT(
509 client_ssl_->IsTlsConnected() && server_ssl_->IsTlsConnected(),
510 handshake_wait_);
511
512 // Until the identity has been verified, the state should still be
513 // SS_OPENING and writes should return SR_BLOCK.
514 EXPECT_EQ(rtc::SS_OPENING, client_ssl_->GetState());
515 EXPECT_EQ(rtc::SS_OPENING, server_ssl_->GetState());
516 unsigned char packet[1];
517 size_t sent;
518 EXPECT_EQ(rtc::SR_BLOCK, client_ssl_->Write(&packet, 1, &sent, 0));
519 EXPECT_EQ(rtc::SR_BLOCK, server_ssl_->Write(&packet, 1, &sent, 0));
520
521 // If we set an invalid identity at this point, SetPeerCertificateDigest
522 // should return false.
523 SetPeerIdentitiesByDigest(valid_identity, valid_identity);
524 // State should then transition to SS_OPEN or SS_CLOSED based on validation
525 // of the identity.
526 if (valid_identity) {
527 EXPECT_EQ(rtc::SS_OPEN, client_ssl_->GetState());
528 EXPECT_EQ(rtc::SS_OPEN, server_ssl_->GetState());
529 } else {
530 EXPECT_EQ(rtc::SS_CLOSED, client_ssl_->GetState());
531 EXPECT_EQ(rtc::SS_CLOSED, server_ssl_->GetState());
532 }
533 }
534
Yves Gerey665174f2018-06-19 15:03:05 +0200535 rtc::StreamResult DataWritten(SSLDummyStreamBase* from,
536 const void* data,
537 size_t data_len,
538 size_t* written,
539 int* error) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000540 // Randomly drop loss_ percent of packets
Peter Boström0c4e06b2015-10-07 12:23:21 +0200541 if (rtc::CreateRandomId() % 100 < static_cast<uint32_t>(loss_)) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100542 RTC_LOG(LS_VERBOSE) << "Randomly dropping packet, size=" << data_len;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000543 *written = data_len;
544 return rtc::SR_SUCCESS;
545 }
546 if (dtls_ && (data_len > mtu_)) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100547 RTC_LOG(LS_VERBOSE) << "Dropping packet > mtu, size=" << data_len;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000548 *written = data_len;
549 return rtc::SR_SUCCESS;
550 }
551
552 // Optionally damage application data (type 23). Note that we don't damage
553 // handshake packets and we damage the last byte to keep the header
554 // intact but break the MAC.
Yves Gerey665174f2018-06-19 15:03:05 +0200555 if (damage_ && (*static_cast<const unsigned char*>(data) == 23)) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000556 std::vector<char> buf(data_len);
557
Mirko Bonadei675513b2017-11-09 11:09:25 +0100558 RTC_LOG(LS_VERBOSE) << "Damaging packet";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000559
560 memcpy(&buf[0], data, data_len);
561 buf[data_len - 1]++;
562
563 return from->WriteData(&buf[0], data_len, written, error);
564 }
565
566 return from->WriteData(data, data_len, written, error);
567 }
568
Yves Gerey665174f2018-06-19 15:03:05 +0200569 void SetDelay(int delay) { delay_ = delay; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000570 int GetDelay() { return delay_; }
571
Yves Gerey665174f2018-06-19 15:03:05 +0200572 void SetLoseFirstPacket(bool lose) { lose_first_packet_ = lose; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000573 bool GetLoseFirstPacket() { return lose_first_packet_; }
574
Yves Gerey665174f2018-06-19 15:03:05 +0200575 void SetLoss(int percent) { loss_ = percent; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000576
Yves Gerey665174f2018-06-19 15:03:05 +0200577 void SetDamage() { damage_ = true; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000578
Yves Gerey665174f2018-06-19 15:03:05 +0200579 void SetMtu(size_t mtu) { mtu_ = mtu; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000580
Yves Gerey665174f2018-06-19 15:03:05 +0200581 void SetHandshakeWait(int wait) { handshake_wait_ = wait; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000582
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800583 void SetDtlsSrtpCryptoSuites(const std::vector<int>& ciphers, bool client) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000584 if (client)
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800585 client_ssl_->SetDtlsSrtpCryptoSuites(ciphers);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000586 else
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800587 server_ssl_->SetDtlsSrtpCryptoSuites(ciphers);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000588 }
589
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800590 bool GetDtlsSrtpCryptoSuite(bool client, int* retval) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000591 if (client)
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800592 return client_ssl_->GetDtlsSrtpCryptoSuite(retval);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000593 else
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800594 return server_ssl_->GetDtlsSrtpCryptoSuite(retval);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000595 }
596
jbauch555604a2016-04-26 03:13:22 -0700597 std::unique_ptr<rtc::SSLCertificate> GetPeerCertificate(bool client) {
Taylor Brandstetterc3928662018-02-23 13:04:51 -0800598 std::unique_ptr<rtc::SSLCertChain> chain;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000599 if (client)
Taylor Brandstetterc3928662018-02-23 13:04:51 -0800600 chain = client_ssl_->GetPeerSSLCertChain();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000601 else
Taylor Brandstetterc3928662018-02-23 13:04:51 -0800602 chain = server_ssl_->GetPeerSSLCertChain();
Steve Antonf25303e2018-10-16 15:23:31 -0700603 return (chain && chain->GetSize()) ? chain->Get(0).Clone() : nullptr;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000604 }
605
Guo-wei Shieh6caafbe2015-10-05 12:43:27 -0700606 bool GetSslCipherSuite(bool client, int* retval) {
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +0000607 if (client)
Guo-wei Shieh456696a2015-09-30 21:48:54 -0700608 return client_ssl_->GetSslCipherSuite(retval);
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +0000609 else
Guo-wei Shieh456696a2015-09-30 21:48:54 -0700610 return server_ssl_->GetSslCipherSuite(retval);
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +0000611 }
612
torbjorng43166b82016-03-11 00:06:47 -0800613 int GetSslVersion(bool client) {
614 if (client)
615 return client_ssl_->GetSslVersion();
616 else
617 return server_ssl_->GetSslVersion();
618 }
619
Yves Gerey665174f2018-06-19 15:03:05 +0200620 bool ExportKeyingMaterial(const char* label,
621 const unsigned char* context,
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000622 size_t context_len,
623 bool use_context,
624 bool client,
Yves Gerey665174f2018-06-19 15:03:05 +0200625 unsigned char* result,
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000626 size_t result_len) {
627 if (client)
Yves Gerey665174f2018-06-19 15:03:05 +0200628 return client_ssl_->ExportKeyingMaterial(label, context, context_len,
629 use_context, result, result_len);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000630 else
Yves Gerey665174f2018-06-19 15:03:05 +0200631 return server_ssl_->ExportKeyingMaterial(label, context, context_len,
632 use_context, result, result_len);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000633 }
634
635 // To be implemented by subclasses.
636 virtual void WriteData() = 0;
Yves Gerey665174f2018-06-19 15:03:05 +0200637 virtual void ReadData(rtc::StreamInterface* stream) = 0;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000638 virtual void TestTransfer(int size) = 0;
639
640 protected:
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100641 rtc::SSLIdentity* client_identity() const {
642 if (!client_ssl_) {
643 return nullptr;
644 }
645 return client_ssl_->GetIdentityForTesting();
646 }
647 rtc::SSLIdentity* server_identity() const {
648 if (!server_ssl_) {
649 return nullptr;
650 }
651 return server_ssl_->GetIdentityForTesting();
652 }
653
jbauche488a0d2015-11-19 05:17:58 -0800654 std::string client_cert_pem_;
655 std::string client_private_key_pem_;
656 rtc::KeyParams client_key_type_;
657 rtc::KeyParams server_key_type_;
Yves Gerey665174f2018-06-19 15:03:05 +0200658 SSLDummyStreamBase* client_stream_; // freed by client_ssl_ destructor
659 SSLDummyStreamBase* server_stream_; // freed by server_ssl_ destructor
jbauch555604a2016-04-26 03:13:22 -0700660 std::unique_ptr<rtc::SSLStreamAdapter> client_ssl_;
661 std::unique_ptr<rtc::SSLStreamAdapter> server_ssl_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000662 int delay_;
663 size_t mtu_;
664 int loss_;
665 bool lose_first_packet_;
666 bool damage_;
667 bool dtls_;
668 int handshake_wait_;
669 bool identities_set_;
670};
671
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200672class SSLStreamAdapterTestTLS
673 : public SSLStreamAdapterTestBase,
torbjorng4e572472015-10-08 09:42:49 -0700674 public WithParamInterface<tuple<rtc::KeyParams, rtc::KeyParams>> {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000675 public:
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200676 SSLStreamAdapterTestTLS()
677 : SSLStreamAdapterTestBase("",
678 "",
679 false,
680 ::testing::get<0>(GetParam()),
jbauche488a0d2015-11-19 05:17:58 -0800681 ::testing::get<1>(GetParam())),
682 client_buffer_(kFifoBufferSize),
Yves Gerey665174f2018-06-19 15:03:05 +0200683 server_buffer_(kFifoBufferSize) {}
jbauche488a0d2015-11-19 05:17:58 -0800684
torbjorng7593aad2015-11-19 12:20:51 -0800685 void CreateStreams() override {
jbauche488a0d2015-11-19 05:17:58 -0800686 client_stream_ =
687 new SSLDummyStreamTLS(this, "c2s", &client_buffer_, &server_buffer_);
688 server_stream_ =
689 new SSLDummyStreamTLS(this, "s2c", &server_buffer_, &client_buffer_);
690 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000691
692 // Test data transfer for TLS
nisseef8b61e2016-04-29 06:09:15 -0700693 void TestTransfer(int size) override {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100694 RTC_LOG(LS_INFO) << "Starting transfer test with " << size << " bytes";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000695 // Create some dummy data to send.
696 size_t received;
697
698 send_stream_.ReserveSize(size);
699 for (int i = 0; i < size; ++i) {
700 char ch = static_cast<char>(i);
deadbeef37f5ecf2017-02-27 14:06:41 -0800701 send_stream_.Write(&ch, 1, nullptr, nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000702 }
703 send_stream_.Rewind();
704
705 // Prepare the receive stream.
706 recv_stream_.ReserveSize(size);
707
708 // Start sending
709 WriteData();
710
711 // Wait for the client to close
712 EXPECT_TRUE_WAIT(server_ssl_->GetState() == rtc::SS_CLOSED, 10000);
713
714 // Now check the data
715 recv_stream_.GetSize(&received);
716
717 EXPECT_EQ(static_cast<size_t>(size), received);
Yves Gerey665174f2018-06-19 15:03:05 +0200718 EXPECT_EQ(0,
719 memcmp(send_stream_.GetBuffer(), recv_stream_.GetBuffer(), size));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000720 }
721
nisseef8b61e2016-04-29 06:09:15 -0700722 void WriteData() override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000723 size_t position, tosend, size;
724 rtc::StreamResult rv;
725 size_t sent;
726 char block[kBlockSize];
727
728 send_stream_.GetSize(&size);
729 if (!size)
730 return;
731
732 for (;;) {
733 send_stream_.GetPosition(&position);
deadbeef37f5ecf2017-02-27 14:06:41 -0800734 if (send_stream_.Read(block, sizeof(block), &tosend, nullptr) !=
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000735 rtc::SR_EOS) {
736 rv = client_ssl_->Write(block, tosend, &sent, 0);
737
738 if (rv == rtc::SR_SUCCESS) {
739 send_stream_.SetPosition(position + sent);
Mirko Bonadei675513b2017-11-09 11:09:25 +0100740 RTC_LOG(LS_VERBOSE) << "Sent: " << position + sent;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000741 } else if (rv == rtc::SR_BLOCK) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100742 RTC_LOG(LS_VERBOSE) << "Blocked...";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000743 send_stream_.SetPosition(position);
744 break;
745 } else {
746 ADD_FAILURE();
747 break;
748 }
749 } else {
750 // Now close
Mirko Bonadei675513b2017-11-09 11:09:25 +0100751 RTC_LOG(LS_INFO) << "Wrote " << position << " bytes. Closing";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000752 client_ssl_->Close();
753 break;
754 }
755 }
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100756 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000757
Yves Gerey665174f2018-06-19 15:03:05 +0200758 void ReadData(rtc::StreamInterface* stream) override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000759 char buffer[1600];
760 size_t bread;
761 int err2;
762 rtc::StreamResult r;
763
764 for (;;) {
765 r = stream->Read(buffer, sizeof(buffer), &bread, &err2);
766
767 if (r == rtc::SR_ERROR || r == rtc::SR_EOS) {
768 // Unfortunately, errors are the way that the stream adapter
torbjorng7593aad2015-11-19 12:20:51 -0800769 // signals close in OpenSSL.
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000770 stream->Close();
771 return;
772 }
773
774 if (r == rtc::SR_BLOCK)
775 break;
776
777 ASSERT_EQ(rtc::SR_SUCCESS, r);
Mirko Bonadei675513b2017-11-09 11:09:25 +0100778 RTC_LOG(LS_VERBOSE) << "Read " << bread;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000779
deadbeef37f5ecf2017-02-27 14:06:41 -0800780 recv_stream_.Write(buffer, bread, nullptr, nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000781 }
782 }
783
784 private:
jbauche488a0d2015-11-19 05:17:58 -0800785 rtc::FifoBuffer client_buffer_;
786 rtc::FifoBuffer server_buffer_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000787 rtc::MemoryStream send_stream_;
788 rtc::MemoryStream recv_stream_;
789};
790
Harald Alvestrand13799132020-03-09 19:39:36 +0100791class SSLStreamAdapterTestDTLSBase : public SSLStreamAdapterTestBase {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000792 public:
Harald Alvestrand13799132020-03-09 19:39:36 +0100793 SSLStreamAdapterTestDTLSBase(rtc::KeyParams param1, rtc::KeyParams param2)
794 : SSLStreamAdapterTestBase("", "", true, param1, param2),
jbauche488a0d2015-11-19 05:17:58 -0800795 client_buffer_(kBufferCapacity, kDefaultBufferSize),
796 server_buffer_(kBufferCapacity, kDefaultBufferSize),
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200797 packet_size_(1000),
798 count_(0),
799 sent_(0) {}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000800
Harald Alvestrand13799132020-03-09 19:39:36 +0100801 SSLStreamAdapterTestDTLSBase(const std::string& cert_pem,
802 const std::string& private_key_pem)
Yves Gerey665174f2018-06-19 15:03:05 +0200803 : SSLStreamAdapterTestBase(cert_pem, private_key_pem, true),
804 client_buffer_(kBufferCapacity, kDefaultBufferSize),
805 server_buffer_(kBufferCapacity, kDefaultBufferSize),
806 packet_size_(1000),
807 count_(0),
808 sent_(0) {}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000809
torbjorng7593aad2015-11-19 12:20:51 -0800810 void CreateStreams() override {
jbauche488a0d2015-11-19 05:17:58 -0800811 client_stream_ =
812 new SSLDummyStreamDTLS(this, "c2s", &client_buffer_, &server_buffer_);
813 server_stream_ =
814 new SSLDummyStreamDTLS(this, "s2c", &server_buffer_, &client_buffer_);
815 }
816
nisseef8b61e2016-04-29 06:09:15 -0700817 void WriteData() override {
Yves Gerey665174f2018-06-19 15:03:05 +0200818 unsigned char* packet = new unsigned char[1600];
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000819
jbauche488a0d2015-11-19 05:17:58 -0800820 while (sent_ < count_) {
torbjorng7593aad2015-11-19 12:20:51 -0800821 unsigned int rand_state = sent_;
822 packet[0] = sent_;
823 for (size_t i = 1; i < packet_size_; i++) {
824 // This is a simple LC PRNG. Keep in synch with identical code below.
825 rand_state = (rand_state * 251 + 19937) >> 7;
826 packet[i] = rand_state & 0xff;
827 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000828
829 size_t sent;
torbjorng7593aad2015-11-19 12:20:51 -0800830 rtc::StreamResult rv = client_ssl_->Write(packet, packet_size_, &sent, 0);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000831 if (rv == rtc::SR_SUCCESS) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100832 RTC_LOG(LS_VERBOSE) << "Sent: " << sent_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000833 sent_++;
834 } else if (rv == rtc::SR_BLOCK) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100835 RTC_LOG(LS_VERBOSE) << "Blocked...";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000836 break;
837 } else {
838 ADD_FAILURE();
839 break;
840 }
jbauche488a0d2015-11-19 05:17:58 -0800841 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000842
Yves Gerey665174f2018-06-19 15:03:05 +0200843 delete[] packet;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000844 }
845
Yves Gerey665174f2018-06-19 15:03:05 +0200846 void ReadData(rtc::StreamInterface* stream) override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000847 unsigned char buffer[2000];
848 size_t bread;
849 int err2;
850 rtc::StreamResult r;
851
852 for (;;) {
853 r = stream->Read(buffer, 2000, &bread, &err2);
854
855 if (r == rtc::SR_ERROR) {
856 // Unfortunately, errors are the way that the stream adapter
857 // signals close right now
858 stream->Close();
859 return;
860 }
861
862 if (r == rtc::SR_BLOCK)
863 break;
864
865 ASSERT_EQ(rtc::SR_SUCCESS, r);
Mirko Bonadei675513b2017-11-09 11:09:25 +0100866 RTC_LOG(LS_VERBOSE) << "Read " << bread;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000867
868 // Now parse the datagram
869 ASSERT_EQ(packet_size_, bread);
torbjorng7593aad2015-11-19 12:20:51 -0800870 unsigned char packet_num = buffer[0];
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000871
torbjorng7593aad2015-11-19 12:20:51 -0800872 unsigned int rand_state = packet_num;
873 for (size_t i = 1; i < packet_size_; i++) {
874 // This is a simple LC PRNG. Keep in synch with identical code above.
875 rand_state = (rand_state * 251 + 19937) >> 7;
876 ASSERT_EQ(rand_state & 0xff, buffer[i]);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000877 }
878 received_.insert(packet_num);
879 }
880 }
881
nisseef8b61e2016-04-29 06:09:15 -0700882 void TestTransfer(int count) override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000883 count_ = count;
884
885 WriteData();
886
887 EXPECT_TRUE_WAIT(sent_ == count_, 10000);
Mirko Bonadei675513b2017-11-09 11:09:25 +0100888 RTC_LOG(LS_INFO) << "sent_ == " << sent_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000889
890 if (damage_) {
891 WAIT(false, 2000);
892 EXPECT_EQ(0U, received_.size());
893 } else if (loss_ == 0) {
Yves Gerey665174f2018-06-19 15:03:05 +0200894 EXPECT_EQ_WAIT(static_cast<size_t>(sent_), received_.size(), 1000);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000895 } else {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100896 RTC_LOG(LS_INFO) << "Sent " << sent_ << " packets; received "
897 << received_.size();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000898 }
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100899 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000900
Harald Alvestrand13799132020-03-09 19:39:36 +0100901 protected:
jbauche488a0d2015-11-19 05:17:58 -0800902 BufferQueueStream client_buffer_;
903 BufferQueueStream server_buffer_;
Harald Alvestrand13799132020-03-09 19:39:36 +0100904
905 private:
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000906 size_t packet_size_;
907 int count_;
908 int sent_;
909 std::set<int> received_;
910};
911
Harald Alvestrand13799132020-03-09 19:39:36 +0100912class SSLStreamAdapterTestDTLS
913 : public SSLStreamAdapterTestDTLSBase,
914 public WithParamInterface<tuple<rtc::KeyParams, rtc::KeyParams>> {
915 public:
916 SSLStreamAdapterTestDTLS()
917 : SSLStreamAdapterTestDTLSBase(::testing::get<0>(GetParam()),
918 ::testing::get<1>(GetParam())) {}
919
920 SSLStreamAdapterTestDTLS(const std::string& cert_pem,
921 const std::string& private_key_pem)
922 : SSLStreamAdapterTestDTLSBase(cert_pem, private_key_pem) {}
923};
924
Yves Gerey665174f2018-06-19 15:03:05 +0200925rtc::StreamResult SSLDummyStreamBase::Write(const void* data,
926 size_t data_len,
927 size_t* written,
928 int* error) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100929 RTC_LOG(LS_VERBOSE) << "Writing to loopback " << data_len;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000930
931 if (first_packet_) {
932 first_packet_ = false;
jbauche488a0d2015-11-19 05:17:58 -0800933 if (test_base_->GetLoseFirstPacket()) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100934 RTC_LOG(LS_INFO) << "Losing initial packet of length " << data_len;
torbjorng7593aad2015-11-19 12:20:51 -0800935 *written = data_len; // Fake successful writing also to writer.
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000936 return rtc::SR_SUCCESS;
937 }
938 }
939
jbauche488a0d2015-11-19 05:17:58 -0800940 return test_base_->DataWritten(this, data, data_len, written, error);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100941}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000942
943class SSLStreamAdapterTestDTLSFromPEMStrings : public SSLStreamAdapterTestDTLS {
944 public:
Yves Gerey665174f2018-06-19 15:03:05 +0200945 SSLStreamAdapterTestDTLSFromPEMStrings()
946 : SSLStreamAdapterTestDTLS(kCERT_PEM, kRSA_PRIVATE_KEY_PEM) {}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000947};
948
Jian Cui0a8798b2017-11-16 16:58:02 -0800949// Test fixture for certificate chaining. Server will push more than one
950// certificate.
951class SSLStreamAdapterTestDTLSCertChain : public SSLStreamAdapterTestDTLS {
952 public:
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100953 SSLStreamAdapterTestDTLSCertChain() : SSLStreamAdapterTestDTLS("", "") {}
Jian Cui0a8798b2017-11-16 16:58:02 -0800954 void SetUp() override {
955 CreateStreams();
956
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100957 client_ssl_ =
958 rtc::SSLStreamAdapter::Create(absl::WrapUnique(client_stream_));
959 server_ssl_ =
960 rtc::SSLStreamAdapter::Create(absl::WrapUnique(server_stream_));
Jian Cui0a8798b2017-11-16 16:58:02 -0800961
962 // Set up the slots
963 client_ssl_->SignalEvent.connect(
964 reinterpret_cast<SSLStreamAdapterTestBase*>(this),
965 &SSLStreamAdapterTestBase::OnEvent);
966 server_ssl_->SignalEvent.connect(
967 reinterpret_cast<SSLStreamAdapterTestBase*>(this),
968 &SSLStreamAdapterTestBase::OnEvent);
969
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100970 std::unique_ptr<rtc::SSLIdentity> client_identity;
Jian Cui0a8798b2017-11-16 16:58:02 -0800971 if (!client_cert_pem_.empty() && !client_private_key_pem_.empty()) {
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100972 client_identity = rtc::SSLIdentity::CreateFromPEMStrings(
Jian Cui0a8798b2017-11-16 16:58:02 -0800973 client_private_key_pem_, client_cert_pem_);
974 } else {
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100975 client_identity = rtc::SSLIdentity::Create("client", client_key_type_);
Jian Cui0a8798b2017-11-16 16:58:02 -0800976 }
977
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100978 client_ssl_->SetIdentity(std::move(client_identity));
Jian Cui0a8798b2017-11-16 16:58:02 -0800979 }
980};
981
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000982// Basic tests: TLS
983
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000984// Test that we can make a handshake work
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200985TEST_P(SSLStreamAdapterTestTLS, TestTLSConnect) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000986 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100987}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000988
Jian Cui0a8798b2017-11-16 16:58:02 -0800989TEST_P(SSLStreamAdapterTestTLS, GetPeerCertChainWithOneCertificate) {
990 TestHandshake();
991 std::unique_ptr<rtc::SSLCertChain> cert_chain =
992 client_ssl_->GetPeerSSLCertChain();
Jian Cui0a8798b2017-11-16 16:58:02 -0800993 ASSERT_NE(nullptr, cert_chain);
994 EXPECT_EQ(1u, cert_chain->GetSize());
Taylor Brandstetterc3928662018-02-23 13:04:51 -0800995 EXPECT_EQ(cert_chain->Get(0).ToPEMString(),
Harald Alvestrand8515d5a2020-03-20 22:51:32 +0100996 server_identity()->certificate().ToPEMString());
Jian Cui0a8798b2017-11-16 16:58:02 -0800997}
998
999TEST_F(SSLStreamAdapterTestDTLSCertChain, TwoCertHandshake) {
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001000 auto server_identity = rtc::SSLIdentity::CreateFromPEMChainStrings(
Jian Cui0a8798b2017-11-16 16:58:02 -08001001 kRSA_PRIVATE_KEY_PEM, std::string(kCERT_PEM) + kCACert);
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001002 server_ssl_->SetIdentity(std::move(server_identity));
Jian Cui0a8798b2017-11-16 16:58:02 -08001003 TestHandshake();
1004 std::unique_ptr<rtc::SSLCertChain> peer_cert_chain =
1005 client_ssl_->GetPeerSSLCertChain();
1006 ASSERT_NE(nullptr, peer_cert_chain);
1007 ASSERT_EQ(2u, peer_cert_chain->GetSize());
1008 EXPECT_EQ(kCERT_PEM, peer_cert_chain->Get(0).ToPEMString());
1009 EXPECT_EQ(kCACert, peer_cert_chain->Get(1).ToPEMString());
1010}
1011
David Benjaminea84b6b2017-12-01 17:25:45 -05001012TEST_F(SSLStreamAdapterTestDTLSCertChain, TwoCertHandshakeWithCopy) {
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001013 server_ssl_->SetIdentity(rtc::SSLIdentity::CreateFromPEMChainStrings(
1014 kRSA_PRIVATE_KEY_PEM, std::string(kCERT_PEM) + kCACert));
David Benjaminea84b6b2017-12-01 17:25:45 -05001015 TestHandshake();
1016 std::unique_ptr<rtc::SSLCertChain> peer_cert_chain =
1017 client_ssl_->GetPeerSSLCertChain();
1018 ASSERT_NE(nullptr, peer_cert_chain);
1019 ASSERT_EQ(2u, peer_cert_chain->GetSize());
1020 EXPECT_EQ(kCERT_PEM, peer_cert_chain->Get(0).ToPEMString());
1021 EXPECT_EQ(kCACert, peer_cert_chain->Get(1).ToPEMString());
1022}
1023
Jian Cui0a8798b2017-11-16 16:58:02 -08001024TEST_F(SSLStreamAdapterTestDTLSCertChain, ThreeCertHandshake) {
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001025 server_ssl_->SetIdentity(rtc::SSLIdentity::CreateFromPEMChainStrings(
1026 kRSA_PRIVATE_KEY_PEM, std::string(kCERT_PEM) + kIntCert1 + kCACert));
Jian Cui0a8798b2017-11-16 16:58:02 -08001027 TestHandshake();
1028 std::unique_ptr<rtc::SSLCertChain> peer_cert_chain =
1029 client_ssl_->GetPeerSSLCertChain();
1030 ASSERT_NE(nullptr, peer_cert_chain);
1031 ASSERT_EQ(3u, peer_cert_chain->GetSize());
1032 EXPECT_EQ(kCERT_PEM, peer_cert_chain->Get(0).ToPEMString());
1033 EXPECT_EQ(kIntCert1, peer_cert_chain->Get(1).ToPEMString());
1034 EXPECT_EQ(kCACert, peer_cert_chain->Get(2).ToPEMString());
1035}
1036
jiayl@webrtc.orgf1d751c2014-09-25 16:38:46 +00001037// Test that closing the connection on one side updates the other side.
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001038TEST_P(SSLStreamAdapterTestTLS, TestTLSClose) {
jiayl@webrtc.orgf1d751c2014-09-25 16:38:46 +00001039 TestHandshake();
1040 client_ssl_->Close();
1041 EXPECT_EQ_WAIT(rtc::SS_CLOSED, server_ssl_->GetState(), handshake_wait_);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001042}
jiayl@webrtc.orgf1d751c2014-09-25 16:38:46 +00001043
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001044// Test transfer -- trivial
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001045TEST_P(SSLStreamAdapterTestTLS, TestTLSTransfer) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001046 TestHandshake();
1047 TestTransfer(100000);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001048}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001049
1050// Test read-write after close.
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001051TEST_P(SSLStreamAdapterTestTLS, ReadWriteAfterClose) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001052 TestHandshake();
1053 TestTransfer(100000);
1054 client_ssl_->Close();
1055
1056 rtc::StreamResult rv;
1057 char block[kBlockSize];
1058 size_t dummy;
1059
1060 // It's an error to write after closed.
deadbeef37f5ecf2017-02-27 14:06:41 -08001061 rv = client_ssl_->Write(block, sizeof(block), &dummy, nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001062 ASSERT_EQ(rtc::SR_ERROR, rv);
1063
1064 // But after closed read gives you EOS.
deadbeef37f5ecf2017-02-27 14:06:41 -08001065 rv = client_ssl_->Read(block, sizeof(block), &dummy, nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001066 ASSERT_EQ(rtc::SR_EOS, rv);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001067}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001068
1069// Test a handshake with a bogus peer digest
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001070TEST_P(SSLStreamAdapterTestTLS, TestTLSBogusDigest) {
deadbeef89824f62016-09-30 11:55:43 -07001071 SetPeerIdentitiesByDigest(false, true);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001072 TestHandshake(false);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001073}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001074
deadbeef89824f62016-09-30 11:55:43 -07001075TEST_P(SSLStreamAdapterTestTLS, TestTLSDelayedIdentity) {
1076 TestHandshakeWithDelayedIdentity(true);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001077}
deadbeef89824f62016-09-30 11:55:43 -07001078
1079TEST_P(SSLStreamAdapterTestTLS, TestTLSDelayedIdentityWithBogusDigest) {
1080 TestHandshakeWithDelayedIdentity(false);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001081}
deadbeef89824f62016-09-30 11:55:43 -07001082
1083// Test that the correct error is returned when SetPeerCertificateDigest is
1084// called with an unknown algorithm.
1085TEST_P(SSLStreamAdapterTestTLS,
1086 TestSetPeerCertificateDigestWithUnknownAlgorithm) {
1087 unsigned char server_digest[20];
1088 size_t server_digest_len;
1089 bool rv;
1090 rtc::SSLPeerCertificateDigestError err;
1091
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001092 rv = server_identity()->certificate().ComputeDigest(
deadbeef89824f62016-09-30 11:55:43 -07001093 rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
1094 ASSERT_TRUE(rv);
1095
1096 rv = client_ssl_->SetPeerCertificateDigest("unknown algorithm", server_digest,
1097 server_digest_len, &err);
1098 EXPECT_EQ(rtc::SSLPeerCertificateDigestError::UNKNOWN_ALGORITHM, err);
1099 EXPECT_FALSE(rv);
1100}
1101
1102// Test that the correct error is returned when SetPeerCertificateDigest is
1103// called with an invalid digest length.
1104TEST_P(SSLStreamAdapterTestTLS, TestSetPeerCertificateDigestWithInvalidLength) {
1105 unsigned char server_digest[20];
1106 size_t server_digest_len;
1107 bool rv;
1108 rtc::SSLPeerCertificateDigestError err;
1109
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001110 rv = server_identity()->certificate().ComputeDigest(
deadbeef89824f62016-09-30 11:55:43 -07001111 rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
1112 ASSERT_TRUE(rv);
1113
1114 rv = client_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, server_digest,
1115 server_digest_len - 1, &err);
1116 EXPECT_EQ(rtc::SSLPeerCertificateDigestError::INVALID_LENGTH, err);
1117 EXPECT_FALSE(rv);
1118}
1119
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001120// Test moving a bunch of data
1121
1122// Basic tests: DTLS
1123// Test that we can make a handshake work
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001124TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnect) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001125 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001126}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001127
1128// Test that we can make a handshake work if the first packet in
1129// each direction is lost. This gives us predictable loss
1130// rather than having to tune random
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001131TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnectWithLostFirstPacket) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001132 SetLoseFirstPacket(true);
1133 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001134}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001135
1136// Test a handshake with loss and delay
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001137TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnectWithLostFirstPacketDelay2s) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001138 SetLoseFirstPacket(true);
1139 SetDelay(2000);
1140 SetHandshakeWait(20000);
1141 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001142}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001143
1144// Test a handshake with small MTU
pbos@webrtc.org127ca3f2014-10-09 07:52:03 +00001145// Disabled due to https://code.google.com/p/webrtc/issues/detail?id=3910
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001146TEST_P(SSLStreamAdapterTestDTLS, DISABLED_TestDTLSConnectWithSmallMtu) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001147 SetMtu(700);
1148 SetHandshakeWait(20000);
1149 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001150}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001151
1152// Test transfer -- trivial
jbauche488a0d2015-11-19 05:17:58 -08001153TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransfer) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001154 TestHandshake();
1155 TestTransfer(100);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001156}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001157
jbauche488a0d2015-11-19 05:17:58 -08001158TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransferWithLoss) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001159 TestHandshake();
1160 SetLoss(10);
1161 TestTransfer(100);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001162}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001163
jbauche488a0d2015-11-19 05:17:58 -08001164TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransferWithDamage) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001165 SetDamage(); // Must be called first because first packet
1166 // write happens at end of handshake.
1167 TestHandshake();
1168 TestTransfer(100);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001169}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001170
deadbeef89824f62016-09-30 11:55:43 -07001171TEST_P(SSLStreamAdapterTestDTLS, TestDTLSDelayedIdentity) {
1172 TestHandshakeWithDelayedIdentity(true);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001173}
deadbeef89824f62016-09-30 11:55:43 -07001174
1175TEST_P(SSLStreamAdapterTestDTLS, TestDTLSDelayedIdentityWithBogusDigest) {
1176 TestHandshakeWithDelayedIdentity(false);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001177}
deadbeef89824f62016-09-30 11:55:43 -07001178
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001179// Test DTLS-SRTP with all high ciphers
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001180TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpHigh) {
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001181 std::vector<int> high;
1182 high.push_back(rtc::SRTP_AES128_CM_SHA1_80);
1183 SetDtlsSrtpCryptoSuites(high, true);
1184 SetDtlsSrtpCryptoSuites(high, false);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001185 TestHandshake();
1186
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001187 int client_cipher;
1188 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1189 int server_cipher;
1190 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001191
1192 ASSERT_EQ(client_cipher, server_cipher);
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001193 ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_80);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001194}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001195
1196// Test DTLS-SRTP with all low ciphers
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001197TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpLow) {
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001198 std::vector<int> low;
1199 low.push_back(rtc::SRTP_AES128_CM_SHA1_32);
1200 SetDtlsSrtpCryptoSuites(low, true);
1201 SetDtlsSrtpCryptoSuites(low, false);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001202 TestHandshake();
1203
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001204 int client_cipher;
1205 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1206 int server_cipher;
1207 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001208
1209 ASSERT_EQ(client_cipher, server_cipher);
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001210 ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_32);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001211}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001212
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001213// Test DTLS-SRTP with a mismatch -- should not converge
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001214TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpHighLow) {
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001215 std::vector<int> high;
1216 high.push_back(rtc::SRTP_AES128_CM_SHA1_80);
1217 std::vector<int> low;
1218 low.push_back(rtc::SRTP_AES128_CM_SHA1_32);
1219 SetDtlsSrtpCryptoSuites(high, true);
1220 SetDtlsSrtpCryptoSuites(low, false);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001221 TestHandshake();
1222
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001223 int client_cipher;
1224 ASSERT_FALSE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1225 int server_cipher;
1226 ASSERT_FALSE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001227}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001228
1229// Test DTLS-SRTP with each side being mixed -- should select high
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001230TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpMixed) {
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001231 std::vector<int> mixed;
1232 mixed.push_back(rtc::SRTP_AES128_CM_SHA1_80);
1233 mixed.push_back(rtc::SRTP_AES128_CM_SHA1_32);
1234 SetDtlsSrtpCryptoSuites(mixed, true);
1235 SetDtlsSrtpCryptoSuites(mixed, false);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001236 TestHandshake();
1237
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001238 int client_cipher;
1239 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1240 int server_cipher;
1241 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001242
1243 ASSERT_EQ(client_cipher, server_cipher);
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001244 ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_80);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001245}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001246
jbauchcb560652016-08-04 05:20:32 -07001247// Test DTLS-SRTP with all GCM-128 ciphers.
1248TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCM128) {
jbauchcb560652016-08-04 05:20:32 -07001249 std::vector<int> gcm128;
1250 gcm128.push_back(rtc::SRTP_AEAD_AES_128_GCM);
1251 SetDtlsSrtpCryptoSuites(gcm128, true);
1252 SetDtlsSrtpCryptoSuites(gcm128, false);
1253 TestHandshake();
1254
1255 int client_cipher;
1256 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1257 int server_cipher;
1258 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
1259
1260 ASSERT_EQ(client_cipher, server_cipher);
1261 ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_128_GCM);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001262}
jbauchcb560652016-08-04 05:20:32 -07001263
1264// Test DTLS-SRTP with all GCM-256 ciphers.
1265TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCM256) {
jbauchcb560652016-08-04 05:20:32 -07001266 std::vector<int> gcm256;
1267 gcm256.push_back(rtc::SRTP_AEAD_AES_256_GCM);
1268 SetDtlsSrtpCryptoSuites(gcm256, true);
1269 SetDtlsSrtpCryptoSuites(gcm256, false);
1270 TestHandshake();
1271
1272 int client_cipher;
1273 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1274 int server_cipher;
1275 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
1276
1277 ASSERT_EQ(client_cipher, server_cipher);
1278 ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_256_GCM);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001279}
jbauchcb560652016-08-04 05:20:32 -07001280
1281// Test DTLS-SRTP with mixed GCM-128/-256 ciphers -- should not converge.
1282TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCMMismatch) {
jbauchcb560652016-08-04 05:20:32 -07001283 std::vector<int> gcm128;
1284 gcm128.push_back(rtc::SRTP_AEAD_AES_128_GCM);
1285 std::vector<int> gcm256;
1286 gcm256.push_back(rtc::SRTP_AEAD_AES_256_GCM);
1287 SetDtlsSrtpCryptoSuites(gcm128, true);
1288 SetDtlsSrtpCryptoSuites(gcm256, false);
1289 TestHandshake();
1290
1291 int client_cipher;
1292 ASSERT_FALSE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1293 int server_cipher;
1294 ASSERT_FALSE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001295}
jbauchcb560652016-08-04 05:20:32 -07001296
1297// Test DTLS-SRTP with both GCM-128/-256 ciphers -- should select GCM-256.
1298TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCMMixed) {
jbauchcb560652016-08-04 05:20:32 -07001299 std::vector<int> gcmBoth;
1300 gcmBoth.push_back(rtc::SRTP_AEAD_AES_256_GCM);
1301 gcmBoth.push_back(rtc::SRTP_AEAD_AES_128_GCM);
1302 SetDtlsSrtpCryptoSuites(gcmBoth, true);
1303 SetDtlsSrtpCryptoSuites(gcmBoth, false);
1304 TestHandshake();
1305
1306 int client_cipher;
1307 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1308 int server_cipher;
1309 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
1310
1311 ASSERT_EQ(client_cipher, server_cipher);
1312 ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_256_GCM);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001313}
jbauchcb560652016-08-04 05:20:32 -07001314
1315// Test SRTP cipher suite lengths.
1316TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpKeyAndSaltLengths) {
1317 int key_len;
1318 int salt_len;
1319
Yves Gerey665174f2018-06-19 15:03:05 +02001320 ASSERT_FALSE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_INVALID_CRYPTO_SUITE,
1321 &key_len, &salt_len));
jbauchcb560652016-08-04 05:20:32 -07001322
Yves Gerey665174f2018-06-19 15:03:05 +02001323 ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_AES128_CM_SHA1_32,
1324 &key_len, &salt_len));
1325 ASSERT_EQ(128 / 8, key_len);
1326 ASSERT_EQ(112 / 8, salt_len);
jbauchcb560652016-08-04 05:20:32 -07001327
Yves Gerey665174f2018-06-19 15:03:05 +02001328 ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_AES128_CM_SHA1_80,
1329 &key_len, &salt_len));
1330 ASSERT_EQ(128 / 8, key_len);
1331 ASSERT_EQ(112 / 8, salt_len);
jbauchcb560652016-08-04 05:20:32 -07001332
Yves Gerey665174f2018-06-19 15:03:05 +02001333 ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_AEAD_AES_128_GCM,
1334 &key_len, &salt_len));
1335 ASSERT_EQ(128 / 8, key_len);
1336 ASSERT_EQ(96 / 8, salt_len);
jbauchcb560652016-08-04 05:20:32 -07001337
Yves Gerey665174f2018-06-19 15:03:05 +02001338 ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_AEAD_AES_256_GCM,
1339 &key_len, &salt_len));
1340 ASSERT_EQ(256 / 8, key_len);
1341 ASSERT_EQ(96 / 8, salt_len);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001342}
jbauchcb560652016-08-04 05:20:32 -07001343
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001344// Test an exporter
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001345TEST_P(SSLStreamAdapterTestDTLS, TestDTLSExporter) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001346 TestHandshake();
1347 unsigned char client_out[20];
1348 unsigned char server_out[20];
1349
1350 bool result;
Yves Gerey665174f2018-06-19 15:03:05 +02001351 result = ExportKeyingMaterial(kExporterLabel, kExporterContext,
1352 kExporterContextLen, true, true, client_out,
1353 sizeof(client_out));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001354 ASSERT_TRUE(result);
1355
Yves Gerey665174f2018-06-19 15:03:05 +02001356 result = ExportKeyingMaterial(kExporterLabel, kExporterContext,
1357 kExporterContextLen, true, false, server_out,
1358 sizeof(server_out));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001359 ASSERT_TRUE(result);
1360
1361 ASSERT_TRUE(!memcmp(client_out, server_out, sizeof(client_out)));
1362}
1363
1364// Test not yet valid certificates are not rejected.
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001365TEST_P(SSLStreamAdapterTestDTLS, TestCertNotYetValid) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001366 long one_day = 60 * 60 * 24;
1367 // Make the certificates not valid until one day later.
1368 ResetIdentitiesWithValidity(one_day, one_day);
1369 TestHandshake();
1370}
1371
1372// Test expired certificates are not rejected.
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001373TEST_P(SSLStreamAdapterTestDTLS, TestCertExpired) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001374 long one_day = 60 * 60 * 24;
1375 // Make the certificates already expired.
1376 ResetIdentitiesWithValidity(-one_day, -one_day);
1377 TestHandshake();
1378}
1379
1380// Test data transfer using certs created from strings.
torbjorng7593aad2015-11-19 12:20:51 -08001381TEST_F(SSLStreamAdapterTestDTLSFromPEMStrings, TestTransfer) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001382 TestHandshake();
1383 TestTransfer(100);
1384}
1385
1386// Test getting the remote certificate.
torbjorng7593aad2015-11-19 12:20:51 -08001387TEST_F(SSLStreamAdapterTestDTLSFromPEMStrings, TestDTLSGetPeerCertificate) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001388 // Peer certificates haven't been received yet.
kwibergb4d01c42016-04-06 05:15:06 -07001389 ASSERT_FALSE(GetPeerCertificate(true));
1390 ASSERT_FALSE(GetPeerCertificate(false));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001391
1392 TestHandshake();
1393
1394 // The client should have a peer certificate after the handshake.
jbauch555604a2016-04-26 03:13:22 -07001395 std::unique_ptr<rtc::SSLCertificate> client_peer_cert =
kwibergb4d01c42016-04-06 05:15:06 -07001396 GetPeerCertificate(true);
1397 ASSERT_TRUE(client_peer_cert);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001398
1399 // It's not kCERT_PEM.
1400 std::string client_peer_string = client_peer_cert->ToPEMString();
1401 ASSERT_NE(kCERT_PEM, client_peer_string);
1402
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001403 // The server should have a peer certificate after the handshake.
jbauch555604a2016-04-26 03:13:22 -07001404 std::unique_ptr<rtc::SSLCertificate> server_peer_cert =
kwibergb4d01c42016-04-06 05:15:06 -07001405 GetPeerCertificate(false);
1406 ASSERT_TRUE(server_peer_cert);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001407
1408 // It's kCERT_PEM
1409 ASSERT_EQ(kCERT_PEM, server_peer_cert->ToPEMString());
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001410}
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +00001411
Joachim Bauch831c5582015-05-20 12:48:41 +02001412// Test getting the used DTLS 1.2 ciphers.
1413// DTLS 1.2 enabled for client and server -> DTLS 1.2 will be used.
Guo-wei Shieh456696a2015-09-30 21:48:54 -07001414TEST_P(SSLStreamAdapterTestDTLS, TestGetSslCipherSuiteDtls12Both) {
Joachim Bauch831c5582015-05-20 12:48:41 +02001415 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_12);
1416 TestHandshake();
1417
Guo-wei Shieh6caafbe2015-10-05 12:43:27 -07001418 int client_cipher;
Guo-wei Shieh456696a2015-09-30 21:48:54 -07001419 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
Guo-wei Shieh6caafbe2015-10-05 12:43:27 -07001420 int server_cipher;
Guo-wei Shieh456696a2015-09-30 21:48:54 -07001421 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
Joachim Bauch831c5582015-05-20 12:48:41 +02001422
torbjorng43166b82016-03-11 00:06:47 -08001423 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(true));
1424 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(false));
1425
Joachim Bauch831c5582015-05-20 12:48:41 +02001426 ASSERT_EQ(client_cipher, server_cipher);
torbjorng43166b82016-03-11 00:06:47 -08001427 ASSERT_TRUE(rtc::SSLStreamAdapter::IsAcceptableCipher(
1428 server_cipher, ::testing::get<1>(GetParam()).type()));
Joachim Bauch831c5582015-05-20 12:48:41 +02001429}
1430
Harald Alvestrand13799132020-03-09 19:39:36 +01001431// Test getting the used DTLS ciphers.
1432// DTLS 1.0 is max version for client and server, this will only work if
1433// legacy is enabled.
1434TEST_P(SSLStreamAdapterTestDTLS, TestGetSslCipherSuite) {
1435 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
Benjamin Wrightaf1f8652019-04-01 18:25:23 +00001436 TestHandshake();
1437
1438 int client_cipher;
1439 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1440 int server_cipher;
1441 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1442
1443 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
1444 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
torbjorng43166b82016-03-11 00:06:47 -08001445
Joachim Bauch831c5582015-05-20 12:48:41 +02001446 ASSERT_EQ(client_cipher, server_cipher);
torbjorng43166b82016-03-11 00:06:47 -08001447 ASSERT_TRUE(rtc::SSLStreamAdapter::IsAcceptableCipher(
1448 server_cipher, ::testing::get<1>(GetParam()).type()));
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +00001449}
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001450
torbjorng4e572472015-10-08 09:42:49 -07001451// The RSA keysizes here might look strange, why not include the RFC's size
1452// 2048?. The reason is test case slowness; testing two sizes to exercise
1453// parametrization is sufficient.
Mirko Bonadeic84f6612019-01-31 12:20:57 +01001454INSTANTIATE_TEST_SUITE_P(
torbjorng4e572472015-10-08 09:42:49 -07001455 SSLStreamAdapterTestsTLS,
1456 SSLStreamAdapterTestTLS,
1457 Combine(Values(rtc::KeyParams::RSA(1024, 65537),
1458 rtc::KeyParams::RSA(1152, 65537),
1459 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256)),
1460 Values(rtc::KeyParams::RSA(1024, 65537),
1461 rtc::KeyParams::RSA(1152, 65537),
1462 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256))));
Mirko Bonadeic84f6612019-01-31 12:20:57 +01001463INSTANTIATE_TEST_SUITE_P(
torbjorng4e572472015-10-08 09:42:49 -07001464 SSLStreamAdapterTestsDTLS,
1465 SSLStreamAdapterTestDTLS,
1466 Combine(Values(rtc::KeyParams::RSA(1024, 65537),
1467 rtc::KeyParams::RSA(1152, 65537),
1468 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256)),
1469 Values(rtc::KeyParams::RSA(1024, 65537),
1470 rtc::KeyParams::RSA(1152, 65537),
1471 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256))));
Harald Alvestrand13799132020-03-09 19:39:36 +01001472
1473// Tests for enabling / disabling legacy TLS protocols in DTLS.
1474class SSLStreamAdapterTestDTLSLegacyProtocols
1475 : public SSLStreamAdapterTestDTLSBase {
1476 public:
1477 SSLStreamAdapterTestDTLSLegacyProtocols()
1478 : SSLStreamAdapterTestDTLSBase(rtc::KeyParams::ECDSA(rtc::EC_NIST_P256),
1479 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256)) {
1480 }
1481
1482 // Do not use the SetUp version from the parent class.
1483 void SetUp() override {}
1484
1485 // The legacy TLS protocols flag is read when the OpenSSLStreamAdapter is
1486 // initialized, so we set the experiment while creationg client_ssl_
1487 // and server_ssl_.
1488
1489 void ConfigureClient(std::string experiment) {
1490 webrtc::test::ScopedFieldTrials trial(experiment);
1491 client_stream_ =
1492 new SSLDummyStreamDTLS(this, "c2s", &client_buffer_, &server_buffer_);
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001493 client_ssl_ =
1494 rtc::SSLStreamAdapter::Create(absl::WrapUnique(client_stream_));
Harald Alvestrand13799132020-03-09 19:39:36 +01001495 client_ssl_->SignalEvent.connect(
1496 static_cast<SSLStreamAdapterTestBase*>(this),
1497 &SSLStreamAdapterTestBase::OnEvent);
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001498 auto client_identity = rtc::SSLIdentity::Create("client", client_key_type_);
1499 client_ssl_->SetIdentity(std::move(client_identity));
Harald Alvestrand13799132020-03-09 19:39:36 +01001500 }
1501
1502 void ConfigureServer(std::string experiment) {
1503 // webrtc::test::ScopedFieldTrials trial(experiment);
1504 server_stream_ =
1505 new SSLDummyStreamDTLS(this, "s2c", &server_buffer_, &client_buffer_);
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001506 server_ssl_ =
1507 rtc::SSLStreamAdapter::Create(absl::WrapUnique(server_stream_));
Harald Alvestrand13799132020-03-09 19:39:36 +01001508 server_ssl_->SignalEvent.connect(
1509 static_cast<SSLStreamAdapterTestBase*>(this),
1510 &SSLStreamAdapterTestBase::OnEvent);
Harald Alvestrand8515d5a2020-03-20 22:51:32 +01001511 server_ssl_->SetIdentity(
1512 rtc::SSLIdentity::Create("server", server_key_type_));
Harald Alvestrand13799132020-03-09 19:39:36 +01001513 }
1514};
1515
1516// Test getting the used DTLS ciphers.
1517// DTLS 1.2 enabled for neither client nor server -> DTLS 1.0 will be used.
1518TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols, TestGetSslCipherSuite) {
1519 ConfigureClient("");
1520 ConfigureServer("");
1521 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
1522 TestHandshake();
1523
1524 int client_cipher;
1525 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1526 int server_cipher;
1527 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1528
1529 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
1530 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
1531
1532 ASSERT_EQ(client_cipher, server_cipher);
1533}
1534
1535// Test getting the used DTLS 1.2 ciphers.
1536// DTLS 1.2 enabled for client and server -> DTLS 1.2 will be used.
1537TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1538 TestGetSslCipherSuiteDtls12Both) {
1539 ConfigureClient("");
1540 ConfigureServer("");
1541 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_12);
1542 TestHandshake();
1543
1544 int client_cipher;
1545 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1546 int server_cipher;
1547 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1548
1549 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(true));
1550 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(false));
1551
1552 ASSERT_EQ(client_cipher, server_cipher);
1553}
1554
1555// DTLS 1.2 enabled for client only -> DTLS 1.0 will be used.
1556TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1557 TestGetSslCipherSuiteDtls12Client) {
1558 ConfigureClient("");
1559 ConfigureServer("");
1560 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_12);
1561 TestHandshake();
1562
1563 int client_cipher;
1564 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1565 int server_cipher;
1566 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1567
1568 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
1569 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
1570
1571 ASSERT_EQ(client_cipher, server_cipher);
1572}
1573
1574// DTLS 1.2 enabled for server only -> DTLS 1.0 will be used.
1575TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1576 TestGetSslCipherSuiteDtls12Server) {
1577 ConfigureClient("");
1578 ConfigureServer("");
1579 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_10);
1580 TestHandshake();
1581
1582 int client_cipher;
1583 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1584 int server_cipher;
1585 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1586
1587 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
1588 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
1589
1590 ASSERT_EQ(client_cipher, server_cipher);
1591}
1592
1593// Client has legacy TLS versions disabled, server has DTLS 1.0 only.
1594// This is meant to cause a failure.
1595TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1596 TestGetSslVersionLegacyDisabledServer10) {
1597 ConfigureClient("WebRTC-LegacyTlsProtocols/Disabled/");
1598 ConfigureServer("");
1599 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_12);
1600 // Handshake should fail.
1601 TestHandshake(false);
1602}
1603
1604// Both client and server have legacy TLS versions disabled and support
1605// DTLS 1.2. This should work.
1606TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1607 TestGetSslVersionLegacyDisabledServer12) {
1608 ConfigureClient("WebRTC-LegacyTlsProtocols/Disabled/");
1609 ConfigureServer("WebRTC-LegacyTlsProtocols/Disabled/");
1610 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_12);
1611 TestHandshake();
1612}
1613
1614// Both client and server have legacy TLS versions enabled and support DTLS 1.0.
1615// This should work.
1616TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1617 TestGetSslVersionLegacyEnabledClient10Server10) {
1618 ConfigureClient("WebRTC-LegacyTlsProtocols/Enabled/");
1619 ConfigureServer("WebRTC-LegacyTlsProtocols/Enabled/");
1620 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
1621 TestHandshake();
1622}
1623
1624// Legacy protocols are disabled, max TLS version is 1.0
1625// This should be a configuration error, and handshake should fail.
1626TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1627 TestGetSslVersionLegacyDisabledClient10Server10) {
1628 ConfigureClient("WebRTC-LegacyTlsProtocols/Disabled/");
1629 ConfigureServer("WebRTC-LegacyTlsProtocols/Disabled/");
1630 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
1631 TestHandshake(false);
1632}