Skip to content
Find file
Fetching contributors…
Cannot retrieve contributors at this time
683 lines (502 sloc) 16.8 KB
/*
* Copyright (C) Igor Sysoev
*/
#include <ngx_config.h>
#include <ngx_core.h>
#include <ngx_http.h>
#include <uuid/uuid.h>
#include <ctype.h>
#define NGX_HTTP_USERID_OFF 0
#define NGX_HTTP_USERID_LOG 1
#define NGX_HTTP_USERID_V1 2
#define NGX_HTTP_USERID_ON 3
/* 31 Dec 2037 23:55:55 GMT */
#define NGX_HTTP_USERID_MAX_EXPIRES 2145916555
int uuid_verify(const char *in)
{
int i;
const char *cp;
if (strlen(in) < 32)
return -1;
for (i=0, cp = in; i < 32; i++,cp++) {
if (!isxdigit(*cp))
return -1;
}
return 0;
}
typedef struct {
ngx_uint_t enable;
ngx_int_t service;
ngx_str_t name;
ngx_str_t domain;
ngx_str_t path;
ngx_str_t p3p;
time_t expires;
u_char mark;
} ngx_http_uuid_conf_t;
typedef struct {
uint32_t uid_got[4];
uint32_t uid_set[4];
ngx_str_t cookie;
} ngx_http_uuid_ctx_t;
static ngx_http_uuid_ctx_t *ngx_http_uuid_get_uid(ngx_http_request_t *r,
ngx_http_uuid_conf_t *conf);
static ngx_int_t ngx_http_uuid_variable(ngx_http_request_t *r,
ngx_http_variable_value_t *v, ngx_str_t *name, uint32_t *uid);
static ngx_int_t ngx_http_uuid_set_uid(ngx_http_request_t *r,
ngx_http_uuid_ctx_t *ctx, ngx_http_uuid_conf_t *conf);
static ngx_int_t ngx_http_uuid_add_variables(ngx_conf_t *cf);
static ngx_int_t ngx_http_uuid_init(ngx_conf_t *cf);
static void *ngx_http_uuid_create_conf(ngx_conf_t *cf);
static char *ngx_http_uuid_merge_conf(ngx_conf_t *cf, void *parent,
void *child);
static char *ngx_http_uuid_domain(ngx_conf_t *cf, void *post, void *data);
static char *ngx_http_uuid_path(ngx_conf_t *cf, void *post, void *data);
static char *ngx_http_uuid_expires(ngx_conf_t *cf, ngx_command_t *cmd,
void *conf);
static char *ngx_http_uuid_p3p(ngx_conf_t *cf, void *post, void *data);
static char *ngx_http_uuid_mark(ngx_conf_t *cf, ngx_command_t *cmd,
void *conf);
static ngx_int_t ngx_http_uuid_init_worker(ngx_cycle_t *cycle);
static uint32_t start_value;
static u_char expires[] = "; expires=Thu, 31-Dec-37 23:55:55 GMT";
static ngx_http_output_header_filter_pt ngx_http_next_header_filter;
static ngx_conf_enum_t ngx_http_uuid_state[] = {
{ ngx_string("off"), NGX_HTTP_USERID_OFF },
{ ngx_string("log"), NGX_HTTP_USERID_LOG },
{ ngx_string("v1"), NGX_HTTP_USERID_V1 },
{ ngx_string("on"), NGX_HTTP_USERID_ON },
{ ngx_null_string, 0 }
};
static ngx_conf_post_handler_pt ngx_http_uuid_domain_p =
ngx_http_uuid_domain;
static ngx_conf_post_handler_pt ngx_http_uuid_path_p = ngx_http_uuid_path;
static ngx_conf_post_handler_pt ngx_http_uuid_p3p_p = ngx_http_uuid_p3p;
static ngx_command_t ngx_http_uuid_commands[] = {
{ ngx_string("uuid"),
NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
ngx_conf_set_enum_slot,
NGX_HTTP_LOC_CONF_OFFSET,
offsetof(ngx_http_uuid_conf_t, enable),
ngx_http_uuid_state },
{ ngx_string("uuid_service"),
NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
ngx_conf_set_num_slot,
NGX_HTTP_LOC_CONF_OFFSET,
offsetof(ngx_http_uuid_conf_t, service),
NULL },
{ ngx_string("uuid_name"),
NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
ngx_conf_set_str_slot,
NGX_HTTP_LOC_CONF_OFFSET,
offsetof(ngx_http_uuid_conf_t, name),
NULL },
{ ngx_string("uuid_domain"),
NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
ngx_conf_set_str_slot,
NGX_HTTP_LOC_CONF_OFFSET,
offsetof(ngx_http_uuid_conf_t, domain),
&ngx_http_uuid_domain_p },
{ ngx_string("uuid_path"),
NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
ngx_conf_set_str_slot,
NGX_HTTP_LOC_CONF_OFFSET,
offsetof(ngx_http_uuid_conf_t, path),
&ngx_http_uuid_path_p },
{ ngx_string("uuid_expires"),
NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
ngx_http_uuid_expires,
NGX_HTTP_LOC_CONF_OFFSET,
0,
NULL },
{ ngx_string("uuid_p3p"),
NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
ngx_conf_set_str_slot,
NGX_HTTP_LOC_CONF_OFFSET,
offsetof(ngx_http_uuid_conf_t, p3p),
&ngx_http_uuid_p3p_p },
{ ngx_string("uuid_mark"),
NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
ngx_http_uuid_mark,
NGX_HTTP_LOC_CONF_OFFSET,
0,
NULL },
ngx_null_command
};
static ngx_http_module_t ngx_http_uuid_filter_module_ctx = {
ngx_http_uuid_add_variables, /* preconfiguration */
ngx_http_uuid_init, /* postconfiguration */
NULL, /* create main configuration */
NULL, /* init main configuration */
NULL, /* create server configuration */
NULL, /* merge server configuration */
ngx_http_uuid_create_conf, /* create location configration */
ngx_http_uuid_merge_conf /* merge location configration */
};
ngx_module_t ngx_http_uuid_filter_module = {
NGX_MODULE_V1,
&ngx_http_uuid_filter_module_ctx, /* module context */
ngx_http_uuid_commands, /* module directives */
NGX_HTTP_MODULE, /* module type */
NULL, /* init master */
NULL, /* init module */
ngx_http_uuid_init_worker, /* init process */
NULL, /* init thread */
NULL, /* exit thread */
NULL, /* exit process */
NULL, /* exit master */
NGX_MODULE_V1_PADDING
};
static ngx_str_t ngx_http_uuid_got = ngx_string("uuid_got");
static ngx_str_t ngx_http_uuid_set = ngx_string("uuid_set");
static ngx_int_t
ngx_http_uuid_filter(ngx_http_request_t *r)
{
ngx_http_uuid_ctx_t *ctx;
ngx_http_uuid_conf_t *conf;
if (r != r->main) {
return ngx_http_next_header_filter(r);
}
conf = ngx_http_get_module_loc_conf(r, ngx_http_uuid_filter_module);
if (conf->enable <= NGX_HTTP_USERID_LOG) {
return ngx_http_next_header_filter(r);
}
ctx = ngx_http_uuid_get_uid(r, conf);
if (ctx == NULL) {
return NGX_ERROR;
}
if (ctx->uid_got[3] != 0) {
return ngx_http_next_header_filter(r);
}
/* ctx->status == NGX_DECLINED */
if (ngx_http_uuid_set_uid(r, ctx, conf) == NGX_OK) {
return ngx_http_next_header_filter(r);
}
return NGX_ERROR;
}
static ngx_int_t
ngx_http_uuid_got_variable(ngx_http_request_t *r,
ngx_http_variable_value_t *v, uintptr_t data)
{
ngx_http_uuid_ctx_t *ctx;
ngx_http_uuid_conf_t *conf;
conf = ngx_http_get_module_loc_conf(r->main, ngx_http_uuid_filter_module);
if (conf->enable == NGX_HTTP_USERID_OFF) {
v->not_found = 1;
return NGX_OK;
}
ctx = ngx_http_uuid_get_uid(r, conf);
if (ctx == NULL) {
return NGX_ERROR;
}
if (ctx->uid_got[3] != 0) {
return ngx_http_uuid_variable(r, v, &conf->name, ctx->uid_got);
}
/* ctx->status == NGX_DECLINED */
v->not_found = 1;
return NGX_OK;
}
static ngx_int_t
ngx_http_uuid_set_variable(ngx_http_request_t *r,
ngx_http_variable_value_t *v, uintptr_t data)
{
ngx_http_uuid_ctx_t *ctx;
ngx_http_uuid_conf_t *conf;
ctx = ngx_http_get_module_ctx(r, ngx_http_uuid_filter_module);
if (ctx == NULL || ctx->uid_set[3] == 0) {
v->not_found = 1;
return NGX_OK;
}
conf = ngx_http_get_module_loc_conf(r, ngx_http_uuid_filter_module);
return ngx_http_uuid_variable(r, v, &conf->name, ctx->uid_set);
}
static ngx_http_uuid_ctx_t *
ngx_http_uuid_get_uid(ngx_http_request_t *r, ngx_http_uuid_conf_t *conf)
{
ngx_int_t n;
ngx_http_uuid_ctx_t *ctx;
ctx = ngx_http_get_module_ctx(r, ngx_http_uuid_filter_module);
if (ctx) {
return ctx;
}
if (ctx == NULL) {
ctx = ngx_pcalloc(r->pool, sizeof(ngx_http_uuid_ctx_t));
if (ctx == NULL) {
return NULL;
}
ngx_http_set_ctx(r, ctx, ngx_http_uuid_filter_module);
}
n = ngx_http_parse_multi_header_lines(&r->headers_in.cookies, &conf->name,
&ctx->cookie);
if (n == NGX_DECLINED) {
return ctx;
}
if(uuid_verify((char *) ctx->cookie.data) == -1) {
ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
"INVALID uid cookie: \"%V\"", &ctx->cookie);
ctx->uid_got[3] = 0;
} else {
ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
"valid uid cookie: \"%V\"", &ctx->cookie);
ctx->uid_got[3] = 1;
}
return ctx;
}
static ngx_int_t
ngx_http_uuid_set_uid(ngx_http_request_t *r, ngx_http_uuid_ctx_t *ctx,
ngx_http_uuid_conf_t *conf)
{
u_char *cookie, *p;
size_t len;
ngx_table_elt_t *set_cookie, *p3p;
if (ctx->uid_got[3] == 0) {
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];
}
len = conf->name.len + 1 + 32 + conf->path.len;
if (conf->expires) {
len += sizeof(expires) - 1 + 2;
}
if (conf->domain.len) {
len += conf->domain.len;
}
cookie = ngx_palloc(r->pool, len);
if (cookie == NULL) {
return NGX_ERROR;
}
p = ngx_copy(cookie, conf->name.data, conf->name.len);
*p++ = '=';
if (ctx->uid_got[3] == 0) {
uuid_t uuid;
uuid_generate_random(uuid);
ngx_hex_dump(p, (u_char *) uuid, 16);
p += 32;
} else {
p = ngx_cpymem(p, ctx->cookie.data, 32);
*p++ = '=';
}
if (conf->expires == NGX_HTTP_USERID_MAX_EXPIRES) {
p = ngx_cpymem(p, expires, sizeof(expires) - 1);
} else if (conf->expires) {
p = ngx_cpymem(p, expires, sizeof("; expires=") - 1);
p = ngx_http_cookie_time(p, ngx_time() + conf->expires);
}
p = ngx_copy(p, conf->domain.data, conf->domain.len);
p = ngx_copy(p, conf->path.data, conf->path.len);
set_cookie = ngx_list_push(&r->headers_out.headers);
if (set_cookie == NULL) {
return NGX_ERROR;
}
set_cookie->hash = 1;
set_cookie->key.len = sizeof("Set-Cookie") - 1;
set_cookie->key.data = (u_char *) "Set-Cookie";
set_cookie->value.len = p - cookie;
set_cookie->value.data = cookie;
ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
"uid cookie: \"%V\"", &set_cookie->value);
if (conf->p3p.len == 0) {
return NGX_OK;
}
p3p = ngx_list_push(&r->headers_out.headers);
if (p3p == NULL) {
return NGX_ERROR;
}
p3p->hash = 1;
p3p->key.len = sizeof("P3P") - 1;
p3p->key.data = (u_char *) "P3P";
p3p->value = conf->p3p;
return NGX_OK;
}
static ngx_int_t
ngx_http_uuid_variable(ngx_http_request_t *r, ngx_http_variable_value_t *v,
ngx_str_t *name, uint32_t *uid)
{
v->len = name->len + sizeof("=00001111222233334444555566667777") - 1;
v->data = ngx_palloc(r->pool, v->len);
if (v->data == NULL) {
return NGX_ERROR;
}
v->valid = 1;
v->no_cacheable = 0;
v->not_found = 0;
ngx_sprintf(v->data, "%V=%08XD%08XD%08XD%08XD",
name, uid[0], uid[1], uid[2], uid[3]);
return NGX_OK;
}
static ngx_int_t
ngx_http_uuid_add_variables(ngx_conf_t *cf)
{
ngx_http_variable_t *var;
var = ngx_http_add_variable(cf, &ngx_http_uuid_got, NGX_HTTP_VAR_NOHASH);
if (var == NULL) {
return NGX_ERROR;
}
var->get_handler = ngx_http_uuid_got_variable;
var = ngx_http_add_variable(cf, &ngx_http_uuid_set, NGX_HTTP_VAR_NOHASH);
if (var == NULL) {
return NGX_ERROR;
}
var->get_handler = ngx_http_uuid_set_variable;
return NGX_OK;
}
static void *
ngx_http_uuid_create_conf(ngx_conf_t *cf)
{
ngx_http_uuid_conf_t *conf;
conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_uuid_conf_t));
if (conf == NULL) {
return NGX_CONF_ERROR;
}
/*
* set by ngx_pcalloc():
*
* conf->name.len = 0;
* conf->name.date = NULL;
* conf->domain.len = 0;
* conf->domain.date = NULL;
* conf->path.len = 0;
* conf->path.date = NULL;
* conf->p3p.len = 0;
* conf->p3p.date = NULL;
*/
conf->enable = NGX_CONF_UNSET_UINT;
conf->service = NGX_CONF_UNSET;
conf->expires = NGX_CONF_UNSET;
conf->mark = (u_char) '\xFF';
return conf;
}
static char *
ngx_http_uuid_merge_conf(ngx_conf_t *cf, void *parent, void *child)
{
ngx_http_uuid_conf_t *prev = parent;
ngx_http_uuid_conf_t *conf = child;
ngx_conf_merge_uint_value(conf->enable, prev->enable,
NGX_HTTP_USERID_OFF);
ngx_conf_merge_str_value(conf->name, prev->name, "uid");
ngx_conf_merge_str_value(conf->domain, prev->domain, "");
ngx_conf_merge_str_value(conf->path, prev->path, "; path=/");
ngx_conf_merge_str_value(conf->p3p, prev->p3p, "");
ngx_conf_merge_value(conf->service, prev->service, NGX_CONF_UNSET);
ngx_conf_merge_sec_value(conf->expires, prev->expires, 0);
if (conf->mark == (u_char) '\xFF') {
if (prev->mark == (u_char) '\xFF') {
conf->mark = '\0';
} else {
conf->mark = prev->mark;
}
}
return NGX_CONF_OK;
}
static ngx_int_t
ngx_http_uuid_init(ngx_conf_t *cf)
{
ngx_http_next_header_filter = ngx_http_top_header_filter;
ngx_http_top_header_filter = ngx_http_uuid_filter;
return NGX_OK;
}
static char *
ngx_http_uuid_domain(ngx_conf_t *cf, void *post, void *data)
{
ngx_str_t *domain = data;
u_char *p, *new;
if (ngx_strcmp(domain->data, "none") == 0) {
domain->len = 0;
domain->data = (u_char *) "";
return NGX_CONF_OK;
}
new = ngx_palloc(cf->pool, sizeof("; domain=") - 1 + domain->len);
if (new == NULL) {
return NGX_CONF_ERROR;
}
p = ngx_cpymem(new, "; domain=", sizeof("; domain=") - 1);
ngx_memcpy(p, domain->data, domain->len);
domain->len += sizeof("; domain=") - 1;
domain->data = new;
return NGX_CONF_OK;
}
static char *
ngx_http_uuid_path(ngx_conf_t *cf, void *post, void *data)
{
ngx_str_t *path = data;
u_char *p, *new;
new = ngx_palloc(cf->pool, sizeof("; path=") - 1 + path->len);
if (new == NULL) {
return NGX_CONF_ERROR;
}
p = ngx_cpymem(new, "; path=", sizeof("; path=") - 1);
ngx_memcpy(p, path->data, path->len);
path->len += sizeof("; path=") - 1;
path->data = new;
return NGX_CONF_OK;
}
static char *
ngx_http_uuid_expires(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
{
ngx_http_uuid_conf_t *ucf = conf;
ngx_str_t *value;
if (ucf->expires != NGX_CONF_UNSET) {
return "is duplicate";
}
value = cf->args->elts;
if (ngx_strcmp(value[1].data, "max") == 0) {
ucf->expires = NGX_HTTP_USERID_MAX_EXPIRES;
return NGX_CONF_OK;
}
if (ngx_strcmp(value[1].data, "off") == 0) {
ucf->expires = 0;
return NGX_CONF_OK;
}
ucf->expires = ngx_parse_time(&value[1], 1);
if (ucf->expires == NGX_ERROR) {
return "invalid value";
}
if (ucf->expires == NGX_PARSE_LARGE_TIME) {
return "value must be less than 68 years";
}
return NGX_CONF_OK;
}
static char *
ngx_http_uuid_p3p(ngx_conf_t *cf, void *post, void *data)
{
ngx_str_t *p3p = data;
if (ngx_strcmp(p3p->data, "none") == 0) {
p3p->len = 0;
p3p->data = (u_char *) "";
}
return NGX_CONF_OK;
}
static char *
ngx_http_uuid_mark(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
{
ngx_http_uuid_conf_t *ucf = conf;
ngx_str_t *value;
if (ucf->mark != (u_char) '\xFF') {
return "is duplicate";
}
value = cf->args->elts;
if (ngx_strcmp(value[1].data, "off") == 0) {
ucf->mark = '\0';
return NGX_CONF_OK;
}
if (value[1].len != 1
|| !((value[1].data[0] >= '0' && value[1].data[0] <= '9')
|| (value[1].data[0] >= 'A' && value[1].data[0] <= 'Z')
|| (value[1].data[0] >= 'a' && value[1].data[0] <= 'z')
|| value[1].data[0] == '='))
{
return "value must be \"off\" or a single letter, digit or \"=\"";
}
ucf->mark = value[1].data[0];
return NGX_CONF_OK;
}
static ngx_int_t
ngx_http_uuid_init_worker(ngx_cycle_t *cycle)
{
struct timeval tp;
ngx_gettimeofday(&tp);
/* use the most significant usec part that fits to 16 bits */
start_value = ((tp.tv_usec / 20) << 16) | ngx_pid;
return NGX_OK;
}
Jump to Line
Something went wrong with that request. Please try again.