/* This file is part of libmicrohttpd Copyright (C) 2007, 2009, 2011 Christian Grothoff Copyright (C) 2014-2022 Evgeny Grin (Karlson2k) - large rework, multithreading. 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_add_conn.c * @brief Testcase for libmicrohttpd GET operations * @author Christian Grothoff * @author Karlson2k (Evgeny Grin) */ #include "MHD_config.h" #include "platform.h" #include #include #include #include #include #include "mhd_has_in_name.h" #include "mhd_has_param.h" #include "mhd_sockets.h" /* only macros used */ #ifdef _WIN32 #ifndef WIN32_LEAN_AND_MEAN #define WIN32_LEAN_AND_MEAN 1 #endif /* !WIN32_LEAN_AND_MEAN */ #include #endif #ifndef WINDOWS #include #include #endif #ifdef HAVE_LIMITS_H #include #endif /* HAVE_LIMITS_H */ #ifdef HAVE_PTHREAD_H #include #endif /* HAVE_PTHREAD_H */ #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 */ /* Could be increased to facilitate debugging */ #define TIMEOUTS_VAL 5 /* Number of requests per daemon in cleanup test, * the number must be more than one as the first connection * will be processed and the rest will stay in the list of unprocessed */ #define CLEANUP_NUM_REQS_PER_DAEMON 6 /* Cleanup test: max number of concurrent daemons depending on maximum number * of open FDs. */ #define CLEANUP_MAX_DAEMONS(max_fds) (unsigned int) \ ( ((max_fds) < 10) ? \ 0 : ( (((max_fds) - 10) / (CLEANUP_NUM_REQS_PER_DAEMON * 5 + 3)) ) ) #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 /* Global parameters */ static int oneone; /**< Use HTTP/1.1 instead of HTTP/1.0 */ static int no_listen; /**< Start MHD daemons without listen socket */ static uint16_t global_port; /**< MHD daemons listen port number */ static int cleanup_test; /**< Test for final cleanup */ 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; static int sys_max_fds; /**< Current system limit for number of open files. */ 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) return 0; /* 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'\n", uri); _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; const char *v; (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; v = MHD_lookup_connection_value (connection, MHD_GET_ARGUMENT_KIND, "a"); if ( (NULL == v) || (0 != strcmp ("&", v)) ) { fprintf (stderr, "Found while looking for 'a=&': 'a=%s'\n", NULL == v ? "NULL" : v); _exit (17); } v = NULL; if (MHD_YES != MHD_lookup_connection_value_n (connection, MHD_GET_ARGUMENT_KIND, "b", 1, &v, NULL)) { fprintf (stderr, "Not found 'b' GET argument.\n"); _exit (18); } if ( (NULL == v) || (0 != strcmp ("c", v)) ) { fprintf (stderr, "Found while looking for 'b=c': 'b=%s'\n", NULL == v ? "NULL" : v); _exit (19); } if (slow_reply) usleep (200000); response = MHD_create_response_from_buffer_copy (strlen (url), (const void *) url); ret = MHD_queue_response (connection, MHD_HTTP_OK, response); MHD_destroy_response (response); if (ret == MHD_NO) { fprintf (stderr, "Failed to queue response.\n"); _exit (19); } return ret; } 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); } #if defined(HAVE___FUNC__) #define externalErrorExit(ignore) \ _externalErrorExit_func(NULL, __func__, __LINE__) #define externalErrorExitDesc(errDesc) \ _externalErrorExit_func(errDesc, __func__, __LINE__) #elif defined(HAVE___FUNCTION__) #define externalErrorExit(ignore) \ _externalErrorExit_func(NULL, __FUNCTION__, __LINE__) #define externalErrorExitDesc(errDesc) \ _externalErrorExit_func(errDesc, __FUNCTION__, __LINE__) #else #define externalErrorExit(ignore) _externalErrorExit_func(NULL, NULL, __LINE__) #define externalErrorExitDesc(errDesc) \ _externalErrorExit_func(errDesc, NULL, __LINE__) #endif /* Static const value, indicates that result value was not set yet */ static const unsigned int eMarker = 0xCE; static MHD_socket createListeningSocket (uint16_t *pport) { MHD_socket skt; struct sockaddr_in sin; socklen_t sin_len; #ifdef MHD_POSIX_SOCKETS static int on = 1; #endif /* MHD_POSIX_SOCKETS */ skt = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP); if (MHD_INVALID_SOCKET == skt) externalErrorExitDesc ("socket() failed"); #ifdef MHD_POSIX_SOCKETS setsockopt (skt, SOL_SOCKET, SO_REUSEADDR, (void *) &on, sizeof (on)); /* Ignore possible error */ #endif /* MHD_POSIX_SOCKETS */ memset (&sin, 0, sizeof(sin)); sin.sin_family = AF_INET; sin.sin_port = htons (*pport); sin.sin_addr.s_addr = htonl (INADDR_LOOPBACK); if (0 != bind (skt, (struct sockaddr *) &sin, sizeof(sin))) externalErrorExitDesc ("bind() failed"); if (0 != listen (skt, SOMAXCONN)) externalErrorExitDesc ("listen() failed"); if (0 == *pport) { memset (&sin, 0, sizeof(sin)); sin_len = (socklen_t) sizeof(sin); if (0 != getsockname (skt, (struct sockaddr *) &sin, &sin_len)) externalErrorExitDesc ("getsockname() failed"); if (sizeof(sin) < (size_t) sin_len) externalErrorExitDesc ("getsockname() failed"); if (AF_INET != sin.sin_family) externalErrorExitDesc ("getsockname() returned wrong socket family"); *pport = ntohs (sin.sin_port); } return skt; } static MHD_socket acceptTimeLimited (MHD_socket lstn_sk, struct sockaddr *paddr, socklen_t *paddr_len) { fd_set rs; struct timeval timeoutval; MHD_socket accepted; FD_ZERO (&rs); FD_SET (lstn_sk, &rs); timeoutval.tv_sec = TIMEOUTS_VAL; timeoutval.tv_usec = 0; if (1 != select (((int) lstn_sk) + 1, &rs, NULL, NULL, &timeoutval)) externalErrorExitDesc ("select() failed"); accepted = accept (lstn_sk, paddr, paddr_len); if (MHD_INVALID_SOCKET == accepted) externalErrorExitDesc ("accept() failed"); return accepted; } struct addConnParam { struct MHD_Daemon *d; MHD_socket lstn_sk; MHD_socket clent_sk; /* Non-zero indicate error */ volatile unsigned int result; #ifdef HAVE_PTHREAD_H pthread_t addConnThread; #endif /* HAVE_PTHREAD_H */ }; static unsigned int doAcceptAndAddConnInThread (struct addConnParam *p) { struct sockaddr addr; socklen_t addr_len = sizeof(addr); p->clent_sk = acceptTimeLimited (p->lstn_sk, &addr, &addr_len); p->result = (MHD_YES == MHD_add_connection (p->d, p->clent_sk, &addr, addr_len)) ? 0 : 1; if (p->result) fprintf (stderr, "MHD_add_connection() failed, errno=%d.\n", errno); return p->result; } #ifdef HAVE_PTHREAD_H static void * doAcceptAndAddConn (void *param) { struct addConnParam *p = param; (void) doAcceptAndAddConnInThread (p); return (void *) p; } static void startThreadAddConn (struct addConnParam *param) { /* thread must reset this value to zero if succeed */ param->result = eMarker; if (0 != pthread_create (¶m->addConnThread, NULL, &doAcceptAndAddConn, (void *) param)) externalErrorExitDesc ("pthread_create() failed"); } static unsigned int finishThreadAddConn (struct addConnParam *param) { struct addConnParam *result; if (0 != pthread_join (param->addConnThread, (void **) &result)) externalErrorExitDesc ("pthread_join() failed"); if (param != result) abort (); /* Test used in a wrong way */ if (eMarker == param->result) abort (); /* Test used in a wrong way */ return result->result; } #endif /* HAVE_PTHREAD_H */ 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; #ifdef HAVE_PTHREAD_H pthread_t queryThread; #endif /* HAVE_PTHREAD_H */ }; static CURL * curlEasyInitForTest (const char *queryPath, uint16_t port, struct CBC *pcbc) { CURL *c; c = curl_easy_init (); if (NULL == c) { fprintf (stderr, "curl_easy_init() failed.\n"); _exit (99); } 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_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"); _exit (99); } return c; } static unsigned int doCurlQueryInThread (struct curlQueryParams *p) { CURL *c; char buf[2048]; struct CBC cbc; CURLcode errornum; if (NULL == p->queryPath) abort (); if (0 == p->queryPort) abort (); cbc.buf = buf; cbc.size = sizeof(buf); cbc.pos = 0; c = curlEasyInitForTest (p->queryPath, p->queryPort, &cbc); errornum = curl_easy_perform (c); if (ignore_response_errors) { p->queryError = 0; curl_easy_cleanup (c); return p->queryError; } if (CURLE_OK != errornum) { fprintf (stderr, "curl_easy_perform failed: `%s'\n", curl_easy_strerror (errornum)); p->queryError = 2; } else { if (cbc.pos != strlen (EXPECTED_URI_BASE_PATH)) { fprintf (stderr, "curl reports wrong size of MHD reply body data.\n"); p->queryError = 4; } 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 = 4; } else p->queryError = 0; } curl_easy_cleanup (c); return p->queryError; } #ifdef HAVE_PTHREAD_H static void * doCurlQuery (void *param) { struct curlQueryParams *p = (struct curlQueryParams *) param; (void) doCurlQueryInThread (p); return param; } static void startThreadCurlQuery (struct curlQueryParams *param) { /* thread must reset this value to zero if succeed */ param->queryError = eMarker; if (0 != pthread_create (¶m->queryThread, NULL, &doCurlQuery, (void *) param)) externalErrorExitDesc ("pthread_create() failed"); } static unsigned int finishThreadCurlQuery (struct curlQueryParams *param) { struct curlQueryParams *result; if (0 != pthread_join (param->queryThread, (void **) &result)) externalErrorExitDesc ("pthread_join() failed"); if (param != result) abort (); /* Test used in wrong way */ if (eMarker == param->queryError) abort (); /* Test used in wrong way */ return result->queryError; } /* Perform test queries and shut down MHD daemon */ static unsigned int performTestQueries (struct MHD_Daemon *d, uint16_t d_port) { struct curlQueryParams qParam; struct addConnParam aParam; uint16_t a_port; /* Additional listening socket port */ unsigned int ret = 0; /* Return value */ qParam.queryPath = "http://127.0.0.1" EXPECTED_URI_FULL_PATH; a_port = 0; /* auto-assign */ aParam.d = d; aParam.lstn_sk = createListeningSocket (&a_port); /* Sets a_port */ /* Test of adding connection in the same thread */ qParam.queryError = eMarker; /* to be zeroed in new thread */ qParam.queryPort = a_port; /* Connect to additional socket */ startThreadCurlQuery (&qParam); ret |= doAcceptAndAddConnInThread (&aParam); ret |= finishThreadCurlQuery (&qParam); if (! no_listen) { /* Test of the daemon itself can accept and process new connection. */ ret <<= 3; /* Remember errors for each step */ qParam.queryPort = d_port; /* Connect to the daemon */ ret |= doCurlQueryInThread (&qParam); } /* Test of adding connection in an external thread */ ret <<= 3; /* Remember errors for each step */ aParam.result = eMarker; /* to be zeroed in new thread */ qParam.queryPort = a_port; /* Connect to the daemon */ startThreadAddConn (&aParam); ret |= doCurlQueryInThread (&qParam); ret |= finishThreadAddConn (&aParam); (void) MHD_socket_close_ (aParam.lstn_sk); MHD_stop_daemon (d); return ret; } /* Perform test for cleanup and shutdown MHD daemon */ static unsigned int performTestCleanup (struct MHD_Daemon *d, unsigned int num_queries) { struct curlQueryParams *qParamList; struct addConnParam aParam; MHD_socket lstn_sk; /* Additional listening socket */ MHD_socket *clntSkList; uint16_t a_port; /* Additional listening socket port */ unsigned int i; unsigned int ret = 0; /* Return value */ a_port = 0; /* auto-assign */ if (0 >= num_queries) abort (); /* Test's API violation */ lstn_sk = createListeningSocket (&a_port); /* Sets a_port */ qParamList = malloc (sizeof(struct curlQueryParams) * num_queries); clntSkList = malloc (sizeof(MHD_socket) * num_queries); if ((NULL == qParamList) || (NULL == clntSkList)) externalErrorExitDesc ("malloc failed"); /* Start CURL queries */ for (i = 0; i < num_queries; i++) { qParamList[i].queryPath = "http://127.0.0.1" EXPECTED_URI_FULL_PATH; qParamList[i].queryError = 0; qParamList[i].queryPort = a_port; startThreadCurlQuery (qParamList + i); } /* Accept and add required number of client sockets */ aParam.d = d; aParam.lstn_sk = lstn_sk; for (i = 0; i < num_queries; i++) { aParam.clent_sk = MHD_INVALID_SOCKET; ret |= doAcceptAndAddConnInThread (&aParam); clntSkList[i] = aParam.clent_sk; } /* Stop daemon while some of new connection are not yet * processed because of slow response to the first queries. */ MHD_stop_daemon (d); (void) MHD_socket_close_ (aParam.lstn_sk); /* Check whether all client sockets were closed by MHD. * Closure of socket by MHD indicate valid cleanup performed. */ for (i = 0; i < num_queries; i++) { if (MHD_INVALID_SOCKET != clntSkList[i]) { /* Check whether socket could be closed one more time. */ if (MHD_socket_close_ (clntSkList[i])) { ret |= 2; fprintf (stderr, "Client socket was not closed by MHD during" \ "cleanup process.\n"); } } } /* Wait for CURL threads to complete. */ /* Ignore soft CURL errors as many connection shouldn't get any response. * Hard failures are detected in processing function. */ for (i = 0; i < num_queries; i++) (void) finishThreadCurlQuery (qParamList + i); free (clntSkList); free (qParamList); return ret; } #endif /* HAVE_PTHREAD_H */ 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; if (! cleanup_test) return numThreads; /* No practical limit for non-cleanup test */ if (CLEANUP_MAX_DAEMONS (sys_max_fds) < numThreads) numThreads = CLEANUP_MAX_DAEMONS (sys_max_fds); if ((testMhdPollBySelect == pollType) && (CLEANUP_MAX_DAEMONS (FD_SETSIZE) < numThreads)) numThreads = CLEANUP_MAX_DAEMONS (FD_SETSIZE); if (2 > numThreads) abort (); return (unsigned int) numThreads; } 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 = 1550; if (oneone) *pport += 1; if (no_listen) *pport += 2; if (cleanup_test) *pport += 4; } if (testMhdThreadInternalPool != thrType) d = MHD_start_daemon (((unsigned int) thrType) | ((unsigned int) pollType) | (thrType == testMhdThreadExternal ? 0 : MHD_USE_ITC) | (no_listen ? MHD_USE_NO_LISTEN_SOCKET : 0) | 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_ITC | (no_listen ? MHD_USE_NO_LISTEN_SOCKET : 0) | 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 ((! no_listen) && (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; } return d; } /* Test runners */ static unsigned int testExternalGet (void) { struct MHD_Daemon *d; CURL *c_d; char buf_d[2048]; struct CBC cbc_d; CURL *c_a; char buf_a[2048]; struct CBC cbc_a; CURLM *multi; time_t start; struct timeval tv; uint16_t d_port = global_port; /* Daemon's port */ uint16_t a_port = 0; /* Additional listening socket port */ struct addConnParam aParam; unsigned int ret = 0; /* Return value of the test */ const int c_no_listen = no_listen; /* Local const value to mute analyzer */ d = startTestMhdDaemon (testMhdThreadExternal, testMhdPollBySelect, &d_port); aParam.d = d; aParam.lstn_sk = createListeningSocket (&a_port); multi = NULL; cbc_d.buf = buf_d; cbc_d.size = sizeof(buf_d); cbc_d.pos = 0; cbc_a.buf = buf_a; cbc_a.size = sizeof(buf_a); cbc_a.pos = 0; if (cleanup_test) abort (); /* Not possible with "external poll" as connections are directly added to the daemon processing in the mode. */ if (! c_no_listen) c_d = curlEasyInitForTest ("http://127.0.0.1" EXPECTED_URI_FULL_PATH, d_port, &cbc_d); else c_d = NULL; /* To mute compiler warning only */ c_a = curlEasyInitForTest ("http://127.0.0.1" EXPECTED_URI_FULL_PATH, a_port, &cbc_a); multi = curl_multi_init (); if (multi == NULL) { fprintf (stderr, "curl_multi_init() failed.\n"); _exit (99); } if (! c_no_listen) { if (CURLM_OK != curl_multi_add_handle (multi, c_d)) { fprintf (stderr, "curl_multi_add_handle() failed.\n"); _exit (99); } } if (CURLM_OK != curl_multi_add_handle (multi, c_a)) { fprintf (stderr, "curl_multi_add_handle() failed.\n"); _exit (99); } 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); 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"); _exit (99); } totalMsgs++; if (CURLMSG_DONE == msg->msg) { if (CURLE_OK != msg->data.result) { fprintf (stderr, "curl_multi_info_read failed, error: '%s'\n", curl_easy_strerror (msg->data.result)); ret |= 2; } } } while (msgLeft > 0); if ((no_listen ? 1 : 2) != totalMsgs) { fprintf (stderr, "curl_multi_info_read returned wrong " "number of results (%d).\n", totalMsgs); _exit (99); } break; /* All transfers have finished. */ } if (CURLM_OK != curl_multi_fdset (multi, &rs, &ws, &es, &maxCurlSk)) { fprintf (stderr, "curl_multi_fdset() failed.\n"); _exit (99); } if (MHD_YES != MHD_get_fdset (d, &rs, &ws, &es, &maxMhdSk)) { ret |= 8; break; } FD_SET (aParam.lstn_sk, &rs); if (maxMhdSk < aParam.lstn_sk) maxMhdSk = aParam.lstn_sk; 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 (FD_ISSET (aParam.lstn_sk, &rs)) ret |= doAcceptAndAddConnInThread (&aParam); if (MHD_YES != MHD_run_from_select (d, &rs, &ws, &es)) { fprintf (stderr, "MHD_run_from_select() failed.\n"); ret |= 1; break; } } MHD_stop_daemon (d); (void) MHD_socket_close_ (aParam.lstn_sk); if (! c_no_listen) { curl_multi_remove_handle (multi, c_d); curl_easy_cleanup (c_d); if (cbc_d.pos != strlen ("/hello_world")) { fprintf (stderr, "curl reports wrong size of MHD reply body data at line %d.\n", __LINE__); ret |= 4; } if (0 != strncmp ("/hello_world", cbc_d.buf, strlen ("/hello_world"))) { fprintf (stderr, "curl reports wrong MHD reply body data at line %d.\n", __LINE__); ret |= 4; } } curl_multi_remove_handle (multi, c_a); curl_easy_cleanup (c_a); curl_multi_cleanup (multi); if (cbc_a.pos != strlen ("/hello_world")) { fprintf (stderr, "curl reports wrong size of MHD reply body data at line %d.\n", __LINE__); ret |= 4; } if (0 != strncmp ("/hello_world", cbc_a.buf, strlen ("/hello_world"))) { fprintf (stderr, "curl reports wrong MHD reply body data at line %d.\n", __LINE__); ret |= 4; } return ret; } #ifdef HAVE_PTHREAD_H 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); if (cleanup_test) return performTestCleanup (d, CLEANUP_NUM_REQS_PER_DAEMON); 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); if (cleanup_test) abort (); /* Cannot be tested as main daemon thread cannot be slowed down by slow responses, so it processes all new connections before daemon could be stopped. */ 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); if (cleanup_test) return performTestCleanup (d, CLEANUP_NUM_REQS_PER_DAEMON * testNumThreadsForPool (pollType)); return performTestQueries (d, d_port); } static unsigned int testStopRace (enum testMhdPollType pollType) { struct MHD_Daemon *d; uint16_t d_port = global_port; /* Daemon's port */ uint16_t a_port = 0; /* Additional listening socket port */ struct sockaddr_in sin; MHD_socket fd1; MHD_socket fd2; struct addConnParam aParam; unsigned int ret = 0; /* Return value of the test */ d = startTestMhdDaemon (testMhdThreadInternal, pollType, &d_port); if (! no_listen) { fd1 = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP); if (MHD_INVALID_SOCKET == fd1) externalErrorExitDesc ("socket() failed"); memset (&sin, 0, sizeof(sin)); sin.sin_family = AF_INET; sin.sin_port = htons (d_port); sin.sin_addr.s_addr = htonl (INADDR_LOOPBACK); if (connect (fd1, (struct sockaddr *) (&sin), sizeof(sin)) < 0) externalErrorExitDesc ("socket() failed"); } else fd1 = MHD_INVALID_SOCKET; aParam.d = d; aParam.lstn_sk = createListeningSocket (&a_port); /* Sets a_port */ startThreadAddConn (&aParam); fd2 = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP); if (MHD_INVALID_SOCKET == fd2) externalErrorExitDesc ("socket() failed"); memset (&sin, 0, sizeof(sin)); sin.sin_family = AF_INET; sin.sin_port = htons (a_port); sin.sin_addr.s_addr = htonl (INADDR_LOOPBACK); if (connect (fd2, (struct sockaddr *) (&sin), sizeof(sin)) < 0) externalErrorExitDesc ("socket() failed"); ret |= finishThreadAddConn (&aParam); /* Let the thread get going. */ usleep (500000); MHD_stop_daemon (d); if (MHD_INVALID_SOCKET != fd1) (void) MHD_socket_close_ (fd1); (void) MHD_socket_close_ (aParam.lstn_sk); (void) MHD_socket_close_ (fd2); return ret; } #endif /* HAVE_PTHREAD_H */ 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], "11"); /* Whether to test MHD daemons without listening socket. */ no_listen = has_in_name (argv[0], "_nolisten"); /* Whether to test for correct final cleanup instead of * of test of normal processing. */ cleanup_test = has_in_name (argv[0], "_cleanup"); /* There are almost nothing that could be tested externally * for final cleanup. Cleanup test actually just tests that * all added client connections were closed by MHD and * nothing fails or crashes when final cleanup is performed. * Mostly useful when configured with '--enable-asserts. */ slow_reply = cleanup_test; ignore_response_errors = cleanup_test; #ifndef HAVE_PTHREAD_H if (cleanup_test) return 77; /* Cannot run without threads */ #endif /* HAVE_PTHREAD_H */ verbose = ! (has_param (argc, argv, "-q") || has_param (argc, argv, "--quiet") || has_param (argc, argv, "-s") || has_param (argc, argv, "--silent")); if (cleanup_test) { #ifndef _WIN32 /* Find system limit for number of open FDs. */ #if defined(HAVE_SYSCONF) && defined(_SC_OPEN_MAX) sys_max_fds = sysconf (_SC_OPEN_MAX) > 500000 ? 500000 : (int) sysconf (_SC_OPEN_MAX); #else /* ! HAVE_SYSCONF || ! _SC_OPEN_MAX */ sys_max_fds = -1; #endif /* ! HAVE_SYSCONF || ! _SC_OPEN_MAX */ if (0 > sys_max_fds) { #if defined(OPEN_MAX) && (0 < ((OPEN_MAX) +1)) sys_max_fds = OPEN_MAX > 500000 ? 500000 : (int) OPEN_MAX; #else /* ! OPEN_MAX */ sys_max_fds = 256; /* Use reasonable value */ #endif /* ! OPEN_MAX */ if (2 > CLEANUP_MAX_DAEMONS (sys_max_fds)) return 77; /* Multithreaded test cannot be run */ } #else /* _WIN32 */ sys_max_fds = 120; /* W32 has problems with ports exhaust */ #endif /* _WIN32 */ } if (0 != curl_global_init (CURL_GLOBAL_WIN32)) return 99; /* Could be set to non-zero value to enforce using specific port * in the test */ global_port = 0; if (! cleanup_test) { test_result = testExternalGet (); if (test_result) fprintf (stderr, "FAILED: testExternalGet () - %u.\n", test_result); else if (verbose) printf ("PASSED: testExternalGet ().\n"); errorCount += test_result; } #ifdef HAVE_PTHREAD_H 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; if (! cleanup_test) { 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; test_result = testStopRace (testMhdPollBySelect); if (test_result) fprintf (stderr, "FAILED: testStopRace (testMhdPollBySelect) - %u.\n", test_result); else if (verbose) printf ("PASSED: testStopRace (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; test_result = testMultithreadedPoolGet (testMhdPollByPoll); if (test_result) fprintf (stderr, "FAILED: testMultithreadedPoolGet (testMhdPollByPoll) - %u.\n", test_result); else if (verbose) printf ("PASSED: testMultithreadedPoolGet (testMhdPollByPoll).\n"); errorCount += test_result; if (! cleanup_test) { test_result = testMultithreadedGet (testMhdPollByPoll); if (test_result) fprintf (stderr, "FAILED: testMultithreadedGet (testMhdPollByPoll) - %u.\n", test_result); else if (verbose) printf ("PASSED: testMultithreadedGet (testMhdPollByPoll).\n"); errorCount += test_result; test_result = testStopRace (testMhdPollByPoll); if (test_result) fprintf (stderr, "FAILED: testStopRace (testMhdPollByPoll) - %u.\n", test_result); else if (verbose) printf ("PASSED: testStopRace (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; test_result = testMultithreadedPoolGet (testMhdPollByEpoll); if (test_result) fprintf (stderr, "FAILED: testMultithreadedPoolGet (testMhdPollByEpoll) - %u.\n", test_result); else if (verbose) printf ("PASSED: testMultithreadedPoolGet (testMhdPollByEpoll).\n"); errorCount += test_result; } } #endif /* HAVE_PTHREAD_H */ if (0 != errorCount) fprintf (stderr, "Error (code: %u)\n", errorCount); else if (verbose) printf ("All tests passed.\n"); curl_global_cleanup (); return (errorCount == 0) ? 0 : 1; /* 0 == pass */ }