// Matrix Construct // // Copyright (C) Matrix Construct Developers, Authors & Contributors // Copyright (C) 2016-2019 Jason Volk // // 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. /////////////////////////////////////////////////////////////////////////////// // // resource/resource.h // decltype(ircd::resource::log) ircd::resource::log { "resource", 'r' }; decltype(ircd::resource::resources) ircd::resource::resources {}; ircd::resource & ircd::resource::find(const string_view &path_) { const string_view path { rstrip(path_, '/') }; auto it { resources.lower_bound(path) }; if(it == end(resources)) try { --it; if(it == begin(resources) || !startswith(path, it->first)) return *resources.at("/"); } catch(const std::out_of_range &e) { throw http::error { http::code::NOT_FOUND }; } auto rpath{it->first}; //assert(!endswith(rpath, '/')); // Exact file or directory match if(path == rpath) return *it->second; // Directories handle all paths under them. while(!startswith(path, rpath)) { // Walk the iterator back to find if there is a directory prefixing this path. if(it == begin(resources)) throw http::error { http::code::NOT_FOUND }; --it; rpath = it->first; if(~it->second->opts->flags & it->second->DIRECTORY) continue; // If the closest directory still doesn't match hand this off to the // webroot which can then service or 404 this itself. if(!startswith(path, rpath)) return *resources.at("/"); } // Check if the resource is a directory; if not, it can only // handle exact path matches. if(~it->second->opts->flags & it->second->DIRECTORY && path != rpath) throw http::error { http::code::NOT_FOUND }; if(it->second->opts->flags & it->second->DIRECTORY) { const auto rem(lstrip(path, rpath)); if(!empty(rem) && !startswith(rem, '/')) throw http::error { http::code::NOT_FOUND }; } return *it->second; } // // resource::resource // ircd::resource::resource(const string_view &path) :resource { path, {} } { } ircd::resource::resource(const string_view &path, struct opts opts) :path { rstrip(path, '/') } ,opts { std::make_unique(std::move(opts)) } ,resources_it{[this] { const auto iit { resources.emplace(this->path, this) }; if(!iit.second) throw error { "resource \"%s\" already registered", this->path }; return unique_const_iterator { resources, iit.first }; }()} ,default_method_head{[this, &opts] () -> std::unique_ptr { if(opts.flags & flag::OVERRIDE_HEAD) return {}; auto handler { std::bind(&resource::handle_head, this, ph::_1, ph::_2) }; return std::make_unique(*this, "HEAD", std::move(handler)); }()} ,default_method_options{[this, &opts] () -> std::unique_ptr { if(opts.flags & flag::OVERRIDE_OPTIONS) return {}; auto handler { std::bind(&resource::handle_options, this, ph::_1, ph::_2) }; return std::make_unique(*this, "OPTIONS", std::move(handler)); }()} { log::debug { log, "Registered resource \"%s\"", path.empty()? string_view{"/"}: this->path }; } ircd::resource::~resource() noexcept { log::debug { log, "Unregistered resource \"%s\"", path.empty()? string_view{"/"}: path }; } ircd::resource::method & ircd::resource::operator[](const string_view &name) const try { return *methods.at(name); } catch(const std::out_of_range &e) { thread_local char buf[512]; const http::header headers[] { { "Allow", method_list(buf) } }; throw http::error { http::METHOD_NOT_ALLOWED, {}, headers }; } ircd::resource::response ircd::resource::handle_head(client &client, const request &request) const { return response { client, http::METHOD_NOT_ALLOWED }; } ircd::resource::response ircd::resource::handle_options(client &client, const request &request) const { const http::headers headers { request.head.headers }; const string_view &request_origin { headers["origin"] }; const string_view &allow_origin { resource::response::access_control_allow_origin }; const string_view &request_headers { headers["access-control-request-headers"] }; const string_view &allow_headers { request_headers }; const string_view &request_method { headers["access-control-request-method"] }; char allow_methods_buf[48]; const string_view &allow_methods { method_list(allow_methods_buf, [&](const method &method) { return true; }) }; const http::header response_headers[] { // ACAO sent further up stack //{ "Access-Control-Allow-Origin", allow_origin }, { "Access-Control-Allow-Methods", allow_methods }, { "Access-Control-Allow-Headers", allow_headers }, }; return response { client, {}, {}, http::OK, response_headers }; } ircd::string_view ircd::resource::method_list(const mutable_buffer &buf) const { return method_list(buf, [] (const method &) { return true; }); } ircd::string_view ircd::resource::method_list(const mutable_buffer &buf, const method_closure &closure) const { size_t len(0); if(likely(size(buf))) buf[len] = '\0'; auto it(begin(methods)); if(it != end(methods)) { assert(it->second); if(closure(*it->second)) len = strlcat(buf, it->first); for(++it; it != end(methods); ++it) { assert(it->second); if(!closure(*it->second)) continue; len = strlcat(buf, ", "); len = strlcat(buf, it->first); } } return { data(buf), len }; } /////////////////////////////////////////////////////////////////////////////// // // resource/method.h // namespace ircd { extern conf::item cache_warmup_time; static void cache_warm_origin(const string_view &origin); } decltype(ircd::resource::method::idle_dock) ircd::resource::method::idle_dock; // // method::opts // decltype(ircd::resource::method::default_timeout) ircd::resource::method::default_timeout { { "name", "ircd.resource.method.default.timeout" }, { "default", 30L }, }; decltype(ircd::resource::method::default_payload_max) ircd::resource::method::default_payload_max { { "name", "ircd.resource.method.default.payload_max" }, { "default", long(128_KiB) }, }; // // method::method // ircd::resource::method::method(struct resource &resource, const string_view &name, handler function) :method { resource, name, std::move(function), {} } { } ircd::resource::method::method(struct resource &resource, const string_view &name, handler function, struct opts opts) :resource { &resource } ,name { name } ,function { std::move(function) } ,opts { std::make_unique(std::move(opts)) } ,stats { std::make_unique() } ,methods_it{[this, &name] { const auto iit { this->resource->methods.emplace(this->name, this) }; if(!iit.second) throw error { "Resource method '%s' already registered to '%s'", name, this->resource->path }; return unique_const_iterator { this->resource->methods, iit.first }; }()} { } ircd::resource::method::~method() noexcept { assert(resource); if(stats && stats->pending) log::dwarning { "Resource '%s' method '%s' still waiting for %zu pending requests", resource->path, name, stats->pending }; idle_dock.wait([this] { return !stats || stats->pending == 0; }); } ircd::resource::response ircd::resource::method::operator()(client &client, const http::request::head &head, const string_view &content_partial) try { const unwind on_idle{[this] { if(stats->pending == 0) idle_dock.notify_all(); }}; ++stats->requests; const scope_count pending { stats->pending }; // Bail out if the method limited the amount of content and it was exceeded. if(!content_length_acceptable(head)) throw http::error { http::PAYLOAD_TOO_LARGE }; // Check if the resource method wants a specific MIME type. If no option // is given by the resource then any Content-Type by the client will pass. if(!mime_type_acceptable(head)) throw http::error { http::UNSUPPORTED_MEDIA_TYPE }; // This timer will keep the request from hanging forever for whatever // reason. The resource method may want to do its own timing and can // disable this in its options structure. const auto &method_timeout { opts->timeout != 0s? opts->timeout: seconds(default_timeout) }; const net::scope_timeout timeout { *client.sock, method_timeout, [this, &client] (const bool &timed_out) { if(timed_out) this->handle_timeout(client); } }; // Content that hasn't yet arrived is remaining const size_t content_remain { head.content_length - client.content_consumed }; // View of the content that will be passed to the resource handler. Starts // with the content received so far which is actually in the head's buffer. // One of three things can happen now: // // - There is no more content so we pass this as-is right to the resource. // - There is more content, so we allocate a content buffer, copy what we // have to it, read the rest off the socket, and then reassign this view. // - There is more content, but the resource wants to read it off the // socket on its own terms, so we pass this as-is. string_view content { content_partial }; if(content_remain && ~opts->flags & CONTENT_DISCRETION) { // Copy any partial content to the final contiguous allocated buffer; client.content_buffer = unique_buffer{head.content_length}; memcpy(data(client.content_buffer), data(content_partial), size(content_partial)); // Setup a window inside the buffer for the remaining socket read. const mutable_buffer content_remain_buffer { data(client.content_buffer) + size(content_partial), content_remain }; // Read the remaining content off the socket. client.content_consumed += read_all(*client.sock, content_remain_buffer); assert(client.content_consumed == head.content_length); content = string_view { data(client.content_buffer), head.content_length }; } // We take the extra step here to clear the assignment to client.request // when this request stack has finished for two reasons: // - It allows other ctxs to peep at the client::list to see what this // client/ctx/request is currently working on with some more safety. // - It prevents an easy source for stale refs wrt the longpoll thing. const scope_restore client_request { client.request, resource::request { head, content } }; client.request.params = lstrip(head.path, resource->path); client.request.params = rstrip(client.request.params, '/'); client.request.parv = { client.request.param, tokens(client.request.params, '/', client.request.param) }; const unwind_nominal completions{[this] { ++stats->completions; }}; // Finally handle the request. return call_handler(client, client.request); } catch(const http::error &e) { if(unlikely(e.code == http::INTERNAL_SERVER_ERROR)) ++stats->internal_errors; throw; } catch(const std::system_error &) { throw; } catch(const ctx::interrupted &) { throw; } catch(...) { ++stats->internal_errors; throw; } ircd::resource::response ircd::resource::method::call_handler(client &client, resource::request &request) try { return function(client, request); } catch(const ctx::timeout &e) { throw http::error { "%s", http::REQUEST_TIMEOUT, e.what() }; } catch(const mods::unavailable &e) { throw http::error { "%s", http::SERVICE_UNAVAILABLE, e.what() }; } catch(const std::bad_function_call &e) { throw http::error { "%s", http::SERVICE_UNAVAILABLE, e.what() }; } catch(const std::out_of_range &e) { throw http::error { "%s", http::NOT_FOUND, e.what() }; } void ircd::resource::method::handle_timeout(client &client) const { log::derror { log, "%s Timed out in %s `%s'", client.loghead(), name, resource->path }; stats->timeouts++; // The interrupt is effective when the socket has already been // closed and/or the client is still stuck in a request for // some reason. if(client.reqctx) ctx::interrupt(*client.reqctx); //TODO: If we know that no response has been sent yet //TODO: we can respond with http::REQUEST_TIMEOUT instead. client.close(net::dc::RST, net::close_ignore); } bool ircd::resource::method::mime_type_acceptable(const http::request::head &head) const { assert(opts); const auto &[required_registry, required_format] { opts->mime }; const auto &[supplied, charset] { split(head.content_type, ';') }; const auto &[supplied_registry, supplied_format] { split(supplied, '/') }; const bool match[] { !required_registry || iequals(required_registry, supplied_registry), !required_format || iequals(required_format, supplied_format), }; return all(match); } bool ircd::resource::method::content_length_acceptable(const http::request::head &head) const { assert(opts); assert(opts->payload_max != 0UL); const auto &payload_max { opts->payload_max != -1UL? opts->payload_max: size_t(default_payload_max) }; return head.content_length <= payload_max; } /////////////////////////////////////////////////////////////////////////////// // // resource/response.h // // // resource::response::chunked // ircd::resource::response::chunked::chunked(client &client, const http::code &code, const size_t &buffer_size) :chunked { client, code, "application/json; charset=utf-8"_sv, string_view{}, buffer_size } { } ircd::resource::response::chunked::chunked(client &client, const http::code &code, const vector_view &headers, const size_t &buffer_size) :chunked { client, code, "application/json; charset=utf-8"_sv, headers, buffer_size } { } ircd::resource::response::chunked::chunked(client &client, const http::code &code, const string_view &content_type, const vector_view &headers, const size_t &buffer_size) :chunked { client, code, content_type, [&headers] { // Note that the headers which are composed into this buffer are // copied again before the response goes out from resource::response. // There must not be any context switch between now and that copy so // we can return a string_view of this TLS buffer. thread_local char buffer[4_KiB]; window_buffer sb{buffer}; http::write(sb, headers); return string_view{sb.completed()}; }(), buffer_size } { } decltype(ircd::resource::response::chunked::default_buffer_size) ircd::resource::response::chunked::default_buffer_size { { "name", "ircd.resource.response.chunked.buffer_size" }, { "default", long(128_KiB) }, }; ircd::resource::response::chunked::chunked(client &client, const http::code &code, const string_view &content_type, const string_view &headers, const size_t &buffer_size) :response { client, code, content_type, size_t(-1), headers } ,c { &client } ,buf { buffer_size } { assert(!empty(content_type)); } ircd::resource::response::chunked::~chunked() noexcept try { if(!c) return; if(std::uncaught_exceptions()) { log::derror { log, "%s HTTP response chunks:%u wrote:%zu flushed:%zu :stream interrupted...", c->loghead(), count, wrote, flushed, }; c->close(net::dc::SSL_NOTIFY, net::close_ignore); return; } finish(); } catch(...) { return; } std::function ircd::resource::response::chunked::flusher() { return std::bind(&chunked::flush, this, ph::_1); } bool ircd::resource::response::chunked::finish() { if(!c) return false; write(const_buffer{}, false); assert(finished); c = nullptr; return true; } ircd::const_buffer ircd::resource::response::chunked::flush(const const_buffer &buf) { assert(size(buf) <= size(this->buf) || empty(this->buf)); const size_t wrote { write(buf, true) }; assert(wrote > 0 || empty(buf)); const size_t flushed { std::min(size(buf), wrote) }; assert(flushed <= size(buf)); this->flushed += flushed; assert(this->flushed <= this->wrote); return const_buffer { data(buf), flushed }; } size_t ircd::resource::response::chunked::write(const const_buffer &chunk, const bool &ignore_empty) try { assert(size(chunk) <= size(this->buf) || empty(this->buf)); assert(!finished); if(!c) return 0UL; if(empty(chunk) && ignore_empty) return 0UL; char headbuf[32]; const size_t wrote { this->wrote }; //TODO: bring iov from net::socket -> net::write_() -> client::write_() const auto head { http::writechunk(headbuf, size(chunk)) }; this->wrote += c->write_all(head); this->wrote += !empty(chunk)? c->write_all(chunk) : 0UL; this->wrote += c->write_all("\r\n"_sv); finished |= empty(chunk); count++; assert(this->wrote >= wrote); assert(this->wrote >= 2 || !finished); return this->wrote - wrote; } catch(...) { this->c = nullptr; throw; } // // resource::response // // A buffer of this size will be passed to the socket and sent // out and must be on stack. decltype(ircd::resource::response::HEAD_BUF_SZ) ircd::resource::response::HEAD_BUF_SZ { 4_KiB }; ircd::resource::response::response(client &client, const http::code &code) :response{client, json::object{json::empty_object}, code} { } ircd::resource::response::response(client &client, const http::code &code, const json::iov &members) :response{client, members, code} { } ircd::resource::response::response(client &client, const json::members &members, const http::code &code) :response{client, code, members} { } ircd::resource::response::response(client &client, const json::value &value, const http::code &code) :response{client, code, value} { } ircd::resource::response::response(client &client, const http::code &code, const json::value &value) try { const auto size { serialized(value) }; const unique_buffer buffer { size }; switch(type(value)) { case json::ARRAY: { response(client, json::array{stringify(mutable_buffer{buffer}, value)}, code); return; } case json::OBJECT: { response(client, json::object{stringify(mutable_buffer{buffer}, value)}, code); return; } default: throw http::error { "Cannot send json::%s as response content", http::INTERNAL_SERVER_ERROR, type(value), }; } } catch(const json::error &e) { throw http::error { "Generator Protection: %s", http::INTERNAL_SERVER_ERROR, e.what(), }; } ircd::resource::response::response(client &client, const http::code &code, const json::members &members) try { const auto size { serialized(members) }; const unique_buffer buffer { size }; const json::object object { stringify(mutable_buffer{buffer}, members) }; response(client, object, code); } catch(const json::error &e) { throw http::error { "Generator Protection: %s", http::INTERNAL_SERVER_ERROR, e.what() }; } ircd::resource::response::response(client &client, const json::iov &members, const http::code &code) try { const auto size { serialized(members) }; const unique_buffer buffer { size }; const json::object object { stringify(mutable_buffer{buffer}, members) }; response(client, object, code); } catch(const json::error &e) { throw http::error { "Generator Protection: %s", http::INTERNAL_SERVER_ERROR, e.what(), }; } ircd::resource::response::response(client &client, const json::object &object, const http::code &code) { static const string_view content_type { "application/json; charset=utf-8" }; assert(json::valid(object, std::nothrow)); response(client, object, content_type, code); } ircd::resource::response::response(client &client, const json::array &array, const http::code &code) { static const string_view content_type { "application/json; charset=utf-8" }; assert(json::valid(array, std::nothrow)); response(client, array, content_type, code); } ircd::resource::response::response(client &client, const string_view &content, const string_view &content_type, const http::code &code, const vector_view &headers) { assert(empty(content) || !empty(content_type)); // contents of this buffer get copied again when further passed to // response{}; we can get this off the stack if that remains true. thread_local char buffer[4_KiB]; window_buffer sb{buffer}; { const critical_assertion ca; http::write(sb, headers); } response { client, content, content_type, code, string_view{sb.completed()} }; } ircd::resource::response::response(client &client, const string_view &content, const string_view &content_type, const http::code &code, const string_view &headers) { assert(empty(content) || !empty(content_type)); // Head gets sent response { client, code, content_type, size(content), headers }; // All content gets sent const size_t written { size(content)? client.write_all(content): 0 }; assert(written == size(content)); } decltype(ircd::resource::response::access_control_allow_origin) ircd::resource::response::access_control_allow_origin { { "name", "ircd.resource.access_control.allow_origin" }, { "default", "*" } }; __attribute__((stack_protect)) ircd::resource::response::response(client &client, const http::code &code, const string_view &content_type, const size_t &content_length, const string_view &headers) { assert(!content_length || !empty(content_type)); const auto request_time { client.timer.at() }; thread_local char rtime_buf[32]; const string_view rtime { pretty(rtime_buf, request_time, true) }; const http::header headers_addl[] { { "X-IRCd-Request-Timer", rtime }, { "Access-Control-Allow-Origin", string_view(access_control_allow_origin) }, }; char head_buf[HEAD_BUF_SZ]; window_buffer head{head_buf}; http::response { head, code, content_length, content_type, headers, headers_addl, }; // Maximum size is realistically ok but ideally a small // maximum; this exception should hit the developer in testing. if(unlikely(!head.remaining())) throw panic { "HTTP headers too large for buffer of %zu", sizeof(head_buf) }; size_t wrote{0}; std::exception_ptr eptr; try { wrote = client.write_all(head.completed()); } catch(...) { eptr = std::current_exception(); } #ifdef RB_DEBUG const log::level level { ushort(code) >= 200 && ushort(code) < 300? log::level::DEBUG: ushort(code) >= 300 && ushort(code) < 400? log::level::DWARNING: ushort(code) >= 400 && ushort(code) < 500? log::level::DERROR: ushort(code) == 500? log::level::ERROR: ushort(code) > 500? log::level::DERROR: log::level::ERROR }; log::logf { log, level, "%s HTTP %u `%s' %s in %s; %s content-length:%s wrote:%zu %s%s", client.loghead(), uint(code), client.request.head.path, http::status(code), rtime, content_type, ssize_t(content_length) >= 0? lex_cast(content_length): "chunked"_sv, wrote, eptr? "error:"_sv: string_view{}, what(eptr) }; #endif if(eptr) std::rethrow_exception(eptr); assert(wrote == size(head.completed())); } /////////////////////////////////////////////////////////////////////////////// // // resource/redirect.h // // // redirect::permanent::permanent // ircd::resource::redirect::permanent::permanent(const string_view &old_path, const string_view &new_path, struct opts opts) :resource { old_path, std::move(opts) } ,new_path { new_path } ,_options { *this, "OPTIONS", std::bind(&permanent::handler, this, ph::_1, ph::_2) } ,_trace { *this, "TRACE", std::bind(&permanent::handler, this, ph::_1, ph::_2) } ,_head { *this, "HEAD", std::bind(&permanent::handler, this, ph::_1, ph::_2) } ,_get { *this, "GET", std::bind(&permanent::handler, this, ph::_1, ph::_2) } ,_put { *this, "PUT", std::bind(&permanent::handler, this, ph::_1, ph::_2) } ,_post { *this, "POST", std::bind(&permanent::handler, this, ph::_1, ph::_2) } ,_patch { *this, "PATCH", std::bind(&permanent::handler, this, ph::_1, ph::_2) } ,_delete { *this, "DELETE", std::bind(&permanent::handler, this, ph::_1, ph::_2) } { } ircd::resource::response ircd::resource::redirect::permanent::handler(client &client, const request &request) { thread_local char buf[response::HEAD_BUF_SZ]; const string_view postfix { lstrip(request.head.uri, this->path) }; const string_view location{fmt::sprintf { buf, "%s/%s", rstrip(this->new_path, '/'), lstrip(postfix, '/') }}; return response { client, {}, {}, http::PERMANENT_REDIRECT, { http::header { "Location", location } } }; }