http://git-wip-us.apache.org/repos/asf/nifi-minifi-cpp/blob/c1101f46/thirdparty/civetweb-1.9.1/test/public_server.c ---------------------------------------------------------------------- diff --git a/thirdparty/civetweb-1.9.1/test/public_server.c b/thirdparty/civetweb-1.9.1/test/public_server.c deleted file mode 100644 index b65f122..0000000 --- a/thirdparty/civetweb-1.9.1/test/public_server.c +++ /dev/null @@ -1,3733 +0,0 @@ -/* Copyright (c) 2015-2017 the Civetweb developers - * - * 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. - */ - -#ifdef _MSC_VER -#ifndef _CRT_SECURE_NO_WARNINGS -#define _CRT_SECURE_NO_WARNINGS -#endif -#endif - -#include <stdlib.h> -#include <stdio.h> -#include <stdarg.h> -#include <string.h> -#include <time.h> -#include <sys/types.h> -#include <sys/stat.h> - -#include "public_server.h" -#include <civetweb.h> - -#if defined(_WIN32) -#include <windows.h> -#define test_sleep(x) (Sleep((x)*1000)) -#else -#include <unistd.h> -#define test_sleep(x) (sleep(x)) -#endif - -#define SLEEP_BEFORE_MG_START (1) -#define SLEEP_AFTER_MG_START (3) -#define SLEEP_BEFORE_MG_STOP (1) -#define SLEEP_AFTER_MG_STOP (5) - -/* This unit test file uses the excellent Check unit testing library. - * The API documentation is available here: - * http://check.sourceforge.net/doc/check_html/index.html - */ - -#if defined(__MINGW32__) || defined(__GNUC__) -/* Return codes of the tested functions are evaluated. Checking all other - * functions, used only to prepare the test environment seems redundant. - * If they fail, the test fails anyway. */ -#pragma GCC diagnostic ignored "-Wunused-result" -#endif - -static const char * -locate_path(const char *a_path) -{ - static char r_path[256]; - -#ifdef _WIN32 -#ifdef LOCAL_TEST - sprintf(r_path, "%s\\", a_path); -#else - /* Appveyor */ - sprintf(r_path, "..\\..\\..\\%s\\", a_path); -/* TODO: the different paths - * used in the different test - * system is an unsolved - * problem. */ -#endif -#else -#ifdef LOCAL_TEST - sprintf(r_path, "%s/", a_path); -#else - /* Travis */ - sprintf(r_path, - "../../%s/", - a_path); // TODO: fix path in CI test environment -#endif -#endif - - return r_path; -} - - -#define locate_resources() locate_path("resources") -#define locate_test_exes() locate_path("output") - - -static const char * -locate_ssl_cert(void) -{ - static char cert_path[256]; - const char *res = locate_resources(); - size_t l; - - ck_assert(res != NULL); - l = strlen(res); - ck_assert_uint_gt(l, 0); - ck_assert_uint_lt(l, 100); /* assume there is enough space left in our - typical 255 character string buffers */ - - strcpy(cert_path, res); - strcat(cert_path, "ssl_cert.pem"); - return cert_path; -} - - -static int -wait_not_null(void *volatile *data) -{ - int i; - for (i = 0; i < 100; i++) { - mark_point(); - test_sleep(1); - - if (*data != NULL) { - mark_point(); - return 1; - } - } - -#if defined(__MINGW32__) || defined(__GNUC__) -#pragma GCC diagnostic push -#pragma GCC diagnostic ignored "-Wunreachable-code" -#endif -#ifdef __clang__ -#pragma clang diagnostic push -#pragma clang diagnostic ignored "-Wunreachable-code" -#endif - - ck_abort_msg("wait_not_null failed"); - - return 0; - -#ifdef __clang__ -#pragma clang diagnostic pop -#endif -#if defined(__MINGW32__) || defined(__GNUC__) -#pragma GCC diagnostic pop -#endif -} - - -START_TEST(test_the_test_environment) -{ - char wd[300]; - char buf[500]; - FILE *f; - struct stat st; - int ret; - const char *ssl_cert = locate_ssl_cert(); - - memset(wd, 0, sizeof(wd)); - memset(buf, 0, sizeof(buf)); - -/* Get the current working directory */ -#ifdef _WIN32 - (void)GetCurrentDirectoryA(sizeof(wd), wd); - wd[sizeof(wd) - 1] = 0; -#else - (void)getcwd(wd, sizeof(wd)); - wd[sizeof(wd) - 1] = 0; -#endif - -/* Check the pem file */ -#ifdef _WIN32 - strcpy(buf, wd); - strcat(buf, "\\"); - strcat(buf, ssl_cert); - f = fopen(buf, "rb"); -#else - strcpy(buf, wd); - strcat(buf, "/"); - strcat(buf, ssl_cert); - f = fopen(buf, "r"); -#endif - - if (f) { - fclose(f); - } else { - fprintf(stderr, "%s not found", buf); - } - -/* Check the test dir */ -#ifdef _WIN32 - strcpy(buf, wd); - strcat(buf, "\\test"); -#else - strcpy(buf, wd); - strcat(buf, "/test"); -#endif - - memset(&st, 0, sizeof(st)); - ret = stat(buf, &st); - - if (ret) { - fprintf(stderr, "%s not found", buf); - } - - -#ifdef _WIN32 -/* Try to copy the files required for AppVeyor */ -#if defined(_WIN64) || defined(__MINGW64__) - (void)system("cmd /c copy C:\\OpenSSL-Win64\\libeay32.dll libeay32.dll"); - (void)system("cmd /c copy C:\\OpenSSL-Win64\\libssl32.dll libssl32.dll"); - (void)system("cmd /c copy C:\\OpenSSL-Win64\\ssleay32.dll ssleay32.dll"); - (void)system("cmd /c copy C:\\OpenSSL-Win64\\libeay32.dll libeay64.dll"); - (void)system("cmd /c copy C:\\OpenSSL-Win64\\libssl32.dll libssl64.dll"); - (void)system("cmd /c copy C:\\OpenSSL-Win64\\ssleay32.dll ssleay64.dll"); -#else - (void)system("cmd /c copy C:\\OpenSSL-Win32\\libeay32.dll libeay32.dll"); - (void)system("cmd /c copy C:\\OpenSSL-Win32\\libssl32.dll libssl32.dll"); - (void)system("cmd /c copy C:\\OpenSSL-Win32\\ssleay32.dll ssleay32.dll"); -#endif -#endif -} -END_TEST - - -static void *threading_data; - -static void * -test_thread_func_t(void *param) -{ - ck_assert_ptr_eq(param, &threading_data); - ck_assert_ptr_eq(threading_data, NULL); - threading_data = &threading_data; - return NULL; -} - - -START_TEST(test_threading) -{ - int ok; - - threading_data = NULL; - - ok = mg_start_thread(test_thread_func_t, &threading_data); - ck_assert_int_eq(ok, 0); - - wait_not_null(&threading_data); - ck_assert_ptr_eq(threading_data, &threading_data); -} -END_TEST - - -static int -log_msg_func(const struct mg_connection *conn, const char *message) -{ - struct mg_context *ctx; - char *ud; - - ck_assert(conn != NULL); - ctx = mg_get_context(conn); - ck_assert(ctx != NULL); - ud = (char *)mg_get_user_data(ctx); - - strncpy(ud, message, 255); - ud[255] = 0; - return 1; -} - - -static struct mg_context * -test_mg_start(const struct mg_callbacks *callbacks, - void *user_data, - const char **configuration_options) -{ - struct mg_context *ctx; - - mark_point(); - test_sleep(SLEEP_BEFORE_MG_START); - mark_point(); - ctx = mg_start(callbacks, user_data, configuration_options); - mark_point(); - if (ctx) { - /* Give the server some time to start in the test VM */ - /* Don't need to do this if mg_start failed */ - test_sleep(SLEEP_AFTER_MG_START); - } - mark_point(); - - return ctx; -} - - -static void -test_mg_stop(struct mg_context *ctx) -{ - mark_point(); - test_sleep(SLEEP_BEFORE_MG_STOP); - mark_point(); - mg_stop(ctx); - mark_point(); - test_sleep(SLEEP_AFTER_MG_STOP); - mark_point(); -} - - -START_TEST(test_mg_start_stop_http_server) -{ - struct mg_context *ctx; - const char *OPTIONS[] = { -#if !defined(NO_FILES) - "document_root", - ".", -#endif - "listening_ports", - "8080", - NULL, - }; - size_t ports_cnt; - int ports[16]; - int ssl[16]; - struct mg_callbacks callbacks; - char errmsg[256]; - - struct mg_connection *client_conn; - char client_err[256]; - const struct mg_request_info *client_ri; - int client_res, ret; - struct mg_server_ports portinfo[8]; - - memset(ports, 0, sizeof(ports)); - memset(ssl, 0, sizeof(ssl)); - memset(portinfo, 0, sizeof(portinfo)); - memset(&callbacks, 0, sizeof(callbacks)); - memset(errmsg, 0, sizeof(errmsg)); - - callbacks.log_message = log_msg_func; - - ctx = test_mg_start(&callbacks, (void *)errmsg, OPTIONS); - - ck_assert_str_eq(errmsg, ""); - ck_assert(ctx != NULL); - - ports_cnt = mg_get_ports(ctx, 16, ports, ssl); - ck_assert_uint_eq(ports_cnt, 1); - ck_assert_int_eq(ports[0], 8080); - ck_assert_int_eq(ssl[0], 0); - ck_assert_int_eq(ports[1], 0); - ck_assert_int_eq(ssl[1], 0); - - ret = mg_get_server_ports(ctx, 0, portinfo); - ck_assert_int_lt(ret, 0); - ck_assert_int_eq(portinfo[0].protocol, 0); - ck_assert_int_eq(portinfo[0].port, 0); - ck_assert_int_eq(portinfo[0].is_ssl, 0); - ck_assert_int_eq(portinfo[0].is_redirect, 0); - ck_assert_int_eq(portinfo[1].protocol, 0); - ck_assert_int_eq(portinfo[1].port, 0); - ck_assert_int_eq(portinfo[1].is_ssl, 0); - ck_assert_int_eq(portinfo[1].is_redirect, 0); - - ret = mg_get_server_ports(ctx, 4, portinfo); - ck_assert_int_eq(ret, 1); - ck_assert_int_eq(portinfo[0].protocol, 1); - ck_assert_int_eq(portinfo[0].port, 8080); - ck_assert_int_eq(portinfo[0].is_ssl, 0); - ck_assert_int_eq(portinfo[0].is_redirect, 0); - ck_assert_int_eq(portinfo[1].protocol, 0); - ck_assert_int_eq(portinfo[1].port, 0); - ck_assert_int_eq(portinfo[1].is_ssl, 0); - ck_assert_int_eq(portinfo[1].is_redirect, 0); - - test_sleep(1); - - /* HTTP 1.0 GET request */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET / HTTP/1.0\r\n\r\n"); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - -#if defined(NO_FILES) - ck_assert_str_eq(client_ri->uri, "404"); -#else - ck_assert_str_eq(client_ri->uri, "200"); - /* TODO: ck_assert_str_eq(client_ri->request_method, "HTTP/1.0"); */ - client_res = (int)mg_read(client_conn, client_err, sizeof(client_err)); - ck_assert_int_gt(client_res, 0); - ck_assert_int_le(client_res, sizeof(client_err)); -#endif - mg_close_connection(client_conn); - - test_sleep(1); - - /* HTTP 1.1 GET request */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET / HTTP/1.1\r\n"); - mg_printf(client_conn, "Host: localhost:8080\r\n"); - mg_printf(client_conn, "Connection: close\r\n\r\n"); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - -#if defined(NO_FILES) - ck_assert_str_eq(client_ri->uri, "404"); -#else - ck_assert_str_eq(client_ri->uri, "200"); - /* TODO: ck_assert_str_eq(client_ri->request_method, "HTTP/1.0"); */ - client_res = (int)mg_read(client_conn, client_err, sizeof(client_err)); - ck_assert_int_gt(client_res, 0); - ck_assert_int_le(client_res, sizeof(client_err)); -#endif - mg_close_connection(client_conn); - - test_sleep(1); - - - /* HTTP 1.7 GET request - this HTTP version does not exist */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET / HTTP/1.7\r\n"); - mg_printf(client_conn, "Host: localhost:8080\r\n"); - mg_printf(client_conn, "Connection: close\r\n\r\n"); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - - /* Response must be 505 HTTP Version not supported */ - ck_assert_str_eq(client_ri->uri, "505"); - mg_close_connection(client_conn); - - test_sleep(1); - - - /* HTTP request with multiline header. - * Multiline header are obsolete with RFC 7230 section-3.2.4 - * and must return "400 Bad Request" */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET / HTTP/1.1\r\n"); - mg_printf(client_conn, "Host: localhost:8080\r\n"); - mg_printf(client_conn, "X-Obsolete-Header: something\r\nfor nothing\r\n"); - mg_printf(client_conn, "Connection: close\r\n\r\n"); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - - /* Response must be 400 Bad Request */ - ck_assert_str_eq(client_ri->uri, "400"); - mg_close_connection(client_conn); - - test_sleep(1); - - /* End test */ - test_mg_stop(ctx); -} -END_TEST - - -START_TEST(test_mg_start_stop_https_server) -{ -#ifndef NO_SSL - - struct mg_context *ctx; - - size_t ports_cnt; - int ports[16]; - int ssl[16]; - struct mg_callbacks callbacks; - char errmsg[256]; - - const char *OPTIONS[8]; /* initializer list here is rejected by CI test */ - int opt_idx = 0; - const char *ssl_cert = locate_ssl_cert(); - - struct mg_connection *client_conn; - char client_err[256]; - const struct mg_request_info *client_ri; - int client_res, ret; - struct mg_server_ports portinfo[8]; - - ck_assert(ssl_cert != NULL); - - memset((void *)OPTIONS, 0, sizeof(OPTIONS)); -#if !defined(NO_FILES) - OPTIONS[opt_idx++] = "document_root"; - OPTIONS[opt_idx++] = "."; -#endif - OPTIONS[opt_idx++] = "listening_ports"; - OPTIONS[opt_idx++] = "8080r,8443s"; - OPTIONS[opt_idx++] = "ssl_certificate"; - OPTIONS[opt_idx++] = ssl_cert; - - ck_assert_int_le(opt_idx, (int)(sizeof(OPTIONS) / sizeof(OPTIONS[0]))); - ck_assert(OPTIONS[sizeof(OPTIONS) / sizeof(OPTIONS[0]) - 1] == NULL); - ck_assert(OPTIONS[sizeof(OPTIONS) / sizeof(OPTIONS[0]) - 2] == NULL); - - memset(ports, 0, sizeof(ports)); - memset(ssl, 0, sizeof(ssl)); - memset(portinfo, 0, sizeof(portinfo)); - memset(&callbacks, 0, sizeof(callbacks)); - memset(errmsg, 0, sizeof(errmsg)); - - callbacks.log_message = log_msg_func; - - ctx = test_mg_start(&callbacks, (void *)errmsg, OPTIONS); - - ck_assert_str_eq(errmsg, ""); - ck_assert(ctx != NULL); - - ports_cnt = mg_get_ports(ctx, 16, ports, ssl); - ck_assert_uint_eq(ports_cnt, 2); - ck_assert_int_eq(ports[0], 8080); - ck_assert_int_eq(ssl[0], 0); - ck_assert_int_eq(ports[1], 8443); - ck_assert_int_eq(ssl[1], 1); - ck_assert_int_eq(ports[2], 0); - ck_assert_int_eq(ssl[2], 0); - - - ret = mg_get_server_ports(ctx, 0, portinfo); - ck_assert_int_lt(ret, 0); - ck_assert_int_eq(portinfo[0].protocol, 0); - ck_assert_int_eq(portinfo[0].port, 0); - ck_assert_int_eq(portinfo[0].is_ssl, 0); - ck_assert_int_eq(portinfo[0].is_redirect, 0); - ck_assert_int_eq(portinfo[1].protocol, 0); - ck_assert_int_eq(portinfo[1].port, 0); - ck_assert_int_eq(portinfo[1].is_ssl, 0); - ck_assert_int_eq(portinfo[1].is_redirect, 0); - - ret = mg_get_server_ports(ctx, 4, portinfo); - ck_assert_int_eq(ret, 2); - ck_assert_int_eq(portinfo[0].protocol, 1); - ck_assert_int_eq(portinfo[0].port, 8080); - ck_assert_int_eq(portinfo[0].is_ssl, 0); - ck_assert_int_eq(portinfo[0].is_redirect, 1); - ck_assert_int_eq(portinfo[1].protocol, 1); - ck_assert_int_eq(portinfo[1].port, 8443); - ck_assert_int_eq(portinfo[1].is_ssl, 1); - ck_assert_int_eq(portinfo[1].is_redirect, 0); - ck_assert_int_eq(portinfo[2].protocol, 0); - ck_assert_int_eq(portinfo[2].port, 0); - ck_assert_int_eq(portinfo[2].is_ssl, 0); - ck_assert_int_eq(portinfo[2].is_redirect, 0); - - test_sleep(1); - - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8443, 1, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET / HTTP/1.0\r\n\r\n"); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - -#if defined(NO_FILES) - ck_assert_str_eq(client_ri->uri, "404"); -#else - ck_assert_str_eq(client_ri->uri, "200"); - /* TODO: ck_assert_str_eq(client_ri->request_method, "HTTP/1.0"); */ - client_res = (int)mg_read(client_conn, client_err, sizeof(client_err)); - ck_assert_int_gt(client_res, 0); - ck_assert_int_le(client_res, sizeof(client_err)); -#endif - mg_close_connection(client_conn); - - test_sleep(1); - - test_mg_stop(ctx); -#endif -} -END_TEST - - -START_TEST(test_mg_server_and_client_tls) -{ -#ifndef NO_SSL - - struct mg_context *ctx; - - int ports_cnt; - struct mg_server_ports ports[16]; - struct mg_callbacks callbacks; - char errmsg[256]; - - struct mg_connection *client_conn; - char client_err[256]; - const struct mg_request_info *client_ri; - int client_res; - struct mg_client_options client_options; - - const char *OPTIONS[32]; /* initializer list here is rejected by CI test */ - int opt_idx = 0; - char server_cert[256]; - char client_cert[256]; - const char *res_dir = locate_resources(); - - ck_assert(res_dir != NULL); - strcpy(server_cert, res_dir); - strcpy(client_cert, res_dir); -#ifdef _WIN32 - strcat(server_cert, "cert\\server.pem"); - strcat(client_cert, "cert\\client.pem"); -#else - strcat(server_cert, "cert/server.pem"); - strcat(client_cert, "cert/client.pem"); -#endif - - memset((void *)OPTIONS, 0, sizeof(OPTIONS)); -#if !defined(NO_FILES) - OPTIONS[opt_idx++] = "document_root"; - OPTIONS[opt_idx++] = "."; -#endif - OPTIONS[opt_idx++] = "listening_ports"; - OPTIONS[opt_idx++] = "8080r,8443s"; - OPTIONS[opt_idx++] = "ssl_certificate"; - OPTIONS[opt_idx++] = server_cert; - OPTIONS[opt_idx++] = "ssl_verify_peer"; - OPTIONS[opt_idx++] = "yes"; - OPTIONS[opt_idx++] = "ssl_ca_file"; - OPTIONS[opt_idx++] = client_cert; - - ck_assert_int_le(opt_idx, (int)(sizeof(OPTIONS) / sizeof(OPTIONS[0]))); - ck_assert(OPTIONS[sizeof(OPTIONS) / sizeof(OPTIONS[0]) - 1] == NULL); - ck_assert(OPTIONS[sizeof(OPTIONS) / sizeof(OPTIONS[0]) - 2] == NULL); - - memset(ports, 0, sizeof(ports)); - memset(&callbacks, 0, sizeof(callbacks)); - memset(errmsg, 0, sizeof(errmsg)); - - callbacks.log_message = log_msg_func; - - ctx = test_mg_start(&callbacks, (void *)errmsg, OPTIONS); - - ck_assert_str_eq(errmsg, ""); - ck_assert(ctx != NULL); - - ports_cnt = mg_get_server_ports(ctx, 16, ports); - ck_assert_int_eq(ports_cnt, 2); - ck_assert_int_eq(ports[0].protocol, 1); - ck_assert_int_eq(ports[0].port, 8080); - ck_assert_int_eq(ports[0].is_ssl, 0); - ck_assert_int_eq(ports[0].is_redirect, 1); - ck_assert_int_eq(ports[1].protocol, 1); - ck_assert_int_eq(ports[1].port, 8443); - ck_assert_int_eq(ports[1].is_ssl, 1); - ck_assert_int_eq(ports[1].is_redirect, 0); - ck_assert_int_eq(ports[2].protocol, 0); - ck_assert_int_eq(ports[2].port, 0); - ck_assert_int_eq(ports[2].is_ssl, 0); - ck_assert_int_eq(ports[2].is_redirect, 0); - - test_sleep(1); - - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8443, 1, client_err, sizeof(client_err)); - ck_assert(client_conn == NULL); - ck_assert_str_ne(client_err, ""); - - memset(client_err, 0, sizeof(client_err)); - memset(&client_options, 0, sizeof(client_options)); - client_options.host = "127.0.0.1"; - client_options.port = 8443; - client_options.client_cert = client_cert; - client_options.server_cert = server_cert; - - client_conn = mg_connect_client_secure(&client_options, - client_err, - sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET / HTTP/1.0\r\n\r\n"); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - -#if defined(NO_FILES) - ck_assert_str_eq(client_ri->uri, "404"); -#else - ck_assert_str_eq(client_ri->uri, "200"); - /* TODO: ck_assert_str_eq(client_ri->request_method, "HTTP/1.0"); */ - client_res = (int)mg_read(client_conn, client_err, sizeof(client_err)); - ck_assert_int_gt(client_res, 0); - ck_assert_int_le(client_res, sizeof(client_err)); -#endif - mg_close_connection(client_conn); - - /* TODO: A client API using a client certificate is missing */ - - test_sleep(1); - - test_mg_stop(ctx); -#endif -} -END_TEST - - -static struct mg_context *g_ctx; - -static int -request_test_handler(struct mg_connection *conn, void *cbdata) -{ - int i; - char chunk_data[32]; - const struct mg_request_info *ri; - struct mg_context *ctx; - void *ud, *cud; - - ctx = mg_get_context(conn); - ud = mg_get_user_data(ctx); - ri = mg_get_request_info(conn); - - ck_assert(ri != NULL); - ck_assert(ctx == g_ctx); - ck_assert(ud == &g_ctx); - - mg_set_user_connection_data(conn, (void *)6543); - cud = mg_get_user_connection_data(conn); - ck_assert_ptr_eq((void *)cud, (void *)6543); - - ck_assert_ptr_eq((void *)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 = 1; i <= 10; i++) { - mg_printf(conn, "%x\r\n", i); - mg_write(conn, chunk_data, (unsigned)i); - mg_printf(conn, "\r\n"); - } - - mg_printf(conn, "0\r\n\r\n"); - - return 1; -} - -#ifdef USE_WEBSOCKET -/****************************************************************************/ -/* WEBSOCKET SERVER */ -/****************************************************************************/ -static const char *websocket_welcome_msg = "websocket welcome\n"; -static const size_t websocket_welcome_msg_len = - 18 /* strlen(websocket_welcome_msg) */; -static const char *websocket_goodbye_msg = "websocket bye\n"; -static const size_t websocket_goodbye_msg_len = - 14 /* strlen(websocket_goodbye_msg) */; - - -static int -websock_server_connect(const struct mg_connection *conn, void *udata) -{ - (void)conn; - - ck_assert_ptr_eq((void *)udata, (void *)7531); - printf("Server: Websocket connected\n"); - - return 0; /* return 0 to accept every connection */ -} - - -static void -websock_server_ready(struct mg_connection *conn, void *udata) -{ - ck_assert_ptr_eq((void *)udata, (void *)7531); - printf("Server: Websocket ready\n"); - - /* Send websocket welcome message */ - mg_lock_connection(conn); - mg_websocket_write(conn, - WEBSOCKET_OPCODE_TEXT, - websocket_welcome_msg, - websocket_welcome_msg_len); - mg_unlock_connection(conn); - - printf("Server: Websocket ready X\n"); -} - - -static int -websock_server_data(struct mg_connection *conn, - int bits, - char *data, - size_t data_len, - void *udata) -{ - (void)bits; - - ck_assert_ptr_eq((void *)udata, (void *)7531); - printf("Server: Got %u bytes from the client\n", (unsigned)data_len); - - if (data_len == 3 && !memcmp(data, "bye", 3)) { - /* Send websocket goodbye message */ - mg_lock_connection(conn); - mg_websocket_write(conn, - WEBSOCKET_OPCODE_TEXT, - websocket_goodbye_msg, - websocket_goodbye_msg_len); - mg_unlock_connection(conn); - } else if (data_len == 5 && !memcmp(data, "data1", 5)) { - mg_lock_connection(conn); - mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, "ok1", 3); - mg_unlock_connection(conn); - } else if (data_len == 5 && !memcmp(data, "data2", 5)) { - mg_lock_connection(conn); - mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, "ok 2", 4); - mg_unlock_connection(conn); - } else if (data_len == 5 && !memcmp(data, "data3", 5)) { - mg_lock_connection(conn); - mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, "ok - 3", 6); - mg_unlock_connection(conn); - } else { - -#if defined(__MINGW32__) || defined(__GNUC__) -#pragma GCC diagnostic push -#pragma GCC diagnostic ignored "-Wunreachable-code" -#endif -#ifdef __clang__ -#pragma clang diagnostic push -#pragma clang diagnostic ignored "-Wunreachable-code" -#endif - - ck_abort_msg("Got unexpected message from websocket client"); - - - return 0; - -#ifdef __clang__ -#pragma clang diagnostic pop -#endif -#if defined(__MINGW32__) || defined(__GNUC__) -#pragma GCC diagnostic pop -#endif - } - - return 1; /* return 1 to keep the connetion open */ -} - -static void -websock_server_close(const struct mg_connection *conn, void *udata) -{ - (void)conn; - - ck_assert_ptr_eq((void *)udata, (void *)7531); - printf("Server: Close connection\n"); - - /* Can not send a websocket goodbye message here - the connection is already - * closed */ -} - - -/****************************************************************************/ -/* WEBSOCKET CLIENT */ -/****************************************************************************/ -struct tclient_data { - void *data; - size_t len; - int closed; - int clientId; -}; - - -static int -websocket_client_data_handler(struct mg_connection *conn, - int flags, - char *data, - size_t data_len, - void *user_data) -{ - struct mg_context *ctx = mg_get_context(conn); - struct tclient_data *pclient_data = - (struct tclient_data *)mg_get_user_data(ctx); - - (void)user_data; /* TODO: check this */ - - ck_assert(pclient_data != NULL); - ck_assert_int_eq(flags, (int)(128 | 1)); - - printf("Client %i received data from server: ", pclient_data->clientId); - fwrite(data, 1, data_len, stdout); - printf("\n"); - - pclient_data->data = malloc(data_len); - ck_assert(pclient_data->data != NULL); - memcpy(pclient_data->data, data, data_len); - pclient_data->len = data_len; - - return 1; -} - - -static void -websocket_client_close_handler(const struct mg_connection *conn, - void *user_data) -{ - struct mg_context *ctx = mg_get_context(conn); - struct tclient_data *pclient_data = - (struct tclient_data *)mg_get_user_data(ctx); - - (void)user_data; /* TODO: check this */ - - ck_assert(pclient_data != NULL); - - printf("Client %i: Close handler\n", pclient_data->clientId); - pclient_data->closed++; -} -#endif - - -START_TEST(test_request_handlers) -{ - char ebuf[100]; - struct mg_context *ctx; - struct mg_connection *client_conn; - const struct mg_request_info *ri; - char uri[64]; - char buf[1 + 2 + 3 + 4 + 5 + 6 + 7 + 8 + 9 + 10 + 8]; - const char *expected = - "112123123412345123456123456712345678123456789123456789A"; - int i; - const char *request = "GET /U7 HTTP/1.0\r\n\r\n"; -#if defined(USE_IPV6) && defined(NO_SSL) - const char *HTTP_PORT = "8084,[::]:8086"; - short ipv4_port = 8084; - short ipv6_port = 8086; -#elif !defined(USE_IPV6) && defined(NO_SSL) - const char *HTTP_PORT = "8084"; - short ipv4_port = 8084; -#elif defined(USE_IPV6) && !defined(NO_SSL) - const char *HTTP_PORT = "8084,[::]:8086,8194r,[::]:8196r,8094s,[::]:8096s"; - short ipv4_port = 8084; - short ipv4s_port = 8094; - short ipv4r_port = 8194; - short ipv6_port = 8086; - short ipv6s_port = 8096; - short ipv6r_port = 8196; -#elif !defined(USE_IPV6) && !defined(NO_SSL) - const char *HTTP_PORT = "8084,8194r,8094s"; - short ipv4_port = 8084; - short ipv4s_port = 8094; - short ipv4r_port = 8194; -#endif - - const char *OPTIONS[16]; - const char *opt; - FILE *f; - const char *plain_file_content; - const char *encoded_file_content; - const char *cgi_script_content; - const char *expected_cgi_result; - int opt_idx = 0; - -#if !defined(NO_SSL) - const char *ssl_cert = locate_ssl_cert(); -#endif - -#if defined(USE_WEBSOCKET) - struct tclient_data ws_client1_data = {NULL, 0, 0, 1}; - struct tclient_data ws_client2_data = {NULL, 0, 0, 2}; - struct tclient_data ws_client3_data = {NULL, 0, 0, 3}; - struct tclient_data ws_client4_data = {NULL, 0, 0, 4}; - struct mg_connection *ws_client1_conn = NULL; - struct mg_connection *ws_client2_conn = NULL; - struct mg_connection *ws_client3_conn = NULL; - struct mg_connection *ws_client4_conn = NULL; -#endif - - char cmd_buf[256]; - - memset((void *)OPTIONS, 0, sizeof(OPTIONS)); - OPTIONS[opt_idx++] = "listening_ports"; - OPTIONS[opt_idx++] = HTTP_PORT; - OPTIONS[opt_idx++] = "authentication_domain"; - OPTIONS[opt_idx++] = "test.domain"; -#if !defined(NO_FILES) - OPTIONS[opt_idx++] = "document_root"; - OPTIONS[opt_idx++] = "."; -#endif -#ifndef NO_SSL - ck_assert(ssl_cert != NULL); - OPTIONS[opt_idx++] = "ssl_certificate"; - OPTIONS[opt_idx++] = ssl_cert; -#endif - ck_assert_int_le(opt_idx, (int)(sizeof(OPTIONS) / sizeof(OPTIONS[0]))); - ck_assert(OPTIONS[sizeof(OPTIONS) / sizeof(OPTIONS[0]) - 1] == NULL); - ck_assert(OPTIONS[sizeof(OPTIONS) / sizeof(OPTIONS[0]) - 2] == NULL); - - ctx = test_mg_start(NULL, &g_ctx, OPTIONS); - - ck_assert(ctx != NULL); - g_ctx = ctx; - - opt = mg_get_option(ctx, "listening_ports"); - ck_assert_str_eq(opt, HTTP_PORT); - opt = mg_get_option(ctx, "cgi_environment"); - ck_assert_str_eq(opt, ""); - opt = mg_get_option(ctx, "unknown_option_name"); - ck_assert(opt == 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 *)(ptrdiff_t)i); - } - -#ifdef USE_WEBSOCKET - mg_set_websocket_handler(ctx, - "/websocket", - websock_server_connect, - websock_server_ready, - websock_server_data, - websock_server_close, - (void *)7531); -#endif - - /* Try to load non existing file */ - client_conn = mg_download("localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "GET /file/not/found HTTP/1.0\r\n\r\n"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "404"); - mg_close_connection(client_conn); - - /* Get data from callback */ - client_conn = mg_download( - "localhost", ipv4_port, 0, ebuf, sizeof(ebuf), "%s", request); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, (int)strlen(expected)); - buf[i] = 0; - ck_assert_str_eq(buf, expected); - mg_close_connection(client_conn); - - /* Get data from callback using http://127.0.0.1 */ - client_conn = mg_download( - "127.0.0.1", ipv4_port, 0, ebuf, sizeof(ebuf), "%s", request); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - if ((i >= 0) && ((size_t)i < sizeof(buf))) { - buf[i] = 0; - } else { - ck_abort_msg( - "ERROR: test_request_handlers: read returned %i (>=0, <%i)", - (int)i, - (int)sizeof(buf)); - } - ck_assert((int)i < (int)sizeof(buf)); - ck_assert(i > 0); - ck_assert_int_eq(i, (int)strlen(expected)); - buf[i] = 0; - ck_assert_str_eq(buf, expected); - mg_close_connection(client_conn); - -#if defined(USE_IPV6) - /* Get data from callback using http://[::1] */ - client_conn = - mg_download("[::1]", ipv6_port, 0, ebuf, sizeof(ebuf), "%s", request); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, (int)strlen(expected)); - buf[i] = 0; - ck_assert_str_eq(buf, expected); - mg_close_connection(client_conn); -#endif - -#if !defined(NO_SSL) - /* Get data from callback using https://127.0.0.1 */ - client_conn = mg_download( - "127.0.0.1", ipv4s_port, 1, ebuf, sizeof(ebuf), "%s", request); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, (int)strlen(expected)); - buf[i] = 0; - ck_assert_str_eq(buf, expected); - mg_close_connection(client_conn); - - /* Get redirect from callback using http://127.0.0.1 */ - client_conn = mg_download( - "127.0.0.1", ipv4r_port, 0, ebuf, sizeof(ebuf), "%s", request); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "302"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, -1); - mg_close_connection(client_conn); -#endif - -#if defined(USE_IPV6) && !defined(NO_SSL) - /* Get data from callback using https://[::1] */ - client_conn = - mg_download("[::1]", ipv6s_port, 1, ebuf, sizeof(ebuf), "%s", request); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, (int)strlen(expected)); - buf[i] = 0; - ck_assert_str_eq(buf, expected); - mg_close_connection(client_conn); - - /* Get redirect from callback using http://127.0.0.1 */ - client_conn = - mg_download("[::1]", ipv6r_port, 0, ebuf, sizeof(ebuf), "%s", request); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "302"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, -1); - mg_close_connection(client_conn); -#endif - -/* It seems to be impossible to find out what the actual working - * directory of the CI test environment is. Before breaking another - * dozen of builds by trying blindly with different paths, just - * create the file here */ -#ifdef _WIN32 - f = fopen("test.txt", "wb"); -#else - f = fopen("test.txt", "w"); -#endif - plain_file_content = "simple text file\n"; - fwrite(plain_file_content, 17, 1, f); - fclose(f); - -#ifdef _WIN32 - f = fopen("test_gz.txt.gz", "wb"); -#else - f = fopen("test_gz.txt.gz", "w"); -#endif - encoded_file_content = "\x1f\x8b\x08\x08\xf8\x9d\xcb\x55\x00\x00" - "test_gz.txt" - "\x00\x01\x11\x00\xee\xff" - "zipped text file" - "\x0a\x34\x5f\xcc\x49\x11\x00\x00\x00"; - fwrite(encoded_file_content, 1, 52, f); - fclose(f); - -#ifdef _WIN32 - f = fopen("test.cgi", "wb"); - cgi_script_content = "#!test.cgi.cmd\r\n"; - fwrite(cgi_script_content, strlen(cgi_script_content), 1, f); - fclose(f); - f = fopen("test.cgi.cmd", "w"); - cgi_script_content = "@echo off\r\n" - "echo Connection: close\r\n" - "echo Content-Type: text/plain\r\n" - "echo.\r\n" - "echo CGI test\r\n" - "\r\n"; - fwrite(cgi_script_content, strlen(cgi_script_content), 1, f); - fclose(f); -#else - f = fopen("test.cgi", "w"); - cgi_script_content = "#!/bin/sh\n\n" - "printf \"Connection: close\\r\\n\"\n" - "printf \"Content-Type: text/plain\\r\\n\"\n" - "printf \"\\r\\n\"\n" - "printf \"CGI test\\r\\n\"\n" - "\n"; - (void)fwrite(cgi_script_content, strlen(cgi_script_content), 1, f); - (void)fclose(f); - (void)system("chmod a+x test.cgi"); -#endif - expected_cgi_result = "CGI test"; - - /* Get static data */ - client_conn = mg_download("localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "GET /test.txt HTTP/1.0\r\n\r\n"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - -#if defined(NO_FILES) - ck_assert_str_eq(ri->uri, "404"); -#else - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, 17); - if ((i >= 0) && (i < (int)sizeof(buf))) { - buf[i] = 0; - } - ck_assert_str_eq(buf, plain_file_content); -#endif - mg_close_connection(client_conn); - - -/* Test with CGI test executable */ -#if defined(_WIN32) - sprintf(cmd_buf, "copy %s\\cgi_test.cgi cgi_test.exe", locate_test_exes()); -#else - sprintf(cmd_buf, "cp %s/cgi_test.cgi cgi_test.cgi", locate_test_exes()); -#endif - (void)system(cmd_buf); - -#if !defined(NO_CGI) && !defined(NO_FILES) && !defined(_WIN32) - /* TODO: add test for windows, check with POST */ - client_conn = mg_download( - "localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "POST /cgi_test.cgi HTTP/1.0\r\nContent-Length: 3\r\n\r\nABC"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); -#endif - - /* Get zipped static data - will not work if Accept-Encoding is not set */ - client_conn = mg_download("localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "GET /test_gz.txt HTTP/1.0\r\n\r\n"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - - ck_assert_str_eq(ri->uri, "404"); - mg_close_connection(client_conn); - - /* Get zipped static data - with Accept-Encoding */ - client_conn = mg_download( - "localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "GET /test_gz.txt HTTP/1.0\r\nAccept-Encoding: gzip\r\n\r\n"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - -#if defined(NO_FILES) - ck_assert_str_eq(ri->uri, "404"); -#else - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, 52); - if ((i >= 0) && (i < (int)sizeof(buf))) { - buf[i] = 0; - } - ck_assert_int_eq(ri->content_length, 52); - ck_assert_str_eq(buf, encoded_file_content); -#endif - mg_close_connection(client_conn); - -/* Get CGI generated data */ -#if !defined(NO_CGI) - client_conn = mg_download("localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "GET /test.cgi HTTP/1.0\r\n\r\n"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - -#if defined(NO_FILES) - ck_assert_str_eq(ri->uri, "404"); - - (void)expected_cgi_result; - (void)cgi_script_content; -#else - i = mg_read(client_conn, buf, sizeof(buf)); - if ((i >= 0) && (i < (int)sizeof(buf))) { - while ((i > 0) && ((buf[i - 1] == '\r') || (buf[i - 1] == '\n'))) { - i--; - } - buf[i] = 0; - } - /* ck_assert_int_eq(i, (int)strlen(expected_cgi_result)); */ - ck_assert_str_eq(buf, expected_cgi_result); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); -#endif - -#else - (void)expected_cgi_result; - (void)cgi_script_content; -#endif - - /* Get directory listing */ - client_conn = mg_download("localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "GET / HTTP/1.0\r\n\r\n"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); -#if defined(NO_FILES) - ck_assert_str_eq(ri->uri, "404"); -#else - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert(i > 6); - buf[6] = 0; - ck_assert_str_eq(buf, "<html>"); -#endif - mg_close_connection(client_conn); - - /* POST to static file (will not work) */ - client_conn = mg_download("localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "POST /test.txt HTTP/1.0\r\n\r\n"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); -#if defined(NO_FILES) - ck_assert_str_eq(ri->uri, "404"); -#else - ck_assert_str_eq(ri->uri, "405"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert(i >= 29); - buf[29] = 0; - ck_assert_str_eq(buf, "Error 405: Method Not Allowed"); -#endif - mg_close_connection(client_conn); - - /* PUT to static file (will not work) */ - client_conn = mg_download("localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "%s", - "PUT /test.txt HTTP/1.0\r\n\r\n"); - ck_assert(client_conn != NULL); - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); -#if defined(NO_FILES) - ck_assert_str_eq(ri->uri, "405"); /* method not allowed */ -#else - ck_assert_str_eq(ri->uri, "401"); /* not authorized */ -#endif - mg_close_connection(client_conn); - - - /* Get data from callback using mg_connect_client instead of mg_download */ - memset(ebuf, 0, sizeof(ebuf)); - client_conn = - mg_connect_client("127.0.0.1", ipv4_port, 0, ebuf, sizeof(ebuf)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(ebuf, ""); - - mg_printf(client_conn, "%s", request); - - i = mg_get_response(client_conn, ebuf, sizeof(ebuf), 10000); - ck_assert_int_ge(i, 0); - ck_assert_str_eq(ebuf, ""); - - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, (int)strlen(expected)); - buf[i] = 0; - ck_assert_str_eq(buf, expected); - mg_close_connection(client_conn); - - /* Get data from callback using mg_connect_client and absolute URI */ - memset(ebuf, 0, sizeof(ebuf)); - client_conn = - mg_connect_client("localhost", ipv4_port, 0, ebuf, sizeof(ebuf)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(ebuf, ""); - - mg_printf(client_conn, - "GET http://test.domain:%d/U7 HTTP/1.0\r\n\r\n", - ipv4_port); - - i = mg_get_response(client_conn, ebuf, sizeof(ebuf), 10000); - ck_assert_int_ge(i, 0); - ck_assert_str_eq(ebuf, ""); - - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, (int)strlen(expected)); - buf[i] = 0; - ck_assert_str_eq(buf, expected); - mg_close_connection(client_conn); - - /* Get data from callback using mg_connect_client and absolute URI with a - * sub-domain */ - memset(ebuf, 0, sizeof(ebuf)); - client_conn = - mg_connect_client("localhost", ipv4_port, 0, ebuf, sizeof(ebuf)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(ebuf, ""); - - mg_printf(client_conn, - "GET http://subdomain.test.domain:%d/U7 HTTP/1.0\r\n\r\n", - ipv4_port); - - i = mg_get_response(client_conn, ebuf, sizeof(ebuf), 10000); - ck_assert_int_ge(i, 0); - ck_assert_str_eq(ebuf, ""); - - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - i = mg_read(client_conn, buf, sizeof(buf)); - ck_assert_int_eq(i, (int)strlen(expected)); - buf[i] = 0; - ck_assert_str_eq(buf, expected); - mg_close_connection(client_conn); - - -/* Websocket test */ -#ifdef USE_WEBSOCKET - /* Then connect a first client */ - ws_client1_conn = - mg_connect_websocket_client("localhost", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "/websocket", - NULL, - websocket_client_data_handler, - websocket_client_close_handler, - &ws_client1_data); - - ck_assert(ws_client1_conn != NULL); - - wait_not_null( - &(ws_client1_data.data)); /* Wait for the websocket welcome message */ - ck_assert_int_eq(ws_client1_data.closed, 0); - ck_assert_int_eq(ws_client2_data.closed, 0); - ck_assert_int_eq(ws_client3_data.closed, 0); - ck_assert(ws_client2_data.data == NULL); - ck_assert_uint_eq(ws_client2_data.len, 0); - ck_assert(ws_client1_data.data != NULL); - ck_assert_uint_eq(ws_client1_data.len, websocket_welcome_msg_len); - ck_assert(!memcmp(ws_client1_data.data, - websocket_welcome_msg, - websocket_welcome_msg_len)); - free(ws_client1_data.data); - ws_client1_data.data = NULL; - ws_client1_data.len = 0; - - mg_websocket_client_write(ws_client1_conn, - WEBSOCKET_OPCODE_TEXT, - "data1", - 5); - - wait_not_null( - &(ws_client1_data - .data)); /* Wait for the websocket acknowledge message */ - ck_assert_int_eq(ws_client1_data.closed, 0); - ck_assert_int_eq(ws_client2_data.closed, 0); - ck_assert(ws_client2_data.data == NULL); - ck_assert_uint_eq(ws_client2_data.len, 0); - ck_assert(ws_client1_data.data != NULL); - ck_assert_uint_eq(ws_client1_data.len, 3); - ck_assert(!memcmp(ws_client1_data.data, "ok1", 3)); - free(ws_client1_data.data); - ws_client1_data.data = NULL; - ws_client1_data.len = 0; - -/* Now connect a second client */ -#ifdef USE_IPV6 - ws_client2_conn = - mg_connect_websocket_client("[::1]", - ipv6_port, - 0, - ebuf, - sizeof(ebuf), - "/websocket", - NULL, - websocket_client_data_handler, - websocket_client_close_handler, - &ws_client2_data); -#else - ws_client2_conn = - mg_connect_websocket_client("127.0.0.1", - ipv4_port, - 0, - ebuf, - sizeof(ebuf), - "/websocket", - NULL, - websocket_client_data_handler, - websocket_client_close_handler, - &ws_client2_data); -#endif - ck_assert(ws_client2_conn != NULL); - - wait_not_null( - &(ws_client2_data.data)); /* Wait for the websocket welcome message */ - ck_assert(ws_client1_data.closed == 0); - ck_assert(ws_client2_data.closed == 0); - ck_assert(ws_client1_data.data == NULL); - ck_assert(ws_client1_data.len == 0); - ck_assert(ws_client2_data.data != NULL); - ck_assert(ws_client2_data.len == websocket_welcome_msg_len); - ck_assert(!memcmp(ws_client2_data.data, - websocket_welcome_msg, - websocket_welcome_msg_len)); - free(ws_client2_data.data); - ws_client2_data.data = NULL; - ws_client2_data.len = 0; - - mg_websocket_client_write(ws_client1_conn, - WEBSOCKET_OPCODE_TEXT, - "data2", - 5); - - wait_not_null( - &(ws_client1_data - .data)); /* Wait for the websocket acknowledge message */ - - ck_assert(ws_client1_data.closed == 0); - ck_assert(ws_client2_data.closed == 0); - ck_assert(ws_client2_data.data == NULL); - ck_assert(ws_client2_data.len == 0); - ck_assert(ws_client1_data.data != NULL); - ck_assert(ws_client1_data.len == 4); - ck_assert(!memcmp(ws_client1_data.data, "ok 2", 4)); - free(ws_client1_data.data); - ws_client1_data.data = NULL; - ws_client1_data.len = 0; - - mg_websocket_client_write(ws_client1_conn, WEBSOCKET_OPCODE_TEXT, "bye", 3); - - wait_not_null( - &(ws_client1_data.data)); /* Wait for the websocket goodbye message */ - - ck_assert(ws_client1_data.closed == 0); - ck_assert(ws_client2_data.closed == 0); - ck_assert(ws_client2_data.data == NULL); - ck_assert(ws_client2_data.len == 0); - ck_assert(ws_client1_data.data != NULL); - ck_assert(ws_client1_data.len == websocket_goodbye_msg_len); - ck_assert(!memcmp(ws_client1_data.data, - websocket_goodbye_msg, - websocket_goodbye_msg_len)); - free(ws_client1_data.data); - ws_client1_data.data = NULL; - ws_client1_data.len = 0; - - mg_close_connection(ws_client1_conn); - - test_sleep(3); /* Won't get any message */ - - ck_assert(ws_client1_data.closed == 1); - ck_assert(ws_client2_data.closed == 0); - ck_assert(ws_client1_data.data == NULL); - ck_assert(ws_client1_data.len == 0); - ck_assert(ws_client2_data.data == NULL); - ck_assert(ws_client2_data.len == 0); - - mg_websocket_client_write(ws_client2_conn, WEBSOCKET_OPCODE_TEXT, "bye", 3); - - wait_not_null( - &(ws_client2_data.data)); /* Wait for the websocket goodbye message */ - - ck_assert(ws_client1_data.closed == 1); - ck_assert(ws_client2_data.closed == 0); - ck_assert(ws_client1_data.data == NULL); - ck_assert(ws_client1_data.len == 0); - ck_assert(ws_client2_data.data != NULL); - ck_assert(ws_client2_data.len == websocket_goodbye_msg_len); - ck_assert(!memcmp(ws_client2_data.data, - websocket_goodbye_msg, - websocket_goodbye_msg_len)); - free(ws_client2_data.data); - ws_client2_data.data = NULL; - ws_client2_data.len = 0; - - mg_close_connection(ws_client2_conn); - - test_sleep(3); /* Won't get any message */ - - ck_assert(ws_client1_data.closed == 1); - ck_assert(ws_client2_data.closed == 1); - ck_assert(ws_client1_data.data == NULL); - ck_assert(ws_client1_data.len == 0); - ck_assert(ws_client2_data.data == NULL); - ck_assert(ws_client2_data.len == 0); - - /* Connect client 3 */ - ws_client3_conn = - mg_connect_websocket_client("localhost", -#if defined(NO_SSL) - ipv4_port, - 0, -#else - ipv4s_port, - 1, -#endif - ebuf, - sizeof(ebuf), - "/websocket", - NULL, - websocket_client_data_handler, - websocket_client_close_handler, - &ws_client3_data); - - ck_assert(ws_client3_conn != NULL); - - wait_not_null( - &(ws_client3_data.data)); /* Wait for the websocket welcome message */ - ck_assert(ws_client1_data.closed == 1); - ck_assert(ws_client2_data.closed == 1); - ck_assert(ws_client3_data.closed == 0); - ck_assert(ws_client1_data.data == NULL); - ck_assert(ws_client1_data.len == 0); - ck_assert(ws_client2_data.data == NULL); - ck_assert(ws_client2_data.len == 0); - ck_assert(ws_client3_data.data != NULL); - ck_assert(ws_client3_data.len == websocket_welcome_msg_len); - ck_assert(!memcmp(ws_client3_data.data, - websocket_welcome_msg, - websocket_welcome_msg_len)); - free(ws_client3_data.data); - ws_client3_data.data = NULL; - ws_client3_data.len = 0; - - /* Disconnect client 3 */ - ck_assert(ws_client3_data.closed == 0); - mg_close_connection(ws_client3_conn); - ck_assert(ws_client3_data.closed == 1); - - /* Connect client 4 */ - ws_client4_conn = - mg_connect_websocket_client("localhost", -#if defined(NO_SSL) - ipv4_port, - 0, -#else - ipv4s_port, - 1, -#endif - ebuf, - sizeof(ebuf), - "/websocket", - NULL, - websocket_client_data_handler, - websocket_client_close_handler, - &ws_client4_data); - - ck_assert(ws_client4_conn != NULL); - - wait_not_null( - &(ws_client4_data.data)); /* Wait for the websocket welcome message */ - ck_assert(ws_client1_data.closed == 1); - ck_assert(ws_client2_data.closed == 1); - ck_assert(ws_client3_data.closed == 1); - ck_assert(ws_client4_data.closed == 0); - ck_assert(ws_client4_data.data != NULL); - ck_assert(ws_client4_data.len == websocket_welcome_msg_len); - ck_assert(!memcmp(ws_client4_data.data, - websocket_welcome_msg, - websocket_welcome_msg_len)); - free(ws_client4_data.data); - ws_client4_data.data = NULL; - ws_client4_data.len = 0; - -/* stop the server without closing this connection */ - -#endif - - /* Close the server */ - g_ctx = NULL; - test_mg_stop(ctx); - mark_point(); - -#ifdef USE_WEBSOCKET - for (i = 0; i < 100; i++) { - test_sleep(1); - if (ws_client3_data.closed != 0) { - mark_point(); - break; - } - } - - ck_assert_int_eq(ws_client4_data.closed, 1); -#endif -} -END_TEST - - -static int g_field_found_return = -999; - -static int -field_found(const char *key, - const char *filename, - char *path, - size_t pathlen, - void *user_data) -{ - ck_assert_ptr_ne(key, NULL); - ck_assert_ptr_ne(filename, NULL); - ck_assert_ptr_ne(path, NULL); - ck_assert_uint_gt(pathlen, 128); - ck_assert_ptr_eq(user_data, (void *)&g_field_found_return); - - ck_assert((g_field_found_return == FORM_FIELD_STORAGE_GET) - || (g_field_found_return == FORM_FIELD_STORAGE_STORE) - || (g_field_found_return == FORM_FIELD_STORAGE_SKIP) - || (g_field_found_return == FORM_FIELD_STORAGE_ABORT)); - - ck_assert_str_ne(key, "dontread"); - - if (!strcmp(key, "break_field_handler")) { - return FORM_FIELD_STORAGE_ABORT; - } - if (!strcmp(key, "continue_field_handler")) { - return FORM_FIELD_STORAGE_SKIP; - } - - if (g_field_found_return == FORM_FIELD_STORAGE_STORE) { - strncpy(path, key, pathlen - 8); - strcat(path, ".txt"); - } - - return g_field_found_return; -} - - -static int g_field_step; - -static int -field_get(const char *key, - const char *value_untruncated, - size_t valuelen, - void *user_data) -{ - /* Copy the untruncated value, so string compare functions can be used. */ - /* The check unit test library does not have build in memcmp functions. */ - char *value = (char *)malloc(valuelen + 1); - ck_assert(value != NULL); - memcpy(value, value_untruncated, valuelen); - value[valuelen] = 0; - - ck_assert_ptr_eq(user_data, (void *)&g_field_found_return); - ck_assert_int_ge(g_field_step, 0); - - ++g_field_step; - switch (g_field_step) { - case 1: - ck_assert_str_eq(key, "textin"); - ck_assert_uint_eq(valuelen, 4); - ck_assert_str_eq(value, "text"); - break; - case 2: - ck_assert_str_eq(key, "passwordin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 3: - ck_assert_str_eq(key, "radio1"); - ck_assert_uint_eq(valuelen, 4); - ck_assert_str_eq(value, "val1"); - break; - case 4: - ck_assert_str_eq(key, "radio2"); - ck_assert_uint_eq(valuelen, 4); - ck_assert_str_eq(value, "val1"); - break; - case 5: - ck_assert_str_eq(key, "check1"); - ck_assert_uint_eq(valuelen, 4); - ck_assert_str_eq(value, "val1"); - break; - case 6: - ck_assert_str_eq(key, "numberin"); - ck_assert_uint_eq(valuelen, 1); - ck_assert_str_eq(value, "1"); - break; - case 7: - ck_assert_str_eq(key, "datein"); - ck_assert_uint_eq(valuelen, 8); - ck_assert_str_eq(value, "1.1.2016"); - break; - case 8: - ck_assert_str_eq(key, "colorin"); - ck_assert_uint_eq(valuelen, 7); - ck_assert_str_eq(value, "#80ff00"); - break; - case 9: - ck_assert_str_eq(key, "rangein"); - ck_assert_uint_eq(valuelen, 1); - ck_assert_str_eq(value, "3"); - break; - case 10: - ck_assert_str_eq(key, "monthin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 11: - ck_assert_str_eq(key, "weekin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 12: - ck_assert_str_eq(key, "timein"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 13: - ck_assert_str_eq(key, "datetimen"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 14: - ck_assert_str_eq(key, "datetimelocalin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 15: - ck_assert_str_eq(key, "emailin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 16: - ck_assert_str_eq(key, "searchin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 17: - ck_assert_str_eq(key, "telin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 18: - ck_assert_str_eq(key, "urlin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 19: - ck_assert_str_eq(key, "filein"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 20: - ck_assert_str_eq(key, "filesin"); - ck_assert_uint_eq(valuelen, 0); - ck_assert_str_eq(value, ""); - break; - case 21: - ck_assert_str_eq(key, "selectin"); - ck_assert_uint_eq(valuelen, 4); - ck_assert_str_eq(value, "opt1"); - break; - case 22: - ck_assert_str_eq(key, "message"); - ck_assert_uint_eq(valuelen, 23); - ck_assert_str_eq(value, "Text area default text."); - break; - default: - ck_abort_msg("field_get called with g_field_step == %i", - (int)g_field_step); - } - - free(value); - - return 0; -} - - -static const char *myfile_content = "Content of myfile.txt\r\n"; -static const int myfile_content_rep = 500; - - -static int -field_store(const char *path, long long file_size, void *user_data) -{ - FILE *f; - ck_assert_ptr_eq(user_data, (void *)&g_field_found_return); - ck_assert_int_ge(g_field_step, 100); - - ++g_field_step; - switch (g_field_step) { - case 101: - ck_assert_str_eq(path, "storeme.txt"); - ck_assert_int_eq(file_size, 9); - f = fopen(path, "r"); - ck_assert_ptr_ne(f, NULL); - if (f) { - char buf[32] = {0}; - int i = (int)fread(buf, 1, 31, f); - ck_assert_int_eq(i, 9); - fclose(f); - ck_assert_str_eq(buf, "storetest"); - } - break; - case 102: - ck_assert_str_eq(path, "file2store.txt"); - ck_assert_uint_eq(23, strlen(myfile_content)); - ck_assert_int_eq(file_size, 23 * myfile_content_rep); -#ifdef _WIN32 - f = fopen(path, "rb"); -#else - f = fopen(path, "r"); -#endif - ck_assert_ptr_ne(f, NULL); - if (f) { - char buf[32] = {0}; - int r, i; - for (r = 0; r < myfile_content_rep; r++) { - i = (int)fread(buf, 1, 23, f); - ck_assert_int_eq(i, 23); - ck_assert_str_eq(buf, myfile_content); - } - i = (int)fread(buf, 1, 23, f); - ck_assert_int_eq(i, 0); - fclose(f); - } - break; - default: - ck_abort_msg("field_get called with g_field_step == %i", - (int)g_field_step); - } - - return 0; -} - - -static int -FormGet(struct mg_connection *conn, void *cbdata) -{ - const struct mg_request_info *req_info = mg_get_request_info(conn); - int ret; - struct mg_form_data_handler fdh = {field_found, field_get, NULL, NULL}; - - (void)cbdata; - - ck_assert(req_info != NULL); - - mg_printf(conn, "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\n"); - fdh.user_data = (void *)&g_field_found_return; - - /* Call the form handler */ - g_field_step = 0; - g_field_found_return = FORM_FIELD_STORAGE_GET; - ret = mg_handle_form_request(conn, &fdh); - g_field_found_return = -888; - ck_assert_int_eq(ret, 22); - ck_assert_int_eq(g_field_step, 22); - mg_printf(conn, "%i\r\n", ret); - g_field_step = 1000; - - return 1; -} - - -static int -FormStore(struct mg_connection *conn, - void *cbdata, - int ret_expected, - int field_step_expected) -{ - const struct mg_request_info *req_info = mg_get_request_info(conn); - int ret; - struct mg_form_data_handler fdh = {field_found, - field_get, - field_store, - NULL}; - - (void)cbdata; - - ck_assert(req_info != NULL); - - mg_printf(conn, "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\n"); - fdh.user_data = (void *)&g_field_found_return; - - /* Call the form handler */ - g_field_step = 100; - g_field_found_return = FORM_FIELD_STORAGE_STORE; - ret = mg_handle_form_request(conn, &fdh); - ck_assert_int_eq(ret, ret_expected); - ck_assert_int_eq(g_field_step, field_step_expected); - mg_printf(conn, "%i\r\n", ret); - g_field_step = 1000; - - return 1; -} - - -static int -FormStore1(struct mg_connection *conn, void *cbdata) -{ - return FormStore(conn, cbdata, 3, 101); -} - - -static int -FormStore2(struct mg_connection *conn, void *cbdata) -{ - return FormStore(conn, cbdata, 4, 102); -} - - -static void -send_chunk_stringl(struct mg_connection *conn, - const char *chunk, - unsigned int chunk_len) -{ - char lenbuf[16]; - size_t lenbuf_len; - int ret; - - /* First store the length information in a text buffer. */ - sprintf(lenbuf, "%x\r\n", chunk_len); - lenbuf_len = strlen(lenbuf); - - /* Then send length information, chunk and terminating \r\n. */ - ret = mg_write(conn, lenbuf, lenbuf_len); - ck_assert_int_eq(ret, (int)lenbuf_len); - - ret = mg_write(conn, chunk, chunk_len); - ck_assert_int_eq(ret, (int)chunk_len); - - ret = mg_write(conn, "\r\n", 2); - ck_assert_int_eq(ret, 2); -} - - -static void -send_chunk_string(struct mg_connection *conn, const char *chunk) -{ - send_chunk_stringl(conn, chunk, (unsigned int)strlen(chunk)); -} - - -START_TEST(test_handle_form) -{ - struct mg_context *ctx; - struct mg_connection *client_conn; - const struct mg_request_info *ri; - const char *OPTIONS[8]; - const char *opt; - int opt_idx = 0; - char ebuf[100]; - const char *multipart_body; - const char *boundary; - size_t body_len, body_sent, chunk_len; - int sleep_cnt; - - memset((void *)OPTIONS, 0, sizeof(OPTIONS)); - OPTIONS[opt_idx++] = "listening_ports"; - OPTIONS[opt_idx++] = "8884"; - ck_assert_int_le(opt_idx, (int)(sizeof(OPTIONS) / sizeof(OPTIONS[0]))); - ck_assert(OPTIONS[sizeof(OPTIONS) / sizeof(OPTIONS[0]) - 1] == NULL); - ck_assert(OPTIONS[sizeof(OPTIONS) / sizeof(OPTIONS[0]) - 2] == NULL); - - ctx = test_mg_start(NULL, &g_ctx, OPTIONS); - - ck_assert(ctx != NULL); - g_ctx = ctx; - - opt = mg_get_option(ctx, "listening_ports"); - ck_assert_str_eq(opt, "8884"); - - mg_set_request_handler(ctx, "/handle_form", FormGet, (void *)0); - mg_set_request_handler(ctx, "/handle_form_store", FormStore1, (void *)0); - mg_set_request_handler(ctx, "/handle_form_store2", FormStore2, (void *)0); - - test_sleep(1); - - /* Handle form: "GET" */ - client_conn = mg_download("localhost", - 8884, - 0, - ebuf, - sizeof(ebuf), - "%s", - "GET /handle_form" - "?textin=text&passwordin=&radio1=val1" - "&radio2=val1&check1=val1&numberin=1" - "&datein=1.1.2016&colorin=%2380ff00" - "&rangein=3&monthin=&weekin=&timein=" - "&datetimen=&datetimelocalin=&emailin=" - "&searchin=&telin=&urlin=&filein=" - "&filesin=&selectin=opt1" - "&message=Text+area+default+text. " - "HTTP/1.0\r\n" - "Host: localhost:8884\r\n" - "Connection: close\r\n\r\n"); - ck_assert(client_conn != NULL); - for (sleep_cnt = 0; sleep_cnt < 30; sleep_cnt++) { - test_sleep(1); - if (g_field_step == 1000) { - break; - } - } - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); - - /* Handle form: "POST x-www-form-urlencoded" */ - client_conn = - mg_download("localhost", - 8884, - 0, - ebuf, - sizeof(ebuf), - "%s", - "POST /handle_form HTTP/1.1\r\n" - "Host: localhost:8884\r\n" - "Connection: close\r\n" - "Content-Type: application/x-www-form-urlencoded\r\n" - "Content-Length: 263\r\n" - "\r\n" - "textin=text&passwordin=&radio1=val1&radio2=val1" - "&check1=val1&numberin=1&datein=1.1.2016" - "&colorin=%2380ff00&rangein=3&monthin=&weekin=" - "&timein=&datetimen=&datetimelocalin=&emailin=" - "&searchin=&telin=&urlin=&filein=&filesin=" - "&selectin=opt1&message=Text+area+default+text."); - ck_assert(client_conn != NULL); - for (sleep_cnt = 0; sleep_cnt < 30; sleep_cnt++) { - test_sleep(1); - if (g_field_step == 1000) { - break; - } - } - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); - - /* Handle form: "POST multipart/form-data" */ - multipart_body = - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"textin\"\r\n" - "\r\n" - "text\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"passwordin\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"radio1\"\r\n" - "\r\n" - "val1\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"radio2\"\r\n" - "\r\n" - "val1\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"check1\"\r\n" - "\r\n" - "val1\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"numberin\"\r\n" - "\r\n" - "1\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"datein\"\r\n" - "\r\n" - "1.1.2016\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"colorin\"\r\n" - "\r\n" - "#80ff00\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"rangein\"\r\n" - "\r\n" - "3\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"monthin\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"weekin\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"timein\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"datetimen\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"datetimelocalin\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"emailin\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"searchin\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"telin\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"urlin\"\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"filein\"; filename=\"\"\r\n" - "Content-Type: application/octet-stream\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"filesin\"; filename=\"\"\r\n" - "Content-Type: application/octet-stream\r\n" - "\r\n" - "\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"selectin\"\r\n" - "\r\n" - "opt1\r\n" - "--multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Disposition: form-data; name=\"message\"\r\n" - "\r\n" - "Text area default text.\r\n" - "--multipart-form-data-boundary--see-RFC-2388--\r\n"; - body_len = strlen(multipart_body); - ck_assert_uint_eq(body_len, 2374); /* not required */ - - client_conn = - mg_download("localhost", - 8884, - 0, - ebuf, - sizeof(ebuf), - "POST /handle_form HTTP/1.1\r\n" - "Host: localhost:8884\r\n" - "Connection: close\r\n" - "Content-Type: multipart/form-data; " - "boundary=multipart-form-data-boundary--see-RFC-2388\r\n" - "Content-Length: %u\r\n" - "\r\n%s", - (unsigned int)body_len, - multipart_body); - - ck_assert(client_conn != NULL); - for (sleep_cnt = 0; sleep_cnt < 30; sleep_cnt++) { - test_sleep(1); - if (g_field_step == 1000) { - break; - } - } - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); - - - /* Handle form: "POST multipart/form-data" with chunked transfer encoding */ - client_conn = - mg_download("localhost", - 8884, - 0, - ebuf, - sizeof(ebuf), - "%s", - "POST /handle_form HTTP/1.1\r\n" - "Host: localhost:8884\r\n" - "Connection: close\r\n" - "Content-Type: multipart/form-data; " - "boundary=multipart-form-data-boundary--see-RFC-2388\r\n" - "Transfer-Encoding: chunked\r\n" - "\r\n"); - - ck_assert(client_conn != NULL); - - body_len = strlen(multipart_body); - chunk_len = 1; - body_sent = 0; - while (body_len > body_sent) { - if (chunk_len > (body_len - body_sent)) { - chunk_len = body_len - body_sent; - } - ck_assert_int_gt((int)chunk_len, 0); - mg_printf(client_conn, "%x\r\n", (unsigned int)chunk_len); - mg_write(client_conn, multipart_body + body_sent, chunk_len); - mg_printf(client_conn, "\r\n"); - body_sent += chunk_len; - chunk_len = (chunk_len % 40) + 1; - } - mg_printf(client_conn, "0\r\n"); - - for (sleep_cnt = 0; sleep_cnt < 30; sleep_cnt++) { - test_sleep(1); - if (g_field_step == 1000) { - break; - } - } - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); - - - /* Now test form_store */ - - /* First test with GET */ - client_conn = mg_download("localhost", - 8884, - 0, - ebuf, - sizeof(ebuf), - "%s", - "GET /handle_form_store" - "?storeme=storetest" - "&continue_field_handler=ignore" - "&break_field_handler=abort" - "&dontread=xyz " - "HTTP/1.0\r\n" - "Host: localhost:8884\r\n" - "Connection: close\r\n\r\n"); - - ck_assert(client_conn != NULL); - - for (sleep_cnt = 0; sleep_cnt < 30; sleep_cnt++) { - test_sleep(1); - if (g_field_step == 1000) { - break; - } - } - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); - - - /* Handle form: "POST x-www-form-urlencoded", chunked, store */ - client_conn = - mg_download("localhost", - 8884, - 0, - ebuf, - sizeof(ebuf), - "%s", - "POST /handle_form_store HTTP/1.0\r\n" - "Host: localhost:8884\r\n" - "Connection: close\r\n" - "Content-Type: application/x-www-form-urlencoded\r\n" - "Transfer-Encoding: chunked\r\n" - "\r\n"); - ck_assert(client_conn != NULL); - - send_chunk_string(client_conn, "storeme=store"); - send_chunk_string(client_conn, "test&"); - send_chunk_string(client_conn, "continue_field_handler=ignore"); - send_chunk_string(client_conn, "&br"); - test_sleep(1); - send_chunk_string(client_conn, "eak_field_handler=abort&"); - send_chunk_string(client_conn, "dontread=xyz"); - mg_printf(client_conn, "0\r\n"); - - for (sleep_cnt = 0; sleep_cnt < 30; sleep_cnt++) { - test_sleep(1); - if (g_field_step == 1000) { - break; - } - } - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); - - /* Handle form: "POST multipart/form-data", chunked, store */ - client_conn = - mg_download("localhost", - 8884, - 0, - ebuf, - sizeof(ebuf), - "%s", - "POST /handle_form_store HTTP/1.0\r\n" - "Host: localhost:8884\r\n" - "Connection: close\r\n" - "Content-Type: multipart/form-data; " - "boundary=multipart-form-data-boundary--see-RFC-2388\r\n" - "Transfer-Encoding: chunked\r\n" - "\r\n"); - ck_assert(client_conn != NULL); - - send_chunk_string(client_conn, "--multipart-form-data-boundary"); - send_chunk_string(client_conn, "--see-RFC-2388\r\n"); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"storeme\"\r\n"); - send_chunk_string(client_conn, "\r\n"); - send_chunk_string(client_conn, "storetest\r\n"); - - send_chunk_string(client_conn, "--multipart-form-data-boundary-"); - send_chunk_string(client_conn, "-see-RFC-2388\r\n"); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"continue_field_handler\"\r\n"); - send_chunk_string(client_conn, "\r\n"); - send_chunk_string(client_conn, "ignore\r\n"); - - send_chunk_string(client_conn, "--multipart-form-data-boundary-"); - send_chunk_string(client_conn, "-see-RFC-2388\r\n"); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"break_field_handler\"\r\n"); - send_chunk_string(client_conn, "\r\n"); - send_chunk_string(client_conn, "abort\r\n"); - - send_chunk_string(client_conn, "--multipart-form-data-boundary-"); - send_chunk_string(client_conn, "-see-RFC-2388\r\n"); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"dontread\"\r\n"); - send_chunk_string(client_conn, "\r\n"); - send_chunk_string(client_conn, "xyz\r\n"); - send_chunk_string(client_conn, "--multipart-form-data-boundary"); - send_chunk_string(client_conn, "--see-RFC-2388--\r\n"); - mg_printf(client_conn, "0\r\n"); - - for (sleep_cnt = 0; sleep_cnt < 30; sleep_cnt++) { - test_sleep(1); - if (g_field_step == 1000) { - break; - } - } - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); - - - /* Handle form: "POST multipart/form-data", chunked, store, with files */ - client_conn = - mg_download("localhost", - 8884, - 0, - ebuf, - sizeof(ebuf), - "%s", - "POST /handle_form_store2 HTTP/1.0\r\n" - "Host: localhost:8884\r\n" - "Connection: close\r\n" - "Content-Type: multipart/form-data; " - "boundary=multipart-form-data-boundary--see-RFC-2388\r\n" - "Transfer-Encoding: chunked\r\n" - "\r\n"); - ck_assert(client_conn != NULL); - - boundary = "--multipart-form-data-boundary--see-RFC-2388\r\n"; - - send_chunk_string(client_conn, boundary); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"storeme\"\r\n"); - send_chunk_string(client_conn, "\r\n"); - send_chunk_string(client_conn, "storetest\r\n"); - - send_chunk_string(client_conn, boundary); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"continue_field_handler\";"); - send_chunk_string(client_conn, "filename=\"file_ignored.txt\"\r\n"); - send_chunk_string(client_conn, "Content-Type: "); - send_chunk_string(client_conn, "application/octet-stream\r\n"); - send_chunk_string(client_conn, "X-Ignored-Header: xyz\r\n"); - send_chunk_string(client_conn, "\r\n"); - - /* send some kilobyte of data */ - /* sending megabytes to localhost does not allways work in CI test - * environments (depending on the network stack) */ - body_sent = 0; - do { - send_chunk_string(client_conn, "ignore\r\n"); - body_sent += 8; - /* send some strings that are almost boundaries */ - for (chunk_len = 1; chunk_len < strlen(boundary); chunk_len++) { - /* chunks from 1 byte to strlen(boundary)-1 */ - send_chunk_stringl(client_conn, boundary, (unsigned int)chunk_len); - body_sent += chunk_len; - } - } while (body_sent < 8 * 1024); - send_chunk_string(client_conn, "\r\n"); - - send_chunk_string(client_conn, boundary); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"file2store\";"); - send_chunk_string(client_conn, "filename=\"myfile.txt\"\r\n"); - send_chunk_string(client_conn, "Content-Type: "); - send_chunk_string(client_conn, "application/octet-stream\r\n"); - send_chunk_string(client_conn, "X-Ignored-Header: xyz\r\n"); - send_chunk_string(client_conn, "\r\n"); - for (body_sent = 0; (int)body_sent < (int)myfile_content_rep; body_sent++) { - send_chunk_string(client_conn, myfile_content); - } - send_chunk_string(client_conn, "\r\n"); - - send_chunk_string(client_conn, boundary); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"break_field_handler\"\r\n"); - send_chunk_string(client_conn, "\r\n"); - send_chunk_string(client_conn, "abort\r\n"); - - send_chunk_string(client_conn, boundary); - send_chunk_string(client_conn, "Content-Disposition: form-data; "); - send_chunk_string(client_conn, "name=\"dontread\"\r\n"); - send_chunk_string(client_conn, "\r\n"); - send_chunk_string(client_conn, "xyz\r\n"); - send_chunk_string(client_conn, "--multipart-form-data-boundary"); - send_chunk_string(client_conn, "--see-RFC-2388--\r\n"); - mg_printf(client_conn, "0\r\n"); - - for (sleep_cnt = 0; sleep_cnt < 30; sleep_cnt++) { - test_sleep(1); - if (g_field_step == 1000) { - break; - } - } - ri = mg_get_request_info(client_conn); - - ck_assert(ri != NULL); - ck_assert_str_eq(ri->uri, "200"); - mg_close_connection(client_conn); - - - /* Close the server */ - g_ctx = NULL; - test_mg_stop(ctx); - mark_point(); -} -END_TEST - - -START_TEST(test_http_auth) -{ -#if !defined(NO_FILES) - const char *OPTIONS[] = { - "document_root", - ".", - "listening_ports", - "8080", -#if !defined(NO_CACHING) - "static_file_max_age", - "0", -#endif - NULL, - }; - struct mg_context *ctx; - struct mg_connection *client_conn; - char client_err[256], nonce[256]; - const struct mg_request_info *client_ri; - int client_res; - FILE *f; - const char *passwd_file = ".htpasswd"; - const char *test_file = "test_http_auth.test_file.txt"; - const char *test_content = "test_http_auth test_file content"; - const char *domain; - const char *doc_root; - const char *auth_request; - const char *str; - size_t len; - int i; - char HA1[256], HA2[256], HA[256]; - char HA1_md5_buf[33], HA2_md5_buf[33], HA_md5_buf[33]; - char *HA1_md5_ret, *HA2_md5_ret, *HA_md5_ret; - const char *nc = "00000001"; - const char *cnonce = "6789ABCD"; - - - /* Start with default options */ - ctx = test_mg_start(NULL, NULL, OPTIONS); - - ck_assert(ctx != NULL); - domain = mg_get_option(ctx, "authentication_domain"); - ck_assert(domain != NULL); - len = strlen(domain); - ck_assert_uint_gt(len, 0); - ck_assert_uint_lt(len, 64); - doc_root = mg_get_option(ctx, "document_root"); - ck_assert_str_eq(doc_root, "."); - - /* Create a default file in the document root */ - f = fopen(test_file, "w"); - if (f) { - fprintf(f, "%s", test_content); - fclose(f); - } else { - ck_abort_msg("Cannot create file %s", test_file); - } - - (void)remove(passwd_file); - - /* Read file before a .htpasswd file has been created */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET /%s HTTP/1.0\r\n\r\n", test_file); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - - ck_assert_str_eq(client_ri->uri, "200"); - client_res = (int)mg_read(client_conn, client_err, sizeof(client_err)); - ck_assert_int_gt(client_res, 0); - ck_assert_int_le(client_res, sizeof(client_err)); - ck_assert_str_eq(client_err, test_content); - mg_close_connection(client_conn); - - test_sleep(1); - - /* Create a .htpasswd file */ - client_res = mg_modify_passwords_file(passwd_file, domain, "user", "pass"); - ck_assert_int_eq(client_res, 1); - - client_res = mg_modify_passwords_file(NULL, domain, "user", "pass"); - ck_assert_int_eq(client_res, 0); /* Filename is required */ - - test_sleep(1); - - /* Repeat test after .htpasswd is created */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET /%s HTTP/1.0\r\n\r\n", test_file); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - - ck_assert_str_eq(client_ri->uri, "401"); - - auth_request = NULL; - for (i = 0; i < client_ri->num_headers; i++) { - if (!mg_strcasecmp(client_ri->http_headers[i].name, - "WWW-Authenticate")) { - ck_assert_ptr_eq(auth_request, NULL); - auth_request = client_ri->http_headers[i].value; - ck_assert_ptr_ne(auth_request, NULL); - } - } - ck_assert_ptr_ne(auth_request, NULL); - str = "Digest qop=\"auth\", realm=\""; - len = strlen(str); - ck_assert(!mg_strncasecmp(auth_request, str, len)); - ck_assert(!strncmp(auth_request + len, domain, strlen(domain))); - len += strlen(domain); - str = "\", nonce=\""; - ck_assert(!strncmp(auth_request + len, str, strlen(str))); - len += strlen(str); - str = strchr(auth_request + len, '\"'); - ck_assert_ptr_ne(str, NULL); - ck_assert_ptr_ne(str, auth_request + len); - /* nonce is from including (auth_request + len) to excluding (str) */ - ck_assert_int_gt((ptrdiff_t)(str) - (ptrdiff_t)(auth_request + len), 0); - ck_assert_int_lt((ptrdiff_t)(str) - (ptrdiff_t)(auth_request + len), - (ptrdiff_t)sizeof(nonce)); - memset(nonce, 0, sizeof(nonce)); - memcpy(nonce, - auth_request + len, - (size_t)((ptrdiff_t)(str) - (ptrdiff_t)(auth_request + len))); - memset(HA1, 0, sizeof(HA1)); - memset(HA2, 0, sizeof(HA2)); - memset(HA, 0, sizeof(HA)); - memset(HA1_md5_buf, 0, sizeof(HA1_md5_buf)); - memset(HA2_md5_buf, 0, sizeof(HA2_md5_buf)); - memset(HA_md5_buf, 0, sizeof(HA_md5_buf)); - - sprintf(HA1, "%s:%s:%s", "user", domain, "pass"); - sprintf(HA2, "%s:/%s", "GET", test_file); - HA1_md5_ret = mg_md5(HA1_md5_buf, HA1, NULL); - HA2_md5_ret = mg_md5(HA2_md5_buf, HA2, NULL); - - ck_assert_ptr_eq(HA1_md5_ret, HA1_md5_buf); - ck_assert_ptr_eq(HA2_md5_ret, HA2_md5_buf); - - HA_md5_ret = mg_md5(HA_md5_buf, "user", ":", domain, ":", "pass", NULL); - ck_assert_ptr_eq(HA_md5_ret, HA_md5_buf); - ck_assert_str_eq(HA1_md5_ret, HA_md5_buf); - - HA_md5_ret = mg_md5(HA_md5_buf, "GET", ":", "/", test_file, NULL); - ck_assert_ptr_eq(HA_md5_ret, HA_md5_buf); - ck_assert_str_eq(HA2_md5_ret, HA_md5_buf); - - HA_md5_ret = mg_md5(HA_md5_buf, - HA1_md5_buf, - ":", - nonce, - ":", - nc, - ":", - cnonce, - ":", - "auth", - ":", - HA2_md5_buf, - NULL); - ck_assert_ptr_eq(HA_md5_ret, HA_md5_buf); - - /* Retry with Authorization */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET /%s HTTP/1.0\r\n", test_file); - mg_printf(client_conn, - "Authorization: Digest " - "username=\"%s\", " - "realm=\"%s\", " - "nonce=\"%s\", " - "uri=\"/%s\", " - "qop=auth, " - "nc=%s, " - "cnonce=\"%s\", " - "response=\"%s\"\r\n\r\n", - "user", - domain, - nonce, - test_file, - nc, - cnonce, - HA_md5_buf); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - - ck_assert_str_eq(client_ri->uri, "200"); - client_res = (int)mg_read(client_conn, client_err, sizeof(client_err)); - ck_assert_int_gt(client_res, 0); - ck_assert_int_le(client_res, sizeof(client_err)); - ck_assert_str_eq(client_err, test_content); - mg_close_connection(client_conn); - - test_sleep(1); - - - /* Remove the user from the .htpasswd file again */ - client_res = mg_modify_passwords_file(passwd_file, domain, "user", NULL); - ck_assert_int_eq(client_res, 1); - - test_sleep(1); - - - /* Try to access the file again. Expected: 401 error */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, ""); - mg_printf(client_conn, "GET /%s HTTP/1.0\r\n\r\n", test_file); - client_res = - mg_get_response(client_conn, client_err, sizeof(client_err), 10000); - ck_assert_int_ge(client_res, 0); - ck_assert_str_eq(client_err, ""); - client_ri = mg_get_request_info(client_conn); - ck_assert(client_ri != NULL); - - ck_assert_str_eq(client_ri->uri, "401"); - mg_close_connection(client_conn); - - test_sleep(1); - - - /* Now remove the password file */ - (void)remove(passwd_file); - test_sleep(1); - - - /* Access to the file must work like before */ - memset(client_err, 0, sizeof(client_err)); - client_conn = - mg_connect_client("127.0.0.1", 8080, 0, client_err, sizeof(client_err)); - ck_assert(client_conn != NULL); - ck_assert_str_eq(client_err, "")
<TRUNCATED>
