src/server.c

   1 /* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 8 -*- */
   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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, 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         return ((avl/2)/nworkers);
  59 }
  60
  61 /* Here we launch the worker threads to attend clients */
  62 void mk_server_launch_workers()
  63 {
  64         int i;
  65
  66         config->worker_capacity = mk_server_worker_capacity(config->workers);
  67
  68         for(i=0; i<config->workers; i++)
  69         {
  70                 mk_sched_launch_thread(config->worker_capacity);
  71         }
  72 }
  73
  74 void mk_server_loop(int server_fd)
  75 {
  76         int remote_fd;
  77         struct sockaddr_in sockaddr;
  78         struct sched_list_node *sched = sched_list;
  79         socklen_t socket_size = sizeof(struct sockaddr_in);
  80
  81         while(1){
  82                 remote_fd = accept(server_fd, (struct sockaddr *)&sockaddr,
  83                                    &socket_size);
  84
  85                 if(remote_fd == -1){
  86                         continue;
  87                 }
  88
  89                 /* Assign socket to worker thread */
  90                 mk_sched_add_client(sched, remote_fd);
  91
  92                 if(sched->next){
  93                         sched = sched->next;
  94                 }
  95                 else{
  96                         sched = sched_list;
  97                 }
  98        }
  99 }