780 lines
27 KiB
C++
780 lines
27 KiB
C++
/*
|
|
* 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.
|
|
*/
|
|
#ifndef UTIL_H
|
|
#define UTIL_H
|
|
|
|
#include "nghttp2_config.h"
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
# include <unistd.h>
|
|
#endif // HAVE_UNISTD_H
|
|
#include <getopt.h>
|
|
#ifdef HAVE_NETDB_H
|
|
# include <netdb.h>
|
|
#endif // HAVE_NETDB_H
|
|
|
|
#include <cmath>
|
|
#include <cstring>
|
|
#include <cassert>
|
|
#include <vector>
|
|
#include <string>
|
|
#include <algorithm>
|
|
#include <sstream>
|
|
#include <memory>
|
|
#include <chrono>
|
|
#include <map>
|
|
#include <random>
|
|
|
|
#include "http-parser/http_parser.h"
|
|
|
|
#include "template.h"
|
|
#include "network.h"
|
|
#include "allocator.h"
|
|
|
|
namespace nghttp2 {
|
|
|
|
constexpr auto NGHTTP2_H2_ALPN = StringRef::from_lit("\x2h2");
|
|
constexpr auto NGHTTP2_H2 = StringRef::from_lit("h2");
|
|
|
|
// The additional HTTP/2 protocol ALPN protocol identifier we also
|
|
// supports for our applications to make smooth migration into final
|
|
// h2 ALPN ID.
|
|
constexpr auto NGHTTP2_H2_16_ALPN = StringRef::from_lit("\x5h2-16");
|
|
constexpr auto NGHTTP2_H2_16 = StringRef::from_lit("h2-16");
|
|
|
|
constexpr auto NGHTTP2_H2_14_ALPN = StringRef::from_lit("\x5h2-14");
|
|
constexpr auto NGHTTP2_H2_14 = StringRef::from_lit("h2-14");
|
|
|
|
constexpr auto NGHTTP2_H1_1_ALPN = StringRef::from_lit("\x8http/1.1");
|
|
constexpr auto NGHTTP2_H1_1 = StringRef::from_lit("http/1.1");
|
|
|
|
constexpr size_t NGHTTP2_MAX_UINT64_DIGITS = str_size("18446744073709551615");
|
|
|
|
namespace util {
|
|
|
|
inline bool is_alpha(const char c) {
|
|
return ('A' <= c && c <= 'Z') || ('a' <= c && c <= 'z');
|
|
}
|
|
|
|
inline bool is_digit(const char c) { return '0' <= c && c <= '9'; }
|
|
|
|
inline bool is_hex_digit(const char c) {
|
|
return is_digit(c) || ('A' <= c && c <= 'F') || ('a' <= c && c <= 'f');
|
|
}
|
|
|
|
// Returns true if |s| is hex string.
|
|
bool is_hex_string(const StringRef &s);
|
|
|
|
bool in_rfc3986_unreserved_chars(const char c);
|
|
|
|
bool in_rfc3986_sub_delims(const char c);
|
|
|
|
// Returns true if |c| is in token (HTTP-p1, Section 3.2.6)
|
|
bool in_token(char c);
|
|
|
|
bool in_attr_char(char c);
|
|
|
|
// Returns integer corresponding to hex notation |c|. If
|
|
// is_hex_digit(c) is false, it returns 256.
|
|
uint32_t hex_to_uint(char c);
|
|
|
|
std::string percent_encode(const unsigned char *target, size_t len);
|
|
|
|
std::string percent_encode(const std::string &target);
|
|
|
|
// percent-encode path component of URI |s|.
|
|
std::string percent_encode_path(const std::string &s);
|
|
|
|
template <typename InputIt>
|
|
std::string percent_decode(InputIt first, InputIt last) {
|
|
std::string result;
|
|
result.resize(last - first);
|
|
auto p = std::begin(result);
|
|
for (; first != last; ++first) {
|
|
if (*first != '%') {
|
|
*p++ = *first;
|
|
continue;
|
|
}
|
|
|
|
if (first + 1 != last && first + 2 != last && is_hex_digit(*(first + 1)) &&
|
|
is_hex_digit(*(first + 2))) {
|
|
*p++ = (hex_to_uint(*(first + 1)) << 4) + hex_to_uint(*(first + 2));
|
|
first += 2;
|
|
continue;
|
|
}
|
|
|
|
*p++ = *first;
|
|
}
|
|
result.resize(p - std::begin(result));
|
|
return result;
|
|
}
|
|
|
|
StringRef percent_decode(BlockAllocator &balloc, const StringRef &src);
|
|
|
|
// Percent encode |target| if character is not in token or '%'.
|
|
StringRef percent_encode_token(BlockAllocator &balloc, const StringRef &target);
|
|
|
|
// Returns quotedString version of |target|. Currently, this function
|
|
// just replace '"' with '\"'.
|
|
StringRef quote_string(BlockAllocator &balloc, const StringRef &target);
|
|
|
|
std::string format_hex(const unsigned char *s, size_t len);
|
|
|
|
template <size_t N> std::string format_hex(const unsigned char (&s)[N]) {
|
|
return format_hex(s, N);
|
|
}
|
|
|
|
template <size_t N> std::string format_hex(const std::array<uint8_t, N> &s) {
|
|
return format_hex(s.data(), s.size());
|
|
}
|
|
|
|
StringRef format_hex(BlockAllocator &balloc, const StringRef &s);
|
|
|
|
static constexpr char LOWER_XDIGITS[] = "0123456789abcdef";
|
|
|
|
template <typename OutputIt>
|
|
OutputIt format_hex(OutputIt it, const StringRef &s) {
|
|
for (auto cc : s) {
|
|
uint8_t c = cc;
|
|
*it++ = LOWER_XDIGITS[c >> 4];
|
|
*it++ = LOWER_XDIGITS[c & 0xf];
|
|
}
|
|
|
|
return it;
|
|
}
|
|
|
|
// decode_hex decodes hex string |s|, returns the decoded byte string.
|
|
// This function assumes |s| is hex string, that is is_hex_string(s)
|
|
// == true.
|
|
StringRef decode_hex(BlockAllocator &balloc, const StringRef &s);
|
|
|
|
// Returns given time |t| from epoch in HTTP Date format (e.g., Mon,
|
|
// 10 Oct 2016 10:25:58 GMT).
|
|
std::string http_date(time_t t);
|
|
// Writes given time |t| from epoch in HTTP Date format into the
|
|
// buffer pointed by |res|. The buffer must be at least 29 bytes
|
|
// long. This function returns the one beyond the last position.
|
|
char *http_date(char *res, time_t t);
|
|
|
|
// Returns given time |t| from epoch in Common Log format (e.g.,
|
|
// 03/Jul/2014:00:19:38 +0900)
|
|
std::string common_log_date(time_t t);
|
|
// Writes given time |t| from epoch in Common Log format into the
|
|
// buffer pointed by |res|. The buffer must be at least 26 bytes
|
|
// long. This function returns the one beyond the last position.
|
|
char *common_log_date(char *res, time_t t);
|
|
|
|
// Returns given millisecond |ms| from epoch in ISO 8601 format (e.g.,
|
|
// 2014-11-15T12:58:24.741Z or 2014-11-15T12:58:24.741+09:00)
|
|
std::string iso8601_date(int64_t ms);
|
|
// Writes given time |t| from epoch in ISO 8601 format into the buffer
|
|
// pointed by |res|. The buffer must be at least 29 bytes long. This
|
|
// function returns the one beyond the last position.
|
|
char *iso8601_date(char *res, int64_t ms);
|
|
|
|
time_t parse_http_date(const StringRef &s);
|
|
|
|
// Parses time formatted as "MMM DD HH:MM:SS YYYY [GMT]" (e.g., Feb 3
|
|
// 00:55:52 2015 GMT), which is specifically used by OpenSSL
|
|
// ASN1_TIME_print().
|
|
time_t parse_openssl_asn1_time_print(const StringRef &s);
|
|
|
|
char upcase(char c);
|
|
|
|
inline char lowcase(char c) {
|
|
constexpr static unsigned char tbl[] = {
|
|
0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
|
|
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29,
|
|
30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44,
|
|
45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59,
|
|
60, 61, 62, 63, 64, 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j',
|
|
'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y',
|
|
'z', 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104,
|
|
105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119,
|
|
120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134,
|
|
135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149,
|
|
150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164,
|
|
165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179,
|
|
180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194,
|
|
195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209,
|
|
210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224,
|
|
225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239,
|
|
240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254,
|
|
255,
|
|
};
|
|
return tbl[static_cast<unsigned char>(c)];
|
|
}
|
|
|
|
template <typename InputIterator1, typename InputIterator2>
|
|
bool starts_with(InputIterator1 first1, InputIterator1 last1,
|
|
InputIterator2 first2, InputIterator2 last2) {
|
|
if (last1 - first1 < last2 - first2) {
|
|
return false;
|
|
}
|
|
return std::equal(first2, last2, first1);
|
|
}
|
|
|
|
template <typename S, typename T> bool starts_with(const S &a, const T &b) {
|
|
return starts_with(a.begin(), a.end(), b.begin(), b.end());
|
|
}
|
|
|
|
struct CaseCmp {
|
|
bool operator()(char lhs, char rhs) const {
|
|
return lowcase(lhs) == lowcase(rhs);
|
|
}
|
|
};
|
|
|
|
template <typename InputIterator1, typename InputIterator2>
|
|
bool istarts_with(InputIterator1 first1, InputIterator1 last1,
|
|
InputIterator2 first2, InputIterator2 last2) {
|
|
if (last1 - first1 < last2 - first2) {
|
|
return false;
|
|
}
|
|
return std::equal(first2, last2, first1, CaseCmp());
|
|
}
|
|
|
|
template <typename S, typename T> bool istarts_with(const S &a, const T &b) {
|
|
return istarts_with(a.begin(), a.end(), b.begin(), b.end());
|
|
}
|
|
|
|
template <typename T, typename CharT, size_t N>
|
|
bool istarts_with_l(const T &a, const CharT (&b)[N]) {
|
|
return istarts_with(a.begin(), a.end(), b, b + N - 1);
|
|
}
|
|
|
|
template <typename InputIterator1, typename InputIterator2>
|
|
bool ends_with(InputIterator1 first1, InputIterator1 last1,
|
|
InputIterator2 first2, InputIterator2 last2) {
|
|
if (last1 - first1 < last2 - first2) {
|
|
return false;
|
|
}
|
|
return std::equal(first2, last2, last1 - (last2 - first2));
|
|
}
|
|
|
|
template <typename T, typename S> bool ends_with(const T &a, const S &b) {
|
|
return ends_with(a.begin(), a.end(), b.begin(), b.end());
|
|
}
|
|
|
|
template <typename T, typename CharT, size_t N>
|
|
bool ends_with_l(const T &a, const CharT (&b)[N]) {
|
|
return ends_with(a.begin(), a.end(), b, b + N - 1);
|
|
}
|
|
|
|
template <typename InputIterator1, typename InputIterator2>
|
|
bool iends_with(InputIterator1 first1, InputIterator1 last1,
|
|
InputIterator2 first2, InputIterator2 last2) {
|
|
if (last1 - first1 < last2 - first2) {
|
|
return false;
|
|
}
|
|
return std::equal(first2, last2, last1 - (last2 - first2), CaseCmp());
|
|
}
|
|
|
|
template <typename T, typename S> bool iends_with(const T &a, const S &b) {
|
|
return iends_with(a.begin(), a.end(), b.begin(), b.end());
|
|
}
|
|
|
|
template <typename T, typename CharT, size_t N>
|
|
bool iends_with_l(const T &a, const CharT (&b)[N]) {
|
|
return iends_with(a.begin(), a.end(), b, b + N - 1);
|
|
}
|
|
|
|
template <typename InputIt1, typename InputIt2>
|
|
bool strieq(InputIt1 first1, InputIt1 last1, InputIt2 first2, InputIt2 last2) {
|
|
if (std::distance(first1, last1) != std::distance(first2, last2)) {
|
|
return false;
|
|
}
|
|
|
|
return std::equal(first1, last1, first2, CaseCmp());
|
|
}
|
|
|
|
template <typename T, typename S> bool strieq(const T &a, const S &b) {
|
|
return strieq(a.begin(), a.end(), b.begin(), b.end());
|
|
}
|
|
|
|
template <typename CharT, typename InputIt, size_t N>
|
|
bool strieq_l(const CharT (&a)[N], InputIt b, size_t blen) {
|
|
return strieq(a, a + (N - 1), b, b + blen);
|
|
}
|
|
|
|
template <typename CharT, size_t N, typename T>
|
|
bool strieq_l(const CharT (&a)[N], const T &b) {
|
|
return strieq(a, a + (N - 1), b.begin(), b.end());
|
|
}
|
|
|
|
template <typename InputIt1, typename InputIt2>
|
|
bool streq(InputIt1 first1, InputIt1 last1, InputIt2 first2, InputIt2 last2) {
|
|
if (std::distance(first1, last1) != std::distance(first2, last2)) {
|
|
return false;
|
|
}
|
|
return std::equal(first1, last1, first2);
|
|
}
|
|
|
|
template <typename T, typename S> bool streq(const T &a, const S &b) {
|
|
return streq(a.begin(), a.end(), b.begin(), b.end());
|
|
}
|
|
|
|
template <typename CharT, typename InputIt, size_t N>
|
|
bool streq_l(const CharT (&a)[N], InputIt b, size_t blen) {
|
|
return streq(a, a + (N - 1), b, b + blen);
|
|
}
|
|
|
|
template <typename CharT, size_t N, typename T>
|
|
bool streq_l(const CharT (&a)[N], const T &b) {
|
|
return streq(a, a + (N - 1), b.begin(), b.end());
|
|
}
|
|
|
|
// Returns true if |a| contains |b|. If both |a| and |b| are empty,
|
|
// this function returns false.
|
|
template <typename S, typename T> bool strifind(const S &a, const T &b) {
|
|
return std::search(a.begin(), a.end(), b.begin(), b.end(), CaseCmp()) !=
|
|
a.end();
|
|
}
|
|
|
|
template <typename InputIt> void inp_strlower(InputIt first, InputIt last) {
|
|
std::transform(first, last, first, lowcase);
|
|
}
|
|
|
|
// Lowercase |s| in place.
|
|
inline void inp_strlower(std::string &s) {
|
|
inp_strlower(std::begin(s), std::end(s));
|
|
}
|
|
|
|
// Returns string representation of |n| with 2 fractional digits.
|
|
std::string dtos(double n);
|
|
|
|
template <typename T> std::string utos(T n) {
|
|
std::string res;
|
|
if (n == 0) {
|
|
res = "0";
|
|
return res;
|
|
}
|
|
size_t nlen = 0;
|
|
for (auto t = n; t; t /= 10, ++nlen)
|
|
;
|
|
res.resize(nlen);
|
|
for (; n; n /= 10) {
|
|
res[--nlen] = (n % 10) + '0';
|
|
}
|
|
return res;
|
|
}
|
|
|
|
template <typename T, typename OutputIt> OutputIt utos(OutputIt dst, T n) {
|
|
if (n == 0) {
|
|
*dst++ = '0';
|
|
return dst;
|
|
}
|
|
size_t nlen = 0;
|
|
for (auto t = n; t; t /= 10, ++nlen)
|
|
;
|
|
auto p = dst + nlen;
|
|
auto res = p;
|
|
for (; n; n /= 10) {
|
|
*--p = (n % 10) + '0';
|
|
}
|
|
return res;
|
|
}
|
|
|
|
template <typename T>
|
|
StringRef make_string_ref_uint(BlockAllocator &balloc, T n) {
|
|
auto iov = make_byte_ref(balloc, NGHTTP2_MAX_UINT64_DIGITS + 1);
|
|
auto p = iov.base;
|
|
p = util::utos(p, n);
|
|
*p = '\0';
|
|
return StringRef{iov.base, p};
|
|
}
|
|
|
|
template <typename T> std::string utos_unit(T n) {
|
|
char u = 0;
|
|
if (n >= (1 << 30)) {
|
|
u = 'G';
|
|
n /= (1 << 30);
|
|
} else if (n >= (1 << 20)) {
|
|
u = 'M';
|
|
n /= (1 << 20);
|
|
} else if (n >= (1 << 10)) {
|
|
u = 'K';
|
|
n /= (1 << 10);
|
|
}
|
|
if (u == 0) {
|
|
return utos(n);
|
|
}
|
|
return utos(n) + u;
|
|
}
|
|
|
|
// Like utos_unit(), but 2 digits fraction part is followed.
|
|
template <typename T> std::string utos_funit(T n) {
|
|
char u = 0;
|
|
int b = 0;
|
|
if (n >= (1 << 30)) {
|
|
u = 'G';
|
|
b = 30;
|
|
} else if (n >= (1 << 20)) {
|
|
u = 'M';
|
|
b = 20;
|
|
} else if (n >= (1 << 10)) {
|
|
u = 'K';
|
|
b = 10;
|
|
}
|
|
if (b == 0) {
|
|
return utos(n);
|
|
}
|
|
return dtos(static_cast<double>(n) / (1 << b)) + u;
|
|
}
|
|
|
|
extern const char UPPER_XDIGITS[];
|
|
|
|
template <typename T> std::string utox(T n) {
|
|
std::string res;
|
|
if (n == 0) {
|
|
res = "0";
|
|
return res;
|
|
}
|
|
int i = 0;
|
|
T t = n;
|
|
for (; t; t /= 16, ++i)
|
|
;
|
|
res.resize(i);
|
|
--i;
|
|
for (; n; --i, n /= 16) {
|
|
res[i] = UPPER_XDIGITS[(n & 0x0f)];
|
|
}
|
|
return res;
|
|
}
|
|
|
|
void to_token68(std::string &base64str);
|
|
|
|
StringRef to_base64(BlockAllocator &balloc, const StringRef &token68str);
|
|
|
|
void show_candidates(const char *unkopt, const option *options);
|
|
|
|
bool has_uri_field(const http_parser_url &u, http_parser_url_fields field);
|
|
|
|
bool fieldeq(const char *uri1, const http_parser_url &u1, const char *uri2,
|
|
const http_parser_url &u2, http_parser_url_fields field);
|
|
|
|
bool fieldeq(const char *uri, const http_parser_url &u,
|
|
http_parser_url_fields field, const char *t);
|
|
|
|
bool fieldeq(const char *uri, const http_parser_url &u,
|
|
http_parser_url_fields field, const StringRef &t);
|
|
|
|
StringRef get_uri_field(const char *uri, const http_parser_url &u,
|
|
http_parser_url_fields field);
|
|
|
|
uint16_t get_default_port(const char *uri, const http_parser_url &u);
|
|
|
|
bool porteq(const char *uri1, const http_parser_url &u1, const char *uri2,
|
|
const http_parser_url &u2);
|
|
|
|
void write_uri_field(std::ostream &o, const char *uri, const http_parser_url &u,
|
|
http_parser_url_fields field);
|
|
|
|
bool numeric_host(const char *hostname);
|
|
|
|
bool numeric_host(const char *hostname, int family);
|
|
|
|
// Returns numeric address string of |addr|. If getnameinfo() is
|
|
// failed, "unknown" is returned.
|
|
std::string numeric_name(const struct sockaddr *sa, socklen_t salen);
|
|
|
|
// Returns string representation of numeric address and port of
|
|
// |addr|. If address family is AF_UNIX, this return path to UNIX
|
|
// domain socket. Otherwise, the format is like <HOST>:<PORT>. For
|
|
// IPv6 address, address is enclosed by square brackets ([]).
|
|
std::string to_numeric_addr(const Address *addr);
|
|
|
|
// Sets |port| to |addr|.
|
|
void set_port(Address &addr, uint16_t port);
|
|
|
|
// Returns ASCII dump of |data| of length |len|. Only ASCII printable
|
|
// characters are preserved. Other characters are replaced with ".".
|
|
std::string ascii_dump(const uint8_t *data, size_t len);
|
|
|
|
// Returns absolute path of executable path. If argc == 0 or |cwd| is
|
|
// nullptr, this function returns nullptr. If argv[0] starts with
|
|
// '/', this function returns argv[0]. Oterwise return cwd + "/" +
|
|
// argv[0]. If non-null is returned, it is NULL-terminated string and
|
|
// dynamically allocated by malloc. The caller is responsible to free
|
|
// it.
|
|
char *get_exec_path(int argc, char **const argv, const char *cwd);
|
|
|
|
// Validates path so that it does not contain directory traversal
|
|
// vector. Returns true if path is safe. The |path| must start with
|
|
// "/" otherwise returns false. This function should be called after
|
|
// percent-decode was performed.
|
|
bool check_path(const std::string &path);
|
|
|
|
// Returns the |tv| value as 64 bit integer using a microsecond as an
|
|
// unit.
|
|
int64_t to_time64(const timeval &tv);
|
|
|
|
// Returns true if ALPN ID |proto| is supported HTTP/2 protocol
|
|
// identifier.
|
|
bool check_h2_is_selected(const StringRef &proto);
|
|
|
|
// Selects h2 protocol ALPN ID if one of supported h2 versions are
|
|
// present in |in| of length inlen. Returns true if h2 version is
|
|
// selected.
|
|
bool select_h2(const unsigned char **out, unsigned char *outlen,
|
|
const unsigned char *in, unsigned int inlen);
|
|
|
|
// Selects protocol ALPN ID if one of identifiers contained in |protolist| is
|
|
// present in |in| of length inlen. Returns true if identifier is
|
|
// selected.
|
|
bool select_protocol(const unsigned char **out, unsigned char *outlen,
|
|
const unsigned char *in, unsigned int inlen,
|
|
std::vector<std::string> proto_list);
|
|
|
|
// Returns default ALPN protocol list, which only contains supported
|
|
// HTTP/2 protocol identifier.
|
|
std::vector<unsigned char> get_default_alpn();
|
|
|
|
// Parses delimited strings in |s| and returns the array of substring,
|
|
// delimited by |delim|. The any white spaces around substring are
|
|
// treated as a part of substring.
|
|
std::vector<std::string> parse_config_str_list(const StringRef &s,
|
|
char delim = ',');
|
|
|
|
// Parses delimited strings in |s| and returns Substrings in |s|
|
|
// delimited by |delim|. The any white spaces around substring are
|
|
// treated as a part of substring.
|
|
std::vector<StringRef> split_str(const StringRef &s, char delim);
|
|
|
|
// Writes given time |tp| in Common Log format (e.g.,
|
|
// 03/Jul/2014:00:19:38 +0900) in buffer pointed by |out|. The buffer
|
|
// must be at least 27 bytes, including terminal NULL byte. Expected
|
|
// type of |tp| is std::chrono::time_point. This function returns
|
|
// StringRef wrapping the buffer pointed by |out|, and this string is
|
|
// terminated by NULL.
|
|
template <typename T> StringRef format_common_log(char *out, const T &tp) {
|
|
auto t =
|
|
std::chrono::duration_cast<std::chrono::seconds>(tp.time_since_epoch());
|
|
auto p = common_log_date(out, t.count());
|
|
*p = '\0';
|
|
return StringRef{out, p};
|
|
}
|
|
|
|
// Returns given time |tp| in ISO 8601 format (e.g.,
|
|
// 2014-11-15T12:58:24.741Z or 2014-11-15T12:58:24.741+09:00).
|
|
// Expected type of |tp| is std::chrono::time_point
|
|
template <typename T> std::string format_iso8601(const T &tp) {
|
|
auto t = std::chrono::duration_cast<std::chrono::milliseconds>(
|
|
tp.time_since_epoch());
|
|
return iso8601_date(t.count());
|
|
}
|
|
|
|
// Writes given time |tp| in ISO 8601 format (e.g.,
|
|
// 2014-11-15T12:58:24.741Z or 2014-11-15T12:58:24.741+09:00) in
|
|
// buffer pointed by |out|. The buffer must be at least 30 bytes,
|
|
// including terminal NULL byte. Expected type of |tp| is
|
|
// std::chrono::time_point. This function returns StringRef wrapping
|
|
// the buffer pointed by |out|, and this string is terminated by NULL.
|
|
template <typename T> StringRef format_iso8601(char *out, const T &tp) {
|
|
auto t = std::chrono::duration_cast<std::chrono::milliseconds>(
|
|
tp.time_since_epoch());
|
|
auto p = iso8601_date(out, t.count());
|
|
*p = '\0';
|
|
return StringRef{out, p};
|
|
}
|
|
|
|
// Writes given time |tp| in HTTP Date format (e.g., Mon, 10 Oct 2016
|
|
// 10:25:58 GMT) in buffer pointed by |out|. The buffer must be at
|
|
// least 30 bytes, including terminal NULL byte. Expected type of
|
|
// |tp| is std::chrono::time_point. This function returns StringRef
|
|
// wrapping the buffer pointed by |out|, and this string is terminated
|
|
// by NULL.
|
|
template <typename T> StringRef format_http_date(char *out, const T &tp) {
|
|
auto t =
|
|
std::chrono::duration_cast<std::chrono::seconds>(tp.time_since_epoch());
|
|
auto p = http_date(out, t.count());
|
|
*p = '\0';
|
|
return StringRef{out, p};
|
|
}
|
|
|
|
// Return the system precision of the template parameter |Clock| as
|
|
// a nanosecond value of type |Rep|
|
|
template <typename Clock, typename Rep> Rep clock_precision() {
|
|
std::chrono::duration<Rep, std::nano> duration = typename Clock::duration(1);
|
|
|
|
return duration.count();
|
|
}
|
|
|
|
int make_socket_closeonexec(int fd);
|
|
int make_socket_nonblocking(int fd);
|
|
int make_socket_nodelay(int fd);
|
|
|
|
int create_nonblock_socket(int family);
|
|
|
|
bool check_socket_connected(int fd);
|
|
|
|
// Returns the error code (errno) by inspecting SO_ERROR of given
|
|
// |fd|. This function returns the error code if it succeeds, or -1.
|
|
// Returning 0 means no error.
|
|
int get_socket_error(int fd);
|
|
|
|
// Returns true if |host| is IPv6 numeric address (e.g., ::1)
|
|
bool ipv6_numeric_addr(const char *host);
|
|
|
|
// Parses NULL terminated string |s| as unsigned integer and returns
|
|
// the parsed integer. Additionally, if |s| ends with 'k', 'm', 'g'
|
|
// and its upper case characters, multiply the integer by 1024, 1024 *
|
|
// 1024 and 1024 * 1024 respectively. If there is an error, returns
|
|
// -1.
|
|
int64_t parse_uint_with_unit(const char *s);
|
|
// The following overload does not require |s| is NULL terminated.
|
|
int64_t parse_uint_with_unit(const uint8_t *s, size_t len);
|
|
int64_t parse_uint_with_unit(const StringRef &s);
|
|
|
|
// Parses NULL terminated string |s| as unsigned integer and returns
|
|
// the parsed integer. If there is an error, returns -1.
|
|
int64_t parse_uint(const char *s);
|
|
// The following overload does not require |s| is NULL terminated.
|
|
int64_t parse_uint(const uint8_t *s, size_t len);
|
|
int64_t parse_uint(const std::string &s);
|
|
int64_t parse_uint(const StringRef &s);
|
|
|
|
// Parses NULL terminated string |s| as unsigned integer and returns
|
|
// the parsed integer casted to double. If |s| ends with "s", the
|
|
// parsed value's unit is a second. If |s| ends with "ms", the unit
|
|
// is millisecond. Similarly, it also supports 'm' and 'h' for
|
|
// minutes and hours respectively. If none of them are given, the
|
|
// unit is second. This function returns
|
|
// std::numeric_limits<double>::infinity() if error occurs.
|
|
double parse_duration_with_unit(const char *s);
|
|
// The following overload does not require |s| is NULL terminated.
|
|
double parse_duration_with_unit(const uint8_t *s, size_t len);
|
|
double parse_duration_with_unit(const StringRef &s);
|
|
|
|
// Returns string representation of time duration |t|. If t has
|
|
// fractional part (at least more than or equal to 1e-3), |t| is
|
|
// multiplied by 1000 and the unit "ms" is appended. Otherwise, |t|
|
|
// is left as is and "s" is appended.
|
|
std::string duration_str(double t);
|
|
|
|
// Returns string representation of time duration |t|. It appends
|
|
// unit after the formatting. The available units are s, ms and us.
|
|
// The unit which is equal to or less than |t| is used and 2
|
|
// fractional digits follow.
|
|
std::string format_duration(const std::chrono::microseconds &u);
|
|
|
|
// Just like above, but this takes |t| as seconds.
|
|
std::string format_duration(double t);
|
|
|
|
// Creates "host:port" string using given |host| and |port|. If
|
|
// |host| is numeric IPv6 address (e.g., ::1), it is enclosed by "["
|
|
// and "]". If |port| is 80 or 443, port part is omitted.
|
|
StringRef make_http_hostport(BlockAllocator &balloc, const StringRef &host,
|
|
uint16_t port);
|
|
|
|
// Just like make_http_hostport(), but doesn't treat 80 and 443
|
|
// specially.
|
|
std::string make_hostport(const StringRef &host, uint16_t port);
|
|
|
|
StringRef make_hostport(BlockAllocator &balloc, const StringRef &host,
|
|
uint16_t port);
|
|
|
|
// Dumps |src| of length |len| in the format similar to `hexdump -C`.
|
|
void hexdump(FILE *out, const uint8_t *src, size_t len);
|
|
|
|
// Copies 2 byte unsigned integer |n| in host byte order to |buf| in
|
|
// network byte order.
|
|
void put_uint16be(uint8_t *buf, uint16_t n);
|
|
|
|
// Copies 4 byte unsigned integer |n| in host byte order to |buf| in
|
|
// network byte order.
|
|
void put_uint32be(uint8_t *buf, uint32_t n);
|
|
|
|
// Retrieves 2 byte unsigned integer stored in |data| in network byte
|
|
// order and returns it in host byte order.
|
|
uint16_t get_uint16(const uint8_t *data);
|
|
|
|
// Retrieves 4 byte unsigned integer stored in |data| in network byte
|
|
// order and returns it in host byte order.
|
|
uint32_t get_uint32(const uint8_t *data);
|
|
|
|
// Retrieves 8 byte unsigned integer stored in |data| in network byte
|
|
// order and returns it in host byte order.
|
|
uint64_t get_uint64(const uint8_t *data);
|
|
|
|
// Reads mime types file (see /etc/mime.types), and stores extension
|
|
// -> MIME type map in |res|. This function returns 0 if it succeeds,
|
|
// or -1.
|
|
int read_mime_types(std::map<std::string, std::string> &res,
|
|
const char *filename);
|
|
|
|
// Fills random alpha and digit byte to the range [|first|, |last|).
|
|
// Returns the one beyond the |last|.
|
|
template <typename OutputIt, typename Generator>
|
|
OutputIt random_alpha_digit(OutputIt first, OutputIt last, Generator &gen) {
|
|
// If we use uint8_t instead char, gcc 6.2.0 complains by shouting
|
|
// char-array initialized from wide string.
|
|
static constexpr char s[] =
|
|
"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
|
|
std::uniform_int_distribution<> dis(0, 26 * 2 + 10 - 1);
|
|
for (; first != last; ++first) {
|
|
*first = s[dis(gen)];
|
|
}
|
|
return first;
|
|
}
|
|
|
|
// Fills random bytes to the range [|first|, |last|).
|
|
template <typename OutputIt, typename Generator>
|
|
void random_bytes(OutputIt first, OutputIt last, Generator &gen) {
|
|
std::uniform_int_distribution<> dis(0, 255);
|
|
std::generate(first, last, [&dis, &gen]() { return dis(gen); });
|
|
}
|
|
|
|
template <typename OutputIterator, typename CharT, size_t N>
|
|
OutputIterator copy_lit(OutputIterator it, CharT (&s)[N]) {
|
|
return std::copy_n(s, N - 1, it);
|
|
}
|
|
|
|
// Returns x**y
|
|
double int_pow(double x, size_t y);
|
|
|
|
uint32_t hash32(const StringRef &s);
|
|
|
|
// Computes SHA-256 of |s|, and stores it in |buf|. This function
|
|
// returns 0 if it succeeds, or -1.
|
|
int sha256(uint8_t *buf, const StringRef &s);
|
|
|
|
// Computes SHA-1 of |s|, and stores it in |buf|. This function
|
|
// returns 0 if it succeeds, or -1.
|
|
int sha1(uint8_t *buf, const StringRef &s);
|
|
|
|
// Returns host from |hostport|. If host cannot be found in
|
|
// |hostport|, returns empty string. The returned string might not be
|
|
// NULL-terminated.
|
|
StringRef extract_host(const StringRef &hostport);
|
|
|
|
// Returns new std::mt19937 object.
|
|
std::mt19937 make_mt19937();
|
|
|
|
} // namespace util
|
|
|
|
} // namespace nghttp2
|
|
|
|
#endif // UTIL_H
|