]>
Commit | Line | Data |
---|---|---|
e9203023 | 1 | /* Test the accept4 function with differing flags arguments. |
d614a753 | 2 | Copyright (C) 2017-2020 Free Software Foundation, Inc. |
e9203023 FW |
3 | This file is part of the GNU C Library. |
4 | ||
5 | The GNU C Library is free software; you can redistribute it and/or | |
6 | modify it under the terms of the GNU Lesser General Public | |
7 | License as published by the Free Software Foundation; either | |
8 | version 2.1 of the License, or (at your option) any later version. | |
9 | ||
10 | The GNU C Library is distributed in the hope that it will be useful, | |
11 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
13 | Lesser General Public License for more details. | |
14 | ||
15 | You should have received a copy of the GNU Lesser General Public | |
16 | License along with the GNU C Library; if not, see | |
5a82c748 | 17 | <https://www.gnu.org/licenses/>. */ |
e9203023 FW |
18 | |
19 | #include <arpa/inet.h> | |
20 | #include <errno.h> | |
21 | #include <fcntl.h> | |
22 | #include <stdbool.h> | |
23 | #include <support/check.h> | |
24 | #include <support/xsocket.h> | |
25 | #include <support/xunistd.h> | |
26 | #include <sys/socket.h> | |
27 | ||
28 | static bool | |
29 | is_nonblocking (int fd) | |
30 | { | |
31 | int status = fcntl (fd, F_GETFL); | |
32 | if (status < 0) | |
33 | FAIL_EXIT1 ("fcntl (F_GETFL): %m"); | |
34 | return status & O_NONBLOCK; | |
35 | } | |
36 | ||
37 | static bool | |
38 | is_cloexec (int fd) | |
39 | { | |
40 | int status = fcntl (fd, F_GETFD); | |
41 | if (status < 0) | |
42 | FAIL_EXIT1 ("fcntl (F_GETFD): %m"); | |
43 | return status & FD_CLOEXEC; | |
44 | } | |
45 | ||
46 | struct client | |
47 | { | |
48 | int socket; | |
49 | struct sockaddr_in address; | |
50 | }; | |
51 | ||
52 | /* Perform a non-blocking connect to *SERVER_ADDRESS. */ | |
53 | static struct client | |
54 | client_connect (const struct sockaddr_in *server_address) | |
55 | { | |
56 | struct client result; | |
57 | result.socket = xsocket (AF_INET, | |
58 | SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, 0); | |
59 | TEST_VERIFY (is_nonblocking (result.socket)); | |
60 | TEST_VERIFY (is_cloexec (result.socket)); | |
61 | int ret = connect (result.socket, (const struct sockaddr *) server_address, | |
62 | sizeof (*server_address)); | |
63 | if (ret < 0 && errno != EINPROGRESS) | |
64 | FAIL_EXIT1 ("client connect: %m"); | |
65 | socklen_t sa_len = sizeof (result.address); | |
66 | xgetsockname (result.socket, (struct sockaddr *) &result.address, | |
67 | &sa_len); | |
68 | TEST_VERIFY (sa_len == sizeof (result.address)); | |
69 | return result; | |
70 | } | |
71 | ||
72 | static void | |
73 | check_same_address (const struct sockaddr_in *left, | |
74 | const struct sockaddr_in *right) | |
75 | { | |
76 | TEST_VERIFY (left->sin_family == AF_INET); | |
77 | TEST_VERIFY (right->sin_family == AF_INET); | |
78 | TEST_VERIFY (left->sin_addr.s_addr == right->sin_addr.s_addr); | |
79 | TEST_VERIFY (left->sin_port == right->sin_port); | |
80 | } | |
81 | ||
82 | static int | |
83 | do_test (void) | |
84 | { | |
85 | /* Create server socket. */ | |
86 | int server_socket = xsocket (AF_INET, SOCK_STREAM, 0); | |
87 | TEST_VERIFY (!is_nonblocking (server_socket)); | |
88 | TEST_VERIFY (!is_cloexec (server_socket)); | |
89 | struct sockaddr_in server_address = | |
90 | { | |
91 | .sin_family = AF_INET, | |
92 | .sin_addr = {.s_addr = htonl (INADDR_LOOPBACK) }, | |
93 | }; | |
94 | xbind (server_socket, | |
95 | (struct sockaddr *) &server_address, sizeof (server_address)); | |
96 | { | |
97 | socklen_t sa_len = sizeof (server_address); | |
98 | xgetsockname (server_socket, (struct sockaddr *) &server_address, | |
99 | &sa_len); | |
100 | TEST_VERIFY (sa_len == sizeof (server_address)); | |
101 | } | |
102 | xlisten (server_socket, 5); | |
103 | ||
104 | for (int do_nonblock = 0; do_nonblock < 2; ++do_nonblock) | |
105 | for (int do_cloexec = 0; do_cloexec < 2; ++do_cloexec) | |
106 | { | |
107 | int sockflags = 0; | |
108 | if (do_nonblock) | |
109 | sockflags |= SOCK_NONBLOCK; | |
110 | if (do_cloexec) | |
111 | sockflags |= SOCK_CLOEXEC; | |
112 | ||
113 | struct client client = client_connect (&server_address); | |
114 | struct sockaddr_in client_address; | |
115 | socklen_t sa_len = sizeof (client_address); | |
116 | int client_socket = xaccept4 (server_socket, | |
117 | (struct sockaddr *) &client_address, | |
118 | &sa_len, sockflags); | |
119 | TEST_VERIFY (sa_len == sizeof (client_address)); | |
120 | TEST_VERIFY (is_nonblocking (client_socket) == do_nonblock); | |
121 | TEST_VERIFY (is_cloexec (client_socket) == do_cloexec); | |
122 | check_same_address (&client.address, &client_address); | |
123 | xclose (client_socket); | |
124 | xclose (client.socket); | |
125 | } | |
126 | ||
127 | xclose (server_socket); | |
128 | return 0; | |
129 | } | |
130 | ||
131 | #include <support/test-driver.c> |