lib/libhttp
1
0
Fork 0
mirror of https://github.com/lammertb/libhttp.git synced 2025-12-22 04:02:04 +03:00
Code Activity
Files
38d130a2cc9a60744a8b83b6467d19a377e52888
libhttp/test/unit_test.c
bel ac3968c850 Add patch of pull request #196 to unit_test.c 
The pull request can not be applied automatically, due to merge conflicts.
It is easier to change the few lines manually than to resolve all these conflicts.
2015-08-25 22:02:20 +02:00

1520 lines
46 KiB
C
Raw Blame History

/* Copyright (c) 2013-2015 the Civetweb developers
* Copyright (c) 2004-2013 Sergey Lyubka
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
/* Unit test for the civetweb web server. Tests embedded API.
*/
#define USE_WEBSOCKET
#ifndef _WIN32
#define __cdecl
#define USE_IPV6
#endif
/* USE_* definitions must be made before #include "civetweb.c" !
* We include the source file so that our object file will have visibility to
* all the static functions.
*/
#include "civetweb.c"
static int s_total_tests = 0;
static int s_failed_tests = 0;
void check_func(int condition, const char *cond_txt, unsigned line)
{
++s_total_tests;
if (!condition) {
printf("Fail on line %d: [%s]\n", line, cond_txt);
++s_failed_tests;
}
}
#define ASSERT(expr) \
do { \
check_func(expr, #expr, __LINE__); \
} while (0)
#define REQUIRE(expr) \
do { \
check_func(expr, #expr, __LINE__); \
if (!(expr)) { \
exit(EXIT_FAILURE); \
} \
} while (0)
#define HTTP_PORT "8080"
#ifdef NO_SSL
#define HTTPS_PORT HTTP_PORT
#define LISTENING_ADDR "127.0.0.1:" HTTP_PORT
#else
#define HTTP_REDIRECT_PORT "8088"
#define HTTPS_PORT "8443"
#define LISTENING_ADDR \
"127.0.0.1:" HTTP_PORT ",127.0.0.1:" HTTP_REDIRECT_PORT "r" \
",127.0.0.1:" HTTPS_PORT "s"
#endif
static void test_parse_http_message()
{
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";
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(parse_http_message(req9, sizeof(req9), &ri) == sizeof(req9) - 1);
ASSERT(ri.num_headers == 1);
ASSERT(parse_http_message(req1, sizeof(req1), &ri) == sizeof(req1) - 1);
ASSERT(strcmp(ri.http_version, "1.1") == 0);
ASSERT(ri.num_headers == 0);
ASSERT(parse_http_message(req2, sizeof(req2), &ri) == -1);
ASSERT(parse_http_message(req3, sizeof(req3), &ri) == 0);
ASSERT(parse_http_message(req6, sizeof(req6), &ri) == 0);
ASSERT(parse_http_message(req7, sizeof(req7), &ri) == 0);
ASSERT(parse_http_message("", 0, &ri) == 0);
ASSERT(parse_http_message(req8, sizeof(req8), &ri) == sizeof(req8) - 1);
/* TODO(lsm): Fix this. Header value may span multiple lines. */
ASSERT(parse_http_message(req4, sizeof(req4), &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), &ri) == sizeof(req5) - 1);
ASSERT(strcmp(ri.request_method, "GET") == 0);
ASSERT(strcmp(ri.http_version, "1.1") == 0);
}
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;
ASSERT(parse_http_message(req1, sizeof(req1), &conn.request_info) ==
sizeof(req1) - 1);
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_message(req2, sizeof(req2), &conn.request_info);
ASSERT(should_keep_alive(&conn) == 0);
parse_http_message(req3, sizeof(req3), &conn.request_info);
ASSERT(should_keep_alive(&conn) == 0);
parse_http_message(req4, sizeof(req4), &conn.request_info);
ASSERT(should_keep_alive(&conn) == 1);
conn.status_code = 401;
ASSERT(should_keep_alive(&conn) == 0);
conn.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);
ASSERT(match_prefix("**o$", 4, "HELLO") == 5);
}
static void test_remove_double_dots()
{
struct {
char before[20], after[20];
} data[] = {
{"////a", "/a"},
{"/.....", "/."},
{"/......", "/"},
{"...", "..."},
{"/...///", "/./"},
{"/a...///", "/a.../"},
{"/.x", "/.x"},
{"/\\", "/"},
{"/a\\", "/a\\"},
{"/a\\\\...", "/a\\."},
};
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);
}
}
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 = mg_malloc(*size);
ASSERT(data != NULL);
ASSERT(fread(data, 1, *size, fp) == (size_t)*size);
fclose(fp);
}
return data;
}
static long fetch_data_size = 1024 * 1024;
static char *fetch_data;
static const char *inmemory_file_data = "hi there";
static const char *upload_filename = "upload_test.txt";
#if 0
static const char *upload_filename2 = "upload_test2.txt";
#endif
static const char *upload_ok_message = "upload successful";
static const char *
open_file_cb(const struct mg_connection *conn, const char *path, size_t *size)
{
(void)conn;
if (!strcmp(path, "./blah")) {
*size = strlen(inmemory_file_data);
return inmemory_file_data;
}
return NULL;
}
static void upload_cb(struct mg_connection *conn, const char *path)
{
const struct mg_request_info *ri = mg_get_request_info(conn);
char *p1, *p2;
int len1, len2;
if (atoi(ri->query_string) == 1) {
ASSERT(!strcmp(path, "./upload_test.txt"));
ASSERT((p1 = read_file("src/civetweb.c", &len1)) != NULL);
ASSERT((p2 = read_file(path, &len2)) != NULL);
ASSERT(len1 == len2);
ASSERT(memcmp(p1, p2, len1) == 0);
mg_free(p1);
mg_free(p2);
remove(upload_filename);
} else if (atoi(ri->query_string) == 2) {
if (!strcmp(path, "./upload_test.txt")) {
ASSERT((p1 = read_file("include/civetweb.h", &len1)) != NULL);
ASSERT((p2 = read_file(path, &len2)) != NULL);
ASSERT(len1 == len2);
ASSERT(memcmp(p1, p2, len1) == 0);
mg_free(p1);
mg_free(p2);
remove(upload_filename);
} else if (!strcmp(path, "./upload_test2.txt")) {
ASSERT((p1 = read_file("README.md", &len1)) != NULL);
ASSERT((p2 = read_file(path, &len2)) != NULL);
ASSERT(len1 == len2);
ASSERT(memcmp(p1, p2, len1) == 0);
mg_free(p1);
mg_free(p2);
remove(upload_filename);
} else {
ASSERT(0);
}
} else {
ASSERT(0);
}
mg_printf(conn,
"HTTP/1.0 200 OK\r\nContent-Length: %d\r\n\r\n%s",
(int)strlen(upload_ok_message),
upload_ok_message);
}
static int begin_request_handler_cb(struct mg_connection *conn)
{
const struct mg_request_info *ri = mg_get_request_info(conn);
int req_len = (int)(ri->content_length);
const char *s_req_len = mg_get_header(conn, "Content-Length");
char *data;
long to_write, write_now;
int bytes_read, bytes_written;
ASSERT(((req_len == -1) && (s_req_len == NULL)) ||
((s_req_len != NULL) && (req_len = atol(s_req_len))));
if (!strncmp(ri->uri, "/data/", 6)) {
if (!strcmp(ri->uri + 6, "all")) {
to_write = fetch_data_size;
} else {
to_write = atol(ri->uri + 6);
}
mg_printf(conn,
"HTTP/1.1 200 OK\r\n"
"Connection: close\r\n"
"Content-Length: %li\r\n"
"Content-Type: text/plain\r\n\r\n",
to_write);
while (to_write > 0) {
write_now = to_write > fetch_data_size ? fetch_data_size : to_write;
bytes_written = mg_write(conn, fetch_data, write_now);
ASSERT(bytes_written == write_now);
if (bytes_written < 0) {
ASSERT(0);
break;
}
to_write -= bytes_written;
}
close_connection(conn);
return 1;
}
if (!strcmp(ri->uri, "/content_length")) {
if (req_len > 0) {
data = mg_malloc(req_len);
assert(data != NULL);
bytes_read = mg_read(conn, data, req_len);
ASSERT(bytes_read == req_len);
mg_printf(conn,
"HTTP/1.1 200 OK\r\n"
"Connection: close\r\n"
"Content-Length: %d\r\n" /* The official definition */
"Content-Type: text/plain\r\n\r\n",
bytes_read);
mg_write(conn, data, bytes_read);
mg_free(data);
} else {
data = mg_malloc(1024);
assert(data != NULL);
bytes_read = mg_read(conn, data, 1024);
mg_printf(conn,
"HTTP/1.1 200 OK\r\n"
"Connection: close\r\n"
"Content-Type: text/plain\r\n\r\n");
mg_write(conn, data, bytes_read);
mg_free(data);
}
close_connection(conn);
return 1;
}
if (!strcmp(ri->uri, "/upload")) {
ASSERT(ri->query_string != NULL);
ASSERT(mg_upload(conn, ".") == atoi(ri->query_string));
}
return 0;
}
static int log_message_cb(const struct mg_connection *conn, const char *msg)
{
(void)conn;
printf("%s\n", msg);
return 0;
}
int (*begin_request)(struct mg_connection *);
void (*end_request)(const struct mg_connection *, int reply_status_code);
int (*log_message)(const struct mg_connection *, const char *message);
int (*init_ssl)(void *ssl_context, void *user_data);
int (*websocket_connect)(const struct mg_connection *);
void (*websocket_ready)(struct mg_connection *);
int (*websocket_data)(struct mg_connection *,
int bits,
char *data,
size_t data_len);
void (*connection_close)(struct mg_connection *);
const char *(*open_file)(const struct mg_connection *,
const char *path,
size_t *data_len);
void (*init_lua)(struct mg_connection *, void *lua_context);
void (*upload)(struct mg_connection *, const char *file_name);
int (*http_error)(struct mg_connection *, int status);
static struct mg_callbacks CALLBACKS;
static void init_CALLBACKS()
{
memset(&CALLBACKS, 0, sizeof(CALLBACKS));
CALLBACKS.begin_request = begin_request_handler_cb;
CALLBACKS.log_message = log_message_cb;
CALLBACKS.open_file = open_file_cb;
CALLBACKS.upload = upload_cb;
};
static const char *OPTIONS[] = {
"document_root",
".",
"listening_ports",
LISTENING_ADDR,
"enable_keep_alive",
"yes",
#ifndef NO_SSL
"ssl_certificate",
"../resources/ssl_cert.pem",
#endif
NULL,
};
static char *read_conn(struct mg_connection *conn, int *size)
{
char buf[100], *data = NULL;
int len;
*size = 0;
while ((len = mg_read(conn, buf, sizeof(buf))) > 0) {
*size += len;
data = mg_realloc(data, *size);
ASSERT(data != NULL);
memcpy(data + *size - len, buf, len);
}
return data;
}
#ifdef MEMORY_DEBUGGING
extern unsigned long mg_memory_debug_blockCount;
extern unsigned long mg_memory_debug_totalMemUsed;
#endif
static void ut_mg_stop(struct mg_context *ctx)
{
/* mg_stop for unit_test */
mg_stop(ctx);
#ifdef MEMORY_DEBUGGING
ASSERT(mg_memory_debug_blockCount == 0);
ASSERT(mg_memory_debug_totalMemUsed == 0);
mg_memory_debug_blockCount = 0;
mg_memory_debug_totalMemUsed = 0;
#endif
mg_sleep(31000); /* This is required to ensure the operating system already
allows to use the port again */
}
static void test_mg_download(int use_ssl)
{
const char *test_data = "123456789A123456789B";
char *p1, *p2, ebuf[100];
const char *h;
int i, len1, len2, port;
struct mg_connection *conn;
struct mg_context *ctx;
const struct mg_request_info *ri;
if (use_ssl) {
port = atoi(HTTPS_PORT);
} else {
port = atoi(HTTP_PORT);
}
ctx = mg_start(&CALLBACKS, NULL, OPTIONS);
ASSERT(ctx != NULL);
ASSERT(mg_download(NULL, port, use_ssl, ebuf, sizeof(ebuf), "%s", "") ==
NULL);
ASSERT(mg_download("localhost", 0, use_ssl, ebuf, sizeof(ebuf), "%s", "") ==
NULL);
ASSERT(
mg_download("localhost", port, use_ssl, ebuf, sizeof(ebuf), "%s", "") ==
NULL);
/* Fetch nonexistent file, should see 404 */
ASSERT((conn = mg_download("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"%s",
"GET /gimbec HTTP/1.0\r\n\r\n")) != NULL);
ASSERT(strcmp(conn->request_info.uri, "404") == 0);
mg_close_connection(conn);
if (use_ssl) {
ASSERT((conn = mg_download("google.com",
443,
1,
ebuf,
sizeof(ebuf),
"%s",
"GET / HTTP/1.0\r\n\r\n")) != NULL);
mg_close_connection(conn);
} else {
ASSERT((conn = mg_download("google.com",
80,
0,
ebuf,
sizeof(ebuf),
"%s",
"GET / HTTP/1.0\r\n\r\n")) != NULL);
mg_close_connection(conn);
}
/* Fetch unit_test.c, should succeed */
ASSERT((conn = mg_download("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"%s",
"GET /unit_test.c HTTP/1.0\r\n\r\n")) != NULL);
ASSERT(&conn->request_info == mg_get_request_info(conn));
ASSERT(!strcmp(conn->request_info.uri, "200"));
ASSERT((p1 = read_conn(conn, &len1)) != NULL);
ASSERT((p2 = read_file("unit_test.c", &len2)) != NULL);
ASSERT(len1 == len2);
ASSERT(memcmp(p1, p2, len1) == 0);
mg_free(p1);
mg_free(p2);
mg_close_connection(conn);
/* Fetch in-memory file, should succeed. */
ASSERT((conn = mg_download("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"%s",
"GET /blah HTTP/1.1\r\n\r\n")) != NULL);
ASSERT((p1 = read_conn(conn, &len1)) != NULL);
ASSERT(len1 == (int)strlen(inmemory_file_data));
ASSERT(memcmp(p1, inmemory_file_data, len1) == 0);
mg_free(p1);
mg_close_connection(conn);
/* Fetch in-memory data with no Content-Length, should succeed. */
ASSERT((conn = mg_download("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"%s",
"GET /data/all HTTP/1.1\r\n\r\n")) != NULL);
ASSERT(conn->request_info.content_length == fetch_data_size);
ASSERT((p1 = read_conn(conn, &len1)) != NULL);
ASSERT(len1 == (int)fetch_data_size);
ASSERT(memcmp(p1, fetch_data, len1) == 0);
mg_free(p1);
mg_close_connection(conn);
/* Fetch in-memory data with no Content-Length, should succeed. */
for (i = 0; i <= 1024 * /* 1024 * */ 8; i += (i < 2 ? 1 : i)) {
ASSERT((conn = mg_download("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"GET /data/%i HTTP/1.1\r\n\r\n",
i)) != NULL);
ASSERT(conn->request_info.content_length == i);
len1 = -1;
p1 = read_conn(conn, &len1);
if (i == 0) {
ASSERT(len1 == 0);
ASSERT(p1 == 0);
} else if (i <= fetch_data_size) {
ASSERT(p1 != NULL);
ASSERT(len1 == i);
ASSERT(memcmp(p1, fetch_data, len1) == 0);
} else {
ASSERT(p1 != NULL);
ASSERT(len1 == i);
ASSERT(memcmp(p1, fetch_data, fetch_data_size) == 0);
}
mg_free(p1);
mg_close_connection(conn);
}
/* Fetch data with Content-Length, should succeed and return the defined
* length. */
ASSERT((conn = mg_download(
"localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"POST /content_length HTTP/1.1\r\nContent-Length: %u\r\n\r\n%s",
(unsigned)strlen(test_data),
test_data)) != NULL);
h = mg_get_header(conn, "Content-Length");
ASSERT((h != NULL) && (atoi(h) == (int)strlen(test_data)));
ASSERT((p1 = read_conn(conn, &len1)) != NULL);
ASSERT(len1 == (int)strlen(test_data));
ASSERT(conn->request_info.content_length == (int)strlen(test_data));
ASSERT(memcmp(p1, test_data, len1) == 0);
ASSERT(strcmp(conn->request_info.request_method, "HTTP/1.1") == 0);
ASSERT(strcmp(conn->request_info.uri, "200") == 0);
ASSERT(strcmp(conn->request_info.http_version, "OK") == 0);
mg_free(p1);
mg_close_connection(conn);
/* A POST request without Content-Length set is only valid, if the request
* used Transfer-Encoding: chunked. Otherwise, it is an HTTP protocol
* violation. Here we send a chunked request, the reply is not chunked. */
ASSERT((conn = mg_download("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"POST /content_length "
"HTTP/1.1\r\n"
"Transfer-Encoding: chunked\r\n"
"\r\n%x\r\n%s\r\n0\r\n\r\n",
strlen(test_data),
test_data)) != NULL);
h = mg_get_header(conn, "Content-Length");
ASSERT(h == NULL);
ASSERT(conn->request_info.content_length == -1);
ASSERT((p1 = read_conn(conn, &len1)) != NULL);
ASSERT(len1 == (int)strlen(test_data));
ASSERT(memcmp(p1, test_data, len1) == 0);
mg_free(p1);
mg_close_connection(conn);
/* Another chunked POST request with different chunk sizes. */
ASSERT((conn = mg_download("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"POST /content_length "
"HTTP/1.1\r\n"
"Transfer-Encoding: chunked\r\n\r\n"
"2\r\n%c%c\r\n"
"1\r\n%c\r\n"
"2\r\n%c%c\r\n"
"2\r\n%c%c\r\n"
"%x\r\n%s\r\n"
"0\r\n\r\n",
test_data[0],
test_data[1],
test_data[2],
test_data[3],
test_data[4],
test_data[5],
test_data[6],
strlen(test_data + 7),
test_data + 7)) != NULL);
h = mg_get_header(conn, "Content-Length");
ASSERT(h == NULL);
ASSERT(conn->request_info.content_length == -1);
ASSERT((p1 = read_conn(conn, &len1)) != NULL);
ASSERT(len1 == (int)strlen(test_data));
ASSERT(memcmp(p1, test_data, len1) == 0);
mg_free(p1);
mg_close_connection(conn);
/* Test non existent */
ASSERT((conn = mg_download("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"%s",
"GET /non_exist HTTP/1.1\r\n\r\n")) != NULL);
ASSERT(strcmp(conn->request_info.request_method, "HTTP/1.1") == 0);
ASSERT(strcmp(conn->request_info.uri, "404") == 0);
ASSERT(strcmp(conn->request_info.http_version, "Not Found") == 0);
mg_close_connection(conn);
if (use_ssl) {
#ifndef NO_SSL
/* Test SSL redirect */
ASSERT((conn = mg_download("localhost",
atoi(HTTP_REDIRECT_PORT),
0,
ebuf,
sizeof(ebuf),
"%s",
"GET /data/4711 HTTP/1.1\r\n\r\n")) != NULL);
ASSERT(strcmp(conn->request_info.uri, "302") == 0);
h = mg_get_header(conn, "Location");
ASSERT(h != NULL);
ASSERT(strcmp(h, "https://127.0.0.1:" HTTPS_PORT "/data/4711") == 0);
mg_close_connection(conn);
#endif
}
/* Test new API */
ebuf[0] = 1;
conn = mg_connect_client("localhost", port, use_ssl, ebuf, sizeof(ebuf));
ASSERT(conn != NULL);
ASSERT(ebuf[0] == 0);
ri = mg_get_request_info(conn);
ASSERT(ri->content_length == 0);
i = mg_get_response(conn, ebuf, sizeof(ebuf), 1000);
ASSERT(ebuf[0] != 0);
ri = mg_get_request_info(conn);
ASSERT(ri->content_length == -1);
mg_printf(conn, "GET /index.html HTTP/1.1\r\n");
mg_printf(conn, "Host: www.example.com\r\n");
mg_printf(conn, "\r\n");
i = mg_get_response(conn, ebuf, sizeof(ebuf), 1000);
ASSERT(ebuf[0] == 0);
ri = mg_get_request_info(conn);
ASSERT(ri->content_length > 0);
mg_read(conn, ebuf, sizeof(ebuf));
ASSERT(!strncmp(ebuf, "Error 404", 9));
mg_close_connection(conn);
/* Stop the test server */
ut_mg_stop(ctx);
}
static int websocket_data_handler(const struct mg_connection *conn,
int flags,
char *data,
size_t data_len,
void *cbdata)
{
(void)conn;
(void)flags;
(void)data;
(void)data_len;
(void)cbdata;
return 1;
}
static void test_mg_websocket_client_connect(int use_ssl)
{
struct mg_connection *conn;
char ebuf[100];
int port;
struct mg_context *ctx;
if (use_ssl)
port = atoi(HTTPS_PORT);
else
port = atoi(HTTP_PORT);
ASSERT((ctx = mg_start(&CALLBACKS, NULL, OPTIONS)) != NULL);
/* Try to connect to our own server */
/* Invalid port test */
conn = mg_connect_websocket_client("localhost",
0,
use_ssl,
ebuf,
sizeof(ebuf),
"/",
"http://localhost",
websocket_data_handler,
NULL,
NULL);
ASSERT(conn == NULL);
/* Should succeed, the default civetweb server should complete the handshake
*/
conn = mg_connect_websocket_client("localhost",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"/",
"http://localhost",
websocket_data_handler,
NULL,
NULL);
ASSERT(conn != NULL);
/* Try an external server test */
port = 80;
if (use_ssl) {
port = 443;
}
/* Not a websocket server path */
conn = mg_connect_websocket_client("websocket.org",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"/",
"http://websocket.org",
websocket_data_handler,
NULL,
NULL);
ASSERT(conn == NULL);
/* Invalid port test */
conn = mg_connect_websocket_client("echo.websocket.org",
0,
use_ssl,
ebuf,
sizeof(ebuf),
"/",
"http://websocket.org",
websocket_data_handler,
NULL,
NULL);
ASSERT(conn == NULL);
/* Should succeed, echo.websocket.org echos the data back */
conn = mg_connect_websocket_client("echo.websocket.org",
port,
use_ssl,
ebuf,
sizeof(ebuf),
"/",
"http://websocket.org",
websocket_data_handler,
NULL,
NULL);
ASSERT(conn != NULL);
ut_mg_stop(ctx);
}
static int alloc_printf(char **buf, size_t size, char *fmt, ...)
{
va_list ap;
int ret = 0;
va_start(ap, fmt);
ret = alloc_vprintf(buf, size, fmt, ap);
va_end(ap);
return ret;
}
static void test_mg_upload(void)
{
static const char *boundary = "OOO___MY_BOUNDARY___OOO";
struct mg_context *ctx;
#if 0
struct mg_connection *conn;
char ebuf[100], buf[20], *file2_data;
int file2_len;
#endif
char *file_data, *post_data;
int file_len, post_data_len;
ASSERT((ctx = mg_start(&CALLBACKS, NULL, OPTIONS)) != NULL);
/* Upload one file */
ASSERT((file_data = read_file("unit_test.c", &file_len)) != NULL);
post_data = NULL;
post_data_len = alloc_printf(&post_data,
0,
"--%s\r\n"
"Content-Disposition: form-data; "
"name=\"file\"; "
"filename=\"%s\"\r\n\r\n"
"%.*s\r\n"
"--%s--\r\n",
boundary,
upload_filename,
file_len,
file_data,
boundary);
ASSERT(post_data_len > 0);
#if 0 /* TODO (bel): ... */
ASSERT((conn = mg_download("localhost", atoi(HTTPS_PORT), 1,
ebuf, sizeof(ebuf),
"POST /upload?1 HTTP/1.1\r\n"
"Content-Length: %d\r\n"
"Content-Type: multipart/form-data; "
"boundary=%s\r\n\r\n"
"%.*s", post_data_len, boundary,
post_data_len, post_data)) != NULL);
mg_free(file_data), mg_free(post_data);
ASSERT(mg_read(conn, buf, sizeof(buf)) == (int) strlen(upload_ok_message));
ASSERT(memcmp(buf, upload_ok_message, strlen(upload_ok_message)) == 0);
mg_close_connection(conn);
/* Upload two files */
ASSERT((file_data = read_file("include/civetweb.h", &file_len)) != NULL);
ASSERT((file2_data = read_file("README.md", &file2_len)) != NULL);
post_data = NULL;
post_data_len = alloc_printf(&post_data, 0,
/* First file */
"--%s\r\n"
"Content-Disposition: form-data; "
"name=\"file\"; "
"filename=\"%s\"\r\n\r\n"
"%.*s\r\n"
/* Second file */
"--%s\r\n"
"Content-Disposition: form-data; "
"name=\"file\"; "
"filename=\"%s\"\r\n\r\n"
"%.*s\r\n"
/* Final boundary */
"--%s--\r\n",
boundary, upload_filename,
file_len, file_data,
boundary, upload_filename2,
file2_len, file2_data,
boundary);
ASSERT(post_data_len > 0);
ASSERT((conn = mg_download("localhost", atoi(HTTPS_PORT), 1,
ebuf, sizeof(ebuf),
"POST /upload?2 HTTP/1.1\r\n"
"Content-Length: %d\r\n"
"Content-Type: multipart/form-data; "
"boundary=%s\r\n\r\n"
"%.*s", post_data_len, boundary,
post_data_len, post_data)) != NULL);
mg_free(file_data), mg_free(file2_data), mg_free(post_data);
ASSERT(mg_read(conn, buf, sizeof(buf)) == (int) strlen(upload_ok_message));
ASSERT(memcmp(buf, upload_ok_message, strlen(upload_ok_message)) == 0);
mg_close_connection(conn);
#endif
ut_mg_stop(ctx);
}
static void 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]));
}
}
static void test_mg_get_var(void)
{
static const char *post[] = {"a=1&&b=2&d&=&c=3%20&e=",
"q=&st=2012%2F11%2F13+17%3A05&et=&team_id=",
NULL};
char buf[20];
ASSERT(mg_get_var(post[0], strlen(post[0]), "a", buf, sizeof(buf)) == 1);
ASSERT(buf[0] == '1' && buf[1] == '\0');
ASSERT(mg_get_var(post[0], strlen(post[0]), "b", buf, sizeof(buf)) == 1);
ASSERT(buf[0] == '2' && buf[1] == '\0');
ASSERT(mg_get_var(post[0], strlen(post[0]), "c", buf, sizeof(buf)) == 2);
ASSERT(buf[0] == '3' && buf[1] == ' ' && buf[2] == '\0');
ASSERT(mg_get_var(post[0], strlen(post[0]), "e", buf, sizeof(buf)) == 0);
ASSERT(buf[0] == '\0');
ASSERT(mg_get_var(post[0], strlen(post[0]), "d", buf, sizeof(buf)) == -1);
ASSERT(mg_get_var(post[0], strlen(post[0]), "c", buf, 2) == -2);
ASSERT(mg_get_var(post[0], strlen(post[0]), "x", NULL, 10) == -2);
ASSERT(mg_get_var(post[0], strlen(post[0]), "x", buf, 0) == -2);
ASSERT(mg_get_var(post[1], strlen(post[1]), "st", buf, 16) == -2);
ASSERT(mg_get_var(post[1], strlen(post[1]), "st", buf, 17) == 16);
}
static void test_set_throttle(void)
{
ASSERT(set_throttle(NULL, 0x0a000001, "/") == 0);
ASSERT(set_throttle("10.0.0.0/8=20", 0x0a000001, "/") == 20);
ASSERT(set_throttle("10.0.0.0/8=0.5k", 0x0a000001, "/") == 512);
ASSERT(set_throttle("10.0.0.0/8=17m", 0x0a000001, "/") == 1048576 * 17);
ASSERT(set_throttle("10.0.0.0/8=1x", 0x0a000001, "/") == 0);
ASSERT(set_throttle("10.0.0.0/8=5,0.0.0.0/0=10", 0x0a000001, "/") == 10);
ASSERT(set_throttle("10.0.0.0/8=5,/foo/**=7", 0x0a000001, "/index") == 5);
ASSERT(set_throttle("10.0.0.0/8=5,/foo/**=7", 0x0a000001, "/foo/x") == 7);
ASSERT(set_throttle("10.0.0.0/8=5,/foo/**=7", 0x0b000001, "/foxo/x") == 0);
ASSERT(set_throttle("10.0.0.0/8=5,*=1", 0x0b000001, "/foxo/x") == 1);
}
static void 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));
}
}
#if defined(USE_LUA)
static void check_lua_expr(lua_State *L, const char *expr, const char *value)
{
const char *v, *var_name = "myVar";
char buf[100];
snprintf(buf, sizeof(buf), "%s = %s", var_name, expr);
(void)luaL_dostring(L, buf);
lua_getglobal(L, var_name);
v = lua_tostring(L, -1);
ASSERT((value == NULL && v == NULL) ||
(value != NULL && v != NULL && !strcmp(value, v)));
}
static void test_lua(void)
{
static struct mg_connection conn;
static struct mg_context ctx;
char http_request[] = "POST /foo/bar HTTP/1.1\r\n"
"Content-Length: 12\r\n"
"Connection: close\r\n\r\nhello world!";
lua_State *L = luaL_newstate();
conn.ctx = &ctx;
conn.buf = http_request;
conn.buf_size = conn.data_len = strlen(http_request);
conn.request_len =
parse_http_message(conn.buf, conn.data_len, &conn.request_info);
conn.content_len = conn.data_len - conn.request_len;
prepare_lua_environment(&conn, L, "unit_test", LUA_ENV_TYPE_PLAIN_LUA_PAGE);
ASSERT(lua_gettop(L) == 4);
check_lua_expr(L, "'hi'", "hi");
check_lua_expr(L, "mg.request_info.request_method", "POST");
check_lua_expr(L, "mg.request_info.uri", "/foo/bar");
check_lua_expr(L, "mg.request_info.num_headers", "2");
check_lua_expr(L, "mg.request_info.remote_ip", "0");
check_lua_expr(L, "mg.request_info.http_headers['Content-Length']", "12");
check_lua_expr(L, "mg.request_info.http_headers['Connection']", "close");
(void)luaL_dostring(L, "post = mg.read()");
check_lua_expr(L, "# post", "12");
check_lua_expr(L, "post", "hello world!");
lua_close(L);
}
#endif
static void test_mg_stat(void)
{
static struct mg_context ctx;
struct file file = STRUCT_FILE_INITIALIZER;
ASSERT(!mg_stat(fc(&ctx), " does not exist ", &file));
}
static void test_skip_quoted(void)
{
char x[] = "a=1, b=2, c='hi \' there', d='here\\, there'", *s = x, *p;
p = skip_quoted(&s, ", ", ", ", 0);
ASSERT(p != NULL && !strcmp(p, "a=1"));
p = skip_quoted(&s, ", ", ", ", 0);
ASSERT(p != NULL && !strcmp(p, "b=2"));
p = skip_quoted(&s, ",", " ", 0);
ASSERT(p != NULL && !strcmp(p, "c='hi \' there'"));
p = skip_quoted(&s, ",", " ", '\\');
ASSERT(p != NULL && !strcmp(p, "d='here, there'"));
ASSERT(*s == 0);
}
static void test_alloc_vprintf(void)
{
char buf[MG_BUF_LEN], *p = buf;
ASSERT(alloc_printf(&p, sizeof(buf), "%s", "hi") == 2);
ASSERT(p == buf);
ASSERT(alloc_printf(&p, sizeof(buf), "%s", "") == 0);
ASSERT(alloc_printf(&p, sizeof(buf), "") == 0);
/* Pass small buffer, make sure alloc_printf allocates */
ASSERT(alloc_printf(&p, 1, "%s", "hello") == 5);
ASSERT(p != buf);
mg_free(p);
}
static void test_request_replies(void)
{
char ebuf[100];
int i;
struct mg_connection *conn;
struct mg_context *ctx;
static struct {
const char *request, *reply_regex;
} tests[] = {
{"GET hello.txt HTTP/1.0\r\nRange: bytes=3-5\r\n\r\n",
"^HTTP/1.1 206 Partial Content"},
{NULL, NULL},
};
ASSERT((ctx = mg_start(&CALLBACKS, NULL, OPTIONS)) != NULL);
for (i = 0; tests[i].request != NULL; i++) {
ASSERT((conn = mg_download("localhost",
atoi(HTTP_PORT),
0,
ebuf,
sizeof(ebuf),
"%s",
tests[i].request)) != NULL);
mg_close_connection(conn);
}
ut_mg_stop(ctx);
#ifndef NO_SSL
ASSERT((ctx = mg_start(&CALLBACKS, NULL, OPTIONS)) != NULL);
for (i = 0; tests[i].request != NULL; i++) {
ASSERT((conn = mg_download("localhost",
atoi(HTTPS_PORT),
1,
ebuf,
sizeof(ebuf),
"%s",
tests[i].request)) != NULL);
mg_close_connection(conn);
}
ut_mg_stop(ctx);
#endif
}
static int request_test_handler(struct mg_connection *conn, void *cbdata)
{
int i;
char chunk_data[32];
ASSERT(cbdata == (void *)7);
strcpy(chunk_data, "123456789A123456789B123456789C");
mg_printf(conn,
"HTTP/1.1 200 OK\r\n"
"Transfer-Encoding: chunked\r\n"
"Content-Type: text/plain\r\n\r\n");
for (i = 0; i < 20; i++) {
mg_printf(conn, "%x\r\n", i);
mg_write(conn, chunk_data, i);
mg_printf(conn, "\r\n");
}
mg_printf(conn, "0\r\n\r\n");
return 1;
}
static void test_request_handlers(void)
{
char ebuf[100];
struct mg_context *ctx;
struct mg_connection *conn;
char uri[64];
int i;
const char *request = "GET /U7 HTTP/1.0\r\n\r\n";
ctx = mg_start(NULL, NULL, OPTIONS);
ASSERT(ctx != NULL);
for (i = 0; i < 1000; i++) {
sprintf(uri, "/U%u", i);
mg_set_request_handler(ctx, uri, request_test_handler, NULL);
}
for (i = 500; i < 800; i++) {
sprintf(uri, "/U%u", i);
mg_set_request_handler(ctx, uri, NULL, (void *)1);
}
for (i = 600; i >= 0; i--) {
sprintf(uri, "/U%u", i);
mg_set_request_handler(ctx, uri, NULL, (void *)2);
}
for (i = 750; i <= 1000; i++) {
sprintf(uri, "/U%u", i);
mg_set_request_handler(ctx, uri, NULL, (void *)3);
}
for (i = 5; i < 9; i++) {
sprintf(uri, "/U%u", i);
mg_set_request_handler(ctx, uri, request_test_handler, (void *)i);
}
conn = mg_download(
"localhost", atoi(HTTP_PORT), 0, ebuf, sizeof(ebuf), "%s", request);
ASSERT(conn != NULL);
mg_sleep(1000);
mg_close_connection(conn);
ut_mg_stop(ctx);
}
static int api_callback(struct mg_connection *conn)
{
const struct mg_request_info *ri = mg_get_request_info(conn);
char post_data[100] = "";
ASSERT(ri->user_data == (void *)123);
ASSERT(ri->num_headers == 2);
ASSERT(strcmp(mg_get_header(conn, "host"), "blah.com") == 0);
ASSERT(mg_read(conn, post_data, sizeof(post_data)) == 3);
ASSERT(memcmp(post_data, "b=1", 3) == 0);
ASSERT(ri->query_string != NULL);
ASSERT(ri->remote_addr[0] != 0);
ASSERT(ri->remote_port > 0);
ASSERT(strcmp(ri->http_version, "1.0") == 0);
mg_printf(conn, "HTTP/1.0 200 OK\r\n\r\n");
return 1;
}
static void test_api_calls(void)
{
char ebuf[100];
struct mg_callbacks callbacks;
struct mg_connection *conn;
struct mg_context *ctx;
static const char *request =
"POST /?a=%20&b=&c=xx HTTP/1.0\r\n"
"Host: blah.com\n" /* More spaces before */
"content-length: 3\r\n" /* Lower case header name */
"\r\nb=123456"; /* Content size > content-length, test for mg_read() */
memset(&callbacks, 0, sizeof(callbacks));
callbacks.begin_request = api_callback;
ASSERT((ctx = mg_start(&callbacks, (void *)123, OPTIONS)) != NULL);
ASSERT((conn = mg_download("localhost",
atoi(HTTP_PORT),
0,
ebuf,
sizeof(ebuf),
"%s",
request)) != NULL);
mg_close_connection(conn);
ut_mg_stop(ctx);
}
static void 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);
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);
}
static void test_mg_strcasestr(void)
{
static const char *big1 = "abcdef";
ASSERT(mg_strcasestr("Y", "X") == NULL);
ASSERT(mg_strcasestr("Y", "y") != NULL);
ASSERT(mg_strcasestr(big1, "X") == NULL);
ASSERT(mg_strcasestr(big1, "CD") == big1 + 2);
ASSERT(mg_strcasestr("aa", "AAB") == NULL);
}
static void test_mg_get_cookie(void)
{
char buf[20];
ASSERT(mg_get_cookie("", "foo", NULL, sizeof(buf)) == -2);
ASSERT(mg_get_cookie("", "foo", buf, 0) == -2);
ASSERT(mg_get_cookie("", "foo", buf, sizeof(buf)) == -1);
ASSERT(mg_get_cookie("", NULL, buf, sizeof(buf)) == -1);
ASSERT(mg_get_cookie("a=1; b=2; c; d", "a", buf, sizeof(buf)) == 1);
ASSERT(strcmp(buf, "1") == 0);
ASSERT(mg_get_cookie("a=1; b=2; c; d", "b", buf, sizeof(buf)) == 1);
ASSERT(strcmp(buf, "2") == 0);
ASSERT(mg_get_cookie("a=1; b=123", "b", buf, sizeof(buf)) == 3);
ASSERT(strcmp(buf, "123") == 0);
ASSERT(mg_get_cookie("a=1; b=2; c; d", "c", buf, sizeof(buf)) == -1);
}
static void test_strtoll(void)
{
ASSERT(strtoll("0", NULL, 10) == 0);
ASSERT(strtoll("123", NULL, 10) == 123);
ASSERT(strtoll("-34", NULL, 10) == -34);
ASSERT(strtoll("3566626116", NULL, 10) == 3566626116);
}
static void test_parse_port_string(void)
{
static const char *valid[] = {
"0",
"1",
"1s",
"1r",
"1.2.3.4:1",
"1.2.3.4:1s",
"1.2.3.4:1r",
#if defined(USE_IPV6)
"[::1]:123",
"[3ffe:2a00:100:7031::1]:900",
#endif
NULL
};
static const char *invalid[] = {
"99999", "1k", "1.2.3", "1.2.3.4:", "1.2.3.4:2p", NULL};
struct socket so;
struct vec vec;
int i;
for (i = 0; valid[i] != NULL; i++) {
vec.ptr = valid[i];
vec.len = strlen(vec.ptr);
ASSERT(parse_port_string(&vec, &so) != 0);
}
for (i = 0; invalid[i] != NULL; i++) {
vec.ptr = invalid[i];
vec.len = strlen(vec.ptr);
ASSERT(parse_port_string(&vec, &so) == 0);
}
}
static void test_md5(void)
{
md5_state_t md5_state;
unsigned char md5_val[16 + 1];
char md5_str[32 + 1];
const char *test_str = "The quick brown fox jumps over the lazy dog";
md5_val[16] = 0;
md5_init(&md5_state);
md5_finish(&md5_state, md5_val);
ASSERT(strcmp((const char *)md5_val,
"\xd4\x1d\x8c\xd9\x8f\x00\xb2\x04\xe9"
"\x80\x09\x98\xec\xf8\x42\x7e") == 0);
sprintf(md5_str,
"%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x",
md5_val[0],
md5_val[1],
md5_val[2],
md5_val[3],
md5_val[4],
md5_val[5],
md5_val[6],
md5_val[7],
md5_val[8],
md5_val[9],
md5_val[10],
md5_val[11],
md5_val[12],
md5_val[13],
md5_val[14],
md5_val[15]);
ASSERT(strcmp(md5_str, "d41d8cd98f00b204e9800998ecf8427e") == 0);
mg_md5(md5_str, "", NULL);
ASSERT(strcmp(md5_str, "d41d8cd98f00b204e9800998ecf8427e") == 0);
md5_init(&md5_state);
md5_append(&md5_state, (const md5_byte_t *)test_str, strlen(test_str));
md5_finish(&md5_state, md5_val);
sprintf(md5_str,
"%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x",
md5_val[0],
md5_val[1],
md5_val[2],
md5_val[3],
md5_val[4],
md5_val[5],
md5_val[6],
md5_val[7],
md5_val[8],
md5_val[9],
md5_val[10],
md5_val[11],
md5_val[12],
md5_val[13],
md5_val[14],
md5_val[15]);
ASSERT(strcmp(md5_str, "9e107d9d372bb6826bd81d3542a419d6") == 0);
mg_md5(md5_str, test_str, NULL);
ASSERT(strcmp(md5_str, "9e107d9d372bb6826bd81d3542a419d6") == 0);
mg_md5(md5_str,
"The",
" ",
"quick brown fox",
"",
" jumps ",
"over the lazy dog",
"",
"",
NULL);
ASSERT(strcmp(md5_str, "9e107d9d372bb6826bd81d3542a419d6") == 0);
mg_md5(md5_str, "civetweb", NULL);
ASSERT(strcmp(md5_str, "95c098bd85b619b24a83d9cea5e8ba54") == 0);
}
int __cdecl main(void)
{
char buffer[512];
FILE *f;
struct mg_context *ctx;
int i;
/* print headline */
printf("Civetweb %s unit test\n", mg_version());
#if defined(_WIN32)
GetCurrentDirectoryA(sizeof(buffer), buffer);
#else
getcwd(buffer, sizeof(buffer));
#endif
printf("Test directory is \"%s\"\n",
buffer); /* should be the "test" directory */
f = fopen("hello.txt", "r");
if (f) {
fclose(f);
} else {
printf("Error: Test directory does not contain hello.txt\n");
}
f = fopen("unit_test.c", "r");
if (f) {
fclose(f);
} else {
printf("Error: Test directory does not contain unit_test.c\n");
}
/* test local functions */
test_parse_port_string();
test_mg_strcasestr();
test_alloc_vprintf();
test_base64_encode();
test_match_prefix();
test_remove_double_dots();
test_should_keep_alive();
test_parse_http_message();
test_mg_get_var();
test_set_throttle();
test_next_option();
test_mg_stat();
test_skip_quoted();
test_url_decode();
test_mg_get_cookie();
test_strtoll();
test_md5();
/* start stop server */
ctx = mg_start(NULL, NULL, OPTIONS);
REQUIRE(ctx != NULL);
mg_sleep(1000);
ut_mg_stop(ctx);
/* create test data */
fetch_data = (char *)mg_malloc(fetch_data_size);
for (i = 0; i < fetch_data_size; i++) {
fetch_data[i] = 'a' + i % 10;
}
/* tests with network access */
init_CALLBACKS();
test_mg_download(0);
#ifndef NO_SSL
test_mg_download(1);
#endif
test_mg_websocket_client_connect(0);
#ifndef NO_SSL
test_mg_websocket_client_connect(1);
#endif
test_mg_upload();
test_request_replies();
test_api_calls();
test_request_handlers();
#if defined(USE_LUA)
test_lua();
#endif
/* test completed */
mg_free(fetch_data);
#ifdef MEMORY_DEBUGGING
{
extern unsigned long mg_memory_debug_blockCount;
extern unsigned long mg_memory_debug_totalMemUsed;
printf("MEMORY DEBUGGING: %u %u\n",
mg_memory_debug_blockCount,
mg_memory_debug_totalMemUsed);
}
#endif
printf("TOTAL TESTS: %d, FAILED: %d\n", s_total_tests, s_failed_tests);
return s_failed_tests == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
}
View Git Blame Copy Permalink