mirror of
https://github.com/cesanta/mongoose.git
synced 2026-10-25 12:04:47 +00:00
Merge dev branch code named Fossa as next stable Mongoose
This commit is contained in:
Marko Mikulicic committed
1 parent
28eb251ccd
commit
8927c9d22b
126 files changed
+162049
-8502
No files matched your search
@@ -1,21 +0,0 @@
|
||||
# Copyright (c) 2014 Cesanta Software
|
||||
# All rights reserved
|
||||
|
||||
PROG = unit_test
|
||||
PROF_FLAGS = -fprofile-arcs -ftest-coverage -g -O0 -DGUI
|
||||
CFLAGS = -W -Wall -pthread -I.. $(PROF_FLAGS) $(CFLAGS_EXTRA)
|
||||
SOURCES = $(PROG).c
|
||||
|
||||
all: $(PROG)
|
||||
./$(PROG)
|
||||
gcov -b $(PROG).c | egrep '^(File|Lines)'
|
||||
|
||||
$(PROG): $(SOURCES) Makefile ../mongoose.c clean
|
||||
$(CC) -o $(PROG) $(SOURCES) $(CFLAGS) -ldl -lssl
|
||||
|
||||
win:
|
||||
wine cl $(SOURCES) /MD /nologo /DNDEBUG /O1 /Fe$(PROG).exe
|
||||
wine $(PROG).exe
|
||||
|
||||
clean:
|
||||
rm -rf $(PROG) *.exe *.dSYM *.obj *.exp .*o *.lib *.gc*
|
||||
@@ -1,533 +0,0 @@
|
||||
// Unit test for the mongoose web server.
|
||||
// g++ -W -Wall -pedantic -g unit_test.c -lssl && ./a.out
|
||||
// cl unit_test.c /MD
|
||||
|
||||
#ifndef _WIN32
|
||||
#define NS_ENABLE_IPV6
|
||||
#define NS_ENABLE_SSL
|
||||
#endif
|
||||
#define MONGOOSE_POST_SIZE_LIMIT 999
|
||||
|
||||
// USE_* definitions must be made before #include "mongoose.c" !
|
||||
#include "../mongoose.c"
|
||||
|
||||
#define FAIL(str, line) do { \
|
||||
printf("Fail on line %d: [%s]\n", line, str); \
|
||||
return str; \
|
||||
} while (0)
|
||||
|
||||
#define ASSERT(expr) do { \
|
||||
static_num_tests++; \
|
||||
if (!(expr)) FAIL(#expr, __LINE__); \
|
||||
} while (0)
|
||||
|
||||
#define RUN_TEST(test) do { const char *msg = test(); \
|
||||
if (msg) return msg; } while (0)
|
||||
|
||||
#define HTTP_PORT "45772"
|
||||
#define LISTENING_ADDR "127.0.0.1:" HTTP_PORT
|
||||
|
||||
static int static_num_tests = 0;
|
||||
|
||||
#if 0
|
||||
// Connects to host:port, and sends formatted request to it. Returns
|
||||
// malloc-ed reply and reply length, or NULL on error. Reply contains
|
||||
// everything including headers, not just the message body.
|
||||
static char *wget(const char *host, int port, int *len, const char *fmt, ...) {
|
||||
char buf[2000], *reply = NULL;
|
||||
int request_len, reply_size = 0, n, sock = -1;
|
||||
struct sockaddr_in sin;
|
||||
struct hostent *he = NULL;
|
||||
va_list ap;
|
||||
|
||||
if (host != NULL &&
|
||||
(he = gethostbyname(host)) != NULL &&
|
||||
(sock = socket(PF_INET, SOCK_STREAM, 0)) != -1) {
|
||||
sin.sin_family = AF_INET;
|
||||
sin.sin_port = htons((uint16_t) port);
|
||||
sin.sin_addr = * (struct in_addr *) he->h_addr_list[0];
|
||||
if (connect(sock, (struct sockaddr *) &sin, sizeof(sin)) == 0) {
|
||||
|
||||
// Format and send the request.
|
||||
va_start(ap, fmt);
|
||||
request_len = vsnprintf(buf, sizeof(buf), fmt, ap);
|
||||
va_end(ap);
|
||||
while (request_len > 0 && (n = send(sock, buf, request_len, 0)) > 0) {
|
||||
request_len -= n;
|
||||
}
|
||||
if (request_len == 0) {
|
||||
*len = 0;
|
||||
while ((n = recv(sock, buf, sizeof(buf), 0)) > 0) {
|
||||
if (*len + n > reply_size) {
|
||||
// Leak possible
|
||||
reply = (char *) realloc(reply, reply_size + sizeof(buf));
|
||||
reply_size += sizeof(buf);
|
||||
}
|
||||
if (reply != NULL) {
|
||||
memcpy(reply + *len, buf, n);
|
||||
*len += n;
|
||||
}
|
||||
}
|
||||
}
|
||||
closesocket(sock);
|
||||
}
|
||||
}
|
||||
|
||||
return reply;
|
||||
}
|
||||
#endif
|
||||
|
||||
static char *read_file(const char *path, int *size) {
|
||||
FILE *fp;
|
||||
struct stat st;
|
||||
char *data = NULL;
|
||||
if ((fp = fopen(path, "rb")) != NULL && !fstat(fileno(fp), &st)) {
|
||||
*size = (int) st.st_size;
|
||||
data = (char *) malloc(*size);
|
||||
fread(data, 1, *size, fp);
|
||||
fclose(fp);
|
||||
}
|
||||
return data;
|
||||
}
|
||||
|
||||
static const char *test_parse_http_message() {
|
||||
struct mg_connection 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";
|
||||
char req4[] = "GET / HTTP/1.1\r\nA: foo bar\r\nB: bar\r\nbaz\r\n\r\n";
|
||||
char req5[] = "GET / HTTP/1.1\r\n\r\n";
|
||||
char req6[] = "G";
|
||||
char req7[] = " blah ";
|
||||
char req8[] = " HTTP/1.1 200 OK \n\n";
|
||||
char req9[] = "HTTP/1.1 200 OK\r\nConnection: close\r\n\r\n";
|
||||
|
||||
ASSERT(get_request_len("\r\n", 3) == -1);
|
||||
ASSERT(get_request_len("\r\n", 2) == 0);
|
||||
ASSERT(get_request_len("GET", 3) == 0);
|
||||
ASSERT(get_request_len("\n\n", 2) == 2);
|
||||
ASSERT(get_request_len("\n\r\n", 3) == 3);
|
||||
ASSERT(get_request_len("\xdd\xdd", 2) == 0);
|
||||
ASSERT(get_request_len("\xdd\x03", 2) == -1);
|
||||
ASSERT(get_request_len(req3, sizeof(req3) - 1) == 0);
|
||||
ASSERT(get_request_len(req6, sizeof(req6) - 1) == 0);
|
||||
ASSERT(get_request_len(req7, sizeof(req7) - 1) == 0);
|
||||
|
||||
ASSERT(parse_http_message(req9, sizeof(req9) - 1, &ri) == sizeof(req9) - 1);
|
||||
ASSERT(ri.num_headers == 1);
|
||||
|
||||
ASSERT(parse_http_message(req1, sizeof(req1) - 1, &ri) == sizeof(req1) - 1);
|
||||
ASSERT(strcmp(ri.http_version, "1.1") == 0);
|
||||
ASSERT(ri.num_headers == 0);
|
||||
|
||||
ASSERT(parse_http_message(req2, sizeof(req2) - 1, &ri) == (size_t) ~0);
|
||||
ASSERT(parse_http_message(req6, 0, &ri) == (size_t) ~0);
|
||||
ASSERT(parse_http_message(req8, sizeof(req8) - 1, &ri) == sizeof(req8) - 1);
|
||||
|
||||
// TODO(lsm): Fix this. Header value may span multiple lines.
|
||||
ASSERT(parse_http_message(req4, sizeof(req4) - 1, &ri) == sizeof(req4) - 1);
|
||||
ASSERT(strcmp(ri.http_version, "1.1") == 0);
|
||||
ASSERT(ri.num_headers == 3);
|
||||
ASSERT(strcmp(ri.http_headers[0].name, "A") == 0);
|
||||
ASSERT(strcmp(ri.http_headers[0].value, "foo bar") == 0);
|
||||
ASSERT(strcmp(ri.http_headers[1].name, "B") == 0);
|
||||
ASSERT(strcmp(ri.http_headers[1].value, "bar") == 0);
|
||||
ASSERT(strcmp(ri.http_headers[2].name, "baz\r\n\r") == 0);
|
||||
ASSERT(strcmp(ri.http_headers[2].value, "") == 0);
|
||||
|
||||
ASSERT(parse_http_message(req5, sizeof(req5) - 1, &ri) == sizeof(req5) - 1);
|
||||
ASSERT(strcmp(ri.request_method, "GET") == 0);
|
||||
ASSERT(strcmp(ri.http_version, "1.1") == 0);
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_should_keep_alive(void) {
|
||||
struct mg_connection conn;
|
||||
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));
|
||||
ASSERT(parse_http_message(req1, sizeof(req1) - 1, &conn) == sizeof(req1) - 1);
|
||||
ASSERT(should_keep_alive(&conn) != 0);
|
||||
|
||||
parse_http_message(req2, sizeof(req2) - 1, &conn);
|
||||
ASSERT(should_keep_alive(&conn) == 0);
|
||||
|
||||
parse_http_message(req3, sizeof(req3) - 1, &conn);
|
||||
ASSERT(should_keep_alive(&conn) == 0);
|
||||
|
||||
parse_http_message(req4, sizeof(req4) - 1, &conn);
|
||||
ASSERT(should_keep_alive(&conn) != 0);
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_match_prefix(void) {
|
||||
ASSERT(mg_match_prefix("/api", 4, "/api") == 4);
|
||||
ASSERT(mg_match_prefix("/a/", 3, "/a/b/c") == 3);
|
||||
ASSERT(mg_match_prefix("/a/", 3, "/ab/c") == -1);
|
||||
ASSERT(mg_match_prefix("/blog/", 6, "/") == -1);
|
||||
ASSERT(mg_match_prefix("/*/", 3, "/ab/c") == 4);
|
||||
ASSERT(mg_match_prefix("**", 2, "/a/b/c") == 6);
|
||||
ASSERT(mg_match_prefix("/*", 2, "/a/b/c") == 2);
|
||||
ASSERT(mg_match_prefix("*/*", 3, "/a/b/c") == 2);
|
||||
ASSERT(mg_match_prefix("**/", 3, "/a/b/c") == 5);
|
||||
ASSERT(mg_match_prefix("**.foo|**.bar", 13, "a.bar") == 5);
|
||||
ASSERT(mg_match_prefix("a|b|cd", 6, "cdef") == 2);
|
||||
ASSERT(mg_match_prefix("a|b|c?", 6, "cdef") == 2);
|
||||
ASSERT(mg_match_prefix("a|?|cd", 6, "cdef") == 1);
|
||||
ASSERT(mg_match_prefix("/a/**.cgi", 9, "/foo/bar/x.cgi") == -1);
|
||||
ASSERT(mg_match_prefix("/a/**.cgi", 9, "/a/bar/x.cgi") == 12);
|
||||
ASSERT(mg_match_prefix("**/", 3, "/a/b/c") == 5);
|
||||
ASSERT(mg_match_prefix("**/$", 4, "/a/b/c") == -1);
|
||||
ASSERT(mg_match_prefix("**/$", 4, "/a/b/") == 5);
|
||||
ASSERT(mg_match_prefix("$", 1, "") == 0);
|
||||
ASSERT(mg_match_prefix("$", 1, "x") == -1);
|
||||
ASSERT(mg_match_prefix("*$", 2, "x") == 1);
|
||||
ASSERT(mg_match_prefix("/$", 2, "/") == 1);
|
||||
ASSERT(mg_match_prefix("**/$", 4, "/a/b/c") == -1);
|
||||
ASSERT(mg_match_prefix("**/$", 4, "/a/b/") == 5);
|
||||
ASSERT(mg_match_prefix("*", 1, "/hello/") == 0);
|
||||
ASSERT(mg_match_prefix("**.a$|**.b$", 11, "/a/b.b/") == -1);
|
||||
ASSERT(mg_match_prefix("**.a$|**.b$", 11, "/a/b.b") == 6);
|
||||
ASSERT(mg_match_prefix("**.a$|**.b$", 11, "/a/B.A") == 6);
|
||||
ASSERT(mg_match_prefix("**o$", 4, "HELLO") == 5);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_remove_double_dots() {
|
||||
struct { char before[30], after[30]; } data[] = {
|
||||
{"////a", "/a"},
|
||||
{"/.....", "/....."},
|
||||
{"/......", "/......"},
|
||||
{"...", "..."},
|
||||
{"/...///", "/.../"},
|
||||
{"/a...///", "/a.../"},
|
||||
{"/.x", "/.x"},
|
||||
{"/\\", "/"},
|
||||
{"/a\\", "/a\\"},
|
||||
{"/a\\\\...", "/a\\..."},
|
||||
{"foo/x..y/././y/../../..", "foo/x..y/y/"},
|
||||
{"foo/..x", "foo/..x"},
|
||||
};
|
||||
size_t i;
|
||||
|
||||
for (i = 0; i < ARRAY_SIZE(data); i++) {
|
||||
remove_double_dots_and_double_slashes(data[i].before);
|
||||
ASSERT(strcmp(data[i].before, data[i].after) == 0);
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_get_var(void) {
|
||||
static const char *data = "a=1&&b=2&d&=&c=3%20&e=&k=aa&a=23";
|
||||
static const char *data2 = "q=&st=2012%2F11%2F13+17%3A05&et=&team_id=";
|
||||
char buf[20];
|
||||
|
||||
ASSERT(get_var(data, strlen(data), "a", buf, sizeof(buf), 0) == 1);
|
||||
ASSERT(buf[0] == '1' && buf[1] == '\0');
|
||||
ASSERT(get_var(data, strlen(data), "a", buf, sizeof(buf), 1) == 2);
|
||||
ASSERT(strcmp(buf, "23") == 0);
|
||||
ASSERT(get_var(data, strlen(data), "b", buf, sizeof(buf), 0) == 1);
|
||||
ASSERT(buf[0] == '2' && buf[1] == '\0');
|
||||
ASSERT(get_var(data, strlen(data), "c", buf, sizeof(buf), 0) == 2);
|
||||
ASSERT(buf[0] == '3' && buf[1] == ' ' && buf[2] == '\0');
|
||||
ASSERT(get_var(data, strlen(data), "e", buf, sizeof(buf), 0) == 0);
|
||||
ASSERT(buf[0] == '\0');
|
||||
|
||||
ASSERT(get_var(data, strlen(data), "d", buf, sizeof(buf), 0) == -1);
|
||||
ASSERT(get_var(data, strlen(data), "c", buf, 2, 0) == -2);
|
||||
|
||||
ASSERT(get_var(data, strlen(data), "x", NULL, 10, 0) == -2);
|
||||
ASSERT(get_var(data, strlen(data), "x", buf, 0, 0) == -2);
|
||||
ASSERT(get_var(data2, strlen(data2), "st", buf, 16, 0) == -2);
|
||||
ASSERT(get_var(data2, strlen(data2), "st", buf, 17, 0) == 16);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_url_decode(void) {
|
||||
char buf[100];
|
||||
|
||||
ASSERT(mg_url_decode("foo", 3, buf, 3, 0) == -1); // No space for \0
|
||||
ASSERT(mg_url_decode("foo", 3, buf, 4, 0) == 3);
|
||||
ASSERT(strcmp(buf, "foo") == 0);
|
||||
|
||||
ASSERT(mg_url_decode("a+", 2, buf, sizeof(buf), 0) == 2);
|
||||
ASSERT(strcmp(buf, "a+") == 0);
|
||||
|
||||
ASSERT(mg_url_decode("a+", 2, buf, sizeof(buf), 1) == 2);
|
||||
ASSERT(strcmp(buf, "a ") == 0);
|
||||
|
||||
ASSERT(mg_url_decode("%61", 1, buf, sizeof(buf), 1) == 1);
|
||||
printf("[%s]\n", buf);
|
||||
ASSERT(strcmp(buf, "%") == 0);
|
||||
|
||||
ASSERT(mg_url_decode("%61", 2, buf, sizeof(buf), 1) == 2);
|
||||
ASSERT(strcmp(buf, "%6") == 0);
|
||||
|
||||
ASSERT(mg_url_decode("%61", 3, buf, sizeof(buf), 1) == 1);
|
||||
ASSERT(strcmp(buf, "a") == 0);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_url_encode(void) {
|
||||
char buf[100];
|
||||
ASSERT(mg_url_encode("", 0, buf, sizeof(buf)) == 0);
|
||||
ASSERT(buf[0] == '\0');
|
||||
ASSERT(mg_url_encode("foo", 3, buf, sizeof(buf)) == 3);
|
||||
ASSERT(strcmp(buf, "foo") == 0);
|
||||
ASSERT(mg_url_encode("f o", 3, buf, sizeof(buf)) == 5);
|
||||
ASSERT(strcmp(buf, "f%20o") == 0);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_to64(void) {
|
||||
ASSERT(to64("0") == 0);
|
||||
ASSERT(to64("") == 0);
|
||||
ASSERT(to64("123") == 123);
|
||||
ASSERT(to64("-34") == -34);
|
||||
ASSERT(to64("3566626116") == 3566626116);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_base64_encode(void) {
|
||||
const char *in[] = {"a", "ab", "abc", "abcd", NULL};
|
||||
const char *out[] = {"YQ==", "YWI=", "YWJj", "YWJjZA=="};
|
||||
char buf[100];
|
||||
int i;
|
||||
|
||||
for (i = 0; in[i] != NULL; i++) {
|
||||
base64_encode((unsigned char *) in[i], strlen(in[i]), buf);
|
||||
ASSERT(!strcmp(buf, out[i]));
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_mg_parse_header(void) {
|
||||
const char *str = "xx=1 kl yy, ert=234 kl=123, uri=\"/?name=x,y\", "
|
||||
"ii=\"12\\\"34\" zz='aa bb',tt=2,gf=\"xx d=1234";
|
||||
char buf[20];
|
||||
ASSERT(mg_parse_header(str, "yy", buf, sizeof(buf)) == 0);
|
||||
ASSERT(mg_parse_header(str, "ert", buf, sizeof(buf)) == 3);
|
||||
ASSERT(strcmp(buf, "234") == 0);
|
||||
ASSERT(mg_parse_header(str, "ert", buf, 2) == 0);
|
||||
ASSERT(mg_parse_header(str, "ert", buf, 3) == 0);
|
||||
ASSERT(mg_parse_header(str, "ert", buf, 4) == 3);
|
||||
ASSERT(mg_parse_header(str, "gf", buf, sizeof(buf)) == 0);
|
||||
ASSERT(mg_parse_header(str, "zz", buf, sizeof(buf)) == 5);
|
||||
ASSERT(strcmp(buf, "aa bb") == 0);
|
||||
ASSERT(mg_parse_header(str, "d", buf, sizeof(buf)) == 4);
|
||||
ASSERT(strcmp(buf, "1234") == 0);
|
||||
buf[0] = 'x';
|
||||
ASSERT(mg_parse_header(str, "MMM", buf, sizeof(buf)) == 0);
|
||||
ASSERT(buf[0] == '\0');
|
||||
ASSERT(mg_parse_header(str, "kl", buf, sizeof(buf)) == 3);
|
||||
ASSERT(strcmp(buf, "123") == 0);
|
||||
ASSERT(mg_parse_header(str, "xx", buf, sizeof(buf)) == 1);
|
||||
ASSERT(strcmp(buf, "1") == 0);
|
||||
ASSERT(mg_parse_header(str, "ii", buf, sizeof(buf)) == 5);
|
||||
ASSERT(strcmp(buf, "12\"34") == 0);
|
||||
ASSERT(mg_parse_header(str, "tt", buf, sizeof(buf)) == 1);
|
||||
ASSERT(strcmp(buf, "2") == 0);
|
||||
ASSERT(mg_parse_header(str, "uri", buf, sizeof(buf)) == 10);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_next_option(void) {
|
||||
const char *p, *list = "x/8,/y**=1;2k,z";
|
||||
struct vec a, b;
|
||||
int i;
|
||||
|
||||
ASSERT(next_option(NULL, &a, &b) == NULL);
|
||||
for (i = 0, p = list; (p = next_option(p, &a, &b)) != NULL; i++) {
|
||||
ASSERT(i != 0 || (a.ptr == list && a.len == 3 && b.len == 0));
|
||||
ASSERT(i != 1 || (a.ptr == list + 4 && a.len == 4 && b.ptr == list + 9 &&
|
||||
b.len == 4));
|
||||
|
||||
ASSERT(i != 2 || (a.ptr == list + 14 && a.len == 1 && b.len == 0));
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static int evh1(struct mg_connection *conn, enum mg_event ev) {
|
||||
char *buf = (char *) conn->connection_param;
|
||||
int result = MG_FALSE;
|
||||
|
||||
switch (ev) {
|
||||
case MG_CONNECT:
|
||||
mg_printf(conn, "GET %s HTTP/1.0\r\n\r\n",
|
||||
buf[0] == '1' ? "/cb1" : "/non_exist");
|
||||
result = MG_TRUE;
|
||||
break;
|
||||
case MG_HTTP_ERROR:
|
||||
mg_printf(conn, "HTTP/1.0 404 NF\r\n\r\nERR: %d", conn->status_code);
|
||||
result = MG_TRUE;
|
||||
break;
|
||||
case MG_REQUEST:
|
||||
if (!strcmp(conn->uri, "/cb1")) {
|
||||
mg_printf(conn, "HTTP/1.0 200 OK\r\n\r\n%s %s %s",
|
||||
(char *) conn->server_param,
|
||||
buf == NULL ? "?" : "!", conn->remote_ip);
|
||||
result = MG_TRUE;
|
||||
}
|
||||
break;
|
||||
case MG_REPLY:
|
||||
if (buf != NULL) {
|
||||
sprintf(buf + 1, "%.*s", (int) conn->content_len, conn->content);
|
||||
}
|
||||
break;
|
||||
case MG_AUTH:
|
||||
result = MG_TRUE;
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
static const char *test_server(void) {
|
||||
char buf1[100] = "1", buf2[100] = "2";
|
||||
struct mg_server *server = mg_create_server((void *) "foo", evh1);
|
||||
struct mg_connection *conn;
|
||||
|
||||
ASSERT(server != NULL);
|
||||
ASSERT(mg_set_option(server, "listening_port", LISTENING_ADDR) == NULL);
|
||||
ASSERT(mg_set_option(server, "document_root", ".") == NULL);
|
||||
|
||||
ASSERT((conn = mg_connect(server, "127.0.0.1:" HTTP_PORT)) != NULL);
|
||||
conn->connection_param = buf1;
|
||||
ASSERT((conn = mg_connect(server, "127.0.0.1:" HTTP_PORT)) != NULL);
|
||||
conn->connection_param = buf2;
|
||||
|
||||
{ int i; for (i = 0; i < 50; i++) mg_poll_server(server, 1); }
|
||||
ASSERT(strcmp(buf1, "1foo ? 127.0.0.1") == 0);
|
||||
ASSERT(strcmp(buf2, "2ERR: 404") == 0);
|
||||
|
||||
ASSERT(strcmp(static_config_options[URL_REWRITES * 2], "url_rewrites") == 0);
|
||||
mg_destroy_server(&server);
|
||||
ASSERT(server == NULL);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
#define DISP "Content-Disposition: form/data; "
|
||||
#define CRLF "\r\n"
|
||||
#define BOUNDARY "--xyz"
|
||||
static const char *test_parse_multipart(void) {
|
||||
char a[100], b[100];
|
||||
const char *p;
|
||||
static const char f1[] = BOUNDARY CRLF DISP "name=f1" CRLF CRLF
|
||||
"some_content" CRLF BOUNDARY CRLF
|
||||
BOUNDARY CRLF DISP "name=f2; filename=\"foo bar.txt\"" CRLF CRLF
|
||||
"another_content" CRLF BOUNDARY CRLF
|
||||
"--" CRLF;
|
||||
int n, n2, len, f1_len = sizeof(f1) - 1;
|
||||
|
||||
ASSERT(mg_parse_multipart("", 0, a, sizeof(a), b, sizeof(b), &p, &len) == 0);
|
||||
ASSERT(mg_parse_multipart("a", 1, a, sizeof(a), b, sizeof(b), &p, &len) == 0);
|
||||
ASSERT((n = mg_parse_multipart(f1, f1_len, a, sizeof(a),
|
||||
b, sizeof(b), &p, &len)) > 0);
|
||||
ASSERT(len == 12);
|
||||
ASSERT(memcmp(p, "some_content", len) == 0);
|
||||
ASSERT(strcmp(a, "f1") == 0);
|
||||
ASSERT(b[0] == '\0');
|
||||
|
||||
ASSERT((n2 = mg_parse_multipart(f1 + n, f1_len - n, a, sizeof(a),
|
||||
b, sizeof(b), &p, &len)) > 0);
|
||||
ASSERT(len == 15);
|
||||
ASSERT(memcmp(p, "another_content", len) == 0);
|
||||
ASSERT(strcmp(a, "f2") == 0);
|
||||
ASSERT(strcmp(b, "foo bar.txt") == 0);
|
||||
|
||||
ASSERT((n2 = mg_parse_multipart(f1 + n + n2, f1_len - (n + n2), a, sizeof(a),
|
||||
b, sizeof(b), &p, &len)) == 0);
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static int evh2(struct mg_connection *conn, enum mg_event ev) {
|
||||
char *file_data, *cp = (char *) conn->connection_param;
|
||||
int file_size, result = MG_FALSE;
|
||||
|
||||
switch (ev) {
|
||||
case MG_AUTH:
|
||||
result = MG_TRUE;
|
||||
break;
|
||||
case MG_CONNECT:
|
||||
mg_printf(conn, "GET /%s HTTP/1.0\r\n\r\n", cp);
|
||||
result = MG_TRUE;
|
||||
break;
|
||||
case MG_REQUEST:
|
||||
break;
|
||||
case MG_REPLY:
|
||||
file_data = read_file("unit_test.c", &file_size);
|
||||
sprintf(cp, "%d %s", (size_t) file_size == conn->content_len &&
|
||||
memcmp(file_data, conn->content, file_size) == 0 ? 1 : 0,
|
||||
conn->query_string == NULL ? "?" : conn->query_string);
|
||||
free(file_data);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
static const char *test_mg_set_option(void) {
|
||||
struct mg_server *server = mg_create_server(NULL, NULL);
|
||||
ASSERT(mg_set_option(server, "listening_port", "0") == NULL);
|
||||
ASSERT(mg_get_option(server, "listening_port")[0] != '\0');
|
||||
mg_destroy_server(&server);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *test_rewrites(void) {
|
||||
char buf1[100] = "xx", addr[50];
|
||||
struct mg_server *server = mg_create_server(NULL, evh2);
|
||||
struct mg_connection *conn;
|
||||
const char *port;
|
||||
|
||||
ASSERT(mg_set_option(server, "listening_port", "0") == NULL);
|
||||
ASSERT(mg_set_option(server, "document_root", ".") == NULL);
|
||||
ASSERT(mg_set_option(server, "url_rewrites", "/xx=unit_test.c") == NULL);
|
||||
ASSERT((port = mg_get_option(server, "listening_port")) != NULL);
|
||||
snprintf(addr, sizeof(addr), "127.0.0.1:%s", port);
|
||||
ASSERT((conn = mg_connect(server, addr)) != NULL);
|
||||
conn->connection_param = buf1;
|
||||
|
||||
{ int i; for (i = 0; i < 50; i++) mg_poll_server(server, 1); }
|
||||
|
||||
ASSERT(strcmp(buf1, "1 ?") == 0);
|
||||
mg_destroy_server(&server);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static const char *run_all_tests(void) {
|
||||
RUN_TEST(test_should_keep_alive);
|
||||
RUN_TEST(test_match_prefix);
|
||||
RUN_TEST(test_remove_double_dots);
|
||||
RUN_TEST(test_parse_http_message);
|
||||
RUN_TEST(test_to64);
|
||||
RUN_TEST(test_url_decode);
|
||||
RUN_TEST(test_url_encode);
|
||||
RUN_TEST(test_base64_encode);
|
||||
RUN_TEST(test_mg_parse_header);
|
||||
RUN_TEST(test_get_var);
|
||||
RUN_TEST(test_next_option);
|
||||
RUN_TEST(test_parse_multipart);
|
||||
RUN_TEST(test_mg_set_option);
|
||||
RUN_TEST(test_server);
|
||||
RUN_TEST(test_rewrites);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int __cdecl main(void) {
|
||||
const char *fail_msg = run_all_tests();
|
||||
printf("%s, tests run: %d\n", fail_msg ? "FAIL" : "PASS", static_num_tests);
|
||||
return fail_msg == NULL ? EXIT_SUCCESS : EXIT_FAILURE;
|
||||
}
|
||||
Reference in new issue
Block a user