diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/libwebrtc/rtc_base/openssl_adapter_unittest.cc | |
parent | Initial commit. (diff) | |
download | firefox-esr-upstream.tar.xz firefox-esr-upstream.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esrupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/rtc_base/openssl_adapter_unittest.cc')
-rw-r--r-- | third_party/libwebrtc/rtc_base/openssl_adapter_unittest.cc | 134 |
1 files changed, 134 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_base/openssl_adapter_unittest.cc b/third_party/libwebrtc/rtc_base/openssl_adapter_unittest.cc new file mode 100644 index 0000000000..ce351dc98e --- /dev/null +++ b/third_party/libwebrtc/rtc_base/openssl_adapter_unittest.cc @@ -0,0 +1,134 @@ +/* + * Copyright 2017 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_adapter.h" + +#include <sstream> +#include <string> +#include <vector> + +#include "absl/memory/memory.h" +#include "rtc_base/gunit.h" +#include "rtc_base/socket.h" +#include "test/gmock.h" + +namespace rtc { +namespace { + +class MockAsyncSocket : public Socket { + public: + virtual ~MockAsyncSocket() = default; + MOCK_METHOD(Socket*, Accept, (SocketAddress*), (override)); + MOCK_METHOD(SocketAddress, GetLocalAddress, (), (const, override)); + MOCK_METHOD(SocketAddress, GetRemoteAddress, (), (const, override)); + MOCK_METHOD(int, Bind, (const SocketAddress&), (override)); + MOCK_METHOD(int, Connect, (const SocketAddress&), (override)); + MOCK_METHOD(int, Send, (const void*, size_t), (override)); + MOCK_METHOD(int, + SendTo, + (const void*, size_t, const SocketAddress&), + (override)); + MOCK_METHOD(int, Recv, (void*, size_t, int64_t*), (override)); + MOCK_METHOD(int, + RecvFrom, + (void*, size_t, SocketAddress*, int64_t*), + (override)); + MOCK_METHOD(int, Listen, (int), (override)); + MOCK_METHOD(int, Close, (), (override)); + MOCK_METHOD(int, GetError, (), (const, override)); + MOCK_METHOD(void, SetError, (int), (override)); + MOCK_METHOD(ConnState, GetState, (), (const, override)); + MOCK_METHOD(int, GetOption, (Option, int*), (override)); + MOCK_METHOD(int, SetOption, (Option, int), (override)); +}; + +class MockCertVerifier : public SSLCertificateVerifier { + public: + virtual ~MockCertVerifier() = default; + MOCK_METHOD(bool, Verify, (const SSLCertificate&), (override)); +}; + +} // namespace + +using ::testing::_; +using ::testing::Return; + +TEST(OpenSSLAdapterTest, TestTransformAlpnProtocols) { + EXPECT_EQ("", TransformAlpnProtocols(std::vector<std::string>())); + + // Protocols larger than 255 characters (whose size can't be fit in a byte), + // can't be converted, and an empty string will be returned. + std::string large_protocol(256, 'a'); + EXPECT_EQ("", + TransformAlpnProtocols(std::vector<std::string>{large_protocol})); + + // One protocol test. + std::vector<std::string> alpn_protos{"h2"}; + std::stringstream expected_response; + expected_response << static_cast<char>(2) << "h2"; + EXPECT_EQ(expected_response.str(), TransformAlpnProtocols(alpn_protos)); + + // Standard protocols test (h2,http/1.1). + alpn_protos.push_back("http/1.1"); + expected_response << static_cast<char>(8) << "http/1.1"; + EXPECT_EQ(expected_response.str(), TransformAlpnProtocols(alpn_protos)); +} + +// Verifies that SSLStart works when OpenSSLAdapter is started in standalone +// mode. +TEST(OpenSSLAdapterTest, TestBeginSSLBeforeConnection) { + rtc::AutoThread main_thread; + Socket* async_socket = new MockAsyncSocket(); + OpenSSLAdapter adapter(async_socket); + EXPECT_EQ(adapter.StartSSL("webrtc.org"), 0); +} + +// Verifies that the adapter factory can create new adapters. +TEST(OpenSSLAdapterFactoryTest, CreateSingleOpenSSLAdapter) { + rtc::AutoThread main_thread; + OpenSSLAdapterFactory adapter_factory; + Socket* async_socket = new MockAsyncSocket(); + auto simple_adapter = std::unique_ptr<OpenSSLAdapter>( + adapter_factory.CreateAdapter(async_socket)); + EXPECT_NE(simple_adapter, nullptr); +} + +// Verifies that setting a custom verifier still allows for adapters to be +// created. +TEST(OpenSSLAdapterFactoryTest, CreateWorksWithCustomVerifier) { + rtc::AutoThread main_thread; + MockCertVerifier* mock_verifier = new MockCertVerifier(); + EXPECT_CALL(*mock_verifier, Verify(_)).WillRepeatedly(Return(true)); + auto cert_verifier = std::unique_ptr<SSLCertificateVerifier>(mock_verifier); + + OpenSSLAdapterFactory adapter_factory; + adapter_factory.SetCertVerifier(cert_verifier.get()); + Socket* async_socket = new MockAsyncSocket(); + auto simple_adapter = std::unique_ptr<OpenSSLAdapter>( + adapter_factory.CreateAdapter(async_socket)); + EXPECT_NE(simple_adapter, nullptr); +} + +TEST(StrJoinTest, SingleElement) { + EXPECT_EQ(webrtc_openssl_adapter_internal::StrJoin({"a"}, ','), "a"); +} + +TEST(StrJoinTest, TwoElements) { + EXPECT_EQ(webrtc_openssl_adapter_internal::StrJoin({"first", "second"}, ':'), + "first:second"); +} + +TEST(StrJoinTest, WithEmptyElement) { + EXPECT_EQ( + webrtc_openssl_adapter_internal::StrJoin({"first", "", "second"}, ':'), + "first::second"); +} + +} // namespace rtc |