2 /* $NetBSD: kttcp.c,v 1.5 2002/07/11 23:32:35 simonb Exp $ */
5 * Copyright (c) 2002 Wasabi Systems, Inc.
8 * Written by Frank van der Linden and Jason R. Thorpe
9 * for Wasabi Systems, Inc.
11 * Redistribution and use in source and binary forms, with or without
12 * modification, are permitted provided that the following conditions
14 * 1. Redistributions of source code must retain the above copyright
15 * notice, this list of conditions and the following disclaimer.
16 * 2. Redistributions in binary form must reproduce the above copyright
17 * notice, this list of conditions and the following disclaimer in the
18 * documentation and/or other materials provided with the distribution.
19 * 3. All advertising materials mentioning features or use of this software
20 * must display the following acknowledgement:
21 * This product includes software developed for the NetBSD Project by
22 * Wasabi Systems, Inc.
23 * 4. The name of Wasabi Systems, Inc. may not be used to endorse
24 * or promote products derived from this software without specific prior
27 * THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND
28 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
29 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
30 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL WASABI SYSTEMS, INC
31 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
32 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
33 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
34 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
35 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
36 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
37 * POSSIBILITY OF SUCH DAMAGE.
40 #include <sys/types.h>
41 #include <sys/param.h>
43 #include <sys/resource.h>
44 #include <sys/socket.h>
45 #include <sys/ioctl.h>
58 #define KTTCP_PORT "22222"
59 #define KTTCP_XMITSIZE (10*1024*1024)
60 #define KTTCP_SOCKBUF_DEFAULT 65536
62 #define KTTCP_DEVICE "/dev/kttcp"
68 "usage: kttcp -r [-b sockbufsize] [-p port] [-q] [-v]\n"
70 " kttcp -t [-b sockbufsize] [-n bytes] [-q] [-v] [-p port]\n"
76 static unsigned long long
77 get_bytes(const char *str)
79 unsigned long long bytes;
82 bytes = strtoull(str, &cp, 10);
83 if (bytes == ULLONG_MAX && errno == ERANGE)
88 errx(1, "invalid byte count: %s", str);
89 if (cp[0] == 'k' || cp[0] == 'K')
91 else if (cp[0] == 'm' || cp[0] == 'M')
93 else if (cp[0] == 'g' || cp[0] == 'G')
94 bytes *= 1024 * 1024 * 1024;
96 errx(1, "invalid byte count modifier: %s", str);
103 main(int argc, char *argv[])
105 int c, error, s, verbose, s2, kfd;
111 struct kttcp_io_args kio;
112 struct addrinfo hints, *addr, *res;
113 struct sockaddr_storage ss;
114 struct rusage rustart, ruend;
115 struct timeval tvtmp;
116 unsigned long long ull, usecs, bytespersec, bitspersec, xmitsize;
117 char connecthost[NI_MAXHOST];
123 portstr = KTTCP_PORT;
126 bufsize = KTTCP_SOCKBUF_DEFAULT;
127 xmitsize = KTTCP_XMITSIZE;
129 while ((c = getopt(argc, argv, "46b:n:p:qrtvw:")) != -1) {
132 if (family != PF_UNSPEC)
137 if (family != PF_UNSPEC)
142 ull = get_bytes(optarg);
145 "invalid socket buffer size: %s\n", optarg);
149 xmitsize = get_bytes(optarg);
182 if (cmd == KTTCP_IO_SEND) {
183 if (xmitsize <= 0 || argc < 1)
189 xmitsize = KTTCP_MAX_XMIT;
191 ai_flag = AI_PASSIVE;
194 if ((kfd = open(KTTCP_DEVICE, O_RDWR, 666)) == -1)
195 err(2, "open %s", KTTCP_DEVICE);
197 memset(&hints, 0, sizeof hints);
198 hints.ai_flags = ai_flag;
199 hints.ai_socktype = SOCK_STREAM;
200 hints.ai_family = family;
201 error = getaddrinfo(host, portstr, &hints, &addr);
204 errx(2, "%s", gai_strerror(error));
207 for (res = addr; res != NULL; res = res->ai_next) {
208 s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
213 err(2, "can't create socket");
215 printf("kttcp: socket buffer size: %d\n", bufsize);
217 if (cmd == KTTCP_IO_SEND) {
218 if (connect(s, res->ai_addr, res->ai_addrlen) < 0)
221 getnameinfo(res->ai_addr, res->ai_addrlen,
222 connecthost, sizeof connecthost, NULL, 0,
224 printf("kttcp: connected to %s\n", connecthost);
226 if (setsockopt(s, SOL_SOCKET, SO_SNDBUF, &bufsize, sizeof (int))
228 err(2, "setsockopt sndbuf");
231 if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &one,
233 err(2, "setsockopt reuseaddr");
234 if (bind(s, res->ai_addr, res->ai_addrlen) < 0)
236 if (listen(s, 1) < 0)
239 printf("kttcp: listening on port %s\n", portstr);
241 s2 = accept(s, (struct sockaddr *)&ss, &slen);
245 getnameinfo((struct sockaddr *)&ss, ss.ss_len,
246 connecthost, sizeof connecthost, NULL, 0,
248 printf("kttcp: connect from %s\n", connecthost);
250 if (setsockopt(s2, SOL_SOCKET, SO_RCVBUF, &bufsize,
252 err(2, "setsockopt rcvbuf");
256 kio.kio_totalsize = xmitsize;
258 getrusage(RUSAGE_SELF, &rustart);
259 if (ioctl(kfd, cmd, &kio) == -1)
260 err(2, "kttcp i/o command");
261 getrusage(RUSAGE_SELF, &ruend);
263 usecs = (unsigned long long)kio.kio_elapsed.tv_sec * 1000000;
264 usecs += kio.kio_elapsed.tv_usec;
266 bytespersec = kio.kio_bytesdone * 1000000LL / usecs;
267 bitspersec = bytespersec * NBBY;
268 printf("kttcp: %llu bytes in %ld.%03ld real seconds ==> %llu bytes/sec\n",
269 kio.kio_bytesdone, kio.kio_elapsed.tv_sec,
270 kio.kio_elapsed.tv_usec / 1000, bytespersec);
272 timersub(&ruend.ru_stime, &rustart.ru_stime, &tvtmp);
273 bytespersec = kio.kio_bytesdone * 1000000LL /
274 (tvtmp.tv_sec * 1000000ULL + tvtmp.tv_usec);
275 printf("kttcp: %llu bytes in %ld.%03ld CPU seconds ==> %llu bytes/CPU sec\n",
276 kio.kio_bytesdone, tvtmp.tv_sec, tvtmp.tv_usec / 1000, bytespersec);
278 printf(" %g (%g) Megabits/sec\n",
279 ((double) bitspersec / 1024.0) / 1024.0,
280 ((double) bitspersec / 1000.0) / 1000.0);
282 timersub(&ruend.ru_utime, &rustart.ru_utime, &tvtmp);
284 * sometimes, this ends up as -1 * hz!?
286 if (tvtmp.tv_sec < 0)
287 tvtmp.tv_sec = tvtmp.tv_usec = 0;
288 printf(" %ld.%02lduser", tvtmp.tv_sec, tvtmp.tv_usec / 10000);
289 ull = tvtmp.tv_sec * 1000000ULL + tvtmp.tv_usec;
291 timersub(&ruend.ru_stime, &rustart.ru_stime, &tvtmp);
292 printf(" %ld.%02ldsys", tvtmp.tv_sec, tvtmp.tv_usec / 10000);
293 ull += tvtmp.tv_sec * 1000000ULL + tvtmp.tv_usec;
295 printf(" %lld.%lldreal", usecs / 1000000, (usecs % 1000000) / 10000);
296 printf(" %lld%%", ull * 100 / usecs);
300 close(kio.kio_socket);
301 if (cmd == KTTCP_IO_RECV)