mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-01-07 19:57:45 +03:00
f883dc3eb0
Starting a gemini request creates a pipe for the output stream for the response, and a Core::Stream::File object is created from that pipe. Previously, the length of the response was computed by calling output_stream.size() which used lseek on the file descriptor. Doing that returned an error from lseek. Computing the value by counting the received bytes (via Gemini::Job::response_length) avoids that crash.
66 lines
2.2 KiB
C++
66 lines
2.2 KiB
C++
/*
|
|
* Copyright (c) 2020, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "ConnectionCache.h"
|
|
#include <LibCore/EventLoop.h>
|
|
#include <LibGemini/GeminiResponse.h>
|
|
#include <LibGemini/Job.h>
|
|
#include <RequestServer/GeminiRequest.h>
|
|
|
|
namespace RequestServer {
|
|
|
|
GeminiRequest::GeminiRequest(ConnectionFromClient& client, NonnullRefPtr<Gemini::Job> job, NonnullOwnPtr<Core::Stream::File>&& output_stream)
|
|
: Request(client, move(output_stream))
|
|
, m_job(move(job))
|
|
{
|
|
m_job->on_finish = [this](bool success) {
|
|
Core::deferred_invoke([url = m_job->url(), socket = m_job->socket()] {
|
|
ConnectionCache::request_did_finish(url, socket);
|
|
});
|
|
if (auto* response = m_job->response()) {
|
|
set_downloaded_size(MUST(m_job->response_length()));
|
|
if (!response->meta().is_empty()) {
|
|
HashMap<DeprecatedString, DeprecatedString, CaseInsensitiveStringTraits> headers;
|
|
headers.set("meta", response->meta());
|
|
// Note: We're setting content-type to meta only on status==SUCCESS
|
|
// we should perhaps have a better mechanism for this, since we
|
|
// are already shoehorning the concept of "headers" here
|
|
if (response->status() >= 20 && response->status() < 30) {
|
|
headers.set("content-type", response->meta());
|
|
}
|
|
set_response_headers(headers);
|
|
}
|
|
}
|
|
|
|
// signal 100% request progress so any listeners can react
|
|
// appropriately
|
|
did_progress(downloaded_size(), downloaded_size());
|
|
|
|
did_finish(success);
|
|
};
|
|
m_job->on_progress = [this](Optional<u32> total, u32 current) {
|
|
did_progress(move(total), current);
|
|
};
|
|
}
|
|
|
|
void GeminiRequest::set_certificate(DeprecatedString, DeprecatedString)
|
|
{
|
|
}
|
|
|
|
GeminiRequest::~GeminiRequest()
|
|
{
|
|
m_job->on_finish = nullptr;
|
|
m_job->on_progress = nullptr;
|
|
m_job->cancel();
|
|
}
|
|
|
|
NonnullOwnPtr<GeminiRequest> GeminiRequest::create_with_job(Badge<GeminiProtocol>, ConnectionFromClient& client, NonnullRefPtr<Gemini::Job> job, NonnullOwnPtr<Core::Stream::File>&& output_stream)
|
|
{
|
|
return adopt_own(*new GeminiRequest(client, move(job), move(output_stream)));
|
|
}
|
|
|
|
}
|