Skip to content

Commit

Permalink
[1.12] [http1] Preserve LWS from the middle of HTTP1 header values th…
Browse files Browse the repository at this point in the history
…at requ… (#12319)

* [http1] Preserve LWS from the middle of HTTP1 header values that require multiple dispatch calls to process (#10886)

Correctly preserve linear whitespace in the middle of HTTP1 header values. The fix in 6a95a21 trimmed away both leading and trailing whitespace when accepting header value fragments which can result in inner LWS in header values being stripped away if the LWS lands at the beginning or end of a buffer slice.

Also various fix to allow build on clang-10

Signed-off-by: Antonio Vicente <avd@google.com>
Signed-off-by: Yuchen Dai <silentdai@gmail.com>
  • Loading branch information
lambdai authored Jul 31, 2020
1 parent e9207f4 commit a624dee
Show file tree
Hide file tree
Showing 15 changed files with 171 additions and 27 deletions.
4 changes: 4 additions & 0 deletions bazel/external/quiche.BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,10 @@ quiche_copts = select({
"-Wno-unused-parameter",
"-Wno-unused-function",
"-Wno-type-limits",
"-Wno-unused-function",
"-Wno-unknown-warning-option",
"-Wno-deprecated-copy",
"-Wno-range-loop-construct",
# quic_inlined_frame.h uses offsetof() to optimize memory usage in frames.
"-Wno-invalid-offsetof",
"-Wno-type-limits",
Expand Down
6 changes: 6 additions & 0 deletions include/envoy/http/header_map.h
Original file line number Diff line number Diff line change
Expand Up @@ -129,6 +129,12 @@ class HeaderString {
*/
char* buffer() { return buffer_.dynamic_; }

/**
* Trim trailing whitespaces from the HeaderString.
* v1.12 supports both Inline and Dynamic, but not Reference type.
*/
void rtrim();

/**
* Get an absl::string_view. It will NOT be NUL terminated!
*
Expand Down
6 changes: 3 additions & 3 deletions source/common/http/async_client_impl.h
Original file line number Diff line number Diff line change
Expand Up @@ -144,9 +144,9 @@ class AsyncStreamImpl : public AsyncClient::Stream,
}
absl::optional<std::chrono::milliseconds> maxInterval() const override { return absl::nullopt; }

const std::vector<uint32_t> retriable_status_codes_;
const std::vector<Http::HeaderMatcherSharedPtr> retriable_headers_;
const std::vector<Http::HeaderMatcherSharedPtr> retriable_request_headers_;
const std::vector<uint32_t> retriable_status_codes_{};
const std::vector<Http::HeaderMatcherSharedPtr> retriable_headers_{};
const std::vector<Http::HeaderMatcherSharedPtr> retriable_request_headers_{};
};

struct NullShadowPolicy : public Router::ShadowPolicy {
Expand Down
9 changes: 9 additions & 0 deletions source/common/http/header_map_impl.cc
Original file line number Diff line number Diff line change
Expand Up @@ -150,6 +150,15 @@ void HeaderString::append(const char* data, uint32_t size) {
string_length_ += size;
}

void HeaderString::rtrim() {
ASSERT(type() == Type::Inline || type() == Type::Dynamic);
absl::string_view original = getStringView();
absl::string_view rtrimmed = StringUtil::rtrim(original);
if (original.size() != rtrimmed.size()) {
string_length_ = rtrimmed.size();
}
}

void HeaderString::clear() {
switch (type_) {
case Type::Reference: {
Expand Down
16 changes: 13 additions & 3 deletions source/common/http/http1/codec_impl.cc
Original file line number Diff line number Diff line change
Expand Up @@ -408,6 +408,11 @@ void ConnectionImpl::completeLastHeader() {
checkHeaderNameForUnderscores();
if (!current_header_field_.empty()) {
toLowerTable().toLowerCase(current_header_field_.buffer(), current_header_field_.size());
// Strip trailing whitespace of the current header value if any. Leading whitespace was trimmed
// in ConnectionImpl::onHeaderValue. http_parser does not strip leading or trailing whitespace
// as the spec requires: https://tools.ietf.org/html/rfc7230#section-3.2.4
current_header_value_.rtrim();

current_header_map_->addViaMove(std::move(current_header_field_),
std::move(current_header_value_));
}
Expand Down Expand Up @@ -518,9 +523,7 @@ void ConnectionImpl::onHeaderValue(const char* data, size_t length) {
return;
}

// Work around a bug in http_parser where trailing whitespace is not trimmed
// as the spec requires: https://tools.ietf.org/html/rfc7230#section-3.2.4
const absl::string_view header_value = StringUtil::trim(absl::string_view(data, length));
absl::string_view header_value{data, length};

if (strict_header_validation_) {
if (!Http::HeaderUtility::headerValueIsValid(header_value)) {
Expand All @@ -538,6 +541,13 @@ void ConnectionImpl::onHeaderValue(const char* data, size_t length) {
}

header_parsing_state_ = HeaderParsingState::Value;
if (current_header_value_.empty()) {
// Strip leading whitespace if the current header value input contains the first bytes of the
// encoded header value. Trailing whitespace is stripped once the full header value is known in
// ConnectionImpl::completeLastHeader. http_parser does not strip leading or trailing whitespace
// as the spec requires: https://tools.ietf.org/html/rfc7230#section-3.2.4 .
header_value = StringUtil::ltrim(header_value);
}
current_header_value_.append(header_value.data(), header_value.length());

checkMaxHeadersSize();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ typename std::enable_if<std::is_signed<T>::value, T>::type leftShift(T left, uin
inline void addByte(Buffer::Instance& buffer, const uint8_t value) { buffer.add(&value, 1); }

void addSeq(Buffer::Instance& buffer, const std::initializer_list<uint8_t>& values) {
for (const int8_t& value : values) {
for (const int8_t value : values) {
buffer.add(&value, 1);
}
}
Expand Down
2 changes: 1 addition & 1 deletion source/extensions/stat_sinks/hystrix/hystrix.cc
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ void HystrixSink::addHistogramToStream(const QuantileLatencyMap& latency_map, ab
// TODO: Consider if we better use join here
ss << ", \"" << key << "\": {";
bool is_first = true;
for (const std::pair<double, double>& element : latency_map) {
for (const std::pair<const double, double>& element : latency_map) {
const std::string quantile = fmt::sprintf("%g", element.first * 100);
HystrixSink::addDoubleToStream(quantile, element.second, ss, is_first);
is_first = false;
Expand Down
3 changes: 1 addition & 2 deletions test/common/http/async_client_impl_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,6 @@ TEST_F(AsyncClientImplTest, Basic) {
response_decoder_ = &decoder;
return nullptr;
}));

TestHeaderMapImpl copy(message_->headers());
copy.addCopy("x-envoy-internal", "true");
copy.addCopy("x-forwarded-for", "127.0.0.1");
Expand All @@ -149,7 +148,7 @@ TEST_F(AsyncClientImplTest, Basic) {
EXPECT_CALL(stream_encoder_, encodeData(BufferEqual(&data), true));
expectSuccess(200);

client_.send(std::move(message_), callbacks_, AsyncClient::RequestOptions());
client_.send(std::move(message_), callbacks_, AsyncClient::RequestOptions().setSendXff(true));

HeaderMapPtr response_headers(new TestHeaderMapImpl{{":status", "200"}});
response_decoder_->decodeHeaders(std::move(response_headers), false);
Expand Down
31 changes: 31 additions & 0 deletions test/common/http/header_map_impl_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -105,6 +105,37 @@ TEST(HeaderStringTest, All) {
EXPECT_EQ("HELLO", string.getStringView());
}

// Inline rtrim removes trailing whitespace only.
{
// This header string is short enough to fit into Inline type.
const std::string data_with_leading_lws = " \t\f\v data";
const std::string data_with_leading_and_trailing_lws = data_with_leading_lws + " \t\f\v";
HeaderString string;
string.append(data_with_leading_and_trailing_lws.data(),
data_with_leading_and_trailing_lws.size());
EXPECT_EQ(data_with_leading_and_trailing_lws, string.getStringView());
EXPECT_EQ(string.type(), HeaderString::Type::Inline);
string.rtrim();
EXPECT_NE(data_with_leading_and_trailing_lws, string.getStringView());
EXPECT_EQ(data_with_leading_lws, string.getStringView());
}

// Dynamic rtrim removes trailing whitespace only.
{
// Making this string longer than Inline can fit.
const std::string padding_data_with_leading_lws = " \t\f\v data" + std::string(128, 'a');
const std::string data_with_leading_and_trailing_lws =
padding_data_with_leading_lws + " \t\f\v";
HeaderString string;
string.append(data_with_leading_and_trailing_lws.data(),
data_with_leading_and_trailing_lws.size());
EXPECT_EQ(data_with_leading_and_trailing_lws, string.getStringView());
EXPECT_EQ(string.type(), HeaderString::Type::Dynamic);
string.rtrim();
EXPECT_NE(data_with_leading_and_trailing_lws, string.getStringView());
EXPECT_EQ(padding_data_with_leading_lws, string.getStringView());
}

// Static clear() does nothing.
{
std::string static_string("HELLO");
Expand Down
57 changes: 54 additions & 3 deletions test/common/http/http1/codec_impl_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,16 @@ std::string createHeaderFragment(int num_headers) {
}
return headers;
}

Buffer::OwnedImpl createBufferWithNByteSlices(absl::string_view input, size_t max_slice_size) {
Buffer::OwnedImpl buffer;
for (size_t offset = 0; offset < input.size(); offset += max_slice_size) {
buffer.appendSliceForTest(input.substr(offset, max_slice_size));
}
// Verify that the buffer contains the right number of slices.
ASSERT(buffer.getRawSlices(nullptr, 0) == (input.size() + max_slice_size - 1) / max_slice_size);
return buffer;
}
} // namespace

class Http1ServerConnectionImplTest : public testing::Test {
Expand All @@ -66,16 +76,21 @@ class Http1ServerConnectionImplTest : public testing::Test {
// Then send a response just to clean up.
void sendAndValidateRequestAndSendResponse(absl::string_view raw_request,
const TestHeaderMapImpl& expected_request_headers) {
NiceMock<Http::MockStreamDecoder> decoder;
Buffer::OwnedImpl buffer(raw_request);
sendAndValidateRequestAndSendResponse(buffer, expected_request_headers);
}

void sendAndValidateRequestAndSendResponse(Buffer::Instance& buffer,
const TestHeaderMapImpl& expected_request_headers) {
NiceMock<MockStreamDecoder> decoder;
Http::StreamEncoder* response_encoder = nullptr;
EXPECT_CALL(callbacks_, newStream(_, _))
.Times(1)
.WillOnce(Invoke([&](Http::StreamEncoder& encoder, bool) -> Http::StreamDecoder& {
response_encoder = &encoder;
return decoder;
}));
EXPECT_CALL(decoder, decodeHeaders_(HeaderMapEqual(&expected_request_headers), true)).Times(1);
Buffer::OwnedImpl buffer(raw_request);
EXPECT_CALL(decoder, decodeHeaders_(HeaderMapEqual(&expected_request_headers), true));
codec_->dispatch(buffer);
EXPECT_EQ(0U, buffer.length());
response_encoder->encodeHeaders(TestHeaderMapImpl{{":status", "200"}}, true);
Expand Down Expand Up @@ -206,6 +221,42 @@ TEST_F(Http1ServerConnectionImplTest, HostWithLWS) {
"GET / HTTP/1.1\r\nHost: host \r\n\r\n", expected_headers);
}

// Regression test for https://github.com/envoyproxy/envoy/issues/10270. Linear whitespace at the
// beginning and end of a header value should be stripped. Whitespace in the middle should be
// preserved.
TEST_F(Http1ServerConnectionImplTest, InnerLWSIsPreserved) {
initialize();

// Header with many spaces surrounded by non-whitespace characters to ensure that dispatching is
// split across multiple dispatch calls. The threshold used here comes from Envoy preferring 16KB
// reads, but the important part is that the header value is split such that the pieces have
// leading and trailing whitespace characters.
const std::string header_value_with_inner_lws = "v" + std::string(32 * 1024, ' ') + "v";
TestHeaderMapImpl expected_headers{{":authority", "host"},
{":path", "/"},
{":method", "GET"},
{"header_field", header_value_with_inner_lws}};

{
// Regression test spaces in the middle are preserved
Buffer::OwnedImpl header_buffer = createBufferWithNByteSlices(
"GET / HTTP/1.1\r\nHost: host\r\nheader_field: " + header_value_with_inner_lws + "\r\n\r\n",
16 * 1024);
EXPECT_EQ(3, header_buffer.getRawSlices(nullptr, 0));
sendAndValidateRequestAndSendResponse(header_buffer, expected_headers);
}

{
// Regression test spaces before and after are removed
Buffer::OwnedImpl header_buffer = createBufferWithNByteSlices(
"GET / HTTP/1.1\r\nHost: host\r\nheader_field: " + header_value_with_inner_lws +
" \r\n\r\n",
16 * 1024);
EXPECT_EQ(3, header_buffer.getRawSlices(nullptr, 0));
sendAndValidateRequestAndSendResponse(header_buffer, expected_headers);
}
}

TEST_F(Http1ServerConnectionImplTest, Http10) {
initialize();

Expand Down
7 changes: 3 additions & 4 deletions test/extensions/filters/http/buffer/config_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -77,10 +77,9 @@ TEST(BufferFilterFactoryTest, BufferFilterEmptyProto) {
TEST(BufferFilterFactoryTest, BufferFilterEmptyRouteProto) {
BufferFilterFactory factory;
EXPECT_NO_THROW({
envoy::config::filter::http::buffer::v2::BufferPerRoute* config =
dynamic_cast<envoy::config::filter::http::buffer::v2::BufferPerRoute*>(
factory.createEmptyRouteConfigProto().get());
EXPECT_NE(nullptr, config);
auto config = factory.createEmptyRouteConfigProto();
EXPECT_NE(nullptr,
dynamic_cast<envoy::config::filter::http::buffer::v2::BufferPerRoute*>(config.get()));
});
}

Expand Down
2 changes: 1 addition & 1 deletion test/extensions/filters/http/dynamo/config_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ namespace {
TEST(DynamoFilterConfigTest, DynamoFilter) {
NiceMock<Server::Configuration::MockFactoryContext> context;
DynamoFilterConfig factory;
const auto proto_config = factory.createEmptyConfigProto().get();
const auto proto_config = factory.createEmptyConfigProto();
Http::FilterFactoryCb cb = factory.createFilterFactoryFromProto(*proto_config, "stats", context);
Http::MockFilterChainFactoryCallbacks filter_callback;
EXPECT_CALL(filter_callback, addStreamFilter(_));
Expand Down
15 changes: 9 additions & 6 deletions test/extensions/filters/http/rbac/config_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -32,16 +32,19 @@ TEST(RoleBasedAccessControlFilterConfigFactoryTest, ValidProto) {

TEST(RoleBasedAccessControlFilterConfigFactoryTest, EmptyProto) {
RoleBasedAccessControlFilterConfigFactory factory;
auto* config = dynamic_cast<envoy::config::filter::http::rbac::v2::RBAC*>(
factory.createEmptyConfigProto().get());
EXPECT_NE(nullptr, config);
auto config =

factory.createEmptyConfigProto();

EXPECT_NE(nullptr, dynamic_cast<envoy::config::filter::http::rbac::v2::RBAC*>(config.get()));
}

TEST(RoleBasedAccessControlFilterConfigFactoryTest, EmptyRouteProto) {
RoleBasedAccessControlFilterConfigFactory factory;
auto* config = dynamic_cast<envoy::config::filter::http::rbac::v2::RBACPerRoute*>(
factory.createEmptyRouteConfigProto().get());
EXPECT_NE(nullptr, config);
auto config = factory.createEmptyRouteConfigProto();

EXPECT_NE(nullptr,
dynamic_cast<envoy::config::filter::http::rbac::v2::RBACPerRoute*>(config.get()));
}

TEST(RoleBasedAccessControlFilterConfigFactoryTest, RouteSpecificConfig) {
Expand Down
5 changes: 2 additions & 3 deletions test/extensions/filters/network/rbac/config_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -66,9 +66,8 @@ TEST_F(RoleBasedAccessControlNetworkFilterConfigFactoryTest, ValidProto) {

TEST_F(RoleBasedAccessControlNetworkFilterConfigFactoryTest, EmptyProto) {
RoleBasedAccessControlNetworkFilterConfigFactory factory;
auto* config = dynamic_cast<envoy::config::filter::network::rbac::v2::RBAC*>(
factory.createEmptyConfigProto().get());
EXPECT_NE(nullptr, config);
auto config = factory.createEmptyConfigProto();
EXPECT_NE(nullptr, dynamic_cast<envoy::config::filter::network::rbac::v2::RBAC*>(config.get()));
}

TEST_F(RoleBasedAccessControlNetworkFilterConfigFactoryTest, InvalidPermission) {
Expand Down
33 changes: 33 additions & 0 deletions test/integration/protocol_integration_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -230,6 +230,39 @@ TEST_P(ProtocolIntegrationTest, DrainClose) {
test_server_->drainManager().draining_ = false;
}

// Regression test for https://github.com/envoyproxy/envoy/issues/10270
TEST_P(ProtocolIntegrationTest, LongHeaderValueWithSpaces) {
// Header with at least 20kb of spaces surrounded by non-whitespace characters to ensure that
// dispatching is split across 2 dispatch calls. This threshold comes from Envoy preferring 16KB
// reads, which the buffer rounds up to about 20KB when allocating slices in
// Buffer::OwnedImpl::reserve().
const std::string long_header_value_with_inner_lws = "v" + std::string(32 * 1024, ' ') + "v";

initialize();
codec_client_ = makeHttpConnection(lookupPort("http"));
auto response = codec_client_->makeHeaderOnlyRequest(
Http::TestHeaderMapImpl{{":method", "GET"},
{":path", "/test/long/url"},
{":scheme", "http"},
{":authority", "host"},
{"longrequestvalue", long_header_value_with_inner_lws}});
waitForNextUpstreamRequest();
EXPECT_EQ(long_header_value_with_inner_lws, upstream_request_->headers()
.get(Http::LowerCaseString("longrequestvalue"))
->value()
.getStringView());
upstream_request_->encodeHeaders(
Http::TestHeaderMapImpl{{":status", "200"},
{"longresponsevalue", long_header_value_with_inner_lws}},
true);
response->waitForEndStream();
EXPECT_TRUE(response->complete());
EXPECT_EQ("200", response->headers().Status()->value().getStringView());
EXPECT_EQ(
long_header_value_with_inner_lws,
response->headers().get(Http::LowerCaseString("longresponsevalue"))->value().getStringView());
}

TEST_P(ProtocolIntegrationTest, Retry) {
initialize();
codec_client_ = makeHttpConnection(lookupPort("http"));
Expand Down

0 comments on commit a624dee

Please sign in to comment.