path: root/nanohttp/nanohttp-server.c

/******************************************************************
*  $Id: nanohttp-server.c,v 1.26 2004/10/15 13:29:36 snowdrop Exp $
*
* CSOAP Project:  A http client/server library in C
* Copyright (C) 2003  Ferhat Ayaz
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library 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
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA  02111-1307, USA.
*
* Email: ayaz@jprogrammer.net
******************************************************************/
#include <nanohttp/nanohttp-server.h>

#include <stdio.h>
#include <stdlib.h>
#include <signal.h>

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#ifndef WIN32
/* According to POSIX 1003.1-2001 */
#include <sys/select.h>

/* According to earlier standards */
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>

#endif

#ifdef MEM_DEBUG
#include <utils/alloc.h>
#endif

typedef struct _conndata
{
	hsocket_t sock;
#ifdef WIN32
	HANDLE tid;
#else
	pthread_t tid;
  	pthread_attr_t attr;
#endif
	time_t atime;
}conndata_t;

/*
 * -----------------------------------------------------
 * nano httpd
 * internally globals
 * -----------------------------------------------------
 */
static int _httpd_port = 10000;
static int _httpd_max_connections = 20;
static int _httpd_max_idle = 120;
static hsocket_t _httpd_socket;
static hservice_t *_httpd_services_head = NULL;
static hservice_t *_httpd_services_tail = NULL;
static int _httpd_run = 1;
#ifdef WIN32
static DWORD _httpd_terminate_signal = CTRL_C_EVENT;
#else
static int _httpd_terminate_signal = SIGTERM;
#endif
static conndata_t *_httpd_connection;


#ifdef WIN32
static void WSAReaper(void *x);
#endif

/*
 * -----------------------------------------------------
 * FUNCTION: httpd_init
 * NOTE: This will be called from soap_server_init_args()
 * -----------------------------------------------------
 */
int
httpd_init (int argc, char *argv[])
{
  int i, status;
  status = hsocket_module_init ();
  if (status != 0)
    return status;

  /* write argument information */
  log_verbose1 ("Arguments:");
  for (i = 0; i < argc; i++)
    log_verbose3 ("argv[%i] = '%s'", i, SAVE_STR (argv[i]));

  /* initialize from arguments */
  for (i = 0; i < argc; i++)
  {
    if (!strcmp (argv[i], NHTTPD_ARG_PORT) && i < argc - 1)
    {
      _httpd_port = atoi (argv[i + 1]);
    }
    else if (!strcmp (argv[i], NHTTPD_ARG_TERMSIG) && i < argc - 1)
    {
      _httpd_terminate_signal = atoi (argv[i + 1]);
    }
    else if (!strcmp (argv[i], NHTTPD_ARG_MAXCONN) && i < argc - 1)
    {
      _httpd_max_connections = atoi (argv[i + 1]);
    }
  }

  log_verbose2 ("socket bind to port '%d'", _httpd_port);

  /* init built-in services */

  /* httpd_register("/httpd/list", service_list); */

  _httpd_connection = calloc (_httpd_max_connections, sizeof (conndata_t));
  for (i = 0; i < _httpd_max_connections; i++)
  {
    memset ((char *) &_httpd_connection[i], 0, sizeof (_httpd_connection[i]));
  }

#ifdef WIN32
  if (_beginthread (WSAReaper, 0, NULL) == -1)
  {
    log_error1 ("Winsock reaper thread failed to start");
    return (-1);
  }
#endif

  /* create socket */
  hsocket_init (&_httpd_socket);
  status = hsocket_bind (&_httpd_socket, _httpd_port);

  return status;
}

/*
 * -----------------------------------------------------
 * FUNCTION: httpd_register
 * -----------------------------------------------------
 */

int
httpd_register (const char *ctx, httpd_service func)
{
  hservice_t *service;

  service = (hservice_t *) malloc (sizeof (hservice_t));
  service->next = NULL;
  service->func = func;
  strcpy (service->ctx, ctx);

  log_verbose3 ("register service:t(%p):%s", service, SAVE_STR (ctx));
  if (_httpd_services_head == NULL)
  {
    _httpd_services_head = _httpd_services_tail = service;
  }
  else
  {
    _httpd_services_tail->next = service;
    _httpd_services_tail = service;
  }

  return 1;
}


/*
 * -----------------------------------------------------
 * FUNCTION: httpd_services
 * -----------------------------------------------------
 */
hservice_t *
httpd_services ()
{
  return _httpd_services_head;
}

/*
 * -----------------------------------------------------
 * FUNCTION: httpd_services
 * -----------------------------------------------------
 */
static 
void hservice_free(hservice_t *service)
{
  free(service);
}

/*
 * -----------------------------------------------------
 * FUNCTION: httpd_find_service
 * -----------------------------------------------------
 */
static hservice_t *
httpd_find_service (const char *ctx)
{
  hservice_t *cur = _httpd_services_head;

  while (cur != NULL)
  {
    if (!strcmp (cur->ctx, ctx))
    {
      return cur;
    }
    cur = cur->next;
  }

  return NULL;
}


/*
 * -----------------------------------------------------
 * FUNCTION: httpd_response_set_content_type
 * -----------------------------------------------------
 */
void
httpd_response_set_content_type (httpd_conn_t * res, const char *content_type)
{
  strncpy (res->content_type, content_type, 25);
}


/*
 * -----------------------------------------------------
 * FUNCTION: httpd_response_send_header
 * -----------------------------------------------------
 */
int
httpd_send_header (httpd_conn_t * res, int code, const char *text)
{
  struct tm stm;
  time_t nw;
  char buffer[255];
  char header[1024];
  hpair_t *cur;
  int status;

  /* set status code */
  sprintf (header, "HTTP/1.1 %d %s\r\n", code, text);

  /* set date */
  nw = time (NULL);
  localtime_r (&nw, &stm);
  strftime (buffer, 255, "Date: %a, %d %b %y %T GMT", &stm);
  strcat (header, buffer);
  strcat (header, "\r\n");

  /* set content-type */
  /*
   * if (res->content_type[0] == '\0') { strcat(header, "Content-Type:
   * text/html\r\n"); } else { sprintf(buffer, "Content-Type: %s\r\n",
   * res->content_type); strcat(header, buffer); }
   */

  /* set server name */
  strcat (header, "Server: Nano HTTPD library\r\n");

  /* set _httpd_connection status */
  strcat (header, "Connection: close\r\n");

  /* add pairs */
  cur = res->header;
  while (cur != NULL)
  {
    sprintf (buffer, "%s: %s\r\n", cur->key, cur->value);
    strcat (header, buffer);
    cur = cur->next;
  }

  /* set end of header */
  strcat (header, "\r\n");

  /* send header */
  status = hsocket_nsend (res->sock, header, strlen (header));

  res->out = http_output_stream_new (res->sock, res->header);
  return status;
}


int
httpd_send_internal_error (httpd_conn_t * conn, const char *errmsg)
{
  const char *template1 =
    "<html><body><h3>Error!</h3><hr> Message: '%s' </body></html>\r\n";

  char buffer[4064];
  sprintf (buffer, template1, errmsg);
  httpd_send_header (conn, 500, "INTERNAL");
  return send (conn->sock, buffer, strlen (buffer), 0);
}

/*
 * -----------------------------------------------------
 * FUNCTION: httpd_request_print
 * -----------------------------------------------------
 */
static void
httpd_request_print (hrequest_t * req)
{
  hpair_t *pair;

  log_verbose1 ("++++++ Request +++++++++");
  log_verbose2 (" Method : '%s'", (req->method == HTTP_REQUEST_POST)?"POST":"GET");
  log_verbose2 (" Path   : '%s'", req->path);
  log_verbose2 (" Spec   : '%s'", (req->version==HTTP_1_0)?"HTTP/1.0":"HTTP/1.1");
  log_verbose1 (" Parsed query string :");

  pair = req->query;
  while (pair != NULL)
  {
    log_verbose3 (" %s = '%s'", pair->key, pair->value);
    pair = pair->next;
  }
  log_verbose1 ("++++++++++++++++++++++++");

}


httpd_conn_t *httpd_new(hsocket_t sock)
{
  httpd_conn_t *conn = (httpd_conn_t *) malloc (sizeof (httpd_conn_t));
  conn->sock = sock;
  conn->out = NULL;
  conn->content_type[0] = '\0';
  conn->header = NULL;

  return conn;
}


void httpd_free(httpd_conn_t *conn)
{
  if (conn->out != NULL)
    http_output_stream_free(conn->out);

  if (conn->header != NULL)
    hpairnode_free_deep(conn->header);

  free(conn);
}

/*
 * -----------------------------------------------------
 * FUNCTION: httpd_session_main
 * -----------------------------------------------------
 */
#ifdef WIN32
static unsigned _stdcall
httpd_session_main (void *data)
#else
static void *
httpd_session_main (void *data)
#endif
{
  conndata_t *conn = (conndata_t *) data;
  const char *msg = "SESSION 1.0\n";
  int len = strlen (msg);
  char ch[2];
  char buffer[256];             /* temp buffer for recv() */
  char header[4064];            /* received header */
  int total;                    /* result from recv() */
  int headerreached = 0;        /* whether reach header * "\n\n" */
  hrequest_t *req = NULL;       /* only for test */
  httpd_conn_t *rconn;
  hservice_t *service = NULL;
  long content_length = 0;

  header[0] = '\0';
  len = 0;


  log_verbose1 ("starting httpd_session_main()");
  conn->atime = time ((time_t) 0);
  /* call the service */
/*  req = hrequest_new_from_buffer (header);*/
  rconn = httpd_new(conn->sock);

  req = hrequest_new_from_socket (conn->sock);
  if (req == NULL)
  {
    httpd_send_internal_error (rconn, "Request parse error!");
  }
  else
  {
    httpd_request_print (req);


    service = httpd_find_service (req->path);
    if (service != NULL)
    {
      log_verbose2 ("service '%s' found", req->path);
      if (service->func != NULL)
      {
        service->func (rconn, req);
      }
      else
      {
        sprintf (buffer,
                 "service '%s' not registered properly (func == NULL)",
                 req->path);
        log_verbose1 (buffer);
        httpd_send_internal_error (rconn, buffer);
      }
    }
    else
    {
      sprintf (buffer, "service '%s' not found", req->path);
      log_verbose1 (buffer);
      httpd_send_internal_error (rconn, buffer);
    }

    /* httpd_response_free(res); */
    /*hrequest_free (req);*/
  }

  hsocket_close(conn->sock);
  conn->sock = 0;
  hrequest_free(req);
  httpd_free(rconn);
#ifdef WIN32
  CloseHandle ((HANDLE) conn->tid);
  _endthread ();
  return 0;
#else
  pthread_exit (NULL);
  return service;
#endif
}

int
httpd_set_header (httpd_conn_t * conn, const char *key, const char *value)
{
  hpair_t *p;

  if (conn == NULL)
  {
    log_warn1 ("Connection object is NULL");
    return 0;
  }
  p = conn->header;
  while (p != NULL)
  {
    if (p->key != NULL)
    {
      if (!strcmp (p->key, key))
      {
        free (p->value);
        p->value = (char *) malloc (strlen (value) + 1);
        strcpy (p->value, value);
        return 1;
      }
    }
    p = p->next;
  }

  conn->header = hpairnode_new (key, value, conn->header);
  return 0;
}


void httpd_set_headers(httpd_conn_t *conn, hpair_t *header)
{
  while (header) {
    httpd_set_header(conn, header->key, header->value);
    header = header->next;
  }
}

/*
 * -----------------------------------------------------
 * FUNCTION: httpd_term
 * -----------------------------------------------------
 */
#ifdef WIN32
BOOL WINAPI
httpd_term (DWORD sig)
{
  log_debug2 ("Got signal %d", sig);
  if (sig == _httpd_terminate_signal)
    _httpd_run = 0;
  return TRUE;
}

#else

void
httpd_term (int sig)
{
  log_debug2 ("Got signal %d", sig);
  if (sig == _httpd_terminate_signal)
    _httpd_run = 0;
}

#endif


/*
 * -----------------------------------------------------
 * FUNCTION: _httpd_register_signal_handler
 * -----------------------------------------------------
 */
static void
_httpd_register_signal_handler ()
{
  log_verbose2 ("registering termination signal handler (SIGNAL:%d)",
                _httpd_terminate_signal);
#ifdef WIN32
  if (SetConsoleCtrlHandler ((PHANDLER_ROUTINE) httpd_term, TRUE) == FALSE)
  {
    log_error1 ("Unable to install console event handler!");
  }

#else
  signal (_httpd_terminate_signal, httpd_term);
#endif
}




/*--------------------------------------------------
FUNCTION: _httpd_wait_for_empty_conn
----------------------------------------------------*/
static conndata_t *
_httpd_wait_for_empty_conn ()
{
  int i;
  for (i = 0;; i++)
  {
    if (!_httpd_run)
      return NULL;

    if (i >= _httpd_max_connections)
    {
      system_sleep (1);
      i = 0;
    }
    else if (_httpd_connection[i].sock == 0)
    {
      break;
    }
  }

  return &_httpd_connection[i];
}

/*
 * -----------------------------------------------------
 * FUNCTION: _httpd_start_thread
 * -----------------------------------------------------
 */
static void
_httpd_start_thread (conndata_t * conn)
{
  int err;

#ifdef WIN32
  conn->tid =
    (HANDLE) _beginthreadex (NULL, 65535, httpd_session_main, conn, 0, &err);
#else
  pthread_attr_init (&(conn->attr));
#ifdef PTHREAD_CREATE_DETACHED
  pthread_attr_setdetachstate (&(conn->attr), PTHREAD_CREATE_DETACHED);
#endif
  err = pthread_create (&(conn->tid), &(conn->attr), httpd_session_main,
                        conn);
  if (err)
  {
    log_error2 ("Error creating thread: ('%d')", err);
  }
#endif
}



/*
 * -----------------------------------------------------
 * FUNCTION: httpd_run
 * -----------------------------------------------------
 */

int
httpd_run ()
{
  int err;
  conndata_t *conn;
  fd_set fds;
  struct timeval timeout;

  log_verbose1 ("starting run routine");

  timeout.tv_sec = 1;
  timeout.tv_usec = 0;

  /* listen to port */
  err = hsocket_listen (_httpd_socket);
  if (err != H_OK)
  {
    log_error2 ("httpd_run(): '%d'", err);
    return err;
  }
  log_verbose2 ("listening to port '%d'", _httpd_port);

  /* register signal handler */
  _httpd_register_signal_handler ();

  /* make the socket non blocking */
  err = hsocket_block (_httpd_socket, 0);
  if (err != H_OK)
  {
    log_error2 ("httpd_run(): '%d'", err);
    return err;
  }


  while (_httpd_run)
  {
    /* Get an empty connection struct */
    conn = _httpd_wait_for_empty_conn ();
    if (!_httpd_run)
      break;


    /* Wait for a socket to accept */
    while (_httpd_run)
    {
      /* zero and set file descriptior */
      FD_ZERO (&fds);
      FD_SET (_httpd_socket, &fds);

      /* select socket descriptor */
      switch (select (_httpd_socket + 1, &fds, NULL, NULL, &timeout))
      {
      case 0:
        /* descriptor is not ready */
        continue;
      case -1:
        /* got a signal? */
        continue;
      default:
        /* no nothing */
        break;
      }
      if (FD_ISSET (_httpd_socket, &fds))
      {
        break;
      }
    }

    /* check signal status */
    if (!_httpd_run)
      break;

    /* Accept a socket */
    err = hsocket_accept (_httpd_socket, &(conn->sock));
    if (err != H_OK)
    {
      log_error2 ("Can not accept socket: %d", err);
      return -1;                /* this is hard core! */
    }

    /* Now start a thread */
    _httpd_start_thread (conn);
  }
  free (_httpd_connection);
  return 0;
}

void httpd_destroy()
{
  hservice_t *tmp, *cur = _httpd_services_head;

  while (cur != NULL)
  {
    tmp = cur->next;
    hservice_free(cur);
    cur = tmp;
  }

}

#ifdef WIN32

static 
void WSAReaper(void *x)
{
	short int connections;
	short int i;
	char junk[10];
	int rc;
	time_t ctime;

	for (;;) {
		connections=0;
		ctime=time((time_t)0);
		for (i=0;i<_httpd_max_connections;i++) {
 			if (_httpd_connection[i].tid==0) continue;
			GetExitCodeThread((HANDLE)_httpd_connection[i].tid,(PDWORD) &rc);
			if (rc!=STILL_ACTIVE) continue;
			connections++;
			if ((ctime-_httpd_connection[i].atime<_httpd_max_idle)||
				(_httpd_connection[i].atime==0)) continue;
			log_verbose3("Reaping socket %u from (runtime ~= %d seconds)", 
				_httpd_connection[i].sock, ctime-_httpd_connection[i].atime);
			shutdown(_httpd_connection[i].sock, 2);
			while (recv(_httpd_connection[i].sock, junk, sizeof(junk), 0)>0) { };
			closesocket(_httpd_connection[i].sock);
			_httpd_connection[i].sock = 0;
			TerminateThread(_httpd_connection[i].tid, (DWORD)&rc);
			CloseHandle(_httpd_connection[i].tid);
			memset((char *)&_httpd_connection[i], 0, sizeof(_httpd_connection[i]));
		}
		Sleep(100);
	}
	return;
}

#endif

unsigned char *
httpd_get_postdata (httpd_conn_t * conn, hrequest_t * req, long *received,
                    long max)
{
  char *content_length_str;
  long content_length = 0;
  long total = 0;
  unsigned char *postdata = NULL;

  if (req->method == HTTP_REQUEST_POST)
  {

    content_length_str =
      hpairnode_get_ignore_case (req->header, HEADER_CONTENT_LENGTH);

    if (content_length_str != NULL)
      content_length = atol (content_length_str);

  }
  else
  {
    log_warn1 ("Not a POST method");
    return NULL;
  }

  if (content_length > max && max != -1)
    return NULL;

  if (content_length == 0)
  {
    *received = 0;
    postdata = (char *) malloc (1);
    postdata[0] = '\0';
    return postdata;
  }
  postdata = (unsigned char *) malloc (content_length + 1);
  if (postdata == NULL)
  {
    log_error1 ("Not enough memory");
    return NULL;
  }
  if (http_input_stream_read(req->in, postdata, (int) content_length) > 0)
  {
    *received = content_length;
    postdata[content_length] = '\0';
    return postdata;
  }
  free (postdata);
  return NULL;
}




/*
  MIME support httpd_mime_* function set
*/

static void
_httpd_mime_get_boundary (httpd_conn_t * conn, char *dest)
{
  sprintf (dest, "---=.Part_NH_%p", conn);
  log_verbose2 ("boundary= \"%s\"", dest);
}


/**
  Begin MIME multipart/related POST 
  Returns: H_OK  or error flag
*/
int
httpd_mime_send_header (httpd_conn_t * conn,
                        const char *related_start,
                        const char *related_start_info,
                        const char *related_type, int code, const char *text)
{
  int status;
  char buffer[300];
  char temp[250];
  char boundary[250];
  hpair_t *header;

  /* 
     Set Content-type 
     Set multipart/related parameter
     type=..; start=.. ; start-info= ..; boundary=...

   */
  sprintf (buffer, "multipart/related;");

  if (related_type)
  {
    snprintf (temp, 75, " type=\"%s\";", related_type);
    strcat (buffer, temp);
  }

  if (related_start)
  {
    snprintf (temp, 250, " start=\"%s\";", related_start);
    strcat (buffer, temp);
  }

  if (related_start_info)
  {
    snprintf (temp, 250, " start-info=\"%s\";", related_start_info);
    strcat (buffer, temp);
  }

  _httpd_mime_get_boundary (conn, boundary);
  snprintf (temp, 250, " boundary=\"%s\"", boundary);
  strcat (buffer, temp);

  httpd_set_header (conn, HEADER_CONTENT_TYPE, buffer);

  return httpd_send_header (conn, code, text);
}


/**
  Send boundary and part header and continue 
  with next part
*/
int
httpd_mime_next (httpd_conn_t * conn,
                 const char *content_id,
                 const char *content_type, const char *transfer_encoding)
{
  int status;
  char buffer[512];
  char boundary[75];

  /* Get the boundary string */
  _httpd_mime_get_boundary (conn, boundary);
  sprintf (buffer, "\r\n--%s\r\n", boundary);

  /* Send boundary */
  status = http_output_stream_write (conn->out,
                                     (const byte_t *) buffer,
                                     strlen (buffer));

  if (status != H_OK)
    return status;

  /* Send Content header */
  sprintf (buffer, "%s: %s\r\n%s: %s\r\n%s: %s\r\n\r\n",
           HEADER_CONTENT_TYPE, content_type ? content_type : "text/plain",
           HEADER_CONTENT_TRANSFER_ENCODING,
           transfer_encoding ? transfer_encoding : "binary",
           HEADER_CONTENT_ID,
           content_id ? content_id : "<content-id-not-set>");

  status = http_output_stream_write (conn->out,
                                     (const byte_t *) buffer,
                                     strlen (buffer));

  return status;
}

/**
  Send boundary and part header and continue 
  with next part
*/
int
httpd_mime_send_file (httpd_conn_t * conn,
                      const char *content_id,
                      const char *content_type,
                      const char *transfer_encoding, const char *filename)
{
  int status;
  FILE *fd = fopen (filename, "rb");
  byte_t buffer[MAX_FILE_BUFFER_SIZE];
  size_t size;

  if (fd == NULL)
    return FILE_ERROR_OPEN;

  status =
    httpd_mime_next (conn, content_id, content_type, transfer_encoding);
  if (status != H_OK)
  {
    fclose (fd);
    return status;
  }

  while (!feof (fd))
  {
    size = fread (buffer, 1, MAX_FILE_BUFFER_SIZE, fd);
    if (size == -1)
    {
      fclose (fd);
      return FILE_ERROR_READ;
    }

    status = http_output_stream_write (conn->out, buffer, size);
    if (status != H_OK) {
      fclose (fd);
      return status;
    }
  }

  fclose (fd);
  return H_OK;
}

/**
  Finish MIME request 
  Returns: H_OK  or error flag
*/
int
httpd_mime_end (httpd_conn_t * conn)
{
  int status;
  char buffer[512];
  char boundary[75];

  /* Get the boundary string */
  _httpd_mime_get_boundary (conn, boundary);
  sprintf (buffer, "\r\n--%s--\r\n\r\n", boundary);

  /* Send boundary */
  status = http_output_stream_write (conn->out,
                                     (const byte_t *) buffer,
                                     strlen (buffer));

  if (status != H_OK)
    return status;

  /* Flush put stream */
  status = http_output_stream_flush (conn->out);

  return status;
}