1 files changed, 145 insertions, 3 deletions

diff --git a/src/daemon/daemontest.c b/src/daemon/daemontest.c
index 1b36e64f..5ebe1e24 100644
--- a/src/daemon/daemontest.c
+++ b/src/daemon/daemontest.c
@@ -1,6 +1,6 @@
 /*
      This file is part of libmicrohttpd
-     (C) 2007 YOUR NAME HERE
+     (C) 2007 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
@@ -21,14 +21,156 @@
 /**
  * @file daemontest.c
  * @brief  Testcase for libmicrohttpd
- * @author FIXME
+ * @author Christian Grothoff
  */
 
 #include "config.h"
+#include "curl/curl.h"
 #include "microhttpd.h"
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
 
+static int testStartError() {
+  return NULL != MHD_start_daemon(0, 0, NULL, NULL, NULL, NULL);
+}
+
+static int apc_nothing(void * cls,
+                       const struct sockaddr * addr,
+                       socklen_t addrlen) {
+  return MHD_NO;
+}
+
+static int apc_all(void * cls,
+                   const struct sockaddr * addr,
+                   socklen_t addrlen) {
+  return MHD_YES;
+}
+
+static int ahc_nothing(void * cls,
+                       struct MHD_Session * session,
+                       const char * url,
+                       const char * method) {
+  return MHD_NO;
+}
+
+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 int ahc_echo(void * cls,
+                    struct MHD_Session * session,
+                    const char * url,
+                    const char * method) {
+  const char * me = cls;
+  struct MHD_Response * response;
+  int ret;
+
+  if (0 != strcmp(me, method))
+    return MHD_NO; /* unexpected method */
+  response = MHD_create_response_from_data(strlen(url),
+                                           (void*) url,
+                                           MHD_NO,
+                                           MHD_YES);
+  ret = MHD_queue_response(session,
+                           MHD_HTTP_OK,
+                           response);
+  MHD_destroy_response(response);
+  return ret;
+}
+
+static int testStartStop() {
+  struct MHD_Daemon * d;
+
+  d = MHD_start_daemon(MHD_USE_SELECT_INTERNALLY | MHD_USE_IPv4,
+                       1080,
+                       &apc_nothing,
+                       NULL,
+                       &ahc_nothing,
+                       NULL);
+  if (d == NULL)
+    return 1;
+  MHD_stop_daemon(d);
+  return 0;
+}
+
+static int testGet() {
+  struct MHD_Daemon * d;
+  CURL * c;
+  int ret;
+  char buf[2048];
+  struct CBC cbc;
+
+  cbc.buf = buf;
+  cbc.size = 2048;
+  cbc.pos = 0;
+  ret = 0;
+  d = MHD_start_daemon(MHD_USE_SELECT_INTERNALLY | MHD_USE_IPv4,
+                       1080,
+                       &apc_all,
+                       NULL,
+                       &ahc_echo,
+                       "GET");
+  if (d == NULL)
+    return 1;
+  c = curl_easy_init();
+  curl_easy_setopt(c,
+                   CURLOPT_URL,
+                   "http://localhost:1080/hello_world");
+  curl_easy_setopt(c,
+                   CURLOPT_WRITEFUNCTION,
+                   &copyBuffer);
+  curl_easy_setopt(c,
+                   CURLOPT_WRITEDATA,
+                   &cbc);
+  curl_easy_setopt(c,
+                   CURLOPT_FAILONERROR,
+                   1);
+  curl_easy_setopt(c,
+                   CURLOPT_CONNECTTIMEOUT,
+                   15L);
+  /* NOTE: use of CONNECTTIMEOUT without also
+     setting NOSIGNAL results in really weird
+     crashes on my system! */
+  curl_easy_setopt(c,
+                   CURLOPT_NOSIGNAL,
+                   1);  
+  if (CURLE_OK != curl_easy_perform(c))
+    ret = 1;
+  curl_easy_cleanup(c);
+  if (cbc.pos != strlen("hello_world"))
+    ret = 2;
+  if (0 != strncmp("hello_world",
+                   cbc.buf,
+                   strlen("hello_world")))
+    ret = 3;
+  MHD_stop_daemon(d);
+  return ret;
+}
 
 int main(int argc,
          char * const * argv) {
-  return 0; /* 0 == pass */
+  unsigned int errorCount = 0;
+  
+  errorCount += testStartError();
+  errorCount += testStartStop();
+  errorCount += testGet();
+  return errorCount != 0; /* 0 == pass */
 }