mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-12-07 21:59:54 +00:00
7021d30288
Rather than needing to set another environment variable for WebDriver's passing socket, let's forward these FDs by command line. This also moves the creation of the WebContent connection to a helper function so that the WebDriver connection can re-use it.
120 lines
4.7 KiB
C++
120 lines
4.7 KiB
C++
/*
|
|
* Copyright (c) 2020-2022, Andreas Kling <kling@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#define AK_DONT_REPLACE_STD
|
|
|
|
#include "../EventLoopPluginQt.h"
|
|
#include "../FontPluginQt.h"
|
|
#include "../ImageCodecPluginLadybird.h"
|
|
#include "../RequestManagerQt.h"
|
|
#include "../Utilities.h"
|
|
#include "../WebSocketClientManagerLadybird.h"
|
|
#include <AK/LexicalPath.h>
|
|
#include <LibCore/ArgsParser.h>
|
|
#include <LibCore/EventLoop.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibCore/LocalServer.h>
|
|
#include <LibCore/System.h>
|
|
#include <LibCore/SystemServerTakeover.h>
|
|
#include <LibIPC/ConnectionFromClient.h>
|
|
#include <LibMain/Main.h>
|
|
#include <LibWeb/Loader/ContentFilter.h>
|
|
#include <LibWeb/Loader/FrameLoader.h>
|
|
#include <LibWeb/Loader/ResourceLoader.h>
|
|
#include <LibWeb/WebSockets/WebSocket.h>
|
|
#include <QGuiApplication>
|
|
#include <QSocketNotifier>
|
|
#include <QTimer>
|
|
#include <WebContent/ConnectionFromClient.h>
|
|
|
|
static ErrorOr<void> load_content_filters();
|
|
|
|
extern String s_serenity_resource_root;
|
|
|
|
struct DeferredInvokerQt final : IPC::DeferredInvoker {
|
|
virtual ~DeferredInvokerQt() = default;
|
|
virtual void schedule(Function<void()> callback) override
|
|
{
|
|
QTimer::singleShot(0, move(callback));
|
|
}
|
|
};
|
|
|
|
template<typename ConnectionType, typename... Args>
|
|
static ErrorOr<NonnullRefPtr<ConnectionType>> create_connection_from_passed_socket(int passing_socket_fd, StringView socket_name, QSocketNotifier& notifier, Args&&... args)
|
|
{
|
|
auto socket = TRY(Core::take_over_socket_from_system_server(socket_name));
|
|
auto client = TRY(ConnectionType::try_create(move(socket), std::forward<Args>(args)...));
|
|
|
|
VERIFY(passing_socket_fd >= 0);
|
|
client->set_fd_passing_socket(TRY(Core::Stream::LocalSocket::adopt_fd(passing_socket_fd)));
|
|
|
|
notifier.setSocket(client->socket().fd().value());
|
|
notifier.setEnabled(true);
|
|
|
|
QObject::connect(¬ifier, &QSocketNotifier::activated, [client]() mutable {
|
|
client->socket().notifier()->on_ready_to_read();
|
|
});
|
|
|
|
client->set_deferred_invoker(make<DeferredInvokerQt>());
|
|
return client;
|
|
}
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
|
{
|
|
// NOTE: This is only used for the Core::Socket inside the IPC connection.
|
|
// FIXME: Refactor things so we can get rid of this somehow.
|
|
Core::EventLoop event_loop;
|
|
|
|
QGuiApplication app(arguments.argc, arguments.argv);
|
|
|
|
platform_init();
|
|
|
|
Web::Platform::EventLoopPlugin::install(*new Ladybird::EventLoopPluginQt);
|
|
Web::Platform::ImageCodecPlugin::install(*new Ladybird::ImageCodecPluginLadybird);
|
|
|
|
Web::ResourceLoader::initialize(RequestManagerQt::create());
|
|
Web::WebSockets::WebSocketClientManager::initialize(Ladybird::WebSocketClientManagerLadybird::create());
|
|
|
|
Web::FrameLoader::set_default_favicon_path(String::formatted("{}/res/icons/16x16/app-browser.png", s_serenity_resource_root));
|
|
|
|
Web::Platform::FontPlugin::install(*new Ladybird::FontPluginQt);
|
|
|
|
Web::FrameLoader::set_error_page_url(String::formatted("file://{}/res/html/error.html", s_serenity_resource_root));
|
|
|
|
auto maybe_content_filter_error = load_content_filters();
|
|
if (maybe_content_filter_error.is_error())
|
|
dbgln("Failed to load content filters: {}", maybe_content_filter_error.error());
|
|
|
|
int webcontent_fd_passing_socket { -1 };
|
|
|
|
Core::ArgsParser args_parser;
|
|
args_parser.add_option(webcontent_fd_passing_socket, "File descriptor of the passing socket for the WebContent connection", "webcontent-fd-passing-socket", 'c', "webcontent_fd_passing_socket");
|
|
args_parser.parse(arguments);
|
|
|
|
QSocketNotifier webcontent_notifier(QSocketNotifier::Type::Read);
|
|
auto webcontent_client = TRY(create_connection_from_passed_socket<WebContent::ConnectionFromClient>(webcontent_fd_passing_socket, "WebContent"sv, webcontent_notifier));
|
|
|
|
return app.exec();
|
|
}
|
|
|
|
static ErrorOr<void> load_content_filters()
|
|
{
|
|
auto file_or_error = Core::Stream::File::open(String::formatted("{}/home/anon/.config/BrowserContentFilters.txt", s_serenity_resource_root), Core::Stream::OpenMode::Read);
|
|
if (file_or_error.is_error())
|
|
file_or_error = Core::Stream::File::open(String::formatted("{}/res/ladybird/BrowserContentFilters.txt", s_serenity_resource_root), Core::Stream::OpenMode::Read);
|
|
if (file_or_error.is_error())
|
|
return file_or_error.release_error();
|
|
auto file = file_or_error.release_value();
|
|
auto ad_filter_list = TRY(Core::Stream::BufferedFile::create(move(file)));
|
|
auto buffer = TRY(ByteBuffer::create_uninitialized(4096));
|
|
while (TRY(ad_filter_list->can_read_line())) {
|
|
auto line = TRY(ad_filter_list->read_line(buffer));
|
|
if (!line.is_empty()) {
|
|
Web::ContentFilter::the().add_pattern(line);
|
|
}
|
|
}
|
|
return {};
|
|
}
|