#include "apr_strings.h"
#include "apr_lib.h"
#include "apr_hash.h"
#include "apr_optional.h"
#define APR_WANT_STRFUNC
#include "apr_want.h"
#include "ap_config.h"
#include "mod_log_config.h"
#include "httpd.h"
#include "http_core.h"
#include "http_config.h"
#include "http_connection.h"
#include "http_protocol.h"
module AP_MODULE_DECLARE_DATA logio_module;
static const char logio_filter_name[] = "LOG_INPUT_OUTPUT";
typedef struct logio_config_t {
apr_off_t bytes_in;
apr_off_t bytes_out;
} logio_config_t;
static void ap_logio_add_bytes_out(conn_rec *c, apr_off_t bytes){
logio_config_t *cf = ap_get_module_config(c->conn_config, &logio_module);
cf->bytes_out += bytes;
}
static void ap_logio_add_bytes_in(conn_rec *c, apr_off_t bytes){
logio_config_t *cf = ap_get_module_config(c->conn_config, &logio_module);
cf->bytes_in += bytes;
}
static const char *log_bytes_in(request_rec *r, char *a)
{
logio_config_t *cf = ap_get_module_config(r->connection->conn_config,
&logio_module);
return apr_off_t_toa(r->pool, cf->bytes_in);
}
static const char *log_bytes_out(request_rec *r, char *a)
{
logio_config_t *cf = ap_get_module_config(r->connection->conn_config,
&logio_module);
return apr_off_t_toa(r->pool, cf->bytes_out);
}
static int logio_transaction(request_rec *r)
{
logio_config_t *cf = ap_get_module_config(r->connection->conn_config,
&logio_module);
cf->bytes_in = cf->bytes_out = 0;
return OK;
}
static apr_status_t logio_in_filter(ap_filter_t *f,
apr_bucket_brigade *bb,
ap_input_mode_t mode,
apr_read_type_e block,
apr_off_t readbytes) {
apr_off_t length;
apr_status_t status;
logio_config_t *cf = ap_get_module_config(f->c->conn_config, &logio_module);
status = ap_get_brigade(f->next, bb, mode, block, readbytes);
apr_brigade_length (bb, 0, &length);
if (length > 0)
cf->bytes_in += length;
return status;
}
static apr_status_t logio_out_filter(ap_filter_t *f,
apr_bucket_brigade *bb) {
apr_bucket *b = APR_BRIGADE_LAST(bb);
if (APR_BUCKET_IS_EOS(b)) {
APR_BUCKET_INSERT_BEFORE(b,
apr_bucket_flush_create(f->c->bucket_alloc));
}
return ap_pass_brigade(f->next, bb);
}
static int logio_pre_conn(conn_rec *c, void *csd) {
logio_config_t *cf = apr_pcalloc(c->pool, sizeof(*cf));
ap_set_module_config(c->conn_config, &logio_module, cf);
ap_add_input_filter(logio_filter_name, NULL, NULL, c);
ap_add_output_filter(logio_filter_name, NULL, NULL, c);
return OK;
}
static int logio_pre_config(apr_pool_t *p, apr_pool_t *plog, apr_pool_t *ptemp)
{
static APR_OPTIONAL_FN_TYPE(ap_register_log_handler) *log_pfn_register;
log_pfn_register = APR_RETRIEVE_OPTIONAL_FN(ap_register_log_handler);
if (log_pfn_register) {
log_pfn_register(p, "I", log_bytes_in, 0);
log_pfn_register(p, "O", log_bytes_out, 0);
}
return OK;
}
static void register_hooks(apr_pool_t *p)
{
static const char *pre[] = { "mod_log_config.c", NULL };
ap_hook_pre_connection(logio_pre_conn, NULL, NULL, APR_HOOK_MIDDLE);
ap_hook_pre_config(logio_pre_config, NULL, NULL, APR_HOOK_REALLY_FIRST);
ap_hook_log_transaction(logio_transaction, pre, NULL, APR_HOOK_MIDDLE);
ap_register_input_filter(logio_filter_name, logio_in_filter, NULL,
AP_FTYPE_NETWORK - 1);
ap_register_output_filter(logio_filter_name, logio_out_filter, NULL,
AP_FTYPE_NETWORK - 1);
APR_REGISTER_OPTIONAL_FN(ap_logio_add_bytes_out);
APR_REGISTER_OPTIONAL_FN(ap_logio_add_bytes_in);
}
module AP_MODULE_DECLARE_DATA logio_module =
{
STANDARD20_MODULE_STUFF,
NULL,
NULL,
NULL,
NULL,
NULL,
register_hooks
};