26N/A

21N/A

7N/A#include "apr.h"

26N/A#include "apr_lib.h"

26N/A#include "apr_strings.h"

38N/A

26N/A#define APR_WANT_STRFUNC

26N/A#include "apr_want.h"

26N/A

30N/A#include "httpd.h"

26N/A#include "http_config.h"

46N/A#include "http_core.h"

46N/A#include "http_request.h"

46N/A

26N/A

26N/Amodule AP_MODULE_DECLARE_DATA usertrack_module;

46N/A

46N/Atypedef struct {

46N/A int always;

2N/A int expires;

46N/A} cookie_log_state;

46N/A

46N/Atypedef struct {

46N/A int enabled;

46N/A char *cookie_name;

26N/A} cookie_dir_rec;

46N/A

46N/A

46N/A#define COOKIE_NAME "Apache"

26N/A

46N/Astatic void make_cookie(request_rec *r)

46N/A{

46N/A cookie_log_state *cls = ap_get_module_config(r->server->module_config,

46N/A &usertrack_module);

46N/A /* 1024 == hardcoded constant */

46N/A char cookiebuf[1024];

46N/A char *new_cookie;

2N/A const char *rname = ap_get_remote_host(r->connection, r->per_dir_config,

46N/A REMOTE_NAME);

38N/A cookie_dir_rec *dcfg;

46N/A

46N/A dcfg = ap_get_module_config(r->per_dir_config, &usertrack_module);

46N/A

38N/A /* XXX: hmm, this should really tie in with mod_unique_id */

46N/A apr_snprintf(cookiebuf, sizeof(cookiebuf), "%s.%qd", rname, apr_time_now());

46N/A

26N/A if (cls->expires) {

26N/A apr_exploded_time_t tms;

46N/A

2N/A apr_explode_gmt(&tms, r->request_time + cls->expires * APR_USEC_PER_SEC);

38N/A

26N/A /* Cookie with date; as strftime '%a, %d-%h-%y %H:%M:%S GMT' */

2N/A new_cookie = apr_psprintf(r->pool,

32N/A "%s=%s; path=/; expires=%s, %.2d-%s-%.2d %.2d:%.2d:%.2d GMT",

32N/A dcfg->cookie_name, cookiebuf, apr_day_snames[tms.tm_wday],

32N/A tms.tm_mday, apr_month_snames[tms.tm_mon],

32N/A tms.tm_year % 100,

32N/A tms.tm_hour, tms.tm_min, tms.tm_sec);

32N/A }

32N/A else {

32N/A new_cookie = apr_psprintf(r->pool, "%s=%s; path=/",

32N/A dcfg->cookie_name, cookiebuf);

32N/A }

38N/A

38N/A apr_table_setn(r->headers_out, "Set-Cookie", new_cookie);

38N/A apr_table_setn(r->notes, "cookie", apr_pstrdup(r->pool, cookiebuf)); /* log first time */

38N/A return;

38N/A}

38N/A

38N/Astatic int spot_cookie(request_rec *r)

38N/A{

38N/A cookie_dir_rec *dcfg = ap_get_module_config(r->per_dir_config,

38N/A &usertrack_module);

38N/A const char *cookie;

38N/A const char *value;

38N/A

26N/A if (!dcfg->enabled) {

26N/A return DECLINED;

26N/A }

26N/A

26N/A if ((cookie = apr_table_get(r->headers_in, "Cookie")))

26N/A if ((value = ap_strstr_c(cookie, dcfg->cookie_name))) {

26N/A char *cookiebuf, *cookieend;

26N/A

26N/A value += strlen(dcfg->cookie_name) + 1; /* Skip over the '=' */

32N/A cookiebuf = apr_pstrdup(r->pool, value);

32N/A cookieend = strchr(cookiebuf, ';');

32N/A if (cookieend)

32N/A *cookieend = '\0'; /* Ignore anything after a ; */

32N/A

32N/A /* Set the cookie in a note, for logging */

32N/A apr_table_setn(r->notes, "cookie", cookiebuf);

32N/A

38N/A return DECLINED; /* There's already a cookie, no new one */

38N/A }

38N/A make_cookie(r);

32N/A return OK; /* We set our cookie */

}

static void *make_cookie_log_state(apr_pool_t *p, server_rec *s)

{

cookie_log_state *cls =

(cookie_log_state *) apr_palloc(p, sizeof(cookie_log_state));

cls->expires = 0;

return (void *) cls;

}

static void *make_cookie_dir(apr_pool_t *p, char *d)

{

cookie_dir_rec *dcfg;

dcfg = (cookie_dir_rec *) apr_pcalloc(p, sizeof(cookie_dir_rec));

dcfg->cookie_name = COOKIE_NAME;

dcfg->enabled = 0;

return dcfg;

}

static const char *set_cookie_enable(cmd_parms *cmd, void *mconfig, int arg)

{

cookie_dir_rec *dcfg = mconfig;

dcfg->enabled = arg;

return NULL;

}

static const char *set_cookie_exp(cmd_parms *parms, void *dummy, const char *arg)

{

cookie_log_state *cls = ap_get_module_config(parms->server->module_config,

&usertrack_module);

time_t factor, modifier = 0;

time_t num = 0;

char *word;

/* The simple case first - all numbers (we assume) */

if (apr_isdigit(arg[0]) && apr_isdigit(arg[strlen(arg) - 1])) {

cls->expires = atol(arg);

return NULL;

}

/*

word = ap_getword_conf(parms->pool, &arg);

if (!strncasecmp(word, "plus", 1)) {

word = ap_getword_conf(parms->pool, &arg);

};

/* {<num> <type>}* */

while (word[0]) {

/* <num> */

if (apr_isdigit(word[0]))

num = atoi(word);

else

return "bad expires code, numeric value expected.";

/* <type> */

word = ap_getword_conf(parms->pool, &arg);

if (!word[0])

return "bad expires code, missing <type>";

factor = 0;

if (!strncasecmp(word, "years", 1))

factor = 60 * 60 * 24 * 365;

else if (!strncasecmp(word, "months", 2))

factor = 60 * 60 * 24 * 30;

else if (!strncasecmp(word, "weeks", 1))

factor = 60 * 60 * 24 * 7;

else if (!strncasecmp(word, "days", 1))

factor = 60 * 60 * 24;

else if (!strncasecmp(word, "hours", 1))

factor = 60 * 60;

else if (!strncasecmp(word, "minutes", 2))

factor = 60;

else if (!strncasecmp(word, "seconds", 1))

factor = 1;

else

return "bad expires code, unrecognized type";

modifier = modifier + factor * num;

/* next <num> */

word = ap_getword_conf(parms->pool, &arg);

}

cls->expires = modifier;

return NULL;

}

static const char *set_cookie_name(cmd_parms *cmd, void *mconfig, const char *name)

{

cookie_dir_rec *dcfg = (cookie_dir_rec *) mconfig;

dcfg->cookie_name = apr_pstrdup(cmd->pool, name);

return NULL;

}

static const command_rec cookie_log_cmds[] = {

AP_INIT_TAKE1("CookieExpires", set_cookie_exp, NULL, RSRC_CONF,

"an expiry date code"),

AP_INIT_FLAG("CookieTracking", set_cookie_enable, NULL, OR_FILEINFO,

"whether or not to enable cookies"),

AP_INIT_TAKE1("CookieName", set_cookie_name, NULL, OR_FILEINFO,

"name of the tracking cookie"),

{NULL}

};

static void register_hooks(apr_pool_t *p)

{

ap_hook_fixups(spot_cookie,NULL,NULL,APR_HOOK_MIDDLE);

}

module AP_MODULE_DECLARE_DATA usertrack_module = {

STANDARD20_MODULE_STUFF,

make_cookie_dir, /* dir config creater */

NULL, /* dir merger --- default is to override */

make_cookie_log_state, /* server config */

NULL, /* merge server configs */

cookie_log_cmds, /* command apr_table_t */

register_hooks /* register hooks */

};