jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2022, Andreas Kling <kling@serenityos.org>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include "RequestManagerQt.h"
8#include <AK/JsonObject.h>
9#include <QNetworkCookie>
10
11RequestManagerQt::RequestManagerQt()
12{
13 m_qnam = new QNetworkAccessManager(this);
14
15 QObject::connect(m_qnam, &QNetworkAccessManager::finished, this, &RequestManagerQt::reply_finished);
16}
17
18void RequestManagerQt::reply_finished(QNetworkReply* reply)
19{
20 auto request = m_pending.get(reply).value();
21 m_pending.remove(reply);
22 request->did_finish();
23}
24
25RefPtr<Web::ResourceLoaderConnectorRequest> RequestManagerQt::start_request(DeprecatedString const& method, AK::URL const& url, HashMap<DeprecatedString, DeprecatedString> const& request_headers, ReadonlyBytes request_body, Core::ProxyData const& proxy)
26{
27 if (!url.scheme().is_one_of_ignoring_ascii_case("http"sv, "https"sv)) {
28 return nullptr;
29 }
30 auto request_or_error = Request::create(*m_qnam, method, url, request_headers, request_body, proxy);
31 if (request_or_error.is_error()) {
32 return nullptr;
33 }
34 auto request = request_or_error.release_value();
35 m_pending.set(&request->reply(), *request);
36 return request;
37}
38
39ErrorOr<NonnullRefPtr<RequestManagerQt::Request>> RequestManagerQt::Request::create(QNetworkAccessManager& qnam, DeprecatedString const& method, AK::URL const& url, HashMap<DeprecatedString, DeprecatedString> const& request_headers, ReadonlyBytes request_body, Core::ProxyData const&)
40{
41 QNetworkRequest request { QString(url.to_deprecated_string().characters()) };
42 request.setAttribute(QNetworkRequest::RedirectPolicyAttribute, QNetworkRequest::ManualRedirectPolicy);
43 request.setAttribute(QNetworkRequest::HttpPipeliningAllowedAttribute, true);
44 request.setAttribute(QNetworkRequest::CookieLoadControlAttribute, QNetworkRequest::Manual);
45 request.setAttribute(QNetworkRequest::CookieSaveControlAttribute, QNetworkRequest::Manual);
46
47 // NOTE: We disable HTTP2 as it's significantly slower (up to 5x, possibly more)
48 request.setAttribute(QNetworkRequest::Http2AllowedAttribute, false);
49
50 QNetworkReply* reply = nullptr;
51
52 for (auto& it : request_headers) {
53 // FIXME: We currently strip the Accept-Encoding header on outgoing requests from LibWeb
54 // since otherwise it'll ask for compression without Qt being aware of it.
55 // This is very hackish and I'm sure we can do it in concert with Qt somehow.
56 if (it.key == "Accept-Encoding")
57 continue;
58 request.setRawHeader(QByteArray(it.key.characters()), QByteArray(it.value.characters()));
59 }
60
61 if (method.equals_ignoring_ascii_case("head"sv)) {
62 reply = qnam.head(request);
63 } else if (method.equals_ignoring_ascii_case("get"sv)) {
64 reply = qnam.get(request);
65 } else if (method.equals_ignoring_ascii_case("post"sv)) {
66 reply = qnam.post(request, QByteArray((char const*)request_body.data(), request_body.size()));
67 } else if (method.equals_ignoring_ascii_case("put"sv)) {
68 reply = qnam.put(request, QByteArray((char const*)request_body.data(), request_body.size()));
69 } else if (method.equals_ignoring_ascii_case("delete"sv)) {
70 reply = qnam.deleteResource(request);
71 } else {
72 reply = qnam.sendCustomRequest(request, QByteArray(method.characters()), QByteArray((char const*)request_body.data(), request_body.size()));
73 }
74
75 return adopt_ref(*new Request(*reply));
76}
77
78RequestManagerQt::Request::Request(QNetworkReply& reply)
79 : m_reply(reply)
80{
81}
82
83RequestManagerQt::Request::~Request() = default;
84
85void RequestManagerQt::Request::did_finish()
86{
87 auto buffer = m_reply.readAll();
88 auto http_status_code = m_reply.attribute(QNetworkRequest::Attribute::HttpStatusCodeAttribute).toInt();
89 HashMap<DeprecatedString, DeprecatedString, CaseInsensitiveStringTraits> response_headers;
90 Vector<DeprecatedString> set_cookie_headers;
91 for (auto& it : m_reply.rawHeaderPairs()) {
92 auto name = DeprecatedString(it.first.data(), it.first.length());
93 auto value = DeprecatedString(it.second.data(), it.second.length());
94 if (name.equals_ignoring_ascii_case("set-cookie"sv)) {
95 // NOTE: Qt may have bundled multiple Set-Cookie headers into a single one.
96 // We have to extract the full list of cookies via QNetworkReply::header().
97 auto set_cookie_list = m_reply.header(QNetworkRequest::SetCookieHeader).value<QList<QNetworkCookie>>();
98 for (auto const& cookie : set_cookie_list) {
99 set_cookie_headers.append(cookie.toRawForm().data());
100 }
101 } else {
102 response_headers.set(name, value);
103 }
104 }
105 if (!set_cookie_headers.is_empty()) {
106 response_headers.set("set-cookie"sv, JsonArray { set_cookie_headers }.to_deprecated_string());
107 }
108 bool success = http_status_code != 0;
109 on_buffered_request_finish(success, buffer.length(), response_headers, http_status_code, ReadonlyBytes { buffer.data(), (size_t)buffer.size() });
110}