| /* |
| This file is part of libmicrohttpd |
| Copyright (C) 2014-2022 Evgeny Grin (Karlson2k) |
| Copyright (C) 2007, 2009, 2011 Christian Grothoff |
| |
| libmicrohttpd is free software; you can redistribute it and/or modify |
| it under the terms of the GNU General Public License as published |
| by the Free Software Foundation; either version 2, or (at your |
| option) any later version. |
| |
| libmicrohttpd is distributed in the hope that it will be useful, but |
| WITHOUT ANY WARRANTY; without even the implied warranty of |
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| General Public License for more details. |
| |
| You should have received a copy of the GNU General Public License |
| along with libmicrohttpd; see the file COPYING. If not, write to the |
| Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, |
| Boston, MA 02110-1301, USA. |
| */ |
| /** |
| * @file test_get_close_keep_alive.c |
| * @brief Testcase for libmicrohttpd "Close" and "Keep-Alive" connection. |
| * @details Testcases for testing of MHD automatic choice between "Close" and |
| * "Keep-Alive" connections. Also tested selected HTTP version and |
| * "Connection:" headers. |
| * @author Karlson2k (Evgeny Grin) |
| * @author Christian Grothoff |
| */ |
| #include "MHD_config.h" |
| #include "platform.h" |
| #include <curl/curl.h> |
| #include <microhttpd.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <time.h> |
| #include "mhd_has_in_name.h" |
| #include "mhd_has_param.h" |
| #include "mhd_sockets.h" /* only macros used */ |
| |
| #ifdef HAVE_STRINGS_H |
| #include <strings.h> |
| #endif /* HAVE_STRINGS_H */ |
| |
| #ifdef _WIN32 |
| #ifndef WIN32_LEAN_AND_MEAN |
| #define WIN32_LEAN_AND_MEAN 1 |
| #endif /* !WIN32_LEAN_AND_MEAN */ |
| #include <windows.h> |
| #endif |
| |
| #ifndef WINDOWS |
| #include <unistd.h> |
| #include <sys/socket.h> |
| #endif |
| |
| #ifdef HAVE_LIMITS_H |
| #include <limits.h> |
| #endif /* HAVE_LIMITS_H */ |
| |
| #ifndef CURL_VERSION_BITS |
| #define CURL_VERSION_BITS(x,y,z) ((x)<<16|(y)<<8|(z)) |
| #endif /* ! CURL_VERSION_BITS */ |
| #ifndef CURL_AT_LEAST_VERSION |
| #define CURL_AT_LEAST_VERSION(x,y,z) \ |
| (LIBCURL_VERSION_NUM >= CURL_VERSION_BITS(x, y, z)) |
| #endif /* ! CURL_AT_LEAST_VERSION */ |
| |
| #if defined(MHD_CPU_COUNT) && (MHD_CPU_COUNT + 0) < 2 |
| #undef MHD_CPU_COUNT |
| #endif |
| #if ! defined(MHD_CPU_COUNT) |
| #define MHD_CPU_COUNT 2 |
| #endif |
| #if MHD_CPU_COUNT > 32 |
| #undef MHD_CPU_COUNT |
| /* Limit to reasonable value */ |
| #define MHD_CPU_COUNT 32 |
| #endif /* MHD_CPU_COUNT > 32 */ |
| |
| |
| #if defined(HAVE___FUNC__) |
| #define externalErrorExit(ignore) \ |
| _externalErrorExit_func(NULL, __func__, __LINE__) |
| #define externalErrorExitDesc(errDesc) \ |
| _externalErrorExit_func(errDesc, __func__, __LINE__) |
| #define libcurlErrorExit(ignore) \ |
| _libcurlErrorExit_func(NULL, __func__, __LINE__) |
| #define libcurlErrorExitDesc(errDesc) \ |
| _libcurlErrorExit_func(errDesc, __func__, __LINE__) |
| #elif defined(HAVE___FUNCTION__) |
| #define externalErrorExit(ignore) \ |
| _externalErrorExit_func(NULL, __FUNCTION__, __LINE__) |
| #define externalErrorExitDesc(errDesc) \ |
| _externalErrorExit_func(errDesc, __FUNCTION__, __LINE__) |
| #define libcurlErrorExit(ignore) \ |
| _libcurlErrorExit_func(NULL, __FUNCTION__, __LINE__) |
| #define libcurlErrorExitDesc(errDesc) \ |
| _libcurlErrorExit_func(errDesc, __FUNCTION__, __LINE__) |
| #else |
| #define externalErrorExit(ignore) _externalErrorExit_func(NULL, NULL, __LINE__) |
| #define externalErrorExitDesc(errDesc) \ |
| _externalErrorExit_func(errDesc, NULL, __LINE__) |
| #define libcurlErrorExit(ignore) _externalErrorExit_func(NULL, NULL, __LINE__) |
| #define libcurlErrorExitDesc(errDesc) \ |
| _externalErrorExit_func(errDesc, NULL, __LINE__) |
| #endif |
| |
| |
| static void |
| _externalErrorExit_func (const char *errDesc, const char *funcName, int lineNum) |
| { |
| if ((NULL != errDesc) && (0 != errDesc[0])) |
| fprintf (stderr, "%s", errDesc); |
| else |
| fprintf (stderr, "System or external library call failed"); |
| if ((NULL != funcName) && (0 != funcName[0])) |
| fprintf (stderr, " in %s", funcName); |
| if (0 < lineNum) |
| fprintf (stderr, " at line %d", lineNum); |
| |
| fprintf (stderr, ".\nLast errno value: %d (%s)\n", (int) errno, |
| strerror (errno)); |
| #ifdef MHD_WINSOCK_SOCKETS |
| fprintf (stderr, "WSAGetLastError() value: %d\n", (int) WSAGetLastError ()); |
| #endif /* MHD_WINSOCK_SOCKETS */ |
| fflush (stderr); |
| exit (99); |
| } |
| |
| |
| static char libcurl_errbuf[CURL_ERROR_SIZE] = ""; |
| |
| static void |
| _libcurlErrorExit_func (const char *errDesc, const char *funcName, int lineNum) |
| { |
| if ((NULL != errDesc) && (0 != errDesc[0])) |
| fprintf (stderr, "%s", errDesc); |
| else |
| fprintf (stderr, "CURL library call failed"); |
| if ((NULL != funcName) && (0 != funcName[0])) |
| fprintf (stderr, " in %s", funcName); |
| if (0 < lineNum) |
| fprintf (stderr, " at line %d", lineNum); |
| |
| fprintf (stderr, ".\nLast errno value: %d (%s)\n", (int) errno, |
| strerror (errno)); |
| if (0 != libcurl_errbuf[0]) |
| fprintf (stderr, "Last libcurl error details: %s\n", libcurl_errbuf); |
| |
| fflush (stderr); |
| exit (99); |
| } |
| |
| |
| /* Could be increased to facilitate debugging */ |
| #define TIMEOUTS_VAL 5 |
| |
| #define EXPECTED_URI_BASE_PATH "/hello_world" |
| #define EXPECTED_URI_QUERY "a=%26&b=c" |
| #define EXPECTED_URI_FULL_PATH EXPECTED_URI_BASE_PATH "?" EXPECTED_URI_QUERY |
| #define HDR_CONN_CLOSE_VALUE "close" |
| #define HDR_CONN_CLOSE MHD_HTTP_HEADER_CONNECTION ": " \ |
| HDR_CONN_CLOSE_VALUE |
| #define HDR_CONN_KEEP_ALIVE_VALUE "Keep-Alive" |
| #define HDR_CONN_KEEP_ALIVE MHD_HTTP_HEADER_CONNECTION ": " \ |
| HDR_CONN_KEEP_ALIVE_VALUE |
| |
| /* Global parameters */ |
| static int oneone; /**< Use HTTP/1.1 instead of HTTP/1.0 for requests*/ |
| static int conn_close; /**< Don't use Keep-Alive */ |
| static uint16_t global_port; /**< MHD daemons listen port number */ |
| static int slow_reply = 0; /**< Slowdown MHD replies */ |
| static int ignore_response_errors = 0; /**< Do not fail test if CURL |
| returns error */ |
| static int response_timeout_val = TIMEOUTS_VAL; |
| |
| /* Current test parameters */ |
| /* Poor thread sync, but enough for the testing */ |
| static volatile int mhd_add_close; /**< Add "Connection: close" header by MHD */ |
| static volatile int mhd_set_10_cmptbl; /**< Set MHD_RF_HTTP_1_0_COMPATIBLE_STRICT response flag */ |
| static volatile int mhd_set_10_server; /**< Set MHD_RF_HTTP_1_0_SERVER response flag */ |
| static volatile int mhd_set_k_a_send; /**< Set MHD_RF_SEND_KEEP_ALIVE_HEADER response flag */ |
| |
| /* Static helper variables */ |
| static struct curl_slist *curl_close_hdr; /**< CURL "Connection: close" header */ |
| static struct curl_slist *curl_k_alive_hdr; /**< CURL "Connection: keep-alive" header */ |
| static struct curl_slist *curl_both_hdrs; /**< CURL both "Connection: keep-alive" and "close" headers */ |
| |
| static void |
| test_global_init (void) |
| { |
| libcurl_errbuf[0] = 0; |
| |
| if (0 != curl_global_init (CURL_GLOBAL_WIN32)) |
| externalErrorExit (); |
| |
| curl_close_hdr = NULL; |
| curl_close_hdr = curl_slist_append (curl_close_hdr, |
| HDR_CONN_CLOSE); |
| if (NULL == curl_close_hdr) |
| externalErrorExit (); |
| |
| curl_k_alive_hdr = NULL; |
| curl_k_alive_hdr = curl_slist_append (curl_k_alive_hdr, |
| HDR_CONN_KEEP_ALIVE); |
| if (NULL == curl_k_alive_hdr) |
| externalErrorExit (); |
| |
| curl_both_hdrs = NULL; |
| curl_both_hdrs = curl_slist_append (curl_both_hdrs, |
| HDR_CONN_KEEP_ALIVE); |
| if (NULL == curl_both_hdrs) |
| externalErrorExit (); |
| curl_both_hdrs = curl_slist_append (curl_both_hdrs, |
| HDR_CONN_CLOSE); |
| if (NULL == curl_both_hdrs) |
| externalErrorExit (); |
| } |
| |
| |
| static void |
| test_global_cleanup (void) |
| { |
| curl_slist_free_all (curl_both_hdrs); |
| curl_slist_free_all (curl_k_alive_hdr); |
| curl_slist_free_all (curl_close_hdr); |
| |
| curl_global_cleanup (); |
| } |
| |
| |
| struct headers_check_result |
| { |
| int found_http11; |
| int found_http10; |
| int found_conn_close; |
| int found_conn_keep_alive; |
| }; |
| |
| static size_t |
| lcurl_hdr_callback (char *buffer, size_t size, size_t nitems, |
| void *userdata) |
| { |
| const size_t data_size = size * nitems; |
| struct headers_check_result *check_res = |
| (struct headers_check_result *) userdata; |
| |
| if ((strlen (MHD_HTTP_VERSION_1_1) < data_size) && |
| (0 == memcmp (MHD_HTTP_VERSION_1_1, buffer, |
| strlen (MHD_HTTP_VERSION_1_1)))) |
| check_res->found_http11 = 1; |
| else if ((strlen (MHD_HTTP_VERSION_1_0) < data_size) && |
| (0 == memcmp (MHD_HTTP_VERSION_1_0, buffer, |
| strlen (MHD_HTTP_VERSION_1_0)))) |
| check_res->found_http10 = 1; |
| else if ((data_size == strlen (HDR_CONN_CLOSE) + 2) && |
| (0 == memcmp (buffer, HDR_CONN_CLOSE "\r\n", data_size))) |
| check_res->found_conn_close = 1; |
| else if ((data_size == strlen (HDR_CONN_KEEP_ALIVE) + 2) && |
| (0 == memcmp (buffer, HDR_CONN_KEEP_ALIVE "\r\n", data_size))) |
| check_res->found_conn_keep_alive = 1; |
| |
| return data_size; |
| } |
| |
| |
| struct CBC |
| { |
| char *buf; |
| size_t pos; |
| size_t size; |
| }; |
| |
| |
| static size_t |
| copyBuffer (void *ptr, size_t size, size_t nmemb, void *ctx) |
| { |
| struct CBC *cbc = ctx; |
| |
| if (cbc->pos + size * nmemb > cbc->size) |
| externalErrorExit (); /* overflow */ |
| memcpy (&cbc->buf[cbc->pos], ptr, size * nmemb); |
| cbc->pos += size * nmemb; |
| return size * nmemb; |
| } |
| |
| |
| static void * |
| log_cb (void *cls, |
| const char *uri, |
| struct MHD_Connection *con) |
| { |
| (void) cls; |
| (void) con; |
| if (0 != strcmp (uri, |
| EXPECTED_URI_FULL_PATH)) |
| { |
| fprintf (stderr, |
| "Wrong URI: `%s', line: %d\n", |
| uri, __LINE__); |
| exit (22); |
| } |
| return NULL; |
| } |
| |
| |
| static enum MHD_Result |
| ahc_echo (void *cls, |
| struct MHD_Connection *connection, |
| const char *url, |
| const char *method, |
| const char *version, |
| const char *upload_data, size_t *upload_data_size, |
| void **req_cls) |
| { |
| static int ptr; |
| struct MHD_Response *response; |
| enum MHD_Result ret; |
| (void) cls; |
| (void) version; |
| (void) upload_data; |
| (void) upload_data_size; /* Unused. Silence compiler warning. */ |
| |
| if (0 != strcmp (MHD_HTTP_METHOD_GET, method)) |
| return MHD_NO; /* unexpected method */ |
| if (&ptr != *req_cls) |
| { |
| *req_cls = &ptr; |
| return MHD_YES; |
| } |
| *req_cls = NULL; |
| if (slow_reply) |
| usleep (200000); |
| |
| response = MHD_create_response_from_buffer_copy (strlen (url), |
| (const void *) url); |
| if (NULL == response) |
| { |
| fprintf (stderr, "Failed to create response. Line: %d\n", __LINE__); |
| exit (19); |
| } |
| if (mhd_add_close) |
| { |
| if (MHD_YES != MHD_add_response_header (response, |
| MHD_HTTP_HEADER_CONNECTION, |
| HDR_CONN_CLOSE_VALUE)) |
| { |
| fprintf (stderr, "Failed to add header. Line: %d\n", __LINE__); |
| exit (19); |
| } |
| } |
| if (MHD_YES != MHD_set_response_options (response, |
| (mhd_set_10_cmptbl ? |
| MHD_RF_HTTP_1_0_COMPATIBLE_STRICT |
| : 0) |
| | (mhd_set_10_server ? |
| MHD_RF_HTTP_1_0_SERVER |
| : 0) |
| | (mhd_set_k_a_send ? |
| MHD_RF_SEND_KEEP_ALIVE_HEADER |
| : 0), MHD_RO_END)) |
| { |
| fprintf (stderr, "Failed to set response flags. Line: %d\n", __LINE__); |
| exit (19); |
| } |
| |
| ret = MHD_queue_response (connection, |
| MHD_HTTP_OK, |
| response); |
| MHD_destroy_response (response); |
| if (MHD_YES != ret) |
| { |
| fprintf (stderr, "Failed to queue response. Line: %d\n", __LINE__); |
| exit (19); |
| } |
| return ret; |
| } |
| |
| |
| struct curlQueryParams |
| { |
| /* Destination path for CURL query */ |
| const char *queryPath; |
| |
| /* Destination port for CURL query */ |
| uint16_t queryPort; |
| |
| /* CURL query result error flag */ |
| volatile unsigned int queryError; |
| }; |
| |
| |
| static CURL * |
| curlEasyInitForTest (const char *queryPath, uint16_t port, struct CBC *pcbc, |
| struct headers_check_result *hdr_chk_result, |
| int add_hdr_close, int add_hdr_k_alive) |
| { |
| CURL *c; |
| |
| c = curl_easy_init (); |
| if (NULL == c) |
| { |
| fprintf (stderr, "curl_easy_init() failed.\n"); |
| externalErrorExit (); |
| } |
| if ((CURLE_OK != curl_easy_setopt (c, CURLOPT_NOSIGNAL, 1L)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_URL, queryPath)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_PORT, (long) port)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, |
| ©Buffer)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_WRITEDATA, pcbc)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_CONNECTTIMEOUT, |
| (long) response_timeout_val)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_TIMEOUT, |
| (long) response_timeout_val)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_ERRORBUFFER, |
| libcurl_errbuf)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_HEADERFUNCTION, |
| lcurl_hdr_callback)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_HEADERDATA, |
| hdr_chk_result)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_FAILONERROR, 1L)) || |
| (CURLE_OK != curl_easy_setopt (c, CURLOPT_HTTP_VERSION, |
| (oneone) ? |
| CURL_HTTP_VERSION_1_1 : |
| CURL_HTTP_VERSION_1_0))) |
| { |
| fprintf (stderr, "curl_easy_setopt() failed.\n"); |
| externalErrorExit (); |
| } |
| if (add_hdr_close && add_hdr_k_alive) |
| { /* This combination is actually incorrect */ |
| if (CURLE_OK != curl_easy_setopt (c, CURLOPT_HTTPHEADER, curl_both_hdrs)) |
| { |
| fprintf (stderr, "Set libcurl HTTP header failed.\n"); |
| externalErrorExit (); |
| } |
| } |
| else if (add_hdr_close) |
| { |
| if (CURLE_OK != curl_easy_setopt (c, CURLOPT_HTTPHEADER, curl_close_hdr)) |
| { |
| fprintf (stderr, "Set libcurl HTTP header failed.\n"); |
| externalErrorExit (); |
| } |
| } |
| else if (add_hdr_k_alive) |
| { |
| if (CURLE_OK != curl_easy_setopt (c, CURLOPT_HTTPHEADER, curl_k_alive_hdr)) |
| { |
| fprintf (stderr, "Set libcurl HTTP header failed.\n"); |
| externalErrorExit (); |
| } |
| } |
| |
| return c; |
| } |
| |
| |
| static void |
| print_test_params (int add_hdr_close, |
| int add_hdr_k_alive) |
| { |
| fprintf (stderr, "Request HTTP/%s| ", oneone ? "1.1" : "1.0"); |
| fprintf (stderr, "Connection must be: %s| ", |
| conn_close ? "close" : "keep-alive"); |
| fprintf (stderr, "Request \"close\": %s| ", |
| add_hdr_close ? " used" : "NOT used"); |
| fprintf (stderr, "Request \"keep-alive\": %s| ", |
| add_hdr_k_alive ? " used" : "NOT used"); |
| fprintf (stderr, "MHD response \"close\": %s| ", |
| mhd_add_close ? " used" : "NOT used"); |
| fprintf (stderr, "MHD response 1.0 strict compatible: %s| ", |
| mhd_set_10_cmptbl ? "yes" : " NO"); |
| fprintf (stderr, "MHD response 1.0 server: %s| ", |
| mhd_set_10_server ? "yes" : " NO"); |
| fprintf (stderr, "MHD response send \"Keep-Alive\": %s|", |
| mhd_set_k_a_send ? "yes" : " NO"); |
| fprintf (stderr, "\n*** "); |
| } |
| |
| |
| static CURLcode |
| performQueryExternal (struct MHD_Daemon *d, CURL *c) |
| { |
| CURLM *multi; |
| time_t start; |
| struct timeval tv; |
| CURLcode ret; |
| |
| ret = CURLE_FAILED_INIT; /* will be replaced with real result */ |
| multi = NULL; |
| multi = curl_multi_init (); |
| if (multi == NULL) |
| { |
| fprintf (stderr, "curl_multi_init() failed.\n"); |
| externalErrorExit (); |
| } |
| if (CURLM_OK != curl_multi_add_handle (multi, c)) |
| { |
| fprintf (stderr, "curl_multi_add_handle() failed.\n"); |
| externalErrorExit (); |
| } |
| |
| start = time (NULL); |
| while (time (NULL) - start <= TIMEOUTS_VAL) |
| { |
| fd_set rs; |
| fd_set ws; |
| fd_set es; |
| MHD_socket maxMhdSk; |
| int maxCurlSk; |
| int running; |
| |
| maxMhdSk = MHD_INVALID_SOCKET; |
| maxCurlSk = -1; |
| FD_ZERO (&rs); |
| FD_ZERO (&ws); |
| FD_ZERO (&es); |
| if (NULL != multi) |
| { |
| curl_multi_perform (multi, &running); |
| if (0 == running) |
| { |
| struct CURLMsg *msg; |
| int msgLeft; |
| int totalMsgs = 0; |
| do |
| { |
| msg = curl_multi_info_read (multi, &msgLeft); |
| if (NULL == msg) |
| { |
| fprintf (stderr, "curl_multi_info_read failed, NULL returned.\n"); |
| externalErrorExit (); |
| } |
| totalMsgs++; |
| if (CURLMSG_DONE == msg->msg) |
| ret = msg->data.result; |
| } while (msgLeft > 0); |
| if (1 != totalMsgs) |
| { |
| fprintf (stderr, |
| "curl_multi_info_read returned wrong " |
| "number of results (%d).\n", |
| totalMsgs); |
| externalErrorExit (); |
| } |
| curl_multi_remove_handle (multi, c); |
| curl_multi_cleanup (multi); |
| multi = NULL; |
| } |
| else |
| { |
| if (CURLM_OK != curl_multi_fdset (multi, &rs, &ws, &es, &maxCurlSk)) |
| { |
| fprintf (stderr, "curl_multi_fdset() failed.\n"); |
| externalErrorExit (); |
| } |
| } |
| } |
| if (NULL == multi) |
| { /* libcurl has finished, check whether MHD still needs to perform cleanup */ |
| if (0 != MHD_get_timeout64s (d)) |
| break; /* MHD finished as well */ |
| } |
| if (MHD_YES != MHD_get_fdset (d, &rs, &ws, &es, &maxMhdSk)) |
| { |
| fprintf (stderr, "MHD_get_fdset() failed. Line: %d\n", __LINE__); |
| exit (11); |
| break; |
| } |
| tv.tv_sec = 0; |
| tv.tv_usec = 1000; |
| #ifdef MHD_POSIX_SOCKETS |
| if (maxMhdSk > maxCurlSk) |
| maxCurlSk = maxMhdSk; |
| #endif /* MHD_POSIX_SOCKETS */ |
| if (-1 == select (maxCurlSk + 1, &rs, &ws, &es, &tv)) |
| { |
| #ifdef MHD_POSIX_SOCKETS |
| if (EINTR != errno) |
| { |
| fprintf (stderr, "Unexpected select() error: %d. Line: %d\n", |
| (int) errno, __LINE__); |
| fflush (stderr); |
| exit (99); |
| } |
| #else |
| if ((WSAEINVAL != WSAGetLastError ()) || |
| (0 != rs.fd_count) || (0 != ws.fd_count) || (0 != es.fd_count) ) |
| { |
| fprintf (stderr, "Unexpected select() error: %d. Line: %d\n", |
| (int) WSAGetLastError (), __LINE__); |
| fflush (stderr); |
| exit (99); |
| } |
| Sleep (1); |
| #endif |
| } |
| if (MHD_YES != MHD_run_from_select (d, &rs, &ws, &es)) |
| { |
| fprintf (stderr, "MHD_run_from_select() failed. Line: %d\n", __LINE__); |
| exit (11); |
| } |
| } |
| |
| return ret; |
| } |
| |
| |
| static unsigned int |
| getMhdActiveConnections (struct MHD_Daemon *d) |
| { |
| const union MHD_DaemonInfo *dinfo; |
| /* The next method is unreliable unless it's known that no |
| * connections are started or finished in parallel */ |
| dinfo = MHD_get_daemon_info (d, MHD_DAEMON_INFO_CURRENT_CONNECTIONS); |
| if (NULL == dinfo) |
| { |
| fprintf (stderr, "MHD_get_daemon_info() failed.\n"); |
| abort (); |
| } |
| return dinfo->num_connections; |
| } |
| |
| |
| static unsigned int |
| doCurlQueryInThread (struct MHD_Daemon *d, |
| struct curlQueryParams *p, |
| int add_hdr_close, |
| int add_hdr_k_alive) |
| { |
| const union MHD_DaemonInfo *dinfo; |
| CURL *c; |
| char buf[2048]; |
| struct CBC cbc; |
| struct headers_check_result hdr_res; |
| CURLcode errornum; |
| int use_external_poll; |
| |
| dinfo = MHD_get_daemon_info (d, MHD_DAEMON_INFO_FLAGS); |
| if (NULL == dinfo) |
| { |
| fprintf (stderr, "MHD_get_daemon_info() failed.\n"); |
| abort (); |
| } |
| use_external_poll = (0 == (dinfo->flags |
| & MHD_USE_INTERNAL_POLLING_THREAD)); |
| |
| if (NULL == p->queryPath) |
| abort (); |
| |
| if (0 == p->queryPort) |
| abort (); |
| |
| cbc.buf = buf; |
| cbc.size = sizeof(buf); |
| cbc.pos = 0; |
| |
| hdr_res.found_http11 = 0; |
| hdr_res.found_http10 = 0; |
| hdr_res.found_conn_close = 0; |
| hdr_res.found_conn_keep_alive = 0; |
| |
| c = curlEasyInitForTest (p->queryPath, p->queryPort, &cbc, &hdr_res, |
| add_hdr_close, add_hdr_k_alive); |
| |
| if (! use_external_poll) |
| errornum = curl_easy_perform (c); |
| else |
| errornum = performQueryExternal (d, c); |
| if (ignore_response_errors) |
| { |
| p->queryError = 0; |
| curl_easy_cleanup (c); |
| |
| return p->queryError; |
| } |
| if (CURLE_OK != errornum) |
| { |
| p->queryError = 1; |
| fprintf (stderr, |
| "libcurl query failed: `%s'\n", |
| curl_easy_strerror (errornum)); |
| libcurlErrorExit (); |
| } |
| else |
| { |
| if (cbc.pos != strlen (EXPECTED_URI_BASE_PATH)) |
| { |
| fprintf (stderr, "curl reports wrong size of MHD reply body data.\n"); |
| p->queryError = 1; |
| } |
| else if (0 != strncmp (EXPECTED_URI_BASE_PATH, cbc.buf, |
| strlen (EXPECTED_URI_BASE_PATH))) |
| { |
| fprintf (stderr, "curl reports wrong MHD reply body data.\n"); |
| p->queryError = 1; |
| } |
| else |
| p->queryError = 0; |
| } |
| |
| if (! hdr_res.found_http11 && ! hdr_res.found_http10) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "No know HTTP versions were found in the " |
| "reply header. Line: %d\n", __LINE__); |
| exit (24); |
| } |
| else if (hdr_res.found_http11 && hdr_res.found_http10) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "Both HTTP/1.1 and HTTP/1.0 were found in the " |
| "reply header. Line: %d\n", __LINE__); |
| exit (24); |
| } |
| |
| if (conn_close) |
| { |
| if (! hdr_res.found_conn_close) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "\"Connection: close\" was not found in" |
| " MHD reply headers.\n"); |
| p->queryError |= 2; |
| } |
| if (hdr_res.found_conn_keep_alive) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "\"Connection: keep-alive\" was found in" |
| " MHD reply headers.\n"); |
| p->queryError |= 2; |
| } |
| if (use_external_poll) |
| { /* The number of MHD connection can queried only with external poll. |
| * otherwise it creates a race condition. */ |
| if (0 != getMhdActiveConnections (d)) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "MHD still has active connection " |
| "after response has been sent.\n"); |
| p->queryError |= 2; |
| } |
| } |
| } |
| else |
| { /* Keep-Alive */ |
| if (! oneone || mhd_set_10_server || mhd_set_k_a_send) |
| { /* Should have "Connection: Keep-Alive" */ |
| if (! hdr_res.found_conn_keep_alive) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "\"Connection: keep-alive\" was not found in" |
| " MHD reply headers.\n"); |
| p->queryError |= 2; |
| } |
| } |
| else |
| { /* Should NOT have "Connection: Keep-Alive" */ |
| if (hdr_res.found_conn_keep_alive) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "\"Connection: keep-alive\" was found in" |
| " MHD reply headers.\n"); |
| p->queryError |= 2; |
| } |
| } |
| if (hdr_res.found_conn_close) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "\"Connection: close\" was found in" |
| " MHD reply headers.\n"); |
| p->queryError |= 2; |
| } |
| if (use_external_poll) |
| { /* The number of MHD connection can be queried only with external poll. |
| * otherwise it creates a race condition. */ |
| unsigned int num_conn = getMhdActiveConnections (d); |
| if (0 == num_conn) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "MHD has no active connection " |
| "after response has been sent.\n"); |
| p->queryError |= 2; |
| } |
| else if (1 != num_conn) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "MHD has wrong number of active connection (%u) " |
| "after response has been sent. Line: %d\n", num_conn, |
| __LINE__); |
| exit (23); |
| } |
| } |
| } |
| |
| #if defined(CURL_AT_LEAST_VERSION) && CURL_AT_LEAST_VERSION (7, 45, 0) |
| if (! use_external_poll) |
| { |
| /* libcurl closes connection socket with curl_multi_remove_handle () / |
| curl_multi_cleanup() */ |
| curl_socket_t curl_sckt; |
| if (CURLE_OK != curl_easy_getinfo (c, CURLINFO_ACTIVESOCKET, &curl_sckt)) |
| { |
| fprintf (stderr, |
| "Failed to get libcurl active socket.\n"); |
| libcurlErrorExit (); |
| } |
| |
| if (conn_close && (CURL_SOCKET_BAD != curl_sckt)) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "libcurl still has active connection " |
| "after performing the test query.\n"); |
| p->queryError |= 2; |
| } |
| else if (! conn_close && (CURL_SOCKET_BAD == curl_sckt)) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "libcurl has no active connection " |
| "after performing the test query.\n"); |
| p->queryError |= 2; |
| } |
| } |
| #endif |
| |
| if (! mhd_set_10_server) |
| { /* Response must be HTTP/1.1 */ |
| if (hdr_res.found_http10) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "Reply has HTTP/1.0 version, while it " |
| "must be HTTP/1.1.\n"); |
| p->queryError |= 4; |
| } |
| } |
| else |
| { /* Response must be HTTP/1.0 */ |
| if (hdr_res.found_http11) |
| { |
| print_test_params (add_hdr_close, add_hdr_k_alive); |
| fprintf (stderr, "Reply has HTTP/1.1 version, while it " |
| "must be HTTP/1.0.\n"); |
| p->queryError |= 4; |
| } |
| } |
| curl_easy_cleanup (c); |
| |
| return p->queryError; |
| } |
| |
| |
| /* Perform test queries and shut down MHD daemon */ |
| static unsigned int |
| performTestQueries (struct MHD_Daemon *d, uint16_t d_port) |
| { |
| struct curlQueryParams qParam; |
| unsigned int ret = 0; /* Return value */ |
| int i = 0; |
| /* masks */ |
| const int m_mhd_close = 1 << (i++); |
| const int m_10_cmptbl = 1 << (i++); |
| const int m_10_server = 1 << (i++); |
| const int m_k_a_send = 1 << (i++); |
| const int m_client_close = 1 << (i++); |
| const int m_client_k_alive = 1 << (i++); |
| |
| qParam.queryPath = "http://127.0.0.1" EXPECTED_URI_FULL_PATH; |
| qParam.queryPort = d_port; /* Connect to the daemon */ |
| |
| for (i = (1 << i) - 1; 0 <= i; i--) |
| { |
| const int f_mhd_close = (0 == (i & m_mhd_close)); /**< Use MHD "close" header */ |
| const int f_10_cmptbl = (0 == (i & m_10_cmptbl)); /**< Use MHD MHD_RF_HTTP_1_0_COMPATIBLE_STRICT flag */ |
| const int f_10_server = (0 == (i & m_10_server)); /**< Use MHD MHD_RF_HTTP_1_0_SERVER flag */ |
| const int f_k_a_send = (0 == (i & m_k_a_send)); /**< Use MHD MHD_RF_SEND_KEEP_ALIVE_HEADER flag */ |
| const int f_client_close = (0 == (i & m_client_close)); /**< Use libcurl "close" header */ |
| const int f_client_k_alive = (0 == (i & m_client_k_alive)); /**< Use libcurl "Keep-Alive" header */ |
| int res_close; /**< Indicate the result of the test query should be closed connection */ |
| |
| if (f_mhd_close) /* Connection with server's "close" header must be always closed */ |
| res_close = 1; |
| else if (f_client_close) /* Connection with client's "close" header must be always closed */ |
| res_close = 1; |
| else if (f_10_cmptbl) /* Connection in strict HTTP/1.0 compatible mode must be always closed */ |
| res_close = 1; |
| else if (! oneone || f_10_server) |
| /* With HTTP/1.0 client or server connection must be close unless client use "keep-alive" header */ |
| res_close = ! f_client_k_alive; |
| else |
| res_close = 0; /* HTTP/1.1 is "keep-alive" by default */ |
| |
| if ((! ! res_close) != (! ! conn_close)) |
| continue; /* Another mode is in test currently */ |
| |
| mhd_add_close = f_mhd_close; |
| mhd_set_10_cmptbl = f_10_cmptbl; |
| mhd_set_10_server = f_10_server; |
| mhd_set_k_a_send = f_k_a_send; |
| |
| ret <<= 3; /* Remember errors for each step */ |
| ret |= doCurlQueryInThread (d, &qParam, f_client_close, f_client_k_alive); |
| } |
| |
| MHD_stop_daemon (d); |
| |
| return ret; |
| } |
| |
| |
| enum testMhdThreadsType |
| { |
| testMhdThreadExternal = 0, |
| testMhdThreadInternal = MHD_USE_INTERNAL_POLLING_THREAD, |
| testMhdThreadInternalPerConnection = MHD_USE_THREAD_PER_CONNECTION |
| | MHD_USE_INTERNAL_POLLING_THREAD, |
| testMhdThreadInternalPool |
| }; |
| |
| enum testMhdPollType |
| { |
| testMhdPollBySelect = 0, |
| testMhdPollByPoll = MHD_USE_POLL, |
| testMhdPollByEpoll = MHD_USE_EPOLL, |
| testMhdPollAuto = MHD_USE_AUTO |
| }; |
| |
| /* Get number of threads for thread pool depending |
| * on used poll function and test type. */ |
| static unsigned int |
| testNumThreadsForPool (enum testMhdPollType pollType) |
| { |
| unsigned int numThreads = MHD_CPU_COUNT; |
| (void) pollType; /* Don't care about pollType for this test */ |
| return numThreads; /* No practical limit for non-cleanup test */ |
| } |
| |
| |
| static struct MHD_Daemon * |
| startTestMhdDaemon (enum testMhdThreadsType thrType, |
| enum testMhdPollType pollType, uint16_t *pport) |
| { |
| struct MHD_Daemon *d; |
| const union MHD_DaemonInfo *dinfo; |
| |
| if ( (0 == *pport) && |
| (MHD_NO == MHD_is_feature_supported (MHD_FEATURE_AUTODETECT_BIND_PORT)) ) |
| { |
| *pport = 4050; |
| if (oneone) |
| *pport += 1; |
| if (! conn_close) |
| *pport += 2; |
| } |
| |
| if (testMhdThreadInternalPool != thrType) |
| d = MHD_start_daemon (((unsigned int) thrType) | ((unsigned int) pollType) |
| | MHD_USE_ERROR_LOG, |
| *pport, NULL, NULL, |
| &ahc_echo, NULL, |
| MHD_OPTION_URI_LOG_CALLBACK, &log_cb, NULL, |
| MHD_OPTION_END); |
| else |
| d = MHD_start_daemon (MHD_USE_INTERNAL_POLLING_THREAD |
| | ((unsigned int) pollType) |
| | MHD_USE_ERROR_LOG, |
| *pport, NULL, NULL, |
| &ahc_echo, NULL, |
| MHD_OPTION_THREAD_POOL_SIZE, |
| testNumThreadsForPool (pollType), |
| MHD_OPTION_URI_LOG_CALLBACK, &log_cb, NULL, |
| MHD_OPTION_END); |
| |
| if (NULL == d) |
| { |
| fprintf (stderr, "Failed to start MHD daemon, errno=%d.\n", errno); |
| abort (); |
| } |
| |
| if (0 == *pport) |
| { |
| dinfo = MHD_get_daemon_info (d, MHD_DAEMON_INFO_BIND_PORT); |
| if ((NULL == dinfo) || (0 == dinfo->port) ) |
| { |
| fprintf (stderr, "MHD_get_daemon_info() failed.\n"); |
| abort (); |
| } |
| *pport = dinfo->port; |
| if (0 == global_port) |
| global_port = *pport; /* Reuse the same port for all tests */ |
| } |
| |
| return d; |
| } |
| |
| |
| /* Test runners */ |
| |
| |
| static unsigned int |
| testExternalGet (void) |
| { |
| struct MHD_Daemon *d; |
| uint16_t d_port = global_port; /* Daemon's port */ |
| |
| d = startTestMhdDaemon (testMhdThreadExternal, testMhdPollBySelect, &d_port); |
| |
| return performTestQueries (d, d_port); |
| } |
| |
| |
| static unsigned int |
| testInternalGet (enum testMhdPollType pollType) |
| { |
| struct MHD_Daemon *d; |
| uint16_t d_port = global_port; /* Daemon's port */ |
| |
| d = startTestMhdDaemon (testMhdThreadInternal, pollType, |
| &d_port); |
| |
| return performTestQueries (d, d_port); |
| } |
| |
| |
| static unsigned int |
| testMultithreadedGet (enum testMhdPollType pollType) |
| { |
| struct MHD_Daemon *d; |
| uint16_t d_port = global_port; /* Daemon's port */ |
| |
| d = startTestMhdDaemon (testMhdThreadInternalPerConnection, pollType, |
| &d_port); |
| return performTestQueries (d, d_port); |
| } |
| |
| |
| static unsigned int |
| testMultithreadedPoolGet (enum testMhdPollType pollType) |
| { |
| struct MHD_Daemon *d; |
| uint16_t d_port = global_port; /* Daemon's port */ |
| |
| d = startTestMhdDaemon (testMhdThreadInternalPool, pollType, |
| &d_port); |
| return performTestQueries (d, d_port); |
| } |
| |
| |
| int |
| main (int argc, char *const *argv) |
| { |
| unsigned int errorCount = 0; |
| unsigned int test_result = 0; |
| int verbose = 0; |
| |
| if ((NULL == argv) || (0 == argv[0])) |
| return 99; |
| oneone = ! has_in_name (argv[0], "10"); |
| conn_close = has_in_name (argv[0], "_close"); |
| if (! conn_close && ! has_in_name (argv[0], "_keep_alive")) |
| return 99; |
| verbose = ! (has_param (argc, argv, "-q") || |
| has_param (argc, argv, "--quiet") || |
| has_param (argc, argv, "-s") || |
| has_param (argc, argv, "--silent")); |
| |
| test_global_init (); |
| |
| /* Could be set to non-zero value to enforce using specific port |
| * in the test */ |
| global_port = 0; |
| test_result = testExternalGet (); |
| if (test_result) |
| fprintf (stderr, "FAILED: testExternalGet () - %u.\n", test_result); |
| else if (verbose) |
| printf ("PASSED: testExternalGet ().\n"); |
| errorCount += test_result; |
| if (MHD_YES == MHD_is_feature_supported (MHD_FEATURE_THREADS)) |
| { |
| test_result = testInternalGet (testMhdPollBySelect); |
| if (test_result) |
| fprintf (stderr, "FAILED: testInternalGet (testMhdPollBySelect) - %u.\n", |
| test_result); |
| else if (verbose) |
| printf ("PASSED: testInternalGet (testMhdPollBySelect).\n"); |
| errorCount += test_result; |
| test_result = testMultithreadedPoolGet (testMhdPollBySelect); |
| if (test_result) |
| fprintf (stderr, |
| "FAILED: testMultithreadedPoolGet (testMhdPollBySelect) - %u.\n", |
| test_result); |
| else if (verbose) |
| printf ("PASSED: testMultithreadedPoolGet (testMhdPollBySelect).\n"); |
| errorCount += test_result; |
| test_result = testMultithreadedGet (testMhdPollBySelect); |
| if (test_result) |
| fprintf (stderr, |
| "FAILED: testMultithreadedGet (testMhdPollBySelect) - %u.\n", |
| test_result); |
| else if (verbose) |
| printf ("PASSED: testMultithreadedGet (testMhdPollBySelect).\n"); |
| errorCount += test_result; |
| if (MHD_YES == MHD_is_feature_supported (MHD_FEATURE_POLL)) |
| { |
| test_result = testInternalGet (testMhdPollByPoll); |
| if (test_result) |
| fprintf (stderr, "FAILED: testInternalGet (testMhdPollByPoll) - %u.\n", |
| test_result); |
| else if (verbose) |
| printf ("PASSED: testInternalGet (testMhdPollByPoll).\n"); |
| errorCount += test_result; |
| } |
| if (MHD_YES == MHD_is_feature_supported (MHD_FEATURE_EPOLL)) |
| { |
| test_result = testInternalGet (testMhdPollByEpoll); |
| if (test_result) |
| fprintf (stderr, "FAILED: testInternalGet (testMhdPollByEpoll) - %u.\n", |
| test_result); |
| else if (verbose) |
| printf ("PASSED: testInternalGet (testMhdPollByEpoll).\n"); |
| errorCount += test_result; |
| } |
| } |
| if (0 != errorCount) |
| fprintf (stderr, |
| "Error (code: %u)\n", |
| errorCount); |
| else if (verbose) |
| printf ("All tests passed.\n"); |
| |
| test_global_cleanup (); |
| |
| return (errorCount == 0) ? 0 : 1; /* 0 == pass */ |
| } |