|
|
|
@ -19,6 +19,8 @@
|
|
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
|
buffer *cookie_name;
|
|
|
|
|
buffer *cookie_domain;
|
|
|
|
|
unsigned short cookie_max_age;
|
|
|
|
|
} plugin_config;
|
|
|
|
|
|
|
|
|
|
typedef struct {
|
|
|
|
@ -52,6 +54,7 @@ FREE_FUNC(mod_usertrack_free) {
|
|
|
|
|
plugin_config *s = p->config_storage[i];
|
|
|
|
|
|
|
|
|
|
buffer_free(s->cookie_name);
|
|
|
|
|
buffer_free(s->cookie_domain);
|
|
|
|
|
|
|
|
|
|
free(s);
|
|
|
|
|
}
|
|
|
|
@ -70,8 +73,12 @@ SETDEFAULTS_FUNC(mod_usertrack_set_defaults) {
|
|
|
|
|
size_t i = 0;
|
|
|
|
|
|
|
|
|
|
config_values_t cv[] = {
|
|
|
|
|
{ "usertrack.cookiename", NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 0 */
|
|
|
|
|
{ NULL, NULL, T_CONFIG_UNSET, T_CONFIG_SCOPE_UNSET }
|
|
|
|
|
{ "usertrack.cookie-name", NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 0 */
|
|
|
|
|
{ "usertrack.cookie-max-age", NULL, T_CONFIG_SHORT, T_CONFIG_SCOPE_CONNECTION }, /* 1 */
|
|
|
|
|
{ "usertrack.cookie-domain", NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION }, /* 2 */
|
|
|
|
|
|
|
|
|
|
{ "usertrack.cookiename", NULL, T_CONFIG_DEPRECATED, T_CONFIG_SCOPE_CONNECTION },
|
|
|
|
|
{ NULL, NULL, T_CONFIG_UNSET, T_CONFIG_SCOPE_UNSET }
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
if (!p) return HANDLER_ERROR;
|
|
|
|
@ -83,8 +90,12 @@ SETDEFAULTS_FUNC(mod_usertrack_set_defaults) {
|
|
|
|
|
|
|
|
|
|
s = malloc(sizeof(plugin_config));
|
|
|
|
|
s->cookie_name = buffer_init();
|
|
|
|
|
s->cookie_domain = buffer_init();
|
|
|
|
|
s->cookie_max_age = 0;
|
|
|
|
|
|
|
|
|
|
cv[0].destination = s->cookie_name;
|
|
|
|
|
cv[1].destination = &(s->cookie_max_age);
|
|
|
|
|
cv[2].destination = s->cookie_domain;
|
|
|
|
|
|
|
|
|
|
p->config_storage[i] = s;
|
|
|
|
|
|
|
|
|
@ -92,7 +103,7 @@ SETDEFAULTS_FUNC(mod_usertrack_set_defaults) {
|
|
|
|
|
return HANDLER_ERROR;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (s->cookie_name->used == 0) {
|
|
|
|
|
if (buffer_is_empty(s->cookie_name)) {
|
|
|
|
|
buffer_copy_string(s->cookie_name, "TRACKID");
|
|
|
|
|
} else {
|
|
|
|
|
size_t j;
|
|
|
|
@ -100,13 +111,27 @@ SETDEFAULTS_FUNC(mod_usertrack_set_defaults) {
|
|
|
|
|
char c = s->cookie_name->ptr[j] | 32;
|
|
|
|
|
if (c < 'a' || c > 'z') {
|
|
|
|
|
log_error_write(srv, __FILE__, __LINE__, "sb",
|
|
|
|
|
"invalid character in usertrack.cookiename:",
|
|
|
|
|
"invalid character in usertrack.cookie-name:",
|
|
|
|
|
s->cookie_name);
|
|
|
|
|
|
|
|
|
|
return HANDLER_ERROR;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!buffer_is_empty(s->cookie_domain)) {
|
|
|
|
|
size_t j;
|
|
|
|
|
for (j = 0; j < s->cookie_domain->used - 1; j++) {
|
|
|
|
|
char c = s->cookie_domain->ptr[j];
|
|
|
|
|
if (c <= 32 || c >= 127 || c == '"' || c == '\\') {
|
|
|
|
|
log_error_write(srv, __FILE__, __LINE__, "sb",
|
|
|
|
|
"invalid character in usertrack.cookie-domain:",
|
|
|
|
|
s->cookie_domain);
|
|
|
|
|
|
|
|
|
|
return HANDLER_ERROR;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return HANDLER_GO_ON;
|
|
|
|
@ -114,16 +139,18 @@ SETDEFAULTS_FUNC(mod_usertrack_set_defaults) {
|
|
|
|
|
|
|
|
|
|
#define PATCH(x) \
|
|
|
|
|
p->conf.x = s->x;
|
|
|
|
|
static int mod_usertrack_patch_connection(server *srv, connection *con, plugin_data *p, const char *stage, size_t stage_len) {
|
|
|
|
|
static int mod_usertrack_patch_connection(server *srv, connection *con, plugin_data *p) {
|
|
|
|
|
size_t i, j;
|
|
|
|
|
plugin_config *s = p->config_storage[0];
|
|
|
|
|
|
|
|
|
|
PATCH(cookie_name);
|
|
|
|
|
PATCH(cookie_domain);
|
|
|
|
|
PATCH(cookie_max_age);
|
|
|
|
|
|
|
|
|
|
/* skip the first, the global context */
|
|
|
|
|
for (i = 1; i < srv->config_context->used; i++) {
|
|
|
|
|
data_config *dc = (data_config *)srv->config_context->data[i];
|
|
|
|
|
plugin_config *s = p->config_storage[i];
|
|
|
|
|
|
|
|
|
|
/* not our stage */
|
|
|
|
|
if (!buffer_is_equal_string(dc->comp_key, stage, stage_len)) continue;
|
|
|
|
|
s = p->config_storage[i];
|
|
|
|
|
|
|
|
|
|
/* condition didn't match */
|
|
|
|
|
if (!config_check_cond(srv, con, dc)) continue;
|
|
|
|
@ -132,44 +159,30 @@ static int mod_usertrack_patch_connection(server *srv, connection *con, plugin_d
|
|
|
|
|
for (j = 0; j < dc->value->used; j++) {
|
|
|
|
|
data_unset *du = dc->value->data[j];
|
|
|
|
|
|
|
|
|
|
if (buffer_is_equal_string(du->key, CONST_STR_LEN("usertrack.cookiename"))) {
|
|
|
|
|
if (buffer_is_equal_string(du->key, CONST_STR_LEN("usertrack.cookie-name"))) {
|
|
|
|
|
PATCH(cookie_name);
|
|
|
|
|
} else if (buffer_is_equal_string(du->key, CONST_STR_LEN("usertrack.cookie-max-age"))) {
|
|
|
|
|
PATCH(cookie_max_age);
|
|
|
|
|
} else if (buffer_is_equal_string(du->key, CONST_STR_LEN("usertrack.cookie-domain"))) {
|
|
|
|
|
PATCH(cookie_domain);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static int mod_usertrack_setup_connection(server *srv, connection *con, plugin_data *p) {
|
|
|
|
|
plugin_config *s = p->config_storage[0];
|
|
|
|
|
UNUSED(srv);
|
|
|
|
|
UNUSED(con);
|
|
|
|
|
|
|
|
|
|
PATCH(cookie_name);
|
|
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
#undef PATCH
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
URIHANDLER_FUNC(mod_usertrack_uri_handler) {
|
|
|
|
|
plugin_data *p = p_d;
|
|
|
|
|
data_string *ds;
|
|
|
|
|
unsigned char h[16];
|
|
|
|
|
MD5_CTX Md5Ctx;
|
|
|
|
|
char hh[32];
|
|
|
|
|
size_t i;
|
|
|
|
|
|
|
|
|
|
if (con->uri.path->used == 0) return HANDLER_GO_ON;
|
|
|
|
|
|
|
|
|
|
mod_usertrack_setup_connection(srv, con, p);
|
|
|
|
|
for (i = 0; i < srv->config_patches->used; i++) {
|
|
|
|
|
buffer *patch = srv->config_patches->ptr[i];
|
|
|
|
|
|
|
|
|
|
mod_usertrack_patch_connection(srv, con, p, CONST_BUF_LEN(patch));
|
|
|
|
|
}
|
|
|
|
|
mod_usertrack_patch_connection(srv, con, p);
|
|
|
|
|
|
|
|
|
|
if (NULL != (ds = (data_string *)array_get_element(con->request.headers, "Cookie"))) {
|
|
|
|
|
char *g;
|
|
|
|
@ -204,7 +217,7 @@ URIHANDLER_FUNC(mod_usertrack_uri_handler) {
|
|
|
|
|
}
|
|
|
|
|
buffer_copy_string(ds->key, "Set-Cookie");
|
|
|
|
|
buffer_copy_string_buffer(ds->value, p->conf.cookie_name);
|
|
|
|
|
buffer_append_string(ds->value, "=");
|
|
|
|
|
buffer_append_string(ds->value, "=\"");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* taken from mod_auth.c */
|
|
|
|
@ -223,7 +236,19 @@ URIHANDLER_FUNC(mod_usertrack_uri_handler) {
|
|
|
|
|
MD5_Final(h, &Md5Ctx);
|
|
|
|
|
|
|
|
|
|
buffer_append_string_hex(ds->value, (char *)h, 16);
|
|
|
|
|
buffer_append_string(ds->value, "; path=/");
|
|
|
|
|
buffer_append_string(ds->value, "\"; Path=\"/\"");
|
|
|
|
|
buffer_append_string(ds->value, "; Version=\"1\"");
|
|
|
|
|
|
|
|
|
|
if (!buffer_is_empty(p->conf.cookie_domain)) {
|
|
|
|
|
buffer_append_string(ds->value, "; Domain=\"");
|
|
|
|
|
buffer_append_string_buffer(ds->value, p->conf.cookie_domain);
|
|
|
|
|
buffer_append_string(ds->value, "\"");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (p->conf.cookie_max_age) {
|
|
|
|
|
buffer_append_string(ds->value, "; max-age=");
|
|
|
|
|
buffer_append_long(ds->value, p->conf.cookie_max_age);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
array_insert_unique(con->response.headers, (data_unset *)ds);
|
|
|
|
|
|
|
|
|
|