#include "apr_strings.h"

#include "ap_config.h"

#include "httpd.h"

#include "http_config.h"

#include "http_request.h"

#include "http_core.h"

#include "http_protocol.h"

#include "http_main.h"

#include "http_log.h"

#include "util_script.h"

typedef struct {

apr_table_t *action_types; /* Added with Action... */

const char *scripted[METHODS]; /* Added with Script... */

} action_dir_config;

module action_module;

static void *create_action_dir_config(apr_pool_t *p, char *dummy)

{

action_dir_config *new =

(action_dir_config *) apr_pcalloc(p, sizeof(action_dir_config));

new->action_types = apr_make_table(p, 4);

return new;

}

static void *merge_action_dir_configs(apr_pool_t *p, void *basev, void *addv)

{

action_dir_config *base = (action_dir_config *) basev;

action_dir_config *add = (action_dir_config *) addv;

action_dir_config *new = (action_dir_config *) apr_palloc(p,

sizeof(action_dir_config));

int i;

new->action_types = apr_overlay_tables(p, add->action_types,

base->action_types);

for (i = 0; i < METHODS; ++i) {

new->scripted[i] = add->scripted[i] ? add->scripted[i]

: base->scripted[i];

}

return new;

}

static const char *add_action(cmd_parms *cmd, void *m_v,

const char *type, const char *script)

{

action_dir_config *m = (action_dir_config *)m_v;

apr_table_setn(m->action_types, type, script);

return NULL;

}

static const char *set_script(cmd_parms *cmd, void *m_v,

const char *method, const char *script)

{

action_dir_config *m = (action_dir_config *)m_v;

int methnum;

methnum = ap_method_number_of(method);

if (methnum == M_TRACE)

return "TRACE not allowed for Script";

else if (methnum == M_INVALID)

return "Unknown method type for Script";

else

m->scripted[methnum] = script;

return NULL;

}

static const command_rec action_cmds[] =

{

AP_INIT_TAKE2("Action", add_action, NULL, OR_FILEINFO,

"a media type followed by a script name"),

AP_INIT_TAKE2("Script", set_script, NULL, ACCESS_CONF | RSRC_CONF,

"a method followed by a script name"),

{NULL}

};

static int action_handler(request_rec *r)

{

action_dir_config *conf = (action_dir_config *)

ap_get_module_config(r->per_dir_config, &action_module);

const char *t, *action = r->handler ? r->handler :

ap_field_noparam(r->pool, r->content_type);

const char *script;

int i;

/* Note that this handler handles _all_ types, so handler is unchecked */

/* Set allowed stuff */

for (i = 0; i < METHODS; ++i) {

if (conf->scripted[i])

r->allowed |= (1 << i);

}

/* First, check for the method-handling scripts */

if (r->method_number == M_GET) {

if (r->args)

script = conf->scripted[M_GET];

else

script = NULL;

}

else {

script = conf->scripted[r->method_number];

}

/* Check for looping, which can happen if the CGI script isn't */

if (script && r->prev && r->prev->prev)

return DECLINED;

/* Second, check for actions (which override the method scripts) */

if ((t = apr_table_get(conf->action_types,

action ? action : ap_default_type(r)))) {

script = t;

if (r->finfo.protection == 0) {

ap_log_rerror(APLOG_MARK, APLOG_NOERRNO|APLOG_ERR, 0, r,

"File does not exist: %s", r->filename);

return HTTP_NOT_FOUND;

}

}

if (script == NULL)

return DECLINED;

ap_internal_redirect_handler(apr_pstrcat(r->pool, script, ap_escape_uri(r->pool,

r->uri), r->args ? "?" : NULL, r->args, NULL), r);

return OK;

}

static void register_hooks(void)

{

ap_hook_handler(action_handler,NULL,NULL,AP_HOOK_LAST);

}

module action_module =

{

STANDARD20_MODULE_STUFF,

create_action_dir_config, /* dir config creater */

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

NULL, /* server config */

NULL, /* merge server config */

action_cmds, /* command apr_table_t */

register_hooks /* register hooks */

};