freebsd-skq/tools/regression/netinet/ipmulticast/ipmulticast.c
2007-06-12 16:29:22 +00:00

788 lines
19 KiB
C

/*-
* Copyright (c) 2007 Bruce M. Simpson
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
/*
* Regression test utility for RFC 3678 Advanced Multicast API in FreeBSD.
*
* TODO: Test the SSM paths.
* TODO: Support INET6. The code has been written to facilitate this later.
* TODO: Merge multicast socket option tests from ipsockopt.
*/
#include <sys/cdefs.h>
__FBSDID("$FreeBSD$");
#include <sys/param.h>
#include <sys/types.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <net/if.h>
#include <net/if_dl.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <assert.h>
#include <err.h>
#include <errno.h>
#include <getopt.h>
#include <libgen.h>
#include <pwd.h>
#include <setjmp.h>
#include <signal.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <time.h>
#include <unistd.h>
#ifndef __SOCKUNION_DECLARED
union sockunion {
struct sockaddr_storage ss;
struct sockaddr sa;
struct sockaddr_dl sdl;
struct sockaddr_in sin;
#ifdef INET6
struct sockaddr_in6 sin6;
#endif
};
typedef union sockunion sockunion_t;
#define __SOCKUNION_DECLARED
#endif /* __SOCKUNION_DECLARED */
#define ADDRBUF_LEN 16
#define DEFAULT_GROUP_STR "238.1.1.0"
#define DEFAULT_IFNAME "lo0"
#define DEFAULT_IFADDR_STR "127.0.0.1"
#define DEFAULT_PORT 6698
#define DEFAULT_TIMEOUT 0 /* don't wait for traffic */
#define RXBUFSIZE 2048
static sockunion_t basegroup;
static const char *basegroup_str = NULL;
static int dobindaddr = 0;
static int dodebug = 1;
static int doipv4 = 0;
static int domiscopts = 0;
static int dorandom = 0;
static int doreuseport = 0;
static int dossm = 0;
static int dossf = 0;
static int doverbose = 0;
static sockunion_t ifaddr;
static const char *ifaddr_str = NULL;
static uint32_t ifindex = 0;
static const char *ifname = NULL;
struct in_addr *ipv4_sources = NULL;
static jmp_buf jmpbuf;
static size_t nmcastgroups = IP_MAX_MEMBERSHIPS;
static size_t nmcastsources = 0;
static uint16_t portno = DEFAULT_PORT;
static char *progname = NULL;
struct sockaddr_storage *ss_sources = NULL;
static uint32_t timeout = 0;
static int do_asm_ipv4(void);
static int do_asm_pim(void);
#ifdef notyet
static int do_misc_opts(void);
#endif
static int do_ssf_ipv4(void);
static int do_ssf_pim(void);
static int do_ssm_ipv4(void);
static int do_ssm_pim(void);
static int open_and_bind_socket(sockunion_t *);
static int recv_loop_with_match(int, sockunion_t *, sockunion_t *);
static void signal_handler(int);
static void usage(void);
/*
* Test the IPv4 set/getipv4sourcefilter() libc API functions.
* Build a single socket.
* Join a source group.
* Repeatedly change the source filters via setipv4sourcefilter.
* Read it back with getipv4sourcefilter up to IP_MAX_SOURCES
* and check for inconsistency.
*/
static int
do_ssf_ipv4(void)
{
fprintf(stderr, "not yet implemented\n");
return (0);
}
/*
* Test the protocol-independent set/getsourcefilter() functions.
*/
static int
do_ssf_pim(void)
{
fprintf(stderr, "not yet implemented\n");
return (0);
}
/*
* Test the IPv4 ASM API.
* Repeatedly join, block sources, unblock and leave groups.
*/
static int
do_asm_ipv4(void)
{
int error;
char gaddrbuf[ADDRBUF_LEN];
int i;
sockunion_t laddr;
struct ip_mreq mreq;
struct ip_mreq_source mreqs;
in_addr_t ngroupbase;
char saddrbuf[ADDRBUF_LEN];
int sock;
sockunion_t tmpgroup;
sockunion_t tmpsource;
memset(&mreq, 0, sizeof(struct ip_mreq));
memset(&mreqs, 0, sizeof(struct ip_mreq_source));
memset(&laddr, 0, sizeof(sockunion_t));
if (dobindaddr) {
laddr = ifaddr;
} else {
laddr.sin.sin_family = AF_INET;
laddr.sin.sin_len = sizeof(struct sockaddr_in);
laddr.sin.sin_addr.s_addr = INADDR_ANY;
}
laddr.sin.sin_port = htons(portno);
tmpgroup = basegroup;
ngroupbase = ntohl(basegroup.sin.sin_addr.s_addr) + 1; /* XXX */
tmpgroup.sin.sin_addr.s_addr = htonl(ngroupbase);
sock = open_and_bind_socket(&laddr);
if (sock == -1)
return (EX_OSERR);
for (i = 0; i < (signed)nmcastgroups; i++) {
mreq.imr_multiaddr.s_addr = htonl((ngroupbase + i));
mreq.imr_interface = ifaddr.sin.sin_addr;
if (doverbose) {
inet_ntop(AF_INET, &mreq.imr_multiaddr, gaddrbuf,
sizeof(gaddrbuf));
fprintf(stderr, "IP_ADD_MEMBERSHIP %s %s\n",
gaddrbuf, inet_ntoa(mreq.imr_interface));
}
error = setsockopt(sock, IPPROTO_IP, IP_ADD_MEMBERSHIP,
&mreq, sizeof(struct ip_mreq));
if (error < 0) {
warn("setsockopt IP_ADD_MEMBERSHIP");
close(sock);
return (EX_OSERR);
}
}
/*
* If no test sources auto-generated or specified on command line,
* skip source filter portion of ASM test.
*/
if (nmcastsources == 0)
goto skipsources;
/*
* Begin blocking sources on the first group chosen.
*/
for (i = 0; i < (signed)nmcastsources; i++) {
mreqs.imr_multiaddr = tmpgroup.sin.sin_addr;
mreqs.imr_interface = ifaddr.sin.sin_addr;
mreqs.imr_sourceaddr = ipv4_sources[i];
if (doverbose) {
inet_ntop(AF_INET, &mreqs.imr_multiaddr, gaddrbuf,
sizeof(gaddrbuf));
inet_ntop(AF_INET, &mreqs.imr_sourceaddr, saddrbuf,
sizeof(saddrbuf));
fprintf(stderr, "IP_BLOCK_SOURCE %s %s %s\n",
gaddrbuf, inet_ntoa(mreqs.imr_interface),
saddrbuf);
}
error = setsockopt(sock, IPPROTO_IP, IP_BLOCK_SOURCE, &mreqs,
sizeof(struct ip_mreq_source));
if (error < 0) {
warn("setsockopt IP_BLOCK_SOURCE");
close(sock);
return (EX_OSERR);
}
}
/*
* Choose the first group and source for a match.
* Enter the I/O loop.
*/
memset(&tmpsource, 0, sizeof(sockunion_t));
tmpsource.sin.sin_family = AF_INET;
tmpsource.sin.sin_len = sizeof(struct sockaddr_in);
tmpsource.sin.sin_addr = ipv4_sources[0];
error = recv_loop_with_match(sock, &tmpgroup, &tmpsource);
/*
* Unblock sources.
*/
for (i = nmcastsources-1; i >= 0; i--) {
mreqs.imr_multiaddr = tmpgroup.sin.sin_addr;
mreqs.imr_interface = ifaddr.sin.sin_addr;
mreqs.imr_sourceaddr = ipv4_sources[i];
if (doverbose) {
inet_ntop(AF_INET, &mreqs.imr_multiaddr, gaddrbuf,
sizeof(gaddrbuf));
inet_ntop(AF_INET, &mreqs.imr_sourceaddr, saddrbuf,
sizeof(saddrbuf));
fprintf(stderr, "IP_UNBLOCK_SOURCE %s %s %s\n",
gaddrbuf, inet_ntoa(mreqs.imr_interface),
saddrbuf);
}
error = setsockopt(sock, IPPROTO_IP, IP_UNBLOCK_SOURCE, &mreqs,
sizeof(struct ip_mreq_source));
if (error < 0) {
warn("setsockopt IP_UNBLOCK_SOURCE");
close(sock);
return (EX_OSERR);
}
}
skipsources:
/*
* Leave groups.
*/
for (i = nmcastgroups-1; i >= 0; i--) {
mreq.imr_multiaddr.s_addr = htonl((ngroupbase + i));
mreq.imr_interface = ifaddr.sin.sin_addr;
if (doverbose) {
inet_ntop(AF_INET, &mreq.imr_multiaddr, gaddrbuf,
sizeof(gaddrbuf));
fprintf(stderr, "IP_DROP_MEMBERSHIP %s %s\n",
gaddrbuf, inet_ntoa(mreq.imr_interface));
}
error = setsockopt(sock, IPPROTO_IP, IP_DROP_MEMBERSHIP,
&mreq, sizeof(struct ip_mreq));
if (error < 0) {
warn("setsockopt IP_DROP_MEMBERSHIP");
close(sock);
return (EX_OSERR);
}
}
return (0);
}
static int
do_asm_pim(void)
{
fprintf(stderr, "not yet implemented\n");
return (0);
}
#ifdef notyet
/*
* Test misceallaneous IPv4 options.
*/
static int
do_misc_opts(void)
{
int sock;
sock = open_and_bind_socket(NULL);
if (sock == -1)
return (EX_OSERR);
test_ip_uchar(sock, socktypename, IP_MULTICAST_TTL,
"IP_MULTICAST_TTL", 1);
close(sock);
sock = open_and_bind_socket(NULL);
if (sock == -1)
return (EX_OSERR);
test_ip_boolean(sock, socktypename, IP_MULTICAST_LOOP,
"IP_MULTICAST_LOOP", 1, BOOLEAN_ANYONE);
close(sock);
return (0);
}
#endif
/*
* Test the IPv4 SSM API.
*/
static int
do_ssm_ipv4(void)
{
fprintf(stderr, "not yet implemented\n");
return (0);
}
/*
* Test the protocol-independent SSM API with IPv4 addresses.
*/
static int
do_ssm_pim(void)
{
fprintf(stderr, "not yet implemented\n");
return (0);
}
int
main(int argc, char *argv[])
{
struct addrinfo aih;
struct addrinfo *aip;
int ch;
int error;
int exitval;
size_t i;
struct in_addr *pina;
struct sockaddr_storage *pbss;
ifname = DEFAULT_IFNAME;
ifaddr_str = DEFAULT_IFADDR_STR;
basegroup_str = DEFAULT_GROUP_STR;
ifname = DEFAULT_IFNAME;
portno = DEFAULT_PORT;
basegroup.ss.ss_family = AF_UNSPEC;
ifaddr.ss.ss_family = AF_UNSPEC;
progname = basename(argv[0]);
while ((ch = getopt(argc, argv, "4bg:i:I:mM:p:rsS:tT:v")) != -1) {
switch (ch) {
case '4':
doipv4 = 1;
break;
case 'b':
dobindaddr = 1;
break;
case 'g':
basegroup_str = optarg;
break;
case 'i':
ifname = optarg;
break;
case 'I':
ifaddr_str = optarg;
break;
case 'm':
usage(); /* notyet */
/*NOTREACHED*/
domiscopts = 1;
break;
case 'M':
nmcastgroups = atoi(optarg);
break;
case 'p':
portno = atoi(optarg);
break;
case 'r':
doreuseport = 1;
break;
case 'S':
nmcastsources = atoi(optarg);
break;
case 's':
dossm = 1;
break;
case 't':
dossf = 1;
break;
case 'T':
timeout = atoi(optarg);
break;
case 'v':
doverbose = 1;
break;
default:
usage();
break;
/*NOTREACHED*/
}
}
argc -= optind;
argv += optind;
memset(&aih, 0, sizeof(struct addrinfo));
aih.ai_flags = AI_NUMERICHOST | AI_PASSIVE;
aih.ai_family = PF_INET;
aih.ai_socktype = SOCK_DGRAM;
aih.ai_protocol = IPPROTO_UDP;
/*
* Fill out base group.
*/
aip = NULL;
error = getaddrinfo(basegroup_str, NULL, &aih, &aip);
if (error != 0) {
fprintf(stderr, "%s: getaddrinfo: %s\n", progname,
gai_strerror(error));
exit(EX_USAGE);
}
memcpy(&basegroup, aip->ai_addr, aip->ai_addrlen);
if (dodebug) {
fprintf(stderr, "debug: gai thinks %s is %s\n",
basegroup_str, inet_ntoa(basegroup.sin.sin_addr));
}
freeaddrinfo(aip);
assert(basegroup.ss.ss_family == AF_INET);
/*
* If user specified interface as an address, and protocol
* specific APIs were selected, parse it.
* Otherwise, parse interface index from name if protocol
* independent APIs were selected (the default).
*/
if (doipv4) {
if (ifaddr_str == NULL) {
warnx("required argument missing: ifaddr");
usage();
/* NOTREACHED */
}
aip = NULL;
error = getaddrinfo(ifaddr_str, NULL, &aih, &aip);
if (error != 0) {
fprintf(stderr, "%s: getaddrinfo: %s\n", progname,
gai_strerror(error));
exit(EX_USAGE);
}
memcpy(&ifaddr, aip->ai_addr, aip->ai_addrlen);
if (dodebug) {
fprintf(stderr, "debug: gai thinks %s is %s\n",
ifaddr_str, inet_ntoa(ifaddr.sin.sin_addr));
}
freeaddrinfo(aip);
}
if (!doipv4) {
if (ifname == NULL) {
warnx("required argument missing: ifname");
usage();
/* NOTREACHED */
}
ifindex = if_nametoindex(ifname);
if (ifindex == 0)
err(EX_USAGE, "if_nametoindex");
}
/*
* Introduce randomness into group base if specified.
*/
if (dorandom) {
in_addr_t ngroupbase;
srandomdev();
ngroupbase = ntohl(basegroup.sin.sin_addr.s_addr);
ngroupbase |= ((random() % ((1 << 11) - 1)) << 16);
basegroup.sin.sin_addr.s_addr = htonl(ngroupbase);
}
if (argc > 0) {
nmcastsources = argc;
if (doipv4) {
ipv4_sources = calloc(nmcastsources,
sizeof(struct in_addr));
if (ipv4_sources == NULL) {
exitval = EX_OSERR;
goto out;
}
} else {
ss_sources = calloc(nmcastsources,
sizeof(struct sockaddr_storage));
if (ss_sources == NULL) {
exitval = EX_OSERR;
goto out;
}
}
}
/*
* Parse source list, if any were specified on the command line.
*/
assert(aih.ai_family == PF_INET);
pbss = ss_sources;
pina = ipv4_sources;
for (i = 0; i < (size_t)argc; i++) {
aip = NULL;
error = getaddrinfo(argv[i], NULL, &aih, &aip);
if (error != 0) {
fprintf(stderr, "getaddrinfo: %s\n",
gai_strerror(error));
exitval = EX_USAGE;
goto out;
}
if (doipv4) {
struct sockaddr_in *sin =
(struct sockaddr_in *)aip->ai_addr;
*pina++ = sin->sin_addr;
} else {
memcpy(pbss++, aip->ai_addr, aip->ai_addrlen);
}
freeaddrinfo(aip);
}
/*
* Perform the regression tests which the user requested.
*/
#ifdef notyet
if (domiscopts) {
exitval = do_misc_opts();
if (exitval)
goto out;
}
#endif
if (doipv4) {
/* IPv4 protocol specific API tests */
if (dossm) {
/* Source-specific multicast */
exitval = do_ssm_ipv4();
if (exitval)
goto out;
if (dossf) {
/* Do setipvsourcefilter() too */
exitval = do_ssf_ipv4();
}
} else {
/* Any-source multicast */
exitval = do_asm_ipv4();
}
} else {
/* Protocol independent API tests */
if (dossm) {
/* Source-specific multicast */
exitval = do_ssm_pim();
if (exitval)
goto out;
if (dossf) {
/* Do setsourcefilter() too */
exitval = do_ssf_pim();
}
} else {
/* Any-source multicast */
exitval = do_asm_pim();
}
}
out:
if (ipv4_sources != NULL)
free(ipv4_sources);
if (ss_sources != NULL)
free(ss_sources);
exit(exitval);
}
static int
open_and_bind_socket(sockunion_t *bsu)
{
int error, optval, sock;
sock = -1;
sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
if (sock == -1) {
warn("socket");
return (-1);
}
if (doreuseport) {
optval = 1;
if (setsockopt(sock, SOL_SOCKET, SO_REUSEPORT, &optval,
sizeof(optval)) < 0) {
warn("setsockopt SO_REUSEPORT");
close(sock);
return (-1);
}
}
if (bsu != NULL) {
error = bind(sock, &bsu->sa, bsu->sa.sa_len);
if (error == -1) {
warn("bind");
close(sock);
return (-1);
}
}
return (sock);
}
/*
* Protocol-agnostic multicast I/O loop.
*
* Wait for 'timeout' seconds looking for traffic on group, so that manual
* or automated regression tests (possibly running on another host) have an
* opportunity to transmit within the group to test source filters.
*
* If the filter failed, this loop will report if we received traffic
* from the source we elected to monitor.
*/
static int
recv_loop_with_match(int sock, sockunion_t *group, sockunion_t *source)
{
int error;
sockunion_t from;
char groupname[NI_MAXHOST];
ssize_t len;
size_t npackets;
int jmpretval;
char rxbuf[RXBUFSIZE];
char sourcename[NI_MAXHOST];
assert(source->sa.sa_family == AF_INET);
/*
* Return immediately if we don't need to wait for traffic.
*/
if (timeout == 0)
return (0);
error = getnameinfo(&group->sa, group->sa.sa_len, groupname,
NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
if (error) {
fprintf(stderr, "getnameinfo: %s\n", gai_strerror(error));
return (error);
}
error = getnameinfo(&source->sa, source->sa.sa_len, sourcename,
NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
if (error) {
fprintf(stderr, "getnameinfo: %s\n", gai_strerror(error));
return (error);
}
fprintf(stdout,
"Waiting %d seconds for inbound traffic on group %s\n"
"Expecting no traffic from blocked source: %s\n",
(int)timeout, groupname, sourcename);
signal(SIGINT, signal_handler);
signal(SIGALRM, signal_handler);
error = 0;
npackets = 0;
alarm(timeout);
while (0 == (jmpretval = setjmp(jmpbuf))) {
len = recvfrom(sock, rxbuf, RXBUFSIZE, 0, &from.sa,
(socklen_t *)&from.sa.sa_len);
if (dodebug) {
fprintf(stderr, "debug: packet received from %s\n",
inet_ntoa(from.sin.sin_addr));
}
if (source &&
source->sin.sin_addr.s_addr == from.sin.sin_addr.s_addr)
break;
npackets++;
}
if (doverbose) {
fprintf(stderr, "Number of datagrams received from "
"non-blocked sources: %d\n", (int)npackets);
}
switch (jmpretval) {
case SIGALRM: /* ok */
break;
case SIGINT: /* go bye bye */
fprintf(stderr, "interrupted\n");
error = 20;
break;
case 0: /* Broke out of loop; saw a bad source. */
fprintf(stderr, "FAIL: got packet from blocked source\n");
error = EX_IOERR;
break;
default:
warnx("recvfrom");
error = EX_OSERR;
break;
}
signal(SIGINT, SIG_DFL);
signal(SIGALRM, SIG_DFL);
return (error);
}
static void
signal_handler(int signo)
{
longjmp(jmpbuf, signo);
}
static void
usage(void)
{
fprintf(stderr, "\nIP multicast regression test utility\n");
fprintf(stderr,
"usage: %s [-4] [-b] [-g groupaddr] [-i ifname] [-I ifaddr] [-m]\n"
" [-M ngroups] [-p portno] [-r] [-R] [-s] [-S nsources] [-t] [-T timeout]\n"
" [-v] [blockaddr ...]\n\n", progname);
fprintf(stderr, "-4: Use IPv4 API "
"(default: Use protocol-independent API)\n");
fprintf(stderr, "-b: bind listening socket to ifaddr "
"(default: INADDR_ANY)\n");
fprintf(stderr, "-g: Base IPv4 multicast group to join (default: %s)\n",
DEFAULT_GROUP_STR);
fprintf(stderr, "-i: interface for multicast joins (default: %s)\n",
DEFAULT_IFNAME);
fprintf(stderr, "-I: IPv4 address to join groups on, if using IPv4 "
"API\n (default: %s)\n", DEFAULT_IFADDR_STR);
#ifdef notyet
fprintf(stderr, "-m: Test misc IPv4 multicast socket options "
"(default: off)\n");
#endif
fprintf(stderr, "-M: Number of multicast groups to join "
"(default: %d)\n", (int)nmcastgroups);
fprintf(stderr, "-p: Set local and remote port (default: %d)\n",
DEFAULT_PORT);
fprintf(stderr, "-r: Set SO_REUSEPORT on (default: off)\n");
fprintf(stderr, "-R: Randomize groups/sources (default: off)\n");
fprintf(stderr, "-s: Test source-specific API "
"(default: test any-source API)\n");
fprintf(stderr, "-S: Number of multicast sources to generate if\n"
" none specified on command line (default: %d)\n",
(int)nmcastsources);
fprintf(stderr, "-t: Test get/setNsourcefilter() (default: off)\n");
fprintf(stderr, "-T: Timeout to wait for blocked traffic on first "
"group (default: %d)\n", DEFAULT_TIMEOUT);
fprintf(stderr, "-v: Be verbose (default: off)\n");
fprintf(stderr, "\nRemaining arguments are treated as a list of IPv4 "
"sources to filter.\n\n");
exit(EX_USAGE);
}