1 #include <stdlib.h>
2 #include <stdio.h>
3 #include <string.h>
4 #include <err.h>
5 #include <errno.h>
6 #include <pthread.h>
7 #include <fcntl.h>
8 #include <poll.h>
9 #include <sys/socket.h>
10 #include <darwintest.h>
11 /*
12 * Tests that a full UNIX domain socket buffer
13 * always reports the right poll() event.
14 */
15
16 static const int MSG1_LEN = 16384;
17
18 static void
do_recv(int sk,char * buf,size_t size)19 do_recv(int sk, char *buf, size_t size)
20 {
21 do{
22 struct iovec iov[1];
23 struct msghdr msg;
24
25 struct pollfd pfd[1] = { { sk, POLLIN, 0 } };
26 T_QUIET; T_ASSERT_POSIX_SUCCESS(poll(pfd, 1, -1), "poll");
27 memset(&msg, 0, sizeof(msg));
28 iov->iov_base = buf;
29 iov->iov_len = size;
30 msg.msg_iov = iov;
31 msg.msg_iovlen = 1;
32 ssize_t res = recvmsg(sk, &msg, 0);
33 T_QUIET; T_ASSERT_POSIX_SUCCESS(res, "recvmsg");
34 buf += res;
35 size -= (size_t)res;
36 } while (size);
37 }
38
39
40 static void *
receiver(void * arg)41 receiver(void *arg)
42 {
43 int sk = (int)arg;
44 char *buf = malloc(MSG1_LEN);
45 for (;;) {
46 do_recv(sk, buf, MSG1_LEN);
47 }
48 }
49
50 static void
do_send(int sk,char * buf,size_t size)51 do_send(int sk, char *buf, size_t size)
52 {
53 do{
54 struct iovec iov[1];
55 struct msghdr msg;
56
57 struct pollfd pfd[1] = { { sk, POLLOUT, 0 } };
58 int res = poll(pfd, 1, -1);
59 T_QUIET; T_ASSERT_POSIX_SUCCESS(res, "poll");
60 if (res == 0) {
61 continue;
62 }
63 if (!(pfd[0].revents & POLLOUT)) {
64 T_FAIL("POLLOUT not set");
65 }
66 memset(&msg, 0, sizeof(msg));
67 iov->iov_base = buf;
68 iov->iov_len = size;
69 msg.msg_iov = iov;
70 msg.msg_iovlen = 1;
71 ssize_t res_sendmsg = sendmsg(sk, &msg, 0);
72 T_QUIET; T_ASSERT_POSIX_SUCCESS(res_sendmsg, "sendmsg");
73 buf += res_sendmsg;
74 size -= (size_t)res_sendmsg;
75 } while (size);
76 }
77
78
79 static void
cfg_sk(int sk)80 cfg_sk(int sk)
81 {
82 int newSndBufSz = MSG1_LEN * 2;
83 socklen_t optLen = sizeof(newSndBufSz);
84 T_ASSERT_POSIX_SUCCESS(setsockopt(sk, SOL_SOCKET, SO_SNDBUF,
85 &newSndBufSz, optLen),
86 "setsockopt");
87 newSndBufSz = MSG1_LEN * 2;
88 optLen = sizeof(newSndBufSz);
89 T_ASSERT_POSIX_SUCCESS(setsockopt(sk, SOL_SOCKET, SO_RCVBUF,
90 &newSndBufSz, optLen),
91 "setsockopt");
92 int flags = fcntl(sk, F_GETFL, 0);
93 T_ASSERT_POSIX_SUCCESS(flags, "fcntl");
94 T_ASSERT_POSIX_SUCCESS(fcntl(sk, F_SETFL, flags | O_NONBLOCK),
95 "fcntl");
96 }
97
98 T_DECL(uipc_sosendcheck, "Tests the UNIX Domain poll filter", T_META_CHECK_LEAKS(false))
99 {
100 int s[2];
101 T_ASSERT_POSIX_SUCCESS(socketpair(AF_UNIX, SOCK_STREAM, 0, s),
102 "socketpair");
103 cfg_sk(s[0]);
104 cfg_sk(s[1]);
105 char *buf = malloc(MSG1_LEN);
106
107 pthread_t receiver_th;
108 if (pthread_create(&receiver_th, 0, receiver, (void *)(uintptr_t)s[1])) {
109 T_FAIL("pthread_create failed");
110 }
111
112 for (unsigned int i = 0; i < 90000; i++) {
113 do_send(s[0], buf, 5);
114 do_send(s[0], buf, MSG1_LEN);
115 }
116 }
117