git.meshlink.io Git - catta/blob - avahi/fdutil.c

   1 /***
   2   This file is part of avahi.
   3
   4   avahi is free software; you can redistribute it and/or modify it
   5   under the terms of the GNU Lesser General Public License as
   6   published by the Free Software Foundation; either version 2.1 of the
   7   License, or (at your option) any later version.
   8
   9   avahi is distributed in the hope that it will be useful, but WITHOUT
  10   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  12   Public License for more details.
  13
  14   You should have received a copy of the GNU Lesser General Public
  15   License along with avahi; if not, write to the Free Software
  16   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  17   USA.
  18 ***/
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include <config.h>
  22 #endif
  23
  24 #include <sys/select.h>
  25 #include <unistd.h>
  26 #include <fcntl.h>
  27 #include <assert.h>
  28
  29 #include "fdutil.h"
  30
  31 int avahi_set_cloexec(int fd) {
  32     int n;
  33
  34     assert(fd >= 0);
  35
  36     if ((n = fcntl(fd, F_GETFD)) < 0)
  37         return -1;
  38
  39     if (n & FD_CLOEXEC)
  40         return 0;
  41
  42     return fcntl(fd, F_SETFD, n|FD_CLOEXEC);
  43 }
  44
  45 int avahi_set_nonblock(int fd) {
  46     int n;
  47
  48     assert(fd >= 0);
  49
  50     if ((n = fcntl(fd, F_GETFL)) < 0)
  51         return -1;
  52
  53     if (n & O_NONBLOCK)
  54         return 0;
  55
  56     return fcntl(fd, F_SETFL, n|O_NONBLOCK);
  57 }
  58
  59 int avahi_wait_for_write(int fd) {
  60     fd_set fds;
  61     int r;
  62
  63     FD_ZERO(&fds);
  64     FD_SET(fd, &fds);
  65
  66     if ((r = select(fd+1, NULL, &fds, NULL, NULL)) < 0)
  67         return -1;
  68
  69     assert(r > 0);
  70
  71     return 0;
  72 }