tcpconnect.c revision 222483
1136337Srwatson/*- 2146251Srwatson * Copyright (c) 2004-2005 Robert N. M. Watson 3136337Srwatson * All rights reserved. 4136337Srwatson * 5136337Srwatson * Redistribution and use in source and binary forms, with or without 6136337Srwatson * modification, are permitted provided that the following conditions 7136337Srwatson * are met: 8136337Srwatson * 1. Redistributions of source code must retain the above copyright 9136337Srwatson * notice, this list of conditions and the following disclaimer. 10136337Srwatson * 2. Redistributions in binary form must reproduce the above copyright 11136337Srwatson * notice, this list of conditions and the following disclaimer in the 12136337Srwatson * documentation and/or other materials provided with the distribution. 13136337Srwatson * 14136337Srwatson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15136337Srwatson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16136337Srwatson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17136337Srwatson * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18136337Srwatson * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19136337Srwatson * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20136337Srwatson * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21136337Srwatson * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22136337Srwatson * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23136337Srwatson * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24136337Srwatson * SUCH DAMAGE. 25136337Srwatson * 26136337Srwatson * $FreeBSD: head/tools/regression/netinet/tcpconnect/tcpconnect.c 222483 2011-05-30 08:54:32Z rwatson $ 27136337Srwatson */ 28136337Srwatson 29136337Srwatson#include <sys/types.h> 30136337Srwatson#include <sys/socket.h> 31136337Srwatson 32136337Srwatson#include <netinet/in.h> 33174032Sbz#include <netinet/tcp.h> 34136337Srwatson 35136337Srwatson#include <arpa/inet.h> 36136337Srwatson 37222483Srwatson#include <err.h> 38136843Srwatson#include <errno.h> 39174032Sbz#include <fcntl.h> 40136337Srwatson#include <stdio.h> 41136337Srwatson#include <stdlib.h> 42136337Srwatson#include <string.h> 43136337Srwatson#include <unistd.h> 44136337Srwatson 45174032Sbz 46136337Srwatsonstatic void 47136613Sruusage(void) 48136337Srwatson{ 49136337Srwatson 50136337Srwatson fprintf(stderr, "tcpconnect server port\n"); 51174032Sbz fprintf(stderr, "tcpconnect client ip port count [nonblock] [tcpmd5]\n"); 52136337Srwatson exit(-1); 53136337Srwatson} 54136337Srwatson 55136337Srwatsonstatic void 56136337Srwatsontcpconnect_server(int argc, char *argv[]) 57136337Srwatson{ 58136337Srwatson int listen_sock, accept_sock; 59136337Srwatson struct sockaddr_in sin; 60136337Srwatson char *dummy; 61136337Srwatson long port; 62136337Srwatson 63136337Srwatson if (argc != 1) 64136337Srwatson usage(); 65136337Srwatson 66136337Srwatson bzero(&sin, sizeof(sin)); 67136337Srwatson sin.sin_len = sizeof(sin); 68136337Srwatson sin.sin_family = AF_INET; 69136337Srwatson sin.sin_addr.s_addr = htonl(INADDR_ANY); 70136337Srwatson 71136337Srwatson port = strtoul(argv[0], &dummy, 10); 72136337Srwatson if (port < 1 || port > 65535 || *dummy != '\0') 73136337Srwatson usage(); 74136337Srwatson sin.sin_port = htons(port); 75136337Srwatson 76136337Srwatson listen_sock = socket(PF_INET, SOCK_STREAM, 0); 77136843Srwatson if (listen_sock == -1) 78174033Sbz err(-1, "socket"); 79136337Srwatson 80136843Srwatson if (bind(listen_sock, (struct sockaddr *)&sin, sizeof(sin)) == -1) 81174033Sbz err(-1, "bind"); 82136337Srwatson 83136843Srwatson if (listen(listen_sock, -1) == -1) 84174033Sbz err(-1, "listen"); 85136337Srwatson 86136337Srwatson while (1) { 87136337Srwatson accept_sock = accept(listen_sock, NULL, NULL); 88136337Srwatson close(accept_sock); 89136337Srwatson } 90136337Srwatson} 91136337Srwatson 92136337Srwatsonstatic void 93136337Srwatsontcpconnect_client(int argc, char *argv[]) 94136337Srwatson{ 95136337Srwatson struct sockaddr_in sin; 96136337Srwatson long count, i, port; 97136337Srwatson char *dummy; 98136337Srwatson int sock; 99174032Sbz int nonblock = 0, md5enable = 0; 100136337Srwatson 101174032Sbz if (argc < 3 || argc > 5) 102136337Srwatson usage(); 103174032Sbz for (i=3; i < argc; i++) { 104174032Sbz if (strcmp(argv[i], "nonblock") == 0) 105174032Sbz nonblock = 1; 106174032Sbz if (strcmp(argv[i], "tcpmd5") == 0) 107174032Sbz md5enable = 1; 108174032Sbz } 109136337Srwatson 110136337Srwatson bzero(&sin, sizeof(sin)); 111136337Srwatson sin.sin_len = sizeof(sin); 112136337Srwatson sin.sin_family = AF_INET; 113136843Srwatson if (inet_aton(argv[0], &sin.sin_addr) == 0) 114174033Sbz err(-1, "listen"); 115136337Srwatson 116136337Srwatson port = strtoul(argv[1], &dummy, 10); 117136337Srwatson if (port < 1 || port > 65535 || *dummy != '\0') 118136337Srwatson usage(); 119136337Srwatson sin.sin_port = htons(port); 120136337Srwatson 121136337Srwatson count = strtoul(argv[2], &dummy, 10); 122136337Srwatson if (count < 1 || count > 100000 || *dummy != '\0') 123136337Srwatson usage(); 124136337Srwatson 125136337Srwatson for (i = 0; i < count; i++) { 126136337Srwatson sock = socket(PF_INET, SOCK_STREAM, 0); 127136843Srwatson if (sock == -1) 128174033Sbz err(-1, "socket"); 129136337Srwatson 130174032Sbz /* No warning in default case on ENOPROTOOPT. */ 131174032Sbz if (setsockopt(sock, IPPROTO_TCP, TCP_MD5SIG, 132174032Sbz &md5enable, sizeof(md5enable)) != 0) { 133174032Sbz if (errno == ENOPROTOOPT && md5enable > 0) 134174032Sbz err(-1, "setsockopt(TCP_MD5SIG)"); 135174032Sbz else if (errno != ENOPROTOOPT) 136174032Sbz warn("setsockopt(TCP_MD5SIG)"); 137174032Sbz } 138146251Srwatson 139174032Sbz if (nonblock) { 140174032Sbz if (fcntl(sock, F_SETFL, O_NONBLOCK) != 0) 141174033Sbz err(-1, "fcntl(F_SETFL)"); 142136337Srwatson 143174032Sbz if (connect(sock, (struct sockaddr *)&sin, 144174032Sbz sizeof(sin)) == -1 && errno != EINPROGRESS) 145174033Sbz err(-1, "connect"); 146174032Sbz } else { 147174032Sbz if (connect(sock, (struct sockaddr *)&sin, 148174032Sbz sizeof(sin)) == -1) 149174033Sbz err(-1, "connect"); 150174032Sbz } 151174032Sbz 152136337Srwatson close(sock); 153136337Srwatson } 154136337Srwatson} 155136337Srwatson 156136337Srwatsonint 157136337Srwatsonmain(int argc, char *argv[]) 158136337Srwatson{ 159136337Srwatson 160136337Srwatson if (argc < 2) 161136337Srwatson usage(); 162136337Srwatson 163136337Srwatson if (strcmp(argv[1], "server") == 0) 164136337Srwatson tcpconnect_server(argc - 2, argv + 2); 165136337Srwatson else if (strcmp(argv[1], "client") == 0) 166136337Srwatson tcpconnect_client(argc - 2, argv + 2); 167136337Srwatson else 168136337Srwatson usage(); 169136337Srwatson 170136337Srwatson exit(0); 171136337Srwatson} 172