Open-Harbor/osrm-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

Handle POST request when spanning multiple packets

Browse Source
This commit is contained in:
Daniel Patterson 2015-05-31 09:50:46 -07:00 committed by Andreas Gruß
parent 6a08d93e2c
commit a87d89302f
2 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
server/request_parser.cpp
View File

@ -42,7 +42,8 @@ namespace http
RequestParser::RequestParser() RequestParser::RequestParser()
: state(internal_state::method_start), current_header({"", ""}), : state(internal_state::method_start), current_header({"", ""}),
selected_compression(no_compression), is_post_header(false) selected_compression(no_compression), is_post_header(false),
content_length(0)
{ {
} }
@ -59,7 +60,7 @@ RequestParser::parse(request &current_request, char *begin, char *end)
} }
osrm::tribool result = osrm::tribool::indeterminate; osrm::tribool result = osrm::tribool::indeterminate;
if(is_post_header && (begin == end)) if(is_post_header && content_length == 0)
{ {
result = osrm::tribool::yes; result = osrm::tribool::yes;
} }
@ -106,6 +107,7 @@ osrm::tribool RequestParser::consume(request &current_request, const char input)
return osrm::tribool::no; return osrm::tribool::no;
case internal_state::post_request: case internal_state::post_request:
current_request.uri.push_back(input); current_request.uri.push_back(input);
--content_length;
return osrm::tribool::indeterminate; return osrm::tribool::indeterminate;
case internal_state::method: case internal_state::method:
if (input == ' ') if (input == ' ')
@ -239,6 +241,17 @@ osrm::tribool RequestParser::consume(request &current_request, const char input)
{ {
current_request.agent = current_header.value; current_request.agent = current_header.value;
} }
if (boost::iequals(current_header.name, "Content-Length"))
{
try
{
content_length = std::stoi(current_header.value);
}
catch (const std::exception &e)
{
// Ignore the header if the parameter isn't an int
}
}
if (input == '\r') if (input == '\r')
{ {
@ -312,6 +325,7 @@ osrm::tribool RequestParser::consume(request &current_request, const char input)
{ {
if(is_post_header) if(is_post_header)
{ {
current_request.uri.push_back('?');
state = internal_state::post_request; state = internal_state::post_request;
return osrm::tribool::indeterminate; return osrm::tribool::indeterminate;
} }

1
server/request_parser.hpp
View File

@ -90,6 +90,7 @@ class RequestParser
header current_header; header current_header;
compression_type selected_compression; compression_type selected_compression;
bool is_post_header; bool is_post_header;
int content_length;
}; };
} // namespace http } // namespace http