1 /* $NetBSD: t_sendmmsg.c,v 1.2 2018/10/16 09:23:29 roy Exp $ */ 2 3 /*- 4 * Copyright (c) 2018 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by Christos Zoulas. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 * POSSIBILITY OF SUCH DAMAGE. 30 */ 31 #include <sys/cdefs.h> 32 __RCSID("$NetBSD: t_sendmmsg.c,v 1.2 2018/10/16 09:23:29 roy Exp $"); 33 34 #include <atf-c.h> 35 #include <sys/types.h> 36 #include <sys/socket.h> 37 #include <sys/ioctl.h> 38 #include <sys/wait.h> 39 40 #include <string.h> 41 #include <time.h> 42 #include <stdint.h> 43 #include <errno.h> 44 #include <signal.h> 45 #include <stdio.h> 46 #include <stdlib.h> 47 #include <unistd.h> 48 #include <sched.h> 49 50 #define BUFSIZE 65536 51 52 #define min(a, b) ((a) < (b) ? (a) : (b)) 53 static int debug = 1; 54 static volatile sig_atomic_t rdied; 55 56 static void 57 handle_sigchld(__unused int pid) 58 { 59 60 rdied = 1; 61 } 62 63 ATF_TC(sendmmsg_basic); 64 ATF_TC_HEAD(sendmmsg_basic, tc) 65 { 66 atf_tc_set_md_var(tc, "descr", "A basic test of sendmmsg(2)"); 67 } 68 69 static void 70 setsock(int fd, int type) 71 { 72 int buflen = BUFSIZE; 73 socklen_t socklen = sizeof(buflen); 74 75 ATF_REQUIRE_MSG(setsockopt(fd, SOL_SOCKET, type, 76 &buflen, socklen) != -1, "%s (%s)", 77 type == SO_RCVBUF ? "rcv" : "snd", strerror(errno)); 78 } 79 80 ATF_TC_BODY(sendmmsg_basic, tc) 81 { 82 int fd[2], error, cnt; 83 uint8_t *buf; 84 struct mmsghdr *mmsghdr; 85 struct iovec *iov; 86 unsigned int mmsgcnt, n; 87 int status; 88 off_t off; 89 uint8_t DGRAM[1316] = { 0, 2, 3, 4, 5, 6, 7, 8, 9, }; 90 uint8_t rgram[sizeof(DGRAM)]; 91 struct sigaction sa; 92 ssize_t overf = 0; 93 94 error = socketpair(AF_UNIX, SOCK_DGRAM, 0, fd); 95 ATF_REQUIRE_MSG(error != -1, "socketpair failed (%s)", strerror(errno)); 96 97 buf = malloc(BUFSIZE); 98 ATF_REQUIRE_MSG(buf != NULL, "malloc failed (%s)", strerror(errno)); 99 100 setsock(fd[1], SO_SNDBUF); 101 // setsock(fd[0], SO_RCVBUF); 102 103 mmsgcnt = BUFSIZE / sizeof(DGRAM); 104 mmsghdr = malloc(sizeof(*mmsghdr) * mmsgcnt); 105 ATF_REQUIRE_MSG(mmsghdr != NULL, "malloc failed (%s)", strerror(errno)); 106 iov = malloc(sizeof(*iov) * mmsgcnt); 107 ATF_REQUIRE_MSG(iov != NULL, "malloc failed (%s)", strerror(errno)); 108 109 for (off = 0, n = 0; n < mmsgcnt; n++) { 110 iov[n].iov_base = buf + off; 111 memcpy(iov[n].iov_base, DGRAM, sizeof(DGRAM)); 112 *(buf + off) = n; 113 iov[n].iov_len = sizeof(DGRAM); 114 off += iov[n].iov_len; 115 mmsghdr[n].msg_hdr.msg_iov = &iov[n]; 116 mmsghdr[n].msg_hdr.msg_iovlen = 1; 117 mmsghdr[n].msg_hdr.msg_name = NULL; 118 mmsghdr[n].msg_hdr.msg_namelen = 0; 119 } 120 121 memset(&sa, 0, sizeof(sa)); 122 sa.sa_flags = SA_RESTART; 123 sa.sa_handler = &handle_sigchld; 124 sigemptyset(&sa.sa_mask); 125 error = sigaction(SIGCHLD, &sa, 0); 126 ATF_REQUIRE_MSG(error != -1, "sigaction failed (%s)", 127 strerror(errno)); 128 129 switch (fork()) { 130 case -1: 131 ATF_REQUIRE_MSG(0, "fork failed (%s)", strerror(errno)); 132 break; 133 case 0: 134 sched_yield(); 135 if (debug) 136 printf("sending %u messages (max %u per syscall)\n", n, 137 mmsgcnt); 138 for (n = 0; n < mmsgcnt;) { 139 if (debug) 140 printf("sending packet %u/%u...\n", n, 141 mmsgcnt); 142 // XXX: ENOBUFS bug, on the receive side!!! 143 // in npkt = min(mmsgsize, mmsgcnt - n); 144 int npkt = min(3, mmsgcnt - n), a; 145 do { 146 a = 0; 147 ATF_REQUIRE(ioctl(fd[1], FIONSPACE, &a) != -1); 148 printf("1 %d\n", a); 149 ATF_REQUIRE(ioctl(fd[0], FIONSPACE, &a) != -1); 150 printf("0 %d\n", a); 151 } while ((size_t)a < sizeof(DGRAM)); 152 cnt = sendmmsg(fd[1], mmsghdr + n, npkt, 0); 153 if (cnt == -1 && errno == ENOBUFS) { 154 overf++; 155 if (debug) 156 printf("send buffer overflowed" 157 " (%zu)\n",overf); 158 if (overf > 100) 159 exit(1); 160 sched_yield(); 161 sched_yield(); 162 sched_yield(); 163 continue; 164 } 165 ATF_REQUIRE_MSG(cnt != -1, "sendmmsg failed (%s)", 166 strerror(errno)); 167 if (debug) 168 printf("sendmmsg: sent %u messages\n", cnt); 169 n += cnt; 170 sched_yield(); 171 sched_yield(); 172 sched_yield(); 173 } 174 if (debug) 175 printf("done!\n"); 176 exit(0); 177 /*NOTREACHED*/ 178 default: 179 for (n = 0; n < mmsgcnt; n++) { 180 if (debug) 181 printf("receiving packet %u/%u...\n", n, 182 mmsgcnt); 183 do { 184 if (rdied) 185 break; 186 cnt = recv(fd[0], rgram, sizeof(rgram), 0); 187 ATF_REQUIRE_MSG(cnt != -1 || errno != ENOBUFS, 188 "recv failed (%s)", strerror(errno)); 189 ATF_CHECK_EQ_MSG(cnt, sizeof(rgram), 190 "packet length"); 191 ATF_CHECK_EQ_MSG(rgram[0], n, 192 "number %u != %u", rgram[0], n); 193 ATF_REQUIRE_MSG(memcmp(rgram + 1, DGRAM + 1, 194 sizeof(rgram) - 1) == 0, "bad data"); 195 } while (cnt == -1 && errno == ENOBUFS); 196 } 197 error = wait(&status); 198 ATF_REQUIRE_MSG(error != -1, "wait failed (%s)", 199 strerror(errno)); 200 ATF_REQUIRE_MSG(WIFEXITED(status) && WEXITSTATUS(status) == 0, 201 "receiver died"); 202 break; 203 } 204 } 205 206 ATF_TP_ADD_TCS(tp) 207 { 208 209 ATF_TP_ADD_TC(tp, sendmmsg_basic); 210 211 return atf_no_error(); 212 } 213