src/testcurl/test_timeout.c - manifest_repos/libmicrohttpd - Git at Google

 /*
      This file is part of libmicrohttpd
      Copyright (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
      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_timeout.c
  * @brief  Testcase for libmicrohttpd PUT operations
  * @author Matthias Wachs
  */

 #include "MHD_config.h"
 #include "platform.h"
 #include <curl/curl.h>
 #include <microhttpd.h>
 #include <stdlib.h>
 #include <string.h>
 #include <time.h>

 #ifndef WINDOWS
 #include <unistd.h>
 #endif

 static int oneone;

 static int withTimeout = 1;

 static int withoutTimeout = 1;

 struct CBC
 {
   char *buf;
   size_t pos;
   size_t size;
 };


 static void
 termination_cb (void *cls,
 		struct MHD_Connection *connection,
 		void **con_cls,
 		enum MHD_RequestTerminationCode toe)
 {
   int *test = cls;

   switch (toe)
     {
     case MHD_REQUEST_TERMINATED_COMPLETED_OK :
       if (test == &withoutTimeout)
 	{
 	  withoutTimeout = 0;
 	}
       break;
     case MHD_REQUEST_TERMINATED_WITH_ERROR :
     case MHD_REQUEST_TERMINATED_READ_ERROR :
       break;
     case MHD_REQUEST_TERMINATED_TIMEOUT_REACHED :
       if (test == &withTimeout)
 	{
 	  withTimeout = 0;
 	}
       break;
     case MHD_REQUEST_TERMINATED_DAEMON_SHUTDOWN:
       break;
     case MHD_REQUEST_TERMINATED_CLIENT_ABORT:
       break;
     }
 }


 static size_t
 putBuffer (void *stream, size_t size, size_t nmemb, void *ptr)
 {
   unsigned int *pos = ptr;
   unsigned int wrt;

   wrt = size * nmemb;
   if (wrt > 8 - (*pos))
 	wrt = 8 - (*pos);
   memcpy (stream, &("Hello123"[*pos]), wrt);
   (*pos) += wrt;
   return wrt;
 }


 static size_t
 putBuffer_fail (void *stream, size_t size, size_t nmemb, void *ptr)
 {
   return 0;
 }


 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_Connection *connection,
           const char *url,
           const char *method,
           const char *version,
           const char *upload_data, size_t *upload_data_size,
           void **unused)
 {
   int *done = cls;
   struct MHD_Response *response;
   int ret;

   if (0 != strcmp ("PUT", method))
     return MHD_NO;              /* unexpected method */
   if ((*done) == 0)
     {
       if (*upload_data_size != 8)
         return MHD_YES;         /* not yet ready */
       if (0 == memcmp (upload_data, "Hello123", 8))
         {
           *upload_data_size = 0;
         }
       else
         {
           printf ("Invalid upload data `%8s'!\n", upload_data);
           return MHD_NO;
         }
       *done = 1;
       return MHD_YES;
     }
   response = MHD_create_response_from_buffer (strlen (url),
 					      (void *) url,
 					      MHD_RESPMEM_MUST_COPY);
   ret = MHD_queue_response (connection, MHD_HTTP_OK, response);
   MHD_destroy_response (response);
   return ret;
 }


 static int
 testWithoutTimeout ()
 {
   struct MHD_Daemon *d;
   CURL *c;
   char buf[2048];
   struct CBC cbc;
   unsigned int pos = 0;
   int done_flag = 0;
   CURLcode errornum;

   cbc.buf = buf;
   cbc.size = 2048;
   cbc.pos = 0;
   d = MHD_start_daemon (MHD_USE_SELECT_INTERNALLY | MHD_USE_DEBUG,
                         1080,
                         NULL, NULL, &ahc_echo, &done_flag,
                         MHD_OPTION_CONNECTION_TIMEOUT, 2,
                         MHD_OPTION_NOTIFY_COMPLETED, &termination_cb, &withTimeout,
                         MHD_OPTION_END);
   if (d == NULL)
     return 1;
   c = curl_easy_init ();
   curl_easy_setopt (c, CURLOPT_URL, "http://127.0.0.1:1080/hello_world");
   curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, &copyBuffer);
   curl_easy_setopt (c, CURLOPT_WRITEDATA, &cbc);
   curl_easy_setopt (c, CURLOPT_READFUNCTION, &putBuffer);
   curl_easy_setopt (c, CURLOPT_READDATA, &pos);
   curl_easy_setopt (c, CURLOPT_UPLOAD, 1L);
   curl_easy_setopt (c, CURLOPT_INFILESIZE_LARGE, (curl_off_t) 8L);
   curl_easy_setopt (c, CURLOPT_FAILONERROR, 1);
   curl_easy_setopt (c, CURLOPT_TIMEOUT, 150L);
   if (oneone)
     curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
   else
     curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
   curl_easy_setopt (c, CURLOPT_CONNECTTIMEOUT, 150L);
   // 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 != (errornum = curl_easy_perform (c)))
     {
       curl_easy_cleanup (c);
       MHD_stop_daemon (d);
       return 2;
     }
   curl_easy_cleanup (c);
   MHD_stop_daemon (d);
   if (cbc.pos != strlen ("/hello_world"))
     return 4;
   if (0 != strncmp ("/hello_world", cbc.buf, strlen ("/hello_world")))
     return 8;
   return 0;
 }

 static int
 testWithTimeout ()
 {
   struct MHD_Daemon *d;
   CURL *c;
   char buf[2048];
   struct CBC cbc;
   int done_flag = 0;
   CURLcode errornum;

   cbc.buf = buf;
   cbc.size = 2048;
   cbc.pos = 0;
   d = MHD_start_daemon (MHD_USE_SELECT_INTERNALLY | MHD_USE_DEBUG,
                         1080,
                         NULL, NULL, &ahc_echo, &done_flag,
                         MHD_OPTION_CONNECTION_TIMEOUT, 2,
                         MHD_OPTION_NOTIFY_COMPLETED, &termination_cb, &withoutTimeout,
                         MHD_OPTION_END);
   if (d == NULL)
     return 16;
   c = curl_easy_init ();
   curl_easy_setopt (c, CURLOPT_URL, "http://127.0.0.1:1080/hello_world");
   curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, &copyBuffer);
   curl_easy_setopt (c, CURLOPT_WRITEDATA, &cbc);
   curl_easy_setopt (c, CURLOPT_READFUNCTION, &putBuffer_fail);
   curl_easy_setopt (c, CURLOPT_READDATA, &testWithTimeout);
   curl_easy_setopt (c, CURLOPT_UPLOAD, 1L);
   curl_easy_setopt (c, CURLOPT_INFILESIZE_LARGE, (curl_off_t) 8L);
   curl_easy_setopt (c, CURLOPT_FAILONERROR, 1);
   curl_easy_setopt (c, CURLOPT_TIMEOUT, 150L);
   if (oneone)
     curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
   else
     curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
   curl_easy_setopt (c, CURLOPT_CONNECTTIMEOUT, 150L);
   // 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 != (errornum = curl_easy_perform (c)))
     {
       curl_easy_cleanup (c);
       MHD_stop_daemon (d);
       if (errornum == CURLE_GOT_NOTHING)
     	  /* mhd had the timeout */
     	  return 0;
       else
     	  /* curl had the timeout first */
     	  return 32;
     }
   curl_easy_cleanup (c);
   MHD_stop_daemon (d);
   return 64;
 }


 int
 main (int argc, char *const *argv)
 {
   unsigned int errorCount = 0;

   oneone = (NULL != strrchr (argv[0], (int) '/')) ?
     (NULL != strstr (strrchr (argv[0], (int) '/'), "11")) : 0;
   if (0 != curl_global_init (CURL_GLOBAL_WIN32))
     return 16;
   errorCount += testWithoutTimeout ();
   errorCount += testWithTimeout ();
   if (errorCount != 0)
     fprintf (stderr,
 	     "Error during test execution (code: %u)\n",
 	     errorCount);
   curl_global_cleanup ();
   if ((withTimeout == 0) && (withoutTimeout == 0))
     return 0;
   else
     return errorCount;       /* 0 == pass */
 }
	/*
	This file is part of libmicrohttpd
	Copyright (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
	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_timeout.c
	* @brief Testcase for libmicrohttpd PUT operations
	* @author Matthias Wachs
	*/

	#include "MHD_config.h"
	#include "platform.h"
	#include <curl/curl.h>
	#include <microhttpd.h>
	#include <stdlib.h>
	#include <string.h>
	#include <time.h>

	#ifndef WINDOWS
	#include <unistd.h>
	#endif

	static int oneone;

	static int withTimeout = 1;

	static int withoutTimeout = 1;

	struct CBC
	{
	char *buf;
	size_t pos;
	size_t size;
	};


	static void
	termination_cb (void *cls,
	struct MHD_Connection *connection,
	void **con_cls,
	enum MHD_RequestTerminationCode toe)
	{
	int *test = cls;

	switch (toe)
	{
	case MHD_REQUEST_TERMINATED_COMPLETED_OK :
	if (test == &withoutTimeout)
	{
	withoutTimeout = 0;
	}
	break;
	case MHD_REQUEST_TERMINATED_WITH_ERROR :
	case MHD_REQUEST_TERMINATED_READ_ERROR :
	break;
	case MHD_REQUEST_TERMINATED_TIMEOUT_REACHED :
	if (test == &withTimeout)
	{
	withTimeout = 0;
	}
	break;
	case MHD_REQUEST_TERMINATED_DAEMON_SHUTDOWN:
	break;
	case MHD_REQUEST_TERMINATED_CLIENT_ABORT:
	break;
	}
	}


	static size_t
	putBuffer (void stream, size_t size, size_t nmemb, void ptr)
	{
	unsigned int *pos = ptr;
	unsigned int wrt;

	wrt = size * nmemb;
	if (wrt > 8 - (*pos))
	wrt = 8 - (*pos);
	memcpy (stream, &("Hello123"[*pos]), wrt);
	(*pos) += wrt;
	return wrt;
	}


	static size_t
	putBuffer_fail (void stream, size_t size, size_t nmemb, void ptr)
	{
	return 0;
	}


	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_Connection *connection,
	const char *url,
	const char *method,
	const char *version,
	const char upload_data, size_t upload_data_size,
	void **unused)
	{
	int *done = cls;
	struct MHD_Response *response;
	int ret;

	if (0 != strcmp ("PUT", method))
	return MHD_NO; /* unexpected method */
	if ((*done) == 0)
	{
	if (*upload_data_size != 8)
	return MHD_YES; /* not yet ready */
	if (0 == memcmp (upload_data, "Hello123", 8))
	{
	*upload_data_size = 0;
	}
	else
	{
	printf ("Invalid upload data `%8s'!\n", upload_data);
	return MHD_NO;
	}
	*done = 1;
	return MHD_YES;
	}
	response = MHD_create_response_from_buffer (strlen (url),
	(void *) url,
	MHD_RESPMEM_MUST_COPY);
	ret = MHD_queue_response (connection, MHD_HTTP_OK, response);
	MHD_destroy_response (response);
	return ret;
	}


	static int
	testWithoutTimeout ()
	{
	struct MHD_Daemon *d;
	CURL *c;
	char buf[2048];
	struct CBC cbc;
	unsigned int pos = 0;
	int done_flag = 0;
	CURLcode errornum;

	cbc.buf = buf;
	cbc.size = 2048;
	cbc.pos = 0;
	d = MHD_start_daemon (MHD_USE_SELECT_INTERNALLY \| MHD_USE_DEBUG,
	1080,
	NULL, NULL, &ahc_echo, &done_flag,
	MHD_OPTION_CONNECTION_TIMEOUT, 2,
	MHD_OPTION_NOTIFY_COMPLETED, &termination_cb, &withTimeout,
	MHD_OPTION_END);
	if (d == NULL)
	return 1;
	c = curl_easy_init ();
	curl_easy_setopt (c, CURLOPT_URL, "http://127.0.0.1:1080/hello_world");
	curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, &copyBuffer);
	curl_easy_setopt (c, CURLOPT_WRITEDATA, &cbc);
	curl_easy_setopt (c, CURLOPT_READFUNCTION, &putBuffer);
	curl_easy_setopt (c, CURLOPT_READDATA, &pos);
	curl_easy_setopt (c, CURLOPT_UPLOAD, 1L);
	curl_easy_setopt (c, CURLOPT_INFILESIZE_LARGE, (curl_off_t) 8L);
	curl_easy_setopt (c, CURLOPT_FAILONERROR, 1);
	curl_easy_setopt (c, CURLOPT_TIMEOUT, 150L);
	if (oneone)
	curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
	else
	curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
	curl_easy_setopt (c, CURLOPT_CONNECTTIMEOUT, 150L);
	// 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 != (errornum = curl_easy_perform (c)))
	{
	curl_easy_cleanup (c);
	MHD_stop_daemon (d);
	return 2;
	}
	curl_easy_cleanup (c);
	MHD_stop_daemon (d);
	if (cbc.pos != strlen ("/hello_world"))
	return 4;
	if (0 != strncmp ("/hello_world", cbc.buf, strlen ("/hello_world")))
	return 8;
	return 0;
	}

	static int
	testWithTimeout ()
	{
	struct MHD_Daemon *d;
	CURL *c;
	char buf[2048];
	struct CBC cbc;
	int done_flag = 0;
	CURLcode errornum;

	cbc.buf = buf;
	cbc.size = 2048;
	cbc.pos = 0;
	d = MHD_start_daemon (MHD_USE_SELECT_INTERNALLY \| MHD_USE_DEBUG,
	1080,
	NULL, NULL, &ahc_echo, &done_flag,
	MHD_OPTION_CONNECTION_TIMEOUT, 2,
	MHD_OPTION_NOTIFY_COMPLETED, &termination_cb, &withoutTimeout,
	MHD_OPTION_END);
	if (d == NULL)
	return 16;
	c = curl_easy_init ();
	curl_easy_setopt (c, CURLOPT_URL, "http://127.0.0.1:1080/hello_world");
	curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, &copyBuffer);
	curl_easy_setopt (c, CURLOPT_WRITEDATA, &cbc);
	curl_easy_setopt (c, CURLOPT_READFUNCTION, &putBuffer_fail);
	curl_easy_setopt (c, CURLOPT_READDATA, &testWithTimeout);
	curl_easy_setopt (c, CURLOPT_UPLOAD, 1L);
	curl_easy_setopt (c, CURLOPT_INFILESIZE_LARGE, (curl_off_t) 8L);
	curl_easy_setopt (c, CURLOPT_FAILONERROR, 1);
	curl_easy_setopt (c, CURLOPT_TIMEOUT, 150L);
	if (oneone)
	curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
	else
	curl_easy_setopt (c, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
	curl_easy_setopt (c, CURLOPT_CONNECTTIMEOUT, 150L);
	// 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 != (errornum = curl_easy_perform (c)))
	{
	curl_easy_cleanup (c);
	MHD_stop_daemon (d);
	if (errornum == CURLE_GOT_NOTHING)
	/* mhd had the timeout */
	return 0;
	else
	/* curl had the timeout first */
	return 32;
	}
	curl_easy_cleanup (c);
	MHD_stop_daemon (d);
	return 64;
	}


	int
	main (int argc, char const argv)
	{
	unsigned int errorCount = 0;

	oneone = (NULL != strrchr (argv[0], (int) '/')) ?
	(NULL != strstr (strrchr (argv[0], (int) '/'), "11")) : 0;
	if (0 != curl_global_init (CURL_GLOBAL_WIN32))
	return 16;
	errorCount += testWithoutTimeout ();
	errorCount += testWithTimeout ();
	if (errorCount != 0)
	fprintf (stderr,
	"Error during test execution (code: %u)\n",
	errorCount);
	curl_global_cleanup ();
	if ((withTimeout == 0) && (withoutTimeout == 0))
	return 0;
	else
	return errorCount; /* 0 == pass */
	}