daemon.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 <sys/un.h>
  37 #include <fcntl.h>
  38
  39 #include "common.h"
  40 #include "daemon.h"
  41
  42 static int setnonblock(int sock)
  43 {
  44     int res = fcntl(sock, F_GETFL);
  45
  46     if (res < 0) {
  47         UNIXERR("fcntl");
  48         return -1;
  49     }
  50
  51     if (fcntl(sock, F_SETFL, res | O_NONBLOCK) < 0) {
  52         UNIXERR("fcntl");
  53         return -1;
  54     }
  55
  56     return 0;
  57 }
  58
  59
  60 int tcp_listen(const struct sockaddr *addr, socklen_t len)
  61 {
  62     int sock;
  63
  64     switch (addr->sa_family) {
  65       case AF_UNIX:
  66         unlink(((struct sockaddr_un *)addr)->sun_path);
  67         sock = socket(PF_UNIX, SOCK_STREAM, 0);
  68         break;
  69       case AF_INET:
  70         sock = socket(PF_INET, SOCK_STREAM, 0);
  71         break;
  72       case AF_INET6:
  73         sock = socket(PF_INET6, SOCK_STREAM, 0);
  74         break;
  75       default:
  76         errno = EINVAL;
  77         return -1;
  78     }
  79
  80     if (sock < 0) {
  81         UNIXERR("socket");
  82         return -1;
  83     }
  84
  85     if (addr->sa_family != AF_UNIX) {
  86         int v = 1;
  87         if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &v, sizeof(v)) < 0) {
  88             UNIXERR("setsockopt(SO_REUSEADDR)");
  89             close(sock);
  90             return -1;
  91         }
  92     }
  93
  94     if (bind(sock, addr, len) < 0) {
  95         UNIXERR("bind");
  96         close(sock);
  97         return -1;
  98     }
  99
 100     if (setnonblock(sock)) {
 101         close(sock);
 102         return -1;
 103     }
 104
 105     if (listen(sock, 0) < 0) {
 106         UNIXERR("bind");
 107         close(sock);
 108         return -1;
 109     }
 110
 111     return sock;
 112 }
 113
 114 int accept_nonblock(int fd)
 115 {
 116     int sock = accept(fd, NULL, 0);
 117
 118     if (sock < 0) {
 119         UNIXERR("accept");
 120         return -1;
 121     }
 122
 123     if (setnonblock(sock)) {
 124         close(sock);
 125         return -1;
 126     }
 127
 128     return sock;
 129 }