arthurscchan commented on code in PR #13826:
URL: https://github.com/apache/trafficserver/pull/13826#discussion_r4280550398
##########
tests/fuzzing/fuzz_header_conversion.cc:
##########
@@ -0,0 +1,152 @@
+#include "proxy/hdrs/HTTP.h"
+#include "proxy/hdrs/HeaderValidator.h"
+#include "proxy/hdrs/VersionConverter.h"
+#include "tscore/Diags.h"
+#include "tsutil/PostScript.h"
+
+#include <fuzzer/FuzzedDataProvider.h>
+
+#include <array>
+#include <cstddef>
+#include <cstdint>
+#include <string>
+#include <string_view>
+#include <vector>
+
+extern int cmd_disable_pfreelist;
+
+namespace
+{
+constexpr size_t MAX_INPUT_LENGTH = 64 * 1024;
+constexpr std::array<std::string_view, 17> FIELD_NAMES = {
+ ":method", ":scheme", ":authority", ":path",
":status", "host",
+ "cookie", "connection", "keep-alive", "proxy-connection",
"upgrade", "transfer-encoding",
+ "te", "content-length", "content-type", "x-test", "",
+};
+
+void
+add_field(HTTPHdr &header, std::string_view name, std::string_view value)
+{
+ MIMEField *field = header.field_create(name);
+
+ field->value_set(header.m_heap, header.m_mime, value);
+ header.field_attach(field);
+}
+
+void
+print_header(const HTTPHdr &header)
+{
+ std::array<char, 32> small_buffer{};
+ int buffer_index = 0;
+ int dump_offset = 0;
+ int length = header.length_get();
+
+ header.print(small_buffer.data(), small_buffer.size(), &buffer_index,
&dump_offset);
+ if (length >= 0 && length <= 2 * static_cast<int>(MAX_INPUT_LENGTH)) {
+ std::vector<char> buffer(static_cast<size_t>(length) + 1);
+ buffer_index = 0;
+ dump_offset = 0;
+ header.print(buffer.data(), static_cast<int>(buffer.size()),
&buffer_index, &dump_offset);
+ }
+}
+
+void
+convert_decoded_header(const HTTPHdr &header, int version, bool is_response,
bool is_trailer)
+{
+ if (!HeaderValidator::is_h2_h3_header_valid(header, is_response, is_trailer)
|| is_trailer) {
+ return;
+ }
+
+ HTTPHdr converted;
+ ts::PostScript cleanup([&]() { converted.destroy(); });
+ VersionConverter converter;
+
+ converted.create(is_response ? HTTPType::RESPONSE : HTTPType::REQUEST);
+ converted.copy(&header);
+ if (converter.convert(converted, version, 1) == 0) {
+ print_header(converted);
+ }
+}
+} // namespace
+
+extern "C" int
+LLVMFuzzerInitialize(int *, char ***)
+{
+ cmd_disable_pfreelist = true;
+ DiagsPtr::set(new Diags("fuzzing", "", "", nullptr));
+ http_init();
+ return 0;
+}
+
+extern "C" int
+LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
+{
+ if (size == 0 || size > MAX_INPUT_LENGTH) {
+ return 0;
+ }
+
+ uint8_t flags = data[0];
+ bool is_response = (flags & 1) != 0;
+ bool is_text = (flags & 2) != 0;
+ int version = (flags & 4) != 0 ? 3 : 2;
+ HTTPType type = is_response ? HTTPType::RESPONSE :
HTTPType::REQUEST;
+ HTTPHdr header;
+ ts::PostScript cleanup([&]() { header.destroy(); });
+ VersionConverter converter;
+
+ if (is_text) {
+ std::string input(reinterpret_cast<const char *>(data + 1), size - 1);
+ HTTPParser parser;
+ ts::PostScript parser_cleanup([&]() { http_parser_clear(&parser); });
+ const char *start = input.data();
+ const char *end = start + input.size();
+
+ header.create(type, version == 2 ? HTTP_2_0 : HTTP_3_0);
Review Comment:
It is necessary.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
