1 /*
2  *  OpenVPN -- An application to securely tunnel IP networks
3  *             over a single UDP port, with support for SSL/TLS-based
4  *             session authentication and key exchange,
5  *             packet encryption, packet authentication, and
6  *             packet compression.
7  *
8  *  Copyright (C) 2002-2022 OpenVPN Inc <sales@openvpn.net>
9  *
10  *  This program is free software; you can redistribute it and/or modify
11  *  it under the terms of the GNU General Public License version 2
12  *  as published by the Free Software Foundation.
13  *
14  *  This program is distributed in the hope that it will be useful,
15  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
16  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  *  GNU General Public License for more details.
18  *
19  *  You should have received a copy of the GNU General Public License along
20  *  with this program; if not, write to the Free Software Foundation, Inc.,
21  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
22  */
23 
24 #ifdef HAVE_CONFIG_H
25 #include "config.h"
26 #elif defined(_MSC_VER)
27 #include "config-msvc.h"
28 #endif
29 
30 #include "syshead.h"
31 
32 #include "fdmisc.h"
33 #include "error.h"
34 
35 #include "memdbg.h"
36 
37 /* Set a file descriptor to non-blocking */
38 bool
set_nonblock_action(int fd)39 set_nonblock_action(int fd)
40 {
41 #ifdef _WIN32
42     u_long arg = 1;
43     if (ioctlsocket(fd, FIONBIO, &arg))
44     {
45         return false;
46     }
47 #else  /* ifdef _WIN32 */
48     if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0)
49     {
50         return false;
51     }
52 #endif
53     return true;
54 }
55 
56 /* Set a file descriptor to not be passed across execs */
57 bool
set_cloexec_action(int fd)58 set_cloexec_action(int fd)
59 {
60 #ifndef _WIN32
61     if (fcntl(fd, F_SETFD, FD_CLOEXEC) < 0)
62     {
63         return false;
64     }
65 #endif
66     return true;
67 }
68 
69 /* Set a file descriptor to non-blocking */
70 void
set_nonblock(int fd)71 set_nonblock(int fd)
72 {
73     if (!set_nonblock_action(fd))
74     {
75         msg(M_ERR, "Set socket to non-blocking mode failed");
76     }
77 }
78 
79 /* Set a file descriptor to not be passed across execs */
80 void
set_cloexec(int fd)81 set_cloexec(int fd)
82 {
83     if (!set_cloexec_action(fd))
84     {
85         msg(M_ERR, "Set FD_CLOEXEC flag on file descriptor failed");
86     }
87 }
88