diff options
Diffstat (limited to 'third_party/libwebrtc/rtc_base/openssl_session_cache_unittest.cc')
-rw-r--r-- | third_party/libwebrtc/rtc_base/openssl_session_cache_unittest.cc | 104 |
1 files changed, 104 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_base/openssl_session_cache_unittest.cc b/third_party/libwebrtc/rtc_base/openssl_session_cache_unittest.cc new file mode 100644 index 0000000000..0441d5c012 --- /dev/null +++ b/third_party/libwebrtc/rtc_base/openssl_session_cache_unittest.cc @@ -0,0 +1,104 @@ +/* + * Copyright 2018 The WebRTC Project Authors. All rights reserved. + * + * Use of this source code is governed by a BSD-style license + * that can be found in the LICENSE file in the root of the source + * tree. An additional intellectual property rights grant can be found + * in the file PATENTS. All contributing project authors may + * be found in the AUTHORS file in the root of the source tree. + */ + +#include "rtc_base/openssl_session_cache.h" + +#include <openssl/ssl.h> +#include <stdlib.h> + +#include <map> +#include <memory> + +#include "rtc_base/gunit.h" +#include "rtc_base/openssl.h" + +namespace { +// Use methods that avoid X509 objects if possible. +SSL_CTX* NewDtlsContext() { +#ifdef OPENSSL_IS_BORINGSSL + return SSL_CTX_new(DTLS_with_buffers_method()); +#else + return SSL_CTX_new(DTLS_method()); +#endif +} +SSL_CTX* NewTlsContext() { +#ifdef OPENSSL_IS_BORINGSSL + return SSL_CTX_new(TLS_with_buffers_method()); +#else + return SSL_CTX_new(TLS_method()); +#endif +} +} // namespace + +namespace rtc { + +TEST(OpenSSLSessionCache, DTLSModeSetCorrectly) { + SSL_CTX* ssl_ctx = NewDtlsContext(); + + OpenSSLSessionCache session_cache(SSL_MODE_DTLS, ssl_ctx); + EXPECT_EQ(session_cache.GetSSLMode(), SSL_MODE_DTLS); + + SSL_CTX_free(ssl_ctx); +} + +TEST(OpenSSLSessionCache, TLSModeSetCorrectly) { + SSL_CTX* ssl_ctx = NewTlsContext(); + + OpenSSLSessionCache session_cache(SSL_MODE_TLS, ssl_ctx); + EXPECT_EQ(session_cache.GetSSLMode(), SSL_MODE_TLS); + + SSL_CTX_free(ssl_ctx); +} + +TEST(OpenSSLSessionCache, SSLContextSetCorrectly) { + SSL_CTX* ssl_ctx = NewDtlsContext(); + + OpenSSLSessionCache session_cache(SSL_MODE_DTLS, ssl_ctx); + EXPECT_EQ(session_cache.GetSSLContext(), ssl_ctx); + + SSL_CTX_free(ssl_ctx); +} + +TEST(OpenSSLSessionCache, InvalidLookupReturnsNullptr) { + SSL_CTX* ssl_ctx = NewDtlsContext(); + + OpenSSLSessionCache session_cache(SSL_MODE_DTLS, ssl_ctx); + EXPECT_EQ(session_cache.LookupSession("Invalid"), nullptr); + EXPECT_EQ(session_cache.LookupSession(""), nullptr); + EXPECT_EQ(session_cache.LookupSession("."), nullptr); + + SSL_CTX_free(ssl_ctx); +} + +TEST(OpenSSLSessionCache, SimpleValidSessionLookup) { + SSL_CTX* ssl_ctx = NewDtlsContext(); + SSL_SESSION* ssl_session = SSL_SESSION_new(ssl_ctx); + + OpenSSLSessionCache session_cache(SSL_MODE_DTLS, ssl_ctx); + session_cache.AddSession("webrtc.org", ssl_session); + EXPECT_EQ(session_cache.LookupSession("webrtc.org"), ssl_session); + + SSL_CTX_free(ssl_ctx); +} + +TEST(OpenSSLSessionCache, AddToExistingReplacesPrevious) { + SSL_CTX* ssl_ctx = NewDtlsContext(); + SSL_SESSION* ssl_session_1 = SSL_SESSION_new(ssl_ctx); + SSL_SESSION* ssl_session_2 = SSL_SESSION_new(ssl_ctx); + + OpenSSLSessionCache session_cache(SSL_MODE_DTLS, ssl_ctx); + session_cache.AddSession("webrtc.org", ssl_session_1); + session_cache.AddSession("webrtc.org", ssl_session_2); + EXPECT_EQ(session_cache.LookupSession("webrtc.org"), ssl_session_2); + + SSL_CTX_free(ssl_ctx); +} + +} // namespace rtc |