[GNUnet-SVN] r16593 - libmicrohttpd/src/testcurl

[gnunet]

Author: grothoff
Date: 2011-08-18 10:56:18 +0200 (Thu, 18 Aug 2011)
New Revision: 16593

Added:
   libmicrohttpd/src/testcurl/test_callback.c
Modified:
   libmicrohttpd/src/testcurl/Makefile.am
Log:
new testcase

Modified: libmicrohttpd/src/testcurl/Makefile.am
===================================================================
--- libmicrohttpd/src/testcurl/Makefile.am      2011-08-16 18:56:15 UTC (rev 
16592)
+++ libmicrohttpd/src/testcurl/Makefile.am      2011-08-18 08:56:18 UTC (rev 
16593)
@@ -48,6 +48,7 @@
   daemontest_iplimit11 \
   daemontest_termination \
   daemontest_timeout \
+  test_callback \
   perf_get $(PERF_GET_CONCURRENT)
 
 
@@ -77,6 +78,12 @@
   $(top_builddir)/src/daemon/libmicrohttpd.la \
   @LIBCURL@ 
 
+test_callback_SOURCES = \
+  test_callback.c
+test_callback_LDADD = \
+  $(top_builddir)/src/daemon/libmicrohttpd.la \
+  @LIBCURL@ 
+
 perf_get_SOURCES = \
   perf_get.c \
   gauger.h

Added: libmicrohttpd/src/testcurl/test_callback.c
===================================================================
--- libmicrohttpd/src/testcurl/test_callback.c                          (rev 0)
+++ libmicrohttpd/src/testcurl/test_callback.c  2011-08-18 08:56:18 UTC (rev 
16593)
@@ -0,0 +1,183 @@
+/*
+     This file is part of libmicrohttpd
+     (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., 59 Temple Place - Suite 330,
+     Boston, MA 02111-1307, USA.
+*/
+
+/**
+ * @file test_callback.c
+ * @brief Testcase for MHD not calling the callback too often
+ * @author Jan Seeger 
+ * @author Christian Grothoff
+ */
+
+
+#include "MHD_config.h"
+#include "platform.h"
+#include <curl/curl.h>
+#include <microhttpd.h>
+
+struct callback_closure {
+  unsigned int called;
+};
+
+
+static ssize_t 
+called_twice(void *cls, uint64_t pos, char *buf, size_t max) 
+{
+  struct callback_closure *cls2 = cls;
+  
+  if (cls2->called == 0) 
+    {
+      memset(buf, 0, max);
+      strcat(buf, "test");
+      cls2->called = 1;
+      return strlen(buf);
+    }
+  if (cls2->called == 1) 
+    {
+      cls2->called = 2;
+      return MHD_CONTENT_READER_END_OF_STREAM;
+    }
+  fprintf(stderr, 
+         "Handler called after returning END_OF_STREAM!\n");
+  return MHD_CONTENT_READER_END_WITH_ERROR;
+}
+
+
+static int
+callback(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 **con_cls) {
+  struct callback_closure *cbc = calloc(1, sizeof(struct callback_closure));
+  struct MHD_Response *r;
+
+  r = MHD_create_response_from_callback (MHD_SIZE_UNKNOWN, 1024, 
+                                        &called_twice, cbc, 
+                                        &free);
+  MHD_queue_response(connection, 200, r);
+  MHD_destroy_response(r);
+  return MHD_YES;
+}
+
+
+static size_t
+discard_buffer (void *ptr, size_t size, size_t nmemb, void *ctx)
+{
+  return size * nmemb;
+}
+
+
+int main(int argc, char **argv) 
+{
+  struct MHD_Daemon *d;
+  fd_set rs;
+  fd_set ws;
+  fd_set es;
+  int max;
+  CURL *c;
+  CURLM *multi;
+  CURLMcode mret;
+  struct CURLMsg *msg;
+  int running;
+  struct timeval tv;
+  int extra;
+
+  d = MHD_start_daemon(MHD_USE_DEBUG,
+                      8000,
+                      NULL,
+                      NULL,
+                      callback,
+                      NULL,
+                      MHD_OPTION_END);
+  c = curl_easy_init ();
+  curl_easy_setopt (c, CURLOPT_URL, "http://localhost:8000/";);
+  curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, &discard_buffer);
+  curl_easy_setopt (c, CURLOPT_FAILONERROR, 1);
+  curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
+  curl_easy_setopt (c, CURLOPT_TIMEOUT, 150L);
+  curl_easy_setopt (c, CURLOPT_CONNECTTIMEOUT, 15L);
+  curl_easy_setopt (c, CURLOPT_NOSIGNAL, 1);
+  multi = curl_multi_init ();
+  if (multi == NULL)
+    {
+      curl_easy_cleanup (c);
+      MHD_stop_daemon (d);
+      return 1;
+    }
+  mret = curl_multi_add_handle (multi, c);
+  if (mret != CURLM_OK)
+    {
+      curl_multi_cleanup (multi);
+      curl_easy_cleanup (c);
+      MHD_stop_daemon (d);
+      return 2;
+    }
+  extra = 10;
+  while ( (c != NULL) || (--extra > 0) )
+    {
+      max = -1;
+      FD_ZERO(&ws);
+      FD_ZERO(&rs);
+      FD_ZERO(&es);
+      curl_multi_perform (multi, &running);
+      mret = curl_multi_fdset (multi, &rs, &ws, &es, &max);
+      if (mret != CURLM_OK)
+        {
+          curl_multi_remove_handle (multi, c);
+          curl_multi_cleanup (multi);
+          curl_easy_cleanup (c);
+          MHD_stop_daemon (d);
+          return 3;
+        }   
+      if (MHD_YES !=
+         MHD_get_fdset(d, &rs, &ws, &es, &max))
+       {
+          curl_multi_remove_handle (multi, c);
+          curl_multi_cleanup (multi);
+          curl_easy_cleanup (c);
+          MHD_stop_daemon (d);
+         return 4;
+       }
+      tv.tv_sec = 0;
+      tv.tv_usec = 1000;
+      select(max + 1, &rs, &ws, &es, &tv);
+      curl_multi_perform (multi, &running);
+      if (running == 0)
+        {
+          msg = curl_multi_info_read (multi, &running);
+          if (msg == NULL)
+            break;
+          if (msg->msg == CURLMSG_DONE)
+            {
+              if (msg->data.result != CURLE_OK)
+                printf ("%s failed at %s:%d: `%s'\n",
+                        "curl_multi_perform",
+                        __FILE__,
+                        __LINE__, curl_easy_strerror (msg->data.result));
+              curl_multi_remove_handle (multi, c);
+              curl_multi_cleanup (multi);
+              curl_easy_cleanup (c);
+              c = NULL;
+              multi = NULL;
+            }
+        }
+      MHD_run(d);
+    }
+  MHD_stop_daemon(d);
+  return 0;
+}