|
Packit |
6c4009 |
/* Test the accept4 function with differing flags arguments.
|
|
Packit |
6c4009 |
Copyright (C) 2017-2018 Free Software Foundation, Inc.
|
|
Packit |
6c4009 |
This file is part of the GNU C Library.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
The GNU C Library is free software; you can redistribute it and/or
|
|
Packit |
6c4009 |
modify it under the terms of the GNU Lesser General Public
|
|
Packit |
6c4009 |
License as published by the Free Software Foundation; either
|
|
Packit |
6c4009 |
version 2.1 of the License, or (at your option) any later version.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
The GNU C Library is distributed in the hope that it will be useful,
|
|
Packit |
6c4009 |
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
6c4009 |
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Packit |
6c4009 |
Lesser General Public License for more details.
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
You should have received a copy of the GNU Lesser General Public
|
|
Packit |
6c4009 |
License along with the GNU C Library; if not, see
|
|
Packit |
6c4009 |
<http://www.gnu.org/licenses/>. */
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#include <arpa/inet.h>
|
|
Packit |
6c4009 |
#include <errno.h>
|
|
Packit |
6c4009 |
#include <fcntl.h>
|
|
Packit |
6c4009 |
#include <stdbool.h>
|
|
Packit |
6c4009 |
#include <support/check.h>
|
|
Packit |
6c4009 |
#include <support/xsocket.h>
|
|
Packit |
6c4009 |
#include <support/xunistd.h>
|
|
Packit |
6c4009 |
#include <sys/socket.h>
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static bool
|
|
Packit |
6c4009 |
is_nonblocking (int fd)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
int status = fcntl (fd, F_GETFL);
|
|
Packit |
6c4009 |
if (status < 0)
|
|
Packit |
6c4009 |
FAIL_EXIT1 ("fcntl (F_GETFL): %m");
|
|
Packit |
6c4009 |
return status & O_NONBLOCK;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static bool
|
|
Packit |
6c4009 |
is_cloexec (int fd)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
int status = fcntl (fd, F_GETFD);
|
|
Packit |
6c4009 |
if (status < 0)
|
|
Packit |
6c4009 |
FAIL_EXIT1 ("fcntl (F_GETFD): %m");
|
|
Packit |
6c4009 |
return status & FD_CLOEXEC;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
struct client
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
int socket;
|
|
Packit |
6c4009 |
struct sockaddr_in address;
|
|
Packit |
6c4009 |
};
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
/* Perform a non-blocking connect to *SERVER_ADDRESS. */
|
|
Packit |
6c4009 |
static struct client
|
|
Packit |
6c4009 |
client_connect (const struct sockaddr_in *server_address)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
struct client result;
|
|
Packit |
6c4009 |
result.socket = xsocket (AF_INET,
|
|
Packit |
6c4009 |
SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, 0);
|
|
Packit |
6c4009 |
TEST_VERIFY (is_nonblocking (result.socket));
|
|
Packit |
6c4009 |
TEST_VERIFY (is_cloexec (result.socket));
|
|
Packit |
6c4009 |
int ret = connect (result.socket, (const struct sockaddr *) server_address,
|
|
Packit |
6c4009 |
sizeof (*server_address));
|
|
Packit |
6c4009 |
if (ret < 0 && errno != EINPROGRESS)
|
|
Packit |
6c4009 |
FAIL_EXIT1 ("client connect: %m");
|
|
Packit |
6c4009 |
socklen_t sa_len = sizeof (result.address);
|
|
Packit |
6c4009 |
xgetsockname (result.socket, (struct sockaddr *) &result.address,
|
|
Packit |
6c4009 |
&sa_len);
|
|
Packit |
6c4009 |
TEST_VERIFY (sa_len == sizeof (result.address));
|
|
Packit |
6c4009 |
return result;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static void
|
|
Packit |
6c4009 |
check_same_address (const struct sockaddr_in *left,
|
|
Packit |
6c4009 |
const struct sockaddr_in *right)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
TEST_VERIFY (left->sin_family == AF_INET);
|
|
Packit |
6c4009 |
TEST_VERIFY (right->sin_family == AF_INET);
|
|
Packit |
6c4009 |
TEST_VERIFY (left->sin_addr.s_addr == right->sin_addr.s_addr);
|
|
Packit |
6c4009 |
TEST_VERIFY (left->sin_port == right->sin_port);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
static int
|
|
Packit |
6c4009 |
do_test (void)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
/* Create server socket. */
|
|
Packit |
6c4009 |
int server_socket = xsocket (AF_INET, SOCK_STREAM, 0);
|
|
Packit |
6c4009 |
TEST_VERIFY (!is_nonblocking (server_socket));
|
|
Packit |
6c4009 |
TEST_VERIFY (!is_cloexec (server_socket));
|
|
Packit |
6c4009 |
struct sockaddr_in server_address =
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
.sin_family = AF_INET,
|
|
Packit |
6c4009 |
.sin_addr = {.s_addr = htonl (INADDR_LOOPBACK) },
|
|
Packit |
6c4009 |
};
|
|
Packit |
6c4009 |
xbind (server_socket,
|
|
Packit |
6c4009 |
(struct sockaddr *) &server_address, sizeof (server_address));
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
socklen_t sa_len = sizeof (server_address);
|
|
Packit |
6c4009 |
xgetsockname (server_socket, (struct sockaddr *) &server_address,
|
|
Packit |
6c4009 |
&sa_len);
|
|
Packit |
6c4009 |
TEST_VERIFY (sa_len == sizeof (server_address));
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
xlisten (server_socket, 5);
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
for (int do_nonblock = 0; do_nonblock < 2; ++do_nonblock)
|
|
Packit |
6c4009 |
for (int do_cloexec = 0; do_cloexec < 2; ++do_cloexec)
|
|
Packit |
6c4009 |
{
|
|
Packit |
6c4009 |
int sockflags = 0;
|
|
Packit |
6c4009 |
if (do_nonblock)
|
|
Packit |
6c4009 |
sockflags |= SOCK_NONBLOCK;
|
|
Packit |
6c4009 |
if (do_cloexec)
|
|
Packit |
6c4009 |
sockflags |= SOCK_CLOEXEC;
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
struct client client = client_connect (&server_address);
|
|
Packit |
6c4009 |
struct sockaddr_in client_address;
|
|
Packit |
6c4009 |
socklen_t sa_len = sizeof (client_address);
|
|
Packit |
6c4009 |
int client_socket = xaccept4 (server_socket,
|
|
Packit |
6c4009 |
(struct sockaddr *) &client_address,
|
|
Packit |
6c4009 |
&sa_len, sockflags);
|
|
Packit |
6c4009 |
TEST_VERIFY (sa_len == sizeof (client_address));
|
|
Packit |
6c4009 |
TEST_VERIFY (is_nonblocking (client_socket) == do_nonblock);
|
|
Packit |
6c4009 |
TEST_VERIFY (is_cloexec (client_socket) == do_cloexec);
|
|
Packit |
6c4009 |
check_same_address (&client.address, &client_address);
|
|
Packit |
6c4009 |
xclose (client_socket);
|
|
Packit |
6c4009 |
xclose (client.socket);
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
xclose (server_socket);
|
|
Packit |
6c4009 |
return 0;
|
|
Packit |
6c4009 |
}
|
|
Packit |
6c4009 |
|
|
Packit |
6c4009 |
#include <support/test-driver.c>
|