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
31int 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
45int 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
59int 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