mirror of
https://github.com/nginx/nginx.git
synced 2024-12-17 15:17:48 +08:00
6a75019ad6
*) use ngx_snprintf() in ngx_inet_ntop() and ngx_sock_ntop() as they are called just once per connection *) NGX_INET_ADDRSTRLEN
310 lines
8.0 KiB
C
310 lines
8.0 KiB
C
|
|
/*
|
|
* Copyright (C) Igor Sysoev
|
|
*/
|
|
|
|
|
|
#include <ngx_config.h>
|
|
#include <ngx_core.h>
|
|
#include <ngx_http.h>
|
|
|
|
|
|
typedef struct {
|
|
ngx_http_request_t *request;
|
|
ngx_pool_t *pool;
|
|
ngx_chain_t *head;
|
|
ngx_buf_t *last;
|
|
size_t size;
|
|
} ngx_http_status_ctx_t;
|
|
|
|
|
|
static ngx_int_t ngx_http_status(ngx_http_status_ctx_t *ctx);
|
|
static char *ngx_http_set_status(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
void *conf);
|
|
|
|
static ngx_command_t ngx_http_status_commands[] = {
|
|
|
|
{ ngx_string("status"),
|
|
NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_FLAG,
|
|
ngx_http_set_status,
|
|
0,
|
|
0,
|
|
NULL },
|
|
|
|
ngx_null_command
|
|
};
|
|
|
|
|
|
|
|
static ngx_http_module_t ngx_http_status_module_ctx = {
|
|
NULL, /* pre conf */
|
|
|
|
NULL, /* create main configuration */
|
|
NULL, /* init main configuration */
|
|
|
|
NULL, /* create server configuration */
|
|
NULL, /* merge server configuration */
|
|
|
|
NULL, /* create location configuration */
|
|
NULL /* merge location configuration */
|
|
};
|
|
|
|
|
|
ngx_module_t ngx_http_status_module = {
|
|
NGX_MODULE,
|
|
&ngx_http_status_module_ctx, /* module context */
|
|
ngx_http_status_commands, /* module directives */
|
|
NGX_HTTP_MODULE, /* module type */
|
|
NULL, /* init module */
|
|
NULL /* init process */
|
|
};
|
|
|
|
|
|
static ngx_int_t ngx_http_status_handler(ngx_http_request_t *r)
|
|
{
|
|
ngx_int_t rc;
|
|
ngx_http_status_ctx_t ctx;
|
|
|
|
if (r->method != NGX_HTTP_GET && r->method != NGX_HTTP_HEAD) {
|
|
return NGX_HTTP_NOT_ALLOWED;
|
|
}
|
|
|
|
rc = ngx_http_discard_body(r);
|
|
|
|
if (rc != NGX_OK && rc != NGX_AGAIN) {
|
|
return rc;
|
|
}
|
|
|
|
r->headers_out.content_type = ngx_list_push(&r->headers_out.headers);
|
|
if (r->headers_out.content_type == NULL) {
|
|
return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
}
|
|
|
|
r->headers_out.content_type->key.len = 0;
|
|
r->headers_out.content_type->key.data = NULL;
|
|
r->headers_out.content_type->value.len = sizeof("text/plain") - 1;
|
|
r->headers_out.content_type->value.data = (u_char *) "text/plain";
|
|
|
|
if (r->method == NGX_HTTP_HEAD) {
|
|
r->headers_out.status = NGX_HTTP_OK;
|
|
|
|
rc = ngx_http_send_header(r);
|
|
|
|
if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) {
|
|
return rc;
|
|
}
|
|
}
|
|
|
|
ctx.request = r;
|
|
ctx.pool = r->pool;
|
|
ctx.head = NULL;
|
|
ctx.size = 0;
|
|
|
|
if (ngx_http_status(&ctx) != NGX_OK) {
|
|
return NGX_HTTP_INTERNAL_SERVER_ERROR;
|
|
}
|
|
|
|
r->headers_out.status = NGX_HTTP_OK;
|
|
r->headers_out.content_length_n = ctx.size;
|
|
|
|
rc = ngx_http_send_header(r);
|
|
|
|
if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) {
|
|
return rc;
|
|
}
|
|
|
|
if (!r->main) {
|
|
ctx.last->last_buf = 1;
|
|
}
|
|
|
|
return ngx_http_output_filter(r, ctx.head);
|
|
}
|
|
|
|
|
|
static ngx_int_t ngx_http_status(ngx_http_status_ctx_t *ctx)
|
|
{
|
|
u_char ch;
|
|
size_t len, n;
|
|
ngx_uint_t i, dash;
|
|
ngx_buf_t *b;
|
|
ngx_chain_t *cl, **ll;
|
|
ngx_connection_t *c;
|
|
ngx_http_request_t *r;
|
|
ngx_http_core_main_conf_t *cmcf;
|
|
|
|
cmcf = ngx_http_get_module_main_conf(ctx->request, ngx_http_core_module);
|
|
|
|
#if (NGX_SUPPRESS_WARN)
|
|
b = NULL;
|
|
ll = NULL;
|
|
#endif
|
|
|
|
dash = 0;
|
|
|
|
/* TODO: old connections */
|
|
|
|
c = ngx_cycle->connections;
|
|
for (i = 0; i < ngx_cycle->connection_n; i++) {
|
|
|
|
/* TODO: trylock connection mutex */
|
|
|
|
r = c[i].data;
|
|
if (r && r->signature == NGX_HTTP_MODULE) {
|
|
|
|
/* STUB: should be NGX_PID_T_LEN */
|
|
len = NGX_INT64_LEN /* pid */
|
|
+ 1 + NGX_INT32_LEN /* connection */
|
|
+ 1 + 1 /* state */
|
|
+ 1 + NGX_INET_ADDRSTRLEN
|
|
+ 1 + (r->server_name ? cmcf->max_server_name_len : 1)
|
|
+ 2; /* "\r\n" */
|
|
|
|
/* BUG: cmcf->max_server_name_len and "*.domain.tld" */
|
|
|
|
|
|
if (r->request_line.len) {
|
|
len += 1 + 1 + r->request_line.len + 1;
|
|
}
|
|
|
|
if (!(b = ngx_create_temp_buf(ctx->pool, len))) {
|
|
/* TODO: unlock mutex */
|
|
return NGX_ERROR;
|
|
}
|
|
|
|
b->last = ngx_sprintf(b->last, "%P %5ui", ngx_pid, i);
|
|
|
|
switch (r->http_state) {
|
|
case NGX_HTTP_INITING_REQUEST_STATE:
|
|
ch = 'I';
|
|
break;
|
|
|
|
case NGX_HTTP_READING_REQUEST_STATE:
|
|
ch = 'R';
|
|
break;
|
|
|
|
case NGX_HTTP_PROCESS_REQUEST_STATE:
|
|
ch = 'P';
|
|
break;
|
|
|
|
case NGX_HTTP_WRITING_REQUEST_STATE:
|
|
ch = 'W';
|
|
break;
|
|
|
|
case NGX_HTTP_KEEPALIVE_STATE:
|
|
ch = 'K';
|
|
break;
|
|
|
|
default:
|
|
ch = '?';
|
|
}
|
|
|
|
*(b->last++) = ' ';
|
|
*(b->last++) = ch;
|
|
|
|
*(b->last++) = ' ';
|
|
b->last = ngx_cpymem(b->last, c[i].addr_text.data,
|
|
c[i].addr_text.len);
|
|
for (n = c[i].addr_text.len; n < NGX_INET_ADDRSTRLEN; n++) {
|
|
*(b->last++) = ' ';
|
|
}
|
|
|
|
*(b->last++) = ' ';
|
|
if (r->server_name) {
|
|
b->last = ngx_cpymem(b->last, r->server_name->data,
|
|
r->server_name->len);
|
|
for (n = r->server_name->len;
|
|
n < cmcf->max_server_name_len;
|
|
n++)
|
|
{
|
|
*(b->last++) = ' ';
|
|
}
|
|
|
|
} else {
|
|
*(b->last++) = '?';
|
|
}
|
|
|
|
if (r->request_line.len) {
|
|
*(b->last++) = ' ';
|
|
*(b->last++) = '"';
|
|
b->last = ngx_cpymem(b->last, r->request_line.data,
|
|
r->request_line.len);
|
|
*(b->last++) = '"';
|
|
|
|
}
|
|
|
|
*(b->last++) = CR; *(b->last++) = LF;
|
|
|
|
dash = 0;
|
|
|
|
} else if (c[i].fd != -1) {
|
|
len = NGX_INT64_LEN /* pid */
|
|
+ 1 + NGX_INT32_LEN /* connection */
|
|
+ 1 + 1 /* state */
|
|
+ 2; /* "\r\n" */
|
|
|
|
if (!(b = ngx_create_temp_buf(ctx->pool, len))) {
|
|
/* TODO: unlock mutex */
|
|
return NGX_ERROR;
|
|
}
|
|
|
|
b->last = ngx_sprintf(b->last, "%P %5ui", ngx_pid, i);
|
|
|
|
*(b->last++) = ' ';
|
|
*(b->last++) = 's';
|
|
|
|
*(b->last++) = CR; *(b->last++) = LF;
|
|
|
|
dash = 0;
|
|
|
|
} else if (!dash) {
|
|
len = 3;
|
|
|
|
if (!(b = ngx_create_temp_buf(ctx->pool, len))) {
|
|
/* TODO: unlock mutex */
|
|
return NGX_ERROR;
|
|
}
|
|
|
|
*(b->last++) = '-'; *(b->last++) = CR; *(b->last++) = LF;
|
|
|
|
dash = 1;
|
|
|
|
} else {
|
|
continue;
|
|
}
|
|
|
|
/* TODO: unlock mutex */
|
|
|
|
if (!(cl = ngx_alloc_chain_link(ctx->pool))) {
|
|
return NGX_ERROR;
|
|
}
|
|
|
|
if (ctx->head) {
|
|
*ll = cl;
|
|
|
|
} else {
|
|
ctx->head = cl;
|
|
}
|
|
|
|
cl->buf = b;
|
|
cl->next = NULL;
|
|
ll = &cl->next;
|
|
|
|
ctx->size += b->last - b->pos;
|
|
}
|
|
|
|
ctx->last = b;
|
|
|
|
return NGX_OK;
|
|
}
|
|
|
|
|
|
static char *ngx_http_set_status(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
{
|
|
ngx_http_core_loc_conf_t *clcf;
|
|
|
|
clcf = ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module);
|
|
clcf->handler = ngx_http_status_handler;
|
|
|
|
return NGX_CONF_OK;
|
|
}
|