#include "ap_config.h"

#include "ap_mmn.h"

#include "httpd.h"

#include "http_config.h"

#include "http_connection.h"

#include "apr_buckets.h"

#include "util_filter.h"

module AP_MODULE_DECLARE_DATA echo_module;

typedef struct {

int bEnabled;

} EchoConfig;

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

{

EchoConfig *pConfig = apr_pcalloc(p, sizeof *pConfig);

pConfig->bEnabled = 0;

return pConfig;

}

static const char *echo_on(cmd_parms *cmd, void *dummy, int arg)

{

EchoConfig *pConfig = ap_get_module_config(cmd->server->module_config,

&echo_module);

pConfig->bEnabled = arg;

return NULL;

}

static int process_echo_connection(conn_rec *c)

{

apr_bucket_brigade *bb;

apr_bucket *b;

apr_status_t rv;

apr_off_t zero = 0;

EchoConfig *pConfig = ap_get_module_config(c->base_server->module_config,

&echo_module);

if (!pConfig->bEnabled) {

return DECLINED;

}

bb = apr_brigade_create(c->pool);

for ( ; ; ) {

/* Get a single line of input from the client */

if ((rv = ap_get_brigade(c->input_filters, bb,

AP_MODE_BLOCKING, &zero) != APR_SUCCESS ||

APR_BRIGADE_EMPTY(bb))) {

apr_brigade_destroy(bb);

break;

}

/* Make sure the data is flushed to the client */

b = apr_bucket_flush_create();

APR_BRIGADE_INSERT_TAIL(bb, b);

ap_pass_brigade(c->output_filters, bb);

}

return OK;

}

static const command_rec echo_cmds[] =

{

AP_INIT_FLAG("ProtocolEcho", echo_on, NULL, RSRC_CONF,

"Run an echo server on this host"),

{ NULL }

};

static void register_hooks(apr_pool_t *p)

{

ap_hook_process_connection(process_echo_connection, NULL, NULL,

APR_HOOK_MIDDLE);

}

module AP_MODULE_DECLARE_DATA echo_module = {

STANDARD20_MODULE_STUFF,

NULL, /* create per-directory config structure */

NULL, /* merge per-directory config structures */

create_echo_server_config, /* create per-server config structure */

NULL, /* merge per-server config structures */

echo_cmds, /* command apr_table_t */

register_hooks /* register hooks */

};