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 
avahi_set_cloexec(int fd)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 
avahi_set_nonblock(int fd)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 
avahi_wait_for_write(int fd)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 }
73