tests/socket_0byte_udp_poll_58140856.c

   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
  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
  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
  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 }