src/server.c

   1 /* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2
   3 /*  Monkey HTTP Daemon
   4  *  ------------------
   5  *  Copyright (C) 2001-2010, Eduardo Silva P. <edsiper@gmail.com>
   6  *
   7  *  This program is free software; you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License as published by
   9  *  the Free Software Foundation; either version 2 of the License, or
  10  *  (at your option) any later version.
  11  *
  12  *  This program is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *  GNU Library General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU General Public License
  18  *  along with this program; if not, write to the Free Software
  19  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  20  */
  21
  22 #include <stdio.h>
  23 #include <sys/socket.h>
  24 #include <netinet/in.h>
  25
  26 #include <sys/time.h>
  27 #include <sys/resource.h>
  28
  29 #include "monkey.h"
  30 #include "config.h"
  31 #include "scheduler.h"
  32 #include "epoll.h"
  33 #include "socket.h"
  34 #include "plugin.h"
  35 #include "utils.h"
  36
  37 /* Return the number of clients that can be attended
  38  * at the same time per worker thread
  39  */
  40 int mk_server_worker_capacity(int nworkers)
  41 {
  42     int max, avl;
  43     struct rlimit lim;
  44
  45     /* Limit by system */
  46     getrlimit(RLIMIT_NOFILE, &lim);
  47     max = lim.rlim_cur;
  48
  49     /* Minimum of fds needed by Monkey:
  50      * --------------------------------
  51      * 3 fds: stdin, stdout, stderr
  52      * 1 fd for main socket server
  53      * 1 fd for epoll array (per thread)
  54      * 1 fd for worker logger when writing to FS
  55      * 2 fd for worker logger pipe
  56      */
  57
  58     avl = max - (3 + 1 + nworkers + 1 + 2);
  59
  60     /* The avl is divided by two as we need to consider
  61      * a possible additional FD for each plugin working
  62      * on the same request.
  63      */
  64     return ((avl / 2) / nworkers);
  65 }
  66
  67 /* Here we launch the worker threads to attend clients */
  68 void mk_server_launch_workers()
  69 {
  70     int i;
  71
  72     /* Look for plugins thread key data */
  73     mk_plugin_preworker_calls();
  74
  75     /* Get each worker clients capacity based on FDs system limits */
  76     config->worker_capacity = mk_server_worker_capacity(config->workers);
  77
  78     /* Launch workers */
  79     for (i = 0; i < config->workers; i++) {
  80         mk_sched_launch_thread(config->worker_capacity);
  81     }
  82 }
  83
  84 void mk_server_loop(int server_fd)
  85 {
  86     int remote_fd;
  87     struct sockaddr_in sockaddr;
  88     struct sched_list_node *sched = sched_list;
  89     socklen_t socket_size = sizeof(struct sockaddr_in);
  90
  91     while (1) {
  92         remote_fd = accept(server_fd, (struct sockaddr *) &sockaddr,
  93                            &socket_size);
  94
  95         if (remote_fd == -1) {
  96             continue;
  97         }
  98
  99 #ifdef TRACE
 100         MK_TRACE("New connection arrived: FD %i", remote_fd);
 101 #endif
 102
 103         /* Assign socket to worker thread */
 104         mk_sched_add_client(sched, remote_fd);
 105
 106         if (sched->next) {
 107             sched = sched->next;
 108         }
 109         else {
 110             sched = sched_list;
 111         }
 112     }
 113 }