1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/* Smoke test for MSG_CMSG_CLOEXEC.
Copyright (C) 2021-2024 Free Software Foundation, Inc.
This file is part of the GNU C Library.
The GNU C Library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
The GNU C Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with the GNU C Library; if not, see
<https://www.gnu.org/licenses/>. */
#include <support/xunistd.h>
#include <support/check.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <string.h>
#include <fcntl.h>
static void
send_fd (int sockfd, int fd)
{
char data[] = "hello";
struct iovec iov = { .iov_base = data, .iov_len = sizeof (data) };
union
{
struct cmsghdr header;
char bytes[CMSG_SPACE (sizeof (fd))];
} cmsg_storage;
struct msghdr msg =
{
.msg_iov = &iov,
.msg_iovlen = 1,
.msg_control = cmsg_storage.bytes,
.msg_controllen = sizeof (cmsg_storage)
};
memset (&cmsg_storage, 0, sizeof (cmsg_storage));
struct cmsghdr *cmsg = CMSG_FIRSTHDR (&msg);
cmsg->cmsg_level = SOL_SOCKET;
cmsg->cmsg_type = SCM_RIGHTS;
cmsg->cmsg_len = CMSG_LEN (sizeof (fd));
memcpy (CMSG_DATA (cmsg), &fd, sizeof (fd));
ssize_t nsent = sendmsg (sockfd, &msg, 0);
if (nsent < 0)
FAIL_EXIT1 ("sendmsg (%d): %m", sockfd);
TEST_COMPARE (nsent, sizeof (data));
}
static int
recv_fd (int sockfd, int flags)
{
char buffer[100];
struct iovec iov = { .iov_base = buffer, .iov_len = sizeof (buffer) };
union
{
struct cmsghdr header;
char bytes[100];
} cmsg_storage;
struct msghdr msg =
{
.msg_iov = &iov,
.msg_iovlen = 1,
.msg_control = cmsg_storage.bytes,
.msg_controllen = sizeof (cmsg_storage)
};
ssize_t nrecv = recvmsg (sockfd, &msg, flags);
if (nrecv < 0)
FAIL_EXIT1 ("recvmsg (%d): %m", sockfd);
TEST_COMPARE (msg.msg_controllen, CMSG_SPACE (sizeof (int)));
struct cmsghdr *cmsg = CMSG_FIRSTHDR (&msg);
TEST_COMPARE (cmsg->cmsg_level, SOL_SOCKET);
TEST_COMPARE (cmsg->cmsg_type, SCM_RIGHTS);
TEST_COMPARE (cmsg->cmsg_len, CMSG_LEN (sizeof (int)));
int fd;
memcpy (&fd, CMSG_DATA (cmsg), sizeof (fd));
return fd;
}
static int
do_test (void)
{
int sockfd[2];
int newfd;
int flags;
int rc = socketpair (AF_UNIX, SOCK_DGRAM | SOCK_CLOEXEC, 0, sockfd);
if (rc < 0)
FAIL_EXIT1 ("socketpair: %m");
send_fd (sockfd[1], STDIN_FILENO);
newfd = recv_fd (sockfd[0], 0);
TEST_VERIFY_EXIT (newfd > 0);
flags = fcntl (newfd, F_GETFD, 0);
TEST_VERIFY_EXIT (flags != -1);
TEST_VERIFY (!(flags & FD_CLOEXEC));
xclose (newfd);
send_fd (sockfd[1], STDIN_FILENO);
newfd = recv_fd (sockfd[0], MSG_CMSG_CLOEXEC);
TEST_VERIFY_EXIT (newfd > 0);
flags = fcntl (newfd, F_GETFD, 0);
TEST_VERIFY_EXIT (flags != -1);
TEST_VERIFY (flags & FD_CLOEXEC);
xclose (newfd);
xclose (sockfd[0]);
xclose (sockfd[1]);
return 0;
}
#include <support/test-driver.c>
|
