| |
| #include <ngx_config.h> |
| #include <ngx_core.h> |
| #include <ngx_event.h> |
| #include <nginx.h> |
| |
| |
| static ngx_int_t ngx_add_inherited_sockets(ngx_cycle_t *cycle); |
| static ngx_int_t ngx_getopt(ngx_master_ctx_t *ctx, ngx_cycle_t *cycle); |
| static void *ngx_core_module_create_conf(ngx_cycle_t *cycle); |
| static char *ngx_core_module_init_conf(ngx_cycle_t *cycle, void *conf); |
| static char *ngx_set_user(ngx_conf_t *cf, ngx_command_t *cmd, void *conf); |
| |
| |
| static ngx_command_t ngx_core_commands[] = { |
| |
| { ngx_string("daemon"), |
| NGX_MAIN_CONF|NGX_DIRECT_CONF|NGX_CONF_TAKE1, |
| ngx_conf_set_flag_slot, |
| 0, |
| offsetof(ngx_core_conf_t, daemon), |
| NULL }, |
| |
| { ngx_string("master_process"), |
| NGX_MAIN_CONF|NGX_DIRECT_CONF|NGX_CONF_TAKE1, |
| ngx_conf_set_flag_slot, |
| 0, |
| offsetof(ngx_core_conf_t, master), |
| NULL }, |
| |
| { ngx_string("worker_processes"), |
| NGX_MAIN_CONF|NGX_DIRECT_CONF|NGX_CONF_TAKE1, |
| ngx_conf_set_num_slot, |
| 0, |
| offsetof(ngx_core_conf_t, worker_processes), |
| NULL }, |
| |
| { ngx_string("user"), |
| NGX_MAIN_CONF|NGX_DIRECT_CONF|NGX_CONF_TAKE12, |
| ngx_set_user, |
| 0, |
| 0, |
| NULL }, |
| |
| { ngx_string("pid"), |
| NGX_MAIN_CONF|NGX_DIRECT_CONF|NGX_CONF_TAKE1, |
| ngx_conf_set_str_slot, |
| 0, |
| offsetof(ngx_core_conf_t, pid), |
| NULL }, |
| |
| ngx_null_command |
| }; |
| |
| |
| static ngx_core_module_t ngx_core_module_ctx = { |
| ngx_string("core"), |
| ngx_core_module_create_conf, |
| ngx_core_module_init_conf |
| }; |
| |
| |
| ngx_module_t ngx_core_module = { |
| NGX_MODULE, |
| &ngx_core_module_ctx, /* module context */ |
| ngx_core_commands, /* module directives */ |
| NGX_CORE_MODULE, /* module type */ |
| NULL, /* init module */ |
| NULL /* init child */ |
| }; |
| |
| |
| ngx_int_t ngx_max_module; |
| |
| ngx_int_t ngx_process; |
| ngx_pid_t ngx_pid; |
| ngx_pid_t ngx_new_binary; |
| ngx_int_t ngx_inherited; |
| |
| |
| int main(int argc, char *const *argv) |
| { |
| ngx_int_t i; |
| ngx_log_t *log; |
| ngx_cycle_t *cycle, init_cycle; |
| ngx_core_conf_t *ccf; |
| ngx_master_ctx_t ctx; |
| #if !(WIN32) |
| size_t len; |
| u_char pid[/* STUB */ 10]; |
| #endif |
| |
| #if __FreeBSD__ |
| ngx_debug_init(); |
| #endif |
| |
| /* TODO */ ngx_max_sockets = -1; |
| |
| ngx_time_init(); |
| |
| #if (HAVE_PCRE) |
| ngx_regex_init(); |
| #endif |
| |
| log = ngx_log_init_errlog(); |
| ngx_pid = ngx_getpid(); |
| |
| /* init_cycle->log is required for signal handlers and ngx_getopt() */ |
| |
| ngx_memzero(&init_cycle, sizeof(ngx_cycle_t)); |
| init_cycle.log = log; |
| ngx_cycle = &init_cycle; |
| |
| #if 0 |
| /* STUB */ log->log_level = NGX_LOG_DEBUG_ALL; |
| #endif |
| |
| ngx_memzero(&ctx, sizeof(ngx_master_ctx_t)); |
| ctx.argc = argc; |
| ctx.argv = argv; |
| |
| #if (NGX_THREADS) |
| if (ngx_time_mutex_init(log) == NGX_ERROR) { |
| return 1; |
| } |
| #endif |
| |
| if (ngx_getopt(&ctx, &init_cycle) == NGX_ERROR) { |
| return 1; |
| } |
| |
| if (ngx_os_init(log) == NGX_ERROR) { |
| return 1; |
| } |
| |
| ngx_max_module = 0; |
| for (i = 0; ngx_modules[i]; i++) { |
| ngx_modules[i]->index = ngx_max_module++; |
| } |
| |
| if (!(init_cycle.pool = ngx_create_pool(1024, log))) { |
| return 1; |
| } |
| |
| if (ngx_add_inherited_sockets(&init_cycle) == NGX_ERROR) { |
| return 1; |
| } |
| |
| cycle = ngx_init_cycle(&init_cycle); |
| if (cycle == NULL) { |
| return 1; |
| } |
| |
| ngx_cycle = cycle; |
| |
| ccf = (ngx_core_conf_t *) ngx_get_conf(cycle->conf_ctx, ngx_core_module); |
| |
| ngx_process = ccf->master ? NGX_PROCESS_MASTER : NGX_PROCESS_SINGLE; |
| |
| #if (WIN32) |
| |
| #if 0 |
| if (ccf->run_as_service) { |
| if (ngx_service(cycle->log) == NGX_ERROR) { |
| return 1; |
| } |
| |
| return 0; |
| } |
| #endif |
| |
| #else |
| |
| if (!ngx_inherited && ccf->daemon) { |
| if (ngx_daemon(cycle->log) == NGX_ERROR) { |
| return 1; |
| } |
| } |
| |
| if (ngx_create_pidfile(cycle, NULL) == NGX_ERROR) { |
| return 1; |
| } |
| |
| #endif |
| |
| ngx_master_process_cycle(cycle, &ctx); |
| |
| return 0; |
| } |
| |
| |
| static ngx_int_t ngx_add_inherited_sockets(ngx_cycle_t *cycle) |
| { |
| u_char *p, *v, *inherited; |
| ngx_socket_t s; |
| ngx_listening_t *ls; |
| |
| inherited = (u_char *) getenv(NGINX_VAR); |
| |
| if (inherited == NULL) { |
| return NGX_OK; |
| } |
| |
| ngx_log_error(NGX_LOG_INFO, cycle->log, 0, |
| "using inherited sockets from \"%s\"", inherited); |
| |
| ngx_init_array(cycle->listening, cycle->pool, |
| 10, sizeof(ngx_listening_t), NGX_ERROR); |
| |
| for (p = inherited, v = p; *p; p++) { |
| if (*p == ':' || *p == ';') { |
| s = ngx_atoi(v, p - v); |
| if (s == NGX_ERROR) { |
| ngx_log_error(NGX_LOG_EMERG, cycle->log, 0, |
| "invalid socket number \"%s\" in " |
| NGINX_VAR " enviroment variable, " |
| "ignoring the rest of the variable", v); |
| break; |
| } |
| |
| v = p + 1; |
| |
| if (!(ls = ngx_push_array(&cycle->listening))) { |
| return NGX_ERROR; |
| } |
| |
| ls->fd = s; |
| } |
| } |
| |
| ngx_inherited = 1; |
| |
| return ngx_set_inherited_sockets(cycle); |
| } |
| |
| |
| ngx_pid_t ngx_exec_new_binary(ngx_cycle_t *cycle, char *const *argv) |
| { |
| char *env[2], *var, *p; |
| ngx_uint_t i; |
| ngx_pid_t pid; |
| ngx_exec_ctx_t ctx; |
| ngx_listening_t *ls; |
| |
| ctx.path = argv[0]; |
| ctx.name = "new binary process"; |
| ctx.argv = argv; |
| |
| var = ngx_alloc(sizeof(NGINX_VAR) |
| + cycle->listening.nelts * (NGX_INT32_LEN + 1) + 2, |
| cycle->log); |
| |
| p = (char *) ngx_cpymem(var, NGINX_VAR "=", sizeof(NGINX_VAR)); |
| |
| ls = cycle->listening.elts; |
| for (i = 0; i < cycle->listening.nelts; i++) { |
| p += ngx_snprintf(p, NGX_INT32_LEN + 2, "%u;", ls[i].fd); |
| } |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_CORE, cycle->log, 0, "inherited: %s", var); |
| |
| env[0] = var; |
| env[1] = NULL; |
| ctx.envp = (char *const *) &env; |
| |
| pid = ngx_execute(cycle, &ctx); |
| |
| ngx_free(var); |
| |
| return pid; |
| } |
| |
| |
| static ngx_int_t ngx_getopt(ngx_master_ctx_t *ctx, ngx_cycle_t *cycle) |
| { |
| ngx_int_t i; |
| |
| for (i = 1; i < ctx->argc; i++) { |
| if (ctx->argv[i][0] != '-') { |
| ngx_log_error(NGX_LOG_EMERG, cycle->log, 0, |
| "invalid option: \"%s\"", ctx->argv[i]); |
| return NGX_ERROR; |
| } |
| |
| switch (ctx->argv[i][1]) { |
| |
| case 'c': |
| cycle->conf_file.data = (u_char *) ctx->argv[++i]; |
| cycle->conf_file.len = ngx_strlen(cycle->conf_file.data); |
| break; |
| |
| default: |
| ngx_log_error(NGX_LOG_EMERG, cycle->log, 0, |
| "invalid option: \"%s\"", ctx->argv[i]); |
| return NGX_ERROR; |
| } |
| } |
| |
| if (cycle->conf_file.data == NULL) { |
| cycle->conf_file.len = sizeof(NGINX_CONF) - 1; |
| cycle->conf_file.data = NGINX_CONF; |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static void *ngx_core_module_create_conf(ngx_cycle_t *cycle) |
| { |
| ngx_core_conf_t *ccf; |
| |
| if (!(ccf = ngx_pcalloc(cycle->pool, sizeof(ngx_core_conf_t)))) { |
| return NULL; |
| } |
| /* set by pcalloc() |
| * |
| * ccf->pid = NULL; |
| * ccf->newpid = NULL; |
| */ |
| ccf->daemon = NGX_CONF_UNSET; |
| ccf->master = NGX_CONF_UNSET; |
| ccf->worker_processes = NGX_CONF_UNSET; |
| ccf->user = (ngx_uid_t) NGX_CONF_UNSET; |
| ccf->group = (ngx_gid_t) NGX_CONF_UNSET; |
| |
| return ccf; |
| } |
| |
| |
| static char *ngx_core_module_init_conf(ngx_cycle_t *cycle, void *conf) |
| { |
| ngx_core_conf_t *ccf = conf; |
| |
| ngx_conf_init_value(ccf->daemon, 1); |
| ngx_conf_init_value(ccf->master, 1); |
| ngx_conf_init_value(ccf->worker_processes, 1); |
| |
| #if !(WIN32) |
| |
| /* TODO: default "nobody" user */ |
| |
| if (ccf->pid.len == 0) { |
| ccf->pid.len = sizeof(NGINX_PID) - 1; |
| ccf->pid.data = NGINX_PID; |
| ccf->newpid.len = sizeof(NGINX_NEWPID) - 1; |
| ccf->newpid.data = NGINX_NEWPID; |
| |
| } else { |
| ccf->newpid.len = ccf->pid.len + sizeof(NGINX_NEWPID_EXT); |
| |
| if (!(ccf->newpid.data = ngx_palloc(cycle->pool, ccf->newpid.len))) { |
| return NGX_CONF_ERROR; |
| } |
| |
| ngx_memcpy(ngx_cpymem(ccf->newpid.data, ccf->pid.data, ccf->pid.len), |
| NGINX_NEWPID_EXT, sizeof(NGINX_NEWPID_EXT)); |
| } |
| #endif |
| |
| return NGX_CONF_OK; |
| } |
| |
| |
| static char *ngx_set_user(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) |
| { |
| #if (WIN32) |
| |
| ngx_conf_log_error(NGX_LOG_WARN, cf, 0, |
| "\"user\" is not supported, ignored"); |
| |
| return NGX_CONF_OK; |
| |
| #else |
| |
| ngx_core_conf_t *ccf = conf; |
| |
| struct passwd *pwd; |
| struct group *grp; |
| ngx_str_t *value; |
| |
| if (ccf->user != (uid_t) NGX_CONF_UNSET) { |
| return "is duplicate"; |
| } |
| |
| value = (ngx_str_t *) cf->args->elts; |
| |
| pwd = getpwnam((const char *) value[1].data); |
| if (pwd == NULL) { |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, ngx_errno, |
| "getpwnam(%s) failed", value[1].data); |
| return NGX_CONF_ERROR; |
| } |
| |
| ccf->user = pwd->pw_uid; |
| |
| if (cf->args->nelts == 2) { |
| return NGX_CONF_OK; |
| } |
| |
| grp = getgrnam((const char *) value[2].data); |
| if (grp == NULL) { |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, ngx_errno, |
| "getgrnam(%s) failed", value[1].data); |
| return NGX_CONF_ERROR; |
| } |
| |
| ccf->group = grp->gr_gid; |
| |
| return NGX_CONF_OK; |
| |
| #endif |
| } |
