X-Git-Url: http://git.madism.org/?a=blobdiff_plain;f=job.c;h=3d99a8f83dea8171c98979905a2d7b1d367d5693;hb=0bfa9f900af9a4f4e65f3f580056de28380b333a;hp=fc0c1b84574b31a1aa9daf913edf0e6596de38b6;hpb=b1315f7af90c4d8047cf0c8f3c0a00b703676f84;p=apps%2Fpfixtools.git

diff --git a/job.c b/job.c
index fc0c1b8..3d99a8f 100644
--- a/job.c
+++ b/job.c
@@ -33,5 +33,181 @@
  * Copyright © 2007 Pierre Habouzit
  */
 
+#include <errno.h>
+#include <fcntl.h>
+#include <signal.h>
+#include <stdbool.h>
+#include <syslog.h>
+#include <sysexits.h>
+#include <sys/epoll.h>
+#include <sys/socket.h>
+#include <sys/types.h>
+#include <time.h>
+#include <unistd.h>
+
+#ifndef EPOLLRDHUP
+#  include <linux/poll.h>
+#  ifdef POLLRDHUP
+#    define EPOLLRDHUP POLLRDHUP
+#  else
+#    define EPOLLRDHUP 0
+#  endif
+#endif
+
 #include "job.h"
 
+static int epollfd = -1;
+static bool sigint = false;
+
+void job_delete(job_t **job)
+{
+    if (*job) {
+        if ((*job)->stop) {
+            (*job)->stop(*job);
+        }
+        if ((*job)->fd >= 0) {
+            close((*job)->fd);
+        }
+        p_delete(job);
+    }
+}
+
+static job_t *job_register_fd(job_t *job)
+{
+    struct epoll_event event = { .data.ptr = job, .events = EPOLLRDHUP };
+
+    if (job->mode & (JOB_READ | JOB_LISTEN)) {
+        event.events |= EPOLLIN;
+    }
+
+    if (job->mode & (JOB_WRITE | JOB_CONN)) {
+        event.events |= EPOLLOUT;
+    }
+
+    if (epoll_ctl(epollfd, EPOLL_CTL_ADD, job->fd, &event) < 0) {
+        syslog(LOG_ERR, "epoll_ctl error: %m");
+        job->error = true;
+        job_delete(&job);
+    }
+
+    return job;
+}
+
+void job_update_mode(job_t *job, int mode)
+{
+    struct epoll_event event = { .data.ptr = job, .events = EPOLLRDHUP };
+
+    if (job->mode == mode)
+        return;
+
+    job->mode = mode;
+    if (job->mode & (JOB_READ | JOB_LISTEN)) {
+        event.events |= EPOLLIN;
+    }
+
+    if (job->mode & (JOB_WRITE | JOB_CONN)) {
+        event.events |= EPOLLOUT;
+    }
+
+    if (epoll_ctl(epollfd, EPOLL_CTL_MOD, job->fd, &event) < 0) {
+        syslog(LOG_ERR, "epoll_ctl error: %m");
+        job->error = true;
+    }
+}
+
+job_t *job_accept(job_t *listener, int mode)
+{
+    int sock;
+    job_t *res;
+
+    if ((sock = accept(listener->fd, NULL, 0)) < 0) {
+        syslog(LOG_ERR, "accept error: %m");
+        return NULL;
+    }
+
+    if (fcntl(sock, F_SETFL, fcntl(sock, F_GETFL) | O_NONBLOCK)) {
+        syslog(LOG_ERR, "fcntl error: %m");
+        return NULL;
+    }
+
+    res          = job_new();
+    res->fd      = sock;
+    res->mode    = mode;
+    res->process = listener->process;
+    res->stop    = listener->stop;
+    return job_register_fd(res);
+}
+
+static void job_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);
+}
+
+void job_initialize(void)
+{
+    signal(SIGPIPE, SIG_IGN);
+    signal(SIGINT,  &job_sighandler);
+    signal(SIGTERM, &job_sighandler);
+
+    epollfd = epoll_create(128);
+    if (epollfd < 0) {
+        syslog(LOG_ERR, "epoll_create error: %m");
+        exit(EX_OSERR);
+    }
+}
+
+void job_loop(void)
+{
+    while (!sigint) {
+        struct epoll_event events[FD_SETSIZE];
+        int todo = epoll_wait(epollfd, events, countof(events), -1);
+
+        if (todo < 0) {
+            if (errno == EAGAIN || errno == EINTR)
+                continue;
+            syslog(LOG_ERR, "epoll_wait error: %m");
+            exit(EX_OSERR);
+        }
+
+        while (todo) {
+            job_t *job = events[--todo].data.ptr;
+
+            assert (job->process);
+            job->process(job);
+
+            if (job->error || job->done) {
+                job_delete(&job);
+            }
+        }
+    }
+}
+
+void job_shutdown(void)
+{
+    if (epollfd >= 0) {
+        close(epollfd);
+        epollfd = -1;
+    }
+}