/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2010 Sun Microsystems, Inc.  All rights reserved.
 */

/*
 * Test helper to abort TCP connections to some server;
 * either all of the, or those to a specified port.
 */

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/sockio.h>
#include <sys/stropts.h>

#include <inet/tcp.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <netdb.h>

#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <zone.h>

/*
 * Abort all connections to the passed address.
 */
static void
tcp_abort_connections(struct sockaddr *rsa)
{
	tcp_ioc_abort_conn_t conn;
	struct strioctl ioc;
	struct sockaddr *lsa;
	int fd;

	(void) memset(&conn, 0, sizeof (conn));
	lsa = (void *)&conn.ac_local;
	lsa->sa_family = rsa->sa_family;
	(void) memcpy(&conn.ac_remote, rsa, sizeof (*rsa));
	conn.ac_start = TCPS_SYN_SENT;
	conn.ac_end = TCPS_CLOSE_WAIT;
	conn.ac_zoneid = ALL_ZONES;

	ioc.ic_cmd = TCP_IOC_ABORT_CONN;
	ioc.ic_timout = -1; /* infinite timeout */
	ioc.ic_len = sizeof (conn);
	ioc.ic_dp = (char *)&conn;

	if ((fd = open("/dev/tcp", O_RDONLY)) < 0) {
		(void) fprintf(stderr, "unable to open %s", "/dev/tcp");
		return;
	}

	if (ioctl(fd, I_STR, &ioc) < 0)
		if (errno != ENOENT)	/* ENOENT is not an error */
			perror("ioctl");

	(void) close(fd);
}

static void
usage(char *arg0)
{
	(void) fprintf(stderr, "usage: %s [-p <PORT>] <ADDR>\n", arg0);
	exit(1);
}

int
main(int argc, char **argv)
{
	extern char *optarg;
	extern int optind, optopt;
	struct addrinfo hints, *res, *ai;
	char *addr_str = NULL;
	char *port_str = NULL;
	int errflag = 0;
	int c, gaierr;

	while ((c = getopt(argc, argv, "p:")) != -1) {
		switch (c) {
		case 'p':
			port_str = optarg;
			break;
		case ':':
			(void) fprintf(stderr,
			    "Option -%c requires an operand\n", optopt);
			errflag++;
			break;
		case '?':
			(void) fprintf(stderr,
			    "Unrecognized option: -%c\n", optopt);
			errflag++;
			break;
		}
	}
	if (errflag)
		usage(argv[0]);
	if (argc <= optind) {
		(void) fprintf(stderr, "No address specified\n");
		usage(argv[0]);
	}
	addr_str = argv[optind];

	/*
	 * Lookup the IP address
	 */
	(void) memset(&hints, 0, sizeof (hints));
	hints.ai_family = PF_UNSPEC;
	hints.ai_socktype = SOCK_STREAM;
	gaierr = getaddrinfo(addr_str, port_str, &hints, &res);
	if (gaierr != 0) {
		(void) fprintf(stderr, "%s: %s\n", addr_str,
		    gai_strerror(gaierr));
		return (1);
	}

	for (ai = res; ai != NULL; ai = ai->ai_next) {
		tcp_abort_connections(ai->ai_addr);
	}

	return (0);
}