src/monkey.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/types.h>
  24
  25 #include <stdlib.h>
  26 #include <unistd.h>
  27 #include <signal.h>
  28 #include <sys/wait.h>
  29 #include <resolv.h>
  30
  31 #include "monkey.h"
  32 #include "socket.h"
  33 #include <sys/time.h>
  34
  35 #include <string.h>
  36
  37 #include "epoll.h"
  38 #include "scheduler.h"
  39 #include "user.h"
  40 #include "info.h"
  41 #include "utils.h"
  42 #include "logfile.h"
  43 #include "signals.h"
  44 #include "config.h"
  45 #include "modules.h"
  46 #include "memory.h"
  47 #include "clock.h"
  48 #include "cache.h"
  49 #include "worker.h"
  50 #include "server.h"
  51 #include "plugin.h"
  52
  53 #if defined(__DATE__) && defined(__TIME__)
  54         static const char MONKEY_BUILT[] = __DATE__ " " __TIME__;
  55 #else
  56         static const char MONKEY_BUILT[] = "Unknown";
  57 #endif
  58
  59 void mk_details()
  60 {
  61         printf("* Process ID is %i", getpid());
  62         printf("\n* Server socket listening on Port %i", config->serverport);
  63         printf("\n* %i threads, %i client connections per thread, total %i\n",
  64                config->workers, config->worker_capacity,
  65                config->workers*config->worker_capacity);
  66         fflush(stdout);
  67 }
  68
  69 void mk_version()
  70 {
  71         printf("Monkey HTTP Daemon %s\n",VERSION);
  72         printf("Built : %s\n", MONKEY_BUILT);
  73         printf("Home  : http://www.monkey-project.com\n");
  74         fflush(stdout);
  75 }
  76
  77 void mk_help()
  78 {
  79         printf("Usage : monkey [-c directory] [-D] [-v] [-h]\n\n");
  80         printf("Available options:\n");
  81         printf("  -b\t\trun Monkey in benchmark mode, limits are disabled\n");
  82         printf("  -c directory\tspecify directory from configuration files\n");
  83         printf("  -D\t\trun Monkey as daemon\n");
  84         printf("  -v\t\tshow version number\n");
  85         printf("  -h\t\tthis help\n\n");
  86         exit(0);
  87 }
  88
  89 /* MAIN */
  90 int main(int argc, char **argv)
  91 {
  92         int opt;
  93         int daemon = 0;
  94
  95         config = mk_mem_malloc(sizeof(struct server_config));
  96         config->file_config=0;
  97
  98         opterr = 0;
  99         while ((opt = getopt(argc, argv, "DSvhc:")) != -1)
 100         {
 101                 switch (opt) {
 102                         case 'v':
 103                                 mk_version() ;
 104                                 exit(0);
 105                                 break;
 106                         case 'h':
 107                                 mk_help();
 108                                 break;
 109                         case 'D':
 110                                 daemon = 1;
 111                                 break;
 112                         case 'c':
 113                                 if (strlen(optarg) != 0) {
 114                                         config->file_config=optarg;
 115                                         break;
 116                                 }
 117                         case '?':
 118                                 printf("Monkey: Invalid option or option needs an argument.\n");
 119                                 mk_help();
 120                                 break;
 121                 }
 122         }
 123
 124         if(!config->file_config){
 125                 config->file_config=MONKEY_PATH_CONF;
 126         }
 127
 128         mk_version();
 129         mk_signal_init();
 130         mk_config_start_configure();
 131         mk_plugin_init();
 132
 133         server_fd = mk_socket_server(config->serverport);
 134
 135         /* Workers: logger and clock */
 136         mk_worker_spawn((void *)mk_logger_worker_init);
 137         mk_worker_spawn((void *)mk_clock_worker_init);
 138
 139         /* Running Monkey as daemon */
 140         if(daemon)
 141         {
 142                 mk_utils_set_daemon();
 143         }
 144
 145         /* Register PID of Monkey */
 146         mk_logger_register_pid();
 147
 148
 149         mk_mem_pointers_init();
 150
 151         /* Create thread keys */
 152         pthread_key_create(&request_index, NULL);
 153         pthread_key_create(&epoll_fd, NULL);
 154         pthread_key_create(&timer, NULL);
 155         pthread_key_create(&mk_cache_iov_log, NULL);
 156         pthread_key_create(&mk_cache_iov_header, NULL);
 157         pthread_key_create(&mk_cache_header_toc, NULL);
 158
 159         /* Change process owner */
 160         mk_user_set_uidgid();
 161
 162         mk_config_sanity_check();
 163
 164         /* Launch monkey http workers */
 165         mk_server_launch_workers();
 166
 167         /* Print server details */
 168         mk_details();
 169
 170         /* Plugins Stage 10 */
 171         mk_plugin_stage_run(MK_PLUGIN_STAGE_10, NULL, NULL);
 172
 173         /* Server loop, let's listen for incomming clients */
 174         mk_server_loop(server_fd);
 175
 176         return 0;
 177 }
 178