Ladybird: Add WebSocket server for use by Lagom networking

Hide its use behind the same flag as RequestServer in WebContent.

Ladybird/CMakeLists.txt

@@ -144,8 +144,9 @@ add_custom_target(debug${LADYBIRD_CUSTOM_TARGET_SUFFIX}
 add_subdirectory(SQLServer)
 add_subdirectory(WebContent)
 add_subdirectory(WebDriver)
+add_subdirectory(WebSocket)
 add_subdirectory(RequestServer)
-add_dependencies(ladybird SQLServer WebContent WebDriver RequestServer headless-browser)
+add_dependencies(ladybird SQLServer WebContent WebDriver WebSocketServer RequestServer headless-browser)
 
 if (APPLE)
     # FIXME: Create a proper app bundle for each helper process
@@ -156,6 +157,7 @@ if (APPLE)
         COMMAND "${CMAKE_COMMAND}" -E copy_if_different "$<TARGET_FILE:SQLServer>" "${app_dir}"
         COMMAND "${CMAKE_COMMAND}" -E copy_if_different "$<TARGET_FILE:WebContent>" "${app_dir}"
         COMMAND "${CMAKE_COMMAND}" -E copy_if_different "$<TARGET_FILE:WebDriver>" "${app_dir}"
+        COMMAND "${CMAKE_COMMAND}" -E copy_if_different "$<TARGET_FILE:WebSocketServer>" "${app_dir}"
         COMMAND "${CMAKE_COMMAND}" -E copy_if_different "$<TARGET_FILE:headless-browser>" "${app_dir}"
     )
 endif()

Ladybird/HelperProcess.cpp

@@ -87,36 +87,37 @@ ErrorOr<NonnullRefPtr<WebView::WebContentClient>> launch_web_content_process(Web
     return new_client;
 }
 
-ErrorOr<NonnullRefPtr<Protocol::RequestClient>> launch_request_server_process(ReadonlySpan<String> candidate_request_server_paths, StringView serenity_resource_root)
+template<typename Client>
+ErrorOr<NonnullRefPtr<Client>> launch_generic_server_process(ReadonlySpan<String> candidate_server_paths, StringView serenity_resource_root, StringView server_name)
 {
     int socket_fds[2] {};
     TRY(Core::System::socketpair(AF_LOCAL, SOCK_STREAM, 0, socket_fds));
 
     int ui_fd = socket_fds[0];
-    int rc_fd = socket_fds[1];
+    int server_fd = socket_fds[1];
 
     int fd_passing_socket_fds[2] {};
     TRY(Core::System::socketpair(AF_LOCAL, SOCK_STREAM, 0, fd_passing_socket_fds));
 
     int ui_fd_passing_fd = fd_passing_socket_fds[0];
-    int rc_fd_passing_fd = fd_passing_socket_fds[1];
+    int server_fd_passing_fd = fd_passing_socket_fds[1];
 
     if (auto child_pid = TRY(Core::System::fork()); child_pid == 0) {
         TRY(Core::System::close(ui_fd));
         TRY(Core::System::close(ui_fd_passing_fd));
 
-        auto takeover_string = TRY(String::formatted("RequestServer:{}", rc_fd));
+        auto takeover_string = TRY(String::formatted("{}:{}", server_name, server_fd));
         TRY(Core::System::setenv("SOCKET_TAKEOVER"sv, takeover_string, true));
 
-        auto fd_passing_socket_string = TRY(String::number(rc_fd_passing_fd));
+        auto fd_passing_socket_string = TRY(String::number(server_fd_passing_fd));
 
         ErrorOr<void> result;
-        for (auto const& path : candidate_request_server_paths) {
+        for (auto const& path : candidate_server_paths) {
 
             if (Core::System::access(path, X_OK).is_error())
                 continue;
 
-            auto arguments = Vector {
+            auto arguments = Vector<StringView, 5> {
                 path.bytes_as_string_view(),
                 "--fd-passing-socket"sv,
                 fd_passing_socket_string,
@@ -130,18 +131,28 @@ ErrorOr<NonnullRefPtr<Protocol::RequestClient>> launch_request_server_process(Re
         }
 
         if (result.is_error())
-            warnln("Could not launch any of {}: {}", candidate_request_server_paths, result.error());
+            warnln("Could not launch any of {}: {}", candidate_server_paths, result.error());
         VERIFY_NOT_REACHED();
     }
 
-    TRY(Core::System::close(rc_fd));
-    TRY(Core::System::close(rc_fd_passing_fd));
+    TRY(Core::System::close(server_fd));
+    TRY(Core::System::close(server_fd_passing_fd));
 
     auto socket = TRY(Core::LocalSocket::adopt_fd(ui_fd));
     TRY(socket->set_blocking(true));
 
-    auto new_client = TRY(try_make_ref_counted<Protocol::RequestClient>(move(socket)));
+    auto new_client = TRY(try_make_ref_counted<Client>(move(socket)));
     new_client->set_fd_passing_socket(TRY(Core::LocalSocket::adopt_fd(ui_fd_passing_fd)));
 
     return new_client;
 }
+
+ErrorOr<NonnullRefPtr<Protocol::RequestClient>> launch_request_server_process(ReadonlySpan<String> candidate_request_server_paths, StringView serenity_resource_root)
+{
+    return launch_generic_server_process<Protocol::RequestClient>(candidate_request_server_paths, serenity_resource_root, "RequestServer"sv);
+}
+
+ErrorOr<NonnullRefPtr<Protocol::WebSocketClient>> launch_web_socket_process(ReadonlySpan<String> candidate_web_socket_paths, StringView serenity_resource_root)
+{
+    return launch_generic_server_process<Protocol::WebSocketClient>(candidate_web_socket_paths, serenity_resource_root, "WebSocket"sv);
+}

Ladybird/HelperProcess.h

@@ -11,6 +11,7 @@
 #include <AK/Span.h>
 #include <AK/StringView.h>
 #include <LibProtocol/RequestClient.h>
+#include <LibProtocol/WebSocketClient.h>
 #include <LibWebView/ViewImplementation.h>
 #include <LibWebView/WebContentClient.h>
 
@@ -22,3 +23,4 @@ ErrorOr<NonnullRefPtr<WebView::WebContentClient>> launch_web_content_process(Web
     Ladybird::UseLagomNetworking);
 
 ErrorOr<NonnullRefPtr<Protocol::RequestClient>> launch_request_server_process(ReadonlySpan<String> candidate_request_server_paths, StringView serenity_resource_root);
+ErrorOr<NonnullRefPtr<Protocol::WebSocketClient>> launch_web_socket_process(ReadonlySpan<String> candidate_web_socket_paths, StringView serenity_resource_root);

Ladybird/RequestServer/main.cpp

@@ -20,6 +20,7 @@
 #include <RequestServer/HttpProtocol.h>
 #include <RequestServer/HttpsProtocol.h>
 
+// FIXME: Share b/w RequestServer and WebSocket
 ErrorOr<String> find_certificates(StringView serenity_resource_root)
 {
     auto cert_path = TRY(String::formatted("{}/res/ladybird/cacert.pem", serenity_resource_root));

Ladybird/WebContent/main.cpp

@@ -30,6 +30,7 @@
 #include <LibWeb/Platform/EventLoopPluginSerenity.h>
 #include <LibWeb/WebSockets/WebSocket.h>
 #include <LibWebView/RequestServerAdapter.h>
+#include <LibWebView/WebSocketClientAdapter.h>
 #include <QCoreApplication>
 #include <WebContent/ConnectionFromClient.h>
 #include <WebContent/PageHost.h>
@@ -56,8 +57,6 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
         return Ladybird::AudioCodecPluginQt::create(move(loader));
     });
 
-    Web::WebSockets::WebSocketClientManager::initialize(Ladybird::WebSocketClientManagerQt::create());
-
     Web::FrameLoader::set_default_favicon_path(DeprecatedString::formatted("{}/res/icons/16x16/app-browser.png", s_serenity_resource_root));
 
     int webcontent_fd_passing_socket { -1 };
@@ -74,10 +73,15 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
 
     if (use_lagom_networking) {
         auto candidate_request_server_paths = TRY(get_paths_for_helper_process("RequestServer"sv));
-        auto protocol_client = TRY(launch_request_server_process(candidate_request_server_paths, s_serenity_resource_root));
-        Web::ResourceLoader::initialize(TRY(WebView::RequestServerAdapter::try_create(move(protocol_client))));
+        auto request_server_client = TRY(launch_request_server_process(candidate_request_server_paths, s_serenity_resource_root));
+        Web::ResourceLoader::initialize(TRY(WebView::RequestServerAdapter::try_create(move(request_server_client))));
+
+        auto candidate_web_socket_paths = TRY(get_paths_for_helper_process("WebSocket"sv));
+        auto web_socket_client = TRY(launch_web_socket_process(candidate_web_socket_paths, s_serenity_resource_root));
+        Web::WebSockets::WebSocketClientManager::initialize(TRY(WebView::WebSocketClientManagerAdapter::try_create(move(web_socket_client))));
     } else {
         Web::ResourceLoader::initialize(Ladybird::RequestManagerQt::create());
+        Web::WebSockets::WebSocketClientManager::initialize(Ladybird::WebSocketClientManagerQt::create());
     }
 
     JS::Bytecode::Interpreter::set_enabled(use_javascript_bytecode);

Ladybird/WebSocket/CMakeLists.txt

@@ -0,0 +1,8 @@
+set(SOURCES
+    "${SERENITY_SOURCE_DIR}/Userland/Services/WebSocket/ConnectionFromClient.cpp"
+    main.cpp
+)
+
+add_executable(WebSocketServer ${SOURCES})
+set_target_properties(WebSocketServer PROPERTIES OUTPUT_NAME WebSocket)
+target_link_libraries(WebSocketServer PRIVATE LibCore LibFileSystem LibIPC LibMain LibTLS LibWebSocket LibWebView)

Ladybird/WebSocket/main.cpp

@@ -0,0 +1,53 @@
+/*
+ * Copyright (c) 2021, Dex♪ <dexes.ttp@gmail.com>
+ * Copyright (c) 2023, Andrew Kaster <akaster@serenityos.org>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <AK/LexicalPath.h>
+#include <LibCore/ArgsParser.h>
+#include <LibCore/EventLoop.h>
+#include <LibCore/LocalServer.h>
+#include <LibCore/System.h>
+#include <LibFileSystem/FileSystem.h>
+#include <LibIPC/SingleServer.h>
+#include <LibMain/Main.h>
+#include <LibTLS/Certificate.h>
+#include <WebSocket/ConnectionFromClient.h>
+
+// FIXME: Share b/w RequestServer and WebSocket
+ErrorOr<String> find_certificates(StringView serenity_resource_root)
+{
+    auto cert_path = TRY(String::formatted("{}/res/ladybird/cacert.pem", serenity_resource_root));
+    if (!FileSystem::exists(cert_path)) {
+        auto app_dir = LexicalPath::dirname(TRY(Core::System::current_executable_path()).to_deprecated_string());
+
+        cert_path = TRY(String::formatted("{}/cacert.pem", LexicalPath(app_dir).parent()));
+        if (!FileSystem::exists(cert_path))
+            return Error::from_string_view("Don't know how to load certs!"sv);
+    }
+    return cert_path;
+}
+
+ErrorOr<int> serenity_main(Main::Arguments arguments)
+{
+    int fd_passing_socket { -1 };
+    StringView serenity_resource_root;
+
+    Core::ArgsParser args_parser;
+    args_parser.add_option(fd_passing_socket, "File descriptor of the fd passing socket", "fd-passing-socket", 'c', "fd-passing-socket");
+    args_parser.add_option(serenity_resource_root, "Absolute path to directory for serenity resources", "serenity-resource-root", 'r', "serenity-resource-root");
+    args_parser.parse(arguments);
+
+    // Ensure the certificates are read out here.
+    DefaultRootCACertificates::set_default_certificate_path(TRY(find_certificates(serenity_resource_root)));
+    [[maybe_unused]] auto& certs = DefaultRootCACertificates::the();
+
+    Core::EventLoop event_loop;
+
+    auto client = TRY(IPC::take_over_accepted_client_from_system_server<WebSocket::ConnectionFromClient>());
+    client->set_fd_passing_socket(TRY(Core::LocalSocket::adopt_fd(fd_passing_socket)));
+
+    return event_loop.exec();
+}

Ladybird/cmake/InstallRules.cmake

@@ -4,7 +4,7 @@ include(GNUInstallDirs)
 
 set(package ladybird)
 
-set(ladybird_applications ladybird SQLServer WebContent WebDriver RequestServer headless-browser)
+set(ladybird_applications ladybird SQLServer WebContent WebDriver WebSocketServer RequestServer headless-browser)
 
 set(app_install_targets ${ladybird_applications})
 if (ANDROID)