diff options
Diffstat (limited to '')
-rw-r--r-- | test/nxt_http_parse_unit_test.c | 457 |
1 files changed, 457 insertions, 0 deletions
diff --git a/test/nxt_http_parse_unit_test.c b/test/nxt_http_parse_unit_test.c new file mode 100644 index 00000000..abe7093d --- /dev/null +++ b/test/nxt_http_parse_unit_test.c @@ -0,0 +1,457 @@ + +/* + * Copyright (C) NGINX, Inc. + * Copyright (C) Valentin V. Bartenev + */ + +#include <nxt_main.h> + + +typedef struct { + nxt_str_t method; + nxt_str_t target; + nxt_str_t exten; + nxt_str_t args; + u_char version[8]; + + /* target with "/." */ + unsigned complex_target:1; + /* target with "%" */ + unsigned quoted_target:1; + /* target with " " */ + unsigned space_in_target:1; + /* target with "+" */ + unsigned plus_in_target:1; +} nxt_http_parse_unit_test_request_line_t; + + +typedef union { + void *pointer; + nxt_http_parse_unit_test_request_line_t request_line; +} nxt_http_parse_unit_test_data_t; + + +typedef struct { + nxt_str_t request; + nxt_int_t result; + nxt_int_t (*handler)(nxt_http_request_parse_t *rp, + nxt_http_parse_unit_test_data_t *data, + nxt_str_t *request, nxt_log_t *log); + + nxt_http_parse_unit_test_data_t data; +} nxt_http_parse_unit_test_case_t; + + +static nxt_int_t nxt_http_parse_unit_test_run(nxt_http_request_parse_t *rp, + nxt_str_t *request); + +static nxt_int_t nxt_http_parse_unit_test_request_line( + nxt_http_request_parse_t *rp, nxt_http_parse_unit_test_data_t *data, + nxt_str_t *request, nxt_log_t *log); + +static nxt_int_t nxt_http_unit_test_header_return(void *ctx, nxt_str_t *name, + nxt_str_t *value, uintptr_t data); + + +static nxt_http_parse_unit_test_case_t nxt_http_unit_test_cases[] = { + { + nxt_string("GET / HTTP/1.0\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("/"), + nxt_null_string, + nxt_null_string, + "HTTP/1.0", + 0, 0, 0, 0 + }} + }, + { + nxt_string("XXX-METHOD /d.ir/fi+le.ext?key=val HTTP/1.2\n\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("XXX-METHOD"), + nxt_string("/d.ir/fi+le.ext?key=val"), + nxt_string("ext?key=val"), + nxt_string("key=val"), + "HTTP/1.2", + 0, 0, 0, 1 + }} + }, + { + nxt_string("GET /di.r/? HTTP/1.0\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("/di.r/?"), + nxt_null_string, + nxt_string(""), + "HTTP/1.0", + 0, 0, 0, 0 + }} + }, + { + nxt_string("GEt / HTTP/1.0\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET /\0 HTTP/1.0\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET /\r HTTP/1.0\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET /\n HTTP/1.0\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET / HTTP/1.0\r\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET /. HTTP/1.0\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("/."), + nxt_null_string, + nxt_null_string, + "HTTP/1.0", + 1, 0, 0, 0 + }} + }, + { + nxt_string("GET /# HTTP/1.0\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("/#"), + nxt_null_string, + nxt_null_string, + "HTTP/1.0", + 1, 0, 0, 0 + }} + }, + { + nxt_string("GET /?# HTTP/1.0\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("/?#"), + nxt_null_string, + nxt_string("#"), + "HTTP/1.0", + 1, 0, 0, 0 + }} + }, + { + nxt_string("GET // HTTP/1.0\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("//"), + nxt_null_string, + nxt_null_string, + "HTTP/1.0", + 1, 0, 0, 0 + }} + }, + { + nxt_string("GET /%20 HTTP/1.0\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("/%20"), + nxt_null_string, + nxt_null_string, + "HTTP/1.0", + 0, 1, 0, 0 + }} + }, + { + nxt_string("GET / a HTTP/1.0\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("/ a"), + nxt_null_string, + nxt_null_string, + "HTTP/1.0", + 0, 0, 1, 0 + }} + }, + { + nxt_string("GET / HTTP/1.0 HTTP/1.1\r\n\r\n"), + NXT_DONE, + &nxt_http_parse_unit_test_request_line, + { .request_line = { + nxt_string("GET"), + nxt_string("/ HTTP/1.0"), + nxt_null_string, + nxt_null_string, + "HTTP/1.1", + 0, 0, 1, 0 + }} + }, + { + nxt_string("GET / HTTP/1.1\r\n" + "Host: example.com\r\n\r\n"), + NXT_DONE, + NULL, { NULL } + }, + { + nxt_string("GET / HTTP/1.1\r\n" + ":Host: example.com\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET / HTTP/1.1\r\n" + "Ho_st: example.com\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET / HTTP/1.1\r\n" + "Ho\0st: example.com\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET / HTTP/1.1\r\n" + "Ho\rst: example.com\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET / HTTP/1.1\r\n" + "Host: exa\0mple.com\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET / HTTP/1.1\r\n" + "Host: exa\rmple.com\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, + { + nxt_string("GET / HTTP/1.1\r\n" + "X-Bad-Header: value\r\n\r\n"), + NXT_ERROR, + NULL, { NULL } + }, +}; + + +static nxt_http_fields_t nxt_http_unit_test_headers[] = { + { nxt_string("X-Bad-Header"), + &nxt_http_unit_test_header_return, + (uintptr_t) NXT_ERROR }, + + { nxt_null_string, NULL, 0 } +}; + + +nxt_int_t +nxt_http_parse_unit_test(nxt_thread_t *thr) +{ + nxt_int_t rc; + nxt_uint_t i; + nxt_mem_pool_t *pool; + nxt_http_fields_hash_t *hash; + nxt_http_request_parse_t rp; + nxt_http_parse_unit_test_case_t *test; + + nxt_thread_time_update(thr); + + pool = nxt_mem_pool_create(512); + if (pool == NULL) { + return NXT_ERROR; + } + + hash = nxt_http_fields_hash(nxt_http_unit_test_headers, pool); + + if (hash == NULL) { + return NXT_ERROR; + } + + for (i = 0; i < nxt_nitems(nxt_http_unit_test_cases); i++) { + test = &nxt_http_unit_test_cases[i]; + + nxt_memzero(&rp, sizeof(nxt_http_request_parse_t)); + + rp.hash = hash; + + rc = nxt_http_parse_unit_test_run(&rp, &test->request); + + if (rc != test->result) { + nxt_log_alert(thr->log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - result: %i (expected: %i)", + &test->request, rc, test->result); + return NXT_ERROR; + } + + if (test->handler != NULL + && test->handler(&rp, &test->data, &test->request, thr->log) + != NXT_OK) + { + return NXT_ERROR; + } + } + + nxt_mem_pool_destroy(pool); + + nxt_log_error(NXT_LOG_NOTICE, thr->log, "http parse unit test passed"); + + return NXT_OK; +} + + +static nxt_int_t +nxt_http_parse_unit_test_run(nxt_http_request_parse_t *rp, nxt_str_t *request) +{ + nxt_int_t rc; + nxt_buf_mem_t buf; + + buf.start = request->start; + buf.end = request->start + request->length; + + buf.pos = buf.start; + buf.free = buf.pos + 1; + + do { + buf.free++; + rc = nxt_http_parse_request(rp, &buf); + } while (buf.free < buf.end && rc == NXT_AGAIN); + + return rc; +} + + +static nxt_int_t +nxt_http_parse_unit_test_request_line(nxt_http_request_parse_t *rp, + nxt_http_parse_unit_test_data_t *data, nxt_str_t *request, nxt_log_t *log) +{ + nxt_str_t str; + + nxt_http_parse_unit_test_request_line_t *test = &data->request_line; + + if (rp->method.start != test->method.start + && !nxt_strstr_eq(&rp->method, &test->method)) + { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - method: \"%V\" (expected: \"%V\")", + request, &rp->method, &test->method); + return NXT_ERROR; + } + + str.length = rp->target_end - rp->target_start; + str.start = rp->target_start; + + if (str.start != test->target.start + && !nxt_strstr_eq(&str, &test->target)) + { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - target: \"%V\" (expected: \"%V\")", + request, &str, &test->target); + return NXT_ERROR; + } + + str.length = (rp->exten_start != NULL) ? rp->target_end - rp->exten_start + : 0; + str.start = rp->exten_start; + + if (str.start != test->exten.start + && !nxt_strstr_eq(&str, &test->exten)) + { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - exten: \"%V\" (expected: \"%V\")", + request, &str, &test->exten); + return NXT_ERROR; + } + + str.length = (rp->args_start != NULL) ? rp->target_end - rp->args_start + : 0; + str.start = rp->args_start; + + if (str.start != test->args.start + && !nxt_strstr_eq(&str, &test->args)) + { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - args: \"%V\" (expected: \"%V\")", + request, &str, &test->args); + return NXT_ERROR; + } + + if (nxt_memcmp(rp->version.str, test->version, 8) != 0) { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - version: \"%*s\" (expected: \"%*s\")", + request, 8, rp->version.str, 8, test->version); + return NXT_ERROR; + } + + if (rp->complex_target != test->complex_target) { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - complex_target: %d (expected: %d)", + request, rp->complex_target, test->complex_target); + return NXT_ERROR; + } + + if (rp->quoted_target != test->quoted_target) { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - quoted_target: %d (expected: %d)", + request, rp->quoted_target, test->quoted_target); + return NXT_ERROR; + } + + if (rp->space_in_target != test->space_in_target) { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - space_in_target: %d (expected: %d)", + request, rp->space_in_target, test->space_in_target); + return NXT_ERROR; + } + + if (rp->plus_in_target != test->plus_in_target) { + nxt_log_alert(log, "http parse unit test case failed:\n" + " - request:\n\"%V\"\n" + " - plus_in_target: %d (expected: %d)", + request, rp->plus_in_target, test->plus_in_target); + return NXT_ERROR; + } + + return NXT_OK; +} + + +static nxt_int_t +nxt_http_unit_test_header_return(void *ctx, nxt_str_t *name, nxt_str_t *value, + uintptr_t data) +{ + return (nxt_int_t) data; +} |