nghttpx: Complete HTTP request and response
This commit is contained in:
parent
12425556c1
commit
b0548b4944
|
@ -157,6 +157,7 @@ NGHTTPX_SRCS = \
|
||||||
shrpx_quic_listener.cc shrpx_quic_listener.h \
|
shrpx_quic_listener.cc shrpx_quic_listener.h \
|
||||||
shrpx_quic_connection_handler.cc shrpx_quic_connection_handler.h \
|
shrpx_quic_connection_handler.cc shrpx_quic_connection_handler.h \
|
||||||
shrpx_http3_upstream.cc shrpx_http3_upstream.h \
|
shrpx_http3_upstream.cc shrpx_http3_upstream.h \
|
||||||
|
http3.cc http3.h \
|
||||||
quic.cc quic.h \
|
quic.cc quic.h \
|
||||||
buffer.h memchunk.h template.h allocator.h \
|
buffer.h memchunk.h template.h allocator.h \
|
||||||
xsi_strerror.c xsi_strerror.h
|
xsi_strerror.c xsi_strerror.h
|
||||||
|
|
|
@ -0,0 +1,215 @@
|
||||||
|
/*
|
||||||
|
* nghttp2 - HTTP/2 C Library
|
||||||
|
*
|
||||||
|
* Copyright (c) 2021 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 "http3.h"
|
||||||
|
|
||||||
|
namespace nghttp2 {
|
||||||
|
|
||||||
|
namespace http3 {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
nghttp3_nv make_nv_internal(const std::string &name, const std::string &value,
|
||||||
|
bool never_index, uint8_t nv_flags) {
|
||||||
|
uint8_t flags;
|
||||||
|
|
||||||
|
flags = nv_flags |
|
||||||
|
(never_index ? NGHTTP3_NV_FLAG_NEVER_INDEX : NGHTTP3_NV_FLAG_NONE);
|
||||||
|
|
||||||
|
return {(uint8_t *)name.c_str(), (uint8_t *)value.c_str(), name.size(),
|
||||||
|
value.size(), flags};
|
||||||
|
}
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
nghttp3_nv make_nv_internal(const StringRef &name, const StringRef &value,
|
||||||
|
bool never_index, uint8_t nv_flags) {
|
||||||
|
uint8_t flags;
|
||||||
|
|
||||||
|
flags = nv_flags |
|
||||||
|
(never_index ? NGHTTP3_NV_FLAG_NEVER_INDEX : NGHTTP3_NV_FLAG_NONE);
|
||||||
|
|
||||||
|
return {(uint8_t *)name.c_str(), (uint8_t *)value.c_str(), name.size(),
|
||||||
|
value.size(), flags};
|
||||||
|
}
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
nghttp3_nv make_nv(const std::string &name, const std::string &value,
|
||||||
|
bool never_index) {
|
||||||
|
return make_nv_internal(name, value, never_index, NGHTTP3_NV_FLAG_NONE);
|
||||||
|
}
|
||||||
|
|
||||||
|
nghttp3_nv make_nv(const StringRef &name, const StringRef &value,
|
||||||
|
bool never_index) {
|
||||||
|
return make_nv_internal(name, value, never_index, NGHTTP3_NV_FLAG_NONE);
|
||||||
|
}
|
||||||
|
|
||||||
|
nghttp3_nv make_nv_nocopy(const std::string &name, const std::string &value,
|
||||||
|
bool never_index) {
|
||||||
|
return make_nv_internal(name, value, never_index,
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME |
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_VALUE);
|
||||||
|
}
|
||||||
|
|
||||||
|
nghttp3_nv make_nv_nocopy(const StringRef &name, const StringRef &value,
|
||||||
|
bool never_index) {
|
||||||
|
return make_nv_internal(name, value, never_index,
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME |
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_VALUE);
|
||||||
|
}
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
void copy_headers_to_nva_internal(std::vector<nghttp3_nv> &nva,
|
||||||
|
const HeaderRefs &headers, uint8_t nv_flags,
|
||||||
|
uint32_t flags) {
|
||||||
|
auto it_forwarded = std::end(headers);
|
||||||
|
auto it_xff = std::end(headers);
|
||||||
|
auto it_xfp = std::end(headers);
|
||||||
|
auto it_via = std::end(headers);
|
||||||
|
|
||||||
|
for (auto it = std::begin(headers); it != std::end(headers); ++it) {
|
||||||
|
auto kv = &(*it);
|
||||||
|
if (kv->name.empty() || kv->name[0] == ':') {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
switch (kv->token) {
|
||||||
|
case http2::HD_COOKIE:
|
||||||
|
case http2::HD_CONNECTION:
|
||||||
|
case http2::HD_HOST:
|
||||||
|
case http2::HD_HTTP2_SETTINGS:
|
||||||
|
case http2::HD_KEEP_ALIVE:
|
||||||
|
case http2::HD_PROXY_CONNECTION:
|
||||||
|
case http2::HD_SERVER:
|
||||||
|
case http2::HD_TE:
|
||||||
|
case http2::HD_TRANSFER_ENCODING:
|
||||||
|
case http2::HD_UPGRADE:
|
||||||
|
continue;
|
||||||
|
case http2::HD_EARLY_DATA:
|
||||||
|
if (flags & http2::HDOP_STRIP_EARLY_DATA) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case http2::HD_SEC_WEBSOCKET_ACCEPT:
|
||||||
|
if (flags & http2::HDOP_STRIP_SEC_WEBSOCKET_ACCEPT) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case http2::HD_SEC_WEBSOCKET_KEY:
|
||||||
|
if (flags & http2::HDOP_STRIP_SEC_WEBSOCKET_KEY) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case http2::HD_FORWARDED:
|
||||||
|
if (flags & http2::HDOP_STRIP_FORWARDED) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (it_forwarded == std::end(headers)) {
|
||||||
|
it_forwarded = it;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
kv = &(*it_forwarded);
|
||||||
|
it_forwarded = it;
|
||||||
|
break;
|
||||||
|
case http2::HD_X_FORWARDED_FOR:
|
||||||
|
if (flags & http2::HDOP_STRIP_X_FORWARDED_FOR) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (it_xff == std::end(headers)) {
|
||||||
|
it_xff = it;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
kv = &(*it_xff);
|
||||||
|
it_xff = it;
|
||||||
|
break;
|
||||||
|
case http2::HD_X_FORWARDED_PROTO:
|
||||||
|
if (flags & http2::HDOP_STRIP_X_FORWARDED_PROTO) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (it_xfp == std::end(headers)) {
|
||||||
|
it_xfp = it;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
kv = &(*it_xfp);
|
||||||
|
it_xfp = it;
|
||||||
|
break;
|
||||||
|
case http2::HD_VIA:
|
||||||
|
if (flags & http2::HDOP_STRIP_VIA) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (it_via == std::end(headers)) {
|
||||||
|
it_via = it;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
kv = &(*it_via);
|
||||||
|
it_via = it;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
nva.push_back(
|
||||||
|
make_nv_internal(kv->name, kv->value, kv->no_index, nv_flags));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
void copy_headers_to_nva(std::vector<nghttp3_nv> &nva,
|
||||||
|
const HeaderRefs &headers, uint32_t flags) {
|
||||||
|
copy_headers_to_nva_internal(nva, headers, NGHTTP3_NV_FLAG_NONE, flags);
|
||||||
|
}
|
||||||
|
|
||||||
|
void copy_headers_to_nva_nocopy(std::vector<nghttp3_nv> &nva,
|
||||||
|
const HeaderRefs &headers, uint32_t flags) {
|
||||||
|
copy_headers_to_nva_internal(
|
||||||
|
nva, headers,
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME | NGHTTP3_NV_FLAG_NO_COPY_VALUE, flags);
|
||||||
|
}
|
||||||
|
|
||||||
|
int check_nv(const uint8_t *name, size_t namelen, const uint8_t *value,
|
||||||
|
size_t valuelen) {
|
||||||
|
if (!nghttp3_check_header_name(name, namelen)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (!nghttp3_check_header_value(value, valuelen)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
void dump_nv(FILE *out, const nghttp3_nv *nva, size_t nvlen) {
|
||||||
|
auto end = nva + nvlen;
|
||||||
|
for (; nva != end; ++nva) {
|
||||||
|
fprintf(out, "%s: %s\n", nva->name, nva->value);
|
||||||
|
}
|
||||||
|
fputc('\n', out);
|
||||||
|
fflush(out);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace http3
|
||||||
|
|
||||||
|
} // namespace nghttp2
|
|
@ -0,0 +1,129 @@
|
||||||
|
/*
|
||||||
|
* nghttp2 - HTTP/2 C Library
|
||||||
|
*
|
||||||
|
* Copyright (c) 2021 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 HTTP3_H
|
||||||
|
#define HTTP3_H
|
||||||
|
|
||||||
|
#include "nghttp2_config.h"
|
||||||
|
|
||||||
|
#include <cstring>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include <nghttp3/nghttp3.h>
|
||||||
|
|
||||||
|
#include "http2.h"
|
||||||
|
#include "template.h"
|
||||||
|
|
||||||
|
namespace nghttp2 {
|
||||||
|
|
||||||
|
namespace http3 {
|
||||||
|
|
||||||
|
// Creates nghttp3_nv using |name| and |value| and returns it. The
|
||||||
|
// returned value only references the data pointer to name.c_str() and
|
||||||
|
// value.c_str(). If |no_index| is true, nghttp3_nv flags member has
|
||||||
|
// NGHTTP3_NV_FLAG_NEVER_INDEX flag set.
|
||||||
|
nghttp3_nv make_nv(const std::string &name, const std::string &value,
|
||||||
|
bool never_index = false);
|
||||||
|
|
||||||
|
nghttp3_nv make_nv(const StringRef &name, const StringRef &value,
|
||||||
|
bool never_index = false);
|
||||||
|
|
||||||
|
nghttp3_nv make_nv_nocopy(const std::string &name, const std::string &value,
|
||||||
|
bool never_index = false);
|
||||||
|
|
||||||
|
nghttp3_nv make_nv_nocopy(const StringRef &name, const StringRef &value,
|
||||||
|
bool never_index = false);
|
||||||
|
|
||||||
|
// Create nghttp3_nv from string literal |name| and |value|.
|
||||||
|
template <size_t N, size_t M>
|
||||||
|
constexpr nghttp3_nv make_nv_ll(const char (&name)[N], const char (&value)[M]) {
|
||||||
|
return {(uint8_t *)name, (uint8_t *)value, N - 1, M - 1,
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME | NGHTTP3_NV_FLAG_NO_COPY_VALUE};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create nghttp3_nv from string literal |name| and c-string |value|.
|
||||||
|
template <size_t N>
|
||||||
|
nghttp3_nv make_nv_lc(const char (&name)[N], const char *value) {
|
||||||
|
return {(uint8_t *)name, (uint8_t *)value, N - 1, strlen(value),
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME};
|
||||||
|
}
|
||||||
|
|
||||||
|
template <size_t N>
|
||||||
|
nghttp3_nv make_nv_lc_nocopy(const char (&name)[N], const char *value) {
|
||||||
|
return {(uint8_t *)name, (uint8_t *)value, N - 1, strlen(value),
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME | NGHTTP3_NV_FLAG_NO_COPY_VALUE};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create nghttp3_nv from string literal |name| and std::string
|
||||||
|
// |value|.
|
||||||
|
template <size_t N>
|
||||||
|
nghttp3_nv make_nv_ls(const char (&name)[N], const std::string &value) {
|
||||||
|
return {(uint8_t *)name, (uint8_t *)value.c_str(), N - 1, value.size(),
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME};
|
||||||
|
}
|
||||||
|
|
||||||
|
template <size_t N>
|
||||||
|
nghttp3_nv make_nv_ls_nocopy(const char (&name)[N], const std::string &value) {
|
||||||
|
return {(uint8_t *)name, (uint8_t *)value.c_str(), N - 1, value.size(),
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME | NGHTTP3_NV_FLAG_NO_COPY_VALUE};
|
||||||
|
}
|
||||||
|
|
||||||
|
template <size_t N>
|
||||||
|
nghttp3_nv make_nv_ls_nocopy(const char (&name)[N], const StringRef &value) {
|
||||||
|
return {(uint8_t *)name, (uint8_t *)value.c_str(), N - 1, value.size(),
|
||||||
|
NGHTTP3_NV_FLAG_NO_COPY_NAME | NGHTTP3_NV_FLAG_NO_COPY_VALUE};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Appends headers in |headers| to |nv|. |headers| must be indexed
|
||||||
|
// before this call (its element's token field is assigned). Certain
|
||||||
|
// headers, including disallowed headers in HTTP/3 spec and headers
|
||||||
|
// which require special handling (i.e. via), are not copied. |flags|
|
||||||
|
// is one or more of HeaderBuildOp flags. They tell function that
|
||||||
|
// certain header fields should not be added.
|
||||||
|
void copy_headers_to_nva(std::vector<nghttp3_nv> &nva,
|
||||||
|
const HeaderRefs &headers, uint32_t flags);
|
||||||
|
|
||||||
|
// Just like copy_headers_to_nva(), but this adds
|
||||||
|
// NGHTTP3_NV_FLAG_NO_COPY_NAME and NGHTTP3_NV_FLAG_NO_COPY_VALUE.
|
||||||
|
void copy_headers_to_nva_nocopy(std::vector<nghttp3_nv> &nva,
|
||||||
|
const HeaderRefs &headers, uint32_t flags);
|
||||||
|
|
||||||
|
// Dumps name/value pairs in |nva| to |out|.
|
||||||
|
void dump_nv(FILE *out, const nghttp3_nv *nva, size_t nvlen);
|
||||||
|
|
||||||
|
// Checks the header name/value pair using nghttp3_check_header_name()
|
||||||
|
// and nghttp3_check_header_value(). If both function returns nonzero,
|
||||||
|
// this function returns nonzero.
|
||||||
|
int check_nv(const uint8_t *name, size_t namelen, const uint8_t *value,
|
||||||
|
size_t valuelen);
|
||||||
|
|
||||||
|
// Dumps name/value pairs in |nva| to |out|.
|
||||||
|
void dump_nv(FILE *out, const nghttp3_nv *nva, size_t nvlen);
|
||||||
|
|
||||||
|
} // namespace http3
|
||||||
|
|
||||||
|
} // namespace nghttp2
|
||||||
|
|
||||||
|
#endif // HTTP3_H
|
|
@ -113,13 +113,22 @@ template <typename T> struct Pool {
|
||||||
|
|
||||||
template <typename Memchunk> struct Memchunks {
|
template <typename Memchunk> struct Memchunks {
|
||||||
Memchunks(Pool<Memchunk> *pool)
|
Memchunks(Pool<Memchunk> *pool)
|
||||||
: pool(pool), head(nullptr), tail(nullptr), len(0) {}
|
: pool(pool),
|
||||||
|
head(nullptr),
|
||||||
|
tail(nullptr),
|
||||||
|
len(0),
|
||||||
|
mark(nullptr),
|
||||||
|
mark_pos(nullptr),
|
||||||
|
mark_offset(0) {}
|
||||||
Memchunks(const Memchunks &) = delete;
|
Memchunks(const Memchunks &) = delete;
|
||||||
Memchunks(Memchunks &&other) noexcept
|
Memchunks(Memchunks &&other) noexcept
|
||||||
: pool{other.pool}, // keep other.pool
|
: pool{other.pool}, // keep other.pool
|
||||||
head{std::exchange(other.head, nullptr)},
|
head{std::exchange(other.head, nullptr)},
|
||||||
tail{std::exchange(other.tail, nullptr)},
|
tail{std::exchange(other.tail, nullptr)},
|
||||||
len{std::exchange(other.len, 0)} {}
|
len{std::exchange(other.len, 0)},
|
||||||
|
mark{std::exchange(other.mark, nullptr)},
|
||||||
|
mark_pos{std::exchange(other.mark_pos, nullptr)},
|
||||||
|
mark_offset{std::exchange(other.mark_offset, 0)} {}
|
||||||
Memchunks &operator=(const Memchunks &) = delete;
|
Memchunks &operator=(const Memchunks &) = delete;
|
||||||
Memchunks &operator=(Memchunks &&other) noexcept {
|
Memchunks &operator=(Memchunks &&other) noexcept {
|
||||||
if (this == &other) {
|
if (this == &other) {
|
||||||
|
@ -132,6 +141,9 @@ template <typename Memchunk> struct Memchunks {
|
||||||
head = std::exchange(other.head, nullptr);
|
head = std::exchange(other.head, nullptr);
|
||||||
tail = std::exchange(other.tail, nullptr);
|
tail = std::exchange(other.tail, nullptr);
|
||||||
len = std::exchange(other.len, 0);
|
len = std::exchange(other.len, 0);
|
||||||
|
mark = std::exchange(other.mark, nullptr);
|
||||||
|
mark_pos = std::exchange(other.mark_pos, nullptr);
|
||||||
|
mark_offset = std::exchange(other.mark_offset, 0);
|
||||||
|
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
@ -200,6 +212,8 @@ template <typename Memchunk> struct Memchunks {
|
||||||
return len;
|
return len;
|
||||||
}
|
}
|
||||||
size_t remove(void *dest, size_t count) {
|
size_t remove(void *dest, size_t count) {
|
||||||
|
assert(mark == nullptr);
|
||||||
|
|
||||||
if (!tail || count == 0) {
|
if (!tail || count == 0) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
@ -231,6 +245,8 @@ template <typename Memchunk> struct Memchunks {
|
||||||
return first - static_cast<uint8_t *>(dest);
|
return first - static_cast<uint8_t *>(dest);
|
||||||
}
|
}
|
||||||
size_t remove(Memchunks &dest, size_t count) {
|
size_t remove(Memchunks &dest, size_t count) {
|
||||||
|
assert(mark == nullptr);
|
||||||
|
|
||||||
if (!tail || count == 0) {
|
if (!tail || count == 0) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
@ -262,6 +278,7 @@ template <typename Memchunk> struct Memchunks {
|
||||||
}
|
}
|
||||||
size_t remove(Memchunks &dest) {
|
size_t remove(Memchunks &dest) {
|
||||||
assert(pool == dest.pool);
|
assert(pool == dest.pool);
|
||||||
|
assert(mark == nullptr);
|
||||||
|
|
||||||
if (head == nullptr) {
|
if (head == nullptr) {
|
||||||
return 0;
|
return 0;
|
||||||
|
@ -284,6 +301,8 @@ template <typename Memchunk> struct Memchunks {
|
||||||
return n;
|
return n;
|
||||||
}
|
}
|
||||||
size_t drain(size_t count) {
|
size_t drain(size_t count) {
|
||||||
|
assert(mark == nullptr);
|
||||||
|
|
||||||
auto ndata = count;
|
auto ndata = count;
|
||||||
auto m = head;
|
auto m = head;
|
||||||
while (m) {
|
while (m) {
|
||||||
|
@ -305,6 +324,38 @@ template <typename Memchunk> struct Memchunks {
|
||||||
}
|
}
|
||||||
return ndata - count;
|
return ndata - count;
|
||||||
}
|
}
|
||||||
|
size_t drain_mark(size_t count) {
|
||||||
|
auto ndata = count;
|
||||||
|
auto m = head;
|
||||||
|
while (m) {
|
||||||
|
auto next = m->next;
|
||||||
|
auto n = std::min(count, m->len());
|
||||||
|
m->pos += n;
|
||||||
|
count -= n;
|
||||||
|
len -= n;
|
||||||
|
mark_offset -= n;
|
||||||
|
|
||||||
|
if (m->len() > 0) {
|
||||||
|
assert(mark != m || m->pos <= mark_pos);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (mark == m) {
|
||||||
|
assert(m->pos <= mark_pos);
|
||||||
|
|
||||||
|
mark = nullptr;
|
||||||
|
mark_pos = nullptr;
|
||||||
|
mark_offset = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
pool->recycle(m);
|
||||||
|
m = next;
|
||||||
|
}
|
||||||
|
head = m;
|
||||||
|
if (head == nullptr) {
|
||||||
|
tail = nullptr;
|
||||||
|
}
|
||||||
|
return ndata - count;
|
||||||
|
}
|
||||||
int riovec(struct iovec *iov, int iovcnt) const {
|
int riovec(struct iovec *iov, int iovcnt) const {
|
||||||
if (!head) {
|
if (!head) {
|
||||||
return 0;
|
return 0;
|
||||||
|
@ -317,7 +368,41 @@ template <typename Memchunk> struct Memchunks {
|
||||||
}
|
}
|
||||||
return i;
|
return i;
|
||||||
}
|
}
|
||||||
|
int riovec_mark(struct iovec *iov, int iovcnt) {
|
||||||
|
if (!head || iovcnt == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int i = 0;
|
||||||
|
Memchunk *m;
|
||||||
|
if (mark) {
|
||||||
|
if (mark_pos != mark->last) {
|
||||||
|
iov[0].iov_base = mark_pos;
|
||||||
|
iov[0].iov_len = mark->len() - (mark_pos - mark->pos);
|
||||||
|
|
||||||
|
mark_pos = mark->last;
|
||||||
|
mark_offset += iov[0].iov_len;
|
||||||
|
i = 1;
|
||||||
|
}
|
||||||
|
m = mark->next;
|
||||||
|
} else {
|
||||||
|
i = 0;
|
||||||
|
m = head;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (; i < iovcnt && m; ++i, m = m->next) {
|
||||||
|
iov[i].iov_base = m->pos;
|
||||||
|
iov[i].iov_len = m->len();
|
||||||
|
|
||||||
|
mark = m;
|
||||||
|
mark_pos = m->last;
|
||||||
|
mark_offset += m->len();
|
||||||
|
}
|
||||||
|
|
||||||
|
return i;
|
||||||
|
}
|
||||||
size_t rleft() const { return len; }
|
size_t rleft() const { return len; }
|
||||||
|
size_t rleft_mark() const { return len - mark_offset; }
|
||||||
void reset() {
|
void reset() {
|
||||||
for (auto m = head; m;) {
|
for (auto m = head; m;) {
|
||||||
auto next = m->next;
|
auto next = m->next;
|
||||||
|
@ -325,12 +410,17 @@ template <typename Memchunk> struct Memchunks {
|
||||||
m = next;
|
m = next;
|
||||||
}
|
}
|
||||||
len = 0;
|
len = 0;
|
||||||
head = tail = nullptr;
|
head = tail = mark = nullptr;
|
||||||
|
mark_pos = nullptr;
|
||||||
|
mark_offset = 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
Pool<Memchunk> *pool;
|
Pool<Memchunk> *pool;
|
||||||
Memchunk *head, *tail;
|
Memchunk *head, *tail;
|
||||||
size_t len;
|
size_t len;
|
||||||
|
Memchunk *mark;
|
||||||
|
uint8_t *mark_pos;
|
||||||
|
size_t mark_offset;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Wrapper around Memchunks to offer "peeking" functionality.
|
// Wrapper around Memchunks to offer "peeking" functionality.
|
||||||
|
|
|
@ -145,7 +145,8 @@ Downstream::Downstream(Upstream *upstream, MemchunkPool *mcpool,
|
||||||
accesslog_written_(false),
|
accesslog_written_(false),
|
||||||
new_affinity_cookie_(false),
|
new_affinity_cookie_(false),
|
||||||
blocked_request_data_eof_(false),
|
blocked_request_data_eof_(false),
|
||||||
expect_100_continue_(false) {
|
expect_100_continue_(false),
|
||||||
|
stop_reading_(false) {
|
||||||
|
|
||||||
auto &timeoutconf = get_config()->http2.timeout;
|
auto &timeoutconf = get_config()->http2.timeout;
|
||||||
|
|
||||||
|
@ -164,6 +165,7 @@ Downstream::Downstream(Upstream *upstream, MemchunkPool *mcpool,
|
||||||
downstream_wtimer_.data = this;
|
downstream_wtimer_.data = this;
|
||||||
|
|
||||||
rcbufs_.reserve(32);
|
rcbufs_.reserve(32);
|
||||||
|
rcbufs3_.reserve(32);
|
||||||
}
|
}
|
||||||
|
|
||||||
Downstream::~Downstream() {
|
Downstream::~Downstream() {
|
||||||
|
@ -203,6 +205,10 @@ Downstream::~Downstream() {
|
||||||
// explicitly.
|
// explicitly.
|
||||||
dconn_.reset();
|
dconn_.reset();
|
||||||
|
|
||||||
|
for (auto rcbuf : rcbufs3_) {
|
||||||
|
nghttp3_rcbuf_decref(rcbuf);
|
||||||
|
}
|
||||||
|
|
||||||
for (auto rcbuf : rcbufs_) {
|
for (auto rcbuf : rcbufs_) {
|
||||||
nghttp2_rcbuf_decref(rcbuf);
|
nghttp2_rcbuf_decref(rcbuf);
|
||||||
}
|
}
|
||||||
|
@ -1128,6 +1134,11 @@ void Downstream::add_rcbuf(nghttp2_rcbuf *rcbuf) {
|
||||||
rcbufs_.push_back(rcbuf);
|
rcbufs_.push_back(rcbuf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Downstream::add_rcbuf(nghttp3_rcbuf *rcbuf) {
|
||||||
|
nghttp3_rcbuf_incref(rcbuf);
|
||||||
|
rcbufs3_.push_back(rcbuf);
|
||||||
|
}
|
||||||
|
|
||||||
void Downstream::set_downstream_addr_group(
|
void Downstream::set_downstream_addr_group(
|
||||||
const std::shared_ptr<DownstreamAddrGroup> &group) {
|
const std::shared_ptr<DownstreamAddrGroup> &group) {
|
||||||
group_ = group;
|
group_ = group;
|
||||||
|
@ -1169,4 +1180,8 @@ bool Downstream::get_expect_100_continue() const {
|
||||||
return expect_100_continue_;
|
return expect_100_continue_;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool Downstream::get_stop_reading() const { return stop_reading_; }
|
||||||
|
|
||||||
|
void Downstream::set_stop_reading(bool f) { stop_reading_ = f; }
|
||||||
|
|
||||||
} // namespace shrpx
|
} // namespace shrpx
|
||||||
|
|
|
@ -38,6 +38,8 @@
|
||||||
|
|
||||||
#include <nghttp2/nghttp2.h>
|
#include <nghttp2/nghttp2.h>
|
||||||
|
|
||||||
|
#include <nghttp3/nghttp3.h>
|
||||||
|
|
||||||
#include "llhttp.h"
|
#include "llhttp.h"
|
||||||
|
|
||||||
#include "shrpx_io_control.h"
|
#include "shrpx_io_control.h"
|
||||||
|
@ -488,6 +490,7 @@ public:
|
||||||
BlockAllocator &get_block_allocator();
|
BlockAllocator &get_block_allocator();
|
||||||
|
|
||||||
void add_rcbuf(nghttp2_rcbuf *rcbuf);
|
void add_rcbuf(nghttp2_rcbuf *rcbuf);
|
||||||
|
void add_rcbuf(nghttp3_rcbuf *rcbuf);
|
||||||
|
|
||||||
void
|
void
|
||||||
set_downstream_addr_group(const std::shared_ptr<DownstreamAddrGroup> &group);
|
set_downstream_addr_group(const std::shared_ptr<DownstreamAddrGroup> &group);
|
||||||
|
@ -513,6 +516,9 @@ public:
|
||||||
|
|
||||||
bool get_expect_100_continue() const;
|
bool get_expect_100_continue() const;
|
||||||
|
|
||||||
|
bool get_stop_reading() const;
|
||||||
|
void set_stop_reading(bool f);
|
||||||
|
|
||||||
enum {
|
enum {
|
||||||
EVENT_ERROR = 0x1,
|
EVENT_ERROR = 0x1,
|
||||||
EVENT_TIMEOUT = 0x2,
|
EVENT_TIMEOUT = 0x2,
|
||||||
|
@ -527,6 +533,7 @@ private:
|
||||||
BlockAllocator balloc_;
|
BlockAllocator balloc_;
|
||||||
|
|
||||||
std::vector<nghttp2_rcbuf *> rcbufs_;
|
std::vector<nghttp2_rcbuf *> rcbufs_;
|
||||||
|
std::vector<nghttp3_rcbuf *> rcbufs3_;
|
||||||
|
|
||||||
Request req_;
|
Request req_;
|
||||||
Response resp_;
|
Response resp_;
|
||||||
|
@ -606,6 +613,7 @@ private:
|
||||||
bool blocked_request_data_eof_;
|
bool blocked_request_data_eof_;
|
||||||
// true if request contains "expect: 100-continue" header field.
|
// true if request contains "expect: 100-continue" header field.
|
||||||
bool expect_100_continue_;
|
bool expect_100_continue_;
|
||||||
|
bool stop_reading_;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace shrpx
|
} // namespace shrpx
|
||||||
|
|
File diff suppressed because it is too large
Load Diff
|
@ -31,6 +31,8 @@
|
||||||
#include <nghttp3/nghttp3.h>
|
#include <nghttp3/nghttp3.h>
|
||||||
|
|
||||||
#include "shrpx_upstream.h"
|
#include "shrpx_upstream.h"
|
||||||
|
#include "shrpx_downstream_queue.h"
|
||||||
|
#include "shrpx_mruby.h"
|
||||||
#include "quic.h"
|
#include "quic.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
|
|
||||||
|
@ -111,6 +113,31 @@ public:
|
||||||
void reset_timer();
|
void reset_timer();
|
||||||
|
|
||||||
int setup_httpconn();
|
int setup_httpconn();
|
||||||
|
void add_pending_downstream(std::unique_ptr<Downstream> downstream);
|
||||||
|
int recv_stream_data(uint32_t flags, int64_t stream_id, const uint8_t *data,
|
||||||
|
size_t datalen);
|
||||||
|
int acked_stream_data_offset(int64_t stream_id, uint64_t datalen);
|
||||||
|
int extend_max_stream_data(int64_t stream_id);
|
||||||
|
void extend_max_remote_streams_bidi(uint64_t max_streams);
|
||||||
|
int error_reply(Downstream *downstream, unsigned int status_code);
|
||||||
|
void http_begin_request_headers(int64_t stream_id);
|
||||||
|
int http_recv_request_header(Downstream *downstream, int32_t token,
|
||||||
|
nghttp3_rcbuf *name, nghttp3_rcbuf *value,
|
||||||
|
uint8_t flags);
|
||||||
|
int http_end_request_headers(Downstream *downstream);
|
||||||
|
int http_end_stream(Downstream *downstream);
|
||||||
|
void start_downstream(Downstream *downstream);
|
||||||
|
void initiate_downstream(Downstream *downstream);
|
||||||
|
int shutdown_stream(Downstream *downstream, uint64_t app_error_code);
|
||||||
|
int redirect_to_https(Downstream *downstream);
|
||||||
|
int http_stream_close(Downstream *downstream, uint64_t app_error_code);
|
||||||
|
void consume(int64_t stream_id, size_t nconsumed);
|
||||||
|
void remove_downstream(Downstream *downstream);
|
||||||
|
int stream_close(int64_t stream_id, uint64_t app_error_code);
|
||||||
|
void log_response_headers(Downstream *downstream,
|
||||||
|
const std::vector<nghttp3_nv> &nva) const;
|
||||||
|
int http_acked_stream_data(Downstream *downstream, size_t datalen);
|
||||||
|
int http_shutdown_stream_read(int64_t stream_id);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
ClientHandler *handler_;
|
ClientHandler *handler_;
|
||||||
|
@ -121,6 +148,7 @@ private:
|
||||||
quic::Error last_error_;
|
quic::Error last_error_;
|
||||||
uint8_t tls_alert_;
|
uint8_t tls_alert_;
|
||||||
nghttp3_conn *httpconn_;
|
nghttp3_conn *httpconn_;
|
||||||
|
DownstreamQueue downstream_queue_;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace shrpx
|
} // namespace shrpx
|
||||||
|
|
Loading…
Reference in New Issue