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