mirror of
https://github.com/cesanta/mongoose.git
synced 2024-11-28 05:39:00 +08:00
130 lines
4.0 KiB
C
130 lines
4.0 KiB
C
#include "mongoose.c"
|
|
|
|
#define FATAL(str, line) do { \
|
|
printf("Fail on line %d: [%s]\n", line, str); \
|
|
abort(); \
|
|
} while (0)
|
|
#define ASSERT(expr) do { if (!(expr)) FATAL(#expr, __LINE__); } while (0)
|
|
|
|
static void test_parse_http_request() {
|
|
struct mg_request_info ri;
|
|
char req1[] = "GET / HTTP/1.1\r\n\r\n";
|
|
char req2[] = "BLAH / HTTP/1.1\r\n\r\n";
|
|
char req3[] = "GET / HTTP/1.1\r\nBah\r\n";
|
|
|
|
ASSERT(parse_http_request(req1, &ri) == 1);
|
|
ASSERT(strcmp(ri.http_version, "1.1") == 0);
|
|
ASSERT(ri.num_headers == 0);
|
|
|
|
ASSERT(parse_http_request(req2, &ri) == 0);
|
|
|
|
// TODO(lsm): Fix this. Bah is not a valid header.
|
|
ASSERT(parse_http_request(req3, &ri) == 1);
|
|
ASSERT(ri.num_headers == 1);
|
|
ASSERT(strcmp(ri.http_headers[0].name, "Bah\r\n") == 0);
|
|
|
|
// TODO(lsm): add more tests.
|
|
}
|
|
|
|
static void test_should_keep_alive(void) {
|
|
struct mg_connection conn;
|
|
struct mg_context ctx;
|
|
char req1[] = "GET / HTTP/1.1\r\n\r\n";
|
|
char req2[] = "GET / HTTP/1.0\r\n\r\n";
|
|
char req3[] = "GET / HTTP/1.1\r\nConnection: close\r\n\r\n";
|
|
char req4[] = "GET / HTTP/1.1\r\nConnection: keep-alive\r\n\r\n";
|
|
|
|
memset(&conn, 0, sizeof(conn));
|
|
conn.ctx = &ctx;
|
|
parse_http_request(req1, &conn.request_info);
|
|
|
|
ctx.config[ENABLE_KEEP_ALIVE] = "no";
|
|
ASSERT(should_keep_alive(&conn) == 0);
|
|
|
|
ctx.config[ENABLE_KEEP_ALIVE] = "yes";
|
|
ASSERT(should_keep_alive(&conn) == 1);
|
|
|
|
conn.must_close = 1;
|
|
ASSERT(should_keep_alive(&conn) == 0);
|
|
|
|
conn.must_close = 0;
|
|
parse_http_request(req2, &conn.request_info);
|
|
ASSERT(should_keep_alive(&conn) == 0);
|
|
|
|
parse_http_request(req3, &conn.request_info);
|
|
ASSERT(should_keep_alive(&conn) == 0);
|
|
|
|
parse_http_request(req4, &conn.request_info);
|
|
ASSERT(should_keep_alive(&conn) == 1);
|
|
|
|
conn.request_info.status_code = 401;
|
|
ASSERT(should_keep_alive(&conn) == 0);
|
|
|
|
conn.request_info.status_code = 200;
|
|
conn.must_close = 1;
|
|
ASSERT(should_keep_alive(&conn) == 0);
|
|
}
|
|
|
|
static void test_match_prefix(void) {
|
|
ASSERT(match_prefix("/api", 4, "/api") == 4);
|
|
ASSERT(match_prefix("/a/", 3, "/a/b/c") == 3);
|
|
ASSERT(match_prefix("/a/", 3, "/ab/c") == -1);
|
|
ASSERT(match_prefix("/*/", 3, "/ab/c") == 4);
|
|
ASSERT(match_prefix("**", 2, "/a/b/c") == 6);
|
|
ASSERT(match_prefix("/*", 2, "/a/b/c") == 2);
|
|
ASSERT(match_prefix("*/*", 3, "/a/b/c") == 2);
|
|
ASSERT(match_prefix("**/", 3, "/a/b/c") == 5);
|
|
ASSERT(match_prefix("**.foo|**.bar", 13, "a.bar") == 5);
|
|
ASSERT(match_prefix("a|b|cd", 6, "cdef") == 2);
|
|
ASSERT(match_prefix("a|b|c?", 6, "cdef") == 2);
|
|
ASSERT(match_prefix("a|?|cd", 6, "cdef") == 1);
|
|
ASSERT(match_prefix("/a/**.cgi", 9, "/foo/bar/x.cgi") == -1);
|
|
ASSERT(match_prefix("/a/**.cgi", 9, "/a/bar/x.cgi") == 12);
|
|
ASSERT(match_prefix("**/", 3, "/a/b/c") == 5);
|
|
ASSERT(match_prefix("**/$", 4, "/a/b/c") == -1);
|
|
ASSERT(match_prefix("**/$", 4, "/a/b/") == 5);
|
|
ASSERT(match_prefix("$", 1, "") == 0);
|
|
ASSERT(match_prefix("$", 1, "x") == -1);
|
|
ASSERT(match_prefix("*$", 2, "x") == 1);
|
|
ASSERT(match_prefix("/$", 2, "/") == 1);
|
|
ASSERT(match_prefix("**/$", 4, "/a/b/c") == -1);
|
|
ASSERT(match_prefix("**/$", 4, "/a/b/") == 5);
|
|
ASSERT(match_prefix("*", 1, "/hello/") == 0);
|
|
ASSERT(match_prefix("**.a$|**.b$", 11, "/a/b.b/") == -1);
|
|
ASSERT(match_prefix("**.a$|**.b$", 11, "/a/b.b") == 6);
|
|
ASSERT(match_prefix("**.a$|**.b$", 11, "/a/b.a") == 6);
|
|
}
|
|
|
|
static void test_remove_double_dots() {
|
|
struct { char before[20], after[20]; } data[] = {
|
|
{"////a", "/a"},
|
|
{"/.....", "/."},
|
|
{"/......", "/"},
|
|
{"...", "..."},
|
|
{"/...///", "/./"},
|
|
{"/a...///", "/a.../"},
|
|
{"/.x", "/.x"},
|
|
#if defined(_WIN32)
|
|
{"/\\", "/"},
|
|
#else
|
|
{"/\\", "/\\"},
|
|
#endif
|
|
{"/a\\", "/a\\"},
|
|
};
|
|
size_t i;
|
|
|
|
for (i = 0; i < ARRAY_SIZE(data); i++) {
|
|
//printf("[%s] -> [%s]\n", data[i].before, data[i].after);
|
|
remove_double_dots_and_double_slashes(data[i].before);
|
|
ASSERT(strcmp(data[i].before, data[i].after) == 0);
|
|
}
|
|
}
|
|
|
|
int main(void) {
|
|
test_match_prefix();
|
|
test_remove_double_dots();
|
|
test_should_keep_alive();
|
|
test_parse_http_request();
|
|
return 0;
|
|
}
|