Initial import of htcd system 1.0

Signed-off-by: Henrik Rydberg <rydberg@euromail.se>

1 files changed, 153 insertions, 0 deletions

diff --git a/src/net/htcd.cc b/src/net/htcd.cc
new file mode 100644
index 0000000..89729e6
--- /dev/null
+++ b/src/net/htcd.cc
@@ -0,0 +1,153 @@
+/*************************************************************************
+ *
+ * HTCd - Copyright (C) 1998-2006 Henrik Rydberg
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+#include <net/thrserv.h>
+
+static fp_stream merr(stderr);
+
+static mstring PATH;
+static htcServ* SERV;
+
+const mstring PIDURI="/sys/htcd";
+
+const int BACKLOG=4;
+const int SERVERS=9;
+const int LOWER=3;
+const int UPPER=6;
+const int KEEPALIVE=300;
+const int SNDBUFSIZE=16384;
+const int RCVBUFSIZE=16384;
+
+///////////////////////////////////////////////////////////
+
+static thrServ* server[SERVERS];
+static thrCron* crontab;
+
+extern "C" {
+  static void* thr_handler(void* p) { ((thrServ*)p)->enter(); return 0; }
+  static void* thr_cron(void* p) { ((thrCron*)p)->enter(); return 0; }
+  static void onTERM(int sig) { SERV->htcd.running=0; }
+  static void onHUP(int sig) { if(crontab) crontab->running=0; }
+  typedef void (*disp_t)(int);
+}
+
+static void MainProc(isocket& sock)
+{
+  int keep=1;
+  while(SERV->htcd.running) {
+    int nvacant=0; for(int i=0;i<SERVERS;i++) if(server[i]->vacant()) nvacant++;
+    if(nvacant>UPPER) keep=1; else if(nvacant<=LOWER) keep=0;
+    if(!keep) for(int i=0;i<SERVERS;i++) server[i]->shutdown();
+    time_t timeout=time(0)-KEEPALIVE;
+    for(int i=0;i<SERVERS;i++) if(timeout>server[i]->last) server[i]->disconnect();
+    int test=sock.test();
+    if(test>0) {
+      thrServ* p=0;
+      for(int i=0;i<SERVERS;i++) if(server[i]->vacant()) { p=server[i]; break; }
+      if(!p) for(int i=0;i<SERVERS;i++) if(!p||server[i]->first<p->first) p=server[i];
+      if(p->shutdown()) {
+        try {
+          isocket* s=sock.accept();
+          if(s) p->reset(s,keep);
+        }
+        catch(const merror_t& e) {
+          merr<<"SOCKET: "<<e.desc<<"\n";
+        }
+      }
+    }
+    else if(test<0) SERV->htcd.running=0;
+  }
+  for(int i=0;i<SERVERS;i++) {
+    server[i]->term();
+    pthread_join(server[i]->ptid,0);
+  }
+  pthread_kill(crontab->tid,SIGHUP);
+  pthread_join(crontab->ptid,0);
+}
+
+///////////////////////////////////////////////////////////////
+
+static void httpServer(int overport)
+{
+  isocket sock(SERV->htcd.node.addr,overport,BACKLOG);
+  sock.keepalive(1);
+  sock.sendbufsize(SNDBUFSIZE);
+  sock.recvbufsize(RCVBUFSIZE);
+
+  sigignore(SIGPIPE);
+  disp_t onterm=signal(SIGTERM,onTERM);
+  disp_t onint=signal(SIGINT,onTERM);
+  disp_t onhup=signal(SIGHUP,onHUP);
+  MainProc(sock);
+  signal(SIGHUP,onhup);
+  signal(SIGINT,onint);
+  signal(SIGTERM,onterm);
+}
+
+static void Daemon(int overport) try
+{
+  PROCLOCK(PATH+PIDURI);
+  SERV=new htcServ(PATH);
+  crontab=new thrCron(*SERV);
+  pthread_create(&crontab->ptid,0,thr_cron,crontab);
+  for(int i=0;i<SERVERS;i++) {
+    server[i]=new thrServ(*SERV);
+    pthread_create(&server[i]->ptid,0,thr_handler,server[i]);
+  }
+  merr<<"htcd: "<<PATH<<" started\n";
+  httpServer(overport<0?0:overport==0?SERV->htcd.port:overport);
+  merr<<"htcd: "<<PATH<<" terminating\n";
+  for(int i=0;i<SERVERS;i++) delete server[i];
+  delete crontab;
+}
+catch(const merror_t& e) {
+  if(SERV) {
+    SERV->htcd.errlog<<"["<<time(0)<<"][MAIN]["<<e.desc<<"]\n";
+    merr<<e.desc<<"\n";
+  }
+  else if(First(e.desc)!=sref("proclock:")) merr<<e.desc<<"\n";
+}
+catch(...) {
+  if(SERV) SERV->htcd.errlog<<"["<<time(0)<<"][MAIN][FATAL]\n";  
+  merr<<"FATAL\n";
+}
+
+main(int argc,char* argv[]) try
+{
+  if(argc<3) {
+    merr<<"Usage: "<<argv[0]<<" <path> <overport=-1|0|port> [-1|0|1]\n"; 
+    return -1;
+  }
+  PATH=argv[1]; if(PATH.back()=='/') PATH=PATH.popb();
+  setenv("DOCROOT",PATH.c_str(),1);
+  int overport=atoi(argv[2]);
+  int cmd=argc>3?atoi(argv[3]):0;
+  if(cmd) ProcTerminate(PATH+PIDURI);
+  if(cmd>=0) { if(fork()==0) Daemon(overport); }
+  delete SERV;
+  return 0;
+}
+catch(const merror_t& e) {
+  merr<<"ERROR: ["<<e.desc<<"]\n";
+  return -1;
+}
+catch(...) {
+  merr<<"ERROR: [FATAL]\n";
+  return -1;
+}