/*
     This file is part of libmicrohttpd
     (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 daemontest.c
 * @brief  Testcase for libmicrohttpd GET operations
 * @author Christian Grothoff
 */

#include "config.h"
#include "curl/curl.h"
#include "microhttpd.h"
#include <stdlib.h>
#include <unistd.h>
#include <string.h>

static int apc_all(void * cls,
		   const struct sockaddr * addr,
		   socklen_t addrlen) {
  return MHD_YES;
}

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 * upload_data,
		    unsigned int * upload_data_size) {
  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 testInternalGet() {
  struct MHD_Daemon * d;
  CURL * c;
  char buf[2048];
  struct CBC cbc;

  cbc.buf = buf;
  cbc.size = 2048;
  cbc.pos = 0;
  d = MHD_start_daemon(MHD_USE_SELECT_INTERNALLY | MHD_USE_IPv4 | MHD_USE_DEBUG,
		       1083,
		       &apc_all,
		       NULL,
		       &ahc_echo,
		       "GET");
  if (d == NULL)
    return 1;

  if (MHD_run(d) == MHD_NO) {
    MHD_stop_daemon(d);
    return 2;
  }
  
  c = curl_easy_init();
  curl_easy_setopt(c,
		   CURLOPT_URL,
		   "http://localhost:1083/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_TIMEOUT,
		   15L);
  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))
    return 3;
    
  curl_easy_cleanup(c);
  
  if (cbc.pos != strlen("hello_world"))
    return 4;
  
  if (0 != strncmp("hello_world",
		   cbc.buf,
		   strlen("hello_world")))
    return 5;
  
  MHD_stop_daemon(d);
  
  return 0;
}

static int testMultithreadedGet() {
  struct MHD_Daemon * d;
  CURL * c;
  char buf[2048];
  struct CBC cbc;

  cbc.buf = buf;
  cbc.size = 2048;
  cbc.pos = 0;
  d = MHD_start_daemon(MHD_USE_THREAD_PER_CONNECTION | MHD_USE_IPv4 | MHD_USE_DEBUG,
		       1084,
		       &apc_all,
		       NULL,
		       &ahc_echo,
		       "GET");
  if (d == NULL)
    return 1;

  if(MHD_run(d) == MHD_NO)
    return 2;
 
  
  c = curl_easy_init();
  curl_easy_setopt(c,
		   CURLOPT_URL,
		   "http://localhost:1084/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))
    return 3;
  curl_easy_cleanup(c);
  if (cbc.pos != strlen("hello_world"))
    return 4;
  
  if (0 != strncmp("hello_world",
		   cbc.buf,
		   strlen("hello_world")))
    return 5;
    
  MHD_stop_daemon(d);
  
  return 0;
}

int main(int argc,
	 char * const * argv) {
  unsigned int errorCount = 0;
  errorCount += testInternalGet();
  errorCount += testMultithreadedGet();  
  if (errorCount != 0)
    fprintf(stderr, 
	    "Error (code: %u)\n", 
	    errorCount);
  return errorCount != 0; /* 0 == pass */
}