Mercurial > hg > nginx
view src/http/modules/ngx_http_status_module.c @ 1417:b23a80f9a7b8
set current peer to use it in ngx_http_upstream_free_round_robin_peer()
author | Igor Sysoev <igor@sysoev.ru> |
---|---|
date | Fri, 24 Aug 2007 15:29:50 +0000 |
parents | 63a820b0bc6c |
children | 74477ea8074f |
line wrap: on
line source
/* * 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 + 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 < 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; }