/* Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/*
 * http_script: keeps all script-related ramblings together.
 *
 * Compliant to cgi/1.1 spec
 *
 * Adapted by rst from original NCSA code by Rob McCool
 *
 * Apache adds some new env vars; REDIRECT_URL and REDIRECT_QUERY_STRING for
 * custom error responses, and DOCUMENT_ROOT because we found it useful.
 * It also adds SERVER_ADMIN - useful for scripts to know who to mail when
 * they fail.
 */

#include "apr_lib.h"
#include "apr_strings.h"
#include "apr_general.h"
#include "apr_file_io.h"
#include "apr_portable.h"
#include "apr_buckets.h"
#include "apr_optional.h"
#include "apr_signal.h"

#define SUCGID_MODULE_PFX "SUCGID: "

#define APR_WANT_STRFUNC
#include "apr_want.h"

#if APR_HAVE_SYS_SOCKET_H
#include <sys/socket.h>
#endif
#if APR_HAVE_UNISTD_H
#include <unistd.h>
#endif
#if APR_HAVE_SYS_TYPES_H
#include <sys/types.h>
#endif

#define CORE_PRIVATE

#include "util_filter.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"
#include "ap_mpm.h"
#include "unixd.h"
#include "mod_include.h"

#include "mod_core.h"


/* ### should be tossed in favor of APR */
#include <sys/stat.h>
#include <sys/un.h> /* for sockaddr_un */
#include <sys/sem.h>

#include "sucgid.h"

static struct rlimit no_limit = { (rlim_t) -1, (rlim_t) -1 };




#ifdef _WIN32
	#if defined(SUCGID_EXPORT)
		#define SUCGID_DECL __declspec(dllexport)
	#else
		#define SUCGID_DECL
	#endif

#else
	#if defined(SUCGID_EXPORT)
		#define SUCGID_DECL __attribute__((visibility("default")))
	#else
		#define SUCGID_DECL
	#endif
#endif//#ifdef _WIN32

SUCGID_DECL module sucgid_module;

static int find_daemonid(pid_t daemon_pid);
static int cgidaemon_start(apr_pool_t *p, server_rec *main_server, apr_proc_t *procnew, int daemon_id);
static int cgidaemons_start(apr_pool_t *p, server_rec *main_server, apr_proc_t *procnew);
static int cgid_init(apr_pool_t *p, apr_pool_t *plog, apr_pool_t *ptemp, server_rec *main_server);
static int handle_exec(include_ctx_t *ctx, ap_filter_t *f, apr_bucket_brigade *bb);

static APR_OPTIONAL_FN_TYPE(ap_register_include_handler) *cgid_pfn_reg_with_ssi;
static APR_OPTIONAL_FN_TYPE(ap_ssi_get_tag_and_value) *cgid_pfn_gtv;
static APR_OPTIONAL_FN_TYPE(ap_ssi_parse_string) *cgid_pfn_ps;

typedef struct __attribute__ ((__packed__)) limits_t
{
        enum _fields
        {
                io_lim = 1L << 0
        } fields;

        unsigned int uid;
        unsigned int gid;
        unsigned int lve_cpu;
        unsigned int lve_nproc;
        unsigned int lve_io;
} limits_t;

APR_DECLARE_OPTIONAL_FN(int, poa_cgid_limits_lookup_ex,
        (conn_rec *, const char*, struct limits_t*));

APR_DECLARE_OPTIONAL_FN(int, poa_cgid_limits_lookup,
	(conn_rec*, 
	const char* host, 
	unsigned int* uid, 
	unsigned int* gid, 
	unsigned int* lve_cpu, 
	unsigned int* lve_nproc
	));

APR_DECLARE_OPTIONAL_FN(const char *, poa_get_err_log_path,
                        (int, conn_rec*, const char*));

static APR_OPTIONAL_FN_TYPE(poa_cgid_limits_lookup_ex) *cgid_pfn_limits_lookup_ex = NULL;
static APR_OPTIONAL_FN_TYPE(poa_cgid_limits_lookup) *cgid_pfn_limits_lookup = NULL;
static APR_OPTIONAL_FN_TYPE(poa_get_err_log_path) *cgid_pfn_err_log_path = NULL;

static int ignorelve_enabled = 0;
static int lve_enabled = 1;
static apr_pool_t *pcgi = NULL;
static int total_modules = 0;
static pid_t *daemon_pids = 0;
static server_rec *root_server = NULL;
static apr_pool_t *root_pool = NULL;
static const char *sockname;
static int daemons_count = 4;
static int script_ttl = 120;
static int io_limit = 100;
static int cpu_limit=-1;
static int mem_limit=-1;
static int ep_limit = -1;
static int ncpu_limit = 1;
static pid_t parent_pid;
static const char *daemon_path = "/usr/sbin/daemon_sucgid";
static int child_count_semid = 0;
static int child_count = 0;

/* The APR other-child API doesn't tell us how the daemon exited
 * (SIGSEGV vs. exit(1)).  The other-child maintenance function
 * needs to decide whether to restart the daemon after a failure
 * based on whether or not it exited due to a fatal startup error
 * or something that happened at steady-state.  This exit status
 * is unlikely to collide with exit signals.
 */
#define DAEMON_STARTUP_ERROR 254

/* Read and discard the data in the brigade produced by a CGI script */
static void discard_script_output(apr_bucket_brigade *bb);

/* This doer will only ever be called when we are sure that we have
 * a valid ugid.
 */
static ap_unix_identity_t *cgid_suexec_id_doer(const request_rec *r)
{
    return (ap_unix_identity_t *)
                        ap_get_module_config(r->request_config, &sucgid_module);
}

/* KLUDGE --- for back-combatibility, we don't have to check ExecCGI
 * in ScriptAliased directories, which means we need to know if this
 * request came through ScriptAlias or not... so the Alias module
 * leaves a note for us.
 */

static int is_scriptaliased(request_rec *r)
{
    const char *t = apr_table_get(r->notes, "alias-forced-type");
    return t && (!strcasecmp(t, "cgi-script"));
}

/* Configuration stuff */

#define DEFAULT_LOGBYTES 10385760
#define DEFAULT_BUFBYTES 1024
#define DEFAULT_SOCKET  DEFAULT_REL_RUNTIMEDIR "/cgisock"

#define CGI_REQ    1
#define SSI_REQ    2
#define GETPID_REQ 3 /* get the pid of script created for prior request */
#define KILLCHILD_REQ 4 /* kill child script process */

#define ERRFN_USERDATA_KEY         "CGIDCHILDERRFN"

/* DEFAULT_CGID_LISTENBACKLOG controls the max depth on the unix socket's
 * pending connection queue.  If a bunch of cgi requests arrive at about
 * the same time, connections from httpd threads/processes will back up
 * in the queue while the cgid process slowly forks off a child to process
 * each connection on the unix socket.  If the queue is too short, the
 * httpd process will get ECONNREFUSED when trying to connect.
 */
#ifndef DEFAULT_CGID_LISTENBACKLOG
#define DEFAULT_CGID_LISTENBACKLOG 100
#endif

/* DEFAULT_CONNECT_ATTEMPTS controls how many times we'll try to connect
 * to the cgi daemon from the thread/process handling the cgi request.
 * Generally we want to retry when we get ECONNREFUSED since it is
 * probably because the listen queue is full.  We need to try harder so
 * the client doesn't see it as a 503 error.
 *
 * Set this to 0 to continually retry until the connect works or Apache
 * terminates.
 */
#ifndef DEFAULT_CONNECT_ATTEMPTS
#define DEFAULT_CONNECT_ATTEMPTS  15
#endif

typedef struct {
    const char *logname;
    long logbytes;
    int bufbytes;
} cgid_server_conf;

typedef struct {
  uint32_t lve_id;
} cgid_dir_conf;

static char * get_daemon_sockname(apr_pool_t *p, const char * sockname, int daemon_id) 
{
    apr_psprintf(p, "%s.%d", sockname, daemon_id);
}

static int get_random_daemon_id()
{
	return rand() % daemons_count;
}

/* This routine is called to create the argument list to be passed
 * to the CGI script.  When suexec is enabled, the suexec path, user, and
 * group are the first three arguments to be passed; if not, all three
 * must be NULL.  The query info is split into separate arguments, where
 * "+" is the separator between keyword arguments.
 *
 * Do not process the args if they containing an '=' assignment.
 */
static char **create_argv(apr_pool_t *p, char *path, char *user, char *group,
                          char *av0, const char *args)
{
    int x, numwords;
    char **av;
    char *w;
    int idx = 0;

    if (!(*args) || ap_strchr_c(args, '=')) {
        numwords = 0;
    }
    else {
        /* count the number of keywords */

        for (x = 0, numwords = 1; args[x]; x++) {
            if (args[x] == '+') {
                ++numwords;
            }
        }
    }

    if (numwords > APACHE_ARG_MAX - 5) {
        numwords = APACHE_ARG_MAX - 5;  /* Truncate args to prevent overrun */
    }
    av = (char **) apr_pcalloc(p, (numwords + 5) * sizeof(char *));

    if (path) {
        av[idx++] = path;
    }
    if (user) {
        av[idx++] = user;
    }
    if (group) {
        av[idx++] = group;
    }

    av[idx++] = apr_pstrdup(p, av0);

    for (x = 1; x <= numwords; x++) {
        w = ap_getword_nulls(p, &args, '+');
        ap_unescape_url(w);
        av[idx++] = ap_escape_shell_cmd(p, w);
    }
    av[idx] = NULL;
    return av;
}

#if APR_HAS_OTHER_CHILD
static void cgid_maint(int reason, void *data, apr_wait_t status)
{
    apr_proc_t *proc = data;
    int mpm_state;
    int stopping;
    int i;
	int daemon_id;

    switch (reason) {
        case APR_OC_REASON_DEATH:
            apr_proc_other_child_unregister(data);
            /* If apache is not terminating or restarting,
             * restart the cgid daemon
             */
            stopping = 1; /* if MPM doesn't support query,
                           * assume we shouldn't restart daemon
                           */
            if (ap_mpm_query(AP_MPMQ_MPM_STATE, &mpm_state) == APR_SUCCESS &&
                mpm_state != AP_MPMQ_STOPPING) {
                stopping = 0;
            }
            if (!stopping) {
                if (status == DAEMON_STARTUP_ERROR) {
                    ap_log_error(APLOG_MARK, APLOG_CRIT, 0, NULL,
                                 "cgid daemon failed to initialize");
                }
                else {
                    ap_log_error(APLOG_MARK, APLOG_ERR, 0, NULL,
                                 "cgid daemon process died, restarting");
				  
					daemon_id = find_daemonid(proc->pid);
                    daemon_pids[daemon_id] = cgidaemon_start(root_pool, root_server, proc, daemon_id);
                }
            }
            break;
        case APR_OC_REASON_RESTART:
            /* don't do anything; server is stopping or restarting */
            apr_proc_other_child_unregister(data);
            break;
        case APR_OC_REASON_LOST:
            /* Restart the child cgid daemon process */
            apr_proc_other_child_unregister(data);
	    daemon_id = find_daemonid(proc->pid);
	    daemon_pids[daemon_id] = cgidaemon_start(root_pool, root_server, 
						     proc, daemon_id);
            break;
        case APR_OC_REASON_UNREGISTER:
            /* we get here when pcgi is cleaned up; pcgi gets cleaned
             * up when pconf gets cleaned up
             */
            kill(proc->pid, SIGHUP); /* send signal to daemon telling it to die */
            daemon_id = find_daemonid(proc->pid);
            /* Remove the cgi socket, we must do it here in order to try and
             * guarantee the same permissions as when the socket was created.
             */
	    
	    char * daemon_sockname = get_daemon_sockname(root_pool, sockname, daemon_id);
	    if (unlink(daemon_sockname) < 0 && errno != ENOENT) {
	      ap_log_error(APLOG_MARK, APLOG_ERR, errno, NULL,
			   "Couldn't unlink unix domain socket %s",
			   daemon_sockname);
	    }
            break;
    }
}
#endif

/* deal with incomplete reads and signals
 * assume you really have to read buf_size bytes
 */
static apr_status_t sock_read(int fd, void *vbuf, size_t buf_size)
{
    char *buf = vbuf;
    int rc;
    size_t bytes_read = 0;

    do {
        do {
            rc = read(fd, buf + bytes_read, buf_size - bytes_read);
        } while (rc < 0 && errno == EINTR);
        switch(rc) {
        case -1:
            return errno;
        case 0: /* unexpected */
            return ECONNRESET;
        default:
            bytes_read += rc;
        }
    } while (bytes_read < buf_size);

    return APR_SUCCESS;
}

/* deal with signals
 */
static apr_status_t sock_write(int fd, const void *buf, size_t buf_size)
{
    int rc;

    do {
        rc = write(fd, buf, buf_size);
    } while (rc < 0 && errno == EINTR);
    if (rc < 0) {
        return errno;
    }

    return APR_SUCCESS;
}

static apr_status_t sock_writev(int fd, request_rec *r, int count, ...)
{
    va_list ap;
    int rc;
    struct iovec *vec;
    int i;
    int total_bytes = 0;

    vec = (struct iovec *)apr_palloc(r->pool, count * sizeof(struct iovec));
    va_start(ap, count);
    for (i = 0; i < count; i++) {
        vec[i].iov_base = va_arg(ap, caddr_t);
        vec[i].iov_len  = va_arg(ap, apr_size_t);
        total_bytes += vec[i].iov_len;
    }
    va_end(ap);

    do {
        rc = writev(fd, vec, count);
    } while (rc < 0 && errno == EINTR);
    if (rc < 0) {
        return errno;
    }

    return APR_SUCCESS;
}

static apr_status_t send_req(int fd, request_rec *r, char *argv0, char **env,
                             int req_type)
{
  int i;
  sucgid_req_t req = {0};
  apr_status_t stat;

  /* TODO
   * Code below is example how to obtain uid, gid.
   * Untill nss plugin is ready just send this data to log
   */
  unsigned int uid, gid, lve_cpu, lve_nproc;
  struct limits_t limits_data;
  if(cgid_pfn_limits_lookup_ex && !cgid_pfn_limits_lookup_ex(r->connection, r->hostname, &limits_data)) {
	  req.uid = limits_data.uid;
	  req.gid = limits_data.gid;
	  req.cpu = cpu_limit==-1?limits_data.lve_cpu:cpu_limit;
	  req.ncpu = ((ncpu_limit==-1)?0:ncpu_limit);
	  req.maxentryproc = ep_limit==-1?limits_data.lve_nproc:ep_limit;
	  if (limits_data.fields & io_lim)
	      req.io = limits_data.lve_io;
	  else
	      req.io = io_limit;
  } else if(cgid_pfn_limits_lookup && !cgid_pfn_limits_lookup(r->connection, r->hostname, &uid, &gid, &lve_cpu, &lve_nproc)) {
	  req.uid = uid;
	  req.gid = gid;
	  req.cpu = cpu_limit==-1?lve_cpu:cpu_limit;
	  req.ncpu = ((ncpu_limit==-1)?0:ncpu_limit);
	  req.maxentryproc = ep_limit==-1?lve_nproc:ep_limit;
	  /* Temporary decision until io not works */
	  req.io = io_limit;
  } else {
      ap_unix_identity_t * ugid = ap_run_get_suexec_identity(r);
      req.cpu = -1;
      if (ugid == NULL) {
        req.uid = empty_ugid;
        req.gid = empty_ugid;
      } else {
    	req.uid = ugid->uid;
		req.gid = ugid->gid;
      }
      /* Temporary decision until io not works */
      req.ncpu = ((ncpu_limit==-1)?0:ncpu_limit);
      req.io = io_limit;
  }

  cgid_dir_conf *cfg = (cgid_dir_conf *) ap_get_module_config(r->per_dir_config, &sucgid_module);
  if (cfg && (cfg->lve_id>0)) {
    req.lve_id = cfg->lve_id;
  } else {
    req.lve_id = req.uid;
  }

  req.req_type = req_type;
  req.ppid = parent_pid;
  req.conn_id = r->connection->id;
  req.core_module_index = core_module.module_index;
  req.ignore_lve = ignorelve_enabled;
  for (req.env_count = 0; env[req.env_count]; req.env_count++) {
    continue;
  }
    
  const char * const *argv = (const char * const *)create_argv(r->pool, 
							       NULL, NULL, NULL, argv0, 
							       r->args ? r->args : "");
    
  for (req.args_count = 0; argv[req.args_count]; req.args_count++) {
    continue;
  }
    
  req.filename_len = strlen(r->filename);
  req.argv0_len = strlen(argv0);
  req.uri_len = strlen(r->uri);
  req.args_len = 0;
  req.loglevel = r->server->loglevel;

  core_dir_config *conf = ap_get_module_config(r->per_dir_config, &core_module);
  if (cpu_limit != -1) {
     req.limit_cpu.rlim_cur = cpu_limit;
     req.limit_cpu.rlim_max = cpu_limit;
  } else if (conf->limit_cpu) {
    req.limit_cpu = *conf->limit_cpu;
  } else {
    req.limit_cpu.rlim_cur = -1;
    req.limit_cpu.rlim_max = -1;
  }
  
  if (mem_limit != -1) {
    req.limit_mem.rlim_cur = mem_limit * 1024 * 1024;    
    req.limit_mem.rlim_max = mem_limit * 1024 * 1024;
  } else if (conf->limit_mem) {
    req.limit_mem = *conf->limit_mem;
  } else {
    req.limit_mem.rlim_cur = -1;
    req.limit_mem.rlim_max = -1;
  }
  
  if (ep_limit != -1) {
    req.limit_nproc.rlim_cur = ep_limit;
    req.limit_nproc.rlim_max = ep_limit;
  } else if (conf->limit_nproc) {
    req.limit_nproc = *conf->limit_nproc;
  } else {
    req.limit_nproc.rlim_cur = -1;
    req.limit_nproc.rlim_max = -1;
  }
  
  req.err_log_len = 0;
  const char* err_log_path = "";
  if(cgid_pfn_err_log_path) {
    const char* elog = cgid_pfn_err_log_path(1, r->connection, r->hostname);
    if(elog) {
      err_log_path = elog;
      req.err_log_len = strlen(err_log_path);
    }
  }  					      

  /* Write the request header */
  
  ap_log_error(APLOG_MARK, APLOG_CRIT, 0, NULL,
		"sucgi mod_sucgid: MEMORY LIMIT LVEID %d rlim_cur %lu, rlim_max %lu", req.lve_id, req.limit_mem.rlim_cur, req.limit_mem.rlim_max);

  stat = sock_writev(fd, r, 4,
		     &req, sizeof(req),
		     r->filename, req.filename_len,
		     argv0, req.argv0_len,
		     r->uri, req.uri_len);

  if (stat != APR_SUCCESS) {
    return stat;
  }

  if (req.err_log_len > 0) {
    stat = sock_writev(fd, r, 1, err_log_path, req.err_log_len);
    if (stat != APR_SUCCESS)
      return stat;
  }

  /* write the environment variables */
  for (i = 0; i < req.env_count; i++) {
    apr_size_t curlen = strlen(env[i]);

    if ((stat = sock_writev(fd, r, 2, &curlen, sizeof(curlen),
			    env[i], curlen)) != APR_SUCCESS) {
      return stat;
    }
  }
    
    
  for (i = 0; i < req.args_count; i++) {
    apr_size_t curlen = strlen(argv[i]);

    if ((stat = sock_writev(fd, r, 2, &curlen, sizeof(curlen),
			    argv[i], curlen)) != APR_SUCCESS) {
      return stat;
    }
  }

  return APR_SUCCESS;
}

static void cgid_child_errfn(apr_pool_t *pool, apr_status_t err,
                             const char *description)
{
    request_rec *r;
    void *vr;

    apr_pool_userdata_get(&vr, ERRFN_USERDATA_KEY, pool);
    r = vr;

    /* sure we got r, but don't call ap_log_rerror() because we don't
     * have r->headers_in and possibly other storage referenced by
     * ap_log_rerror()
     */
    ap_log_error(APLOG_MARK, APLOG_ERR, err, r->server, "%s", description);
}

static int find_daemonid(pid_t daemon_pid)
{
	// TODO: remove this function as well as daemon_pids, get_daemon_sockname after
	// sucgid will use shared hash for KILLCHILD_REQ and same socket
    int i;
	for(i=0; i<daemons_count; i++)
		if(daemon_pids[i] == daemon_pid)
			return i;

	// impossible situation
	ap_log_error(APLOG_MARK, APLOG_CRIT, 0, NULL,
				 "sucgi: find_daemonid failed for pid %d", daemon_pid);
	return 0;
}

static int cgidaemons_start(apr_pool_t *p, server_rec *main_server,
                      apr_proc_t *procnew)
{
    int i;

    daemon_pids = (pid_t *)apr_palloc(p, daemons_count * sizeof(pid_t));
    memset(daemon_pids, 0, daemons_count * sizeof(pid_t));

	for(i=0; i < daemons_count; i++) 
	{
		daemon_pids[i] = cgidaemon_start(p, main_server, &procnew[i], i);
		if(daemon_pids[i] < 0)
			return DECLINED;
	}
    return OK;
}

static pid_t cgidaemon_start(apr_pool_t *p, server_rec *main_server,
                      apr_proc_t *procnew, int daemon_id)
{
	pid_t daemon_pid;
	if ((daemon_pid = fork()) < 0) {
		ap_log_error(APLOG_MARK, APLOG_ERR, errno, main_server,
					 "sucgi: Couldn't spawn cgid daemon process");
		return daemon_pid;
	}
	else if (daemon_pid == 0) {
		/* Close our copy of the listening sockets */
		ap_close_listeners();
		char apid[16];
		sprintf(apid,"%d",getppid());
		char semid[16];
		sprintf(semid,"%d",child_count_semid);
		const char * daemon_sockname = get_daemon_sockname(p, sockname, daemon_id);
		const char * const argv[] = {daemon_path,
				       apid,
				       daemon_sockname,
				       (lve_enabled?"1":"0"),
				       semid,
				       NULL};
		execvp(daemon_path, (char * const *) argv);
		exit(-1);
	}
	procnew->pid = daemon_pid;
	procnew->err = procnew->in = procnew->out = NULL;
	apr_pool_note_subprocess(p, procnew, APR_KILL_AFTER_TIMEOUT);
#if APR_HAS_OTHER_CHILD
	apr_proc_other_child_register(procnew, cgid_maint, procnew, NULL, p);
#endif

	return daemon_pid;
}

static int cgid_pre_config(apr_pool_t *pconf, apr_pool_t *plog,
                           apr_pool_t *ptemp)
{
    sockname = ap_append_pid(pconf, DEFAULT_SOCKET, ".");
    return OK;
}

static int cgid_init(apr_pool_t *p, apr_pool_t *plog, apr_pool_t *ptemp,
                     server_rec *main_server)
{
    apr_proc_t *procnew = NULL;
    int first_time = 0;
    const char *userdata_key = "cgid_init";
    module **m;
    int ret = OK;
    void *data;
	int i;

    root_server = main_server;
    root_pool = p;

    apr_pool_userdata_get(&data, userdata_key, main_server->process->pool);
    if (!data) {
        first_time = 1;
        procnew = apr_pcalloc(main_server->process->pool, daemons_count * sizeof(*procnew));
		for(i=0; i<daemons_count; i++)
		{
        	procnew[i].pid = -1;
            procnew[i].err = procnew[i].in = procnew[i].out = NULL;
		}
		apr_pool_userdata_set((const void *)procnew, userdata_key,
					 apr_pool_cleanup_null, main_server->process->pool);
    }
    else {
        procnew = data;
    }

    if (!first_time) {
        total_modules = 0;
        for (m = ap_preloaded_modules; *m != NULL; m++)
            total_modules++;

        parent_pid = getpid();
        sockname = ap_server_root_relative(p, sockname);
	if(child_count){
		child_count_semid = semget(IPC_PRIVATE, 1, 0644);
		semctl(child_count_semid, 0, SETVAL, child_count);
	}
        ret = cgidaemons_start(p, main_server, procnew);
        if (ret != OK ) {
            return ret;
        }
        cgid_pfn_reg_with_ssi = APR_RETRIEVE_OPTIONAL_FN(ap_register_include_handler);
        cgid_pfn_gtv          = APR_RETRIEVE_OPTIONAL_FN(ap_ssi_get_tag_and_value);
        cgid_pfn_ps           = APR_RETRIEVE_OPTIONAL_FN(ap_ssi_parse_string);
        cgid_pfn_limits_lookup = APR_RETRIEVE_OPTIONAL_FN(poa_cgid_limits_lookup);
        cgid_pfn_limits_lookup_ex = APR_RETRIEVE_OPTIONAL_FN(poa_cgid_limits_lookup_ex);
        cgid_pfn_err_log_path = APR_RETRIEVE_OPTIONAL_FN(poa_get_err_log_path);
        if ((cgid_pfn_reg_with_ssi) && (cgid_pfn_gtv) && (cgid_pfn_ps)) {
            /* Required by mod_include filter. This is how mod_cgid registers
             *   with mod_include to provide processing of the exec directive.
             */
            cgid_pfn_reg_with_ssi("exec", handle_exec);
        }
    }
    return ret;
}

static void *create_cgid_dir_config(apr_pool_t *p, char *dirspec) {
  cgid_dir_conf *cfg = (cgid_dir_conf *) apr_pcalloc(p, sizeof(cgid_dir_conf));
  cfg->lve_id = 0;
}

static void *merge_cgid_dir_config(apr_pool_t *p, void *basev, void *overridesv) {
  cgid_dir_conf *base = basev;
  cgid_dir_conf *add = overridesv;
  cgid_dir_conf *cfg = (cgid_dir_conf *) apr_pcalloc(p, sizeof(cgid_dir_conf));
  cfg->lve_id = add->lve_id?add->lve_id:base->lve_id;
  return cfg;
}

static void *create_cgid_config(apr_pool_t *p, server_rec *s)
{
  cgid_server_conf *c =
    (cgid_server_conf *) apr_pcalloc(p, sizeof(cgid_server_conf));
  
  c->logname = NULL;
  c->logbytes = DEFAULT_LOGBYTES;
  c->bufbytes = DEFAULT_BUFBYTES;
  return c;
}

static void *merge_cgid_config(apr_pool_t *p, void *basev, void *overridesv)
{
    cgid_server_conf *base = (cgid_server_conf *) basev, *add = (cgid_server_conf *) overridesv;
    cgid_server_conf* cfg = (cgid_server_conf *) 
      apr_pcalloc(p, sizeof(cgid_server_conf));
    cfg->logname = add->logname?add->logname:base->logname;
    cfg->logbytes = add->logbytes?add->logbytes:base->logbytes;
    cfg->bufbytes = add->bufbytes?add->bufbytes:base->bufbytes;
    return cfg;
}

static const char *set_scriptlog(cmd_parms *cmd, void *dummy, const char *arg)
{
    server_rec *s = cmd->server;
    cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                  &sucgid_module);

    conf->logname = ap_server_root_relative(cmd->pool, arg);

    if (!conf->logname) {
        return apr_pstrcat(cmd->pool, "Invalid ScriptLog path ",
                           arg, NULL);
    }
    return NULL;
}

static const char *set_lve_id(cmd_parms *cmd, void *mcfg,
			      const char *lve_id)
{
  cgid_dir_conf *cfg = (cgid_dir_conf *) mcfg;
  if (lve_id) {
    cfg->lve_id = (uint32_t)apr_atoi64(lve_id);;
  }
  return NULL;
}

static const char *set_cpu_limit(cmd_parms *cmd, void *dummy,
                                const char *arg) {
  server_rec *s = cmd->server;
  cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                &sucgid_module);
  if (arg) {
    cpu_limit = atol(arg);
  }
  return NULL;
}

static const char *set_ncpu_limit(cmd_parms *cmd, void *dummy,
                                const char *arg) {
  server_rec *s = cmd->server;
  cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                &sucgid_module);
  if (arg) {
    ncpu_limit = atol(arg);
  }
  return NULL;
}

static const char *set_mem_limit(cmd_parms *cmd, void *dummy,
                                const char *arg) {
  server_rec *s = cmd->server;
  cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                &sucgid_module);
  if (arg) {
    mem_limit = atol(arg);
  }
  return NULL;
}

static const char *set_ep_limit(cmd_parms *cmd, void *dummy,
                                const char *arg) {
  server_rec *s = cmd->server;
  cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                &sucgid_module);
  if (arg) {
    ep_limit = atol(arg);
  }
  return NULL;
}



static const char *set_io_limit(cmd_parms *cmd, void *dummy,
                                const char *arg) {
  server_rec *s = cmd->server;
  cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                &sucgid_module);
  if (arg) {
    io_limit = atol(arg);
  }
  return NULL;
}

static const char *set_scriptlog_length(cmd_parms *cmd, void *dummy, const char *arg)
{
    server_rec *s = cmd->server;
    cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                  &sucgid_module);

    conf->logbytes = atol(arg);
    return NULL;
}

static const char *set_scriptlog_buffer(cmd_parms *cmd, void *dummy, const char *arg)
{
    server_rec *s = cmd->server;
    cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                  &sucgid_module);

    conf->bufbytes = atoi(arg);
    return NULL;
}

static const char *set_script_socket(cmd_parms *cmd, void *dummy, const char *arg)
{
    const char *err = ap_check_cmd_context(cmd, GLOBAL_ONLY);
    if (err != NULL) {
        return err;
    }

    /* Make sure the pid is appended to the sockname */
    sockname = ap_append_pid(cmd->pool, arg, ".");
    sockname = ap_server_root_relative(cmd->pool, sockname);

    if (!sockname) {
        return apr_pstrcat(cmd->pool, "Invalid ScriptSock path",
                           arg, NULL);
    }

    return NULL;
}

static const char *set_script_daemons_count(cmd_parms *cmd, void *dummy, const char *arg)
{
    server_rec *s = cmd->server;
    cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                  &sucgid_module);

    daemons_count = atol(arg);

    return NULL;
}

static const char *set_daemon_path(cmd_parms *cmd, void *dummy, const char *arg)
{
    const char *err = ap_check_cmd_context(cmd, GLOBAL_ONLY);
    if (err != NULL) {
        return err;
    }
    const char * path = ap_server_root_relative(cmd->pool, arg);

    if (path) {
      daemon_path = path;
    }
    //TODO: test that path actually leads to right file -- or should we hard code it?

    return NULL;
}

static const char *set_enable_lve(cmd_parms *cmd, void *dummy, int flag) {
  lve_enabled = flag;
  if (!lve_enabled) {
    ap_log_error(APLOG_MARK, APLOG_NOTICE, 0, NULL,
		 "sucgid: LVE is disabled");
  }
  return NULL;
}

static const char *set_enable_ignorelve(cmd_parms *cmd, void *dummy, int flag) {
  ignorelve_enabled = flag;
  return NULL;
}

static const char *set_script_ttl(cmd_parms *cmd, void *dummy, const char *arg)
{
    server_rec *s = cmd->server;
    cgid_server_conf *conf = ap_get_module_config(s->module_config,
                                                   &sucgid_module);

    script_ttl = atol(arg);
    /* ap_log_error(APLOG_MARK, APLOG_NOTICE, 0, NULL, 
		 SUCGID_MODULE_PFX"cgid script TTL set to %d.", script_ttl);
    */
    return NULL;
}

static const char *set_child_count(cmd_parms *cmd, void *dummy, const char *arg)
{
    child_count = atol(arg);
    return NULL;
}

static const command_rec cgid_cmds[] =
{
    AP_INIT_TAKE1("ScriptLog", set_scriptlog, NULL, RSRC_CONF,
                  "the name of a log for script debugging info"),
    AP_INIT_TAKE1("ScriptLogLength", set_scriptlog_length, NULL, RSRC_CONF,
                  "the maximum length (in bytes) of the script debug log"),
    AP_INIT_TAKE1("ScriptLogBuffer", set_scriptlog_buffer, NULL, RSRC_CONF,
                  "the maximum size (in bytes) to record of a POST request"),
    AP_INIT_TAKE1("ScriptSock", set_script_socket, NULL, RSRC_CONF,
                  "the name of the socket to use for communication with "
                  "the cgi daemon."),
    AP_INIT_TAKE1("ScriptDaemonsCount", set_script_daemons_count, NULL, RSRC_CONF,
                  "the count of the child daemons that accept connections on the socket"),
    AP_INIT_TAKE1("ScriptDaemon", set_daemon_path, NULL, RSRC_CONF,
		  "the path to daemon executable"),
    AP_INIT_FLAG("EnableLVE", set_enable_lve, NULL, RSRC_CONF, 
		 "Enable LVE"),
    AP_INIT_FLAG("LVEIgnoreErrors", set_enable_ignorelve, NULL, RSRC_CONF, 
		 "Ignore E2BIG or EALEADY lve errors"),
    AP_INIT_TAKE1("ScriptTTL", set_script_ttl, NULL, RSRC_CONF,
                  "number of seconds after which script will be killed"),
    AP_INIT_TAKE1("LVEId", set_lve_id, NULL, ACCESS_CONF | RSRC_CONF, "LVE Id"),
    AP_INIT_TAKE1("MaxChildCount", set_child_count, NULL, RSRC_CONF,
                  "maximum number of child processes to execute"),
    AP_INIT_TAKE1("SuCGIDIOLimit", set_io_limit, NULL, RSRC_CONF,
                  "IO Limit in KB/s"),
    AP_INIT_TAKE1("OverrideCPULimit", set_cpu_limit, NULL, RSRC_CONF,
                  "CPU Limit"),
    AP_INIT_TAKE1("OverrideMEMLimit", set_mem_limit, NULL, RSRC_CONF,
                  "MEM Limit"),
    AP_INIT_TAKE1("OverrideEPROCLimit", set_ep_limit, NULL, RSRC_CONF,
                  "EP Limit"),
    AP_INIT_TAKE1("OverrideNCPULimit", set_ncpu_limit, NULL, RSRC_CONF,
                                    "NCPU Limit"),


    {NULL}
};

static int log_scripterror(request_rec *r, cgid_server_conf * conf, int ret,
                           apr_status_t rv, char *error)
{
    apr_file_t *f = NULL;
    struct stat finfo;
    char time_str[APR_CTIME_LEN];
    int log_flags = rv ? APLOG_ERR : APLOG_ERR;

    ap_log_rerror(APLOG_MARK, log_flags, rv, r,
                "%s: %s", error, r->filename);

    /* XXX Very expensive mainline case! Open, then getfileinfo! */
    if (!conf->logname ||
        ((stat(conf->logname, &finfo) == 0)
         && (finfo.st_size > conf->logbytes)) ||
         (apr_file_open(&f, conf->logname,
                  APR_APPEND|APR_WRITE|APR_CREATE, APR_OS_DEFAULT, r->pool) != APR_SUCCESS)) {
        return ret;
    }

    /* "%% [Wed Jun 19 10:53:21 1996] GET /cgid-bin/printenv HTTP/1.0" */
    apr_ctime(time_str, apr_time_now());
    apr_file_printf(f, "%%%% [%s] %s %s%s%s %s\n", time_str, r->method, r->uri,
            r->args ? "?" : "", r->args ? r->args : "", r->protocol);
    /* "%% 500 /usr/local/apache/cgid-bin */
    apr_file_printf(f, "%%%% %d %s\n", ret, r->filename);

    apr_file_printf(f, "%%error\n%s\n", error);

    apr_file_close(f);
    return ret;
}

static int log_script(request_rec *r, cgid_server_conf * conf, int ret,
                      char *dbuf, const char *sbuf, apr_bucket_brigade *bb,
                      apr_file_t *script_err)
{
    const apr_array_header_t *hdrs_arr = apr_table_elts(r->headers_in);
    const apr_table_entry_t *hdrs = (apr_table_entry_t *) hdrs_arr->elts;
    char argsbuffer[HUGE_STRING_LEN];
    apr_file_t *f = NULL;
    apr_bucket *e;
    const char *buf;
    apr_size_t len;
    apr_status_t rv;
    int first;
    int i;
    struct stat finfo;
    char time_str[APR_CTIME_LEN];

    /* XXX Very expensive mainline case! Open, then getfileinfo! */
    if (!conf->logname ||
        ((stat(conf->logname, &finfo) == 0)
         && (finfo.st_size > conf->logbytes)) ||
         (apr_file_open(&f, conf->logname,
                  APR_APPEND|APR_WRITE|APR_CREATE, APR_OS_DEFAULT, r->pool) != APR_SUCCESS)) {
        /* Soak up script output */
        discard_script_output(bb);
        if (script_err) {
            while (apr_file_gets(argsbuffer, HUGE_STRING_LEN,
                                 script_err) == APR_SUCCESS)
                continue;
        }
        return ret;
    }

    /* "%% [Wed Jun 19 10:53:21 1996] GET /cgid-bin/printenv HTTP/1.0" */
    apr_ctime(time_str, apr_time_now());
    apr_file_printf(f, "%%%% [%s] %s %s%s%s %s\n", time_str, r->method, r->uri,
            r->args ? "?" : "", r->args ? r->args : "", r->protocol);
    /* "%% 500 /usr/local/apache/cgid-bin" */
    apr_file_printf(f, "%%%% %d %s\n", ret, r->filename);

    apr_file_puts("%request\n", f);
    for (i = 0; i < hdrs_arr->nelts; ++i) {
        if (!hdrs[i].key)
            continue;
        apr_file_printf(f, "%s: %s\n", hdrs[i].key, hdrs[i].val);
    }
    if ((r->method_number == M_POST || r->method_number == M_PUT)
        && *dbuf) {
        apr_file_printf(f, "\n%s\n", dbuf);
    }

    apr_file_puts("%response\n", f);
    hdrs_arr = apr_table_elts(r->err_headers_out);
    hdrs = (const apr_table_entry_t *) hdrs_arr->elts;

    for (i = 0; i < hdrs_arr->nelts; ++i) {
        if (!hdrs[i].key)
            continue;
        apr_file_printf(f, "%s: %s\n", hdrs[i].key, hdrs[i].val);
    }

    if (sbuf && *sbuf)
        apr_file_printf(f, "%s\n", sbuf);

    first = 1;

    for (e = APR_BRIGADE_FIRST(bb);
         e != APR_BRIGADE_SENTINEL(bb);
         e = APR_BUCKET_NEXT(e))
    {
        if (APR_BUCKET_IS_EOS(e)) {
            break;
        }
        rv = apr_bucket_read(e, &buf, &len, APR_BLOCK_READ);
        if (rv != APR_SUCCESS || (len == 0)) {
            break;
        }
        if (first) {
            apr_file_puts("%stdout\n", f);
            first = 0;
        }
        apr_file_write(f, buf, &len);
        apr_file_puts("\n", f);
    }

    if (script_err) {
        if (apr_file_gets(argsbuffer, HUGE_STRING_LEN,
                          script_err) == APR_SUCCESS) {
            apr_file_puts("%stderr\n", f);
            apr_file_puts(argsbuffer, f);
            while (apr_file_gets(argsbuffer, HUGE_STRING_LEN,
                                 script_err) == APR_SUCCESS)
                apr_file_puts(argsbuffer, f);
            apr_file_puts("\n", f);
        }
    }

    if (script_err) {
        apr_file_close(script_err);
    }

    apr_file_close(f);
    return ret;
}

static apr_status_t close_unix_socket(void *thefd)
{
    int fd = (int)((long)thefd);

    return close(fd);
}

static int connect_to_daemon(int *sdptr, request_rec *r,
                             cgid_server_conf *conf, int daemon_id)
{
    struct sockaddr_un unix_addr;
    int sd;
    int connect_tries;
    apr_interval_time_t sliding_timer;
	char * daemon_sockname = get_daemon_sockname(r->pool, sockname, daemon_id);

    memset(&unix_addr, 0, sizeof(unix_addr));
    unix_addr.sun_family = AF_UNIX;
    apr_cpystrn(unix_addr.sun_path, daemon_sockname, sizeof unix_addr.sun_path);

    connect_tries = 0;
    sliding_timer = 100000; /* 100 milliseconds */
    while (1) {
        ++connect_tries;
        if ((sd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
            return log_scripterror(r, conf, HTTP_INTERNAL_SERVER_ERROR, errno,
                                   "unable to create socket to cgi daemon");
        }
        if (connect(sd, (struct sockaddr *)&unix_addr, sizeof(unix_addr)) < 0) {
            if (/*errno == ECONNREFUSED && */ connect_tries < DEFAULT_CONNECT_ATTEMPTS) {
                ap_log_rerror(APLOG_MARK, APLOG_DEBUG, errno, r,
                              "connect #%d to cgi daemon failed, sleeping before retry",
                              connect_tries);
                close(sd);
                apr_sleep(sliding_timer);
                if (sliding_timer < apr_time_from_sec(2)) {
                    sliding_timer *= 2;
                }
            }
            else {
                close(sd);
                return log_scripterror(r, conf, HTTP_SERVICE_UNAVAILABLE, errno,
                                       "unable to connect to cgi daemon after multiple tries");
            }
        }
        else {
            apr_pool_cleanup_register(r->pool, (void *)((long)sd),
                                      close_unix_socket, apr_pool_cleanup_null);
            break; /* we got connected! */
        }
        /* gotta try again, but make sure the cgid daemon is still around */
		if (kill(daemon_pids[daemon_id], 0) != 0) {
			return log_scripterror(r, conf, HTTP_SERVICE_UNAVAILABLE, errno,
					       "cgid daemon is gone; is Apache terminating?");
		}
    }
    *sdptr = sd;
    return OK;
}

static void discard_script_output(apr_bucket_brigade *bb)
{
    apr_bucket *e;
    const char *buf;
    apr_size_t len;
    apr_status_t rv;

    for (e = APR_BRIGADE_FIRST(bb);
         e != APR_BRIGADE_SENTINEL(bb);
         e = APR_BUCKET_NEXT(e))
    {
        if (APR_BUCKET_IS_EOS(e)) {
            break;
        }
        rv = apr_bucket_read(e, &buf, &len, APR_BLOCK_READ);
        if (rv != APR_SUCCESS) {
            break;
        }
    }
}

/****************************************************************
 *
 * Actual cgid handling...
 */

struct cleanup_script_info {
    request_rec *r;
    unsigned long conn_id;
    cgid_server_conf *conf;
    int daemon_id;
};

static apr_status_t cleanup_script(void *vptr)
{
    struct cleanup_script_info *info = vptr;
    int sd;
    int rc;
    sucgid_req_t req = {0};
    pid_t pid;
    apr_status_t stat;

    rc = connect_to_daemon(&sd, info->r, info->conf, info->daemon_id);
    if (rc != OK) {
        return APR_EGENERAL;
    }

    /* we got a socket, and there is already a cleanup registered for it */

    req.req_type = KILLCHILD_REQ;
    req.ppid = parent_pid;
    req.conn_id = info->r->connection->id;

    stat = sock_write(sd, &req, sizeof(req));
    if (stat != APR_SUCCESS) {
        return stat;
    }

    /* wait for pid of script */
    stat = sock_read(sd, &pid, sizeof(pid));
    if (stat != APR_SUCCESS) {
        return stat;
    }

    if (pid == 0) {
        ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, info->r,
                      "daemon couldn't find CGI process for connection %lu",
                      info->conn_id);
        return APR_EGENERAL;
    }
    return APR_SUCCESS;
}

static int cgid_handler(request_rec *r)
{
    conn_rec *c = r->connection;
    int retval, nph, dbpos = 0;
    char *argv0, *dbuf = NULL;
    apr_bucket_brigade *bb;
    apr_bucket *b;
    cgid_server_conf *conf;
    int is_included;
    int seen_eos, child_stopped_reading;
    int sd;
    char **env;
    apr_file_t *tempsock;
    struct cleanup_script_info *info;
    apr_status_t rv;
    int daemon_id = get_random_daemon_id();

    if (strcmp(r->handler,CGI_MAGIC_TYPE) && strcmp(r->handler,"cgi-script"))
        return DECLINED;

    conf = ap_get_module_config(r->server->module_config, &sucgid_module);
    is_included = !strcmp(r->protocol, "INCLUDED");

    if ((argv0 = strrchr(r->filename, '/')) != NULL)
        argv0++;
    else
        argv0 = r->filename;

    nph = !(strncmp(argv0, "nph-", 4));

    argv0 = r->filename;

    if (!(ap_allow_options(r) & OPT_EXECCGI) && !is_scriptaliased(r))
        return log_scripterror(r, conf, HTTP_FORBIDDEN, 0,
                               "Options ExecCGI is off in this directory");
    if (nph && is_included)
        return log_scripterror(r, conf, HTTP_FORBIDDEN, 0,
                               "attempt to include NPH CGI script");

#if defined(OS2) || defined(WIN32)
#error mod_cgid does not work on this platform.  If you teach it to, look
#error at mod_cgi.c for required code in this path.
#else
    if (r->finfo.filetype == 0)
        return log_scripterror(r, conf, HTTP_NOT_FOUND, 0,
                               "script not found or unable to stat");
#endif
    if (r->finfo.filetype == APR_DIR)
        return log_scripterror(r, conf, HTTP_FORBIDDEN, 0,
                               "attempt to invoke directory as script");

    if ((r->used_path_info == AP_REQ_REJECT_PATH_INFO) &&
        r->path_info && *r->path_info)
    {
        /* default to accept */
        return log_scripterror(r, conf, HTTP_NOT_FOUND, 0,
                               "AcceptPathInfo off disallows user's path");
    }

    int current_child_count = child_count ? semctl(child_count_semid, 0, GETVAL) : 0;
    if ( child_count && current_child_count == 0 ) {
      return log_scripterror(r, conf, HTTP_SERVICE_UNAVAILABLE, 0, "MaxChildCount reached, dropping connection");
    } else if ( child_count && current_child_count <= child_count / 10 ) {
      ap_log_rerror(APLOG_MARK, APLOG_ERR, rv, r, "MaxChildCount exhausted to 90%%");
    }
/*
    if (!ap_suexec_enabled) {
        if (!ap_can_exec(&r->finfo))
            return log_scripterror(r, conf, HTTP_FORBIDDEN, 0,
                                   "file permissions deny server execution");
    }
*/
    ap_add_common_vars(r);
    ap_add_cgi_vars(r);
    env = ap_create_environment(r->pool, r->subprocess_env);

    if ((retval = connect_to_daemon(&sd, r, conf, daemon_id)) != OK) {
        return retval;
    }

    rv = send_req(sd, r, argv0, env, CGI_REQ);
    if (rv != APR_SUCCESS) {
        ap_log_rerror(APLOG_MARK, APLOG_ERR, rv, r,
                     "write to cgi daemon process");
    }

    info = apr_palloc(r->pool, sizeof(struct cleanup_script_info));
    info->r = r;
    info->conn_id = r->connection->id;
    info->conf = conf;
    info->daemon_id = daemon_id;
    apr_pool_cleanup_register(r->pool, info,
                              cleanup_script,
                              apr_pool_cleanup_null);
    /* We are putting the socket discriptor into an apr_file_t so that we can
     * use a pipe bucket to send the data to the client.  APR will create
     * a cleanup for the apr_file_t which will close the socket, so we'll
     * get rid of the cleanup we registered when we created the socket.
     */

    apr_os_pipe_put_ex(&tempsock, &sd, 1, r->pool);
    apr_pool_cleanup_kill(r->pool, (void *)((long)sd), close_unix_socket);

    if ((argv0 = strrchr(r->filename, '/')) != NULL)
        argv0++;
    else
        argv0 = r->filename;

    /* Transfer any put/post args, CERN style...
     * Note that we already ignore SIGPIPE in the core server.
     */
    bb = apr_brigade_create(r->pool, r->connection->bucket_alloc);
    seen_eos = 0;
    child_stopped_reading = 0;
    if (conf->logname) {
        dbuf = apr_palloc(r->pool, conf->bufbytes + 1);
        dbpos = 0;
    }
    do {
        apr_bucket *bucket;

        rv = ap_get_brigade(r->input_filters, bb, AP_MODE_READBYTES,
                            APR_BLOCK_READ, HUGE_STRING_LEN);

        if (rv != APR_SUCCESS) {
            ap_log_rerror(APLOG_MARK, APLOG_ERR, rv, r,
                          "Error reading request entity data");
            return HTTP_INTERNAL_SERVER_ERROR;
        }

        for (bucket = APR_BRIGADE_FIRST(bb);
             bucket != APR_BRIGADE_SENTINEL(bb);
             bucket = APR_BUCKET_NEXT(bucket))
        {
            const char *data;
            apr_size_t len;

            if (APR_BUCKET_IS_EOS(bucket)) {
                seen_eos = 1;
                break;
            }

            /* We can't do much with this. */
            if (APR_BUCKET_IS_FLUSH(bucket)) {
                continue;
            }

            /* If the child stopped, we still must read to EOS. */
            if (child_stopped_reading) {
                continue;
            }

            /* read */
            apr_bucket_read(bucket, &data, &len, APR_BLOCK_READ);

            if (conf->logname && dbpos < conf->bufbytes) {
                int cursize;

                if ((dbpos + len) > conf->bufbytes) {
                    cursize = conf->bufbytes - dbpos;
                }
                else {
                    cursize = len;
                }
                memcpy(dbuf + dbpos, data, cursize);
                dbpos += cursize;
            }

            /* Keep writing data to the child until done or too much time
             * elapses with no progress or an error occurs.
             */
            rv = apr_file_write_full(tempsock, data, len, NULL);

            if (rv != APR_SUCCESS) {
                /* silly script stopped reading, soak up remaining message */
                child_stopped_reading = 1;
            }
        }
        apr_brigade_cleanup(bb);
    }
    while (!seen_eos);

    if (conf->logname) {
        dbuf[dbpos] = '\0';
    }

    /* we're done writing, or maybe we didn't write at all;
     * force EOF on child's stdin so that the cgi detects end (or
     * absence) of data
     */
    shutdown(sd, 1);

    /* Handle script return... */
    if (!nph) {
        const char *location;
        char sbuf[MAX_STRING_LEN];
        int ret;

        bb = apr_brigade_create(r->pool, c->bucket_alloc);
        b = apr_bucket_pipe_create(tempsock, c->bucket_alloc);
	apr_file_pipe_timeout_set(b->data, apr_time_from_sec(script_ttl));
        APR_BRIGADE_INSERT_TAIL(bb, b);
        b = apr_bucket_eos_create(c->bucket_alloc);
        APR_BRIGADE_INSERT_TAIL(bb, b);

        if ((ret = ap_scan_script_header_err_brigade(r, bb, sbuf))) {
	    cleanup_script(info);
            ret = log_script(r, conf, ret, dbuf, sbuf, bb, NULL);

            /*
             * ret could be HTTP_NOT_MODIFIED in the case that the CGI script
             * does not set an explicit status and ap_meets_conditions, which
             * is called by ap_scan_script_header_err_brigade, detects that
             * the conditions of the requests are met and the response is
             * not modified.
             * In this case set r->status and return OK in order to prevent
             * running through the error processing stack as this would
             * break with mod_cache, if the conditions had been set by
             * mod_cache itself to validate a stale entity.
             * BTW: We circumvent the error processing stack anyway if the
             * CGI script set an explicit status code (whatever it is) and
             * the only possible values for ret here are:
             *
             * HTTP_NOT_MODIFIED          (set by ap_meets_conditions)
             * HTTP_PRECONDITION_FAILED   (set by ap_meets_conditions)
             * HTTP_INTERNAL_SERVER_ERROR (if something went wrong during the
             * processing of the response of the CGI script, e.g broken headers
             * or a crashed CGI process).
             */
            if (ret == HTTP_NOT_MODIFIED) {
                r->status = ret;
                return OK;
            }

            return ret;
        }

	const char *lve_error = apr_table_get(r->err_headers_out, "X-LVE-Error-Status");
	if (lve_error!=NULL){
		ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, lve_error);
		discard_script_output(bb);
            	apr_brigade_destroy(bb);
            	return HTTP_INTERNAL_SERVER_ERROR;
	}	

        location = apr_table_get(r->headers_out, "Location");

        if (location && location[0] == '/' && r->status == 200) {

            /* Soak up all the script output */
            discard_script_output(bb);
            apr_brigade_destroy(bb);
            /* This redirect needs to be a GET no matter what the original
             * method was.
             */
            r->method = apr_pstrdup(r->pool, "GET");
            r->method_number = M_GET;

            /* We already read the message body (if any), so don't allow
             * the redirected request to think it has one. We can ignore
             * Transfer-Encoding, since we used REQUEST_CHUNKED_ERROR.
             */
            apr_table_unset(r->headers_in, "Content-Length");

            ap_internal_redirect_handler(location, r);
            return OK;
        }
        else if (location && r->status == 200) {
            /* XX Note that if a script wants to produce its own Redirect
             * body, it now has to explicitly *say* "Status: 302"
             */
            discard_script_output(bb);
            apr_brigade_destroy(bb);
            return HTTP_MOVED_TEMPORARILY;
        }

        int pass_rv = ap_pass_brigade(r->output_filters, bb);
	if(pass_rv != APR_SUCCESS){
		char buff[128];
		ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, "ap_pass_brigade failed with error %d: %s", pass_rv, apr_strerror(pass_rv, buff, sizeof(buff)));
	}
    } else {
        struct ap_filter_t *cur;

        /* get rid of all filters up through protocol...  since we
         * haven't parsed off the headers, there is no way they can
         * work
         */

        cur = r->proto_output_filters;
        while (cur && cur->frec->ftype < AP_FTYPE_CONNECTION) {
            cur = cur->next;
        }
        r->output_filters = r->proto_output_filters = cur;

        bb = apr_brigade_create(r->pool, c->bucket_alloc);
        b = apr_bucket_pipe_create(tempsock, c->bucket_alloc);
        APR_BRIGADE_INSERT_TAIL(bb, b);
        b = apr_bucket_eos_create(c->bucket_alloc);
        APR_BRIGADE_INSERT_TAIL(bb, b);
        ap_pass_brigade(r->output_filters, bb);
    }

    return OK; /* NOT r->status, even if it has changed. */
}




/*============================================================================
 *============================================================================
 * This is the beginning of the cgi filter code moved from mod_include. This
 *   is the code required to handle the "exec" SSI directive.
 *============================================================================
 *============================================================================*/
static apr_status_t include_cgi(include_ctx_t *ctx, ap_filter_t *f,
                                apr_bucket_brigade *bb, char *s)
{
    request_rec *r = f->r;
    request_rec *rr = ap_sub_req_lookup_uri(s, r, f->next);
    int rr_status;

    if (rr->status != HTTP_OK) {
        ap_destroy_sub_req(rr);
        return APR_EGENERAL;
    }

    /* No hardwired path info or query allowed */
    if ((rr->path_info && rr->path_info[0]) || rr->args) {
        ap_destroy_sub_req(rr);
        return APR_EGENERAL;
    }
    if (rr->finfo.filetype != APR_REG) {
        ap_destroy_sub_req(rr);
        return APR_EGENERAL;
    }

    /* Script gets parameters of the *document*, for back compatibility */
    rr->path_info = r->path_info;       /* hard to get right; see mod_cgi.c */
    rr->args = r->args;

    /* Force sub_req to be treated as a CGI request, even if ordinary
     * typing rules would have called it something else.
     */
    ap_set_content_type(rr, CGI_MAGIC_TYPE);

    /* Run it. */
    rr_status = ap_run_sub_req(rr);
    if (ap_is_HTTP_REDIRECT(rr_status)) {
        const char *location = apr_table_get(rr->headers_out, "Location");

        if (location) {
            char *buffer;

            location = ap_escape_html(rr->pool, location);
            buffer = apr_pstrcat(ctx->pool, "<a href=\"", location, "\">",
                                 location, "</a>", NULL);

            APR_BRIGADE_INSERT_TAIL(bb, apr_bucket_pool_create(buffer,
                                    strlen(buffer), ctx->pool,
                                    f->c->bucket_alloc));
        }
    }

    ap_destroy_sub_req(rr);

    return APR_SUCCESS;
}

/* This is the special environment used for running the "exec cmd="
 *   variety of SSI directives.
 */
static void add_ssi_vars(request_rec *r)
{
    apr_table_t *e = r->subprocess_env;

    if (r->path_info && r->path_info[0] != '\0') {
        request_rec *pa_req;

        apr_table_setn(e, "PATH_INFO", ap_escape_shell_cmd(r->pool, r->path_info));

        pa_req = ap_sub_req_lookup_uri(ap_escape_uri(r->pool, r->path_info), r, NULL);
        if (pa_req->filename) {
            apr_table_setn(e, "PATH_TRANSLATED",
                           apr_pstrcat(r->pool, pa_req->filename, pa_req->path_info, NULL));
        }
        ap_destroy_sub_req(pa_req);
    }

    if (r->args) {
        char *arg_copy = apr_pstrdup(r->pool, r->args);

        apr_table_setn(e, "QUERY_STRING", r->args);
        ap_unescape_url(arg_copy);
        apr_table_setn(e, "QUERY_STING_UNESCAPED", ap_escape_shell_cmd(r->pool, arg_copy));
    }
}

static int include_cmd(include_ctx_t *ctx, ap_filter_t *f,
                       apr_bucket_brigade *bb, char *command)
{
    char **env;
    int sd;
    int retval;
    apr_file_t *tempsock = NULL;
    request_rec *r = f->r;
    cgid_server_conf *conf = ap_get_module_config(r->server->module_config,
                                                  &sucgid_module);
    struct cleanup_script_info *info;
    int daemon_id = get_random_daemon_id();

    add_ssi_vars(r);
    env = ap_create_environment(r->pool, r->subprocess_env);

    if ((retval = connect_to_daemon(&sd, r, conf, daemon_id)) != OK) {
        return retval;
    }

    send_req(sd, r, command, env, SSI_REQ);

    info = apr_palloc(r->pool, sizeof(struct cleanup_script_info));
    info->r = r;
    info->conn_id = r->connection->id;
    info->conf = conf;
    info->daemon_id = daemon_id;
    /* for this type of request, the script is invoked through an
     * intermediate shell process...  cleanup_script is only able
     * to knock out the shell process, not the actual script
     */
    apr_pool_cleanup_register(r->pool, info,
                              cleanup_script,
                              apr_pool_cleanup_null);

    /* We are putting the socket discriptor into an apr_file_t so that we can
     * use a pipe bucket to send the data to the client.  APR will create
     * a cleanup for the apr_file_t which will close the socket, so we'll
     * get rid of the cleanup we registered when we created the socket.
     */
    apr_os_pipe_put_ex(&tempsock, &sd, 1, r->pool);
    apr_pool_cleanup_kill(r->pool, (void *)((long)sd), close_unix_socket);

    APR_BRIGADE_INSERT_TAIL(bb, apr_bucket_pipe_create(tempsock,
                            f->c->bucket_alloc));
    ctx->flush_now = 1;

    return APR_SUCCESS;
}

static apr_status_t handle_exec(include_ctx_t *ctx, ap_filter_t *f,
                                apr_bucket_brigade *bb)
{
    char *tag     = NULL;
    char *tag_val = NULL;
    request_rec *r = f->r;
    char *file = r->filename;
    char parsed_string[MAX_STRING_LEN];

    if (!ctx->argc) {
        ap_log_rerror(APLOG_MARK,
                      (ctx->flags & SSI_FLAG_PRINTING)
                          ? APLOG_ERR : APLOG_WARNING,
                      0, r, "missing argument for exec element in %s",
                      r->filename);
    }

    if (!(ctx->flags & SSI_FLAG_PRINTING)) {
        return APR_SUCCESS;
    }

    if (!ctx->argc) {
        SSI_CREATE_ERROR_BUCKET(ctx, f, bb);
        return APR_SUCCESS;
    }

    if (ctx->flags & SSI_FLAG_NO_EXEC) {
        ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, "exec used but not allowed "
                      "in %s", r->filename);
        SSI_CREATE_ERROR_BUCKET(ctx, f, bb);
        return APR_SUCCESS;
    }

    while (1) {
        cgid_pfn_gtv(ctx, &tag, &tag_val, SSI_VALUE_DECODED);
        if (!tag || !tag_val) {
            break;
        }

        if (!strcmp(tag, "cmd")) {
            apr_status_t rv;

            cgid_pfn_ps(ctx, tag_val, parsed_string, sizeof(parsed_string),
                        SSI_EXPAND_LEAVE_NAME);

            rv = include_cmd(ctx, f, bb, parsed_string);
            if (rv != APR_SUCCESS) {
                ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r,
                              "execution failure for parameter \"%s\" "
                              "to tag exec in file %s", tag, r->filename);
                SSI_CREATE_ERROR_BUCKET(ctx, f, bb);
                break;
            }
        }
        else if (!strcmp(tag, "cgi")) {
            apr_status_t rv;

            cgid_pfn_ps(ctx, tag_val, parsed_string, sizeof(parsed_string),
                        SSI_EXPAND_DROP_NAME);

            rv = include_cgi(ctx, f, bb, parsed_string);
            if (rv != APR_SUCCESS) {
                ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, "invalid CGI ref "
                              "\"%s\" in %s", tag_val, file);
                SSI_CREATE_ERROR_BUCKET(ctx, f, bb);
                break;
            }
        }
        else {
            ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, "unknown parameter "
                          "\"%s\" to tag exec in %s", tag, file);
            SSI_CREATE_ERROR_BUCKET(ctx, f, bb);
            break;
        }
    }

    return APR_SUCCESS;
}
/*============================================================================
 *============================================================================
 * This is the end of the cgi filter code moved from mod_include.
 *============================================================================
 *============================================================================*/


static void register_hook(apr_pool_t *p)
{
    static const char * const aszPre[] = { "mod_include.c", NULL };

    ap_hook_pre_config(cgid_pre_config, NULL, NULL, APR_HOOK_MIDDLE);
    ap_hook_post_config(cgid_init, aszPre, NULL, APR_HOOK_MIDDLE);
    ap_hook_handler(cgid_handler, NULL, NULL, APR_HOOK_FIRST);
}

SUCGID_DECL module sucgid_module = 
{
    STANDARD20_MODULE_STUFF,
    create_cgid_dir_config, /* dir config creater */
    merge_cgid_dir_config, /* dir merger --- default is to override */
    create_cgid_config, /* server config */
    merge_cgid_config, /* merge server config */
    cgid_cmds, /* command table */
    register_hook /* register_handlers */
};

