RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-30 18:28:10 +00:00
Code Issues Activity Actions

LibCore: Add Content-Encoding handling to CHttpJob

Browse source
This commit is contained in:
Marcel Schneider 2019-11-09 18:33:37 +01:00 • committed by Andreas Kling
parent 4fe5503b17
commit c40935e79f
1 changed files with 34 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

35
Libraries/LibCore/CHttpJob.cpp
View file

@ -4,8 +4,37 @@
#include <stdio.h> #include <stdio.h>
#include <unistd.h> #include <unistd.h>
#include <CGzip.h>
#define CHTTPJOB_DEBUG #define CHTTPJOB_DEBUG
static ByteBuffer handle_content_encoding(const ByteBuffer& buf, const String& content_encoding)
{
dbg() << "CHttpJob::handle_content_encoding: buf has content_encoding = " << content_encoding;
if (content_encoding == "gzip") {
if (!Gzip::is_compressed(buf)) {
dbg() << "CHttpJob::handle_content_encoding: buf is not gzip compressed!";
}
dbg() << "CHttpJob::handle_content_encoding: buf is gzip compressed!";
auto uncompressed = Gzip::decompress(buf);
if (!uncompressed.has_value()) {
dbg() << "CHttpJob::handle_content_encoding: Gzip::decompress() failed. Returning original buffer.";
return buf;
}
dbg() << "CHttpJob::handle_content_encoding: Gzip::decompress() successful.\n"
<< " Input size = " << buf.size() << "\n"
<< " Output size = " << uncompressed.value().size();
return uncompressed.value();
}
return buf;
}
CHttpJob::CHttpJob(const CHttpRequest& request) CHttpJob::CHttpJob(const CHttpRequest& request)
: m_request(request) : m_request(request)
{ {
@ -113,6 +142,11 @@ void CHttpJob::finish_up()
} }
m_received_buffers.clear(); m_received_buffers.clear();
auto content_encoding = m_headers.get("Content-Encoding");
if (content_encoding.has_value()) {
flattened_buffer = handle_content_encoding(flattened_buffer, content_encoding.value());
}
auto response = CHttpResponse::create(m_code, move(m_headers), move(flattened_buffer)); auto response = CHttpResponse::create(m_code, move(m_headers), move(flattened_buffer));
deferred_invoke([this, response](auto&) { deferred_invoke([this, response](auto&) {
did_finish(move(response)); did_finish(move(response));
@ -146,4 +180,3 @@ void CHttpJob::shutdown()
remove_child(*m_socket); remove_child(*m_socket);
m_socket = nullptr; m_socket = nullptr;
} }