/* * nghttp2 - HTTP/2 C Library * * Copyright (c) 2012 Tatsuhiro Tsujikawa * * Permission is hereby granted, free of charge, to any person obtaining * a copy of this software and associated documentation files (the * "Software"), to deal in the Software without restriction, including * without limitation the rights to use, copy, modify, merge, publish, * distribute, sublicense, and/or sell copies of the Software, and to * permit persons to whom the Software is furnished to do so, subject to * the following conditions: * * The above copyright notice and this permission notice shall be * included in all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ #include "shrpx_http.h" #include "shrpx_config.h" #include "shrpx_log.h" #include "http2.h" #include "util.h" using namespace nghttp2; namespace shrpx { namespace http { std::string create_error_html(unsigned int status_code) { std::string res; res.reserve(512); auto status = http2::get_status_string(status_code); res += R"()"; res += status; res += "

"; res += status; res += "

"; return res; } std::string create_via_header_value(int major, int minor) { std::string hdrs; hdrs += static_cast(major + '0'); if (major < 2) { hdrs += '.'; hdrs += static_cast(minor + '0'); } hdrs += " nghttpx"; return hdrs; } std::string create_forwarded(int params, const StringRef &node_by, const std::string &node_for, const std::string &host, const std::string &proto) { std::string res; if ((params & FORWARDED_BY) && !node_by.empty()) { // This must be quoted-string unless it is obfuscated version // (which starts with "_") or some special value (e.g., // "localhost" for UNIX domain socket), since ':' is not allowed // in token. ':' is used to separate host and port. if (node_by[0] == '_' || node_by[0] == 'l') { res += "by="; res += node_by; res += ";"; } else { res += "by=\""; res += node_by; res += "\";"; } } if ((params & FORWARDED_FOR) && !node_for.empty()) { // We only quote IPv6 literal address only, which starts with '['. if (node_for[0] == '[') { res += "for=\""; res += node_for; res += "\";"; } else { res += "for="; res += node_for; res += ";"; } } if ((params & FORWARDED_HOST) && !host.empty()) { // Just be quoted to skip checking characters. res += "host=\""; res += host; res += "\";"; } if ((params & FORWARDED_PROTO) && !proto.empty()) { // Scheme production rule only allow characters which are all in // token. res += "proto="; res += proto; res += ";"; } if (res.empty()) { return res; } res.erase(res.size() - 1); return res; } std::string colorizeHeaders(const char *hdrs) { std::string nhdrs; const char *p = strchr(hdrs, '\n'); if (!p) { // Not valid HTTP header return hdrs; } nhdrs.append(hdrs, p + 1); ++p; while (1) { const char *np = strchr(p, ':'); if (!np) { nhdrs.append(p); break; } nhdrs += TTY_HTTP_HD; nhdrs.append(p, np); nhdrs += TTY_RST; p = np; np = strchr(p, '\n'); if (!np) { nhdrs.append(p); break; } nhdrs.append(p, np + 1); p = np + 1; } return nhdrs; } ssize_t select_padding_callback(nghttp2_session *session, const nghttp2_frame *frame, size_t max_payload, void *user_data) { return std::min(max_payload, frame->hd.length + get_config()->padding); } } // namespace http } // namespace shrpx