Everywhere: Hoist the Libraries folder to the top-level

Author: https://github.com/trflynn89 Commit: 93712b24bf Pull-request: https://github.com/LadybirdBrowser/ladybird/pull/2256 Reviewed-by: https://github.com/sideshowbarker
2025-08-11 18:50:50 +00:00 · 2024-11-09 12:25:08 -05:00 · 2024-11-09 12:25:08 -05:00 · 93712b24bf · 2024-11-10 11:51:52 +00:00
commit 93712b24bf
parent 950e819ee7
4547 changed files with 104 additions and 113 deletions
--- a/Libraries/LibCore/LocalServer.cpp
+++ b/Libraries/LibCore/LocalServer.cpp
@ -0,0 +1,150 @@
+/*
+ * Copyright (c) 2018-2021, Andreas Kling <andreas@ladybird.org>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <LibCore/LocalServer.h>
+#include <LibCore/Notifier.h>
+#include <LibCore/SessionManagement.h>
+#include <LibCore/Socket.h>
+#include <LibCore/System.h>
+#include <LibCore/SystemServerTakeover.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <sys/socket.h>
+#include <sys/stat.h>
+#include <unistd.h>
+
+#ifndef SOCK_NONBLOCK
+#    include <sys/ioctl.h>
+#endif
+
+namespace Core {
+
+LocalServer::LocalServer(EventReceiver* parent)
+    : EventReceiver(parent)
+{
+}
+
+LocalServer::~LocalServer()
+{
+    if (m_fd >= 0)
+        ::close(m_fd);
+}
+
+ErrorOr<void> LocalServer::take_over_from_system_server(ByteString const& socket_path)
+{
+    if (m_listening)
+        return Error::from_string_literal("Core::LocalServer: Can't perform socket takeover when already listening");
+
+    auto const parsed_path = TRY(Core::SessionManagement::parse_path_with_sid(socket_path));
+    auto socket = TRY(take_over_socket_from_system_server(parsed_path));
+    m_fd = TRY(socket->release_fd());
+
+    m_listening = true;
+    setup_notifier();
+    return {};
+}
+
+ErrorOr<void> LocalServer::take_over_fd(int socket_fd)
+{
+    if (m_listening)
+        return Error::from_string_literal("Core::LocalServer: Can't perform socket takeover when already listening");
+
+    m_fd = socket_fd;
+    m_listening = true;
+    setup_notifier();
+    return {};
+}
+
+void LocalServer::setup_notifier()
+{
+    m_notifier = Notifier::construct(m_fd, Notifier::Type::Read, this);
+    m_notifier->on_activation = [this] {
+        if (on_accept) {
+            auto maybe_client_socket = accept();
+            if (maybe_client_socket.is_error()) {
+                dbgln("LocalServer::on_ready_to_read: Error accepting a connection: {}", maybe_client_socket.error());
+                if (on_accept_error)
+                    on_accept_error(maybe_client_socket.release_error());
+                return;
+            }
+
+            on_accept(maybe_client_socket.release_value());
+        }
+    };
+}
+
+bool LocalServer::listen(ByteString const& address)
+{
+    if (m_listening)
+        return false;
+
+    int rc;
+
+#ifdef SOCK_NONBLOCK
+    m_fd = socket(AF_LOCAL, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
+#else
+    m_fd = socket(AF_LOCAL, SOCK_STREAM, 0);
+    int option = 1;
+    ioctl(m_fd, FIONBIO, &option);
+    fcntl(m_fd, F_SETFD, FD_CLOEXEC);
+#endif
+    VERIFY(m_fd >= 0);
+#if !defined(AK_OS_MACOS) && !defined(AK_OS_IOS)
+    rc = fchmod(m_fd, 0600);
+    if (rc < 0) {
+        perror("fchmod");
+        VERIFY_NOT_REACHED();
+    }
+#endif
+
+    auto socket_address = SocketAddress::local(address);
+    auto un_optional = socket_address.to_sockaddr_un();
+    if (!un_optional.has_value()) {
+        perror("bind");
+        return false;
+    }
+    auto un = un_optional.value();
+    rc = ::bind(m_fd, (sockaddr const*)&un, sizeof(un));
+    if (rc < 0) {
+        perror("bind");
+        return false;
+    }
+
+    rc = ::listen(m_fd, 5);
+    if (rc < 0) {
+        perror("listen");
+        return false;
+    }
+
+    m_listening = true;
+    setup_notifier();
+    return true;
+}
+
+ErrorOr<NonnullOwnPtr<LocalSocket>> LocalServer::accept()
+{
+    VERIFY(m_listening);
+    sockaddr_un un;
+    socklen_t un_size = sizeof(un);
+#if !defined(AK_OS_MACOS) && !defined(AK_OS_IOS) && !defined(AK_OS_HAIKU)
+    int accepted_fd = ::accept4(m_fd, (sockaddr*)&un, &un_size, SOCK_NONBLOCK | SOCK_CLOEXEC);
+#else
+    int accepted_fd = ::accept(m_fd, (sockaddr*)&un, &un_size);
+#endif
+    if (accepted_fd < 0) {
+        return Error::from_syscall("accept"sv, -errno);
+    }
+
+#if defined(AK_OS_MACOS) || defined(AK_OS_IOS) || defined(AK_OS_HAIKU)
+    int option = 1;
+    ioctl(m_fd, FIONBIO, &option);
+    (void)fcntl(accepted_fd, F_SETFD, FD_CLOEXEC);
+#endif
+
+    return LocalSocket::adopt_fd(accepted_fd);
+}
+
+}