serenity/Ladybird/main.cpp

/*
 * Copyright (c) 2022, Andreas Kling <kling@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include "BrowserWindow.h"
#include "Settings.h"
#include "Utilities.h"
#include <Browser/CookieJar.h>
#include <Browser/Database.h>
#include <LibCore/ArgsParser.h>
#include <LibCore/EventLoop.h>
#include <LibCore/File.h>
#include <LibCore/Stream.h>
#include <LibCore/System.h>
#include <LibGfx/Font/FontDatabase.h>
#include <LibMain/Main.h>
#include <LibSQL/SQLClient.h>
#include <QApplication>

Browser::Settings* s_settings;

static ErrorOr<void> handle_attached_debugger()
{
#ifdef AK_OS_LINUX
    // Let's ignore SIGINT if we're being debugged because GDB
    // incorrectly forwards the signal to us even when it's set to
    // "nopass". See https://sourceware.org/bugzilla/show_bug.cgi?id=9425
    // for details.
    auto unbuffered_status_file = TRY(Core::Stream::File::open("/proc/self/status"sv, Core::Stream::OpenMode::Read));
    auto status_file = TRY(Core::Stream::BufferedFile::create(move(unbuffered_status_file)));
    auto buffer = TRY(ByteBuffer::create_uninitialized(4096));
    while (TRY(status_file->can_read_line())) {
        auto line = TRY(status_file->read_line(buffer));
        auto const parts = line.split_view(':');
        if (parts.size() < 2 || parts[0] != "TracerPid"sv)
            continue;
        auto tracer_pid = parts[1].to_uint<u32>();
        if (tracer_pid != 0UL) {
            dbgln("Debugger is attached, ignoring SIGINT");
            TRY(Core::System::signal(SIGINT, SIG_IGN));
        }
        break;
    }
#endif
    return {};
}

ErrorOr<NonnullRefPtr<Browser::Database>> create_database()
{
    int socket_fds[2] {};
    TRY(Core::System::socketpair(AF_LOCAL, SOCK_STREAM, 0, socket_fds));
    auto [browser_fd, sql_server_fd] = socket_fds;

    int fd_passing_socket_fds[2] {};
    TRY(Core::System::socketpair(AF_LOCAL, SOCK_STREAM, 0, fd_passing_socket_fds));
    auto [browser_fd_passing_fd, sql_server_fd_passing_fd] = fd_passing_socket_fds;

    auto sql_server_pid = TRY(Core::System::fork());

    if (sql_server_pid == 0) {
        TRY(Core::System::close(browser_fd_passing_fd));
        TRY(Core::System::close(browser_fd));

        DeprecatedString takeover_string;
        if (auto* socket_takeover = getenv("SOCKET_TAKEOVER"))
            takeover_string = DeprecatedString::formatted("{} SQLServer:{}", socket_takeover, sql_server_fd);
        else
            takeover_string = DeprecatedString::formatted("SQLServer:{}", sql_server_fd);
        TRY(Core::System::setenv("SOCKET_TAKEOVER"sv, takeover_string, true));

        auto sql_server_fd_passing_fd_string = DeprecatedString::number(sql_server_fd_passing_fd);

        char const* argv[] = {
            "SQLServer",
            "--sql-server-fd-passing-socket",
            sql_server_fd_passing_fd_string.characters(),
            nullptr,
        };

        if (execvp("./SQLServer/SQLServer", const_cast<char**>(argv)) < 0)
            perror("execvp");
        VERIFY_NOT_REACHED();
    }

    TRY(Core::System::close(sql_server_fd_passing_fd));
    TRY(Core::System::close(sql_server_fd));

    auto socket = TRY(Core::Stream::LocalSocket::adopt_fd(browser_fd));
    TRY(socket->set_blocking(true));

    auto sql_client = TRY(adopt_nonnull_ref_or_enomem(new (nothrow) SQL::SQLClient(std::move(socket))));
    sql_client->set_fd_passing_socket(TRY(Core::Stream::LocalSocket::adopt_fd(browser_fd_passing_fd)));

    return Browser::Database::create(move(sql_client));
}

ErrorOr<int> serenity_main(Main::Arguments arguments)
{
    // NOTE: This is only used for the Core::Socket inside the IPC connections.
    // FIXME: Refactor things so we can get rid of this somehow.
    Core::EventLoop event_loop;

    TRY(handle_attached_debugger());

    QApplication app(arguments.argc, arguments.argv);

    platform_init();

    // NOTE: We only instantiate this to ensure that Gfx::FontDatabase has its default queries initialized.
    Gfx::FontDatabase::set_default_font_query("Katica 10 400 0");
    Gfx::FontDatabase::set_fixed_width_font_query("Csilla 10 400 0");

    StringView raw_url;
    int webdriver_fd_passing_socket { -1 };

    Core::ArgsParser args_parser;
    args_parser.set_general_help("The Ladybird web browser :^)");
    args_parser.add_positional_argument(raw_url, "URL to open", "url", Core::ArgsParser::Required::No);
    args_parser.add_option(webdriver_fd_passing_socket, "File descriptor of the passing socket for the WebDriver connection", "webdriver-fd-passing-socket", 'd', "webdriver_fd_passing_socket");
    args_parser.parse(arguments);

    auto database = TRY(create_database());
    auto cookie_jar = TRY(Browser::CookieJar::create(*database));

    BrowserWindow window(cookie_jar, webdriver_fd_passing_socket);
    s_settings = new Browser::Settings(&window);
    window.setWindowTitle("Ladybird");
    window.resize(800, 600);
    window.show();

    URL url = raw_url;
    if (Core::File::exists(raw_url))
        url = URL::create_with_file_scheme(Core::File::real_path_for(raw_url));
    else if (!url.is_valid())
        url = DeprecatedString::formatted("http://{}", raw_url);

    if (url.is_valid())
        window.view().load(url);

    return app.exec();
}