/* MAIN */ int main(int argc, char **argv) { int opt, run_daemon = 0; char *file_config = NULL; static const struct option long_opts[] = { { "configdir", required_argument, NULL, 'c' }, { "daemon", no_argument, NULL, 'D' }, { "version", no_argument, NULL, 'v' }, { "help", no_argument, NULL, 'h' }, { NULL, 0, NULL, 0 } }; while ((opt = getopt_long(argc, argv, "DSvhc:", long_opts, NULL)) != -1) { switch (opt) { case 'v': mk_version(); exit(EXIT_SUCCESS); case 'h': mk_help(EXIT_SUCCESS); case 'D': run_daemon = 1; break; case 'c': file_config = optarg; break; case '?': printf("Monkey: Invalid option or option needs an argument.\n"); mk_help(EXIT_FAILURE); } } /* setup basic configurations */ config = mk_mem_malloc_z(sizeof(struct server_config)); if (!file_config) config->file_config = MONKEY_PATH_CONF; else config->file_config = file_config; if (run_daemon) config->is_daemon = MK_TRUE; else config->is_daemon = MK_FALSE; #ifdef TRACE monkey_init_time = time(NULL); MK_TRACE("Monkey TRACE is enabled"); env_trace_filter = getenv("MK_TRACE_FILTER"); pthread_mutex_init(&mutex_trace, (pthread_mutexattr_t *) NULL); #endif mk_version(); mk_signal_init(); mk_config_start_configure(); mk_sched_init(); mk_plugin_init(); /* Server listening socket */ config->server_fd = mk_socket_server(config->serverport, config->listen_addr); /* Running Monkey as daemon */ if (config->is_daemon == MK_TRUE) { mk_utils_set_daemon(); } /* Register PID of Monkey */ mk_utils_register_pid(); /* Workers: logger and clock */ mk_utils_worker_spawn((void *) mk_clock_worker_init); /* Init mk pointers */ mk_mem_pointers_init(); /* Init thread keys */ mk_thread_keys_init(); /* Change process owner */ mk_user_set_uidgid(); /* Configuration sanity check */ mk_config_sanity_check(); /* Print server details */ mk_details(); /* Invoke Plugin PRCTX hooks */ mk_plugin_core_process(); /* Launch monkey http workers */ mk_server_launch_workers(); /* Server loop, let's listen for incomming clients */ mk_server_loop(config->server_fd); mk_mem_free(config); return 0; }
/* MAIN */ int main(int argc, char **argv) { int opt; int port_override = -1; int workers_override = -1; int run_daemon = 0; char *path_config = NULL; char *server_config = NULL; static const struct option long_opts[] = { { "configdir", required_argument, NULL, 'c' }, { "serverconf",required_argument, NULL, 's' }, { "build", no_argument, NULL, 'b' }, { "daemon", no_argument, NULL, 'D' }, { "port", required_argument, NULL, 'p' }, { "workers", required_argument, NULL, 'w' }, { "version", no_argument, NULL, 'v' }, { "help", no_argument, NULL, 'h' }, { NULL, 0, NULL, 0 } }; while ((opt = getopt_long(argc, argv, "bDSvhp:w:c:s:", long_opts, NULL)) != -1) { switch (opt) { case 'b': mk_build_info(); exit(EXIT_SUCCESS); case 'v': mk_version(); exit(EXIT_SUCCESS); case 'h': mk_help(EXIT_SUCCESS); case 'D': run_daemon = 1; break; case 'p': port_override = atoi(optarg); break; case 'w': workers_override = atoi(optarg); break; case 'c': path_config = optarg; break; case 's': server_config = optarg; break; case '?': mk_help(EXIT_FAILURE); } } /* setup basic configurations */ config = mk_mem_malloc_z(sizeof(struct server_config)); /* set configuration path */ if (!path_config) { config->path_config = MONKEY_PATH_CONF; } else { config->path_config = path_config; } /* set target configuration file for the server */ if (!server_config) { config->server_config = M_DEFAULT_CONFIG_FILE; } else { config->server_config = server_config; } if (run_daemon) config->is_daemon = MK_TRUE; else config->is_daemon = MK_FALSE; #ifdef TRACE monkey_init_time = time(NULL); MK_TRACE("Monkey TRACE is enabled"); env_trace_filter = getenv("MK_TRACE_FILTER"); pthread_mutex_init(&mutex_trace, (pthread_mutexattr_t *) NULL); #endif mk_version(); mk_signal_init(); #ifdef LINUX_TRACE mk_info("Linux Trace enabled"); #endif /* Override number of thread workers */ if (workers_override >= 0) { config->workers = workers_override; } else { config->workers = -1; } /* Core and Scheduler setup */ mk_config_start_configure(); mk_sched_init(); /* Clock init that must happen before starting threads */ mk_clock_sequential_init(); /* Load plugins */ mk_plugin_init(); mk_plugin_read_config(); /* Override TCP port if it was set in the command line */ if (port_override > 0) { config->serverport = port_override; } /* Server listening socket */ config->server_fd = mk_socket_server(config->serverport, config->listen_addr); /* Running Monkey as daemon */ if (config->is_daemon == MK_TRUE) { mk_utils_set_daemon(); } /* Register PID of Monkey */ mk_utils_register_pid(); /* Workers: logger and clock */ mk_utils_worker_spawn((void *) mk_clock_worker_init, NULL); /* Init mk pointers */ mk_mem_pointers_init(); /* Init thread keys */ mk_thread_keys_init(); /* Change process owner */ mk_user_set_uidgid(); /* Configuration sanity check */ mk_config_sanity_check(); /* Print server details */ mk_details(); /* Invoke Plugin PRCTX hooks */ mk_plugin_core_process(); /* Launch monkey http workers */ mk_server_launch_workers(); /* Wait until all workers report as ready */ while (1) { int i, ready = 0; pthread_mutex_lock(&mutex_worker_init); for (i = 0; i < config->workers; i++) { if (sched_list[i].initialized) ready++; } pthread_mutex_unlock(&mutex_worker_init); if (ready == config->workers) break; usleep(10000); } /* Server loop, let's listen for incomming clients */ mk_server_loop(config->server_fd); mk_mem_free(config); return 0; }