job.c

   1 /******************************************************************************/
   2 /*          postlicyd: a postfix policy daemon with a lot of features         */
   3 /*          ~~~~~~~~~                                                         */
   4 /*  ________________________________________________________________________  */
   5 /*                                                                            */
   6 /*  Redistribution and use in source and binary forms, with or without        */
   7 /*  modification, are permitted provided that the following conditions        */
   8 /*  are met:                                                                  */
   9 /*                                                                            */
  10 /*  1. Redistributions of source code must retain the above copyright         */
  11 /*     notice, this list of conditions and the following disclaimer.          */
  12 /*  2. Redistributions in binary form must reproduce the above copyright      */
  13 /*     notice, this list of conditions and the following disclaimer in the    */
  14 /*     documentation and/or other materials provided with the distribution.   */
  15 /*  3. The names of its contributors may not be used to endorse or promote    */
  16 /*     products derived from this software without specific prior written     */
  17 /*     permission.                                                            */
  18 /*                                                                            */
  19 /*  THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND   */
  20 /*  ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE     */
  21 /*  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR        */
  22 /*  PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS    */
  23 /*  BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR    */
  24 /*  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF      */
  25 /*  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS  */
  26 /*  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN   */
  27 /*  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)   */
  28 /*  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF    */
  29 /*  THE POSSIBILITY OF SUCH DAMAGE.                                           */
  30 /******************************************************************************/
  31
  32 /*
  33  * Copyright © 2007 Pierre Habouzit
  34  */
  35
  36 #include <fcntl.h>
  37 #include <stdbool.h>
  38 #include <sys/epoll.h>
  39 #include <sys/socket.h>
  40 #include <sys/types.h>
  41 #include <unistd.h>
  42
  43 #ifndef EPOLLRDHUP
  44 #  include <linux/poll.h>
  45 #  ifdef POLLRDHUP
  46 #    define EPOLLRDHUP POLLRDHUP
  47 #  else
  48 #    define EPOLLRDHUP 0
  49 #  endif
  50 #endif
  51
  52
  53 #include "job.h"
  54
  55 static int epollfd;
  56
  57 static void job_wipe(job_t *job)
  58 {
  59     if (job->fd >= 0) {
  60         close(job->fd);
  61         job->fd = -1;
  62     }
  63 }
  64 DO_DELETE(job_t, job);
  65
  66 void job_release(job_t **job)
  67 {
  68     if (*job) {
  69         if ((*job)->stop) {
  70             (*job)->stop(*job);
  71         }
  72         job_delete(job);
  73     }
  74 }
  75
  76 static job_t *job_register_fd(job_t *job)
  77 {
  78     struct epoll_event event = { .data.ptr = job, .events = EPOLLRDHUP };
  79
  80     if (job->state & JOB_READ || job->state & JOB_LISTEN) {
  81         event.events |= EPOLLIN;
  82     }
  83
  84     if (job->state & JOB_WRITE || job->state & JOB_CONN) {
  85         event.events |= EPOLLIN;
  86     }
  87
  88     if (epoll_ctl(epollfd, EPOLL_CTL_ADD, job->fd, &event) < 0) {
  89         job->error = true;
  90         job_release(&job);
  91     }
  92
  93     return job;
  94 }
  95
  96 void job_update_events(job_t *job)
  97 {
  98     struct epoll_event event = { .data.ptr = job, .events = EPOLLRDHUP };
  99
 100     if (job->state & JOB_READ || job->state & JOB_LISTEN) {
 101         event.events |= EPOLLIN;
 102     }
 103
 104     if (job->state & JOB_WRITE || job->state & JOB_CONN) {
 105         event.events |= EPOLLIN;
 106     }
 107
 108     epoll_ctl(epollfd, EPOLL_CTL_MOD, job->fd, &event);
 109 }
 110
 111 job_t *job_accept(job_t *listener, int state)
 112 {
 113     int sock;
 114     job_t *res;
 115
 116     if ((sock = accept(listener->fd, NULL, 0)) < 0) {
 117         return NULL;
 118     }
 119
 120     if (fcntl(sock, F_SETFL, fcntl(sock, F_GETFL) | O_NONBLOCK)) {
 121         return NULL;
 122     }
 123
 124     res          = job_new();
 125     res->fd      = sock;
 126     res->state   = state;
 127     res->process = listener->process;
 128     res->stop    = listener->stop;
 129     return job_register_fd(res);
 130 }