initial commit into repository0.0

1 files changed, 325 insertions, 0 deletions

diff --git a/src/sitecing-fastcgi.cc b/src/sitecing-fastcgi.cc
new file mode 100644
index 0000000..963c257
--- a/dev/null
+++ b/src/sitecing-fastcgi.cc
@@ -0,0 +1,325 @@
+#include <sys/types.h>
+#include <unistd.h>
+#include <signal.h>
+#include <getopt.h>
+#include <pwd.h>
+#include <grp.h>
+#include <errno.h>
+#include <syslog.h>
+#include <iostream>
+#include <memory>
+#include <typeinfo>
+using namespace std;
+#include "kingate/fastcgi.h"
+#include "kingate/cgi_gateway.h"
+using namespace kingate;
+#include "sitecing/sitecing_util.h"
+#include "sitecing/util.h"
+#include "sitecing/sitespace.h"
+#include "sitecing/sitecing_interface_cgi.h"
+#include "sitecing/cgi_component.h"
+#include "sitecing/configuration.h"
+#include "sitecing/magic.h"
+#include "sitecing/sitecing_exception.h"
+#include "sitecing/exception.h"
+#include "sitecing/process_manager.h"
+using namespace sitecing;
+
+#include "config.h"
+#define PHEADER PACKAGE " Version " VERSION
+#define PCOPY   "Copyright (c) 2004 Klever Group"
+
+class cdummyClass : public acomponent {
+    public:
+	void main(int _magic,va_list _args) {}
+	void *__the_most_derived_this() { return NULL; }
+} cdummyInstance;
+class adummyClass : public cgi_component {
+    public:
+	void main(int _magic,va_list _args) {}
+	void *__the_most_derived_this() { return NULL; }
+} adummyInstance;
+
+class sitecing_fastcgi_pm : public process_manager {
+    public:
+	configuration config;
+	fcgi_socket *fss;
+	semaphore sem;
+	bool multi;
+	uid_t uid;
+	gid_t gid;
+	pid_file pidfile;
+
+	sitecing_fastcgi_pm(const string& config_file);
+	~sitecing_fastcgi_pm();
+
+	void process(int slot);
+
+	void run();
+	void give_up_privs();
+};
+
+sitecing_fastcgi_pm::sitecing_fastcgi_pm(const string& config_file)
+    : config(config_file), multi(false) {
+	if(( (config.flags&configuration::flag_user) || (config.flags&configuration::flag_group) || (config.flags&configuration::flag_chroot) ) && geteuid() )
+	    throw konforka::exception(CODEPOINT,"can't use User, Group or Chroot when started without root privileges");
+	if(config.flags&configuration::flag_user) {
+	    struct passwd *ptmp = getpwnam(config.user.c_str());
+	    if(ptmp) {
+		uid = ptmp->pw_uid;
+	    }else{
+		errno=0;
+		uid = strtol(config.user.c_str(),NULL,0);
+		if(errno)
+		    throw konforka::exception(CODEPOINT,"failed to resolve User value to uid");
+	    }
+	}
+	if(config.flags&configuration::flag_group) {
+	    struct group *gtmp = getgrnam(config.group.c_str());
+	    if(gtmp) {
+		gid = gtmp->gr_gid;
+	    }else{
+		errno=0;
+		gid = strtol(config.group.c_str(),NULL,0);
+		if(errno)
+		    throw konforka::exception(CODEPOINT,"failed to resolve Group value to gid");
+	    }
+	}
+	if(!(config.flags&configuration::flag_root_source))
+	    throw konforka::exception(CODEPOINT,"unspecified root for sources");
+	if(!(config.flags&configuration::flag_root_intermediate))
+	    throw konforka::exception(CODEPOINT,"unspecified root for intermediate files");
+	if(!(config.flags&configuration::flag_root_so))
+	    throw konforka::exception(CODEPOINT,"unspecified root for shared objects");
+	if(config.flags&configuration::flag_min_children)
+	    min_children = config.min_children;
+	if(config.flags&configuration::flag_max_children)
+	    max_children = config.max_children;
+	if(config.flags&configuration::flag_min_spare_children)
+	    min_spare_children = config.min_spare_children;
+	if(config.flags&configuration::flag_max_spare_children)
+	    max_spare_children = config.max_spare_children;
+	if(max_children<min_spare_children)
+	    throw konforka::exception(CODEPOINT,"inconsistent numbers of MaxChildren and MinSpareChildren");
+	if(min_children>max_spare_children && max_spare_children>=0)
+	    throw konforka::exception(CODEPOINT,"inconsistent numbers of MinChildren and MaxSpareChildren");
+	if(config.flags&configuration::flag_multi_process) {
+	    multi = config.multi_process;
+	}else{
+	    if(config.flags&configuration::flag_listen_socket)
+		multi = true;
+	    else
+		multi = false;
+	}
+	fss = (config.flags&configuration::flag_listen_socket)?new fcgi_socket(config.listen_socket.c_str(),5):new fcgi_socket(0);
+	if(!fss)
+	    throw konforka::exception(CODEPOINT,"failed to establish listening socket");
+	if(config.flags&configuration::flag_daemonize && config.daemonize) {
+	    pid_t pf = fork();
+	    if(pf<0)
+		throw konforka::exception(CODEPOINT,"failed to fork()");
+	    if(pf) {
+		die_humbly=true;
+		_exit(0);
+	    }
+	}
+	if(config.flags&configuration::flag_pid_file) {
+	    pidfile.set(config.pid_file);
+	}
+	if(multi)
+	    sem.init();
+    }
+sitecing_fastcgi_pm::~sitecing_fastcgi_pm() {
+    if(fss)
+	delete fss;
+}
+
+void sitecing_fastcgi_pm::process(int slot) {
+    signal(SIGINT,SIG_DFL);
+    signal(SIGABRT,SIG_DFL);
+    signal(SIGTERM,SIG_DFL);
+    give_up_privs();
+    scoreboard_slot *sslot = sboard.get_slot(slot);
+    try {
+	sitespace ss(config);
+	fcgi_socket& fs = *fss;
+	sitecing_interface_cgi scif(&ss);
+	string component_path;
+	string action;
+	config_options::action_handler_t *action_handler;
+	int rpc = 0;
+	if(config.flags&configuration::flag_requests_per_child)
+	    rpc = config.requests_per_child;
+	for(int req=0;(rpc<=0) || (req<rpc);rpc++) {
+	    semaphore_lock sl;
+	    if(multi) {
+		sslot->state = scoreboard_slot::state_idle;
+		sl.sem = &sem;
+		sl.lock();
+	    }
+	    sslot->state = scoreboard_slot::state_accept;
+	    fcgi_interface fi(fs);
+	    sslot->state = scoreboard_slot::state_processing;
+	    if(multi)
+		sl.unlock();
+	    cgi_gateway gw(fi);
+	    scif.prepare(&gw);
+	    try {
+		component_path = normalize_path(gw.get_meta("PATH_INFO"),strip_leading_slash|strip_trailing_slash);
+		string full_component_path;
+		while(true) {
+		    full_component_path = config.root_source+'/'+component_path;
+		    if(!access(full_component_path.c_str(),F_OK))
+			break;
+		    string::size_type sl = component_path.rfind('/');
+		    if(sl==string::npos)
+			throw konforka::exception(CODEPOINT,"can't find the target component");
+		    component_path.erase(sl);
+		}
+		action = component_path;
+		action_handler = config.lookup_action_handler(component_path);
+		if(action_handler) {
+		    action = action_handler->action;
+		}
+		string pwd = dir_name(full_component_path);
+		if(chdir(pwd.c_str()))
+		    throw konforka::exception(CODEPOINT,"failed to chdir() into document's directory");
+		so_component soc = ss.fetch(action,&scif);
+		if(action_handler) {
+		    soc.ac->run(__magic_action,
+			    config.root_source.c_str(), config.root_intermediate.c_str(), config.root_so.c_str(),
+			    &(action_handler->args)
+			    );
+		}else{
+		    soc.ac->main(0,NULL);
+		}
+	    }catch(http_status& hs) {
+		scif.headers["Status"] = hs.status+" "+hs.message;
+		string hshp = config.lookup_http_status_handler(component_path,hs.status);
+		if(!hshp.empty()) {
+		    so_component hsh = ss.fetch(hshp,&scif); // TODO: handle error trying to handle status
+		    hsh.ac->run(__magic_http_status,config.root_source.c_str(),config.root_intermediate.c_str(),
+			    config.root_so.c_str(),action.c_str(),&hs);
+		}
+	    }catch(compile_error& ce) {
+		config_options *co_exception_handler = config.lookup_config(component_path,config_options::flag_exception_handler);
+		if(co_exception_handler) {
+		    so_component eh = ss.fetch(co_exception_handler->exception_handler,&scif); // TODO: handle error trying to handle error.
+		    eh.ac->run(__magic_compile_error,ce.what(),config.root_source.c_str(),config.root_intermediate.c_str(),config.root_so.c_str(),ce.component_path.c_str());
+		}else{
+		    ce.see(CODEPOINT);
+		    throw;
+		}
+	    }catch(preprocessor_error& pe) {
+		config_options *co_exception_handler = config.lookup_config(component_path,config_options::flag_exception_handler);
+		if(co_exception_handler) {
+		    so_component eh = ss.fetch(co_exception_handler->exception_handler,&scif); // TODO: handle error trying to handle error.
+		    eh.ac->run(__magic_preprocess_error,pe.what(),config.root_source.c_str(),config.root_intermediate.c_str(),config.root_so.c_str(),pe.component_name.c_str(),pe.line_number);
+		}else{
+		    pe.see(CODEPOINT);
+		    throw;
+		}
+	    }catch(exception& e) {
+		config_options *co_exception_handler = config.lookup_config(component_path,config_options::flag_exception_handler);
+		if(co_exception_handler) {
+		    so_component eh = ss.fetch(co_exception_handler->exception_handler,&scif); // TODO: handle error trying to handle error.
+		    eh.ac->run(__magic_generic_exception,e.what(),config.root_source.c_str(),config.root_intermediate.c_str(),config.root_so.c_str(),component_path.c_str(),&e);
+		}
+	    }
+	    scif.flush();
+	}
+    }catch(exception& e) {
+	cerr << "->Oops: " << e.what() << endl;
+    }
+}
+
+void sitecing_fastcgi_pm::run() {
+    if(multi)
+	manage();
+    else
+	process(0);
+}
+
+void sitecing_fastcgi_pm::give_up_privs() {
+    if(config.flags&configuration::flag_chroot) {
+	if(chroot(config.chroot.c_str()))
+	    throw konforka::exception(CODEPOINT,"failed to chroot()");
+    }
+    if(config.flags&configuration::flag_group) {
+	if((getgid()!=gid) && setgid(gid))
+	    throw konforka::exception(CODEPOINT,"failed to setgid()");
+    }
+    if(config.flags&configuration::flag_user) {
+	if((getuid()!=uid) && setuid(uid))
+	    throw konforka::exception(CODEPOINT,"failed to setuid()");
+    }
+}
+
+static sitecing_fastcgi_pm* _process_manager = NULL;
+
+static void lethal_signal_handler(int signum) {
+    _process_manager->finishing=true;
+}
+
+int main(int argc,char **argv) {
+    const char* id = *argv;
+    const char* t;
+    while(t = index(id,'/')) {
+	id=t; id++;
+    }
+    openlog(id,LOG_PERROR|LOG_PID,LOG_USER);
+    try {
+	string config_file = "sitecing.conf";
+	while(true) {
+	    static struct option opts[] = {
+		{ "help", no_argument, 0, 'h' },
+		{ "usage", no_argument, 0, 'h' },
+		{ "version", no_argument, 0, 'V' },
+		{ "license", no_argument, 0, 'L' },
+		{ "config", required_argument, 0, 'f' },
+		{ NULL, 0, 0, 0 }
+	    };
+	    int c = getopt_long(argc,argv,"f:hVL",opts,NULL);
+	    if(c==-1)
+		break;
+	    switch(c) {
+		case 'h':
+		    cerr << PHEADER << endl
+			<< PCOPY << endl << endl
+			<< " -h, --help" << endl
+			<< " --usage          display this text" << endl
+			<< " -V, --version    display version number" << endl
+			<< " -L, --license    show license" << endl
+			<< " -f filename, --config=filename" << endl
+			<< "                  specify configuration file to use" << endl;
+		    exit(0);
+		    break;
+		case 'V':
+		    cerr << VERSION << endl;
+		    exit(0);
+		    break;
+		case 'L':
+		    extern const char *COPYING;
+		    cerr << COPYING << endl;
+		    exit(0);
+		    break;
+		case 'f':
+		    config_file = optarg;
+		    break;
+		default:
+		    cerr << "Huh??" << endl;
+		    break;
+	    }
+	}
+	sitecing_fastcgi_pm sfpm(config_file);
+	_process_manager = &sfpm;
+	signal(SIGINT,lethal_signal_handler);
+	signal(SIGABRT,lethal_signal_handler);
+	signal(SIGTERM,lethal_signal_handler);
+	sfpm.run();
+    }catch(exception& e) {
+	/* cerr << "Oops: " << e.what() << endl; */
+	syslog(LOG_ERR,"uncaught exception: %s",e.what());
+    }
+    closelog();
+}