0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-11-30 10:42:47 +01:00
construct/ircd/http.cc

1371 lines
33 KiB
C++
Raw Normal View History

2018-02-04 03:22:01 +01:00
// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
2016-11-29 16:23:38 +01:00
namespace ircd { namespace http
__attribute__((visibility("hidden")))
{
using namespace ircd::spirit;
struct grammar;
struct parser extern const parser;
extern const std::unordered_map<ircd::http::code, ircd::string_view> reason;
[[noreturn]] void throw_error(const qi::expectation_failure<const char *> &, const bool &internal = false);
}}
BOOST_FUSION_ADAPT_STRUCT
(
ircd::http::query,
( decltype(ircd::http::query::first), first )
( decltype(ircd::http::query::second), second )
)
BOOST_FUSION_ADAPT_STRUCT
(
ircd::http::header,
( decltype(ircd::http::header::first), first )
( decltype(ircd::http::header::second), second )
)
BOOST_FUSION_ADAPT_STRUCT
(
ircd::http::line::response,
( decltype(ircd::http::line::response::version), version )
( decltype(ircd::http::line::response::status), status )
( decltype(ircd::http::line::response::reason), reason )
)
BOOST_FUSION_ADAPT_STRUCT
(
ircd::http::line::request,
( decltype(ircd::http::line::request::method), method )
( decltype(ircd::http::line::request::path), path )
( decltype(ircd::http::line::request::query), query )
( decltype(ircd::http::line::request::fragment), fragment )
( decltype(ircd::http::line::request::version), version )
)
decltype(ircd::http::reason)
ircd::http::reason
{
2017-04-03 05:59:18 +02:00
{ code::CONTINUE, "Continue" },
{ code::SWITCHING_PROTOCOLS, "Switching Protocols" },
{ code::PROCESSING, "Processing" },
{ code::EARLY_HINTS, "Early Hints" },
2017-04-03 05:59:18 +02:00
{ code::OK, "OK" },
{ code::CREATED, "Created" },
{ code::ACCEPTED, "Accepted" },
{ code::NON_AUTHORITATIVE_INFORMATION, "Non-Authoritative Information" },
{ code::NO_CONTENT, "No Content" },
2017-04-07 01:54:20 +02:00
{ code::PARTIAL_CONTENT, "Partial Content" },
{ code::MULTIPLE_CHOICES, "Multiple Choices" },
2017-04-07 01:54:20 +02:00
{ code::MOVED_PERMANENTLY, "Moved Permanently" },
{ code::FOUND, "Found" },
{ code::SEE_OTHER, "See Other" },
{ code::NOT_MODIFIED, "Not Modified" },
2019-02-19 20:34:48 +01:00
{ code::USE_PROXY, "Use Proxy" },
{ code::SWITCH_PROXY, "Switch Proxy" },
2017-04-07 01:54:20 +02:00
{ code::TEMPORARY_REDIRECT, "Temporary Redirect" },
{ code::PERMANENT_REDIRECT, "Permanent Redirect" },
2017-04-03 05:59:18 +02:00
{ code::BAD_REQUEST, "Bad Request" },
{ code::UNAUTHORIZED, "Unauthorized" },
{ code::FORBIDDEN, "Forbidden" },
{ code::NOT_FOUND, "Not Found" },
{ code::METHOD_NOT_ALLOWED, "Method Not Allowed" },
{ code::NOT_ACCEPTABLE, "Not Acceptable" },
2017-04-03 05:59:18 +02:00
{ code::REQUEST_TIMEOUT, "Request Time-out" },
{ code::CONFLICT, "Conflict" },
2019-09-09 03:29:19 +02:00
{ code::GONE, "Gone" },
2018-03-22 07:01:23 +01:00
{ code::LENGTH_REQUIRED, "Length Required" },
2018-01-12 03:41:58 +01:00
{ code::PAYLOAD_TOO_LARGE, "Payload Too Large" },
2017-04-03 05:59:18 +02:00
{ code::REQUEST_URI_TOO_LONG, "Request URI Too Long" },
{ code::UNSUPPORTED_MEDIA_TYPE, "Unsupported Media Type" },
{ code::RANGE_NOT_SATISFIABLE, "Range Not Satisfiable" },
2017-04-03 05:59:18 +02:00
{ code::EXPECTATION_FAILED, "Expectation Failed" },
2017-04-07 01:54:20 +02:00
{ code::IM_A_TEAPOT, "Negative, I Am A Meat Popsicle" },
2017-04-03 05:59:18 +02:00
{ code::UNPROCESSABLE_ENTITY, "Unprocessable Entity" },
{ code::PRECONDITION_REQUIRED, "Precondition Required" },
2017-04-03 05:59:18 +02:00
{ code::TOO_MANY_REQUESTS, "Too Many Requests" },
{ code::REQUEST_HEADER_FIELDS_TOO_LARGE, "Request Header Fields Too Large" },
2017-04-03 05:59:18 +02:00
{ code::INTERNAL_SERVER_ERROR, "Internal Server Error" },
{ code::NOT_IMPLEMENTED, "Not Implemented" },
2018-02-07 04:33:33 +01:00
{ code::BAD_GATEWAY, "Bad Gateway" },
2017-04-03 05:59:18 +02:00
{ code::SERVICE_UNAVAILABLE, "Service Unavailable" },
{ code::GATEWAY_TIMEOUT, "Gateway Timeout" },
2017-04-03 05:59:18 +02:00
{ code::HTTP_VERSION_NOT_SUPPORTED, "HTTP Version Not Supported" },
{ code::INSUFFICIENT_STORAGE, "Insufficient Storage" },
{ code::CLOUDFLARE_REFUSED, "Cloudflare Customer Connection Refused" },
{ code::CLOUDFLARE_TIMEDOUT, "Cloudflare Customer Connection Timed-out" },
{ code::CLOUDFLARE_UNREACHABLE, "Cloudflare Customer Unreachable" },
{ code::CLOUDFLARE_REQUEST_TIMEOUT, "Cloudflare Customer Request Time-out" },
};
struct ircd::http::grammar
:qi::grammar<const char *, unused_type>
2016-11-29 16:23:38 +01:00
{
using it = const char *;
template<class R = unused_type,
class... S>
using rule = qi::rule<it, R, S...>;
2016-11-29 16:23:38 +01:00
rule<> NUL { lit('\0') ,"nul" };
2016-11-29 16:23:38 +01:00
// insignificant whitespaces
rule<> SP { lit('\x20') ,"space" };
rule<> HT { lit('\x09') ,"horizontal tab" };
rule<> ws { SP | HT ,"whitespace" };
rule<> CR { lit('\x0D') ,"carriage return" };
rule<> LF { lit('\x0A') ,"line feed" };
rule<> CRLF { CR >> LF ,"carriage return, line feed" };
2016-11-29 16:23:38 +01:00
rule<> illegal { NUL | CR | LF ,"illegal" };
rule<> colon { lit(':') ,"colon" };
rule<> slash { lit('/') ,"forward solidus" };
rule<> question { lit('?') ,"question mark" };
rule<> pound { lit('#') ,"pound sign" };
rule<> equal { lit('=') ,"equal sign" };
rule<> ampersand { lit('&') ,"ampersand" };
rule<string_view> token { raw[+(char_ - (illegal | ws))] ,"token" };
rule<string_view> string { raw[+(char_ - illegal)] ,"string" };
rule<string_view> line { *ws >> -string >> CRLF ,"line" };
rule<string_view> status { raw[repeat(3)[char_("0-9")]] ,"status" };
rule<string_view> reason { string ,"status" };
rule<string_view> head_key { raw[+(char_ - (illegal | ws | colon))] ,"head key" };
rule<string_view> head_val { string ,"head value" };
rule<http::header> header { head_key >> *ws >> colon >> *ws >> head_val ,"header" };
rule<unused_type> headers { (header % (*ws >> CRLF)) ,"headers" };
rule<> query_terminator { equal | question | ampersand | pound ,"query terminator" };
rule<> query_illegal { illegal | ws | query_terminator ,"query illegal" };
rule<string_view> query_key { raw[+(char_ - query_illegal)] ,"query key" };
rule<string_view> query_val { raw[*(char_ - query_illegal)] ,"query value" };
rule<string_view> method { token ,"method" };
rule<string_view> path { raw[-slash >> *(char_ - query_illegal)] ,"path" };
rule<string_view> fragment { pound >> -token ,"fragment" };
rule<string_view> version { token ,"version" };
rule<uint32_t> chunk_size
{
qi::uint_parser<uint32_t, 16, 1, 8>{}
,"chunk size"
};
rule<string_view> chunk_extensions
{
';' >> raw[string] //TODO: extensions
,"chunk extensions"
};
rule<http::query> query
{
query_key >> -(equal >> query_val)
,"query"
};
rule<string_view> query_string
{
question >> -raw[(query_key >> -(equal >> query_val)) % ampersand]
,"query string"
};
rule<line::request> request_line
{
method >> +SP >> path >> -query_string >> -fragment >> +SP >> version
,"request line"
};
rule<line::response> response_line
{
version >> +SP >> status >> -(+SP >> reason)
,"response line"
};
rule<unused_type> request
{
request_line >> *ws >> CRLF >> -headers >> CRLF
,"request"
};
rule<unused_type> response
{
response_line >> *ws >> CRLF >> -headers >> CRLF
,"response"
};
grammar()
:qi::grammar<const char *, unused_type>{rule<>{}}
{}
};
2016-11-29 16:23:38 +01:00
struct ircd::http::parser
:grammar
2016-11-29 16:23:38 +01:00
{
static size_t content_length(const string_view &val);
template<class gen,
class... attr>
bool operator()(const char *&start, const char *const &stop, gen&&, attr&&...) const;
2016-11-29 16:23:38 +01:00
}
const ircd::http::parser;
2016-11-29 16:23:38 +01:00
template<class gen,
class... attr>
inline bool
ircd::http::parser::operator()(const char *&start,
const char *const &stop,
gen&& g,
attr&&... a)
const
{
return ircd::parse(start, stop, std::forward<gen>(g), std::forward<attr>(a)...);
}
namespace ircd { namespace http
__attribute__((visibility("default")))
{
// stub needed for clang
}}
/// Compose a request. This prints an HTTP head into the buffer. No real IO is
/// done here. After composing into the buffer, the user can then drive the
/// socket by sending the header and the content as specified.
///
/// If termination is false, no extra CRLF is printed to the buffer allowing
/// additional headers not specified to be appended later.
ircd::http::request::request(window_buffer &out,
const string_view &host,
const string_view &method,
const string_view &uri,
const size_t &content_length,
const string_view &content_type,
const vector_view<const header> &headers,
const bool &termination)
{
writeline(out, [&method, &uri](const mutable_buffer &out) -> size_t
{
assert(!method.empty());
assert(!uri.empty());
return fmt::sprintf
{
out, "%s %s HTTP/1.1", method, uri
};
});
if(!has(headers, "host"))
writeline(out, [&host](const mutable_buffer &out) -> size_t
{
assert(!host.empty());
return fmt::sprintf
{
out, "Host: %s", host
};
});
if(content_length && !has(headers, "content-type"))
writeline(out, [&content_type](const mutable_buffer &out) -> size_t
{
return fmt::sprintf
{
out, "Content-Type: %s", content_type?: "text/plain; charset=utf-8"
};
});
if(!has(headers, "content-length"))
writeline(out, [&content_length](const mutable_buffer &out) -> size_t
{
return fmt::sprintf
{
out, "Content-Length: %zu", content_length
};
});
write(out, headers);
if(termination)
writeline(out);
}
namespace ircd::http
{
static void assign(request::head &, const header &);
}
2017-03-13 23:24:42 +01:00
ircd::http::request::head::head(parse::capstan &pc,
const headers::closure &closure)
:line::request{pc}
,uri
{
fragment? string_view { begin(path), end(fragment) }:
query? string_view { begin(path), end(query) }:
string_view { begin(path), end(path) }
}
,headers{[this, &pc, &closure]
{
2019-04-12 17:35:13 +02:00
const bool version_compatible
{
this->version == "HTTP/1.0" ||
this->version == "HTTP/1.1"
};
if(unlikely(!version_compatible))
throw error
{
HTTP_VERSION_NOT_SUPPORTED, fmt::snstringf
{
2019-04-12 17:35:13 +02:00
128, "Sorry, no speak '%s'. Try HTTP/1.1 here.",
this->version
}
};
if(closure)
return http::headers
{
pc, [this, &closure](const auto &header)
{
assign(*this, header);
closure(header);
}
};
return http::headers
{
pc, [this](const auto &header)
{
assign(*this, header);
}
};
}()}
{
}
ircd::http::request::head::operator
string_view()
const
{
const string_view request_line
{
static_cast<const line::request &>(*this)
};
if(request_line.empty() || headers.empty())
return request_line;
return string_view
{
request_line.begin(), headers.end()
};
}
void
ircd::http::assign(request::head &head,
const header &header)
{
char buf[64];
const auto &[key_, val] {header};
assert(size(key_) <= sizeof(buf));
const auto &key
{
tolower(buf, key_)
};
if(key == "content-length"_sv)
head.content_length = parser.content_length(val);
else if(key == "host"_sv)
head.host = val;
else if(key == "expect"_sv)
head.expect = val;
else if(key == "te"_sv)
head.te = val;
else if(key == "authorization"_sv)
head.authorization = val;
else if(key == "connection"_sv)
head.connection = val;
else if(key == "content-type"_sv)
head.content_type = val;
else if(key == "user-agent"_sv)
head.user_agent = val;
else if(key == "upgrade"_sv)
head.upgrade = val;
else if(key == "range"_sv)
head.range = val;
else if(key == "if-range"_sv)
head.if_range = val;
else if(key == "x-forwarded-for"_sv)
head.forwarded_for = val;
}
ircd::http::response::response(window_buffer &out,
const code &code,
const size_t &content_length,
const string_view &content_type,
const http::headers &headers_s,
const vector_view<const header> &headers_v,
const bool &termination)
{
const auto has_header{[&headers_s, &headers_v]
(const string_view &key) -> bool
{
return has(headers_v, key) || has(headers_s, key);
}};
writeline(out, [&code](const mutable_buffer &out) -> size_t
{
return fmt::sprintf
{
out, "HTTP/1.1 %u %s", uint(code), status(code)
};
});
const bool write_server_header
{
code >= 200 && code < 300
&& !has_header("server")
};
if(write_server_header)
writeline(out, [&code](const mutable_buffer &out) -> size_t
{
size_t ret{0};
ret += copy(out, "Server: "_sv);
ret += copy(out + ret, ircd::info::server_agent);
return ret;
});
const bool write_date_header
{
code < 400
&& !has_header("date")
};
if(write_date_header)
writeline(out, [](const mutable_buffer &out) -> size_t
{
2018-05-20 04:58:37 +02:00
thread_local char date_buf[96];
return fmt::sprintf
{
out, "Date: %s", timef(date_buf, ircd::localtime)
};
});
const bool write_content_type_header
{
code != NO_CONTENT
&& content_type && content_length
&& !has_header("content-type")
};
if(write_content_type_header)
writeline(out, [&content_type](const mutable_buffer &out) -> size_t
{
return fmt::sprintf
{
out, "Content-Type: %s", content_type?: "text/plain; charset=utf-8"
};
});
const bool write_content_length_header
{
code != NO_CONTENT
&& content_length != size_t(-1) // chunked encoding indication
&& !has_header("content-length")
};
if(write_content_length_header)
writeline(out, [&content_length](const mutable_buffer &out) -> size_t
{
return fmt::sprintf
{
out, "Content-Length: %zu", content_length
};
});
const bool write_transfer_encoding_chunked
{
content_length == size_t(-1)
&& !has_header("transfer-encoding")
};
if(write_transfer_encoding_chunked)
writeline(out, [&content_length](const mutable_buffer &out) -> size_t
{
return copy(out, "Transfer-Encoding: chunked"_sv);
});
if(!headers_s.empty())
out([&headers_s](const mutable_buffer &out)
{
assert(endswith(headers_s, "\r\n"));
return copy(out, headers_s);
});
if(!headers_v.empty())
write(out, headers_v);
if(termination)
writeline(out);
}
namespace ircd::http
{
static void assign(response::head &, const header &);
}
ircd::http::response::head::head(parse::capstan &pc)
:line::response{pc}
,headers
{
http::headers
{
pc, [this](const auto &header)
{
assign(*this, header);
}
}
}
{
}
ircd::http::response::head::head(parse::capstan &pc,
const headers::closure &closure)
:line::response{pc}
,headers
{
http::headers
{
pc, [this, &closure](const auto &header)
{
assign(*this, header);
closure(header);
}
}
}
{
}
void
ircd::http::assign(response::head &head,
const header &header)
{
char buf[64];
const auto &[key_, val] {header};
assert(size(key_) <= sizeof(buf));
const auto &key
{
tolower(buf, key_)
};
if(key == "content-length"_sv)
head.content_length = parser.content_length(val);
else if(key == "content-type"_sv)
head.content_type = val;
else if(key == "content-range"_sv)
head.content_range = val;
else if(key == "accept-range"_sv)
head.content_range = val;
else if(key == "transfer-encoding"_sv)
head.transfer_encoding = val;
else if(key == "server"_sv)
head.server = val;
else if(key == "location"_sv)
head.location = val;
}
ircd::http::response::chunk::chunk(parse::capstan &pc)
try
:line{pc}
{
static const parser::rule<size_t> grammar
{
eoi | (eps > (parser.chunk_size >> -parser.chunk_extensions))
,"chunk head"
};
const char *start(line::begin());
const auto res
{
parser(start, line::end(), grammar, this->size)
};
assert(res == true);
}
catch(const qi::expectation_failure<const char *> &e)
{
throw_error(e, true);
}
//
// headers
//
bool
ircd::http::has(const vector_view<const header> &headers,
const string_view &key)
{
return end(headers) != std::find_if(begin(headers), end(headers), [&key]
(const header &header)
{
return header == key;
});
}
bool
ircd::http::has(const headers &headers,
const string_view &key)
{
return headers.has(key);
}
//
// headers::headers
//
decltype(ircd::http::headers::terminator)
ircd::http::headers::terminator
{
"\r\n\r\n"
};
2017-03-13 23:24:42 +01:00
ircd::http::headers::headers(parse::capstan &pc,
const closure &c)
:headers
{
pc, closure_bool{[&c](const auto &header)
{
if(c)
c(header);
return true;
}}
}
{
}
ircd::http::headers::headers(parse::capstan &pc,
closure_bool c)
:string_view{[&pc, &c]
() -> string_view
{
header h{pc};
const char *const &started{h.first.data()}, *stopped{started};
for(; !h.first.empty(); stopped = h.second.data() + h.second.size(), h = header{pc})
if(c && !c(h))
c = {};
return { started, stopped };
}()}
{
}
bool
ircd::http::headers::has(const string_view &key)
const
{
// has header if break early from for_each
return !for_each([&key]
(const header &header)
{
// true to continue; false to break (for_each protocol)
return iequals(header.first, key)? false : true;
});
}
ircd::string_view
ircd::http::headers::at(const string_view &key)
const
{
const string_view ret
{
this->operator[](key)
};
if(unlikely(!ret))
throw std::out_of_range{key};
return ret;
}
ircd::string_view
ircd::http::headers::operator[](const string_view &key)
const
{
string_view ret;
for_each([&key, &ret](const auto &header)
{
if(iequals(header.first, key))
{
ret = header.second;
return false;
}
else return true;
});
return ret;
}
bool
ircd::http::headers::for_each(const closure_bool &closure)
const
{
if(empty())
return true;
parse::buffer pb{const_buffer{*this}};
parse::capstan pc{pb};
header h{pc};
for(; !h.first.empty(); h = header{pc})
if(!closure(h))
return false;
return true;
}
//
// header
//
ircd::http::header::header(const line &line)
2016-11-29 16:23:38 +01:00
try
{
static const auto grammar
{
eps > parser.header
};
if(line.empty())
return;
const char *start(line.data());
const char *const stop(line.data() + line.size());
parser(start, stop, grammar, *this);
2016-11-29 16:23:38 +01:00
}
catch(const qi::expectation_failure<const char *> &e)
{
throw_error(e);
2016-11-29 16:23:38 +01:00
}
ircd::http::line::response::response(const line &line)
{
static const auto grammar
{
eps > parser.response_line
2016-11-29 16:23:38 +01:00
};
const char *start(line.data());
const char *const stop(line.data() + line.size());
parser(start, stop, grammar, *this);
2016-11-29 16:23:38 +01:00
}
ircd::http::line::request::request(const line &line)
try
{
static const auto grammar
{
eps > parser.request_line
};
const char *start(line.data());
const char *const stop(line.data() + line.size());
parser(start, stop, grammar, *this);
2016-11-29 16:23:38 +01:00
}
catch(const qi::expectation_failure<const char *> &e)
{
throw_error(e);
2016-11-29 16:23:38 +01:00
}
ircd::http::line::request::operator
string_view()
const
{
if(method.empty())
return string_view{};
assert(!version.empty());
assert(method.begin() < version.end());
return string_view
{
method.begin(), version.end()
};
}
//
// http::line
//
decltype(ircd::http::line::terminator)
ircd::http::line::terminator
{
"\r\n"
};
2017-03-13 23:24:42 +01:00
ircd::http::line::line(parse::capstan &pc)
2016-11-29 16:23:38 +01:00
:string_view{[&pc]
{
static const auto grammar
{
parser.line
};
string_view ret;
pc([&ret](const char *&start, const char *const &stop)
2016-11-29 16:23:38 +01:00
{
if(!parser(start, stop, grammar, ret))
{
ret = {};
return false;
}
else return true;
2016-11-29 16:23:38 +01:00
});
return ret;
}()}
{
}
//
// query::string
//
bool
ircd::http::query::string::has(const string_view &key)
const
{
bool ret{false};
for_each(key, [&ret]
(const auto &)
{
ret = true;
return false;
});
return ret;
}
size_t
ircd::http::query::string::count(const string_view &key)
const
{
size_t ret{0};
for_each(key, [&ret]
(const auto &)
{
++ret;
return true;
});
return ret;
}
ircd::vector_view<ircd::string_view>
ircd::http::query::string::array(const mutable_buffer &buf,
const string_view &key,
string_view *const &out,
const size_t &max)
const
{
if(unlikely(!max))
return {};
size_t ret(0);
window_buffer wb(buf);
for_each(key, [&out, &max, &ret, &wb]
(const auto &query)
{
wb([&out, &max, &ret, &query]
(const mutable_buffer &buf)
{
assert(ret < max);
const auto &[_, server_name] {query};
out[ret] = url::decode(buf, server_name);
return out[ret];
});
return ++ret < max;
});
return vector_view<string_view>
{
out, ret
};
}
ircd::string_view
ircd::http::query::string::at(const string_view &key,
const size_t &idx)
const
{
const string_view &ret
{
_get(key, idx)
};
if(!ret)
{
thread_local char buf[1024];
const string_view msg{fmt::sprintf
{
buf, "Failed to find value for required query string key '%s' #%zu.",
key,
idx
}};
throw std::out_of_range
{
msg.c_str() // fmt::sprintf() will null terminate msg
};
}
return ret;
}
ircd::string_view
ircd::http::query::string::operator[](const string_view &key)
const
{
return _get(key, 0);
}
ircd::string_view
ircd::http::query::string::_get(const string_view &key,
size_t idx)
const
{
string_view ret;
for_each(key, [&idx, &ret]
(const auto &query)
{
if(!idx--)
{
ret = query.second;
return false;
}
else return true;
});
return ret;
}
bool
ircd::http::query::string::for_each(const string_view &key,
const closure &closure)
const
{
return for_each([&key, &closure]
(const auto &query)
{
2018-08-31 03:58:59 +02:00
if(query.first != key)
return true;
return closure(query);
});
}
bool
2018-08-31 03:58:59 +02:00
ircd::http::query::string::for_each(const closure &view)
const
{
bool ret{true};
const auto action{[&view, &ret]
2018-08-31 03:58:59 +02:00
(const auto &attribute, const auto &context, auto &continue_)
{
ret = view(attribute);
continue_ = ret;
2018-08-31 03:58:59 +02:00
}};
const parser::rule<unused_type> grammar
{
-parser.question >> (parser.query[action] % parser.ampersand)
};
const string_view &s(*this);
2018-08-31 03:58:59 +02:00
const char *start(s.begin()), *const stop(s.end());
parser(start, stop, grammar);
return ret;
}
//
// parser util
//
size_t
ircd::http::parser::content_length(const string_view &str)
{
static const parser::rule<size_t> grammar
{
ulong_
};
size_t ret;
const char *start(str.data());
const bool parsed
{
http::parser(start, start + str.size(), grammar, ret)
};
if(!parsed || ret >= 256_GiB)
2017-12-30 09:07:23 +01:00
throw error
{
BAD_REQUEST, "Invalid content-length value"
};
return ret;
}
//
// util
//
ircd::const_buffer
ircd::http::writechunk(const mutable_buffer &buf,
const uint32_t &chunk_size)
{
window_buffer wb{buf};
writechunk(wb, chunk_size);
return wb.completed();
}
void
ircd::http::writechunk(window_buffer &buf,
const uint32_t &chunk_size)
{
writeline(buf, [&chunk_size]
(const mutable_buffer &out) -> size_t
{
return ::snprintf(data(out), size(out), "%08x", chunk_size);
});
}
std::string
ircd::http::strung(const vector_view<const header> &headers)
{
return ircd::string(serialized(headers), [&]
(window_buffer out)
{
write(out, headers);
return out.consumed();
});
}
/// Indicates the buffer size required to write these headers. This size
/// may include room for a terminating null character which may be written
/// by write(headers). Only use write(headers) to know the actually written
/// string size (without null) not this.
size_t
ircd::http::serialized(const vector_view<const header> &headers)
{
// Because the write(header) functions use fmt::sprintf we have to
// indicate an extra space for a null string terminator to not overlof
const size_t initial{!headers.empty()};
return std::accumulate(std::begin(headers), std::end(headers), initial, []
(auto &ret, const auto &pair)
{
// key : SP value CRLF
return ret += pair.first.size() + 1 + 1 + pair.second.size() + 2;
});
}
void
ircd::http::write(window_buffer &out,
const vector_view<const header> &headers)
{
for(const auto &header : headers)
write(out, header);
}
void
ircd::http::write(window_buffer &out,
const header &header)
{
if(header.second.empty())
return;
assert(!header.first.empty());
if(unlikely(header.first.empty()))
return;
writeline(out, [&header](const mutable_buffer &out) -> size_t
{
return fmt::sprintf
{
out, "%s: %s", header.first, header.second
};
});
}
/// Close over the user's closure to append a newline.
void
ircd::http::writeline(window_buffer &write,
const window_buffer::closure &closure)
{
// A new window_buffer is implicit constructed out of the mutable_buffer
// otherwise presented to this closure as its write window.
write([&closure](window_buffer write)
{
const auto newline{[](const mutable_buffer &out)
{
return copy(out, line::terminator);
}};
write(closure);
write(newline);
return write.consumed();
});
}
void
ircd::http::writeline(window_buffer &write)
{
writeline(write, [](const mutable_buffer &out)
{
return 0;
});
}
/// Called to translate a grammar exception into an http::error within our
/// system. This will then usually propagate back to our client.
///
/// If we are a client to another server, set internal=true. Even though this
/// still generates an HTTP error, the code is 500 so if it propagates back to
/// a client it does not indicate to *that* client that *they* made a bad
/// request from a 400 back to them.
void
ircd::http::throw_error(const qi::expectation_failure<const char *> &e,
const bool &internal)
{
const auto &code_
{
internal?
code::INTERNAL_SERVER_ERROR:
code::BAD_REQUEST
};
const char *const &fmtstr
{
internal?
"I expected a valid HTTP %s. Server sent %zu invalid characters starting with `%s'.":
"I require a valid HTTP %s. You sent %zu invalid characters starting with `%s'."
};
const auto &rule
{
ircd::string(e.what_)
};
throw error
{
code_, fmt::snstringf
{
512, fmtstr,
between(rule, "<", ">"),
size_t(e.last - e.first),
string_view{e.first, e.last}
}
};
}
//
// error
//
ircd::http::error::error(const http::code &code,
std::string content,
const vector_view<const header> &headers)
:http::error
{
code, std::move(content), strung(headers)
}
{
}
ircd::http::error::error(const http::code &code,
std::string content,
std::string headers)
:ircd::error
{
generate_skip
}
,content
{
std::move(content)
}
,headers
{
std::move(headers)
}
,code{code}
{
auto &buf(ircd::error::buf);
::snprintf(buf, sizeof(buf), "%u %s", uint(code), status(code).c_str());
}
// Out-of-line placement.
ircd::http::error::~error()
noexcept
{
}
//
// status
//
#pragma GCC visibility push(internal)
namespace ircd::http
{
extern const parser::rule<uint8_t> status_codepoint;
extern const parser::rule<enum category> status_category;
extern const parser::rule<enum code> status_code;
}
#pragma GCC visibility pop
decltype(ircd::http::status_codepoint)
ircd::http::status_codepoint
{
qi::uint_parser<uint8_t, 10, 1, 1>{}
,"status codepoint"
};
decltype(ircd::http::status_category)
ircd::http::status_category
{
&char_("1-9") >> status_codepoint >> omit[repeat(2)[char_("0-9")] >> (eoi | parser.ws)]
};
decltype(ircd::http::status_code)
ircd::http::status_code
{
qi::uint_parser<uint16_t, 10, 3, 3>{} >> omit[eoi | parser.ws]
,"status code"
};
enum ircd::http::code
ircd::http::status(const string_view &str)
{
short ret;
const char *start(begin(str)), *const stop(end(str));
const bool parsed
{
parser(start, stop, status_code, ret)
};
if(!parsed)
throw ircd::error
{
"Invalid HTTP status code"
};
assert(ret >= 0 && ret < 1000);
return http::code(ret);
}
ircd::string_view
ircd::http::status(const enum code &code)
noexcept try
{
return reason.at(code);
}
catch(const std::out_of_range &e)
{
log::dwarning
{
"No reason string for HTTP status code %u", uint(code)
};
return ""_sv;
}
enum ircd::http::category
ircd::http::category(const string_view &str)
noexcept
{
enum category ret;
const char *start(begin(str)), *const stop(end(str));
const bool parsed
{
parser(start, stop, status_category, ret)
};
if(!parsed || ret > category::UNKNOWN)
ret = category::UNKNOWN;
assert(uint8_t(ret) <= uint8_t(category::UNKNOWN));
return ret;
}
enum ircd::http::category
ircd::http::category(const enum code &code)
noexcept
{
if(ushort(code) == 0)
return category::NONE;
if(ushort(code) < 200)
return category::INFO;
if(ushort(code) < 300)
return category::SUCCESS;
if(ushort(code) < 400)
return category::REDIRECT;
if(ushort(code) < 500)
return category::ERROR;
if(ushort(code) < 600)
return category::SERVER;
return category::UNKNOWN;
}
ircd::string_view
ircd::http::category(const enum category &category)
noexcept
{
switch(category)
{
case category::NONE: return "NONE";
case category::INFO: return "INFO";
case category::SUCCESS: return "SUCCESS";
case category::REDIRECT: return "REDIRECT";
case category::ERROR: return "ERROR";
case category::SERVER: return "SERVER";
default:
case category::UNKNOWN: return "UNKNOWN";
}
}