nginx/src/core/nginx.c

567 lines
14 KiB
C
Raw Normal View History

#include <ngx_config.h>
2003-01-10 14:09:20 +08:00
#include <ngx_core.h>
2003-06-11 23:28:34 +08:00
#include <ngx_event.h>
2003-05-30 22:27:59 +08:00
#include <nginx.h>
2003-07-02 22:41:17 +08:00
static ngx_cycle_t *ngx_init_cycle(ngx_cycle_t *old_cycle, ngx_log_t *log);
static int ngx_open_listening_sockets(ngx_cycle_t *cycle, ngx_log_t *log);
2003-07-07 14:11:50 +08:00
static void ngx_clean_old_cycles(ngx_event_t *ev);
2003-07-07 14:11:50 +08:00
#if (NGX_DEBUG) && (__FreeBSD__)
extern char *malloc_options;
#endif
2002-08-20 22:48:28 +08:00
2003-07-07 14:11:50 +08:00
int ngx_max_module;
ngx_os_io_t ngx_io;
2003-07-07 14:11:50 +08:00
ngx_cycle_t *ngx_cycle;
ngx_pool_t *ngx_temp_pool;
ngx_array_t ngx_old_cycles;
ngx_event_t ngx_cleaner_event;
2003-07-03 02:51:41 +08:00
2003-07-07 14:11:50 +08:00
/* STUB NAME */
ngx_connection_t dumb;
2003-04-08 23:40:10 +08:00
2002-12-15 14:25:09 +08:00
int ngx_connection_counter;
2002-08-16 23:27:03 +08:00
2003-07-04 00:30:22 +08:00
int restart;
int rotate;
2003-07-01 23:00:03 +08:00
int main(int argc, char *const *argv)
{
2003-07-02 22:41:17 +08:00
int i;
2003-07-02 13:01:53 +08:00
ngx_log_t *log;
2003-07-04 23:10:33 +08:00
ngx_cycle_t *cycle;
2002-12-23 14:29:22 +08:00
2003-07-07 14:11:50 +08:00
#if (NGX_DEBUG) && (__FreeBSD__)
malloc_options = "J";
#endif
2003-07-02 13:01:53 +08:00
/* TODO */ ngx_max_sockets = -1;
2003-05-20 23:37:55 +08:00
2003-06-03 23:42:58 +08:00
log = ngx_log_init_errlog();
2002-08-20 22:48:28 +08:00
2003-06-03 23:42:58 +08:00
if (ngx_os_init(log) == NGX_ERROR) {
2003-05-20 00:39:14 +08:00
return 1;
2003-05-15 01:13:13 +08:00
}
2003-04-08 23:40:10 +08:00
ngx_max_module = 0;
for (i = 0; ngx_modules[i]; i++) {
ngx_modules[i]->index = ngx_max_module++;
}
2003-07-02 13:01:53 +08:00
cycle = ngx_init_cycle(NULL, log);
if (cycle == NULL) {
2003-06-26 23:35:36 +08:00
return 1;
}
2003-07-07 14:11:50 +08:00
ngx_cycle = cycle;
2003-07-04 23:10:33 +08:00
2003-06-26 23:35:36 +08:00
/* daemon */
2003-05-20 00:39:14 +08:00
/* life cycle */
2003-05-16 23:27:48 +08:00
2003-07-01 23:00:03 +08:00
for ( ;; ) {
2003-07-07 14:11:50 +08:00
/* STUB */ ngx_cycle->log->log_level = NGX_LOG_DEBUG;
2003-07-03 02:51:41 +08:00
2003-06-26 23:35:36 +08:00
/* forks */
2003-07-01 23:00:03 +08:00
ngx_init_temp_number();
2003-07-04 23:10:33 +08:00
for (i = 0; ngx_modules[i]; i++) {
if (ngx_modules[i]->init_child) {
2003-07-07 14:11:50 +08:00
if (ngx_modules[i]->init_child(ngx_cycle) == NGX_ERROR) {
2003-07-04 23:10:33 +08:00
/* fatal */
exit(1);
}
}
}
2003-07-03 02:51:41 +08:00
2003-06-26 23:35:36 +08:00
/* threads */
2003-07-04 00:30:22 +08:00
restart = 0;
rotate = 0;
2003-06-26 23:35:36 +08:00
for ( ;; ) {
2003-07-04 00:30:22 +08:00
for ( ;; ) {
2003-07-07 14:11:50 +08:00
ngx_log_debug(ngx_cycle->log, "worker cycle");
2003-07-04 00:30:22 +08:00
2003-07-07 14:11:50 +08:00
ngx_process_events(ngx_cycle->log);
2003-07-04 00:30:22 +08:00
if (rotate) {
2003-07-07 14:11:50 +08:00
ngx_log_debug(ngx_cycle->log, "rotate");
2003-07-04 00:30:22 +08:00
}
if (restart) {
2003-07-07 14:11:50 +08:00
ngx_log_debug(ngx_cycle->log, "restart");
2003-07-04 00:30:22 +08:00
break;
}
}
2003-07-01 23:00:03 +08:00
2003-07-07 14:11:50 +08:00
cycle = ngx_init_cycle(ngx_cycle, ngx_cycle->log);
2003-07-04 23:10:33 +08:00
if (cycle == NULL) {
2003-06-26 23:35:36 +08:00
continue;
}
2003-07-07 14:11:50 +08:00
ngx_cycle = cycle;
2003-06-26 23:35:36 +08:00
break;
}
}
return 0;
2003-07-01 23:00:03 +08:00
}
2003-07-02 22:41:17 +08:00
static ngx_cycle_t *ngx_init_cycle(ngx_cycle_t *old_cycle, ngx_log_t *log)
2003-07-01 23:00:03 +08:00
{
2003-07-02 22:41:17 +08:00
int i, n, failed;
ngx_str_t conf_file;
2003-07-02 13:01:53 +08:00
ngx_conf_t conf;
ngx_pool_t *pool;
2003-07-07 14:11:50 +08:00
ngx_cycle_t *cycle, **old;
2003-07-02 13:01:53 +08:00
ngx_open_file_t *file;
ngx_listening_t *ls, *nls;
2003-07-01 23:00:03 +08:00
pool = ngx_create_pool(16 * 1024, log);
if (pool == NULL) {
return NULL;
}
cycle = ngx_pcalloc(pool, sizeof(ngx_cycle_t));
if (cycle == NULL) {
ngx_destroy_pool(pool);
return NULL;
}
cycle->pool = pool;
2003-07-07 14:11:50 +08:00
cycle->old_cycle = old_cycle;
2003-07-01 23:00:03 +08:00
n = old_cycle ? old_cycle->open_files.nelts : 20;
cycle->open_files.elts = ngx_pcalloc(pool, n * sizeof(ngx_open_file_t));
if (cycle->open_files.elts == NULL) {
ngx_destroy_pool(pool);
return NULL;
}
cycle->open_files.nelts = 0;
cycle->open_files.size = sizeof(ngx_open_file_t);
cycle->open_files.nalloc = n;
cycle->open_files.pool = pool;
2003-07-03 02:51:41 +08:00
cycle->log = ngx_log_create_errlog(cycle);
if (cycle->log == NULL) {
ngx_destroy_pool(pool);
return NULL;
}
2003-07-01 23:00:03 +08:00
n = old_cycle ? old_cycle->listening.nelts : 10;
cycle->listening.elts = ngx_pcalloc(pool, n * sizeof(ngx_listening_t));
if (cycle->listening.elts == NULL) {
ngx_destroy_pool(pool);
return NULL;
}
cycle->listening.nelts = 0;
cycle->listening.size = sizeof(ngx_listening_t);
cycle->listening.nalloc = n;
cycle->listening.pool = pool;
cycle->conf_ctx = ngx_pcalloc(pool, ngx_max_module * sizeof(void *));
if (cycle->conf_ctx == NULL) {
ngx_destroy_pool(pool);
return NULL;
}
ngx_memzero(&conf, sizeof(ngx_conf_t));
/* STUB: init array ? */
conf.args = ngx_create_array(pool, 10, sizeof(ngx_str_t));
if (conf.args == NULL) {
ngx_destroy_pool(pool);
return NULL;
}
conf.ctx = cycle->conf_ctx;
conf.cycle = cycle;
2003-07-07 14:11:50 +08:00
/* STUB */ conf.pool = cycle->pool;
2003-07-01 23:00:03 +08:00
conf.log = log;
conf.module_type = NGX_CORE_MODULE;
conf.cmd_type = NGX_MAIN_CONF;
conf_file.len = sizeof(NGINX_CONF) - 1;
conf_file.data = NGINX_CONF;
if (ngx_conf_parse(&conf, &conf_file) != NGX_CONF_OK) {
ngx_destroy_pool(pool);
return NULL;
}
2003-07-02 22:41:17 +08:00
failed = 0;
2003-07-04 23:10:33 +08:00
file = cycle->open_files.elts;
for (i = 0; i < cycle->open_files.nelts; i++) {
if (file->name.data == NULL) {
continue;
}
2003-07-01 23:00:03 +08:00
2003-07-04 23:10:33 +08:00
file->fd = ngx_open_file(file->name.data,
NGX_FILE_RDWR,
NGX_FILE_CREATE_OR_OPEN|NGX_FILE_APPEND);
2003-07-01 23:00:03 +08:00
2003-07-07 14:11:50 +08:00
ngx_log_debug(log, "OPEN: %d:%s" _ file->fd _ file->name.data);
2003-07-04 23:10:33 +08:00
if (file->fd == NGX_INVALID_FILE) {
ngx_log_error(NGX_LOG_EMERG, log, ngx_errno,
ngx_open_file_n " \"%s\" failed",
file->name.data);
failed = 1;
break;
2003-07-01 23:00:03 +08:00
}
2003-07-04 23:10:33 +08:00
/* TODO: Win32 append */
2003-07-01 23:00:03 +08:00
}
2003-07-07 14:11:50 +08:00
/* STUB */ cycle->log->log_level = NGX_LOG_DEBUG;
2003-07-04 23:10:33 +08:00
2003-07-01 23:00:03 +08:00
if (!failed) {
2003-07-03 02:51:41 +08:00
if (old_cycle) {
ls = old_cycle->listening.elts;
2003-07-02 13:01:53 +08:00
for (i = 0; i < old_cycle->listening.nelts; i++) {
2003-07-03 02:51:41 +08:00
ls[i].remain = 0;
2003-07-02 13:01:53 +08:00
}
2003-07-03 02:51:41 +08:00
nls = cycle->listening.elts;
for (n = 0; n < cycle->listening.nelts; n++) {
for (i = 0; i < old_cycle->listening.nelts; i++) {
if (ngx_memcmp(nls[n].sockaddr,
ls[i].sockaddr, ls[i].socklen) == 0)
{
nls[n].fd = ls[i].fd;
2003-07-07 14:11:50 +08:00
nls[i].remain = 1;
2003-07-03 02:51:41 +08:00
ls[i].remain = 1;
break;
}
}
if (nls[n].fd == -1) {
nls[n].new = 1;
}
2003-07-02 13:01:53 +08:00
}
2003-07-07 14:11:50 +08:00
} else {
ls = cycle->listening.elts;
for (i = 0; i < cycle->listening.nelts; i++) {
ls[i].new = 1;
}
2003-07-02 13:01:53 +08:00
}
2003-07-02 22:41:17 +08:00
if (ngx_open_listening_sockets(cycle, log) == NGX_ERROR) {
2003-07-01 23:00:03 +08:00
failed = 1;
}
}
if (failed) {
/* rollback the new cycle configuration */
file = cycle->open_files.elts;
for (i = 0; i < cycle->open_files.nelts; i++) {
2003-07-02 13:01:53 +08:00
if (file->fd == NGX_INVALID_FILE) {
continue;
}
2003-07-02 22:41:17 +08:00
if (ngx_close_file(file->fd) == NGX_FILE_ERROR) {
2003-07-02 13:01:53 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_errno,
ngx_close_file_n " \"%s\" failed",
file->name.data);
2003-07-01 23:00:03 +08:00
}
}
2003-07-02 22:41:17 +08:00
ls = cycle->listening.elts;
2003-07-01 23:00:03 +08:00
for (i = 0; i < cycle->listening.nelts; i++) {
2003-07-02 13:01:53 +08:00
if (ls[i].new && ls[i].fd == -1) {
continue;
}
2003-07-02 22:41:17 +08:00
if (ngx_close_socket(ls[i].fd) == -1) {
2003-07-02 13:01:53 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno,
ngx_close_socket_n " %s failed",
ls[i].addr_text.data);
2003-07-01 23:00:03 +08:00
}
}
ngx_destroy_pool(pool);
return NULL;
2003-07-02 22:41:17 +08:00
}
2003-07-01 23:00:03 +08:00
2003-07-02 22:41:17 +08:00
/* commit the new cycle configuration */
2003-07-01 23:00:03 +08:00
2003-07-04 23:10:33 +08:00
pool->log = cycle->log;
2003-07-07 14:11:50 +08:00
#if 1
/* STUB */ cycle->one_process = 1;
#endif
2003-07-02 22:41:17 +08:00
for (i = 0; ngx_modules[i]; i++) {
2003-07-04 23:10:33 +08:00
if (ngx_modules[i]->init_module) {
if (ngx_modules[i]->init_module(cycle) == NGX_ERROR) {
/* fatal */
exit(1);
}
2003-07-01 23:00:03 +08:00
}
}
2003-07-03 02:51:41 +08:00
if (old_cycle == NULL) {
return cycle;
}
2003-07-02 13:01:53 +08:00
ls = old_cycle->listening.elts;
for (i = 0; i < old_cycle->listening.nelts; i++) {
if (ls[i].remain) {
continue;
2003-07-01 23:00:03 +08:00
}
2003-07-02 22:41:17 +08:00
if (ngx_close_socket(ls[i].fd) == -1) {
2003-07-02 13:01:53 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno,
ngx_close_socket_n " %s failed",
ls[i].addr_text.data);
}
2003-07-01 23:00:03 +08:00
}
2003-07-02 13:01:53 +08:00
file = old_cycle->open_files.elts;
2003-07-02 22:41:17 +08:00
for (i = 0; i < old_cycle->open_files.nelts; i++) {
2003-07-02 13:01:53 +08:00
if (file->fd == NGX_INVALID_FILE) {
continue;
2003-07-01 23:00:03 +08:00
}
2003-07-02 22:41:17 +08:00
if (ngx_close_file(file->fd) == NGX_FILE_ERROR) {
2003-07-02 13:01:53 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_errno,
ngx_close_file_n " \"%s\" failed",
file->name.data);
2003-07-01 23:00:03 +08:00
}
}
2003-06-26 23:35:36 +08:00
2003-07-07 14:11:50 +08:00
if (!old_cycle->one_process) {
ngx_destroy_pool(old_cycle->pool);
return cycle;
2003-07-01 23:00:03 +08:00
}
2003-07-07 14:11:50 +08:00
if (ngx_temp_pool == NULL) {
ngx_temp_pool = ngx_create_pool(128, cycle->log);
if (ngx_temp_pool == NULL) {
ngx_log_error(NGX_LOG_EMERG, cycle->log, 0,
"can not create ngx_temp_pool");
exit(1);
2003-01-10 14:09:20 +08:00
}
2003-07-07 14:11:50 +08:00
n = 10;
ngx_old_cycles.elts = ngx_pcalloc(ngx_temp_pool,
n * sizeof(ngx_cycle_t *));
if (ngx_old_cycles.elts == NULL) {
exit(1);
2003-05-30 22:27:59 +08:00
}
2003-07-07 14:11:50 +08:00
ngx_old_cycles.nelts = 0;
ngx_old_cycles.size = sizeof(ngx_cycle_t *);
ngx_old_cycles.nalloc = n;
ngx_old_cycles.pool = ngx_temp_pool;
ngx_cleaner_event.event_handler = ngx_clean_old_cycles;
ngx_cleaner_event.log = cycle->log;
ngx_cleaner_event.data = &dumb;
dumb.fd = -1;
}
2003-07-07 14:11:50 +08:00
ngx_temp_pool->log = cycle->log;
2003-07-07 14:11:50 +08:00
old = ngx_push_array(&ngx_old_cycles);
if (old == NULL) {
exit(1);
}
*old = old_cycle;
2003-07-07 14:11:50 +08:00
if (!ngx_cleaner_event.timer_set) {
ngx_add_timer(&ngx_cleaner_event, 30000);
ngx_cleaner_event.timer_set = 1;
2003-05-27 20:18:54 +08:00
}
2002-09-02 22:48:24 +08:00
2003-07-07 14:11:50 +08:00
return cycle;
2002-08-20 22:48:28 +08:00
}
2002-12-11 02:05:12 +08:00
2003-07-02 22:41:17 +08:00
static int ngx_open_listening_sockets(ngx_cycle_t *cycle, ngx_log_t *log)
2002-08-20 22:48:28 +08:00
{
2003-06-11 23:28:34 +08:00
int times, failed, reuseaddr, i;
ngx_err_t err;
ngx_socket_t s;
ngx_listening_t *ls;
2002-08-20 22:48:28 +08:00
reuseaddr = 1;
2003-07-02 22:41:17 +08:00
/* TODO: times configurable */
2002-08-20 22:48:28 +08:00
for (times = 10; times; times--) {
failed = 0;
/* for each listening socket */
2003-06-11 23:28:34 +08:00
2003-07-02 22:41:17 +08:00
ls = cycle->listening.elts;
for (i = 0; i < cycle->listening.nelts; i++) {
2002-08-26 23:18:19 +08:00
2003-07-02 22:41:17 +08:00
if (ls[i].fd != -1) {
2002-08-20 22:48:28 +08:00
continue;
2003-07-02 22:41:17 +08:00
}
2002-08-20 22:48:28 +08:00
2002-08-26 23:18:19 +08:00
if (ls[i].inherited) {
/* TODO: close on exit */
/* TODO: nonblocking */
/* TODO: deferred accept */
continue;
}
s = ngx_socket(ls[i].family, ls[i].type, ls[i].protocol,
ls[i].flags);
2003-07-02 22:41:17 +08:00
2002-08-26 23:18:19 +08:00
if (s == -1) {
2002-08-20 22:48:28 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno,
2002-12-19 15:08:55 +08:00
ngx_socket_n " %s falied", ls[i].addr_text.data);
2003-05-30 22:27:59 +08:00
return NGX_ERROR;
2002-08-26 23:18:19 +08:00
}
2002-08-20 22:48:28 +08:00
2003-06-11 23:28:34 +08:00
#if (WIN32)
/*
* Winsock assignes a socket number divisible by 4
* so to find a connection we divide a socket number by 4.
*/
if (s % 4) {
ngx_log_error(NGX_LOG_EMERG, ls->log, 0,
ngx_socket_n " created socket %d", s);
return NGX_ERROR;
}
#endif
2002-08-20 22:48:28 +08:00
if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
2002-08-26 23:18:19 +08:00
(const void *) &reuseaddr, sizeof(int)) == -1) {
2002-08-20 22:48:28 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno,
2002-08-26 23:18:19 +08:00
"setsockopt(SO_REUSEADDR) %s failed",
2002-12-19 15:08:55 +08:00
ls[i].addr_text.data);
2003-05-30 22:27:59 +08:00
return NGX_ERROR;
2002-08-26 23:18:19 +08:00
}
2002-08-20 22:48:28 +08:00
/* TODO: close on exit */
if (ls[i].nonblocking) {
2002-08-26 23:18:19 +08:00
if (ngx_nonblocking(s) == -1) {
2002-08-20 22:48:28 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno,
ngx_nonblocking_n " %s failed",
2002-12-19 15:08:55 +08:00
ls[i].addr_text.data);
2003-05-30 22:27:59 +08:00
return NGX_ERROR;
2002-08-26 23:18:19 +08:00
}
2002-08-20 22:48:28 +08:00
}
2002-08-26 23:18:19 +08:00
if (bind(s, ls[i].sockaddr, ls[i].socklen) == -1) {
2002-08-20 22:48:28 +08:00
err = ngx_socket_errno;
2002-08-26 23:18:19 +08:00
ngx_log_error(NGX_LOG_EMERG, log, err,
2002-12-19 15:08:55 +08:00
"bind() to %s failed", ls[i].addr_text.data);
2002-08-20 22:48:28 +08:00
if (err != NGX_EADDRINUSE)
2003-05-30 22:27:59 +08:00
return NGX_ERROR;
2002-08-20 22:48:28 +08:00
if (ngx_close_socket(s) == -1)
2002-08-26 23:18:19 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno,
2002-08-20 22:48:28 +08:00
ngx_close_socket_n " %s failed",
2002-12-19 15:08:55 +08:00
ls[i].addr_text.data);
2002-08-20 22:48:28 +08:00
failed = 1;
continue;
}
2002-08-26 23:18:19 +08:00
if (listen(s, ls[i].backlog) == -1) {
2002-08-20 22:48:28 +08:00
ngx_log_error(NGX_LOG_EMERG, log, ngx_socket_errno,
2002-12-19 15:08:55 +08:00
"listen() to %s failed", ls[i].addr_text.data);
2003-05-30 22:27:59 +08:00
return NGX_ERROR;
2002-08-26 23:18:19 +08:00
}
2002-08-20 22:48:28 +08:00
/* TODO: deferred accept */
ls[i].fd = s;
}
if (!failed)
break;
2003-07-02 22:41:17 +08:00
/* TODO: delay configurable */
2002-08-26 23:18:19 +08:00
ngx_log_error(NGX_LOG_NOTICE, log, 0,
"try again to bind() after 500ms");
2002-08-20 22:48:28 +08:00
ngx_msleep(500);
}
2002-08-26 23:18:19 +08:00
if (failed) {
2003-07-02 22:41:17 +08:00
ngx_log_error(NGX_LOG_EMERG, log, 0, "still can not bind()");
2003-05-30 22:27:59 +08:00
return NGX_ERROR;
2002-08-26 23:18:19 +08:00
}
2003-05-30 22:27:59 +08:00
return NGX_OK;
}
2003-07-07 14:11:50 +08:00
static void ngx_clean_old_cycles(ngx_event_t *ev)
{
int i, n, found, live;
ngx_cycle_t **cycle;
ngx_temp_pool->log = ngx_cycle->log;
ngx_log_debug(ngx_cycle->log, "clean old cycles");
live = 0;
cycle = ngx_old_cycles.elts;
for (i = 0; i < ngx_old_cycles.nelts; i++) {
if (cycle[i] == NULL) {
continue;
}
found = 0;
for (n = 0; n < cycle[i]->connection_n; n++) {
if (cycle[i]->connections[n].fd != -1) {
found = 1;
ngx_log_debug(ngx_cycle->log, "live fd: %d" _ n);
break;
}
}
if (found) {
live = 1;
continue;
}
ngx_log_debug(ngx_cycle->log, "clean old cycle: %d" _ i);
ngx_destroy_pool(cycle[i]->pool);
cycle[i] = NULL;
}
ngx_log_debug(ngx_cycle->log, "old cycles status: %d" _ live);
if (live) {
ngx_log_debug(ngx_cycle->log, "TIMER");
ngx_add_timer(ev, 30000);
} else {
ngx_cleaner_event.timer_set = 0;
ngx_destroy_pool(ngx_temp_pool);
ngx_temp_pool = NULL;
ngx_old_cycles.nelts = 0;
}
}