Special buffer initializer
[apps/pfixtools.git] / postlicyd.c
index 7877665..e9eb1c2 100644 (file)
 /******************************************************************************/
 
 /*
- * Copyright © 2006 Pierre Habouzit
+ * Copyright © 2006-2007 Pierre Habouzit
  */
 
-#include <stdbool.h>
-#include <stdio.h>
-#include <sysexits.h>
-#include <syslog.h>
+#include <signal.h>
+#include <time.h>
+#include <getopt.h>
 
-#include "job.h"
+#include "common.h"
 
-bool cleanexit = false;
+static sig_atomic_t cleanexit = false;
+static sig_atomic_t sigint    = false;
+static volatile int nbthreads = 0;
 
-static void shutdown(void)
+static void main_sighandler(int sig)
+{
+    static time_t lastintr = 0;
+    time_t now = time(NULL);
+
+    switch (sig) {
+      case SIGINT:
+        if (sigint) {
+            if (now - lastintr >= 1)
+                break;
+        } else {
+            lastintr = now;
+            sigint   = true;
+        }
+        return;
+
+      case SIGTERM:
+        break;
+
+      default:
+        return;
+    }
+
+    syslog(LOG_ERR, "Killed...");
+    exit(-1);
+}
+
+static void main_initialize(void)
+{
+    openlog("postlicyd", LOG_PID, LOG_MAIL);
+    signal(SIGPIPE, SIG_IGN);
+    signal(SIGINT,  &main_sighandler);
+    signal(SIGTERM, &main_sighandler);
+    syslog(LOG_INFO, "Starting...");
+}
+
+void *job_run(void *_fd)
+{
+    int fd = (intptr_t)_fd;
+
+    close(fd);
+    return NULL;
+}
+
+static void main_loop(void)
+{
+    int sock = -1;
+
+    while (!sigint) {
+        int fd = accept(sock, NULL, 0);
+        pthread_attr_t attr;
+        pthread_t dummy;
+
+        if (fd < 0) {
+            if (errno != EINTR || errno != EAGAIN)
+                UNIXERR("accept");
+            continue;
+        }
+
+        pthread_attr_init(&attr);
+        pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
+        pthread_create(&dummy, &attr, job_run, (void *)(intptr_t)fd);
+        pthread_attr_destroy(&attr);
+    }
+
+    cleanexit = true;
+    close(sock);
+}
+
+static void main_shutdown(void)
 {
     syslog(LOG_INFO, cleanexit ? "Stopping..." : "Unclean exit...");
-    job_shutdown();
     closelog();
 }
 
 int main(void)
 {
-    if (atexit(shutdown)) {
+    if (atexit(main_shutdown)) {
         fputs("Cannot hook my atexit function, quitting !\n", stderr);
-        return EX_CONFIG;
+        return EXIT_FAILURE;
     }
 
-    openlog("postlicyd", LOG_PID, LOG_MAIL);
-    job_initialize();
-    syslog(LOG_INFO, "Starting...");
-
-    job_loop();
-
-    cleanexit = true;
-    shutdown();
-    return 0;
+    main_initialize();
+    main_loop();
+    main_shutdown();
+    return EXIT_SUCCESS;
 }