X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=statserve.c;h=9bee73b2c7dd62a86f35d497cfbfa09e453464f2;hb=a823d5b4a231f42a598ac3624f9271a2a3b29a50;hp=ed4938822e93c60d9acfe95f82ae5cc839e22055;hpb=34d725a595138f81ba84d67d7b84766c3a270d9f;p=statserve.git

diff --git a/statserve.c b/statserve.c
index ed49388..9bee73b 100644
--- a/statserve.c
+++ b/statserve.c
@@ -3,6 +3,7 @@
 #include <unistd.h>
 #include <errno.h>
 #include <string.h>
+#include <sys/socket.h>
 #include <ashd/req.h>
 #include <ashd/resp.h>
 #include <ashd/log.h>
@@ -59,7 +60,7 @@ out:
 
 static void usage(FILE *out)
 {
-    fprintf(out, "usage: statserve [-h] SOURCE...\n");
+    fprintf(out, "usage: statserve [-h] [-P PAGESIZE] SOURCE...\n");
 }
 
 static void listenloop(struct muth *muth, va_list args)
@@ -84,13 +85,31 @@ static void listenloop(struct muth *muth, va_list args)
     }
 }
 
+static void sigterm(int sig)
+{
+    shutdown(0, SHUT_RDWR);
+}
+
+static void closeall(void)
+{
+    struct source *src;
+    
+    for(src = sources; src != NULL; src = src->next) {
+	if(src->close)
+	    src->close(src);
+    }
+}
+
 int main(int argc, char **argv)
 {
     int c;
     struct source *last, *src;
     
-    while((c = getopt(argc, argv, "+h")) >= 0) {
+    while((c = getopt(argc, argv, "+hP:")) >= 0) {
 	switch(c) {
+	case 'P':
+	    dbpagesize = atoi(optarg);
+	    break;
 	case 'h':
 	    usage(stdout);
 	    return(0);
@@ -101,7 +120,10 @@ int main(int argc, char **argv)
     }
     last = NULL;
     while(optind < argc) {
-	src = parsesource(argv[optind++]);
+	if((src = parsesource(argv[optind++])) == NULL) {
+	    closeall();
+	    return(1);
+	}
 	if(!sources)
 	    sources = src;
 	if(last)
@@ -110,9 +132,13 @@ int main(int argc, char **argv)
     }
     if(!sources) {
 	usage(stderr);
+	closeall();
 	return(1);
     }
     mustart(listenloop, 0);
+    signal(SIGINT, sigterm);
+    signal(SIGTERM, sigterm);
     ioloop();
+    closeall();
     return(0);
 }