1 #include <darwintest.h>
2 #include <arpa/inet.h>
3 #include <sys/types.h>
4 #include <sys/event.h>
5 #include <sys/time.h>
6 #include <sys/socket.h>
7 #include <netinet/in.h>
8 #include <string.h>
9 #include <stdio.h>
10 #include <unistd.h>
11
12 T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true));
13
14 #define TEST_ADDR "127.0.0.1"
15 #define TEST_PORT 4242
16
17 static struct {
18 int fd;
19 struct sockaddr_in addr;
20 } server;
21
22 static void
server_listen(void)23 server_listen(void)
24 {
25 int r;
26
27 server.fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
28 T_ASSERT_POSIX_SUCCESS(server.fd, "socket");
29
30 memset(&server.addr, 0, sizeof(server.addr));
31 server.addr.sin_family = AF_INET;
32 server.addr.sin_port = htons(TEST_PORT);
33
34 inet_pton(AF_INET, TEST_ADDR, &server.addr.sin_addr);
35
36 r = bind(server.fd, (struct sockaddr*) &server.addr, sizeof(server.addr));
37 T_ASSERT_POSIX_SUCCESS(r, "bind");
38 }
39
40 static void
send_message(void)41 send_message(void)
42 {
43 int fd;
44 struct msghdr msg;
45 struct iovec iov;
46
47 fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
48 T_ASSERT_POSIX_SUCCESS(fd, "socket");
49
50 memset(&msg, 0, sizeof(msg));
51
52 msg.msg_name = &server.addr;
53 msg.msg_namelen = sizeof(server.addr);
54
55 iov.iov_base = "";
56 iov.iov_len = 0;
57 msg.msg_iov = &iov;
58 msg.msg_iovlen = 1;
59
60 ssize_t r = sendmsg(fd, &msg, 0);
61 T_ASSERT_EQ(r, (ssize_t)iov.iov_len, "sendmsg");
62
63 close(fd);
64 }
65
66 static void
server_poll(void)67 server_poll(void)
68 {
69 int kq;
70 struct kevent event = {
71 .flags = EV_ADD,
72 .filter = EVFILT_READ,
73 .ident = (unsigned long)server.fd,
74 };
75 int r;
76
77 kq = kqueue();
78 T_ASSERT_POSIX_SUCCESS(kq, "kqueue");
79
80 /* Add and poll */
81 r = kevent(kq, &event, 1, &event, 1, NULL);
82 T_EXPECT_EQ(r, 1, "should return an event");
83
84 close(kq);
85 }
86
87 T_DECL(socket_0byte_udp_poll_58140856,
88 "Tests that 0-sized UDP packets wake up kevent")
89 {
90 T_LOG("Starting...\n");
91
92 /* Listen on UDP port */
93 server_listen();
94
95 T_LOG("Server bound to [%s]:%d\n", TEST_ADDR, TEST_PORT);
96
97 /* Send 0-UDP packet to that port */
98 send_message();
99
100 T_LOG("Sent message to server\n");
101
102 /* Poll kqueue events */
103 server_poll();
104
105 T_LOG("Got kqueue event\n");
106
107 close(server.fd);
108 }
109