2 * Copyright (c) 2007 Bruce M. Simpson
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 * Test utility for IPv4 broadcast sockets.
31 #include <sys/cdefs.h>
32 __FBSDID("$FreeBSD$");
34 #include <sys/param.h>
35 #include <sys/types.h>
36 #include <sys/ioctl.h>
37 #include <sys/socket.h>
40 #include <net/if_dl.h>
41 #include <netinet/in.h>
42 #include <arpa/inet.h>
60 #define IP_SENDIF 24 /* XXX */
63 #ifndef IPPROTO_ZEROHOP
64 #define IPPROTO_ZEROHOP 114 /* any 0-hop protocol */
67 #define DEFAULT_PORT 6698
68 #define DEFAULT_PAYLOAD_SIZE 24
71 #define MY_CMSG_SIZE \
72 CMSG_SPACE(sizeof(struct in_addr)) + \
73 CMSG_SPACE(sizeof(struct sockaddr_dl))
75 static char *progname = NULL;
81 fprintf(stderr, "IPv4 broadcast test program. Sends a %d byte UDP "
82 "datagram to <dest>:<port>.\n\n", DEFAULT_PAYLOAD_SIZE);
84 "usage: %s [-1] [-A laddr] [-b] [-B] [-d] [-i iface] [-l len]\n"
85 " [-p port] [-R] [-s srcaddr] [-t ttl] <dest>\n",
87 fprintf(stderr, "-1: Set IP_ONESBCAST\n");
88 fprintf(stderr, "-A: specify laddr (default: INADDR_ANY)\n");
89 fprintf(stderr, "-b: bind socket to <laddr>:<lport>\n");
90 fprintf(stderr, "-B: Set SO_BROADCAST\n");
91 fprintf(stderr, "-d: Set SO_DONTROUTE\n");
92 fprintf(stderr, "-i: Set IP_SENDIF <iface> (if supported)\n");
93 fprintf(stderr, "-l: Set payload size to <len>\n");
94 fprintf(stderr, "-p: Set local and remote port (default: %d)\n",
96 fprintf(stderr, "-R: Use raw IP (protocol %d)\n", IPPROTO_ZEROHOP);
98 fprintf(stderr, "-r: Fill datagram with random bytes\n");
100 fprintf(stderr, "-s: Set IP_SENDSRCADDR to <srcaddr>\n");
101 fprintf(stderr, "-t: Set IP_TTL to <ttl>\n");
107 main(int argc, char *argv[])
110 char cmsgbuf[MY_CMSG_SIZE];
113 struct sockaddr_in dsin;
114 struct sockaddr_in laddr;
115 struct sockaddr_dl *sdl;
116 struct cmsghdr *cmsgp;
117 struct in_addr dstaddr;
118 struct in_addr *srcaddrp;
146 dstaddr.s_addr = INADDR_ANY;
149 portno = DEFAULT_PORT;
153 buflen = DEFAULT_PAYLOAD_SIZE;
155 progname = basename(argv[0]);
156 while ((ch = getopt(argc, argv, "1A:bBdi:l:p:Rrs:t:")) != -1) {
177 buflen = atoi(optarg);
180 portno = atoi(optarg);
204 if (argv[0] == NULL || inet_aton(argv[0], &dstaddr) == 0)
206 /* IP_SENDSRCADDR and IP_SENDIF are mutually exclusive just now. */
207 if (srcaddr_s != NULL && ifname != NULL)
211 fprintf(stderr, "WARNING: not running as root.\n");
212 s = socket(PF_INET, SOCK_RAW, IPPROTO_ZEROHOP);
214 s = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
223 soptlen = sizeof(soptval);
224 ret = setsockopt(s, SOL_SOCKET, SO_DONTROUTE, &soptval,
227 perror("setsockopt SO_DONTROUTE");
235 soptlen = sizeof(soptval);
236 ret = setsockopt(s, SOL_SOCKET, SO_BROADCAST, &soptval,
239 perror("setsockopt SO_BROADCAST");
246 soptlen = sizeof(soptval);
247 ret = setsockopt(s, IPPROTO_IP, IP_TTL, &soptval, soptlen);
249 perror("setsockopt IPPROTO_IP IP_TTL");
256 soptlen = sizeof(soptval);
257 ret = setsockopt(s, IPPROTO_IP, IP_ONESBCAST, &soptval,
260 perror("setsockopt IP_ONESBCAST");
267 memset(&laddr, 0, sizeof(struct sockaddr_in));
268 laddr.sin_family = AF_INET;
269 laddr.sin_len = sizeof(struct sockaddr_in);
270 if (laddr_s != NULL) {
271 laddr.sin_addr.s_addr = inet_addr(laddr_s);
273 laddr.sin_addr.s_addr = INADDR_ANY;
274 laddr.sin_port = htons(portno);
275 ret = bind(s, (struct sockaddr *)&laddr, sizeof(laddr));
283 memset(&dsin, 0, sizeof(struct sockaddr_in));
284 dsin.sin_family = AF_INET;
285 dsin.sin_len = sizeof(struct sockaddr_in);
286 dsin.sin_addr.s_addr = dstaddr.s_addr;
287 dsin.sin_port = htons(portno);
289 buf = malloc(buflen);
295 memset(iov, 0, sizeof(iov));
296 iov[0].iov_base = buf;
297 iov[0].iov_len = buflen;
299 memset(&msg, 0, sizeof(struct msghdr));
300 msg.msg_name = &dsin;
301 msg.msg_namelen = sizeof(dsin);
305 /* Assume we fill out a control msg; macros need to see buf ptr */
306 msg.msg_control = cmsgbuf;
307 msg.msg_controllen = 0;
308 memset(cmsgbuf, 0, MY_CMSG_SIZE);
310 /* IP_SENDSRCADDR and IP_SENDIF are mutually exclusive just now. */
311 if (srcaddr_s != NULL) {
312 msg.msg_controllen += CMSG_SPACE(sizeof(struct in_addr));
313 cmsgp = CMSG_FIRSTHDR(&msg);
314 cmsgp->cmsg_len = CMSG_LEN(sizeof(struct in_addr));
315 cmsgp->cmsg_level = IPPROTO_IP;
316 cmsgp->cmsg_type = IP_SENDSRCADDR;
317 srcaddrp = (struct in_addr *)CMSG_DATA(cmsgp);
318 srcaddrp->s_addr = inet_addr(srcaddr_s);
321 if (ifname != NULL) {
323 msg.msg_controllen += CMSG_SPACE(sizeof(struct sockaddr_dl));
324 cmsgp = CMSG_FIRSTHDR(&msg);
325 cmsgp->cmsg_len = CMSG_LEN(sizeof(struct sockaddr_dl));
326 cmsgp->cmsg_level = IPPROTO_IP;
327 cmsgp->cmsg_type = IP_SENDIF;
330 fprintf(stderr, "DEBUG: cmsgp->cmsg_len is %d\n",
334 sdl = (struct sockaddr_dl *)CMSG_DATA(cmsgp);
335 memset(sdl, 0, sizeof(struct sockaddr_dl));
336 sdl->sdl_family = AF_LINK;
337 sdl->sdl_len = sizeof(struct sockaddr_dl);
338 sdl->sdl_index = if_nametoindex(ifname);
341 fprintf(stderr, "DEBUG: sdl->sdl_family is %d\n",
343 fprintf(stderr, "DEBUG: sdl->sdl_len is %d\n",
345 fprintf(stderr, "DEBUG: sdl->sdl_index is %d\n",
349 fprintf(stderr, "WARNING: IP_SENDIF not supported, ignored.\n");
353 if (msg.msg_controllen == 0)
354 msg.msg_control = NULL;
356 nbytes = sendmsg(s, &msg, (dontroute ? MSG_DONTROUTE : 0));