mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-10-14 12:09:45 +00:00
e636851481
Instead of using a HashMap<ByteString, ByteString, CaseInsensitive...> everywhere, we now encapsulate this in a class. Even better, the new class also allows keeping track of multiple headers with the same name! This will make it possible for HTTP responses to actually retain all their headers on the perilous journey from RequestServer to LibWeb.
75 lines
2 KiB
C++
75 lines
2 KiB
C++
/*
|
|
* Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
|
|
* Copyright (c) 2022, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/Function.h>
|
|
#include <AK/Stream.h>
|
|
#include <LibCore/EventReceiver.h>
|
|
#include <LibCore/Forward.h>
|
|
#include <LibHTTP/HeaderMap.h>
|
|
|
|
namespace Core {
|
|
|
|
class NetworkJob : public EventReceiver {
|
|
C_OBJECT_ABSTRACT(NetworkJob)
|
|
public:
|
|
enum class Error {
|
|
None,
|
|
ConnectionFailed,
|
|
TransmissionFailed,
|
|
ProtocolFailed,
|
|
Cancelled,
|
|
};
|
|
virtual ~NetworkJob() override = default;
|
|
|
|
// Could fire twice, after Headers and after Trailers!
|
|
Function<void(HTTP::HeaderMap const& response_headers, Optional<u32> response_code)> on_headers_received;
|
|
Function<void(bool success)> on_finish;
|
|
Function<void(Optional<u64>, u64)> on_progress;
|
|
|
|
bool is_cancelled() const { return m_error == Error::Cancelled; }
|
|
bool has_error() const { return m_error != Error::None; }
|
|
Error error() const { return m_error; }
|
|
NetworkResponse* response() { return m_response.ptr(); }
|
|
NetworkResponse const* response() const { return m_response.ptr(); }
|
|
|
|
enum class ShutdownMode {
|
|
DetachFromSocket,
|
|
CloseSocket,
|
|
};
|
|
virtual void start(Core::BufferedSocketBase&) = 0;
|
|
virtual void shutdown(ShutdownMode) = 0;
|
|
virtual void fail(Error error) { did_fail(error); }
|
|
|
|
void cancel()
|
|
{
|
|
shutdown(ShutdownMode::DetachFromSocket);
|
|
m_error = Error::Cancelled;
|
|
}
|
|
|
|
protected:
|
|
NetworkJob(Stream&);
|
|
void did_finish(NonnullRefPtr<NetworkResponse>&&);
|
|
void did_fail(Error);
|
|
void did_progress(Optional<u64> total_size, u64 downloaded);
|
|
|
|
ErrorOr<size_t> do_write(ReadonlyBytes bytes) { return m_output_stream.write_some(bytes); }
|
|
|
|
private:
|
|
RefPtr<NetworkResponse> m_response;
|
|
Stream& m_output_stream;
|
|
Error m_error { Error::None };
|
|
};
|
|
|
|
char const* to_string(NetworkJob::Error);
|
|
|
|
}
|
|
|
|
template<>
|
|
struct AK::Formatter<Core::NetworkJob> : Formatter<Core::EventReceiver> {
|
|
};
|