| |
| /* |
| * 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; |
| } |