move tribool obj into request_parser

This commit is contained in:
karenzshea 2016-01-28 18:43:19 -05:00 committed by Patrick Niklaus
parent efee07e20b
commit 262cdd7bc0
4 changed files with 80 additions and 93 deletions

View File

@ -3,7 +3,6 @@
#include "server/http/compression_type.hpp"
#include "server/http/header.hpp"
#include "util/tribool.hpp"
#include <tuple>
@ -22,11 +21,18 @@ class RequestParser
public:
RequestParser();
std::tuple<util::tribool, http::compression_type>
std::tuple<RequestStatus, http::compression_type>
parse(http::request &current_request, char *begin, char *end);
enum class RequestStatus : char
{
valid,
invalid,
indeterminate
};
private:
util::tribool consume(http::request &current_request, const char input);
RequestStatus consume(http::request &current_request, const char input);
bool is_char(const int character) const;

View File

@ -1,17 +0,0 @@
#ifndef TRIBOOL_HPP
#define TRIBOOL_HPP
namespace osrm
{
namespace util
{
enum class tribool : char
{
yes,
no,
indeterminate
};
}
}
#endif // TRIBOOL_HPP

View File

@ -41,13 +41,13 @@ void Connection::handle_read(const boost::system::error_code &error, std::size_t
// no error detected, let's parse the request
http::compression_type compression_type(http::no_compression);
util::tribool result;
RequestStatus result;
std::tie(result, compression_type) =
request_parser.parse(current_request, incoming_data_buffer.data(),
incoming_data_buffer.data() + bytes_transferred);
// the request has been parsed
if (result == util::tribool::yes)
if (result == RequestStatus::yes)
{
current_request.endpoint = TCP_socket.remote_endpoint().address();
request_handler.handle_request(current_request, current_reply);
@ -85,7 +85,7 @@ void Connection::handle_read(const boost::system::error_code &error, std::size_t
strand.wrap(boost::bind(&Connection::handle_write, this->shared_from_this(),
boost::asio::placeholders::error)));
}
else if (result == util::tribool::no)
else if (result == RequestStatus::no)
{ // request is not parseable
current_reply = http::reply::stock_reply(http::reply::bad_request);

View File

@ -4,8 +4,6 @@
#include "server/http/header.hpp"
#include "server/http/request.hpp"
#include "util/tribool.hpp"
#include <boost/algorithm/string/predicate.hpp>
#include <string>
@ -21,177 +19,177 @@ RequestParser::RequestParser()
{
}
std::tuple<util::tribool, http::compression_type>
std::tuple<RequestStatus, http::compression_type>
RequestParser::parse(http::request &current_request, char *begin, char *end)
{
while (begin != end)
{
util::tribool result = consume(current_request, *begin++);
if (result != util::tribool::indeterminate)
RequestStatus result = consume(current_request, *begin++);
if (result != RequestStatus::indeterminate)
{
return std::make_tuple(result, selected_compression);
}
}
util::tribool result = util::tribool::indeterminate;
RequestStatus result = RequestStatus::indeterminate;
if (state == internal_state::post_request && content_length <= 0)
{
result = util::tribool::yes;
result = RequestStatus::valid;
}
return std::make_tuple(result, selected_compression);
}
util::tribool RequestParser::consume(http::request &current_request, const char input)
RequestStatus RequestParser::consume(http::request &current_request, const char input)
{
switch (state)
{
case internal_state::method_start:
if (!is_char(input) || is_CTL(input) || is_special(input))
{
return util::tribool::no;
return RequestStatus::invalid;
}
if (input == 'P')
{
state = internal_state::post_O;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
state = internal_state::method;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::post_O:
if (input == 'O')
{
state = internal_state::post_S;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::post_S:
if (input == 'S')
{
state = internal_state::post_T;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::post_T:
if (input == 'T')
{
is_post_header = true;
state = internal_state::method;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::post_request:
current_request.uri.push_back(input);
--content_length;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::method:
if (input == ' ')
{
state = internal_state::uri;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (!is_char(input) || is_CTL(input) || is_special(input))
{
return util::tribool::no;
return RequestStatus::invalid;
}
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::uri_start:
if (is_CTL(input))
{
return util::tribool::no;
return RequestStatus::invalid;
}
state = internal_state::uri;
current_request.uri.push_back(input);
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::uri:
if (input == ' ')
{
state = internal_state::http_version_h;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (is_CTL(input))
{
return util::tribool::no;
return RequestStatus::invalid;
}
current_request.uri.push_back(input);
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::http_version_h:
if (input == 'H')
{
state = internal_state::http_version_t_1;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::http_version_t_1:
if (input == 'T')
{
state = internal_state::http_version_t_2;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::http_version_t_2:
if (input == 'T')
{
state = internal_state::http_version_p;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::http_version_p:
if (input == 'P')
{
state = internal_state::http_version_slash;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::http_version_slash:
if (input == '/')
{
state = internal_state::http_version_major_start;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::http_version_major_start:
if (is_digit(input))
{
state = internal_state::http_version_major;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::http_version_major:
if (input == '.')
{
state = internal_state::http_version_minor_start;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (is_digit(input))
{
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::http_version_minor_start:
if (is_digit(input))
{
state = internal_state::http_version_minor;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::http_version_minor:
if (input == '\r')
{
state = internal_state::expecting_newline_1;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (is_digit(input))
{
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::expecting_newline_1:
if (input == '\n')
{
state = internal_state::header_line_start;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::header_line_start:
if (boost::iequals(current_header.name, "Accept-Encoding"))
{
@ -230,77 +228,77 @@ util::tribool RequestParser::consume(http::request &current_request, const char
{
if (!boost::icontains(current_header.value, "application/x-www-form-urlencoded"))
{
return util::tribool::no;
return RequestStatus::invalid;
}
}
if (input == '\r')
{
state = internal_state::expecting_newline_3;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (!is_char(input) || is_CTL(input) || is_special(input))
{
return util::tribool::no;
return RequestStatus::invalid;
}
state = internal_state::header_name;
current_header.clear();
current_header.name.push_back(input);
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::header_lws:
if (input == '\r')
{
state = internal_state::expecting_newline_2;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (input == ' ' || input == '\t')
{
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (is_CTL(input))
{
return util::tribool::no;
return RequestStatus::invalid;
}
state = internal_state::header_value;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::header_name:
if (input == ':')
{
state = internal_state::space_before_header_value;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (!is_char(input) || is_CTL(input) || is_special(input))
{
return util::tribool::no;
return RequestStatus::invalid;
}
current_header.name.push_back(input);
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::space_before_header_value:
if (input == ' ')
{
state = internal_state::header_value;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::header_value:
if (input == '\r')
{
state = internal_state::expecting_newline_2;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
if (is_CTL(input))
{
return util::tribool::no;
return RequestStatus::invalid;
}
current_header.value.push_back(input);
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
case internal_state::expecting_newline_2:
if (input == '\n')
{
state = internal_state::header_line_start;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::no;
return RequestStatus::invalid;
case internal_state::expecting_newline_3:
if (input == '\n')
{
@ -311,13 +309,13 @@ util::tribool RequestParser::consume(http::request &current_request, const char
current_request.uri.push_back('?');
}
state = internal_state::post_request;
return util::tribool::indeterminate;
return RequestStatus::indeterminate;
}
return util::tribool::yes;
return RequestStatus::valid;
}
return util::tribool::no;
return RequestStatus::invalid;
default: // should never be reached
return input == '\n' ? util::tribool::yes : util::tribool::no;
return input == '\n' ? RequestStatus::valid : RequestStatus::invalid;
}
}