blob: e0ddafcec2deae735339bd499102f4e06ed02e63 [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
Steve Anton10542f22019-01-11 09:11:00 -080016#include "rtc_base/buffer_queue.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020017#include "rtc_base/checks.h"
18#include "rtc_base/gunit.h"
19#include "rtc_base/helpers.h"
Niels Möller13339482019-03-28 13:30:15 +010020#include "rtc_base/memory/fifo_buffer.h"
Niels Möllere7547d52018-11-01 09:33:08 +010021#include "rtc_base/memory_stream.h"
Steve Anton10542f22019-01-11 09:11:00 -080022#include "rtc_base/message_digest.h"
23#include "rtc_base/ssl_adapter.h"
24#include "rtc_base/ssl_identity.h"
25#include "rtc_base/ssl_stream_adapter.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020026#include "rtc_base/stream.h"
Harald Alvestrand13799132020-03-09 19:39:36 +010027#include "test/field_trial.h"
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000028
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +020029using ::testing::Combine;
30using ::testing::tuple;
Harald Alvestrand13799132020-03-09 19:39:36 +010031using ::testing::Values;
32using ::testing::WithParamInterface;
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +020033
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000034static const int kBlockSize = 4096;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000035static const char kExporterLabel[] = "label";
36static const unsigned char kExporterContext[] = "context";
37static int kExporterContextLen = sizeof(kExporterContext);
38
Harald Alvestrand13799132020-03-09 19:39:36 +010039// A private key used for testing, broken into pieces in order to avoid
40// issues with Git's checks for private keys in repos.
41#define RSA_PRIVATE_KEY_HEADER "-----BEGIN RSA PRIVATE KEY-----\n"
42
43static const char kRSA_PRIVATE_KEY_PEM[] = RSA_PRIVATE_KEY_HEADER
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000044 "MIICdwIBADANBgkqhkiG9w0BAQEFAASCAmEwggJdAgEAAoGBAMYRkbhmI7kVA/rM\n"
45 "czsZ+6JDhDvnkF+vn6yCAGuRPV03zuRqZtDy4N4to7PZu9PjqrRl7nDMXrG3YG9y\n"
46 "rlIAZ72KjcKKFAJxQyAKLCIdawKRyp8RdK3LEySWEZb0AV58IadqPZDTNHHRX8dz\n"
47 "5aTSMsbbkZ+C/OzTnbiMqLL/vg6jAgMBAAECgYAvgOs4FJcgvp+TuREx7YtiYVsH\n"
48 "mwQPTum2z/8VzWGwR8BBHBvIpVe1MbD/Y4seyI2aco/7UaisatSgJhsU46/9Y4fq\n"
49 "2TwXH9QANf4at4d9n/R6rzwpAJOpgwZgKvdQjkfrKTtgLV+/dawvpxUYkRH4JZM1\n"
50 "CVGukMfKNrSVH4Ap4QJBAOJmGV1ASPnB4r4nc99at7JuIJmd7fmuVUwUgYi4XgaR\n"
51 "WhScBsgYwZ/JoywdyZJgnbcrTDuVcWG56B3vXbhdpMsCQQDf9zeJrjnPZ3Cqm79y\n"
52 "kdqANep0uwZciiNiWxsQrCHztywOvbFhdp8iYVFG9EK8DMY41Y5TxUwsHD+67zao\n"
53 "ZNqJAkEA1suLUP/GvL8IwuRneQd2tWDqqRQ/Td3qq03hP7e77XtF/buya3Ghclo5\n"
54 "54czUR89QyVfJEC6278nzA7n2h1uVQJAcG6mztNL6ja/dKZjYZye2CY44QjSlLo0\n"
55 "MTgTSjdfg/28fFn2Jjtqf9Pi/X+50LWI/RcYMC2no606wRk9kyOuIQJBAK6VSAim\n"
56 "1pOEjsYQn0X5KEIrz1G3bfCbB848Ime3U2/FWlCHMr6ch8kCZ5d1WUeJD3LbwMNG\n"
57 "UCXiYxSsu20QNVw=\n"
58 "-----END RSA PRIVATE KEY-----\n";
59
Harald Alvestrand13799132020-03-09 19:39:36 +010060#undef RSA_PRIVATE_KEY_HEADER
61
henrike@webrtc.orgf0488722014-05-13 18:00:26 +000062static const char kCERT_PEM[] =
63 "-----BEGIN CERTIFICATE-----\n"
64 "MIIBmTCCAQKgAwIBAgIEbzBSAjANBgkqhkiG9w0BAQsFADARMQ8wDQYDVQQDEwZX\n"
65 "ZWJSVEMwHhcNMTQwMTAyMTgyNDQ3WhcNMTQwMjAxMTgyNDQ3WjARMQ8wDQYDVQQD\n"
66 "EwZXZWJSVEMwgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAMYRkbhmI7kVA/rM\n"
67 "czsZ+6JDhDvnkF+vn6yCAGuRPV03zuRqZtDy4N4to7PZu9PjqrRl7nDMXrG3YG9y\n"
68 "rlIAZ72KjcKKFAJxQyAKLCIdawKRyp8RdK3LEySWEZb0AV58IadqPZDTNHHRX8dz\n"
69 "5aTSMsbbkZ+C/OzTnbiMqLL/vg6jAgMBAAEwDQYJKoZIhvcNAQELBQADgYEAUflI\n"
70 "VUe5Krqf5RVa5C3u/UTAOAUJBiDS3VANTCLBxjuMsvqOG0WvaYWP3HYPgrz0jXK2\n"
71 "LJE/mGw3MyFHEqi81jh95J+ypl6xKW6Rm8jKLR87gUvCaVYn/Z4/P3AqcQTB7wOv\n"
72 "UD0A8qfhfDM+LK6rPAnCsVN0NRDY3jvd6rzix9M=\n"
73 "-----END CERTIFICATE-----\n";
74
Jian Cui0a8798b2017-11-16 16:58:02 -080075static const char kIntCert1[] =
76 "-----BEGIN CERTIFICATE-----\n"
77 "MIIEUjCCAjqgAwIBAgIBAjANBgkqhkiG9w0BAQsFADCBljELMAkGA1UEBhMCVVMx\n"
78 "EzARBgNVBAgMCkNhbGlmb3JuaWExFjAUBgNVBAcMDU1vdW50YWluIFZpZXcxFDAS\n"
79 "BgNVBAoMC0dvb2dsZSwgSW5jMQwwCgYDVQQLDANHVFAxFzAVBgNVBAMMDnRlbGVw\n"
80 "aG9ueS5nb29nMR0wGwYJKoZIhvcNAQkBFg5ndHBAZ29vZ2xlLmNvbTAeFw0xNzA5\n"
81 "MjYwNDA5MDNaFw0yMDA2MjIwNDA5MDNaMGQxCzAJBgNVBAYTAlVTMQswCQYDVQQI\n"
82 "DAJDQTEWMBQGA1UEBwwNTW91bnRhaW4gVmlldzEXMBUGA1UECgwOdGVsZXBob255\n"
83 "Lmdvb2cxFzAVBgNVBAMMDnRlbGVwaG9ueS5nb29nMIGfMA0GCSqGSIb3DQEBAQUA\n"
84 "A4GNADCBiQKBgQDJXWeeU1v1+wlqkVobzI3aN7Uh2iVQA9YCdq5suuabtiD/qoOD\n"
85 "NKpmQqsx7WZGGWSZTDFEBaUpvIK7Hb+nzRqk6iioPCFOFuarm6GxO1xVneImMuE6\n"
86 "tuWb3YZPr+ikChJbl11y5UcSbg0QsbeUc+jHl5umNvrL85Y+z8SP0rxbBwIDAQAB\n"
87 "o2AwXjAdBgNVHQ4EFgQU7tdZobqlN8R8V72FQnRxmqq8tKswHwYDVR0jBBgwFoAU\n"
88 "5GgKMUtcxkQ2dJrtNR5YOlIAPDswDwYDVR0TAQH/BAUwAwEB/zALBgNVHQ8EBAMC\n"
89 "AQYwDQYJKoZIhvcNAQELBQADggIBADObh9Z+z14FmP9zSenhFtq7hFnmNrSkklk8\n"
90 "eyYWXKfOuIriEQQBZsz76ZcnzStih8Rj+yQ0AXydk4fJ5LOwC2cUqQBar17g6Pd2\n"
91 "8g4SIL4azR9WvtiSvpuGlwp25b+yunaacDne6ebnf/MUiiKT5w61Xo3cEPVfl38e\n"
92 "/Up2l0bioid5enUTmg6LY6RxDO6tnZQkz3XD+nNSwT4ehtkqFpHYWjErj0BbkDM2\n"
93 "hiVc/JsYOZn3DmuOlHVHU6sKwqh3JEyvHO/d7DGzMGWHpHwv2mCTJq6l/sR95Tc2\n"
94 "GaQZgGDVNs9pdEouJCDm9e/PbQWRYhnat82PTkXx/6mDAAwdZlIi/pACzq8K4p7e\n"
95 "6hF0t8uKGnXJubHPXxlnJU6yxZ0yWmivAGjwWK4ur832gKlho4jeMDhiI/T3QPpl\n"
96 "iMNsIvxRhdD+GxJkQP1ezayw8s+Uc9KwKglrkBSRRDLCJUfPOvMmXLUDSTMX7kp4\n"
97 "/Ak1CA8dVLJIlfEjLBUuvAttlP7+7lsKNgxAjCxZkWLXIyGULzNPQwVWkGfCbrQs\n"
98 "XyMvSbFsSIb7blV7eLlmf9a+2RprUUkc2ALXLLCI9YQXmxm2beBfMyNmmebwBJzT\n"
99 "B0OR+5pFFNTJPoNlqpdrDsGrDu7JlUtk0ZLZzYyKXbgy2qXxfd4OWzXXjxpLMszZ\n"
100 "LDIpOAkj\n"
101 "-----END CERTIFICATE-----\n";
102
103static const char kCACert[] =
104 "-----BEGIN CERTIFICATE-----\n"
105 "MIIGETCCA/mgAwIBAgIJAKN9r/BdbGUJMA0GCSqGSIb3DQEBCwUAMIGWMQswCQYD\n"
106 "VQQGEwJVUzETMBEGA1UECAwKQ2FsaWZvcm5pYTEWMBQGA1UEBwwNTW91bnRhaW4g\n"
107 "VmlldzEUMBIGA1UECgwLR29vZ2xlLCBJbmMxDDAKBgNVBAsMA0dUUDEXMBUGA1UE\n"
108 "AwwOdGVsZXBob255Lmdvb2cxHTAbBgkqhkiG9w0BCQEWDmd0cEBnb29nbGUuY29t\n"
109 "MB4XDTE3MDcyNzIzMDE0NVoXDTE3MDgyNjIzMDE0NVowgZYxCzAJBgNVBAYTAlVT\n"
110 "MRMwEQYDVQQIDApDYWxpZm9ybmlhMRYwFAYDVQQHDA1Nb3VudGFpbiBWaWV3MRQw\n"
111 "EgYDVQQKDAtHb29nbGUsIEluYzEMMAoGA1UECwwDR1RQMRcwFQYDVQQDDA50ZWxl\n"
112 "cGhvbnkuZ29vZzEdMBsGCSqGSIb3DQEJARYOZ3RwQGdvb2dsZS5jb20wggIiMA0G\n"
113 "CSqGSIb3DQEBAQUAA4ICDwAwggIKAoICAQCfvpF7aBV5Hp1EHsWoIlL3GeHwh8dS\n"
114 "lv9VQCegN9rD06Ny7MgcED5AiK2vqXmUmOVS+7NbATkdVYN/eozDhKtN3Q3n87kJ\n"
115 "Nt/TD/TcZZHOZIGsRPbrf2URK26E/5KzTzbzXVBOA1e+gSj+EBbltGqb01ZO5ErF\n"
116 "iPGViPM/HpYKdq6mfz2bS5PhU67XZMM2zvToyReQ/Fjm/6PJhwKSRXSgZF5djPhk\n"
117 "2LfOKMLS0AeZtd2C4DFsCU41lfLUkybioDgFuzTQ3TFi1K8A07KYTMmLY/yQppnf\n"
118 "SpNX58shlVhM+Ed37K1Z0rU0OfVCZ5P+KKaSSfMranjlU7zeUIhZYjqq/EYrEhbS\n"
119 "dLnNHwgJrqxzId3kq8uuLM6+VB7JZKnZLfT90GdAbX4+tutNe21smmogF9f80vEy\n"
120 "gM4tOp9rXrvz9vCwWHXVY9kdKemdLAsREoO6MS9k2ctK4jj80o2dROuFC6Q3e7mz\n"
121 "RjvZr5Tvi464c2o9o/jNlJ0O6q7V2eQzohD+7VnV5QPpRGXxlIeqpR2zoAg+WtRS\n"
122 "4OgHOVYiD3M6uAlggJA5pcDjMfkEZ+pkhtVcT4qMCEoruk6GbyPxS565oSHu16bH\n"
123 "EjeCqbZOVND5T3oA7nz6aQSs8sJabt0jmxUkGVnE+4ZDIuuRtkRma+0P/96Mtqor\n"
124 "OlpNWY1OBDY64QIDAQABo2AwXjAdBgNVHQ4EFgQU5GgKMUtcxkQ2dJrtNR5YOlIA\n"
125 "PDswHwYDVR0jBBgwFoAU5GgKMUtcxkQ2dJrtNR5YOlIAPDswDwYDVR0TAQH/BAUw\n"
126 "AwEB/zALBgNVHQ8EBAMCAQYwDQYJKoZIhvcNAQELBQADggIBAARQly5/bB6VUL2C\n"
127 "ykDYgWt48go407pAra6tL2kjpdfxV5PdL7iMZRkeht00vj+BVahIqZKrNOa/f5Fx\n"
128 "vlpahZFu0PDN436aQwRZ9qWut2qDOK0/z9Hhj6NWybquRFwMwqkPG/ivLMDU8Dmj\n"
129 "CIplpngPYNwXCs0KzdjSXYxqxJbwMjQXELD+/RcurY0oTtJMM1/2vKQMzw24UJqe\n"
130 "XLJAlsnd2AnWzWNUEviDZY89j9NdkHerBmV2gGzcU+X5lgOO5M8odBv0ZC9D+a6Z\n"
131 "QPZAOfdGVw60hhGvTW5s/s0dHwCpegRidhs0MD0fTmwwjYFBSmUx3Gztr4JTzOOr\n"
132 "7e5daJuak2ujQ5DqcGBvt1gePjSudb5brS7JQtN8tI/FyrnR4q/OuOwv1EvlC5RG\n"
133 "hLX+TXaWqFxB1Hd8ebKRR40mboFG6KcUI3lLBthDvQE7jnq48QfZMjlMQK0ZF1l7\n"
134 "SrlwRXWA74bU8CLJvnZKKo9p4TsTiDYGSYC6tNHKj5s3TGWL46oqGyZ0KdGNhrtC\n"
135 "rIGenMhth1vPYjyy0XuGBndXT85yi+IM2l8g8oU845+plxIhgpSI8bbC0oLwnhQ5\n"
136 "ARfsiYLkXDE7imSS0CSUmye76372mlzAIB1is4bBB/SzpPQtBuB9LDKtONgpSGHn\n"
137 "dGaXBy+qbVXVyGXaeEbIRjtJ6m92\n"
138 "-----END CERTIFICATE-----\n";
139
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000140class SSLStreamAdapterTestBase;
141
jbauche488a0d2015-11-19 05:17:58 -0800142class SSLDummyStreamBase : public rtc::StreamInterface,
143 public sigslot::has_slots<> {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000144 public:
jbauche488a0d2015-11-19 05:17:58 -0800145 SSLDummyStreamBase(SSLStreamAdapterTestBase* test,
Yves Gerey665174f2018-06-19 15:03:05 +0200146 const std::string& side,
jbauche488a0d2015-11-19 05:17:58 -0800147 rtc::StreamInterface* in,
Yves Gerey665174f2018-06-19 15:03:05 +0200148 rtc::StreamInterface* out)
149 : test_base_(test), side_(side), in_(in), out_(out), first_packet_(true) {
jbauche488a0d2015-11-19 05:17:58 -0800150 in_->SignalEvent.connect(this, &SSLDummyStreamBase::OnEventIn);
151 out_->SignalEvent.connect(this, &SSLDummyStreamBase::OnEventOut);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000152 }
153
jbauche488a0d2015-11-19 05:17:58 -0800154 rtc::StreamState GetState() const override { return rtc::SS_OPEN; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000155
Yves Gerey665174f2018-06-19 15:03:05 +0200156 rtc::StreamResult Read(void* buffer,
157 size_t buffer_len,
158 size_t* read,
159 int* error) override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000160 rtc::StreamResult r;
161
162 r = in_->Read(buffer, buffer_len, read, error);
163 if (r == rtc::SR_BLOCK)
164 return rtc::SR_BLOCK;
165 if (r == rtc::SR_EOS)
166 return rtc::SR_EOS;
167
168 if (r != rtc::SR_SUCCESS) {
169 ADD_FAILURE();
170 return rtc::SR_ERROR;
171 }
172
173 return rtc::SR_SUCCESS;
174 }
175
176 // Catch readability events on in and pass them up.
jbauche488a0d2015-11-19 05:17:58 -0800177 void OnEventIn(rtc::StreamInterface* stream, int sig, int err) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000178 int mask = (rtc::SE_READ | rtc::SE_CLOSE);
179
180 if (sig & mask) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100181 RTC_LOG(LS_VERBOSE) << "SSLDummyStreamBase::OnEvent side=" << side_
182 << " sig=" << sig << " forwarding upward";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000183 PostEvent(sig & mask, 0);
184 }
185 }
186
187 // Catch writeability events on out and pass them up.
jbauche488a0d2015-11-19 05:17:58 -0800188 void OnEventOut(rtc::StreamInterface* stream, int sig, int err) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000189 if (sig & rtc::SE_WRITE) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100190 RTC_LOG(LS_VERBOSE) << "SSLDummyStreamBase::OnEvent side=" << side_
191 << " sig=" << sig << " forwarding upward";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000192
193 PostEvent(sig & rtc::SE_WRITE, 0);
194 }
195 }
196
197 // Write to the outgoing FifoBuffer
Yves Gerey665174f2018-06-19 15:03:05 +0200198 rtc::StreamResult WriteData(const void* data,
199 size_t data_len,
200 size_t* written,
201 int* error) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000202 return out_->Write(data, data_len, written, error);
203 }
204
Yves Gerey665174f2018-06-19 15:03:05 +0200205 rtc::StreamResult Write(const void* data,
206 size_t data_len,
207 size_t* written,
208 int* error) override;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000209
jbauche488a0d2015-11-19 05:17:58 -0800210 void Close() override {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100211 RTC_LOG(LS_INFO) << "Closing outbound stream";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000212 out_->Close();
213 }
214
jbauche488a0d2015-11-19 05:17:58 -0800215 protected:
216 SSLStreamAdapterTestBase* test_base_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000217 const std::string side_;
jbauche488a0d2015-11-19 05:17:58 -0800218 rtc::StreamInterface* in_;
219 rtc::StreamInterface* out_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000220 bool first_packet_;
221};
222
jbauche488a0d2015-11-19 05:17:58 -0800223class SSLDummyStreamTLS : public SSLDummyStreamBase {
224 public:
225 SSLDummyStreamTLS(SSLStreamAdapterTestBase* test,
226 const std::string& side,
227 rtc::FifoBuffer* in,
Yves Gerey665174f2018-06-19 15:03:05 +0200228 rtc::FifoBuffer* out)
229 : SSLDummyStreamBase(test, side, in, out) {}
jbauche488a0d2015-11-19 05:17:58 -0800230};
231
Yves Gerey665174f2018-06-19 15:03:05 +0200232class BufferQueueStream : public rtc::BufferQueue, public rtc::StreamInterface {
jbauche488a0d2015-11-19 05:17:58 -0800233 public:
234 BufferQueueStream(size_t capacity, size_t default_size)
Yves Gerey665174f2018-06-19 15:03:05 +0200235 : rtc::BufferQueue(capacity, default_size) {}
jbauche488a0d2015-11-19 05:17:58 -0800236
237 // Implementation of abstract StreamInterface methods.
238
239 // A buffer queue stream is always "open".
240 rtc::StreamState GetState() const override { return rtc::SS_OPEN; }
241
242 // Reading a buffer queue stream will either succeed or block.
Yves Gerey665174f2018-06-19 15:03:05 +0200243 rtc::StreamResult Read(void* buffer,
244 size_t buffer_len,
245 size_t* read,
246 int* error) override {
jbauche488a0d2015-11-19 05:17:58 -0800247 if (!ReadFront(buffer, buffer_len, read)) {
248 return rtc::SR_BLOCK;
249 }
250 return rtc::SR_SUCCESS;
251 }
252
253 // Writing to a buffer queue stream will either succeed or block.
Yves Gerey665174f2018-06-19 15:03:05 +0200254 rtc::StreamResult Write(const void* data,
255 size_t data_len,
256 size_t* written,
257 int* error) override {
jbauche488a0d2015-11-19 05:17:58 -0800258 if (!WriteBack(data, data_len, written)) {
259 return rtc::SR_BLOCK;
260 }
261 return rtc::SR_SUCCESS;
262 }
263
264 // A buffer queue stream can not be closed.
265 void Close() override {}
266
267 protected:
Yves Gerey665174f2018-06-19 15:03:05 +0200268 void NotifyReadableForTest() override { PostEvent(rtc::SE_READ, 0); }
jbauche488a0d2015-11-19 05:17:58 -0800269
Yves Gerey665174f2018-06-19 15:03:05 +0200270 void NotifyWritableForTest() override { PostEvent(rtc::SE_WRITE, 0); }
jbauche488a0d2015-11-19 05:17:58 -0800271};
272
273class SSLDummyStreamDTLS : public SSLDummyStreamBase {
274 public:
275 SSLDummyStreamDTLS(SSLStreamAdapterTestBase* test,
276 const std::string& side,
277 BufferQueueStream* in,
Yves Gerey665174f2018-06-19 15:03:05 +0200278 BufferQueueStream* out)
279 : SSLDummyStreamBase(test, side, in, out) {}
jbauche488a0d2015-11-19 05:17:58 -0800280};
281
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000282static const int kFifoBufferSize = 4096;
jbauche488a0d2015-11-19 05:17:58 -0800283static const int kBufferCapacity = 1;
284static const size_t kDefaultBufferSize = 2048;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000285
Mirko Bonadei6a489f22019-04-09 15:11:12 +0200286class SSLStreamAdapterTestBase : public ::testing::Test,
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000287 public sigslot::has_slots<> {
288 public:
torbjorng4e572472015-10-08 09:42:49 -0700289 SSLStreamAdapterTestBase(
290 const std::string& client_cert_pem,
291 const std::string& client_private_key_pem,
292 bool dtls,
293 rtc::KeyParams client_key_type = rtc::KeyParams(rtc::KT_DEFAULT),
294 rtc::KeyParams server_key_type = rtc::KeyParams(rtc::KT_DEFAULT))
jbauche488a0d2015-11-19 05:17:58 -0800295 : client_cert_pem_(client_cert_pem),
296 client_private_key_pem_(client_private_key_pem),
297 client_key_type_(client_key_type),
298 server_key_type_(server_key_type),
deadbeef37f5ecf2017-02-27 14:06:41 -0800299 client_stream_(nullptr),
300 server_stream_(nullptr),
301 client_identity_(nullptr),
302 server_identity_(nullptr),
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200303 delay_(0),
304 mtu_(1460),
305 loss_(0),
306 lose_first_packet_(false),
307 damage_(false),
308 dtls_(dtls),
309 handshake_wait_(5000),
Benjamin Wrightaf1f8652019-04-01 18:25:23 +0000310 identities_set_(false) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000311 // Set use of the test RNG to get predictable loss patterns.
312 rtc::SetRandomTestMode(true);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000313 }
314
Steve Anton9de3aac2017-10-24 10:08:26 -0700315 ~SSLStreamAdapterTestBase() override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000316 // Put it back for the next test.
317 rtc::SetRandomTestMode(false);
318 }
319
torbjorng7593aad2015-11-19 12:20:51 -0800320 void SetUp() override {
jbauche488a0d2015-11-19 05:17:58 -0800321 CreateStreams();
322
323 client_ssl_.reset(rtc::SSLStreamAdapter::Create(client_stream_));
324 server_ssl_.reset(rtc::SSLStreamAdapter::Create(server_stream_));
325
326 // Set up the slots
327 client_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
328 server_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
329
330 if (!client_cert_pem_.empty() && !client_private_key_pem_.empty()) {
331 client_identity_ = rtc::SSLIdentity::FromPEMStrings(
332 client_private_key_pem_, client_cert_pem_);
333 } else {
334 client_identity_ = rtc::SSLIdentity::Generate("client", client_key_type_);
335 }
336 server_identity_ = rtc::SSLIdentity::Generate("server", server_key_type_);
337
338 client_ssl_->SetIdentity(client_identity_);
339 server_ssl_->SetIdentity(server_identity_);
340 }
341
torbjorng7593aad2015-11-19 12:20:51 -0800342 void TearDown() override {
jbauche488a0d2015-11-19 05:17:58 -0800343 client_ssl_.reset(nullptr);
344 server_ssl_.reset(nullptr);
345 }
346
347 virtual void CreateStreams() = 0;
348
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000349 // Recreate the client/server identities with the specified validity period.
350 // |not_before| and |not_after| are offsets from the current time in number
351 // of seconds.
352 void ResetIdentitiesWithValidity(int not_before, int not_after) {
jbauche488a0d2015-11-19 05:17:58 -0800353 CreateStreams();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000354
355 client_ssl_.reset(rtc::SSLStreamAdapter::Create(client_stream_));
356 server_ssl_.reset(rtc::SSLStreamAdapter::Create(server_stream_));
357
358 client_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
359 server_ssl_->SignalEvent.connect(this, &SSLStreamAdapterTestBase::OnEvent);
360
Torbjorn Granlund46c9cc02015-12-01 13:06:34 +0100361 time_t now = time(nullptr);
362
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000363 rtc::SSLIdentityParams client_params;
torbjorng4e572472015-10-08 09:42:49 -0700364 client_params.key_params = rtc::KeyParams(rtc::KT_DEFAULT);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000365 client_params.common_name = "client";
Torbjorn Granlund46c9cc02015-12-01 13:06:34 +0100366 client_params.not_before = now + not_before;
367 client_params.not_after = now + not_after;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000368 client_identity_ = rtc::SSLIdentity::GenerateForTest(client_params);
369
370 rtc::SSLIdentityParams server_params;
torbjorng4e572472015-10-08 09:42:49 -0700371 server_params.key_params = rtc::KeyParams(rtc::KT_DEFAULT);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000372 server_params.common_name = "server";
Torbjorn Granlund46c9cc02015-12-01 13:06:34 +0100373 server_params.not_before = now + not_before;
374 server_params.not_after = now + not_after;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000375 server_identity_ = rtc::SSLIdentity::GenerateForTest(server_params);
376
377 client_ssl_->SetIdentity(client_identity_);
378 server_ssl_->SetIdentity(server_identity_);
379 }
380
Yves Gerey665174f2018-06-19 15:03:05 +0200381 virtual void OnEvent(rtc::StreamInterface* stream, int sig, int err) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100382 RTC_LOG(LS_VERBOSE) << "SSLStreamAdapterTestBase::OnEvent sig=" << sig;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000383
384 if (sig & rtc::SE_READ) {
385 ReadData(stream);
386 }
387
388 if ((stream == client_ssl_.get()) && (sig & rtc::SE_WRITE)) {
389 WriteData();
390 }
391 }
392
deadbeef89824f62016-09-30 11:55:43 -0700393 void SetPeerIdentitiesByDigest(bool correct, bool expect_success) {
394 unsigned char server_digest[20];
395 size_t server_digest_len;
396 unsigned char client_digest[20];
397 size_t client_digest_len;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000398 bool rv;
deadbeef89824f62016-09-30 11:55:43 -0700399 rtc::SSLPeerCertificateDigestError err;
400 rtc::SSLPeerCertificateDigestError expected_err =
401 expect_success
402 ? rtc::SSLPeerCertificateDigestError::NONE
403 : rtc::SSLPeerCertificateDigestError::VERIFICATION_FAILED;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000404
Mirko Bonadei675513b2017-11-09 11:09:25 +0100405 RTC_LOG(LS_INFO) << "Setting peer identities by digest";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000406
deadbeef89824f62016-09-30 11:55:43 -0700407 rv = server_identity_->certificate().ComputeDigest(
408 rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000409 ASSERT_TRUE(rv);
deadbeef89824f62016-09-30 11:55:43 -0700410 rv = client_identity_->certificate().ComputeDigest(
411 rtc::DIGEST_SHA_1, client_digest, 20, &client_digest_len);
412 ASSERT_TRUE(rv);
413
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000414 if (!correct) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100415 RTC_LOG(LS_INFO) << "Setting bogus digest for server cert";
deadbeef89824f62016-09-30 11:55:43 -0700416 server_digest[0]++;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000417 }
deadbeef89824f62016-09-30 11:55:43 -0700418 rv = client_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, server_digest,
419 server_digest_len, &err);
420 EXPECT_EQ(expected_err, err);
421 EXPECT_EQ(expect_success, rv);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000422
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000423 if (!correct) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100424 RTC_LOG(LS_INFO) << "Setting bogus digest for client cert";
deadbeef89824f62016-09-30 11:55:43 -0700425 client_digest[0]++;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000426 }
deadbeef89824f62016-09-30 11:55:43 -0700427 rv = server_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, client_digest,
428 client_digest_len, &err);
429 EXPECT_EQ(expected_err, err);
430 EXPECT_EQ(expect_success, rv);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000431
432 identities_set_ = true;
433 }
434
Joachim Bauch831c5582015-05-20 12:48:41 +0200435 void SetupProtocolVersions(rtc::SSLProtocolVersion server_version,
436 rtc::SSLProtocolVersion client_version) {
437 server_ssl_->SetMaxProtocolVersion(server_version);
438 client_ssl_->SetMaxProtocolVersion(client_version);
439 }
440
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000441 void TestHandshake(bool expect_success = true) {
Yves Gerey665174f2018-06-19 15:03:05 +0200442 server_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
443 client_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000444
445 if (!dtls_) {
446 // Make sure we simulate a reliable network for TLS.
447 // This is just a check to make sure that people don't write wrong
448 // tests.
nissec16fa5e2017-02-07 07:18:43 -0800449 RTC_CHECK_EQ(1460, mtu_);
450 RTC_CHECK(!loss_);
451 RTC_CHECK(!lose_first_packet_);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000452 }
453
454 if (!identities_set_)
deadbeef89824f62016-09-30 11:55:43 -0700455 SetPeerIdentitiesByDigest(true, true);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000456
457 // Start the handshake
458 int rv;
459
460 server_ssl_->SetServerRole();
Taylor Brandstetterc8762a82016-08-11 12:01:49 -0700461 rv = server_ssl_->StartSSL();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000462 ASSERT_EQ(0, rv);
463
Taylor Brandstetterc8762a82016-08-11 12:01:49 -0700464 rv = client_ssl_->StartSSL();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000465 ASSERT_EQ(0, rv);
466
467 // Now run the handshake
468 if (expect_success) {
Yves Gerey665174f2018-06-19 15:03:05 +0200469 EXPECT_TRUE_WAIT((client_ssl_->GetState() == rtc::SS_OPEN) &&
470 (server_ssl_->GetState() == rtc::SS_OPEN),
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000471 handshake_wait_);
472 } else {
473 EXPECT_TRUE_WAIT(client_ssl_->GetState() == rtc::SS_CLOSED,
474 handshake_wait_);
475 }
476 }
477
deadbeef89824f62016-09-30 11:55:43 -0700478 // This tests that the handshake can complete before the identity is
479 // verified, and the identity will be verified after the fact.
480 void TestHandshakeWithDelayedIdentity(bool valid_identity) {
481 server_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
482 client_ssl_->SetMode(dtls_ ? rtc::SSL_MODE_DTLS : rtc::SSL_MODE_TLS);
483
484 if (!dtls_) {
485 // Make sure we simulate a reliable network for TLS.
486 // This is just a check to make sure that people don't write wrong
487 // tests.
nissec16fa5e2017-02-07 07:18:43 -0800488 RTC_CHECK_EQ(1460, mtu_);
489 RTC_CHECK(!loss_);
490 RTC_CHECK(!lose_first_packet_);
deadbeef89824f62016-09-30 11:55:43 -0700491 }
492
493 // Start the handshake
494 int rv;
495
496 server_ssl_->SetServerRole();
497 rv = server_ssl_->StartSSL();
498 ASSERT_EQ(0, rv);
499
500 rv = client_ssl_->StartSSL();
501 ASSERT_EQ(0, rv);
502
503 // Now run the handshake.
504 EXPECT_TRUE_WAIT(
505 client_ssl_->IsTlsConnected() && server_ssl_->IsTlsConnected(),
506 handshake_wait_);
507
508 // Until the identity has been verified, the state should still be
509 // SS_OPENING and writes should return SR_BLOCK.
510 EXPECT_EQ(rtc::SS_OPENING, client_ssl_->GetState());
511 EXPECT_EQ(rtc::SS_OPENING, server_ssl_->GetState());
512 unsigned char packet[1];
513 size_t sent;
514 EXPECT_EQ(rtc::SR_BLOCK, client_ssl_->Write(&packet, 1, &sent, 0));
515 EXPECT_EQ(rtc::SR_BLOCK, server_ssl_->Write(&packet, 1, &sent, 0));
516
517 // If we set an invalid identity at this point, SetPeerCertificateDigest
518 // should return false.
519 SetPeerIdentitiesByDigest(valid_identity, valid_identity);
520 // State should then transition to SS_OPEN or SS_CLOSED based on validation
521 // of the identity.
522 if (valid_identity) {
523 EXPECT_EQ(rtc::SS_OPEN, client_ssl_->GetState());
524 EXPECT_EQ(rtc::SS_OPEN, server_ssl_->GetState());
525 } else {
526 EXPECT_EQ(rtc::SS_CLOSED, client_ssl_->GetState());
527 EXPECT_EQ(rtc::SS_CLOSED, server_ssl_->GetState());
528 }
529 }
530
Yves Gerey665174f2018-06-19 15:03:05 +0200531 rtc::StreamResult DataWritten(SSLDummyStreamBase* from,
532 const void* data,
533 size_t data_len,
534 size_t* written,
535 int* error) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000536 // Randomly drop loss_ percent of packets
Peter Boström0c4e06b2015-10-07 12:23:21 +0200537 if (rtc::CreateRandomId() % 100 < static_cast<uint32_t>(loss_)) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100538 RTC_LOG(LS_VERBOSE) << "Randomly dropping packet, size=" << data_len;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000539 *written = data_len;
540 return rtc::SR_SUCCESS;
541 }
542 if (dtls_ && (data_len > mtu_)) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100543 RTC_LOG(LS_VERBOSE) << "Dropping packet > mtu, size=" << data_len;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000544 *written = data_len;
545 return rtc::SR_SUCCESS;
546 }
547
548 // Optionally damage application data (type 23). Note that we don't damage
549 // handshake packets and we damage the last byte to keep the header
550 // intact but break the MAC.
Yves Gerey665174f2018-06-19 15:03:05 +0200551 if (damage_ && (*static_cast<const unsigned char*>(data) == 23)) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000552 std::vector<char> buf(data_len);
553
Mirko Bonadei675513b2017-11-09 11:09:25 +0100554 RTC_LOG(LS_VERBOSE) << "Damaging packet";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000555
556 memcpy(&buf[0], data, data_len);
557 buf[data_len - 1]++;
558
559 return from->WriteData(&buf[0], data_len, written, error);
560 }
561
562 return from->WriteData(data, data_len, written, error);
563 }
564
Yves Gerey665174f2018-06-19 15:03:05 +0200565 void SetDelay(int delay) { delay_ = delay; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000566 int GetDelay() { return delay_; }
567
Yves Gerey665174f2018-06-19 15:03:05 +0200568 void SetLoseFirstPacket(bool lose) { lose_first_packet_ = lose; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000569 bool GetLoseFirstPacket() { return lose_first_packet_; }
570
Yves Gerey665174f2018-06-19 15:03:05 +0200571 void SetLoss(int percent) { loss_ = percent; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000572
Yves Gerey665174f2018-06-19 15:03:05 +0200573 void SetDamage() { damage_ = true; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000574
Yves Gerey665174f2018-06-19 15:03:05 +0200575 void SetMtu(size_t mtu) { mtu_ = mtu; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000576
Yves Gerey665174f2018-06-19 15:03:05 +0200577 void SetHandshakeWait(int wait) { handshake_wait_ = wait; }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000578
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800579 void SetDtlsSrtpCryptoSuites(const std::vector<int>& ciphers, bool client) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000580 if (client)
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800581 client_ssl_->SetDtlsSrtpCryptoSuites(ciphers);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000582 else
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800583 server_ssl_->SetDtlsSrtpCryptoSuites(ciphers);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000584 }
585
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800586 bool GetDtlsSrtpCryptoSuite(bool client, int* retval) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000587 if (client)
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800588 return client_ssl_->GetDtlsSrtpCryptoSuite(retval);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000589 else
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -0800590 return server_ssl_->GetDtlsSrtpCryptoSuite(retval);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000591 }
592
jbauch555604a2016-04-26 03:13:22 -0700593 std::unique_ptr<rtc::SSLCertificate> GetPeerCertificate(bool client) {
Taylor Brandstetterc3928662018-02-23 13:04:51 -0800594 std::unique_ptr<rtc::SSLCertChain> chain;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000595 if (client)
Taylor Brandstetterc3928662018-02-23 13:04:51 -0800596 chain = client_ssl_->GetPeerSSLCertChain();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000597 else
Taylor Brandstetterc3928662018-02-23 13:04:51 -0800598 chain = server_ssl_->GetPeerSSLCertChain();
Steve Antonf25303e2018-10-16 15:23:31 -0700599 return (chain && chain->GetSize()) ? chain->Get(0).Clone() : nullptr;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000600 }
601
Guo-wei Shieh6caafbe2015-10-05 12:43:27 -0700602 bool GetSslCipherSuite(bool client, int* retval) {
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +0000603 if (client)
Guo-wei Shieh456696a2015-09-30 21:48:54 -0700604 return client_ssl_->GetSslCipherSuite(retval);
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +0000605 else
Guo-wei Shieh456696a2015-09-30 21:48:54 -0700606 return server_ssl_->GetSslCipherSuite(retval);
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +0000607 }
608
torbjorng43166b82016-03-11 00:06:47 -0800609 int GetSslVersion(bool client) {
610 if (client)
611 return client_ssl_->GetSslVersion();
612 else
613 return server_ssl_->GetSslVersion();
614 }
615
Yves Gerey665174f2018-06-19 15:03:05 +0200616 bool ExportKeyingMaterial(const char* label,
617 const unsigned char* context,
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000618 size_t context_len,
619 bool use_context,
620 bool client,
Yves Gerey665174f2018-06-19 15:03:05 +0200621 unsigned char* result,
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000622 size_t result_len) {
623 if (client)
Yves Gerey665174f2018-06-19 15:03:05 +0200624 return client_ssl_->ExportKeyingMaterial(label, context, context_len,
625 use_context, result, result_len);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000626 else
Yves Gerey665174f2018-06-19 15:03:05 +0200627 return server_ssl_->ExportKeyingMaterial(label, context, context_len,
628 use_context, result, result_len);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000629 }
630
631 // To be implemented by subclasses.
632 virtual void WriteData() = 0;
Yves Gerey665174f2018-06-19 15:03:05 +0200633 virtual void ReadData(rtc::StreamInterface* stream) = 0;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000634 virtual void TestTransfer(int size) = 0;
635
636 protected:
jbauche488a0d2015-11-19 05:17:58 -0800637 std::string client_cert_pem_;
638 std::string client_private_key_pem_;
639 rtc::KeyParams client_key_type_;
640 rtc::KeyParams server_key_type_;
Yves Gerey665174f2018-06-19 15:03:05 +0200641 SSLDummyStreamBase* client_stream_; // freed by client_ssl_ destructor
642 SSLDummyStreamBase* server_stream_; // freed by server_ssl_ destructor
jbauch555604a2016-04-26 03:13:22 -0700643 std::unique_ptr<rtc::SSLStreamAdapter> client_ssl_;
644 std::unique_ptr<rtc::SSLStreamAdapter> server_ssl_;
Yves Gerey665174f2018-06-19 15:03:05 +0200645 rtc::SSLIdentity* client_identity_; // freed by client_ssl_ destructor
646 rtc::SSLIdentity* server_identity_; // freed by server_ssl_ destructor
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000647 int delay_;
648 size_t mtu_;
649 int loss_;
650 bool lose_first_packet_;
651 bool damage_;
652 bool dtls_;
653 int handshake_wait_;
654 bool identities_set_;
655};
656
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200657class SSLStreamAdapterTestTLS
658 : public SSLStreamAdapterTestBase,
torbjorng4e572472015-10-08 09:42:49 -0700659 public WithParamInterface<tuple<rtc::KeyParams, rtc::KeyParams>> {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000660 public:
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200661 SSLStreamAdapterTestTLS()
662 : SSLStreamAdapterTestBase("",
663 "",
664 false,
665 ::testing::get<0>(GetParam()),
jbauche488a0d2015-11-19 05:17:58 -0800666 ::testing::get<1>(GetParam())),
667 client_buffer_(kFifoBufferSize),
Yves Gerey665174f2018-06-19 15:03:05 +0200668 server_buffer_(kFifoBufferSize) {}
jbauche488a0d2015-11-19 05:17:58 -0800669
torbjorng7593aad2015-11-19 12:20:51 -0800670 void CreateStreams() override {
jbauche488a0d2015-11-19 05:17:58 -0800671 client_stream_ =
672 new SSLDummyStreamTLS(this, "c2s", &client_buffer_, &server_buffer_);
673 server_stream_ =
674 new SSLDummyStreamTLS(this, "s2c", &server_buffer_, &client_buffer_);
675 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000676
677 // Test data transfer for TLS
nisseef8b61e2016-04-29 06:09:15 -0700678 void TestTransfer(int size) override {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100679 RTC_LOG(LS_INFO) << "Starting transfer test with " << size << " bytes";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000680 // Create some dummy data to send.
681 size_t received;
682
683 send_stream_.ReserveSize(size);
684 for (int i = 0; i < size; ++i) {
685 char ch = static_cast<char>(i);
deadbeef37f5ecf2017-02-27 14:06:41 -0800686 send_stream_.Write(&ch, 1, nullptr, nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000687 }
688 send_stream_.Rewind();
689
690 // Prepare the receive stream.
691 recv_stream_.ReserveSize(size);
692
693 // Start sending
694 WriteData();
695
696 // Wait for the client to close
697 EXPECT_TRUE_WAIT(server_ssl_->GetState() == rtc::SS_CLOSED, 10000);
698
699 // Now check the data
700 recv_stream_.GetSize(&received);
701
702 EXPECT_EQ(static_cast<size_t>(size), received);
Yves Gerey665174f2018-06-19 15:03:05 +0200703 EXPECT_EQ(0,
704 memcmp(send_stream_.GetBuffer(), recv_stream_.GetBuffer(), size));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000705 }
706
nisseef8b61e2016-04-29 06:09:15 -0700707 void WriteData() override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000708 size_t position, tosend, size;
709 rtc::StreamResult rv;
710 size_t sent;
711 char block[kBlockSize];
712
713 send_stream_.GetSize(&size);
714 if (!size)
715 return;
716
717 for (;;) {
718 send_stream_.GetPosition(&position);
deadbeef37f5ecf2017-02-27 14:06:41 -0800719 if (send_stream_.Read(block, sizeof(block), &tosend, nullptr) !=
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000720 rtc::SR_EOS) {
721 rv = client_ssl_->Write(block, tosend, &sent, 0);
722
723 if (rv == rtc::SR_SUCCESS) {
724 send_stream_.SetPosition(position + sent);
Mirko Bonadei675513b2017-11-09 11:09:25 +0100725 RTC_LOG(LS_VERBOSE) << "Sent: " << position + sent;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000726 } else if (rv == rtc::SR_BLOCK) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100727 RTC_LOG(LS_VERBOSE) << "Blocked...";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000728 send_stream_.SetPosition(position);
729 break;
730 } else {
731 ADD_FAILURE();
732 break;
733 }
734 } else {
735 // Now close
Mirko Bonadei675513b2017-11-09 11:09:25 +0100736 RTC_LOG(LS_INFO) << "Wrote " << position << " bytes. Closing";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000737 client_ssl_->Close();
738 break;
739 }
740 }
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100741 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000742
Yves Gerey665174f2018-06-19 15:03:05 +0200743 void ReadData(rtc::StreamInterface* stream) override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000744 char buffer[1600];
745 size_t bread;
746 int err2;
747 rtc::StreamResult r;
748
749 for (;;) {
750 r = stream->Read(buffer, sizeof(buffer), &bread, &err2);
751
752 if (r == rtc::SR_ERROR || r == rtc::SR_EOS) {
753 // Unfortunately, errors are the way that the stream adapter
torbjorng7593aad2015-11-19 12:20:51 -0800754 // signals close in OpenSSL.
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000755 stream->Close();
756 return;
757 }
758
759 if (r == rtc::SR_BLOCK)
760 break;
761
762 ASSERT_EQ(rtc::SR_SUCCESS, r);
Mirko Bonadei675513b2017-11-09 11:09:25 +0100763 RTC_LOG(LS_VERBOSE) << "Read " << bread;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000764
deadbeef37f5ecf2017-02-27 14:06:41 -0800765 recv_stream_.Write(buffer, bread, nullptr, nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000766 }
767 }
768
769 private:
jbauche488a0d2015-11-19 05:17:58 -0800770 rtc::FifoBuffer client_buffer_;
771 rtc::FifoBuffer server_buffer_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000772 rtc::MemoryStream send_stream_;
773 rtc::MemoryStream recv_stream_;
774};
775
Harald Alvestrand13799132020-03-09 19:39:36 +0100776class SSLStreamAdapterTestDTLSBase : public SSLStreamAdapterTestBase {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000777 public:
Harald Alvestrand13799132020-03-09 19:39:36 +0100778 SSLStreamAdapterTestDTLSBase(rtc::KeyParams param1, rtc::KeyParams param2)
779 : SSLStreamAdapterTestBase("", "", true, param1, param2),
jbauche488a0d2015-11-19 05:17:58 -0800780 client_buffer_(kBufferCapacity, kDefaultBufferSize),
781 server_buffer_(kBufferCapacity, kDefaultBufferSize),
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200782 packet_size_(1000),
783 count_(0),
784 sent_(0) {}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000785
Harald Alvestrand13799132020-03-09 19:39:36 +0100786 SSLStreamAdapterTestDTLSBase(const std::string& cert_pem,
787 const std::string& private_key_pem)
Yves Gerey665174f2018-06-19 15:03:05 +0200788 : SSLStreamAdapterTestBase(cert_pem, private_key_pem, true),
789 client_buffer_(kBufferCapacity, kDefaultBufferSize),
790 server_buffer_(kBufferCapacity, kDefaultBufferSize),
791 packet_size_(1000),
792 count_(0),
793 sent_(0) {}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000794
torbjorng7593aad2015-11-19 12:20:51 -0800795 void CreateStreams() override {
jbauche488a0d2015-11-19 05:17:58 -0800796 client_stream_ =
797 new SSLDummyStreamDTLS(this, "c2s", &client_buffer_, &server_buffer_);
798 server_stream_ =
799 new SSLDummyStreamDTLS(this, "s2c", &server_buffer_, &client_buffer_);
800 }
801
nisseef8b61e2016-04-29 06:09:15 -0700802 void WriteData() override {
Yves Gerey665174f2018-06-19 15:03:05 +0200803 unsigned char* packet = new unsigned char[1600];
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000804
jbauche488a0d2015-11-19 05:17:58 -0800805 while (sent_ < count_) {
torbjorng7593aad2015-11-19 12:20:51 -0800806 unsigned int rand_state = sent_;
807 packet[0] = sent_;
808 for (size_t i = 1; i < packet_size_; i++) {
809 // This is a simple LC PRNG. Keep in synch with identical code below.
810 rand_state = (rand_state * 251 + 19937) >> 7;
811 packet[i] = rand_state & 0xff;
812 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000813
814 size_t sent;
torbjorng7593aad2015-11-19 12:20:51 -0800815 rtc::StreamResult rv = client_ssl_->Write(packet, packet_size_, &sent, 0);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000816 if (rv == rtc::SR_SUCCESS) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100817 RTC_LOG(LS_VERBOSE) << "Sent: " << sent_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000818 sent_++;
819 } else if (rv == rtc::SR_BLOCK) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100820 RTC_LOG(LS_VERBOSE) << "Blocked...";
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000821 break;
822 } else {
823 ADD_FAILURE();
824 break;
825 }
jbauche488a0d2015-11-19 05:17:58 -0800826 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000827
Yves Gerey665174f2018-06-19 15:03:05 +0200828 delete[] packet;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000829 }
830
Yves Gerey665174f2018-06-19 15:03:05 +0200831 void ReadData(rtc::StreamInterface* stream) override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000832 unsigned char buffer[2000];
833 size_t bread;
834 int err2;
835 rtc::StreamResult r;
836
837 for (;;) {
838 r = stream->Read(buffer, 2000, &bread, &err2);
839
840 if (r == rtc::SR_ERROR) {
841 // Unfortunately, errors are the way that the stream adapter
842 // signals close right now
843 stream->Close();
844 return;
845 }
846
847 if (r == rtc::SR_BLOCK)
848 break;
849
850 ASSERT_EQ(rtc::SR_SUCCESS, r);
Mirko Bonadei675513b2017-11-09 11:09:25 +0100851 RTC_LOG(LS_VERBOSE) << "Read " << bread;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000852
853 // Now parse the datagram
854 ASSERT_EQ(packet_size_, bread);
torbjorng7593aad2015-11-19 12:20:51 -0800855 unsigned char packet_num = buffer[0];
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000856
torbjorng7593aad2015-11-19 12:20:51 -0800857 unsigned int rand_state = packet_num;
858 for (size_t i = 1; i < packet_size_; i++) {
859 // This is a simple LC PRNG. Keep in synch with identical code above.
860 rand_state = (rand_state * 251 + 19937) >> 7;
861 ASSERT_EQ(rand_state & 0xff, buffer[i]);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000862 }
863 received_.insert(packet_num);
864 }
865 }
866
nisseef8b61e2016-04-29 06:09:15 -0700867 void TestTransfer(int count) override {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000868 count_ = count;
869
870 WriteData();
871
872 EXPECT_TRUE_WAIT(sent_ == count_, 10000);
Mirko Bonadei675513b2017-11-09 11:09:25 +0100873 RTC_LOG(LS_INFO) << "sent_ == " << sent_;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000874
875 if (damage_) {
876 WAIT(false, 2000);
877 EXPECT_EQ(0U, received_.size());
878 } else if (loss_ == 0) {
Yves Gerey665174f2018-06-19 15:03:05 +0200879 EXPECT_EQ_WAIT(static_cast<size_t>(sent_), received_.size(), 1000);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000880 } else {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100881 RTC_LOG(LS_INFO) << "Sent " << sent_ << " packets; received "
882 << received_.size();
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000883 }
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100884 }
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000885
Harald Alvestrand13799132020-03-09 19:39:36 +0100886 protected:
jbauche488a0d2015-11-19 05:17:58 -0800887 BufferQueueStream client_buffer_;
888 BufferQueueStream server_buffer_;
Harald Alvestrand13799132020-03-09 19:39:36 +0100889
890 private:
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000891 size_t packet_size_;
892 int count_;
893 int sent_;
894 std::set<int> received_;
895};
896
Harald Alvestrand13799132020-03-09 19:39:36 +0100897class SSLStreamAdapterTestDTLS
898 : public SSLStreamAdapterTestDTLSBase,
899 public WithParamInterface<tuple<rtc::KeyParams, rtc::KeyParams>> {
900 public:
901 SSLStreamAdapterTestDTLS()
902 : SSLStreamAdapterTestDTLSBase(::testing::get<0>(GetParam()),
903 ::testing::get<1>(GetParam())) {}
904
905 SSLStreamAdapterTestDTLS(const std::string& cert_pem,
906 const std::string& private_key_pem)
907 : SSLStreamAdapterTestDTLSBase(cert_pem, private_key_pem) {}
908};
909
Yves Gerey665174f2018-06-19 15:03:05 +0200910rtc::StreamResult SSLDummyStreamBase::Write(const void* data,
911 size_t data_len,
912 size_t* written,
913 int* error) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100914 RTC_LOG(LS_VERBOSE) << "Writing to loopback " << data_len;
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000915
916 if (first_packet_) {
917 first_packet_ = false;
jbauche488a0d2015-11-19 05:17:58 -0800918 if (test_base_->GetLoseFirstPacket()) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100919 RTC_LOG(LS_INFO) << "Losing initial packet of length " << data_len;
torbjorng7593aad2015-11-19 12:20:51 -0800920 *written = data_len; // Fake successful writing also to writer.
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000921 return rtc::SR_SUCCESS;
922 }
923 }
924
jbauche488a0d2015-11-19 05:17:58 -0800925 return test_base_->DataWritten(this, data, data_len, written, error);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100926}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000927
928class SSLStreamAdapterTestDTLSFromPEMStrings : public SSLStreamAdapterTestDTLS {
929 public:
Yves Gerey665174f2018-06-19 15:03:05 +0200930 SSLStreamAdapterTestDTLSFromPEMStrings()
931 : SSLStreamAdapterTestDTLS(kCERT_PEM, kRSA_PRIVATE_KEY_PEM) {}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000932};
933
Jian Cui0a8798b2017-11-16 16:58:02 -0800934// Test fixture for certificate chaining. Server will push more than one
935// certificate.
936class SSLStreamAdapterTestDTLSCertChain : public SSLStreamAdapterTestDTLS {
937 public:
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100938 SSLStreamAdapterTestDTLSCertChain() : SSLStreamAdapterTestDTLS("", "") {}
Jian Cui0a8798b2017-11-16 16:58:02 -0800939 void SetUp() override {
940 CreateStreams();
941
942 client_ssl_.reset(rtc::SSLStreamAdapter::Create(client_stream_));
943 server_ssl_.reset(rtc::SSLStreamAdapter::Create(server_stream_));
944
945 // Set up the slots
946 client_ssl_->SignalEvent.connect(
947 reinterpret_cast<SSLStreamAdapterTestBase*>(this),
948 &SSLStreamAdapterTestBase::OnEvent);
949 server_ssl_->SignalEvent.connect(
950 reinterpret_cast<SSLStreamAdapterTestBase*>(this),
951 &SSLStreamAdapterTestBase::OnEvent);
952
953 if (!client_cert_pem_.empty() && !client_private_key_pem_.empty()) {
954 client_identity_ = rtc::SSLIdentity::FromPEMStrings(
955 client_private_key_pem_, client_cert_pem_);
956 } else {
957 client_identity_ = rtc::SSLIdentity::Generate("client", client_key_type_);
958 }
959
960 client_ssl_->SetIdentity(client_identity_);
961 }
962};
963
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000964// Basic tests: TLS
965
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000966// Test that we can make a handshake work
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +0200967TEST_P(SSLStreamAdapterTestTLS, TestTLSConnect) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000968 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +0100969}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +0000970
Jian Cui0a8798b2017-11-16 16:58:02 -0800971TEST_P(SSLStreamAdapterTestTLS, GetPeerCertChainWithOneCertificate) {
972 TestHandshake();
973 std::unique_ptr<rtc::SSLCertChain> cert_chain =
974 client_ssl_->GetPeerSSLCertChain();
Jian Cui0a8798b2017-11-16 16:58:02 -0800975 ASSERT_NE(nullptr, cert_chain);
976 EXPECT_EQ(1u, cert_chain->GetSize());
Taylor Brandstetterc3928662018-02-23 13:04:51 -0800977 EXPECT_EQ(cert_chain->Get(0).ToPEMString(),
978 server_identity_->certificate().ToPEMString());
Jian Cui0a8798b2017-11-16 16:58:02 -0800979}
980
981TEST_F(SSLStreamAdapterTestDTLSCertChain, TwoCertHandshake) {
982 server_identity_ = rtc::SSLIdentity::FromPEMChainStrings(
983 kRSA_PRIVATE_KEY_PEM, std::string(kCERT_PEM) + kCACert);
984 server_ssl_->SetIdentity(server_identity_);
985 TestHandshake();
986 std::unique_ptr<rtc::SSLCertChain> peer_cert_chain =
987 client_ssl_->GetPeerSSLCertChain();
988 ASSERT_NE(nullptr, peer_cert_chain);
989 ASSERT_EQ(2u, peer_cert_chain->GetSize());
990 EXPECT_EQ(kCERT_PEM, peer_cert_chain->Get(0).ToPEMString());
991 EXPECT_EQ(kCACert, peer_cert_chain->Get(1).ToPEMString());
992}
993
David Benjaminea84b6b2017-12-01 17:25:45 -0500994TEST_F(SSLStreamAdapterTestDTLSCertChain, TwoCertHandshakeWithCopy) {
995 std::unique_ptr<rtc::SSLIdentity> identity(
996 rtc::SSLIdentity::FromPEMChainStrings(kRSA_PRIVATE_KEY_PEM,
997 std::string(kCERT_PEM) + kCACert));
998 server_identity_ = identity->GetReference();
999 server_ssl_->SetIdentity(server_identity_);
1000 TestHandshake();
1001 std::unique_ptr<rtc::SSLCertChain> peer_cert_chain =
1002 client_ssl_->GetPeerSSLCertChain();
1003 ASSERT_NE(nullptr, peer_cert_chain);
1004 ASSERT_EQ(2u, peer_cert_chain->GetSize());
1005 EXPECT_EQ(kCERT_PEM, peer_cert_chain->Get(0).ToPEMString());
1006 EXPECT_EQ(kCACert, peer_cert_chain->Get(1).ToPEMString());
1007}
1008
Jian Cui0a8798b2017-11-16 16:58:02 -08001009TEST_F(SSLStreamAdapterTestDTLSCertChain, ThreeCertHandshake) {
1010 server_identity_ = rtc::SSLIdentity::FromPEMChainStrings(
1011 kRSA_PRIVATE_KEY_PEM, std::string(kCERT_PEM) + kIntCert1 + kCACert);
1012 server_ssl_->SetIdentity(server_identity_);
1013 TestHandshake();
1014 std::unique_ptr<rtc::SSLCertChain> peer_cert_chain =
1015 client_ssl_->GetPeerSSLCertChain();
1016 ASSERT_NE(nullptr, peer_cert_chain);
1017 ASSERT_EQ(3u, peer_cert_chain->GetSize());
1018 EXPECT_EQ(kCERT_PEM, peer_cert_chain->Get(0).ToPEMString());
1019 EXPECT_EQ(kIntCert1, peer_cert_chain->Get(1).ToPEMString());
1020 EXPECT_EQ(kCACert, peer_cert_chain->Get(2).ToPEMString());
1021}
1022
jiayl@webrtc.orgf1d751c2014-09-25 16:38:46 +00001023// Test that closing the connection on one side updates the other side.
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001024TEST_P(SSLStreamAdapterTestTLS, TestTLSClose) {
jiayl@webrtc.orgf1d751c2014-09-25 16:38:46 +00001025 TestHandshake();
1026 client_ssl_->Close();
1027 EXPECT_EQ_WAIT(rtc::SS_CLOSED, server_ssl_->GetState(), handshake_wait_);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001028}
jiayl@webrtc.orgf1d751c2014-09-25 16:38:46 +00001029
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001030// Test transfer -- trivial
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001031TEST_P(SSLStreamAdapterTestTLS, TestTLSTransfer) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001032 TestHandshake();
1033 TestTransfer(100000);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001034}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001035
1036// Test read-write after close.
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001037TEST_P(SSLStreamAdapterTestTLS, ReadWriteAfterClose) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001038 TestHandshake();
1039 TestTransfer(100000);
1040 client_ssl_->Close();
1041
1042 rtc::StreamResult rv;
1043 char block[kBlockSize];
1044 size_t dummy;
1045
1046 // It's an error to write after closed.
deadbeef37f5ecf2017-02-27 14:06:41 -08001047 rv = client_ssl_->Write(block, sizeof(block), &dummy, nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001048 ASSERT_EQ(rtc::SR_ERROR, rv);
1049
1050 // But after closed read gives you EOS.
deadbeef37f5ecf2017-02-27 14:06:41 -08001051 rv = client_ssl_->Read(block, sizeof(block), &dummy, nullptr);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001052 ASSERT_EQ(rtc::SR_EOS, rv);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001053}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001054
1055// Test a handshake with a bogus peer digest
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001056TEST_P(SSLStreamAdapterTestTLS, TestTLSBogusDigest) {
deadbeef89824f62016-09-30 11:55:43 -07001057 SetPeerIdentitiesByDigest(false, true);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001058 TestHandshake(false);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001059}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001060
deadbeef89824f62016-09-30 11:55:43 -07001061TEST_P(SSLStreamAdapterTestTLS, TestTLSDelayedIdentity) {
1062 TestHandshakeWithDelayedIdentity(true);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001063}
deadbeef89824f62016-09-30 11:55:43 -07001064
1065TEST_P(SSLStreamAdapterTestTLS, TestTLSDelayedIdentityWithBogusDigest) {
1066 TestHandshakeWithDelayedIdentity(false);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001067}
deadbeef89824f62016-09-30 11:55:43 -07001068
1069// Test that the correct error is returned when SetPeerCertificateDigest is
1070// called with an unknown algorithm.
1071TEST_P(SSLStreamAdapterTestTLS,
1072 TestSetPeerCertificateDigestWithUnknownAlgorithm) {
1073 unsigned char server_digest[20];
1074 size_t server_digest_len;
1075 bool rv;
1076 rtc::SSLPeerCertificateDigestError err;
1077
1078 rv = server_identity_->certificate().ComputeDigest(
1079 rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
1080 ASSERT_TRUE(rv);
1081
1082 rv = client_ssl_->SetPeerCertificateDigest("unknown algorithm", server_digest,
1083 server_digest_len, &err);
1084 EXPECT_EQ(rtc::SSLPeerCertificateDigestError::UNKNOWN_ALGORITHM, err);
1085 EXPECT_FALSE(rv);
1086}
1087
1088// Test that the correct error is returned when SetPeerCertificateDigest is
1089// called with an invalid digest length.
1090TEST_P(SSLStreamAdapterTestTLS, TestSetPeerCertificateDigestWithInvalidLength) {
1091 unsigned char server_digest[20];
1092 size_t server_digest_len;
1093 bool rv;
1094 rtc::SSLPeerCertificateDigestError err;
1095
1096 rv = server_identity_->certificate().ComputeDigest(
1097 rtc::DIGEST_SHA_1, server_digest, 20, &server_digest_len);
1098 ASSERT_TRUE(rv);
1099
1100 rv = client_ssl_->SetPeerCertificateDigest(rtc::DIGEST_SHA_1, server_digest,
1101 server_digest_len - 1, &err);
1102 EXPECT_EQ(rtc::SSLPeerCertificateDigestError::INVALID_LENGTH, err);
1103 EXPECT_FALSE(rv);
1104}
1105
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001106// Test moving a bunch of data
1107
1108// Basic tests: DTLS
1109// Test that we can make a handshake work
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001110TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnect) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001111 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001112}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001113
1114// Test that we can make a handshake work if the first packet in
1115// each direction is lost. This gives us predictable loss
1116// rather than having to tune random
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001117TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnectWithLostFirstPacket) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001118 SetLoseFirstPacket(true);
1119 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001120}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001121
1122// Test a handshake with loss and delay
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001123TEST_P(SSLStreamAdapterTestDTLS, TestDTLSConnectWithLostFirstPacketDelay2s) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001124 SetLoseFirstPacket(true);
1125 SetDelay(2000);
1126 SetHandshakeWait(20000);
1127 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001128}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001129
1130// Test a handshake with small MTU
pbos@webrtc.org127ca3f2014-10-09 07:52:03 +00001131// Disabled due to https://code.google.com/p/webrtc/issues/detail?id=3910
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001132TEST_P(SSLStreamAdapterTestDTLS, DISABLED_TestDTLSConnectWithSmallMtu) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001133 SetMtu(700);
1134 SetHandshakeWait(20000);
1135 TestHandshake();
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001136}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001137
1138// Test transfer -- trivial
jbauche488a0d2015-11-19 05:17:58 -08001139TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransfer) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001140 TestHandshake();
1141 TestTransfer(100);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001142}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001143
jbauche488a0d2015-11-19 05:17:58 -08001144TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransferWithLoss) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001145 TestHandshake();
1146 SetLoss(10);
1147 TestTransfer(100);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001148}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001149
jbauche488a0d2015-11-19 05:17:58 -08001150TEST_P(SSLStreamAdapterTestDTLS, TestDTLSTransferWithDamage) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001151 SetDamage(); // Must be called first because first packet
1152 // write happens at end of handshake.
1153 TestHandshake();
1154 TestTransfer(100);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001155}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001156
deadbeef89824f62016-09-30 11:55:43 -07001157TEST_P(SSLStreamAdapterTestDTLS, TestDTLSDelayedIdentity) {
1158 TestHandshakeWithDelayedIdentity(true);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001159}
deadbeef89824f62016-09-30 11:55:43 -07001160
1161TEST_P(SSLStreamAdapterTestDTLS, TestDTLSDelayedIdentityWithBogusDigest) {
1162 TestHandshakeWithDelayedIdentity(false);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001163}
deadbeef89824f62016-09-30 11:55:43 -07001164
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001165// Test DTLS-SRTP with all high ciphers
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001166TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpHigh) {
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001167 std::vector<int> high;
1168 high.push_back(rtc::SRTP_AES128_CM_SHA1_80);
1169 SetDtlsSrtpCryptoSuites(high, true);
1170 SetDtlsSrtpCryptoSuites(high, false);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001171 TestHandshake();
1172
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001173 int client_cipher;
1174 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1175 int server_cipher;
1176 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001177
1178 ASSERT_EQ(client_cipher, server_cipher);
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001179 ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_80);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001180}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001181
1182// Test DTLS-SRTP with all low ciphers
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001183TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpLow) {
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001184 std::vector<int> low;
1185 low.push_back(rtc::SRTP_AES128_CM_SHA1_32);
1186 SetDtlsSrtpCryptoSuites(low, true);
1187 SetDtlsSrtpCryptoSuites(low, false);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001188 TestHandshake();
1189
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001190 int client_cipher;
1191 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1192 int server_cipher;
1193 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001194
1195 ASSERT_EQ(client_cipher, server_cipher);
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001196 ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_32);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001197}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001198
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001199// Test DTLS-SRTP with a mismatch -- should not converge
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001200TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpHighLow) {
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001201 std::vector<int> high;
1202 high.push_back(rtc::SRTP_AES128_CM_SHA1_80);
1203 std::vector<int> low;
1204 low.push_back(rtc::SRTP_AES128_CM_SHA1_32);
1205 SetDtlsSrtpCryptoSuites(high, true);
1206 SetDtlsSrtpCryptoSuites(low, false);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001207 TestHandshake();
1208
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001209 int client_cipher;
1210 ASSERT_FALSE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1211 int server_cipher;
1212 ASSERT_FALSE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001213}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001214
1215// Test DTLS-SRTP with each side being mixed -- should select high
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001216TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpMixed) {
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001217 std::vector<int> mixed;
1218 mixed.push_back(rtc::SRTP_AES128_CM_SHA1_80);
1219 mixed.push_back(rtc::SRTP_AES128_CM_SHA1_32);
1220 SetDtlsSrtpCryptoSuites(mixed, true);
1221 SetDtlsSrtpCryptoSuites(mixed, false);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001222 TestHandshake();
1223
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001224 int client_cipher;
1225 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1226 int server_cipher;
1227 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001228
1229 ASSERT_EQ(client_cipher, server_cipher);
Guo-wei Shieh521ed7b2015-11-18 19:41:53 -08001230 ASSERT_EQ(client_cipher, rtc::SRTP_AES128_CM_SHA1_80);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001231}
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001232
jbauchcb560652016-08-04 05:20:32 -07001233// Test DTLS-SRTP with all GCM-128 ciphers.
1234TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCM128) {
jbauchcb560652016-08-04 05:20:32 -07001235 std::vector<int> gcm128;
1236 gcm128.push_back(rtc::SRTP_AEAD_AES_128_GCM);
1237 SetDtlsSrtpCryptoSuites(gcm128, true);
1238 SetDtlsSrtpCryptoSuites(gcm128, false);
1239 TestHandshake();
1240
1241 int client_cipher;
1242 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1243 int server_cipher;
1244 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
1245
1246 ASSERT_EQ(client_cipher, server_cipher);
1247 ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_128_GCM);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001248}
jbauchcb560652016-08-04 05:20:32 -07001249
1250// Test DTLS-SRTP with all GCM-256 ciphers.
1251TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCM256) {
jbauchcb560652016-08-04 05:20:32 -07001252 std::vector<int> gcm256;
1253 gcm256.push_back(rtc::SRTP_AEAD_AES_256_GCM);
1254 SetDtlsSrtpCryptoSuites(gcm256, true);
1255 SetDtlsSrtpCryptoSuites(gcm256, false);
1256 TestHandshake();
1257
1258 int client_cipher;
1259 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1260 int server_cipher;
1261 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
1262
1263 ASSERT_EQ(client_cipher, server_cipher);
1264 ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_256_GCM);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001265}
jbauchcb560652016-08-04 05:20:32 -07001266
1267// Test DTLS-SRTP with mixed GCM-128/-256 ciphers -- should not converge.
1268TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCMMismatch) {
jbauchcb560652016-08-04 05:20:32 -07001269 std::vector<int> gcm128;
1270 gcm128.push_back(rtc::SRTP_AEAD_AES_128_GCM);
1271 std::vector<int> gcm256;
1272 gcm256.push_back(rtc::SRTP_AEAD_AES_256_GCM);
1273 SetDtlsSrtpCryptoSuites(gcm128, true);
1274 SetDtlsSrtpCryptoSuites(gcm256, false);
1275 TestHandshake();
1276
1277 int client_cipher;
1278 ASSERT_FALSE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1279 int server_cipher;
1280 ASSERT_FALSE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001281}
jbauchcb560652016-08-04 05:20:32 -07001282
1283// Test DTLS-SRTP with both GCM-128/-256 ciphers -- should select GCM-256.
1284TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpGCMMixed) {
jbauchcb560652016-08-04 05:20:32 -07001285 std::vector<int> gcmBoth;
1286 gcmBoth.push_back(rtc::SRTP_AEAD_AES_256_GCM);
1287 gcmBoth.push_back(rtc::SRTP_AEAD_AES_128_GCM);
1288 SetDtlsSrtpCryptoSuites(gcmBoth, true);
1289 SetDtlsSrtpCryptoSuites(gcmBoth, false);
1290 TestHandshake();
1291
1292 int client_cipher;
1293 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(true, &client_cipher));
1294 int server_cipher;
1295 ASSERT_TRUE(GetDtlsSrtpCryptoSuite(false, &server_cipher));
1296
1297 ASSERT_EQ(client_cipher, server_cipher);
1298 ASSERT_EQ(client_cipher, rtc::SRTP_AEAD_AES_256_GCM);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001299}
jbauchcb560652016-08-04 05:20:32 -07001300
1301// Test SRTP cipher suite lengths.
1302TEST_P(SSLStreamAdapterTestDTLS, TestDTLSSrtpKeyAndSaltLengths) {
1303 int key_len;
1304 int salt_len;
1305
Yves Gerey665174f2018-06-19 15:03:05 +02001306 ASSERT_FALSE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_INVALID_CRYPTO_SUITE,
1307 &key_len, &salt_len));
jbauchcb560652016-08-04 05:20:32 -07001308
Yves Gerey665174f2018-06-19 15:03:05 +02001309 ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_AES128_CM_SHA1_32,
1310 &key_len, &salt_len));
1311 ASSERT_EQ(128 / 8, key_len);
1312 ASSERT_EQ(112 / 8, salt_len);
jbauchcb560652016-08-04 05:20:32 -07001313
Yves Gerey665174f2018-06-19 15:03:05 +02001314 ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_AES128_CM_SHA1_80,
1315 &key_len, &salt_len));
1316 ASSERT_EQ(128 / 8, key_len);
1317 ASSERT_EQ(112 / 8, salt_len);
jbauchcb560652016-08-04 05:20:32 -07001318
Yves Gerey665174f2018-06-19 15:03:05 +02001319 ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_AEAD_AES_128_GCM,
1320 &key_len, &salt_len));
1321 ASSERT_EQ(128 / 8, key_len);
1322 ASSERT_EQ(96 / 8, salt_len);
jbauchcb560652016-08-04 05:20:32 -07001323
Yves Gerey665174f2018-06-19 15:03:05 +02001324 ASSERT_TRUE(rtc::GetSrtpKeyAndSaltLengths(rtc::SRTP_AEAD_AES_256_GCM,
1325 &key_len, &salt_len));
1326 ASSERT_EQ(256 / 8, key_len);
1327 ASSERT_EQ(96 / 8, salt_len);
Mirko Bonadeic4dd7302019-02-25 09:12:02 +01001328}
jbauchcb560652016-08-04 05:20:32 -07001329
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001330// Test an exporter
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001331TEST_P(SSLStreamAdapterTestDTLS, TestDTLSExporter) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001332 TestHandshake();
1333 unsigned char client_out[20];
1334 unsigned char server_out[20];
1335
1336 bool result;
Yves Gerey665174f2018-06-19 15:03:05 +02001337 result = ExportKeyingMaterial(kExporterLabel, kExporterContext,
1338 kExporterContextLen, true, true, client_out,
1339 sizeof(client_out));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001340 ASSERT_TRUE(result);
1341
Yves Gerey665174f2018-06-19 15:03:05 +02001342 result = ExportKeyingMaterial(kExporterLabel, kExporterContext,
1343 kExporterContextLen, true, false, server_out,
1344 sizeof(server_out));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001345 ASSERT_TRUE(result);
1346
1347 ASSERT_TRUE(!memcmp(client_out, server_out, sizeof(client_out)));
1348}
1349
1350// Test not yet valid certificates are not rejected.
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001351TEST_P(SSLStreamAdapterTestDTLS, TestCertNotYetValid) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001352 long one_day = 60 * 60 * 24;
1353 // Make the certificates not valid until one day later.
1354 ResetIdentitiesWithValidity(one_day, one_day);
1355 TestHandshake();
1356}
1357
1358// Test expired certificates are not rejected.
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001359TEST_P(SSLStreamAdapterTestDTLS, TestCertExpired) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001360 long one_day = 60 * 60 * 24;
1361 // Make the certificates already expired.
1362 ResetIdentitiesWithValidity(-one_day, -one_day);
1363 TestHandshake();
1364}
1365
1366// Test data transfer using certs created from strings.
torbjorng7593aad2015-11-19 12:20:51 -08001367TEST_F(SSLStreamAdapterTestDTLSFromPEMStrings, TestTransfer) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001368 TestHandshake();
1369 TestTransfer(100);
1370}
1371
1372// Test getting the remote certificate.
torbjorng7593aad2015-11-19 12:20:51 -08001373TEST_F(SSLStreamAdapterTestDTLSFromPEMStrings, TestDTLSGetPeerCertificate) {
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001374 // Peer certificates haven't been received yet.
kwibergb4d01c42016-04-06 05:15:06 -07001375 ASSERT_FALSE(GetPeerCertificate(true));
1376 ASSERT_FALSE(GetPeerCertificate(false));
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001377
1378 TestHandshake();
1379
1380 // The client should have a peer certificate after the handshake.
jbauch555604a2016-04-26 03:13:22 -07001381 std::unique_ptr<rtc::SSLCertificate> client_peer_cert =
kwibergb4d01c42016-04-06 05:15:06 -07001382 GetPeerCertificate(true);
1383 ASSERT_TRUE(client_peer_cert);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001384
1385 // It's not kCERT_PEM.
1386 std::string client_peer_string = client_peer_cert->ToPEMString();
1387 ASSERT_NE(kCERT_PEM, client_peer_string);
1388
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001389 // The server should have a peer certificate after the handshake.
jbauch555604a2016-04-26 03:13:22 -07001390 std::unique_ptr<rtc::SSLCertificate> server_peer_cert =
kwibergb4d01c42016-04-06 05:15:06 -07001391 GetPeerCertificate(false);
1392 ASSERT_TRUE(server_peer_cert);
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001393
1394 // It's kCERT_PEM
1395 ASSERT_EQ(kCERT_PEM, server_peer_cert->ToPEMString());
henrike@webrtc.orgf0488722014-05-13 18:00:26 +00001396}
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +00001397
Joachim Bauch831c5582015-05-20 12:48:41 +02001398// Test getting the used DTLS 1.2 ciphers.
1399// DTLS 1.2 enabled for client and server -> DTLS 1.2 will be used.
Guo-wei Shieh456696a2015-09-30 21:48:54 -07001400TEST_P(SSLStreamAdapterTestDTLS, TestGetSslCipherSuiteDtls12Both) {
Joachim Bauch831c5582015-05-20 12:48:41 +02001401 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_12);
1402 TestHandshake();
1403
Guo-wei Shieh6caafbe2015-10-05 12:43:27 -07001404 int client_cipher;
Guo-wei Shieh456696a2015-09-30 21:48:54 -07001405 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
Guo-wei Shieh6caafbe2015-10-05 12:43:27 -07001406 int server_cipher;
Guo-wei Shieh456696a2015-09-30 21:48:54 -07001407 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
Joachim Bauch831c5582015-05-20 12:48:41 +02001408
torbjorng43166b82016-03-11 00:06:47 -08001409 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(true));
1410 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(false));
1411
Joachim Bauch831c5582015-05-20 12:48:41 +02001412 ASSERT_EQ(client_cipher, server_cipher);
torbjorng43166b82016-03-11 00:06:47 -08001413 ASSERT_TRUE(rtc::SSLStreamAdapter::IsAcceptableCipher(
1414 server_cipher, ::testing::get<1>(GetParam()).type()));
Joachim Bauch831c5582015-05-20 12:48:41 +02001415}
1416
Harald Alvestrand13799132020-03-09 19:39:36 +01001417// Test getting the used DTLS ciphers.
1418// DTLS 1.0 is max version for client and server, this will only work if
1419// legacy is enabled.
1420TEST_P(SSLStreamAdapterTestDTLS, TestGetSslCipherSuite) {
1421 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
Benjamin Wrightaf1f8652019-04-01 18:25:23 +00001422 TestHandshake();
1423
1424 int client_cipher;
1425 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1426 int server_cipher;
1427 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1428
1429 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
1430 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
torbjorng43166b82016-03-11 00:06:47 -08001431
Joachim Bauch831c5582015-05-20 12:48:41 +02001432 ASSERT_EQ(client_cipher, server_cipher);
torbjorng43166b82016-03-11 00:06:47 -08001433 ASSERT_TRUE(rtc::SSLStreamAdapter::IsAcceptableCipher(
1434 server_cipher, ::testing::get<1>(GetParam()).type()));
pthatcher@webrtc.org3ee4fe52015-02-11 22:34:36 +00001435}
Torbjorn Granlundb6d4ec42015-08-17 14:08:59 +02001436
torbjorng4e572472015-10-08 09:42:49 -07001437// The RSA keysizes here might look strange, why not include the RFC's size
1438// 2048?. The reason is test case slowness; testing two sizes to exercise
1439// parametrization is sufficient.
Mirko Bonadeic84f6612019-01-31 12:20:57 +01001440INSTANTIATE_TEST_SUITE_P(
torbjorng4e572472015-10-08 09:42:49 -07001441 SSLStreamAdapterTestsTLS,
1442 SSLStreamAdapterTestTLS,
1443 Combine(Values(rtc::KeyParams::RSA(1024, 65537),
1444 rtc::KeyParams::RSA(1152, 65537),
1445 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256)),
1446 Values(rtc::KeyParams::RSA(1024, 65537),
1447 rtc::KeyParams::RSA(1152, 65537),
1448 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256))));
Mirko Bonadeic84f6612019-01-31 12:20:57 +01001449INSTANTIATE_TEST_SUITE_P(
torbjorng4e572472015-10-08 09:42:49 -07001450 SSLStreamAdapterTestsDTLS,
1451 SSLStreamAdapterTestDTLS,
1452 Combine(Values(rtc::KeyParams::RSA(1024, 65537),
1453 rtc::KeyParams::RSA(1152, 65537),
1454 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256)),
1455 Values(rtc::KeyParams::RSA(1024, 65537),
1456 rtc::KeyParams::RSA(1152, 65537),
1457 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256))));
Harald Alvestrand13799132020-03-09 19:39:36 +01001458
1459// Tests for enabling / disabling legacy TLS protocols in DTLS.
1460class SSLStreamAdapterTestDTLSLegacyProtocols
1461 : public SSLStreamAdapterTestDTLSBase {
1462 public:
1463 SSLStreamAdapterTestDTLSLegacyProtocols()
1464 : SSLStreamAdapterTestDTLSBase(rtc::KeyParams::ECDSA(rtc::EC_NIST_P256),
1465 rtc::KeyParams::ECDSA(rtc::EC_NIST_P256)) {
1466 }
1467
1468 // Do not use the SetUp version from the parent class.
1469 void SetUp() override {}
1470
1471 // The legacy TLS protocols flag is read when the OpenSSLStreamAdapter is
1472 // initialized, so we set the experiment while creationg client_ssl_
1473 // and server_ssl_.
1474
1475 void ConfigureClient(std::string experiment) {
1476 webrtc::test::ScopedFieldTrials trial(experiment);
1477 client_stream_ =
1478 new SSLDummyStreamDTLS(this, "c2s", &client_buffer_, &server_buffer_);
1479 client_ssl_.reset(rtc::SSLStreamAdapter::Create(client_stream_));
1480 client_ssl_->SignalEvent.connect(
1481 static_cast<SSLStreamAdapterTestBase*>(this),
1482 &SSLStreamAdapterTestBase::OnEvent);
1483 client_identity_ = rtc::SSLIdentity::Generate("client", client_key_type_);
1484 client_ssl_->SetIdentity(client_identity_);
1485 }
1486
1487 void ConfigureServer(std::string experiment) {
1488 // webrtc::test::ScopedFieldTrials trial(experiment);
1489 server_stream_ =
1490 new SSLDummyStreamDTLS(this, "s2c", &server_buffer_, &client_buffer_);
1491 server_ssl_.reset(rtc::SSLStreamAdapter::Create(server_stream_));
1492 server_ssl_->SignalEvent.connect(
1493 static_cast<SSLStreamAdapterTestBase*>(this),
1494 &SSLStreamAdapterTestBase::OnEvent);
1495 server_identity_ = rtc::SSLIdentity::Generate("server", server_key_type_);
1496 server_ssl_->SetIdentity(server_identity_);
1497 }
1498};
1499
1500// Test getting the used DTLS ciphers.
1501// DTLS 1.2 enabled for neither client nor server -> DTLS 1.0 will be used.
1502TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols, TestGetSslCipherSuite) {
1503 ConfigureClient("");
1504 ConfigureServer("");
1505 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
1506 TestHandshake();
1507
1508 int client_cipher;
1509 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1510 int server_cipher;
1511 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1512
1513 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
1514 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
1515
1516 ASSERT_EQ(client_cipher, server_cipher);
1517}
1518
1519// Test getting the used DTLS 1.2 ciphers.
1520// DTLS 1.2 enabled for client and server -> DTLS 1.2 will be used.
1521TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1522 TestGetSslCipherSuiteDtls12Both) {
1523 ConfigureClient("");
1524 ConfigureServer("");
1525 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_12);
1526 TestHandshake();
1527
1528 int client_cipher;
1529 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1530 int server_cipher;
1531 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1532
1533 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(true));
1534 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_12, GetSslVersion(false));
1535
1536 ASSERT_EQ(client_cipher, server_cipher);
1537}
1538
1539// DTLS 1.2 enabled for client only -> DTLS 1.0 will be used.
1540TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1541 TestGetSslCipherSuiteDtls12Client) {
1542 ConfigureClient("");
1543 ConfigureServer("");
1544 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_12);
1545 TestHandshake();
1546
1547 int client_cipher;
1548 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1549 int server_cipher;
1550 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1551
1552 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
1553 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
1554
1555 ASSERT_EQ(client_cipher, server_cipher);
1556}
1557
1558// DTLS 1.2 enabled for server only -> DTLS 1.0 will be used.
1559TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1560 TestGetSslCipherSuiteDtls12Server) {
1561 ConfigureClient("");
1562 ConfigureServer("");
1563 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_10);
1564 TestHandshake();
1565
1566 int client_cipher;
1567 ASSERT_TRUE(GetSslCipherSuite(true, &client_cipher));
1568 int server_cipher;
1569 ASSERT_TRUE(GetSslCipherSuite(false, &server_cipher));
1570
1571 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(true));
1572 ASSERT_EQ(rtc::SSL_PROTOCOL_DTLS_10, GetSslVersion(false));
1573
1574 ASSERT_EQ(client_cipher, server_cipher);
1575}
1576
1577// Client has legacy TLS versions disabled, server has DTLS 1.0 only.
1578// This is meant to cause a failure.
1579TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1580 TestGetSslVersionLegacyDisabledServer10) {
1581 ConfigureClient("WebRTC-LegacyTlsProtocols/Disabled/");
1582 ConfigureServer("");
1583 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_12);
1584 // Handshake should fail.
1585 TestHandshake(false);
1586}
1587
1588// Both client and server have legacy TLS versions disabled and support
1589// DTLS 1.2. This should work.
1590TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1591 TestGetSslVersionLegacyDisabledServer12) {
1592 ConfigureClient("WebRTC-LegacyTlsProtocols/Disabled/");
1593 ConfigureServer("WebRTC-LegacyTlsProtocols/Disabled/");
1594 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_12, rtc::SSL_PROTOCOL_DTLS_12);
1595 TestHandshake();
1596}
1597
1598// Both client and server have legacy TLS versions enabled and support DTLS 1.0.
1599// This should work.
1600TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1601 TestGetSslVersionLegacyEnabledClient10Server10) {
1602 ConfigureClient("WebRTC-LegacyTlsProtocols/Enabled/");
1603 ConfigureServer("WebRTC-LegacyTlsProtocols/Enabled/");
1604 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
1605 TestHandshake();
1606}
1607
1608// Legacy protocols are disabled, max TLS version is 1.0
1609// This should be a configuration error, and handshake should fail.
1610TEST_F(SSLStreamAdapterTestDTLSLegacyProtocols,
1611 TestGetSslVersionLegacyDisabledClient10Server10) {
1612 ConfigureClient("WebRTC-LegacyTlsProtocols/Disabled/");
1613 ConfigureServer("WebRTC-LegacyTlsProtocols/Disabled/");
1614 SetupProtocolVersions(rtc::SSL_PROTOCOL_DTLS_10, rtc::SSL_PROTOCOL_DTLS_10);
1615 TestHandshake(false);
1616}