mirror of
https://github.com/SerenityOS/serenity
synced 2024-10-16 04:42:57 +00:00
e5192073d9
This allows us to use standard Serenity IPC infrastructure rather than manually creating FD-passing sockets. This also lets us use Serenity's WebDriver Session class, removing the copy previously used in Ladybird. This ensures any changes to Session in the future will be picked up by Ladybird for free.
131 lines
5.2 KiB
C++
131 lines
5.2 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>
|
|
#include <WebContent/PageHost.h>
|
|
#include <WebContent/WebDriverConnection.h>
|
|
|
|
static ErrorOr<void> load_content_filters();
|
|
|
|
extern DeprecatedString 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 ClientType>
|
|
static void proxy_socket_through_notifier(ClientType& client, QSocketNotifier& notifier)
|
|
{
|
|
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>());
|
|
}
|
|
|
|
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(DeprecatedString::formatted("{}/res/icons/16x16/app-browser.png", s_serenity_resource_root));
|
|
|
|
Web::Platform::FontPlugin::install(*new Ladybird::FontPluginQt);
|
|
|
|
Web::FrameLoader::set_error_page_url(DeprecatedString::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 };
|
|
DeprecatedString webdriver_content_ipc_path;
|
|
|
|
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.add_option(webdriver_content_ipc_path, "Path to WebDriver IPC for WebContent", "webdriver-content-path", 0, "path");
|
|
args_parser.parse(arguments);
|
|
|
|
VERIFY(webcontent_fd_passing_socket >= 0);
|
|
|
|
auto webcontent_socket = TRY(Core::take_over_socket_from_system_server("WebContent"sv));
|
|
auto webcontent_client = TRY(WebContent::ConnectionFromClient::try_create(move(webcontent_socket)));
|
|
webcontent_client->set_fd_passing_socket(TRY(Core::Stream::LocalSocket::adopt_fd(webcontent_fd_passing_socket)));
|
|
|
|
QSocketNotifier webcontent_notifier(QSocketNotifier::Type::Read);
|
|
proxy_socket_through_notifier(*webcontent_client, webcontent_notifier);
|
|
|
|
QSocketNotifier webdriver_notifier(QSocketNotifier::Type::Read);
|
|
RefPtr<WebContent::WebDriverConnection> webdriver_client;
|
|
if (!webdriver_content_ipc_path.is_empty()) {
|
|
webdriver_client = TRY(WebContent::WebDriverConnection::connect(webcontent_client->page_host(), webdriver_content_ipc_path));
|
|
proxy_socket_through_notifier(*webdriver_client, webdriver_notifier);
|
|
}
|
|
|
|
return app.exec();
|
|
}
|
|
|
|
static ErrorOr<void> load_content_filters()
|
|
{
|
|
auto file_or_error = Core::Stream::File::open(DeprecatedString::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(DeprecatedString::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 {};
|
|
}
|