mirror of
https://github.com/The-Powder-Toy/The-Powder-Toy.git
synced 2025-08-16 21:34:01 +02:00
Fix parsing HTTP headers after redirects
This commit is contained in:
@@ -63,6 +63,7 @@ namespace http
|
|||||||
char curlErrorBuffer[CURL_ERROR_SIZE];
|
char curlErrorBuffer[CURL_ERROR_SIZE];
|
||||||
bool curlAddedToMulti = false;
|
bool curlAddedToMulti = false;
|
||||||
bool gotStatusLine = false;
|
bool gotStatusLine = false;
|
||||||
|
bool gotAllHeaders = false;
|
||||||
|
|
||||||
RequestHandleHttp() : RequestHandle(CtorTag{})
|
RequestHandleHttp() : RequestHandle(CtorTag{})
|
||||||
{
|
{
|
||||||
@@ -74,7 +75,18 @@ namespace http
|
|||||||
auto bytes = size * count;
|
auto bytes = size * count;
|
||||||
if (bytes >= 2 && ptr[bytes - 2] == '\r' && ptr[bytes - 1] == '\n')
|
if (bytes >= 2 && ptr[bytes - 2] == '\r' && ptr[bytes - 1] == '\n')
|
||||||
{
|
{
|
||||||
if (bytes > 2 && handle->gotStatusLine) // Don't include header list terminator or the status line.
|
if (handle->gotAllHeaders)
|
||||||
|
{
|
||||||
|
// Reset response headers if we see a new header arrive after seeing a header that we thought was the last.
|
||||||
|
// This happens when a Request takes multiple HTTP requests to complete; think redirects.
|
||||||
|
// We are interested only in the response headers of the last response.
|
||||||
|
handle->responseHeaders.clear();
|
||||||
|
handle->gotStatusLine = false;
|
||||||
|
handle->gotAllHeaders = false;
|
||||||
|
}
|
||||||
|
if (bytes > 2) // Don't include header list terminator
|
||||||
|
{
|
||||||
|
if (handle->gotStatusLine) // ... or the status line.
|
||||||
{
|
{
|
||||||
auto line = ByteString(ptr, ptr + bytes - 2);
|
auto line = ByteString(ptr, ptr + bytes - 2);
|
||||||
if (auto split = line.SplitBy(':'))
|
if (auto split = line.SplitBy(':'))
|
||||||
@@ -95,7 +107,15 @@ namespace http
|
|||||||
std::cerr << "skipping weird header: " << line << std::endl;
|
std::cerr << "skipping weird header: " << line << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
handle->gotStatusLine = true;
|
handle->gotStatusLine = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
handle->gotAllHeaders = true;
|
||||||
|
}
|
||||||
return bytes;
|
return bytes;
|
||||||
}
|
}
|
||||||
return 0;
|
return 0;
|
||||||
|
Reference in New Issue
Block a user