/*
* Copyright (c) 1993 W. Richard Stevens. All rights reserved.
* Permission to use or modify this software and its documentation only for
* educational purposes and without fee is hereby granted, provided that
* the above copyright notice appear in all copies. The author makes no
* representations about the suitability of this software for any purpose.
* It is provided "as is" without express or implied warranty.
*/
#include "sock.h"
char *host; /* hostname or dotted-decimal string */
char *port;
/* DefinE global variables */
int bindport; /* 0 or TCP or UDP port number to bind */
/* set by -b or -l options */
int broadcast; /* SO_BROADCAST */
int cbreak; /* set terminal to cbreak mode */
int chunkwrite; /* write in small chunks; not all-at-once */
int client = 1; /* acting as client is the default */
int connectudp = 1; /* connect UDP client */
int crlf; /* convert newline to CR/LF & vice versa */
int debug; /* SO_DEBUG */
int dofork; /* concurrent server, do a fork() */
int dontroute; /* SO_DONTROUTE */
char foreignip[32]; /* foreign IP address, dotted-decimal string */
int foreignport; /* foreign port number */
int halfclose; /* TCP half close option */
int ignorewerr; /* true if write() errors should be ignored */
int iptos = -1; /* IP_TOS opton */
int ipttl = -1; /* IP_TTL opton */
char joinip[32]; /* multicast IP address, dotted-decimal string */
int keepalive; /* SO_KEEPALIVE */
long linger = -1; /* 0 or positive turns on option */
int listenq = 5; /* listen queue for TCP Server */
char localip[32]; /* local IP address, dotted-decimal string */
int maxseg; /* TCP_MAXSEG */
int mcastttl; /* multicast TTL */
int msgpeek; /* MSG_PEEK */
int nodelay; /* TCP_NODELAY (Nagle algorithm) */
int nbuf = 1024; /* number of buffers to write (sink mode) */
int onesbcast; /* set IP_ONESBCAST for 255.255.255.255 bcasts */
int pauseclose; /* #ms to sleep after recv FIN, before close */
int pauseinit; /* #ms to sleep before first read */
int pauselisten; /* #ms to sleep after listen() */
int pauserw; /* #ms to sleep before each read or write */
int reuseaddr; /* SO_REUSEADDR */
int reuseport; /* SO_REUSEPORT */
int readlen = 1024; /* default read length for socket */
int writelen = 1024; /* default write length for socket */
int recvdstaddr; /* IP_RECVDSTADDR option */
int rcvbuflen; /* size for SO_RCVBUF */
int sndbuflen; /* size for SO_SNDBUF */
long rcvtimeo; /* SO_RCVTIMEO */
long sndtimeo; /* SO_SNDTIMEO */
int sroute_cnt; /* count of #IP addresses in route */
char *rbuf; /* pointer that is malloc'ed */
char *wbuf; /* pointer that is malloc'ed */
int server; /* to act as server requires -s option */
int sigio; /* send SIGIO */
int sourcesink; /* source/sink mode */
int udp; /* use UDP instead of TCP */
int urgwrite; /* write urgent byte after this write */
int verbose; /* each -v increments this by 1 */
int usewritev; /* use writev() instead of write() */
struct sockaddr_in cliaddr, servaddr;
static void usage(const char *);
int
main(int argc, char *argv[])
{
int c, fd;
char *ptr;
if (argc < 2)
usage("");
opterr = 0; /* don't want getopt() writing to stderr */
while ( (c = getopt(argc, argv, "2b:cf:g:hij:kl:n:op:q:r:st:uvw:x:y:ABCDEFG:H:IJ:KL:NO:P:Q:R:S:TU:VWX:YZ")) != -1) {
switch (c) {
#ifdef IP_ONESBCAST
case '2': /* use 255.255.255.255 as broadcast address */
onesbcast = 1;
break;
#endif
case 'b':
bindport = atoi(optarg);
break;
case 'c': /* convert newline to CR/LF & vice versa */
crlf = 1;
break;
case 'f': /* foreign IP address and port#: a.b.c.d.p */
if ( (ptr = strrchr(optarg, '.')) == NULL)
usage("invalid -f option");
*ptr++ = 0; /* null replaces final period */
foreignport = atoi(ptr); /* port number */
strcpy(foreignip, optarg); /* save dotted-decimal IP */
break;
case 'g': /* loose source route */
sroute_doopt(0, optarg);
break;
case 'h': /* TCP half-close option */
halfclose = 1;
break;
case 'i': /* source/sink option */
sourcesink = 1;
break;
#ifdef IP_ADD_MEMBERSHIP
case 'j': /* join multicast group a.b.c.d */
strcpy(joinip, optarg); /* save dotted-decimal IP */
break;
#endif
case 'k': /* chunk-write option */
chunkwrite = 1;
break;
case 'l': /* local IP address and port#: a.b.c.d.p */
if ( (ptr = strrchr(optarg, '.')) == NULL)
usage("invalid -l option");
*ptr++ = 0; /* null replaces final period */
bindport = atoi(ptr); /* port number */
strcpy(localip, optarg); /* save dotted-decimal IP */
break;
case 'n': /* number of buffers to write */
nbuf = atol(optarg);
break;
case 'o': /* do not connect UDP client */
connectudp = 0;
break;
case 'p': /* pause before each read or write */
pauserw = atoi(optarg);
break;
case 'q': /* listen queue for TCP server */
listenq = atoi(optarg);
break;
case 'r': /* read() length */
readlen = atoi(optarg);
break;
case 's': /* server */
server = 1;
client = 0;
break;
#ifdef IP_MULTICAST_TTL
case 't': /* IP_MULTICAST_TTL */
mcastttl = atoi(optarg);
break;
#endif
case 'u': /* use UDP instead of TCP */
udp = 1;
break;
case 'v': /* output what's going on */
verbose++;
break;
case 'w': /* write() length */
writelen = atoi(optarg);
break;
case 'x': /* SO_RCVTIMEO socket option */
rcvtimeo = atol(optarg);
break;
case 'y': /* SO_SNDTIMEO socket option */
sndtimeo = atol(optarg);
break;
case 'A': /* SO_REUSEADDR socket option */
reuseaddr = 1;
break;
case 'B': /* SO_BROADCAST socket option */
broadcast = 1;
break;
case 'C': /* set standard input to cbreak mode */
cbreak = 1;
break;
case 'D': /* SO_DEBUG socket option */
debug = 1;
break;
case 'E': /* IP_RECVDSTADDR socket option */
recvdstaddr = 1;
break;
case 'F': /* concurrent server, do a fork() */
dofork = 1;
break;
case 'G': /* strict source route */
sroute_doopt(1, optarg);
break;
#ifdef IP_TOS
case 'H': /* IP_TOS socket option */
iptos = atoi(optarg);
break;
#endif
case 'I': /* SIGIO signal */
sigio = 1;
break;
#ifdef IP_TTL
case 'J': /* IP_TTL socket option */
ipttl = atoi(optarg);
break;
#endif
case 'K': /* SO_KEEPALIVE socket option */
keepalive = 1;
break;
case 'L': /* SO_LINGER socket option */
linger = atol(optarg);
break;
case 'N': /* SO_NODELAY socket option */
nodelay = 1;
break;
case 'O': /* pause before listen(), before first accept() */
pauselisten = atoi(optarg);
break;
case 'P': /* pause before first read() */
pauseinit = atoi(optarg);
break;
case 'Q': /* pause after receiving FIN, but before close() */
pauseclose = atoi(optarg);
break;
case 'R': /* SO_RCVBUF socket option */
rcvbuflen = atoi(optarg);
break;
case 'S': /* SO_SNDBUF socket option */
sndbuflen = atoi(optarg);
break;
#ifdef SO_REUSEPORT
case 'T': /* SO_REUSEPORT socket option */
reuseport = 1;
break;
#endif
case 'U': /* when to write urgent byte */
urgwrite = atoi(optarg);
break;
case 'V': /* use writev() instead of write() */
usewritev = 1;
chunkwrite = 1; /* implies this option too */
break;
case 'W': /* ignore write errors */
ignorewerr = 1;
break;
case 'X': /* TCP maximum segment size option */
maxseg = atoi(optarg);
break;
case 'Y': /* SO_DONTROUTE socket option */
dontroute = 1;
break;
case 'Z': /* MSG_PEEK option */
msgpeek = MSG_PEEK;
break;
case '?':
usage("unrecognized option");
}
}
/* check for options that don't make sense */
if (udp && halfclose)
usage("can't specify -h and -u");
if (udp && debug)
usage("can't specify -D and -u");
if (udp && linger >= 0)
usage("can't specify -L and -u");
if (udp && nodelay)
usage("can't specify -N and -u");
#ifdef notdef
if (udp == 0 && broadcast)
usage("can't specify -B with TCP");
#endif
if (udp == 0 && foreignip[0] != 0)
usage("can't specify -f with TCP");
if (client) {
if (optind != argc-2)
usage("missing <hostname> and/or <port>");
host = argv[optind];
port = argv[optind+1];
} else {
/* If server specifies host and port, then local address is
bound to the "host" argument, instead of being wildcarded. */
if (optind == argc-2) {
host = argv[optind];
port = argv[optind+1];
} else if (optind == argc-1) {
host = NULL;
port = argv[optind];
} else
usage("missing <port>");
}
if (client)
fd = cliopen(host, port);
else
fd = servopen(host, port);
if (sourcesink) { /* ignore stdin/stdout */
if (client) {
if (udp)
source_udp(fd);
else
source_tcp(fd);
} else {
if (udp)
sink_udp(fd);
else
sink_tcp(fd);
}
} else { /* copy stdin/stdout to/from socket */
if (udp)
loop_udp(fd);
else
loop_tcp(fd);
}
exit(0);
}
static void
usage(const char *msg)
{
err_msg(
"usage: sock [ options ] <host> <port> (for client; default)\n"
" sock [ options ] -s [ <IPaddr> ] <port> (for server)\n"
" sock [ options ] -i <host> <port> (for \"source\" client)\n"
" sock [ options ] -i -s [ <IPaddr> ] <port> (for \"sink\" server)\n"
"options: -b n bind n as client's local port number\n"
" -c convert newline to CR/LF & vice versa\n"
" -f a.b.c.d.p foreign IP address = a.b.c.d, foreign port# = p\n"
" -g a.b.c.d loose source route\n"
" -h issue TCP half close on standard input EOF\n"
" -i \"source\" data to socket, \"sink\" data from socket (w/-s)\n"
#ifdef IP_ADD_MEMBERSHIP
" -j a.b.c.d join multicast group\n"
#endif
" -k write or writev in chunks\n"
" -l a.b.c.d.p client's local IP address = a.b.c.d, local port# = p\n"
" -n n #buffers to write for \"source\" client (default 1024)\n"
" -o do NOT connect UDP client\n"
" -p n #ms to pause before each read or write (source/sink)\n"
" -q n size of listen queue for TCP server (default 5)\n"
" -r n #bytes per read() for \"sink\" server (default 1024)\n"
" -s operate as server instead of client\n"
#ifdef IP_MULTICAST_TTL
" -t n set multicast ttl\n"
#endif
" -u use UDP instead of TCP\n"
" -v verbose\n"
" -w n #bytes per write() for \"source\" client (default 1024)\n"
" -x n #ms for SO_RCVTIMEO (receive timeout)\n"
" -y n #ms for SO_SNDTIMEO (send timeout)\n"
" -A SO_REUSEADDR option\n"
" -B SO_BROADCAST option\n"
" -C set terminal to cbreak mode\n"
" -D SO_DEBUG option\n"
" -E IP_RECVDSTADDR option\n"
" -F fork after connection accepted (TCP concurrent server)\n"
" -G a.b.c.d strict source route\n"
#ifdef IP_TOS
" -H n IP_TOS option (16=min del, 8=max thru, 4=max rel, 2=min$)\n"
#endif
" -I SIGIO signal\n"
#ifdef IP_TTL
" -J n IP_TTL option\n"
#endif
" -K SO_KEEPALIVE option\n"
" -L n SO_LINGER option, n = linger time\n"
" -N TCP_NODELAY option\n"
" -O n #ms to pause after listen, but before first accept\n"
" -P n #ms to pause before first read or write (source/sink)\n"
" -Q n #ms to pause after receiving FIN, but before close\n"
" -R n SO_RCVBUF option\n"
" -S n SO_SNDBUF option\n"
#ifdef SO_REUSEPORT
" -T SO_REUSEPORT option\n"
#endif
" -U n enter urgent mode before write number n (source only)\n"
" -V use writev() instead of write(); enables -k too\n"
" -W ignore write errors for sink client\n"
" -X n TCP_MAXSEG option (set MSS)\n"
" -Y SO_DONTROUTE option\n"
" -Z MSG_PEEK\n"
#ifdef IP_ONESBCAST
" -2 IP_ONESBCAST option (255.255.255.255 for broadcast\n"
#endif
);
if (msg[0] != 0)
err_quit("%s", msg);
exit(1);
}
syntax highlighted by Code2HTML, v. 0.9.1