|
|
@@ -1,1549 +0,0 @@
|
|
|
-/* 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.
|
|
|
- */
|
|
|
-
|
|
|
-#if defined(_MSC_VER)
|
|
|
-#pragma message("Warning: " __FILE__ " is obsolete. See note below.")
|
|
|
-#else
|
|
|
-#pragma message __FILE__ ": This file is obsolete. See note below."
|
|
|
-#endif
|
|
|
-
|
|
|
-/* Note: The unit_test.c file is mostly obsolete, since the current unit
|
|
|
- * tests are performed by the CMake build framework based on the
|
|
|
- * CMakeList (https://github.com/civetweb/civetweb/blob/master/test/CMakeLists.txt).
|
|
|
- * The tests do no longer use unit_test.c but public_func.c, public_server.c,
|
|
|
- * private.c, private_exe.c and others.
|
|
|
- * This file is left here for reference and will be removed in the future.
|
|
|
- * It is no longer actively maintained.
|
|
|
- */
|
|
|
-
|
|
|
-
|
|
|
-/* 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"
|
|
|
-
|
|
|
-void check_func(int condition, const char *cond_txt, unsigned line);
|
|
|
-
|
|
|
-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") == 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;
|
|
|
-}
|
|
|
-
|
|
|
-#if defined(MG_LEGACY_INTERFACE)
|
|
|
-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);
|
|
|
-}
|
|
|
-#endif
|
|
|
-
|
|
|
-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 defined(MG_LEGACY_INTERFACE)
|
|
|
- if (!strcmp(ri->uri, "/upload")) {
|
|
|
- ASSERT(ri->query_string != NULL);
|
|
|
- ASSERT(mg_upload(conn, ".") == atoi(ri->query_string));
|
|
|
- }
|
|
|
-#endif
|
|
|
-
|
|
|
- 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;
|
|
|
-#if defined(MG_LEGACY_INTERFACE)
|
|
|
- CALLBACKS.upload = upload_cb;
|
|
|
-#endif
|
|
|
-};
|
|
|
-
|
|
|
-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",
|
|
|
- (uint32_t)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],
|
|
|
- (uint32_t)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",
|
|
|
- (mg_websocket_data_handler)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",
|
|
|
- (mg_websocket_data_handler)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",
|
|
|
- (mg_websocket_data_handler)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",
|
|
|
- (mg_websocket_data_handler)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",
|
|
|
- (mg_websocket_data_handler)websocket_data_handler,
|
|
|
- NULL,
|
|
|
- NULL);
|
|
|
- ASSERT(conn != NULL);
|
|
|
-
|
|
|
- ut_mg_stop(ctx);
|
|
|
-}
|
|
|
-
|
|
|
-static int alloc_printf(char **out_buf, char *buf, size_t size, char *fmt, ...)
|
|
|
-{
|
|
|
- va_list ap;
|
|
|
- int ret = 0;
|
|
|
- va_start(ap, fmt);
|
|
|
- ret = alloc_vprintf(out_buf, buf, size, fmt, ap);
|
|
|
- va_end(ap);
|
|
|
- return ret;
|
|
|
-}
|
|
|
-
|
|
|
-#if defined(MG_LEGACY_INTERFACE)
|
|
|
-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,
|
|
|
- NULL,
|
|
|
- 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);
|
|
|
-}
|
|
|
-#endif
|
|
|
-
|
|
|
-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];
|
|
|
-
|
|
|
- mg_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(&ctx, &conn, NULL, 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, buf, sizeof(buf), "%s", "hi") == 2);
|
|
|
- ASSERT(p == buf);
|
|
|
- ASSERT(alloc_printf(&p, buf, sizeof(buf), "%s", "") == 0);
|
|
|
- ASSERT(alloc_printf(&p, buf, sizeof(buf), "") == 0);
|
|
|
-
|
|
|
- /* Pass small buffer, make sure alloc_printf allocates */
|
|
|
- ASSERT(alloc_printf(&p, buf, 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 *)(intptr_t)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 ipv;
|
|
|
- int i;
|
|
|
-
|
|
|
- for (i = 0; valid[i] != NULL; i++) {
|
|
|
- vec.ptr = valid[i];
|
|
|
- vec.len = strlen(vec.ptr);
|
|
|
- ASSERT(parse_port_string(&vec, &so, &ipv) != 0);
|
|
|
- }
|
|
|
-
|
|
|
- for (i = 0; invalid[i] != NULL; i++) {
|
|
|
- vec.ptr = invalid[i];
|
|
|
- vec.len = strlen(vec.ptr);
|
|
|
- ASSERT(parse_port_string(&vec, &so, &ipv) == 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
|
|
|
-
|
|
|
-#if defined(MG_LEGACY_INTERFACE)
|
|
|
- test_mg_upload();
|
|
|
-#endif
|
|
|
- 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: %lu %lu\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;
|
|
|
-}
|
kakwa
