1 /* $OpenBSD: unfdpassfail.c,v 1.2 2021/12/15 21:25:55 bluhm Exp $ */
2
3 /*
4 * Copyright (c) 2021 Vitaliy Makkoveev <mvs@openbsd.org>
5 *
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 */
18
19 #include <sys/types.h>
20 #include <sys/event.h>
21 #include <sys/socket.h>
22 #include <sys/un.h>
23 #include <stdio.h>
24 #include <err.h>
25 #include <errno.h>
26 #include <fcntl.h>
27 #include <pthread.h>
28 #include <string.h>
29 #include <time.h>
30 #include <unistd.h>
31
32 union msg_control{
33 struct cmsghdr cmsgh;
34 char control[CMSG_SPACE(sizeof(int)*2)];
35 };
36
thr_close(void * arg)37 static void *thr_close(void *arg)
38 {
39 close(*(int *)arg);
40 return NULL;
41 }
42
main(int argc,char * argv[])43 int main(int argc, char *argv[])
44 {
45 struct timespec ts_start, ts_now, ts_time;
46 union msg_control msg_control;
47 int iov_buf;
48 struct iovec iov;
49 struct msghdr msgh;
50 struct cmsghdr *cmsgh;
51 pthread_t thr;
52 int s[2], fd, kqfd;
53 int infinite = 0, error;
54
55 if (argc > 1 && !strcmp(argv[1], "--infinite"))
56 infinite = 1;
57
58 if ((kqfd = kqueue()) < 0)
59 err(1, "kqueue");
60
61 if (!infinite)
62 if (clock_gettime(CLOCK_BOOTTIME, &ts_start) <0)
63 err(1, "clock_gettime");
64
65 while (1) {
66 if (socketpair(AF_UNIX, SOCK_STREAM|O_NONBLOCK, 0, s) < 0)
67 err(1, "socketpair");
68 if ((fd = socket(AF_UNIX, SOCK_DGRAM, 0)) < 0)
69 err(1, "open");
70
71 iov_buf = 0;
72 iov.iov_base = &iov_buf;
73 iov.iov_len = sizeof(iov_buf);
74 msgh.msg_control = msg_control.control;
75 msgh.msg_controllen = sizeof(msg_control.control);
76 msgh.msg_iov = &iov;
77 msgh.msg_iovlen = 1;
78 msgh.msg_name = NULL;
79 msgh.msg_namelen = 0;
80 cmsgh = CMSG_FIRSTHDR(&msgh);
81 cmsgh->cmsg_len = CMSG_LEN(sizeof(int) * 2);
82 cmsgh->cmsg_level = SOL_SOCKET;
83 cmsgh->cmsg_type = SCM_RIGHTS;
84 *((int *)CMSG_DATA(cmsgh) + 0) = fd;
85 *((int *)CMSG_DATA(cmsgh) + 1) = kqfd;
86
87 error = pthread_create(&thr, NULL, thr_close, &fd);
88 if (error)
89 errc(1, error, "pthread_create");
90
91 if (sendmsg(s[0], &msgh, 0) < 0) {
92 if (errno != EINVAL)
93 err(1, "sendmsg");
94 }
95
96 error = pthread_join(thr, NULL);
97 if (error)
98 errc(1, error, "pthread_join");
99
100 close(s[0]);
101 close(s[1]);
102 close(fd);
103
104 if (!infinite) {
105 if (clock_gettime(CLOCK_BOOTTIME, &ts_now) <0)
106 err(1, "clock_gettime");
107
108 timespecsub(&ts_now, &ts_start, &ts_time);
109 if (ts_time.tv_sec >= 20)
110 break;
111 }
112 }
113
114 close(kqfd);
115
116 return 0;
117 }
118