mirror/nginx
1
0
Fork 0
mirror of https://github.com/nginx/nginx.git synced 2025-06-07 17:52:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

$uid_reset

Browse Source
This commit is contained in:
Igor Sysoev 2011-07-19 11:21:50 +00:00
parent a0202f26c6
commit 82afb9cc86
1 changed files with 64 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
src/http/modules/ngx_http_userid_filter_module.c
View File

@ -38,6 +38,7 @@ typedef struct {
uint32_t uid_got[4]; uint32_t uid_got[4];
uint32_t uid_set[4]; uint32_t uid_set[4];
ngx_str_t cookie; ngx_str_t cookie;
ngx_uint_t reset;
} ngx_http_userid_ctx_t; } ngx_http_userid_ctx_t;
@ -185,8 +186,10 @@ ngx_module_t ngx_http_userid_filter_module = {
}; };
static ngx_str_t ngx_http_userid_got = ngx_string("uid_got"); static ngx_str_t ngx_http_userid_got = ngx_string("uid_got");
static ngx_str_t ngx_http_userid_set = ngx_string("uid_set"); static ngx_str_t ngx_http_userid_set = ngx_string("uid_set");
static ngx_str_t ngx_http_userid_reset = ngx_string("uid_reset");
static ngx_uint_t ngx_http_userid_reset_index;
static ngx_int_t static ngx_int_t
@ -387,7 +390,7 @@ ngx_http_userid_set_uid(ngx_http_request_t *r, ngx_http_userid_ctx_t *ctx,
p = ngx_copy(cookie, conf->name.data, conf->name.len); p = ngx_copy(cookie, conf->name.data, conf->name.len);
*p++ = '='; *p++ = '=';
if (ctx->uid_got[3] == 0) { if (ctx->uid_got[3] == 0 || ctx->reset) {
src.len = 16; src.len = 16;
src.data = (u_char *) ctx->uid_set; src.data = (u_char *) ctx->uid_set;
dst.data = p; dst.data = p;
@ -452,11 +455,12 @@ static ngx_int_t
ngx_http_userid_create_uid(ngx_http_request_t *r, ngx_http_userid_ctx_t *ctx, ngx_http_userid_create_uid(ngx_http_request_t *r, ngx_http_userid_ctx_t *ctx,
ngx_http_userid_conf_t *conf) ngx_http_userid_conf_t *conf)
{ {
ngx_connection_t *c; ngx_connection_t *c;
struct sockaddr_in *sin; struct sockaddr_in *sin;
ngx_http_variable_value_t *vv;
#if (NGX_HAVE_INET6) #if (NGX_HAVE_INET6)
u_char *p; u_char *p;
struct sockaddr_in6 *sin6; struct sockaddr_in6 *sin6;
#endif #endif
if (ctx->uid_set[3] != 0) { if (ctx->uid_set[3] != 0) {
@ -465,20 +469,35 @@ ngx_http_userid_create_uid(ngx_http_request_t *r, ngx_http_userid_ctx_t *ctx,
if (ctx->uid_got[3] != 0) { if (ctx->uid_got[3] != 0) {
if (conf->mark == '\0' vv = ngx_http_get_indexed_variable(r, ngx_http_userid_reset_index);
|| (ctx->cookie.len > 23
&& ctx->cookie.data[22] == conf->mark if (vv->len == 0 || (vv->len == 1 && vv->data[0] == '0')) {
&& ctx->cookie.data[23] == '='))
{ if (conf->mark == '\0'
|| (ctx->cookie.len > 23
&& ctx->cookie.data[22] == conf->mark
&& ctx->cookie.data[23] == '='))
{
return NGX_OK;
}
ctx->uid_set[0] = ctx->uid_got[0];
ctx->uid_set[1] = ctx->uid_got[1];
ctx->uid_set[2] = ctx->uid_got[2];
ctx->uid_set[3] = ctx->uid_got[3];
return NGX_OK; return NGX_OK;
} else {
ctx->reset = 1;
if (vv->len == 3 && ngx_strncmp(vv->data, "log", 3) == 0) {
ngx_log_error(NGX_LOG_NOTICE, r->connection->log, 0,
"userid cookie \"%V=%08XD%08XD%08XD%08XD\" was reset",
&conf->name, ctx->uid_got[0], ctx->uid_got[1],
ctx->uid_got[2], ctx->uid_got[3]);
}
} }
ctx->uid_set[0] = ctx->uid_got[0];
ctx->uid_set[1] = ctx->uid_got[1];
ctx->uid_set[2] = ctx->uid_got[2];
ctx->uid_set[3] = ctx->uid_got[3];
return NGX_OK;
} }
/* /*
@ -565,9 +584,20 @@ ngx_http_userid_variable(ngx_http_request_t *r, ngx_http_variable_value_t *v,
} }
static ngx_int_t
ngx_http_userid_reset_variable(ngx_http_request_t *r,
ngx_http_variable_value_t *v, uintptr_t data)
{
*v = ngx_http_variable_null_value;
return NGX_OK;
}
static ngx_int_t static ngx_int_t
ngx_http_userid_add_variables(ngx_conf_t *cf) ngx_http_userid_add_variables(ngx_conf_t *cf)
{ {
ngx_int_t n;
ngx_http_variable_t *var; ngx_http_variable_t *var;
var = ngx_http_add_variable(cf, &ngx_http_userid_got, 0); var = ngx_http_add_variable(cf, &ngx_http_userid_got, 0);
@ -584,6 +614,21 @@ ngx_http_userid_add_variables(ngx_conf_t *cf)
var->get_handler = ngx_http_userid_set_variable; var->get_handler = ngx_http_userid_set_variable;
var = ngx_http_add_variable(cf, &ngx_http_userid_reset,
NGX_HTTP_VAR_CHANGEABLE);
if (var == NULL) {
return NGX_ERROR;
}
var->get_handler = ngx_http_userid_reset_variable;
n = ngx_http_get_variable_index(cf, &ngx_http_userid_reset);
if (n == NGX_ERROR) {
return NGX_ERROR;
}
ngx_http_userid_reset_index = n;
return NGX_OK; return NGX_OK;
} }