Blame standalone/api.c

Packit 284210
/*
Packit 284210
* ModSecurity for Apache 2.x, http://www.modsecurity.org/
Packit 284210
* Copyright (c) 2004-2013 Trustwave Holdings, Inc. (http://www.trustwave.com/)
Packit 284210
*
Packit 284210
* You may not use this file except in compliance with
Packit 284210
* the License.  You may obtain a copy of the License at
Packit 284210
*
Packit 284210
*     http://www.apache.org/licenses/LICENSE-2.0
Packit 284210
*
Packit 284210
* If any of the files related to licensing are missing or if you have any
Packit 284210
* other questions related to licensing please contact Trustwave Holdings, Inc.
Packit 284210
* directly using the email address security@modsecurity.org.
Packit 284210
*/
Packit 284210
Packit 284210
#include <limits.h>
Packit 284210
#include <stdio.h>
Packit 284210
Packit 284210
#include "http_core.h"
Packit 284210
#include "http_request.h"
Packit 284210
Packit 284210
#include "modsecurity.h"
Packit 284210
#include "apache2.h"
Packit 284210
#include "http_main.h"
Packit 284210
#include "http_connection.h"
Packit 284210
Packit 284210
#include "apr_optional.h"
Packit 284210
#include "mod_log_config.h"
Packit 284210
Packit 284210
#include "msc_logging.h"
Packit 284210
#include "msc_util.h"
Packit 284210
Packit 284210
#include "ap_mpm.h"
Packit 284210
#include "scoreboard.h"
Packit 284210
Packit 284210
#include "apr_version.h"
Packit 284210
Packit 284210
#include "apr_lib.h"
Packit 284210
#include "ap_config.h"
Packit 284210
#include "http_config.h"
Packit 284210
Packit 284210
#include "api.h"
Packit 284210
Packit 284210
#ifdef WIN32
Packit 284210
#include "msc_status_engine.h"
Packit 284210
#endif
Packit 284210
Packit 284210
extern void *modsecLogObj;
Packit 284210
extern void (*modsecLogHook)(void *obj, int level, char *str);
Packit 284210
extern int (*modsecDropAction)(request_rec *r);
Packit 284210
apr_status_t (*modsecReadBody)(request_rec *r, char *buf, unsigned int length, unsigned int *readcnt, int *is_eos);
Packit 284210
apr_status_t (*modsecReadResponse)(request_rec *r, char *buf, unsigned int length, unsigned int *readcnt, int *is_eos);
Packit 284210
apr_status_t (*modsecWriteBody)(request_rec *r, char *buf, unsigned int length);
Packit 284210
apr_status_t (*modsecWriteResponse)(request_rec *r, char *buf, unsigned int length);
Packit 284210
Packit 284210
extern const char *process_command_config(server_rec *s,
Packit 284210
                                          void *mconfig,
Packit 284210
                                          apr_pool_t *p,
Packit 284210
                                          apr_pool_t *ptemp,
Packit 284210
                                          const char *filename);
Packit 284210
Packit 284210
#define DECLARE_EXTERNAL_HOOK(ns,link,ret,name,args) \
Packit 284210
extern ns##_HOOK_##name##_t *hookfn_##name;
Packit 284210
Packit 284210
#define DECLARE_HOOK(ret,name,args) \
Packit 284210
    DECLARE_EXTERNAL_HOOK(ap,AP,ret,name,args)
Packit 284210
Packit 284210
DECLARE_HOOK(int,pre_config,(apr_pool_t *pconf,apr_pool_t *plog, apr_pool_t *ptemp))
Packit 284210
DECLARE_HOOK(int,post_config,(apr_pool_t *pconf,apr_pool_t *plog, apr_pool_t *ptemp,server_rec *s))
Packit 284210
DECLARE_HOOK(void,child_init,(apr_pool_t *pchild, server_rec *s))
Packit 284210
DECLARE_HOOK(int,process_connection,(conn_rec *c))
Packit 284210
DECLARE_HOOK(int,post_read_request,(request_rec *r))
Packit 284210
DECLARE_HOOK(int,fixups,(request_rec *r))
Packit 284210
DECLARE_HOOK(void, error_log, (const char *file, int line, int level,
Packit 284210
                       apr_status_t status, const server_rec *s,
Packit 284210
                       const request_rec *r, apr_pool_t *pool,
Packit 284210
                       const char *errstr))
Packit 284210
DECLARE_HOOK(int,log_transaction,(request_rec *r))
Packit 284210
DECLARE_HOOK(void,insert_filter,(request_rec *r))
Packit 284210
DECLARE_HOOK(void,insert_error_filter,(request_rec *r))
Packit 284210
Packit 284210
char *sa_name = "standalone";
Packit 284210
const char *sa_name_argv[] = { "standalone", NULL };
Packit 284210
server_rec *server;
Packit 284210
apr_pool_t *pool = NULL;
Packit 284210
Packit 284210
apr_status_t ap_http_in_filter(ap_filter_t *f, apr_bucket_brigade *b,
Packit 284210
                            ap_input_mode_t mode, apr_read_type_e block,
Packit 284210
                            apr_off_t readbytes);
Packit 284210
apr_status_t ap_http_out_filter(ap_filter_t *f, apr_bucket_brigade *b);
Packit 284210
Packit 284210
server_rec *modsecInit()    {
Packit 284210
    apr_initialize();
Packit 284210
Packit 284210
    apr_pool_create(&pool, NULL);
Packit 284210
Packit 284210
    apr_hook_global_pool = pool;
Packit 284210
Packit 284210
    server = apr_palloc(pool, sizeof(server_rec));
Packit 284210
Packit 284210
    server->addrs = apr_palloc(pool, sizeof(server_addr_rec));
Packit 284210
    server->addrs->host_addr = apr_palloc(pool, sizeof(apr_sockaddr_t));
Packit 284210
    server->addrs->host_addr->addr_str_len = 16;
Packit 284210
    server->addrs->host_addr->family = AF_INET;
Packit 284210
    server->addrs->host_addr->hostname = sa_name;
Packit 284210
#ifdef WIN32
Packit 284210
    server->addrs->host_addr->ipaddr_len = sizeof(IN_ADDR);
Packit 284210
#else
Packit 284210
    server->addrs->host_addr->ipaddr_len = sizeof(struct in_addr);
Packit 284210
#endif
Packit 284210
    server->addrs->host_addr->ipaddr_ptr = &server->addrs->host_addr->sa.sin.sin_addr;
Packit 284210
    server->addrs->host_addr->pool = pool;
Packit 284210
    server->addrs->host_addr->port = 80;
Packit 284210
#ifdef WIN32
Packit 284210
    server->addrs->host_addr->sa.sin.sin_addr.S_un.S_addr = 0x0100007f;
Packit 284210
#else
Packit 284210
    server->addrs->host_addr->sa.sin.sin_addr.s_addr = 0x0100007f;
Packit 284210
#endif
Packit 284210
    server->addrs->host_addr->sa.sin.sin_family = AF_INET;
Packit 284210
    server->addrs->host_addr->sa.sin.sin_port = 80;
Packit 284210
    server->addrs->host_addr->salen = sizeof(server->addrs->host_addr->sa);
Packit 284210
    server->addrs->host_addr->servname = sa_name;
Packit 284210
    server->addrs->host_port = 80;
Packit 284210
    server->error_fname = "error.log";
Packit 284210
    server->error_log = NULL;
Packit 284210
    server->limit_req_fields = 1024;
Packit 284210
    server->limit_req_fieldsize = 1024;
Packit 284210
    server->limit_req_line = 1024;
Packit 284210
#if AP_SERVER_MAJORVERSION_NUMBER > 1 && AP_SERVER_MINORVERSION_NUMBER < 3
Packit 284210
    server->loglevel = APLOG_DEBUG;
Packit 284210
#endif
Packit 284210
    server->lookup_defaults = NULL;
Packit 284210
    server->module_config = NULL;
Packit 284210
    server->names = NULL;
Packit 284210
#ifdef WIN32
Packit 284210
    server->path = "c:\\inetpub\\wwwroot";
Packit 284210
#else
Packit 284210
    server->path = "/var/www";
Packit 284210
#endif
Packit 284210
    server->pathlen = strlen(server->path);
Packit 284210
    server->port = 80;
Packit 284210
    server->process = apr_palloc(pool, sizeof(process_rec));
Packit 284210
    server->process->argc = 1;
Packit 284210
    server->process->argv = sa_name_argv;
Packit 284210
    server->process->pconf = pool;
Packit 284210
    server->process->pool = pool;
Packit 284210
    server->process->short_name = sa_name;
Packit 284210
    server->server_admin = sa_name;
Packit 284210
    server->server_hostname = sa_name;
Packit 284210
    server->server_scheme = "";
Packit 284210
    server->timeout = 60 * 1000000;// 60 seconds
Packit 284210
    server->wild_names = NULL;
Packit 284210
    server->is_virtual = 0;
Packit 284210
Packit 284210
    ap_server_config_defines = apr_array_make(pool, 1, sizeof(char *));
Packit 284210
Packit 284210
    // here we should add scoreboard handling for multiple processes and threads
Packit 284210
    //
Packit 284210
    ap_scoreboard_image = (scoreboard *)apr_palloc(pool, sizeof(scoreboard));
Packit 284210
Packit 284210
    memset(ap_scoreboard_image, 0, sizeof(scoreboard));
Packit 284210
Packit 284210
    // ----------
Packit 284210
Packit 284210
    security2_module.module_index = 0;
Packit 284210
Packit 284210
    security2_module.register_hooks(pool);
Packit 284210
Packit 284210
    ap_register_input_filter("HTTP_IN", ap_http_in_filter, NULL, AP_FTYPE_RESOURCE);
Packit 284210
    ap_register_output_filter("HTTP_OUT", ap_http_out_filter, NULL, AP_FTYPE_CONTENT_SET);
Packit 284210
Packit 284210
    return server;
Packit 284210
}
Packit 284210
Packit 284210
apr_status_t ap_http_in_filter(ap_filter_t *f, apr_bucket_brigade *bb_out,
Packit 284210
        ap_input_mode_t mode, apr_read_type_e block,
Packit 284210
        apr_off_t readbytes)    {
Packit 284210
    char *tmp = NULL;
Packit 284210
    apr_bucket *e = NULL;
Packit 284210
    unsigned int readcnt = 0;
Packit 284210
    int is_eos = 0;
Packit 284210
    apr_bucket_brigade *bb_in;
Packit 284210
    apr_bucket *after;
Packit 284210
    apr_status_t rv;
Packit 284210
Packit 284210
    bb_in = modsecGetBodyBrigade(f->r);
Packit 284210
Packit 284210
    /* use request brigade */
Packit 284210
    if (bb_in != NULL) {
Packit 284210
        if (!APR_BUCKET_IS_EOS(APR_BRIGADE_LAST(bb_in))) {
Packit 284210
            e = apr_bucket_eos_create(f->c->bucket_alloc);
Packit 284210
            APR_BRIGADE_INSERT_TAIL(bb_in, e);
Packit 284210
        }
Packit 284210
Packit 284210
        rv = apr_brigade_partition(bb_in, readbytes, &after);
Packit 284210
        if (rv != APR_SUCCESS && rv != APR_INCOMPLETE) {
Packit 284210
            return rv;
Packit 284210
        }
Packit 284210
        
Packit 284210
        for (e = APR_BRIGADE_FIRST(bb_in); e != after; e = APR_BRIGADE_FIRST(bb_in)) {
Packit 284210
            APR_BUCKET_REMOVE(e);
Packit 284210
            APR_BRIGADE_INSERT_TAIL(bb_out, e);
Packit 284210
        }
Packit 284210
Packit 284210
        return APR_SUCCESS;
Packit 284210
    }
Packit 284210
Packit 284210
    /* call the callback */
Packit 284210
    if(modsecReadBody != NULL) {
Packit 284210
Packit 284210
        tmp = (char *)apr_palloc(f->r->pool, readbytes);
Packit 284210
        modsecReadBody(f->r, tmp, readbytes, &readcnt, &is_eos);
Packit 284210
Packit 284210
        e = apr_bucket_pool_create(tmp, readcnt, f->r->pool, f->c->bucket_alloc);
Packit 284210
        APR_BRIGADE_INSERT_TAIL(bb_out, e);
Packit 284210
Packit 284210
        if(is_eos)  {
Packit 284210
            e = apr_bucket_eos_create(f->c->bucket_alloc);
Packit 284210
            APR_BRIGADE_INSERT_TAIL(bb_out, e);
Packit 284210
        }
Packit 284210
        return APR_SUCCESS;
Packit 284210
    }
Packit 284210
Packit 284210
    /* cannot read request body */
Packit 284210
    e = apr_bucket_eos_create(f->c->bucket_alloc);
Packit 284210
    APR_BRIGADE_INSERT_TAIL(bb_out, e);
Packit 284210
Packit 284210
    return APR_SUCCESS;
Packit 284210
}
Packit 284210
Packit 284210
apr_status_t ap_http_out_filter(ap_filter_t *f, apr_bucket_brigade *b)  {
Packit 284210
    apr_bucket_brigade *bb_out = (apr_bucket_brigade *)f->ctx;
Packit 284210
Packit 284210
    APR_BRIGADE_CONCAT(bb_out, b);
Packit 284210
    return APR_SUCCESS;
Packit 284210
}
Packit 284210
Packit 284210
void modsecTerminate()  {
Packit 284210
    apr_pool_destroy(pool);
Packit 284210
    pool = NULL;
Packit 284210
    apr_terminate();
Packit 284210
}
Packit 284210
Packit 284210
void modsecStartConfig()    {
Packit 284210
    apr_pool_t *ptemp = NULL;
Packit 284210
Packit 284210
    apr_pool_create(&ptemp, pool);
Packit 284210
Packit 284210
    hookfn_pre_config(pool, pool, ptemp);
Packit 284210
Packit 284210
    apr_pool_destroy(ptemp);
Packit 284210
}
Packit 284210
Packit 284210
directory_config *modsecGetDefaultConfig()  {
Packit 284210
    return (directory_config *)security2_module.create_dir_config(pool, NULL);
Packit 284210
}
Packit 284210
Packit 284210
const char *modsecProcessConfig(directory_config *config, const char *file, const char *dir)  {
Packit 284210
    apr_pool_t *ptemp = NULL;
Packit 284210
    const char *err;
Packit 284210
	apr_status_t status;
Packit 284210
	const char *rootpath, *incpath;
Packit 284210
Packit 284210
	if(dir == NULL || strlen(dir) == 0)
Packit 284210
#ifdef	WIN32
Packit 284210
		dir = "\\";
Packit 284210
#else
Packit 284210
		dir = "/";
Packit 284210
#endif
Packit 284210
Packit 284210
	incpath = file;
Packit 284210
Packit 284210
	/* locate the start of the directories proper */
Packit 284210
	status = apr_filepath_root(&rootpath, &incpath, APR_FILEPATH_TRUENAME | APR_FILEPATH_NATIVE, config->mp);
Packit 284210
Packit 284210
	/* we allow APR_SUCCESS and APR_EINCOMPLETE */
Packit 284210
	if (APR_ERELATIVE == status) {
Packit 284210
		int li = strlen(dir) - 1;
Packit 284210
Packit 284210
		if(dir[li] != '/' && dir[li] != '\\')
Packit 284210
#ifdef	WIN32
Packit 284210
			file = apr_pstrcat(config->mp, dir, "\\", file, NULL);
Packit 284210
#else
Packit 284210
			file = apr_pstrcat(config->mp, dir, "/", file, NULL);
Packit 284210
#endif
Packit 284210
		else
Packit 284210
			file = apr_pstrcat(config->mp, dir, file, NULL);
Packit 284210
	}
Packit 284210
	else if (APR_EBADPATH == status) {
Packit 284210
		return apr_pstrcat(config->mp, "Config file has a bad path, ", file, NULL);
Packit 284210
	}
Packit 284210
Packit 284210
	apr_pool_create(&ptemp, config->mp);
Packit 284210
Packit 284210
    err = process_command_config(server, config, config->mp, ptemp, file);
Packit 284210
Packit 284210
    apr_pool_destroy(ptemp);
Packit 284210
Packit 284210
    return err;
Packit 284210
}
Packit 284210
Packit 284210
void modsecFinalizeConfig() {
Packit 284210
    apr_pool_t *ptemp = NULL;
Packit 284210
Packit 284210
    apr_pool_create(&ptemp, pool);
Packit 284210
Packit 284210
    hookfn_post_config(pool, pool, ptemp, server);
Packit 284210
    hookfn_post_config(pool, pool, ptemp, server);
Packit 284210
Packit 284210
    apr_pool_destroy(ptemp);
Packit 284210
}
Packit 284210
Packit 284210
void modsecInitProcess()    {
Packit 284210
    hookfn_child_init(pool, server);
Packit 284210
}
Packit 284210
Packit 284210
conn_rec *modsecNewConnection() {
Packit 284210
    conn_rec *c;
Packit 284210
    apr_pool_t *pc = NULL;
Packit 284210
Packit 284210
    apr_pool_create(&pc, pool);
Packit 284210
Packit 284210
    c = apr_pcalloc(pc, sizeof(conn_rec));
Packit 284210
Packit 284210
    c->base_server = server;
Packit 284210
    c->id = 1;
Packit 284210
    c->local_addr = server->addrs->host_addr;
Packit 284210
    c->local_host = sa_name;
Packit 284210
    c->local_ip = "127.0.0.1";
Packit 284210
    c->pool = pc;
Packit 284210
    c->remote_host = sa_name;
Packit 284210
#if AP_SERVER_MAJORVERSION_NUMBER > 1 && AP_SERVER_MINORVERSION_NUMBER < 3
Packit 284210
    c->remote_ip = "127.0.0.1";
Packit 284210
    c->remote_addr = server->addrs->host_addr;
Packit 284210
#else
Packit 284210
    c->client_ip = "127.0.0.1";
Packit 284210
    c->client_addr = server->addrs->host_addr;
Packit 284210
#endif
Packit 284210
    c->input_filters = NULL;
Packit 284210
    c->output_filters = NULL;
Packit 284210
    c->bucket_alloc = apr_bucket_alloc_create(pc);
Packit 284210
Packit 284210
    return c;
Packit 284210
}
Packit 284210
Packit 284210
void modsecProcessConnection(conn_rec *c)   {
Packit 284210
    hookfn_process_connection(c);
Packit 284210
}
Packit 284210
Packit 284210
request_rec *modsecNewRequest(conn_rec *connection, directory_config *config)   {
Packit 284210
    request_rec *r;
Packit 284210
    apr_pool_t *pr = NULL;
Packit 284210
Packit 284210
    apr_pool_create(&pr, connection->pool);
Packit 284210
Packit 284210
    r = apr_pcalloc(pr, sizeof(request_rec));
Packit 284210
Packit 284210
    r->connection = connection;
Packit 284210
    r->server = server;
Packit 284210
    r->pool = pr;
Packit 284210
    r->main = NULL;
Packit 284210
    r->next = NULL;
Packit 284210
    r->notes = apr_table_make(pr, 10);
Packit 284210
    r->per_dir_config = apr_palloc(pr, sizeof(void *));
Packit 284210
    ((void **)r->per_dir_config)[0] = config;
Packit 284210
    r->prev = NULL;
Packit 284210
    r->subprocess_env = apr_table_make(pr, 10);
Packit 284210
    apr_table_setn(r->subprocess_env, "UNIQUE_ID", "unique_id");
Packit 284210
    r->user = NULL;
Packit 284210
Packit 284210
    r->headers_in = apr_table_make(pr, 10);
Packit 284210
    r->headers_out = apr_table_make(pr, 10);
Packit 284210
    r->err_headers_out = apr_table_make(pr, 10);
Packit 284210
    //apr_table_setn(r->headers_in, "Host", "www.google.com");
Packit 284210
    //apr_table_setn(r->headers_in, "", "");
Packit 284210
Packit 284210
    r->the_request = "GET /../../index.html HTTP/1.1";
Packit 284210
    r->method = "GET";
Packit 284210
    r->method_number = M_GET;
Packit 284210
    r->protocol = "HTTP/1.1";
Packit 284210
    r->uri = "http://www.google.com/../../index.html";
Packit 284210
    r->args = "";
Packit 284210
    r->filename = "/../../index.html";
Packit 284210
    r->handler = "IIS";
Packit 284210
Packit 284210
    r->parsed_uri.scheme = "http";
Packit 284210
    r->parsed_uri.path = "/../../index.html";
Packit 284210
    r->parsed_uri.hostname = "www.google.com";
Packit 284210
    r->parsed_uri.is_initialized = 1;
Packit 284210
    r->parsed_uri.port = 1234;
Packit 284210
    r->parsed_uri.port_str = "1234";
Packit 284210
    r->parsed_uri.query = "";
Packit 284210
    r->parsed_uri.dns_looked_up = 0;
Packit 284210
    r->parsed_uri.dns_resolved = 0;
Packit 284210
    r->parsed_uri.password = NULL;
Packit 284210
    r->parsed_uri.user = NULL;
Packit 284210
    r->parsed_uri.fragment = "";
Packit 284210
Packit 284210
    r->input_filters = NULL;
Packit 284210
    r->output_filters = NULL;
Packit 284210
Packit 284210
    return r;
Packit 284210
}
Packit 284210
Packit 284210
static modsec_rec *retrieve_msr(request_rec *r) {
Packit 284210
    modsec_rec *msr = NULL;
Packit 284210
    request_rec *rx = NULL;
Packit 284210
Packit 284210
    /* Look in the current request first. */
Packit 284210
    msr = (modsec_rec *)apr_table_get(r->notes, NOTE_MSR);
Packit 284210
    if (msr != NULL) {
Packit 284210
        msr->r = r;
Packit 284210
        return msr;
Packit 284210
    }
Packit 284210
Packit 284210
    /* If this is a subrequest then look in the main request. */
Packit 284210
    if (r->main != NULL) {
Packit 284210
        msr = (modsec_rec *)apr_table_get(r->main->notes, NOTE_MSR);
Packit 284210
        if (msr != NULL) {
Packit 284210
            msr->r = r;
Packit 284210
            return msr;
Packit 284210
        }
Packit 284210
    }
Packit 284210
Packit 284210
    /* If the request was redirected then look in the previous requests. */
Packit 284210
    rx = r->prev;
Packit 284210
    while(rx != NULL) {
Packit 284210
        msr = (modsec_rec *)apr_table_get(rx->notes, NOTE_MSR);
Packit 284210
        if (msr != NULL) {
Packit 284210
            msr->r = r;
Packit 284210
            return msr;
Packit 284210
        }
Packit 284210
        rx = rx->prev;
Packit 284210
    }
Packit 284210
Packit 284210
    return NULL;
Packit 284210
}
Packit 284210
Packit 284210
int modsecProcessRequestHeaders(request_rec *r) {
Packit 284210
    return hookfn_post_read_request(r);
Packit 284210
}
Packit 284210
Packit 284210
int modsecProcessRequestBody(request_rec *r) {
Packit 284210
    int status = DECLINED;
Packit 284210
    modsec_rec *msr = NULL;
Packit 284210
Packit 284210
    ap_filter_t *f = ap_add_input_filter("HTTP_IN", NULL, r, r->connection);
Packit 284210
    apr_bucket_brigade* bb_out;
Packit 284210
Packit 284210
    status = hookfn_fixups(r);
Packit 284210
Packit 284210
    ap_remove_input_filter(f);
Packit 284210
Packit 284210
	if (status != DECLINED) {
Packit 284210
		return status;
Packit 284210
	}
Packit 284210
Packit 284210
    hookfn_insert_filter(r);
Packit 284210
Packit 284210
    /* Find the transaction context first. */
Packit 284210
    msr = retrieve_msr(r);
Packit 284210
Packit 284210
    if (msr == NULL)
Packit 284210
        return status;
Packit 284210
Packit 284210
    bb_out = modsecGetBodyBrigade(r);
Packit 284210
    if (bb_out) {
Packit 284210
        (void) apr_brigade_cleanup(bb_out);
Packit 284210
        status = ap_get_brigade(r->input_filters, bb_out, AP_MODE_READBYTES, APR_BLOCK_READ, -1);
Packit 284210
        if (status == APR_SUCCESS) {
Packit 284210
            return DECLINED;
Packit 284210
        }
Packit 284210
        return status;
Packit 284210
    }
Packit 284210
Packit 284210
    if(msr->stream_input_data != NULL && modsecWriteBody != NULL)
Packit 284210
    {
Packit 284210
        // target is responsible for copying the data into correctly managed buffer
Packit 284210
        //
Packit 284210
        modsecWriteBody(r, msr->stream_input_data, msr->stream_input_length);
Packit 284210
Packit 284210
        free(msr->stream_input_data);
Packit 284210
Packit 284210
        msr->stream_input_data = NULL;
Packit 284210
    }
Packit 284210
Packit 284210
    // leftover code possibly for future use
Packit 284210
    //
Packit 284210
    //if(r->input_filters != NULL && r->input_filters->frec->filter_init_func != NULL)
Packit 284210
    //r->input_filters->frec->filter_init_func(r->input_filters);
Packit 284210
    //if(r->input_filters != NULL && r->input_filters->frec->filter_func.in_func != NULL)
Packit 284210
    //r->input_filters->frec->filter_func.in_func(r->input_filters, NULL, 0, 0, 0);
Packit 284210
Packit 284210
    return status;
Packit 284210
}
Packit 284210
Packit 284210
void modsecSetConfigForIISRequestBody(request_rec *r)
Packit 284210
{
Packit 284210
    modsec_rec *msr = retrieve_msr(r);
Packit 284210
Packit 284210
    if(msr == NULL || msr->txcfg == NULL)
Packit 284210
        return;
Packit 284210
Packit 284210
    if(msr->txcfg->reqbody_access)
Packit 284210
        msr->txcfg->stream_inbody_inspection = 1;
Packit 284210
}
Packit 284210
Packit 284210
int modsecContextState(request_rec *r)
Packit 284210
{
Packit 284210
    modsec_rec *msr = retrieve_msr(r);
Packit 284210
Packit 284210
    if(msr == NULL || msr->txcfg == NULL)
Packit 284210
        return NOT_SET;
Packit 284210
Packit 284210
    return msr->txcfg->is_enabled;
Packit 284210
}
Packit 284210
Packit 284210
int modsecIsRequestBodyAccessEnabled(request_rec *r)
Packit 284210
{
Packit 284210
    modsec_rec *msr = retrieve_msr(r);
Packit 284210
Packit 284210
    if(msr == NULL || msr->txcfg == NULL)
Packit 284210
        return 0;
Packit 284210
Packit 284210
    return msr->txcfg->reqbody_access;
Packit 284210
}
Packit 284210
Packit 284210
int modsecIsResponseBodyAccessEnabled(request_rec *r)
Packit 284210
{
Packit 284210
    modsec_rec *msr = retrieve_msr(r);
Packit 284210
Packit 284210
    if(msr == NULL || msr->txcfg == NULL)
Packit 284210
        return 0;
Packit 284210
Packit 284210
    return msr->txcfg->resbody_access;
Packit 284210
}
Packit 284210
Packit 284210
int modsecProcessResponse(request_rec *r)   {
Packit 284210
    int status;
Packit 284210
    modsec_rec *msr;
Packit 284210
    apr_bucket *e;
Packit 284210
    ap_filter_t *f;
Packit 284210
    apr_bucket_brigade *bb_in, *bb_out, *bb;
Packit 284210
Packit 284210
    if(r->output_filters == NULL) {
Packit 284210
        return DECLINED;
Packit 284210
    }
Packit 284210
Packit 284210
    msr = (modsec_rec *)r->output_filters->ctx;
Packit 284210
    if (msr == NULL) {
Packit 284210
        ap_log_error(APLOG_MARK, APLOG_ERR | APLOG_NOERRNO, 0, r->server,
Packit 284210
                "ModSecurity: Internal Error: msr is null in output filter.");
Packit 284210
        ap_remove_output_filter(r->output_filters);
Packit 284210
        return APR_EGENERAL;
Packit 284210
    }
Packit 284210
Packit 284210
    msr->r = r;
Packit 284210
Packit 284210
    /* create input response brigade */
Packit 284210
    bb_in = apr_brigade_create(msr->mp, r->connection->bucket_alloc);
Packit 284210
Packit 284210
    if (bb_in == NULL) {
Packit 284210
        msr_log(msr, 1, "Process response: Failed to create brigade.");
Packit 284210
        return APR_EGENERAL;
Packit 284210
    }
Packit 284210
Packit 284210
    /* get input response brigade */
Packit 284210
    bb = modsecGetResponseBrigade(r);
Packit 284210
    if (bb != NULL) {
Packit 284210
        APR_BRIGADE_CONCAT(bb_in, bb);
Packit 284210
        if (!APR_BUCKET_IS_EOS(APR_BRIGADE_LAST(bb_in))) {
Packit 284210
            e = apr_bucket_eos_create(bb_in->bucket_alloc);
Packit 284210
            APR_BRIGADE_INSERT_TAIL(bb_in, e);
Packit 284210
        }
Packit 284210
    } else if (modsecReadResponse != NULL) {
Packit 284210
        unsigned int readcnt = 0;
Packit 284210
        int is_eos = 0;
Packit 284210
        char buf[8192];
Packit 284210
        while(!is_eos)  {
Packit 284210
            modsecReadResponse(r, buf, 8192, &readcnt, &is_eos);
Packit 284210
Packit 284210
            if(readcnt > 0) {
Packit 284210
                char *tmp = (char *)apr_palloc(r->pool, readcnt);
Packit 284210
                memcpy(tmp, buf, readcnt);
Packit 284210
                e = apr_bucket_pool_create(tmp, readcnt, r->pool, r->connection->bucket_alloc);
Packit 284210
                APR_BRIGADE_INSERT_TAIL(bb_in, e);
Packit 284210
            }
Packit 284210
        }
Packit 284210
Packit 284210
        e = apr_bucket_eos_create(r->connection->bucket_alloc);
Packit 284210
        APR_BRIGADE_INSERT_TAIL(bb_in, e);
Packit 284210
    } else {
Packit 284210
        /* cannot read response body process header only */
Packit 284210
Packit 284210
        e = apr_bucket_eos_create(r->connection->bucket_alloc);
Packit 284210
        APR_BRIGADE_INSERT_TAIL(bb_in, e);
Packit 284210
    }
Packit 284210
Packit 284210
    bb_out = bb ? bb : apr_brigade_create(msr->mp, r->connection->bucket_alloc);
Packit 284210
Packit 284210
    if (bb_out == NULL) {
Packit 284210
        msr_log(msr, 1, "Process response: Failed to create brigade.");
Packit 284210
        return APR_EGENERAL;
Packit 284210
    }
Packit 284210
Packit 284210
    /* concat output bucket to bb_out */
Packit 284210
    f = ap_add_output_filter("HTTP_OUT", bb_out, r, r->connection);
Packit 284210
    status = ap_pass_brigade(r->output_filters, bb_in);
Packit 284210
    ap_remove_output_filter(f);
Packit 284210
Packit 284210
    if (status == APR_EGENERAL) {
Packit 284210
        /* retrive response status from bb_out */
Packit 284210
        for(e = APR_BRIGADE_FIRST(bb_out);
Packit 284210
                e != APR_BRIGADE_SENTINEL(bb_out);
Packit 284210
                e = APR_BUCKET_NEXT(e)) {
Packit 284210
            if (AP_BUCKET_IS_ERROR(e)) {
Packit 284210
                return ((ap_bucket_error*) e->data)->status;
Packit 284210
            }
Packit 284210
        }
Packit 284210
        return APR_EGENERAL;
Packit 284210
    }
Packit 284210
Packit 284210
    if (status != DECLINED) {
Packit 284210
        return status;
Packit 284210
    }
Packit 284210
Packit 284210
    /* copy bb_out */
Packit 284210
    // is there a way to tell whether the response body was modified or not?
Packit 284210
    if (modsecWriteResponse != NULL
Packit 284210
            && (msr->txcfg->content_injection_enabled || msr->content_prepend_len != 0 || msr->content_append_len != 0)
Packit 284210
            && msr->txcfg->resbody_access)   {
Packit 284210
Packit 284210
        char *data = NULL;
Packit 284210
        apr_size_t length;
Packit 284210
Packit 284210
        status = apr_brigade_pflatten(msr->of_brigade, &data, &length, msr->mp);
Packit 284210
Packit 284210
        if (status != APR_SUCCESS) {
Packit 284210
            msr_log(msr, 1, "Output filter: Failed to flatten brigade (%d): %s", status,
Packit 284210
                    get_apr_error(msr->mp, status));
Packit 284210
            return APR_EGENERAL;
Packit 284210
        }
Packit 284210
Packit 284210
        if ( modsecWriteResponse(msr->r, data, msr->stream_output_length) != APR_SUCCESS) {
Packit 284210
            return APR_EGENERAL;
Packit 284210
        }
Packit 284210
    }
Packit 284210
    
Packit 284210
    return DECLINED;
Packit 284210
}
Packit 284210
Packit 284210
int modsecFinishRequest(request_rec *r) {
Packit 284210
    // run output filter
Packit 284210
    //if(r->output_filters != NULL && r->output_filters->frec->filter_init_func != NULL)
Packit 284210
    //r->output_filters->frec->filter_init_func(r->output_filters);
Packit 284210
Packit 284210
    hookfn_log_transaction(r);
Packit 284210
Packit 284210
    // make sure you cleanup before calling apr_terminate()
Packit 284210
    // otherwise double-free might occur, because of the request body pool cleanup function
Packit 284210
    //
Packit 284210
    apr_pool_destroy(r->pool);
Packit 284210
Packit 284210
    return DECLINED;
Packit 284210
}
Packit 284210
Packit 284210
// destroy only the connection pool
Packit 284210
int modsecFinishConnection(conn_rec *c)
Packit 284210
{
Packit 284210
    apr_pool_destroy(c->pool);
Packit 284210
Packit 284210
    return 0;
Packit 284210
}
Packit 284210
Packit 284210
Packit 284210
void modsecSetLogHook(void *obj, void (*hook)(void *obj, int level, char *str)) {
Packit 284210
    modsecLogObj = obj;
Packit 284210
    modsecLogHook = hook;
Packit 284210
}
Packit 284210
Packit 284210
void modsecSetReadBody(apr_status_t (*func)(request_rec *r, char *buf, unsigned int length, unsigned int *readcnt, int *is_eos))    {
Packit 284210
    modsecReadBody = func;
Packit 284210
}
Packit 284210
Packit 284210
void modsecSetReadResponse(apr_status_t (*func)(request_rec *r, char *buf, unsigned int length, unsigned int *readcnt, int *is_eos))    {
Packit 284210
    modsecReadResponse = func;
Packit 284210
}
Packit 284210
Packit 284210
void modsecSetWriteBody(apr_status_t (*func)(request_rec *r, char *buf, unsigned int length))   {
Packit 284210
    modsecWriteBody = func;
Packit 284210
}
Packit 284210
Packit 284210
void modsecSetWriteResponse(apr_status_t (*func)(request_rec *r, char *buf, unsigned int length))   {
Packit 284210
    modsecWriteResponse = func;
Packit 284210
}
Packit 284210
Packit 284210
void modsecSetDropAction(int (*func)(request_rec *r)) {
Packit 284210
    modsecDropAction = func;
Packit 284210
}
Packit 284210
Packit 284210
/*
Packit 284210
 * Case SecServerSignature was used, this function returns the banner that
Packit 284210
 * should be used, otherwise it returns NULL.
Packit 284210
 */
Packit 284210
const char *modsecIsServerSignatureAvailale(void) {
Packit 284210
    return new_server_signature;
Packit 284210
}
Packit 284210
Packit 284210
#ifdef VERSION_IIS
Packit 284210
void modsecStatusEngineCall()
Packit 284210
{
Packit 284210
    if (status_engine_state != STATUS_ENGINE_DISABLED) {
Packit 284210
        msc_status_engine_call();
Packit 284210
    }
Packit 284210
    else {
Packit 284210
        ap_log_error(APLOG_MARK, APLOG_NOTICE, 0, NULL,
Packit 284210
            "Status engine is currently disabled, enable it by set " \
Packit 284210
            "SecStatusEngine to On.\n");
Packit 284210
    }
Packit 284210
}
Packit 284210
Packit 284210
void modsecReportRemoteLoadedRules()
Packit 284210
{
Packit 284210
#ifdef WITH_REMOTE_RULES
Packit 284210
    if (remote_rules_server != NULL)
Packit 284210
    {
Packit 284210
        if (remote_rules_server->amount_of_rules == 1)
Packit 284210
        {
Packit 284210
            ap_log_error(APLOG_MARK, APLOG_NOTICE, 0, NULL,
Packit 284210
                "ModSecurity: Loaded %d rule from: '%s'.",
Packit 284210
                remote_rules_server->amount_of_rules,
Packit 284210
                remote_rules_server->uri);
Packit 284210
        }
Packit 284210
        else
Packit 284210
        {
Packit 284210
            ap_log_error(APLOG_MARK, APLOG_NOTICE, 0, NULL,
Packit 284210
                "ModSecurity: Loaded %d rules from: '%s'.",
Packit 284210
                remote_rules_server->amount_of_rules,
Packit 284210
                remote_rules_server->uri);
Packit 284210
        }
Packit 284210
    }
Packit 284210
#endif
Packit 284210
    if (remote_rules_fail_message != NULL)
Packit 284210
    {
Packit 284210
        ap_log_error(APLOG_MARK, APLOG_NOTICE, 0, NULL, "ModSecurity: " \
Packit 284210
            "Problems loading external resources: %s",
Packit 284210
            remote_rules_fail_message);
Packit 284210
    }
Packit 284210
Packit 284210
}
Packit 284210
#endif