| /* | |
| * Copyright (C) Igor Sysoev | |
| * Copyright (C) Nginx, Inc. | |
| */ | |
| #include <ngx_config.h> | |
| #include <ngx_core.h> | |
| #include <ngx_stream.h> | |
| #if (NGX_ZLIB) | |
| #include <zlib.h> | |
| #endif | |
| typedef struct ngx_stream_log_op_s ngx_stream_log_op_t; | |
| typedef u_char *(*ngx_stream_log_op_run_pt) (ngx_stream_session_t *s, | |
| u_char *buf, ngx_stream_log_op_t *op); | |
| typedef size_t (*ngx_stream_log_op_getlen_pt) (ngx_stream_session_t *s, | |
| uintptr_t data); | |
| struct ngx_stream_log_op_s { | |
| size_t len; | |
| ngx_stream_log_op_getlen_pt getlen; | |
| ngx_stream_log_op_run_pt run; | |
| uintptr_t data; | |
| }; | |
| typedef struct { | |
| ngx_str_t name; | |
| ngx_array_t *flushes; | |
| ngx_array_t *ops; /* array of ngx_stream_log_op_t */ | |
| } ngx_stream_log_fmt_t; | |
| typedef struct { | |
| ngx_array_t formats; /* array of ngx_stream_log_fmt_t */ | |
| } ngx_stream_log_main_conf_t; | |
| typedef struct { | |
| u_char *start; | |
| u_char *pos; | |
| u_char *last; | |
| ngx_event_t *event; | |
| ngx_msec_t flush; | |
| ngx_int_t gzip; | |
| } ngx_stream_log_buf_t; | |
| typedef struct { | |
| ngx_array_t *lengths; | |
| ngx_array_t *values; | |
| } ngx_stream_log_script_t; | |
| typedef struct { | |
| ngx_open_file_t *file; | |
| ngx_stream_log_script_t *script; | |
| time_t disk_full_time; | |
| time_t error_log_time; | |
| ngx_syslog_peer_t *syslog_peer; | |
| ngx_stream_log_fmt_t *format; | |
| ngx_stream_complex_value_t *filter; | |
| } ngx_stream_log_t; | |
| typedef struct { | |
| ngx_array_t *logs; /* array of ngx_stream_log_t */ | |
| ngx_open_file_cache_t *open_file_cache; | |
| time_t open_file_cache_valid; | |
| ngx_uint_t open_file_cache_min_uses; | |
| ngx_uint_t off; /* unsigned off:1 */ | |
| } ngx_stream_log_srv_conf_t; | |
| typedef struct { | |
| ngx_str_t name; | |
| size_t len; | |
| ngx_stream_log_op_run_pt run; | |
| } ngx_stream_log_var_t; | |
| static void ngx_stream_log_write(ngx_stream_session_t *s, ngx_stream_log_t *log, | |
| u_char *buf, size_t len); | |
| static ssize_t ngx_stream_log_script_write(ngx_stream_session_t *s, | |
| ngx_stream_log_script_t *script, u_char **name, u_char *buf, size_t len); | |
| #if (NGX_ZLIB) | |
| static ssize_t ngx_stream_log_gzip(ngx_fd_t fd, u_char *buf, size_t len, | |
| ngx_int_t level, ngx_log_t *log); | |
| static void *ngx_stream_log_gzip_alloc(void *opaque, u_int items, u_int size); | |
| static void ngx_stream_log_gzip_free(void *opaque, void *address); | |
| #endif | |
| static void ngx_stream_log_flush(ngx_open_file_t *file, ngx_log_t *log); | |
| static void ngx_stream_log_flush_handler(ngx_event_t *ev); | |
| static ngx_int_t ngx_stream_log_variable_compile(ngx_conf_t *cf, | |
| ngx_stream_log_op_t *op, ngx_str_t *value, ngx_uint_t json); | |
| static size_t ngx_stream_log_variable_getlen(ngx_stream_session_t *s, | |
| uintptr_t data); | |
| static u_char *ngx_stream_log_variable(ngx_stream_session_t *s, u_char *buf, | |
| ngx_stream_log_op_t *op); | |
| static uintptr_t ngx_stream_log_escape(u_char *dst, u_char *src, size_t size); | |
| static size_t ngx_stream_log_json_variable_getlen(ngx_stream_session_t *s, | |
| uintptr_t data); | |
| static u_char *ngx_stream_log_json_variable(ngx_stream_session_t *s, | |
| u_char *buf, ngx_stream_log_op_t *op); | |
| static void *ngx_stream_log_create_main_conf(ngx_conf_t *cf); | |
| static void *ngx_stream_log_create_srv_conf(ngx_conf_t *cf); | |
| static char *ngx_stream_log_merge_srv_conf(ngx_conf_t *cf, void *parent, | |
| void *child); | |
| static char *ngx_stream_log_set_log(ngx_conf_t *cf, ngx_command_t *cmd, | |
| void *conf); | |
| static char *ngx_stream_log_set_format(ngx_conf_t *cf, ngx_command_t *cmd, | |
| void *conf); | |
| static char *ngx_stream_log_compile_format(ngx_conf_t *cf, | |
| ngx_array_t *flushes, ngx_array_t *ops, ngx_array_t *args, ngx_uint_t s); | |
| static char *ngx_stream_log_open_file_cache(ngx_conf_t *cf, ngx_command_t *cmd, | |
| void *conf); | |
| static ngx_int_t ngx_stream_log_init(ngx_conf_t *cf); | |
| static ngx_command_t ngx_stream_log_commands[] = { | |
| { ngx_string("log_format"), | |
| NGX_STREAM_MAIN_CONF|NGX_CONF_2MORE, | |
| ngx_stream_log_set_format, | |
| NGX_STREAM_MAIN_CONF_OFFSET, | |
| 0, | |
| NULL }, | |
| { ngx_string("access_log"), | |
| NGX_STREAM_MAIN_CONF|NGX_STREAM_SRV_CONF|NGX_CONF_1MORE, | |
| ngx_stream_log_set_log, | |
| NGX_STREAM_SRV_CONF_OFFSET, | |
| 0, | |
| NULL }, | |
| { ngx_string("open_log_file_cache"), | |
| NGX_STREAM_MAIN_CONF|NGX_STREAM_SRV_CONF|NGX_CONF_TAKE1234, | |
| ngx_stream_log_open_file_cache, | |
| NGX_STREAM_SRV_CONF_OFFSET, | |
| 0, | |
| NULL }, | |
| ngx_null_command | |
| }; | |
| static ngx_stream_module_t ngx_stream_log_module_ctx = { | |
| NULL, /* preconfiguration */ | |
| ngx_stream_log_init, /* postconfiguration */ | |
| ngx_stream_log_create_main_conf, /* create main configuration */ | |
| NULL, /* init main configuration */ | |
| ngx_stream_log_create_srv_conf, /* create server configuration */ | |
| ngx_stream_log_merge_srv_conf /* merge server configuration */ | |
| }; | |
| ngx_module_t ngx_stream_log_module = { | |
| NGX_MODULE_V1, | |
| &ngx_stream_log_module_ctx, /* module context */ | |
| ngx_stream_log_commands, /* module directives */ | |
| NGX_STREAM_MODULE, /* module type */ | |
| NULL, /* init master */ | |
| NULL, /* init module */ | |
| NULL, /* init process */ | |
| NULL, /* init thread */ | |
| NULL, /* exit thread */ | |
| NULL, /* exit process */ | |
| NULL, /* exit master */ | |
| NGX_MODULE_V1_PADDING | |
| }; | |
| static ngx_int_t | |
| ngx_stream_log_handler(ngx_stream_session_t *s) | |
| { | |
| u_char *line, *p; | |
| size_t len, size; | |
| ssize_t n; | |
| ngx_str_t val; | |
| ngx_uint_t i, l; | |
| ngx_stream_log_t *log; | |
| ngx_stream_log_op_t *op; | |
| ngx_stream_log_buf_t *buffer; | |
| ngx_stream_log_srv_conf_t *lscf; | |
| ngx_log_debug0(NGX_LOG_DEBUG_STREAM, s->connection->log, 0, | |
| "stream log handler"); | |
| lscf = ngx_stream_get_module_srv_conf(s, ngx_stream_log_module); | |
| if (lscf->off || lscf->logs == NULL) { | |
| return NGX_OK; | |
| } | |
| log = lscf->logs->elts; | |
| for (l = 0; l < lscf->logs->nelts; l++) { | |
| if (log[l].filter) { | |
| if (ngx_stream_complex_value(s, log[l].filter, &val) != NGX_OK) { | |
| return NGX_ERROR; | |
| } | |
| if (val.len == 0 || (val.len == 1 && val.data[0] == '0')) { | |
| continue; | |
| } | |
| } | |
| if (ngx_time() == log[l].disk_full_time) { | |
| /* | |
| * on FreeBSD writing to a full filesystem with enabled softupdates | |
| * may block process for much longer time than writing to non-full | |
| * filesystem, so we skip writing to a log for one second | |
| */ | |
| continue; | |
| } | |
| ngx_stream_script_flush_no_cacheable_variables(s, | |
| log[l].format->flushes); | |
| len = 0; | |
| op = log[l].format->ops->elts; | |
| for (i = 0; i < log[l].format->ops->nelts; i++) { | |
| if (op[i].len == 0) { | |
| len += op[i].getlen(s, op[i].data); | |
| } else { | |
| len += op[i].len; | |
| } | |
| } | |
| if (log[l].syslog_peer) { | |
| /* length of syslog's PRI and HEADER message parts */ | |
| len += sizeof("<255>Jan 01 00:00:00 ") - 1 | |
| + ngx_cycle->hostname.len + 1 | |
| + log[l].syslog_peer->tag.len + 2; | |
| goto alloc_line; | |
| } | |
| len += NGX_LINEFEED_SIZE; | |
| buffer = log[l].file ? log[l].file->data : NULL; | |
| if (buffer) { | |
| if (len > (size_t) (buffer->last - buffer->pos)) { | |
| ngx_stream_log_write(s, &log[l], buffer->start, | |
| buffer->pos - buffer->start); | |
| buffer->pos = buffer->start; | |
| } | |
| if (len <= (size_t) (buffer->last - buffer->pos)) { | |
| p = buffer->pos; | |
| if (buffer->event && p == buffer->start) { | |
| ngx_add_timer(buffer->event, buffer->flush); | |
| } | |
| for (i = 0; i < log[l].format->ops->nelts; i++) { | |
| p = op[i].run(s, p, &op[i]); | |
| } | |
| ngx_linefeed(p); | |
| buffer->pos = p; | |
| continue; | |
| } | |
| if (buffer->event && buffer->event->timer_set) { | |
| ngx_del_timer(buffer->event); | |
| } | |
| } | |
| alloc_line: | |
| line = ngx_pnalloc(s->connection->pool, len); | |
| if (line == NULL) { | |
| return NGX_ERROR; | |
| } | |
| p = line; | |
| if (log[l].syslog_peer) { | |
| p = ngx_syslog_add_header(log[l].syslog_peer, line); | |
| } | |
| for (i = 0; i < log[l].format->ops->nelts; i++) { | |
| p = op[i].run(s, p, &op[i]); | |
| } | |
| if (log[l].syslog_peer) { | |
| size = p - line; | |
| n = ngx_syslog_send(log[l].syslog_peer, line, size); | |
| if (n < 0) { | |
| ngx_log_error(NGX_LOG_WARN, s->connection->log, 0, | |
| "send() to syslog failed"); | |
| } else if ((size_t) n != size) { | |
| ngx_log_error(NGX_LOG_WARN, s->connection->log, 0, | |
| "send() to syslog has written only %z of %uz", | |
| n, size); | |
| } | |
| continue; | |
| } | |
| ngx_linefeed(p); | |
| ngx_stream_log_write(s, &log[l], line, p - line); | |
| } | |
| return NGX_OK; | |
| } | |
| static void | |
| ngx_stream_log_write(ngx_stream_session_t *s, ngx_stream_log_t *log, | |
| u_char *buf, size_t len) | |
| { | |
| u_char *name; | |
| time_t now; | |
| ssize_t n; | |
| ngx_err_t err; | |
| #if (NGX_ZLIB) | |
| ngx_stream_log_buf_t *buffer; | |
| #endif | |
| if (log->script == NULL) { | |
| name = log->file->name.data; | |
| #if (NGX_ZLIB) | |
| buffer = log->file->data; | |
| if (buffer && buffer->gzip) { | |
| n = ngx_stream_log_gzip(log->file->fd, buf, len, buffer->gzip, | |
| s->connection->log); | |
| } else { | |
| n = ngx_write_fd(log->file->fd, buf, len); | |
| } | |
| #else | |
| n = ngx_write_fd(log->file->fd, buf, len); | |
| #endif | |
| } else { | |
| name = NULL; | |
| n = ngx_stream_log_script_write(s, log->script, &name, buf, len); | |
| } | |
| if (n == (ssize_t) len) { | |
| return; | |
| } | |
| now = ngx_time(); | |
| if (n == -1) { | |
| err = ngx_errno; | |
| if (err == NGX_ENOSPC) { | |
| log->disk_full_time = now; | |
| } | |
| if (now - log->error_log_time > 59) { | |
| ngx_log_error(NGX_LOG_ALERT, s->connection->log, err, | |
| ngx_write_fd_n " to \"%s\" failed", name); | |
| log->error_log_time = now; | |
| } | |
| return; | |
| } | |
| if (now - log->error_log_time > 59) { | |
| ngx_log_error(NGX_LOG_ALERT, s->connection->log, 0, | |
| ngx_write_fd_n " to \"%s\" was incomplete: %z of %uz", | |
| name, n, len); | |
| log->error_log_time = now; | |
| } | |
| } | |
| static ssize_t | |
| ngx_stream_log_script_write(ngx_stream_session_t *s, | |
| ngx_stream_log_script_t *script, u_char **name, u_char *buf, size_t len) | |
| { | |
| ssize_t n; | |
| ngx_str_t log; | |
| ngx_open_file_info_t of; | |
| ngx_stream_log_srv_conf_t *lscf; | |
| if (ngx_stream_script_run(s, &log, script->lengths->elts, 1, | |
| script->values->elts) | |
| == NULL) | |
| { | |
| /* simulate successful logging */ | |
| return len; | |
| } | |
| log.data[log.len - 1] = '\0'; | |
| *name = log.data; | |
| ngx_log_debug1(NGX_LOG_DEBUG_STREAM, s->connection->log, 0, | |
| "stream log \"%s\"", log.data); | |
| lscf = ngx_stream_get_module_srv_conf(s, ngx_stream_log_module); | |
| ngx_memzero(&of, sizeof(ngx_open_file_info_t)); | |
| of.log = 1; | |
| of.valid = lscf->open_file_cache_valid; | |
| of.min_uses = lscf->open_file_cache_min_uses; | |
| of.directio = NGX_OPEN_FILE_DIRECTIO_OFF; | |
| if (ngx_open_cached_file(lscf->open_file_cache, &log, &of, | |
| s->connection->pool) | |
| != NGX_OK) | |
| { | |
| if (of.err == 0) { | |
| /* simulate successful logging */ | |
| return len; | |
| } | |
| ngx_log_error(NGX_LOG_CRIT, s->connection->log, ngx_errno, | |
| "%s \"%s\" failed", of.failed, log.data); | |
| /* simulate successful logging */ | |
| return len; | |
| } | |
| ngx_log_debug1(NGX_LOG_DEBUG_STREAM, s->connection->log, 0, | |
| "stream log #%d", of.fd); | |
| n = ngx_write_fd(of.fd, buf, len); | |
| return n; | |
| } | |
| #if (NGX_ZLIB) | |
| static ssize_t | |
| ngx_stream_log_gzip(ngx_fd_t fd, u_char *buf, size_t len, ngx_int_t level, | |
| ngx_log_t *log) | |
| { | |
| int rc, wbits, memlevel; | |
| u_char *out; | |
| size_t size; | |
| ssize_t n; | |
| z_stream zstream; | |
| ngx_err_t err; | |
| ngx_pool_t *pool; | |
| wbits = MAX_WBITS; | |
| memlevel = MAX_MEM_LEVEL - 1; | |
| while ((ssize_t) len < ((1 << (wbits - 1)) - 262)) { | |
| wbits--; | |
| memlevel--; | |
| } | |
| /* | |
| * This is a formula from deflateBound() for conservative upper bound of | |
| * compressed data plus 18 bytes of gzip wrapper. | |
| */ | |
| size = len + ((len + 7) >> 3) + ((len + 63) >> 6) + 5 + 18; | |
| ngx_memzero(&zstream, sizeof(z_stream)); | |
| pool = ngx_create_pool(256, log); | |
| if (pool == NULL) { | |
| /* simulate successful logging */ | |
| return len; | |
| } | |
| pool->log = log; | |
| zstream.zalloc = ngx_stream_log_gzip_alloc; | |
| zstream.zfree = ngx_stream_log_gzip_free; | |
| zstream.opaque = pool; | |
| out = ngx_pnalloc(pool, size); | |
| if (out == NULL) { | |
| goto done; | |
| } | |
| zstream.next_in = buf; | |
| zstream.avail_in = len; | |
| zstream.next_out = out; | |
| zstream.avail_out = size; | |
| rc = deflateInit2(&zstream, (int) level, Z_DEFLATED, wbits + 16, memlevel, | |
| Z_DEFAULT_STRATEGY); | |
| if (rc != Z_OK) { | |
| ngx_log_error(NGX_LOG_ALERT, log, 0, "deflateInit2() failed: %d", rc); | |
| goto done; | |
| } | |
| ngx_log_debug4(NGX_LOG_DEBUG_STREAM, log, 0, | |
| "deflate in: ni:%p no:%p ai:%ud ao:%ud", | |
| zstream.next_in, zstream.next_out, | |
| zstream.avail_in, zstream.avail_out); | |
| rc = deflate(&zstream, Z_FINISH); | |
| if (rc != Z_STREAM_END) { | |
| ngx_log_error(NGX_LOG_ALERT, log, 0, | |
| "deflate(Z_FINISH) failed: %d", rc); | |
| goto done; | |
| } | |
| ngx_log_debug5(NGX_LOG_DEBUG_STREAM, log, 0, | |
| "deflate out: ni:%p no:%p ai:%ud ao:%ud rc:%d", | |
| zstream.next_in, zstream.next_out, | |
| zstream.avail_in, zstream.avail_out, | |
| rc); | |
| size -= zstream.avail_out; | |
| rc = deflateEnd(&zstream); | |
| if (rc != Z_OK) { | |
| ngx_log_error(NGX_LOG_ALERT, log, 0, "deflateEnd() failed: %d", rc); | |
| goto done; | |
| } | |
| n = ngx_write_fd(fd, out, size); | |
| if (n != (ssize_t) size) { | |
| err = (n == -1) ? ngx_errno : 0; | |
| ngx_destroy_pool(pool); | |
| ngx_set_errno(err); | |
| return -1; | |
| } | |
| done: | |
| ngx_destroy_pool(pool); | |
| /* simulate successful logging */ | |
| return len; | |
| } | |
| static void * | |
| ngx_stream_log_gzip_alloc(void *opaque, u_int items, u_int size) | |
| { | |
| ngx_pool_t *pool = opaque; | |
| ngx_log_debug2(NGX_LOG_DEBUG_STREAM, pool->log, 0, | |
| "gzip alloc: n:%ud s:%ud", items, size); | |
| return ngx_palloc(pool, items * size); | |
| } | |
| static void | |
| ngx_stream_log_gzip_free(void *opaque, void *address) | |
| { | |
| #if 0 | |
| ngx_pool_t *pool = opaque; | |
| ngx_log_debug1(NGX_LOG_DEBUG_STREAM, pool->log, 0, | |
| "gzip free: %p", address); | |
| #endif | |
| } | |
| #endif | |
| static void | |
| ngx_stream_log_flush(ngx_open_file_t *file, ngx_log_t *log) | |
| { | |
| size_t len; | |
| ssize_t n; | |
| ngx_stream_log_buf_t *buffer; | |
| buffer = file->data; | |
| len = buffer->pos - buffer->start; | |
| if (len == 0) { | |
| return; | |
| } | |
| #if (NGX_ZLIB) | |
| if (buffer->gzip) { | |
| n = ngx_stream_log_gzip(file->fd, buffer->start, len, buffer->gzip, | |
| log); | |
| } else { | |
| n = ngx_write_fd(file->fd, buffer->start, len); | |
| } | |
| #else | |
| n = ngx_write_fd(file->fd, buffer->start, len); | |
| #endif | |
| if (n == -1) { | |
| ngx_log_error(NGX_LOG_ALERT, log, ngx_errno, | |
| ngx_write_fd_n " to \"%s\" failed", | |
| file->name.data); | |
| } else if ((size_t) n != len) { | |
| ngx_log_error(NGX_LOG_ALERT, log, 0, | |
| ngx_write_fd_n " to \"%s\" was incomplete: %z of %uz", | |
| file->name.data, n, len); | |
| } | |
| buffer->pos = buffer->start; | |
| if (buffer->event && buffer->event->timer_set) { | |
| ngx_del_timer(buffer->event); | |
| } | |
| } | |
| static void | |
| ngx_stream_log_flush_handler(ngx_event_t *ev) | |
| { | |
| ngx_log_debug0(NGX_LOG_DEBUG_EVENT, ev->log, 0, | |
| "stream log buffer flush handler"); | |
| ngx_stream_log_flush(ev->data, ev->log); | |
| } | |
| static u_char * | |
| ngx_stream_log_copy_short(ngx_stream_session_t *s, u_char *buf, | |
| ngx_stream_log_op_t *op) | |
| { | |
| size_t len; | |
| uintptr_t data; | |
| len = op->len; | |
| data = op->data; | |
| while (len--) { | |
| *buf++ = (u_char) (data & 0xff); | |
| data >>= 8; | |
| } | |
| return buf; | |
| } | |
| static u_char * | |
| ngx_stream_log_copy_long(ngx_stream_session_t *s, u_char *buf, | |
| ngx_stream_log_op_t *op) | |
| { | |
| return ngx_cpymem(buf, (u_char *) op->data, op->len); | |
| } | |
| static ngx_int_t | |
| ngx_stream_log_variable_compile(ngx_conf_t *cf, ngx_stream_log_op_t *op, | |
| ngx_str_t *value, ngx_uint_t json) | |
| { | |
| ngx_int_t index; | |
| index = ngx_stream_get_variable_index(cf, value); | |
| if (index == NGX_ERROR) { | |
| return NGX_ERROR; | |
| } | |
| op->len = 0; | |
| if (json) { | |
| op->getlen = ngx_stream_log_json_variable_getlen; | |
| op->run = ngx_stream_log_json_variable; | |
| } else { | |
| op->getlen = ngx_stream_log_variable_getlen; | |
| op->run = ngx_stream_log_variable; | |
| } | |
| op->data = index; | |
| return NGX_OK; | |
| } | |
| static size_t | |
| ngx_stream_log_variable_getlen(ngx_stream_session_t *s, uintptr_t data) | |
| { | |
| uintptr_t len; | |
| ngx_stream_variable_value_t *value; | |
| value = ngx_stream_get_indexed_variable(s, data); | |
| if (value == NULL || value->not_found) { | |
| return 1; | |
| } | |
| len = ngx_stream_log_escape(NULL, value->data, value->len); | |
| value->escape = len ? 1 : 0; | |
| return value->len + len * 3; | |
| } | |
| static u_char * | |
| ngx_stream_log_variable(ngx_stream_session_t *s, u_char *buf, | |
| ngx_stream_log_op_t *op) | |
| { | |
| ngx_stream_variable_value_t *value; | |
| value = ngx_stream_get_indexed_variable(s, op->data); | |
| if (value == NULL || value->not_found) { | |
| *buf = '-'; | |
| return buf + 1; | |
| } | |
| if (value->escape == 0) { | |
| return ngx_cpymem(buf, value->data, value->len); | |
| } else { | |
| return (u_char *) ngx_stream_log_escape(buf, value->data, value->len); | |
| } | |
| } | |
| static uintptr_t | |
| ngx_stream_log_escape(u_char *dst, u_char *src, size_t size) | |
| { | |
| ngx_uint_t n; | |
| static u_char hex[] = "0123456789ABCDEF"; | |
| static uint32_t escape[] = { | |
| 0xffffffff, /* 1111 1111 1111 1111 1111 1111 1111 1111 */ | |
| /* ?>=< ;:98 7654 3210 /.-, +*)( '&%$ #"! */ | |
| 0x00000004, /* 0000 0000 0000 0000 0000 0000 0000 0100 */ | |
| /* _^]\ [ZYX WVUT SRQP ONML KJIH GFED CBA@ */ | |
| 0x10000000, /* 0001 0000 0000 0000 0000 0000 0000 0000 */ | |
| /* ~}| {zyx wvut srqp onml kjih gfed cba` */ | |
| 0x80000000, /* 1000 0000 0000 0000 0000 0000 0000 0000 */ | |
| 0xffffffff, /* 1111 1111 1111 1111 1111 1111 1111 1111 */ | |
| 0xffffffff, /* 1111 1111 1111 1111 1111 1111 1111 1111 */ | |
| 0xffffffff, /* 1111 1111 1111 1111 1111 1111 1111 1111 */ | |
| 0xffffffff, /* 1111 1111 1111 1111 1111 1111 1111 1111 */ | |
| }; | |
| if (dst == NULL) { | |
| /* find the number of the characters to be escaped */ | |
| n = 0; | |
| while (size) { | |
| if (escape[*src >> 5] & (1U << (*src & 0x1f))) { | |
| n++; | |
| } | |
| src++; | |
| size--; | |
| } | |
| return (uintptr_t) n; | |
| } | |
| while (size) { | |
| if (escape[*src >> 5] & (1U << (*src & 0x1f))) { | |
| *dst++ = '\\'; | |
| *dst++ = 'x'; | |
| *dst++ = hex[*src >> 4]; | |
| *dst++ = hex[*src & 0xf]; | |
| src++; | |
| } else { | |
| *dst++ = *src++; | |
| } | |
| size--; | |
| } | |
| return (uintptr_t) dst; | |
| } | |
| static size_t | |
| ngx_stream_log_json_variable_getlen(ngx_stream_session_t *s, uintptr_t data) | |
| { | |
| uintptr_t len; | |
| ngx_stream_variable_value_t *value; | |
| value = ngx_stream_get_indexed_variable(s, data); | |
| if (value == NULL || value->not_found) { | |
| return 0; | |
| } | |
| len = ngx_escape_json(NULL, value->data, value->len); | |
| value->escape = len ? 1 : 0; | |
| return value->len + len; | |
| } | |
| static u_char * | |
| ngx_stream_log_json_variable(ngx_stream_session_t *s, u_char *buf, | |
| ngx_stream_log_op_t *op) | |
| { | |
| ngx_stream_variable_value_t *value; | |
| value = ngx_stream_get_indexed_variable(s, op->data); | |
| if (value == NULL || value->not_found) { | |
| return buf; | |
| } | |
| if (value->escape == 0) { | |
| return ngx_cpymem(buf, value->data, value->len); | |
| } else { | |
| return (u_char *) ngx_escape_json(buf, value->data, value->len); | |
| } | |
| } | |
| static void * | |
| ngx_stream_log_create_main_conf(ngx_conf_t *cf) | |
| { | |
| ngx_stream_log_main_conf_t *conf; | |
| conf = ngx_pcalloc(cf->pool, sizeof(ngx_stream_log_main_conf_t)); | |
| if (conf == NULL) { | |
| return NULL; | |
| } | |
| if (ngx_array_init(&conf->formats, cf->pool, 4, | |
| sizeof(ngx_stream_log_fmt_t)) | |
| != NGX_OK) | |
| { | |
| return NULL; | |
| } | |
| return conf; | |
| } | |
| static void * | |
| ngx_stream_log_create_srv_conf(ngx_conf_t *cf) | |
| { | |
| ngx_stream_log_srv_conf_t *conf; | |
| conf = ngx_pcalloc(cf->pool, sizeof(ngx_stream_log_srv_conf_t)); | |
| if (conf == NULL) { | |
| return NULL; | |
| } | |
| conf->open_file_cache = NGX_CONF_UNSET_PTR; | |
| return conf; | |
| } | |
| static char * | |
| ngx_stream_log_merge_srv_conf(ngx_conf_t *cf, void *parent, void *child) | |
| { | |
| ngx_stream_log_srv_conf_t *prev = parent; | |
| ngx_stream_log_srv_conf_t *conf = child; | |
| if (conf->open_file_cache == NGX_CONF_UNSET_PTR) { | |
| conf->open_file_cache = prev->open_file_cache; | |
| conf->open_file_cache_valid = prev->open_file_cache_valid; | |
| conf->open_file_cache_min_uses = prev->open_file_cache_min_uses; | |
| if (conf->open_file_cache == NGX_CONF_UNSET_PTR) { | |
| conf->open_file_cache = NULL; | |
| } | |
| } | |
| if (conf->logs || conf->off) { | |
| return NGX_CONF_OK; | |
| } | |
| conf->logs = prev->logs; | |
| conf->off = prev->off; | |
| return NGX_CONF_OK; | |
| } | |
| static char * | |
| ngx_stream_log_set_log(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
| { | |
| ngx_stream_log_srv_conf_t *lscf = conf; | |
| ssize_t size; | |
| ngx_int_t gzip; | |
| ngx_uint_t i, n; | |
| ngx_msec_t flush; | |
| ngx_str_t *value, name, s; | |
| ngx_stream_log_t *log; | |
| ngx_syslog_peer_t *peer; | |
| ngx_stream_log_buf_t *buffer; | |
| ngx_stream_log_fmt_t *fmt; | |
| ngx_stream_script_compile_t sc; | |
| ngx_stream_log_main_conf_t *lmcf; | |
| ngx_stream_compile_complex_value_t ccv; | |
| value = cf->args->elts; | |
| if (ngx_strcmp(value[1].data, "off") == 0) { | |
| lscf->off = 1; | |
| if (cf->args->nelts == 2) { | |
| return NGX_CONF_OK; | |
| } | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "invalid parameter \"%V\"", &value[2]); | |
| return NGX_CONF_ERROR; | |
| } | |
| if (lscf->logs == NULL) { | |
| lscf->logs = ngx_array_create(cf->pool, 2, sizeof(ngx_stream_log_t)); | |
| if (lscf->logs == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| } | |
| lmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_log_module); | |
| log = ngx_array_push(lscf->logs); | |
| if (log == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| ngx_memzero(log, sizeof(ngx_stream_log_t)); | |
| if (ngx_strncmp(value[1].data, "syslog:", 7) == 0) { | |
| peer = ngx_pcalloc(cf->pool, sizeof(ngx_syslog_peer_t)); | |
| if (peer == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| if (ngx_syslog_process_conf(cf, peer) != NGX_CONF_OK) { | |
| return NGX_CONF_ERROR; | |
| } | |
| log->syslog_peer = peer; | |
| goto process_formats; | |
| } | |
| n = ngx_stream_script_variables_count(&value[1]); | |
| if (n == 0) { | |
| log->file = ngx_conf_open_file(cf->cycle, &value[1]); | |
| if (log->file == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| } else { | |
| if (ngx_conf_full_name(cf->cycle, &value[1], 0) != NGX_OK) { | |
| return NGX_CONF_ERROR; | |
| } | |
| log->script = ngx_pcalloc(cf->pool, sizeof(ngx_stream_log_script_t)); | |
| if (log->script == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| ngx_memzero(&sc, sizeof(ngx_stream_script_compile_t)); | |
| sc.cf = cf; | |
| sc.source = &value[1]; | |
| sc.lengths = &log->script->lengths; | |
| sc.values = &log->script->values; | |
| sc.variables = n; | |
| sc.complete_lengths = 1; | |
| sc.complete_values = 1; | |
| if (ngx_stream_script_compile(&sc) != NGX_OK) { | |
| return NGX_CONF_ERROR; | |
| } | |
| } | |
| process_formats: | |
| if (cf->args->nelts >= 3) { | |
| name = value[2]; | |
| } else { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "log format is not specified"); | |
| return NGX_CONF_ERROR; | |
| } | |
| fmt = lmcf->formats.elts; | |
| for (i = 0; i < lmcf->formats.nelts; i++) { | |
| if (fmt[i].name.len == name.len | |
| && ngx_strcasecmp(fmt[i].name.data, name.data) == 0) | |
| { | |
| log->format = &fmt[i]; | |
| break; | |
| } | |
| } | |
| if (log->format == NULL) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "unknown log format \"%V\"", &name); | |
| return NGX_CONF_ERROR; | |
| } | |
| size = 0; | |
| flush = 0; | |
| gzip = 0; | |
| for (i = 3; i < cf->args->nelts; i++) { | |
| if (ngx_strncmp(value[i].data, "buffer=", 7) == 0) { | |
| s.len = value[i].len - 7; | |
| s.data = value[i].data + 7; | |
| size = ngx_parse_size(&s); | |
| if (size == NGX_ERROR || size == 0) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "invalid buffer size \"%V\"", &s); | |
| return NGX_CONF_ERROR; | |
| } | |
| continue; | |
| } | |
| if (ngx_strncmp(value[i].data, "flush=", 6) == 0) { | |
| s.len = value[i].len - 6; | |
| s.data = value[i].data + 6; | |
| flush = ngx_parse_time(&s, 0); | |
| if (flush == (ngx_msec_t) NGX_ERROR || flush == 0) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "invalid flush time \"%V\"", &s); | |
| return NGX_CONF_ERROR; | |
| } | |
| continue; | |
| } | |
| if (ngx_strncmp(value[i].data, "gzip", 4) == 0 | |
| && (value[i].len == 4 || value[i].data[4] == '=')) | |
| { | |
| #if (NGX_ZLIB) | |
| if (size == 0) { | |
| size = 64 * 1024; | |
| } | |
| if (value[i].len == 4) { | |
| gzip = Z_BEST_SPEED; | |
| continue; | |
| } | |
| s.len = value[i].len - 5; | |
| s.data = value[i].data + 5; | |
| gzip = ngx_atoi(s.data, s.len); | |
| if (gzip < 1 || gzip > 9) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "invalid compression level \"%V\"", &s); | |
| return NGX_CONF_ERROR; | |
| } | |
| continue; | |
| #else | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "nginx was built without zlib support"); | |
| return NGX_CONF_ERROR; | |
| #endif | |
| } | |
| if (ngx_strncmp(value[i].data, "if=", 3) == 0) { | |
| s.len = value[i].len - 3; | |
| s.data = value[i].data + 3; | |
| ngx_memzero(&ccv, sizeof(ngx_stream_compile_complex_value_t)); | |
| ccv.cf = cf; | |
| ccv.value = &s; | |
| ccv.complex_value = ngx_palloc(cf->pool, | |
| sizeof(ngx_stream_complex_value_t)); | |
| if (ccv.complex_value == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| if (ngx_stream_compile_complex_value(&ccv) != NGX_OK) { | |
| return NGX_CONF_ERROR; | |
| } | |
| log->filter = ccv.complex_value; | |
| continue; | |
| } | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "invalid parameter \"%V\"", &value[i]); | |
| return NGX_CONF_ERROR; | |
| } | |
| if (flush && size == 0) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "no buffer is defined for access_log \"%V\"", | |
| &value[1]); | |
| return NGX_CONF_ERROR; | |
| } | |
| if (size) { | |
| if (log->script) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "buffered logs cannot have variables in name"); | |
| return NGX_CONF_ERROR; | |
| } | |
| if (log->syslog_peer) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "logs to syslog cannot be buffered"); | |
| return NGX_CONF_ERROR; | |
| } | |
| if (log->file->data) { | |
| buffer = log->file->data; | |
| if (buffer->last - buffer->start != size | |
| || buffer->flush != flush | |
| || buffer->gzip != gzip) | |
| { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "access_log \"%V\" already defined " | |
| "with conflicting parameters", | |
| &value[1]); | |
| return NGX_CONF_ERROR; | |
| } | |
| return NGX_CONF_OK; | |
| } | |
| buffer = ngx_pcalloc(cf->pool, sizeof(ngx_stream_log_buf_t)); | |
| if (buffer == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| buffer->start = ngx_pnalloc(cf->pool, size); | |
| if (buffer->start == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| buffer->pos = buffer->start; | |
| buffer->last = buffer->start + size; | |
| if (flush) { | |
| buffer->event = ngx_pcalloc(cf->pool, sizeof(ngx_event_t)); | |
| if (buffer->event == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| buffer->event->data = log->file; | |
| buffer->event->handler = ngx_stream_log_flush_handler; | |
| buffer->event->log = &cf->cycle->new_log; | |
| buffer->event->cancelable = 1; | |
| buffer->flush = flush; | |
| } | |
| buffer->gzip = gzip; | |
| log->file->flush = ngx_stream_log_flush; | |
| log->file->data = buffer; | |
| } | |
| return NGX_CONF_OK; | |
| } | |
| static char * | |
| ngx_stream_log_set_format(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
| { | |
| ngx_stream_log_main_conf_t *lmcf = conf; | |
| ngx_str_t *value; | |
| ngx_uint_t i; | |
| ngx_stream_log_fmt_t *fmt; | |
| value = cf->args->elts; | |
| fmt = lmcf->formats.elts; | |
| for (i = 0; i < lmcf->formats.nelts; i++) { | |
| if (fmt[i].name.len == value[1].len | |
| && ngx_strcmp(fmt[i].name.data, value[1].data) == 0) | |
| { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "duplicate \"log_format\" name \"%V\"", | |
| &value[1]); | |
| return NGX_CONF_ERROR; | |
| } | |
| } | |
| fmt = ngx_array_push(&lmcf->formats); | |
| if (fmt == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| fmt->name = value[1]; | |
| fmt->flushes = ngx_array_create(cf->pool, 4, sizeof(ngx_int_t)); | |
| if (fmt->flushes == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| fmt->ops = ngx_array_create(cf->pool, 16, sizeof(ngx_stream_log_op_t)); | |
| if (fmt->ops == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| return ngx_stream_log_compile_format(cf, fmt->flushes, fmt->ops, | |
| cf->args, 2); | |
| } | |
| static char * | |
| ngx_stream_log_compile_format(ngx_conf_t *cf, ngx_array_t *flushes, | |
| ngx_array_t *ops, ngx_array_t *args, ngx_uint_t s) | |
| { | |
| u_char *data, *p, ch; | |
| size_t i, len; | |
| ngx_str_t *value, var; | |
| ngx_int_t *flush; | |
| ngx_uint_t bracket, json; | |
| ngx_stream_log_op_t *op; | |
| json = 0; | |
| value = args->elts; | |
| if (s < args->nelts && ngx_strncmp(value[s].data, "escape=", 7) == 0) { | |
| data = value[s].data + 7; | |
| if (ngx_strcmp(data, "json") == 0) { | |
| json = 1; | |
| } else if (ngx_strcmp(data, "default") != 0) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "unknown log format escaping \"%s\"", data); | |
| return NGX_CONF_ERROR; | |
| } | |
| s++; | |
| } | |
| for ( /* void */ ; s < args->nelts; s++) { | |
| i = 0; | |
| while (i < value[s].len) { | |
| op = ngx_array_push(ops); | |
| if (op == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| data = &value[s].data[i]; | |
| if (value[s].data[i] == '$') { | |
| if (++i == value[s].len) { | |
| goto invalid; | |
| } | |
| if (value[s].data[i] == '{') { | |
| bracket = 1; | |
| if (++i == value[s].len) { | |
| goto invalid; | |
| } | |
| var.data = &value[s].data[i]; | |
| } else { | |
| bracket = 0; | |
| var.data = &value[s].data[i]; | |
| } | |
| for (var.len = 0; i < value[s].len; i++, var.len++) { | |
| ch = value[s].data[i]; | |
| if (ch == '}' && bracket) { | |
| i++; | |
| bracket = 0; | |
| break; | |
| } | |
| if ((ch >= 'A' && ch <= 'Z') | |
| || (ch >= 'a' && ch <= 'z') | |
| || (ch >= '0' && ch <= '9') | |
| || ch == '_') | |
| { | |
| continue; | |
| } | |
| break; | |
| } | |
| if (bracket) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "the closing bracket in \"%V\" " | |
| "variable is missing", &var); | |
| return NGX_CONF_ERROR; | |
| } | |
| if (var.len == 0) { | |
| goto invalid; | |
| } | |
| if (ngx_stream_log_variable_compile(cf, op, &var, json) | |
| != NGX_OK) | |
| { | |
| return NGX_CONF_ERROR; | |
| } | |
| if (flushes) { | |
| flush = ngx_array_push(flushes); | |
| if (flush == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| *flush = op->data; /* variable index */ | |
| } | |
| continue; | |
| } | |
| i++; | |
| while (i < value[s].len && value[s].data[i] != '$') { | |
| i++; | |
| } | |
| len = &value[s].data[i] - data; | |
| if (len) { | |
| op->len = len; | |
| op->getlen = NULL; | |
| if (len <= sizeof(uintptr_t)) { | |
| op->run = ngx_stream_log_copy_short; | |
| op->data = 0; | |
| while (len--) { | |
| op->data <<= 8; | |
| op->data |= data[len]; | |
| } | |
| } else { | |
| op->run = ngx_stream_log_copy_long; | |
| p = ngx_pnalloc(cf->pool, len); | |
| if (p == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| ngx_memcpy(p, data, len); | |
| op->data = (uintptr_t) p; | |
| } | |
| } | |
| } | |
| } | |
| return NGX_CONF_OK; | |
| invalid: | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "invalid parameter \"%s\"", data); | |
| return NGX_CONF_ERROR; | |
| } | |
| static char * | |
| ngx_stream_log_open_file_cache(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
| { | |
| ngx_stream_log_srv_conf_t *lscf = conf; | |
| time_t inactive, valid; | |
| ngx_str_t *value, s; | |
| ngx_int_t max, min_uses; | |
| ngx_uint_t i; | |
| if (lscf->open_file_cache != NGX_CONF_UNSET_PTR) { | |
| return "is duplicate"; | |
| } | |
| value = cf->args->elts; | |
| max = 0; | |
| inactive = 10; | |
| valid = 60; | |
| min_uses = 1; | |
| for (i = 1; i < cf->args->nelts; i++) { | |
| if (ngx_strncmp(value[i].data, "max=", 4) == 0) { | |
| max = ngx_atoi(value[i].data + 4, value[i].len - 4); | |
| if (max == NGX_ERROR) { | |
| goto failed; | |
| } | |
| continue; | |
| } | |
| if (ngx_strncmp(value[i].data, "inactive=", 9) == 0) { | |
| s.len = value[i].len - 9; | |
| s.data = value[i].data + 9; | |
| inactive = ngx_parse_time(&s, 1); | |
| if (inactive == (time_t) NGX_ERROR) { | |
| goto failed; | |
| } | |
| continue; | |
| } | |
| if (ngx_strncmp(value[i].data, "min_uses=", 9) == 0) { | |
| min_uses = ngx_atoi(value[i].data + 9, value[i].len - 9); | |
| if (min_uses == NGX_ERROR) { | |
| goto failed; | |
| } | |
| continue; | |
| } | |
| if (ngx_strncmp(value[i].data, "valid=", 6) == 0) { | |
| s.len = value[i].len - 6; | |
| s.data = value[i].data + 6; | |
| valid = ngx_parse_time(&s, 1); | |
| if (valid == (time_t) NGX_ERROR) { | |
| goto failed; | |
| } | |
| continue; | |
| } | |
| if (ngx_strcmp(value[i].data, "off") == 0) { | |
| lscf->open_file_cache = NULL; | |
| continue; | |
| } | |
| failed: | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "invalid \"open_log_file_cache\" parameter \"%V\"", | |
| &value[i]); | |
| return NGX_CONF_ERROR; | |
| } | |
| if (lscf->open_file_cache == NULL) { | |
| return NGX_CONF_OK; | |
| } | |
| if (max == 0) { | |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| "\"open_log_file_cache\" must have \"max\" parameter"); | |
| return NGX_CONF_ERROR; | |
| } | |
| lscf->open_file_cache = ngx_open_file_cache_init(cf->pool, max, inactive); | |
| if (lscf->open_file_cache) { | |
| lscf->open_file_cache_valid = valid; | |
| lscf->open_file_cache_min_uses = min_uses; | |
| return NGX_CONF_OK; | |
| } | |
| return NGX_CONF_ERROR; | |
| } | |
| static ngx_int_t | |
| ngx_stream_log_init(ngx_conf_t *cf) | |
| { | |
| ngx_stream_handler_pt *h; | |
| ngx_stream_core_main_conf_t *cmcf; | |
| cmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_core_module); | |
| h = ngx_array_push(&cmcf->phases[NGX_STREAM_LOG_PHASE].handlers); | |
| if (h == NULL) { | |
| return NGX_ERROR; | |
| } | |
| *h = ngx_stream_log_handler; | |
| return NGX_OK; | |
| } |