gnunet-svn
[Top][All Lists]
Advanced

[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

[libmicrohttpd] branch master updated: Added two tests for correct choic


From: gnunet
Subject: [libmicrohttpd] branch master updated: Added two tests for correct choice of "Keep-Alive" or "Close"
Date: Tue, 17 Aug 2021 20:00:51 +0200

This is an automated email from the git hooks/post-receive script.

karlson2k pushed a commit to branch master
in repository libmicrohttpd.

The following commit(s) were added to refs/heads/master by this push:
     new 21b807ae Added two tests for correct choice of "Keep-Alive" or "Close"
21b807ae is described below

commit 21b807ae1ca0a29c4a83ae28076cd2c5cd11b97c
Author: Evgeny Grin (Karlson2k) <k2k@narod.ru>
AuthorDate: Tue Aug 17 20:57:08 2021 +0300

    Added two tests for correct choice of "Keep-Alive" or "Close"
---
 src/testcurl/.gitignore                  |    4 +
 src/testcurl/Makefile.am                 |   28 +
 src/testcurl/test_get_close_keep_alive.c | 1001 ++++++++++++++++++++++++++++++
 3 files changed, 1033 insertions(+)

diff --git a/src/testcurl/.gitignore b/src/testcurl/.gitignore
index cf4ba3bd..2e2c388a 100644
--- a/src/testcurl/.gitignore
+++ b/src/testcurl/.gitignore
@@ -24,6 +24,10 @@
 /test_iplimit11
 /test_get_sendfile11
 /test_get_sendfile
+/test_get_close
+/test_get_close10
+/test_get_keep_alive
+/test_get_keep_alive10
 /test_get_response_cleanup
 /test_get_chunked
 /test_get_chunked_close
diff --git a/src/testcurl/Makefile.am b/src/testcurl/Makefile.am
index cc13962c..3912c716 100644
--- a/src/testcurl/Makefile.am
+++ b/src/testcurl/Makefile.am
@@ -82,6 +82,10 @@ check_PROGRAMS = \
   test_get \
   test_get_iovec \
   test_get_sendfile \
+  test_get_close \
+  test_get_close10 \
+  test_get_keep_alive \
+  test_get_keep_alive10 \
   test_delete \
   test_patch \
   test_put \
@@ -476,6 +480,30 @@ test_get_sendfile11_LDADD = \
   $(top_builddir)/src/microhttpd/libmicrohttpd.la \
   @LIBCURL@
 
+test_get_close_SOURCES = \
+  test_get_close_keep_alive.c $(top_srcdir)/src/microhttpd/test_helpers.h
+test_get_close_LDADD = \
+  $(top_builddir)/src/microhttpd/libmicrohttpd.la \
+  @LIBCURL@
+
+test_get_close10_SOURCES = \
+  test_get_close_keep_alive.c $(top_srcdir)/src/microhttpd/test_helpers.h
+test_get_close10_LDADD = \
+  $(top_builddir)/src/microhttpd/libmicrohttpd.la \
+  @LIBCURL@
+
+test_get_keep_alive_SOURCES = \
+  test_get_close_keep_alive.c $(top_srcdir)/src/microhttpd/test_helpers.h
+test_get_keep_alive_LDADD = \
+  $(top_builddir)/src/microhttpd/libmicrohttpd.la \
+  @LIBCURL@
+
+test_get_keep_alive10_SOURCES = \
+  test_get_close_keep_alive.c $(top_srcdir)/src/microhttpd/test_helpers.h
+test_get_keep_alive10_LDADD = \
+  $(top_builddir)/src/microhttpd/libmicrohttpd.la \
+  @LIBCURL@
+
 test_post11_SOURCES = \
   test_post.c mhd_has_in_name.h
 test_post11_LDADD = \
diff --git a/src/testcurl/test_get_close_keep_alive.c 
b/src/testcurl/test_get_close_keep_alive.c
new file mode 100644
index 00000000..21461e0d
--- /dev/null
+++ b/src/testcurl/test_get_close_keep_alive.c
@@ -0,0 +1,1001 @@
+/*
+     This file is part of libmicrohttpd
+     Copyright (C) 2017-2021 Karlson2k (Evgeny Grin)
+     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
+ * @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 "test_helpers.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 */
+
+#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 */
+static int conn_close;     /**< Don't use Keep-Alive */
+static int 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 add_mhd_close; /**< Add "Connection: close" header by MHD 
*/
+
+/* 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_conn_close;
+  int found_conn_keep_alive;
+};
+
+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 ((data_size == strlen (HDR_CONN_CLOSE) + 2) &&
+      (0 == strncasecmp (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 == strncasecmp (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 **unused)
+{
+  static int ptr;
+  const char *me = cls;
+  struct MHD_Response *response;
+  enum MHD_Result ret;
+  (void) version;
+  (void) upload_data;
+  (void) upload_data_size;       /* Unused. Silence compiler warning. */
+
+  if (0 != strcasecmp (me, method))
+    return MHD_NO;              /* unexpected method */
+  if (&ptr != *unused)
+  {
+    *unused = &ptr;
+    return MHD_YES;
+  }
+  *unused = NULL;
+  if (slow_reply)
+    usleep (200000);
+
+  response = MHD_create_response_from_buffer (strlen (url),
+                                              (void *) url,
+                                              MHD_RESPMEM_MUST_COPY);
+  if (NULL == response)
+  {
+    fprintf (stderr, "Failed to create response. Line: %d\n", __LINE__);
+    _exit (19);
+  }
+  if (add_mhd_close)
+  {
+    if (MHD_NO == 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);
+    }
+  }
+  ret = MHD_queue_response (connection,
+                            MHD_HTTP_OK,
+                            response);
+  MHD_destroy_response (response);
+  if (ret == MHD_NO)
+  {
+    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 */
+  int queryPort;
+
+  /* CURL query result error flag */
+  volatile int queryError;
+};
+
+
+static CURL *
+curlEasyInitForTest (const char *queryPath, int 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,
+                                     &copyBuffer)) ||
+      (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)) ||
+      (add_hdr_k_alive && (CURLE_OK != curl_easy_setopt (c, CURLOPT_HTTPHEADER,
+                                                         curl_k_alive_hdr))) ||
+      (CURLE_OK != curl_easy_setopt (c, CURLOPT_FAILONERROR, 1L)) ||
+      (oneone) ?
+      (CURLE_OK != curl_easy_setopt (c, CURLOPT_HTTP_VERSION,
+                                     CURL_HTTP_VERSION_1_1)) :
+      (CURLE_OK != curl_easy_setopt (c, CURLOPT_HTTP_VERSION,
+                                     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, "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. ",
+           add_mhd_close ? "    used" : "NOT used");
+}
+
+
+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 */
+      unsigned long long to;
+      if ((MHD_YES != MHD_get_timeout (d, &to)) || (0 != to))
+        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 = 10000;
+#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)
+        externalErrorExitDesc ("select() failed");
+#else
+      if ((WSAEINVAL != WSAGetLastError ()) ||
+          (0 != rs.fd_count) || (0 != ws.fd_count) || (0 != es.fd_count))
+        externalErrorExitDesc ("select() failed");
+      Sleep (10);
+#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 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_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)
+  {
+    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 (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 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 headers.\n");
+      p->queryError |= 2;
+    }
+    if (use_external_poll)
+    { /* The number of MHD connection can be used 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 && ! 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 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 headers.\n");
+      p->queryError |= 2;
+    }
+    if (use_external_poll)
+    { /* The number of MHD connection can be used only with external poll.
+       * otherwise it creates a race condition. */
+      if (1 != getMhdActiveConnections (d))
+      {
+        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;
+      }
+    }
+  }
+  curl_easy_cleanup (c);
+
+  return p->queryError;
+}
+
+
+/* Perform test queries and shut down MHD daemon */
+static int
+performTestQueries (struct MHD_Daemon *d, int d_port)
+{
+  struct curlQueryParams qParam;
+  int ret = 0;          /* Return value */
+
+  qParam.queryPath = "http://127.0.0.1"; EXPECTED_URI_FULL_PATH;
+  qParam.queryPort = d_port;   /* Connect to the daemon */
+
+  if (conn_close)
+  {
+    if (! oneone)
+    {
+      /* Client without preferences, no MHD preferences. */
+      ret <<= 3;                   /* Remember errors for each step */
+      add_mhd_close = 0;
+      ret |= doCurlQueryInThread (d, &qParam, 0, 0);
+    }
+    /* Client requested "close", no MHD preferences. */
+    ret <<= 3;                   /* Remember errors for each step */
+    add_mhd_close = 0;
+    ret |= doCurlQueryInThread (d, &qParam, 1, 0);
+
+    /* Client requested both "close" and "Keep-Alive", no MHD preferences. */
+    ret <<= 3;                   /* Remember errors for each step */
+    add_mhd_close = 0;
+    ret |= doCurlQueryInThread (d, &qParam, 1, 1);
+
+    /* Client without preferences, MHD set to "close". */
+    ret <<= 3;                   /* Remember errors for each step */
+    add_mhd_close = 1;
+    ret |= doCurlQueryInThread (d, &qParam, 0, 0);
+
+    /* Client requested "close", MHD set to "close". */
+    ret <<= 3;                   /* Remember errors for each step */
+    add_mhd_close = 1;
+    ret |= doCurlQueryInThread (d, &qParam, 1, 0);
+
+    /* Client requested "Keep-Alive", MHD set to "close". */
+    ret <<= 3;                   /* Remember errors for each step */
+    add_mhd_close = 1;
+    ret |= doCurlQueryInThread (d, &qParam, 0, 1);
+
+    /* Client requested both "close" and "Keep-Alive", MHD set to "close". */
+    ret <<= 3;                   /* Remember errors for each step */
+    add_mhd_close = 1;
+    ret |= doCurlQueryInThread (d, &qParam, 1, 1);
+  }
+  else
+  { /* Keep-Alive */
+    if (oneone)
+    {
+      /* Client without preferences, no MHD preferences. */
+      ret <<= 3;                   /* Remember errors for each step */
+      add_mhd_close = 0;
+      ret |= doCurlQueryInThread (d, &qParam, 0, 0);
+    }
+    /* Client requested "Keep-Alive", no MHD preferences. */
+    ret <<= 3;                   /* Remember errors for each step */
+    add_mhd_close = 0;
+    ret |= doCurlQueryInThread (d, &qParam, 0, 1);
+  }
+
+  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)
+{
+  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, int *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 (((int) thrType) | ((int) pollType)
+                          | MHD_USE_ERROR_LOG,
+                          *pport, NULL, NULL,
+                          &ahc_echo, "GET",
+                          MHD_OPTION_URI_LOG_CALLBACK, &log_cb, NULL,
+                          MHD_OPTION_END);
+  else
+    d = MHD_start_daemon (MHD_USE_INTERNAL_POLLING_THREAD | ((int) pollType)
+                          | MHD_USE_ERROR_LOG,
+                          *pport, NULL, NULL,
+                          &ahc_echo, "GET",
+                          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 = (int) dinfo->port;
+    if (0 == global_port)
+      global_port = *pport; /* Reuse the same port for all tests */
+  }
+
+  return d;
+}
+
+
+/* Test runners */
+
+
+static int
+testExternalGet (void)
+{
+  struct MHD_Daemon *d;
+  int d_port = global_port; /* Daemon's port */
+
+  d = startTestMhdDaemon (testMhdThreadExternal, testMhdPollBySelect, &d_port);
+
+  return performTestQueries (d, d_port);
+}
+
+
+static int
+testInternalGet (enum testMhdPollType pollType)
+{
+  struct MHD_Daemon *d;
+  int d_port = global_port; /* Daemon's port */
+
+  d = startTestMhdDaemon (testMhdThreadInternal, pollType,
+                          &d_port);
+
+  return performTestQueries (d, d_port);
+}
+
+
+static int
+testMultithreadedGet (enum testMhdPollType pollType)
+{
+  struct MHD_Daemon *d;
+  int d_port = global_port; /* Daemon's port */
+
+  d = startTestMhdDaemon (testMhdThreadInternalPerConnection, pollType,
+                          &d_port);
+  return performTestQueries (d, d_port);
+}
+
+
+static int
+testMultithreadedPoolGet (enum testMhdPollType pollType)
+{
+  struct MHD_Daemon *d;
+  int 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");
+
+  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 */
+}

-- 
To stop receiving notification emails like this one, please contact
gnunet@gnunet.org.



reply via email to

[Prev in Thread] Current Thread [Next in Thread]