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-2008, Eduardo Silva P.
   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
  36 /* Return the number of clients that can be attended
  37  * at the same time per worker thread
  38  */
  39 int mk_server_worker_capacity(int nworkers)
  40 {
  41     int max, avl;
  42     struct rlimit lim;
  43
  44     /* Limit by system */
  45     getrlimit(RLIMIT_NOFILE, &lim);
  46     max = lim.rlim_cur;
  47
  48     /* Minimum of fds needed by Monkey:
  49      * --------------------------------
  50      * 3 fds: stdin, stdout, stderr
  51      * 1 fd for main socket server
  52      * 1 fd for epoll array (per thread)
  53      * 1 fd for worker logger when writing to FS
  54      * 2 fd for worker logger pipe
  55      */
  56
  57     avl = max - (3 + 1 + nworkers + 1 + 2);
  58
  59     /* The avl is divided by two as we need to consider
  60      * a possible additional FD for each plugin working
  61      * on the same request.
  62      */
  63     return ((avl / 2) / nworkers);
  64 }
  65
  66 /* Here we launch the worker threads to attend clients */
  67 void mk_server_launch_workers()
  68 {
  69     int i;
  70
  71     /* Look for plugins thread key data */
  72     mk_plugin_preworker_calls();
  73
  74     /* Get each worker clients capacity based on FDs system limits */
  75     config->worker_capacity = mk_server_worker_capacity(config->workers);
  76
  77     /* Launch workers */
  78     for (i = 0; i < config->workers; i++) {
  79         mk_sched_launch_thread(config->worker_capacity);
  80     }
  81 }
  82
  83 void mk_server_loop(int server_fd)
  84 {
  85     int remote_fd;
  86     struct sockaddr_in sockaddr;
  87     struct sched_list_node *sched = sched_list;
  88     socklen_t socket_size = sizeof(struct sockaddr_in);
  89
  90     while (1) {
  91         remote_fd = accept(server_fd, (struct sockaddr *) &sockaddr,
  92                            &socket_size);
  93
  94         if (remote_fd == -1) {
  95             continue;
  96         }
  97
  98         /* Assign socket to worker thread */
  99         mk_sched_add_client(sched, remote_fd);
 100
 101         if (sched->next) {
 102             sched = sched->next;
 103         }
 104         else {
 105             sched = sched_list;
 106         }
 107     }
 108 }