|
|
BSD 4.3reno
/*
* Copyright (c) 1989, 1990 Regents of the University of California.
* All rights reserved.
*
* %sccs.include.redist.c%
*/
#ifndef lint
static char sccsid[] = "%W% (Berkeley) %G%";
#endif /* not lint */
/*
* This is a test program to be a source for ISO packets.
*/
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/uio.h>
#include <sys/ioctl.h>
#include <net/route.h>
#include <net/if.h>
#define TCPT_NTIMERS 4
#include <netiso/iso.h>
#include <netiso/tp_user.h>
#include <stdio.h>
#include <errno.h>
#include <ctype.h>
#include <netdb.h>
#define dbprintf if(verbose)printf
#define try(a,b,c) {x = (a b);dbprintf("%s%s returns %d\n",c,"a",x);\
if (x < 0) {perror("a"); exit(1);}}
struct iso_addr eon = {20, 0x47, 0, 6, 3, 0, 0, 0, 25 /*EGP for Berkeley*/};
struct sockaddr_iso to_s = { sizeof(to_s), AF_ISO }, *to = &to_s;
struct sockaddr_iso old_s = { sizeof(to_s), AF_ISO }, *old = &old_s;
fd_set readfds, writefds, exceptfds;
long size, count = 10;
int verbose = 1, selectp, type = SOCK_SEQPACKET, nobuffs, errno, playtag = 0;
int verify = 0, dgramp = 1, debug = 0;
short portnumber = 3000;
char your_it[] = "You're it!";
char *Servername, *conndata, data_msg[2048];
char Serverbuf[128];
char name[128];
struct iovec iov[1] = {data_msg};
union {
struct {
struct cmsghdr cmhdr;
char cmdata[128 - sizeof(struct cmsghdr)];
} cm;
char data[128];
} cm;
struct msghdr msg = { 0, 0, iov, 1, 0, 0, 0};
main(argc, argv)
int argc;
char *argv[];
{
register char **av = argv;
register char *cp;
struct iso_addr iso_addr();
u_long len;
int handy;
while(--argc > 0) {
av++;
if(strcmp(*av,"Servername")==0) {
av++;
Servername = *av;
argc--;
} else if(strcmp(*av,"conndata")==0) {
av++;
conndata = *av;
argc--;
} else if(strcmp(*av,"host")==0) {
av++;
to_s.siso_addr = iso_addr(*av);
argc--;
} else if(strcmp(*av,"port")==0) {
av++;
sscanf(*av,"%hd",&portnumber);
argc--;
} else if(strcmp(*av,"count")==0) {
av++;
sscanf(*av,"%ld",&count);
argc--;
} else if(strcmp(*av,"size")==0) {
av++;
sscanf(*av,"%ld",&size);
iov->iov_len = size;
} else if (strcmp(*av,"eon") == 0) {
unsigned long l, inet_addr();
l = inet_addr(*++av); argc--;
to_s.siso_addr = eon;
bcopy((char *)&l, &to_s.siso_data[15], 4);
}
}
maketoaddr();
tisrc();
}
maketoaddr()
{
if (Servername) {
int tlen = strlen(Servername);
int len = tlen + TSEL(to) - (caddr_t) to;
if (len < sizeof(*to)) len = sizeof(*to);
if (len > to->siso_len) {
if (old != &old_s) free(old);
old = (struct sockaddr_iso *)malloc(len);
*old = *to; /* We dont care if all old tsel is copied*/
old->siso_len = len;
if (to != &to_s) free(to);
to = (struct sockaddr_iso *)malloc(len);
}
bcopy(Servername, TSEL(old), tlen);
} else {
old->siso_tlen = sizeof(portnumber);
portnumber = htons(portnumber);
bcopy((char *)&portnumber, TSEL(old), sizeof(portnumber));
}
bcopy(old, to, old->siso_len);
}
tisrc() {
int x, s, pid, on = 1, flags = 8, n;
if (dgramp) type = SOCK_DGRAM;
try(socket, (AF_ISO, type, 0),"");
s = x;
if (debug)
try(setsockopt, (s, SOL_SOCKET, SO_DEBUG, &on, sizeof on), "");
if (dgramp == 0) {
if (conndata)
doconndata(s);
try(connect, (s, (struct sockaddr *) to, to->siso_len), "");
}
if (selectp) {
FD_ZERO(&writefds); FD_SET(s, &writefds);
select(1, &writefds, 0, 0, 0);
}
while (count-- > 0) {
if (size <= 0 && get_record(&flags) == EOF)
exit(0);
n = put_record(s, flags);
if (n < iov->iov_len) {
if (n==-1 && errno == 55) {
nobuffs++;
count++;
continue;
}
fprintf(stderr, "wrote %d < %d, count %d,",
n, iov->iov_len, count);
perror("due to");
}
}
if (playtag) {
printf("Tag time!\n");
iov->iov_base = your_it;
iov->iov_len = sizeof your_it;
sendmsg(s, &msg, MSG_EOR);
sendmsg(s, &msg, MSG_EOR);
iov->iov_base = data_msg;
iov->iov_len = sizeof data_msg;
try(recvmsg, (s, &msg, flags), " playtag ");
}
if(nobuffs) {
printf("looped %d times waiting for bufs\n", nobuffs);
}
}
int localsize;
char dupbuf[4096];
put_record(s, flags)
int s, flags;
{
int fd, buflen;
char *buf;
int x, saved_x;
msg.msg_flags = flags;
if (verbose) {
if (msg.msg_controllen) {
printf("(CMessage Type is %x) ", cm.cm.cmhdr.cmsg_type);
dumpit("CMsg data:\n", &msg.msg_control, msg.msg_controllen);
}
if (iov->iov_len) {
printf("sending: %s %s",
(flags & MSG_OOB ? "(OOB Data)" : ""),
(flags & MSG_EOR ? "(Record Mark)" : ""));
dumpit("data: ", data_msg, localsize);
}
}
if (verify) {
buflen = iov->iov_len;
bcopy(iov->iov_base, dupbuf, buflen);
}
if (dgramp) {
msg.msg_name = (caddr_t)to;
msg.msg_namelen = to->siso_len;
}
try(sendmsg, (s, &msg, flags), " put_record ");
saved_x = x;
while (verify && buflen > 0) {
iov->iov_len = buflen;
iov->iov_base = dupbuf;
try(recvmsg, (s, &msg, flags), " put_record ");
if (dgramp) {
if (msg.msg_namelen)
dumpit("from: ", to, msg.msg_namelen);
msg.msg_namelen = old->siso_len;
}
printf("verify got %d\n", x);
buflen -= x;
}
bcopy(old, to, old->siso_len);
msg.msg_control = 0;
return (saved_x);
}
dumpit(what, where, n)
char *what; unsigned short *where; int n;
{
unsigned short *s = where;
unsigned short *z = where + (n+1)/2;
int count = 0;
if (verbose == 0)
return;
printf(what);
while(s < z) {
count++;
printf("%x ",*s++);
if ((count & 15) == 0)
putchar('\n');
}
if (count & 15)
putchar('\n');
fflush(stdout);
}
int *datasize = &iov->iov_len;
char *cp, *cplim;
get_control_data(type)
{
datasize = &msg.msg_controllen;
cp = cm.cm.cmdata;
cplim = cp + sizeof(cm.cm.cmdata);
cm.cm.cmhdr.cmsg_level = SOL_TRANSPORT;
cm.cm.cmhdr.cmsg_type = type;
msg.msg_control = cm.data;
}
doconndata(s)
{
get_control_data(TPOPT_CONN_DATA);
*datasize = strlen(conndata) + sizeof(cm.cm.cmhdr);
cm.cm.cmhdr.cmsg_len = *datasize;
bcopy(conndata, cp, *datasize);
put_record(s, 0);
}
get_altbuf(addrbuf)
char *addrbuf;
{
if (dgramp == 0) {
printf("illegal option for stream\n");
return 1;
}
return (scanf("%s", addrbuf) == EOF ? 1 : 0);
}
get_record(flags)
int *flags;
{
int factor = 1, x = 0, newaddr = 0;
char workbuf[10240];
char addrbuf[128];
*flags = 0;
*datasize = 0;
datasize = &iov->iov_len;
cp = data_msg;
cplim = cp + sizeof(data_msg);
for(;;) {
x = scanf("%s", workbuf);
if (x == EOF)
break;
if (strcmp(workbuf, "disc") == 0)
x = get_control_data(TPOPT_DISC_DATA);
else if (strcmp(workbuf, "cfrm") == 0)
x = get_control_data(TPOPT_CFRM_DATA);
else if (strcmp(workbuf, "oob") == 0)
*flags |= MSG_OOB;
else if (strcmp(workbuf, "eom") == 0)
*flags |= MSG_EOR;
else if (strcmp(workbuf, "host") == 0) {
if (get_altbuf(addrbuf))
break;
to->siso_addr = iso_addr(addrbuf);
newaddr = 1;
} else if (strcmp(workbuf, "Servername") == 0) {
if (get_altbuf(Serverbuf))
break;
Servername = Serverbuf;
newaddr = 1;
} else if (strcmp(workbuf, "port") == 0) {
x = scanf("%hd", &portnumber);
if (x == EOF)
break;
Servername = 0;
newaddr = 1;
} else if (strcmp(workbuf, "factor") == 0) {
x = scanf("%d", &factor);
if (factor <= 0) factor = 1;
if (x == EOF)
break;
} else {
int len = strlen(workbuf);
localsize = 0;
while ((factor-- > 0) &&
((cp + len) < cplim)) {
strcpy(cp, workbuf);
cp += len;
localsize += len;
}
*datasize = localsize;
if (datasize != &iov->iov_len) {
*datasize += sizeof(cm.cm.cmhdr);
cm.cm.cmhdr.cmsg_len = *datasize;
}
break;
}
}
errno = 0;
if (newaddr)
maketoaddr();
return (x);
}
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.