mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-10-19 07:33:20 +00:00
9b8f6b8108
If transferring a cached response body fails for any reason, we will now issue a network request instead of failing the request outright. The catch here is that we will have already transferred the response code and headers to the client, and potentially some of the body. So we attempt to only request the remaining data over the network using a range request. This feels a bit sketchy, but this is also how Chromium behaves. However, the server may or may not support range requests. If they do, we can expect an HTTP 206 response with the bytes we need. If not, we will receive an HTTP 200 (assuming the request succeeded), along with the entire object's body. In this case, we also behave like Chromium, and internally drop number of bytes we had already transferred.
92 lines
3.7 KiB
C++
92 lines
3.7 KiB
C++
/*
|
|
* Copyright (c) 2018-2020, Andreas Kling <andreas@ladybird.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/HashMap.h>
|
|
#include <LibDNS/Resolver.h>
|
|
#include <LibIPC/ConnectionFromClient.h>
|
|
#include <LibWebSocket/WebSocket.h>
|
|
#include <RequestServer/RequestClientEndpoint.h>
|
|
#include <RequestServer/RequestServerEndpoint.h>
|
|
|
|
namespace RequestServer {
|
|
|
|
struct Resolver : public RefCounted<Resolver>
|
|
, Weakable<Resolver> {
|
|
Resolver(Function<ErrorOr<DNS::Resolver::SocketResult>()> create_socket)
|
|
: dns(move(create_socket))
|
|
{
|
|
}
|
|
|
|
DNS::Resolver dns;
|
|
};
|
|
|
|
class ConnectionFromClient final
|
|
: public IPC::ConnectionFromClient<RequestClientEndpoint, RequestServerEndpoint> {
|
|
C_OBJECT(ConnectionFromClient);
|
|
|
|
public:
|
|
~ConnectionFromClient() override;
|
|
|
|
virtual void die() override;
|
|
|
|
private:
|
|
explicit ConnectionFromClient(NonnullOwnPtr<IPC::Transport>);
|
|
|
|
virtual Messages::RequestServer::InitTransportResponse init_transport(int peer_pid) override;
|
|
virtual Messages::RequestServer::ConnectNewClientResponse connect_new_client() override;
|
|
virtual Messages::RequestServer::ConnectNewClientsResponse connect_new_clients(size_t count) override;
|
|
|
|
virtual Messages::RequestServer::IsSupportedProtocolResponse is_supported_protocol(ByteString) override;
|
|
virtual void set_dns_server(ByteString host_or_address, u16 port, bool use_tls, bool validate_dnssec_locally) override;
|
|
virtual void set_use_system_dns() override;
|
|
virtual void start_request(i32 request_id, ByteString, URL::URL, HTTP::HeaderMap, ByteBuffer, Core::ProxyData) override;
|
|
virtual Messages::RequestServer::StopRequestResponse stop_request(i32) override;
|
|
virtual Messages::RequestServer::SetCertificateResponse set_certificate(i32, ByteString, ByteString) override;
|
|
virtual void ensure_connection(URL::URL url, ::RequestServer::CacheLevel cache_level) override;
|
|
|
|
virtual void clear_cache() override;
|
|
|
|
virtual void websocket_connect(i64 websocket_id, URL::URL, ByteString, Vector<ByteString>, Vector<ByteString>, HTTP::HeaderMap) override;
|
|
virtual void websocket_send(i64 websocket_id, bool, ByteBuffer) override;
|
|
virtual void websocket_close(i64 websocket_id, u16, ByteString) override;
|
|
virtual Messages::RequestServer::WebsocketSetCertificateResponse websocket_set_certificate(i64, ByteString, ByteString) override;
|
|
|
|
struct ResumeRequestForFailedCacheEntry {
|
|
size_t start_offset { 0 };
|
|
int writer_fd { 0 };
|
|
};
|
|
void issue_network_request(i32 request_id, ByteString, URL::URL, HTTP::HeaderMap, ByteBuffer, Core::ProxyData, Optional<ResumeRequestForFailedCacheEntry> = {});
|
|
|
|
HashMap<i32, RefPtr<WebSocket::WebSocket>> m_websockets;
|
|
|
|
struct ActiveRequest;
|
|
friend struct ActiveRequest;
|
|
|
|
static ErrorOr<IPC::File> create_client_socket();
|
|
|
|
static int on_socket_callback(void*, int sockfd, int what, void* user_data, void*);
|
|
static int on_timeout_callback(void*, long timeout_ms, void* user_data);
|
|
static size_t on_header_received(void* buffer, size_t size, size_t nmemb, void* user_data);
|
|
static size_t on_data_received(void* buffer, size_t size, size_t nmemb, void* user_data);
|
|
|
|
HashMap<i32, NonnullOwnPtr<ActiveRequest>> m_active_requests;
|
|
|
|
void check_active_requests();
|
|
void* m_curl_multi { nullptr };
|
|
RefPtr<Core::Timer> m_timer;
|
|
HashMap<int, NonnullRefPtr<Core::Notifier>> m_read_notifiers;
|
|
HashMap<int, NonnullRefPtr<Core::Notifier>> m_write_notifiers;
|
|
NonnullRefPtr<Resolver> m_resolver;
|
|
ByteString m_alt_svc_cache_path;
|
|
};
|
|
|
|
// FIXME: Find a good home for this
|
|
ByteString build_curl_resolve_list(DNS::LookupResult const&, StringView host, u16 port);
|
|
constexpr inline uintptr_t websocket_private_tag = 0x1;
|
|
|
|
}
|