src/mod_usertrack.c

   1 #include "first.h"
   2
   3 #include "base.h"
   4 #include "log.h"
   5 #include "buffer.h"
   6 #include "rand.h"
   7
   8 #include "plugin.h"
   9
  10 #include <ctype.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13
  14 #include "md5.h"
  15
  16 /* plugin config for all request/connections */
  17
  18 typedef struct {
  19         buffer *cookie_name;
  20         buffer *cookie_domain;
  21         unsigned int cookie_max_age;
  22 } plugin_config;
  23
  24 typedef struct {
  25         PLUGIN_DATA;
  26
  27         plugin_config **config_storage;
  28
  29         plugin_config conf;
  30 } plugin_data;
  31
  32 /* init the plugin data */
  33 INIT_FUNC(mod_usertrack_init) {
  34         plugin_data *p;
  35
  36         p = calloc(1, sizeof(*p));
  37
  38         return p;
  39 }
  40
  41 /* detroy the plugin data */
  42 FREE_FUNC(mod_usertrack_free) {
  43         plugin_data *p = p_d;
  44
  45         UNUSED(srv);
  46
  47         if (!p) return HANDLER_GO_ON;
  48
  49         if (p->config_storage) {
  50                 size_t i;
  51                 for (i = 0; i < srv->config_context->used; i++) {
  52                         plugin_config *s = p->config_storage[i];
  53
  54                         if (NULL == s) continue;
  55
  56                         buffer_free(s->cookie_name);
  57                         buffer_free(s->cookie_domain);
  58
  59                         free(s);
  60                 }
  61                 free(p->config_storage);
  62         }
  63
  64         free(p);
  65
  66         return HANDLER_GO_ON;
  67 }
  68
  69 /* handle plugin config and check values */
  70
  71 SETDEFAULTS_FUNC(mod_usertrack_set_defaults) {
  72         plugin_data *p = p_d;
  73         size_t i = 0;
  74
  75         config_values_t cv[] = {
  76                 { "usertrack.cookie-name",       NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION },       /* 0 */
  77                 { "usertrack.cookie-max-age",    NULL, T_CONFIG_INT, T_CONFIG_SCOPE_CONNECTION },          /* 1 */
  78                 { "usertrack.cookie-domain",     NULL, T_CONFIG_STRING, T_CONFIG_SCOPE_CONNECTION },       /* 2 */
  79
  80                 { NULL,                          NULL, T_CONFIG_UNSET, T_CONFIG_SCOPE_UNSET }
  81         };
  82
  83         if (!p) return HANDLER_ERROR;
  84
  85         p->config_storage = calloc(1, srv->config_context->used * sizeof(plugin_config *));
  86
  87         for (i = 0; i < srv->config_context->used; i++) {
  88                 data_config const* config = (data_config const*)srv->config_context->data[i];
  89                 plugin_config *s;
  90
  91                 s = calloc(1, sizeof(plugin_config));
  92                 s->cookie_name    = buffer_init();
  93                 s->cookie_domain  = buffer_init();
  94                 s->cookie_max_age = 0;
  95
  96                 cv[0].destination = s->cookie_name;
  97                 cv[1].destination = &(s->cookie_max_age);
  98                 cv[2].destination = s->cookie_domain;
  99
 100                 p->config_storage[i] = s;
 101
 102                 if (0 != config_insert_values_global(srv, config->value, cv, i == 0 ? T_CONFIG_SCOPE_SERVER : T_CONFIG_SCOPE_CONNECTION)) {
 103                         return HANDLER_ERROR;
 104                 }
 105
 106                 if (buffer_string_is_empty(s->cookie_name)) {
 107                         buffer_copy_string_len(s->cookie_name, CONST_STR_LEN("TRACKID"));
 108                 } else {
 109                         size_t j, len = buffer_string_length(s->cookie_name);
 110                         for (j = 0; j < len; j++) {
 111                                 char c = s->cookie_name->ptr[j] | 32;
 112                                 if (c < 'a' || c > 'z') {
 113                                         log_error_write(srv, __FILE__, __LINE__, "sb",
 114                                                         "invalid character in usertrack.cookie-name:",
 115                                                         s->cookie_name);
 116
 117                                         return HANDLER_ERROR;
 118                                 }
 119                         }
 120                 }
 121
 122                 if (!buffer_string_is_empty(s->cookie_domain)) {
 123                         size_t j, len = buffer_string_length(s->cookie_domain);
 124                         for (j = 0; j < len; j++) {
 125                                 char c = s->cookie_domain->ptr[j];
 126                                 if (c <= 32 || c >= 127 || c == '"' || c == '\\') {
 127                                         log_error_write(srv, __FILE__, __LINE__, "sb",
 128                                                         "invalid character in usertrack.cookie-domain:",
 129                                                         s->cookie_domain);
 130
 131                                         return HANDLER_ERROR;
 132                                 }
 133                         }
 134                 }
 135         }
 136
 137         return HANDLER_GO_ON;
 138 }
 139
 140 #define PATCH(x) \
 141         p->conf.x = s->x;
 142 static int mod_usertrack_patch_connection(server *srv, connection *con, plugin_data *p) {
 143         size_t i, j;
 144         plugin_config *s = p->config_storage[0];
 145
 146         PATCH(cookie_name);
 147         PATCH(cookie_domain);
 148         PATCH(cookie_max_age);
 149
 150         /* skip the first, the global context */
 151         for (i = 1; i < srv->config_context->used; i++) {
 152                 data_config *dc = (data_config *)srv->config_context->data[i];
 153                 s = p->config_storage[i];
 154
 155                 /* condition didn't match */
 156                 if (!config_check_cond(srv, con, dc)) continue;
 157
 158                 /* merge config */
 159                 for (j = 0; j < dc->value->used; j++) {
 160                         data_unset *du = dc->value->data[j];
 161
 162                         if (buffer_is_equal_string(du->key, CONST_STR_LEN("usertrack.cookie-name"))) {
 163                                 PATCH(cookie_name);
 164                         } else if (buffer_is_equal_string(du->key, CONST_STR_LEN("usertrack.cookie-max-age"))) {
 165                                 PATCH(cookie_max_age);
 166                         } else if (buffer_is_equal_string(du->key, CONST_STR_LEN("usertrack.cookie-domain"))) {
 167                                 PATCH(cookie_domain);
 168                         }
 169                 }
 170         }
 171
 172         return 0;
 173 }
 174 #undef PATCH
 175
 176 URIHANDLER_FUNC(mod_usertrack_uri_handler) {
 177         plugin_data *p = p_d;
 178         data_string *ds;
 179         unsigned char h[16];
 180         li_MD5_CTX Md5Ctx;
 181         char hh[LI_ITOSTRING_LENGTH];
 182
 183         if (buffer_is_empty(con->uri.path)) return HANDLER_GO_ON;
 184
 185         mod_usertrack_patch_connection(srv, con, p);
 186
 187         if (NULL != (ds = (data_string *)array_get_element(con->request.headers, "Cookie"))) {
 188                 char *g;
 189                 /* we have a cookie, does it contain a valid name ? */
 190
 191                 /* parse the cookie
 192                  *
 193                  * check for cookiename + (WS | '=')
 194                  *
 195                  */
 196
 197                 if (NULL != (g = strstr(ds->value->ptr, p->conf.cookie_name->ptr))) {
 198                         char *nc;
 199
 200                         /* skip WS */
 201                         for (nc = g + buffer_string_length(p->conf.cookie_name); *nc == ' ' || *nc == '\t'; nc++);
 202
 203                         if (*nc == '=') {
 204                                 /* ok, found the key of our own cookie */
 205
 206                                 if (strlen(nc) > 32) {
 207                                         /* i'm lazy */
 208                                         return HANDLER_GO_ON;
 209                                 }
 210                         }
 211                 }
 212         }
 213
 214         /* set a cookie */
 215         if (NULL == (ds = (data_string *)array_get_unused_element(con->response.headers, TYPE_STRING))) {
 216                 ds = data_response_init();
 217         }
 218         buffer_copy_string_len(ds->key, CONST_STR_LEN("Set-Cookie"));
 219         buffer_copy_buffer(ds->value, p->conf.cookie_name);
 220         buffer_append_string_len(ds->value, CONST_STR_LEN("="));
 221
 222
 223         /* taken from mod_auth.c */
 224
 225         /* generate shared-secret */
 226         li_MD5_Init(&Md5Ctx);
 227         li_MD5_Update(&Md5Ctx, CONST_BUF_LEN(con->uri.path));
 228         li_MD5_Update(&Md5Ctx, CONST_STR_LEN("+"));
 229
 230         /* we assume sizeof(time_t) == 4 here, but if not it ain't a problem at all */
 231         li_itostrn(hh, sizeof(hh), srv->cur_ts);
 232         li_MD5_Update(&Md5Ctx, (unsigned char *)hh, strlen(hh));
 233         li_MD5_Update(&Md5Ctx, (unsigned char *)srv->entropy, sizeof(srv->entropy));
 234         li_itostrn(hh, sizeof(hh), li_rand());
 235         li_MD5_Update(&Md5Ctx, (unsigned char *)hh, strlen(hh));
 236
 237         li_MD5_Final(h, &Md5Ctx);
 238
 239         buffer_append_string_encoded(ds->value, (char *)h, 16, ENCODING_HEX);
 240         buffer_append_string_len(ds->value, CONST_STR_LEN("; Path=/"));
 241         buffer_append_string_len(ds->value, CONST_STR_LEN("; Version=1"));
 242
 243         if (!buffer_string_is_empty(p->conf.cookie_domain)) {
 244                 buffer_append_string_len(ds->value, CONST_STR_LEN("; Domain="));
 245                 buffer_append_string_encoded(ds->value, CONST_BUF_LEN(p->conf.cookie_domain), ENCODING_REL_URI);
 246         }
 247
 248         if (p->conf.cookie_max_age) {
 249                 buffer_append_string_len(ds->value, CONST_STR_LEN("; max-age="));
 250                 buffer_append_int(ds->value, p->conf.cookie_max_age);
 251         }
 252
 253         array_insert_unique(con->response.headers, (data_unset *)ds);
 254
 255         return HANDLER_GO_ON;
 256 }
 257
 258 /* this function is called at dlopen() time and inits the callbacks */
 259
 260 int mod_usertrack_plugin_init(plugin *p);
 261 int mod_usertrack_plugin_init(plugin *p) {
 262         p->version     = LIGHTTPD_VERSION_ID;
 263         p->name        = buffer_init_string("usertrack");
 264
 265         p->init        = mod_usertrack_init;
 266         p->handle_uri_clean  = mod_usertrack_uri_handler;
 267         p->set_defaults  = mod_usertrack_set_defaults;
 268         p->cleanup     = mod_usertrack_free;
 269
 270         p->data        = NULL;
 271
 272         return 0;
 273 }