4.4BSD/usr/src/sys/tests/netccitt/xi_src.c
/*-
* Copyright (c) 1988, 1991 The Regents of the University of California.
* 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.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the University of
* California, Berkeley and its contributors.
* 4. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS 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 REGENTS 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.
*/
#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1988, 1991 The Regents of the University of California.\n\
All rights reserved.\n";
#endif /* not lint */
#ifndef lint
static char sccsid[] = "@(#)xi_src.c 7.6 (Berkeley) 7/15/91";
#endif /* not lint */
/*
* This is a test program to be a source for X.25 connections.
*/
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/uio.h>
#include <sys/ioctl.h>
#include <net/route.h>
#include <net/if.h>
#include <netccitt/x25.h>
#include <stdio.h>
#include <errno.h>
#include <ctype.h>
#include <netdb.h>
#define dbprintf if(verbose)printf
#ifdef __STDC__
#define try(a,b,c) {x = (a b); dbprintf("%s%s returns %d\n",c,#a,x);\
if(x<0) {perror(#a); exit(1);}}
#else
#define try(a,b,c) {x = (a b); dbprintf("%s%s returns %d\n",c,"a",x);\
if(x<0) {perror("a"); exit(1);}}
#endif
fd_set readfds, writefds, exceptfds;
long size, count = 10;
int verbose = 1, selectp, type = SOCK_STREAM, nobuffs, errno, playtag = 0;
int verify = 0, mqdata, protolisten = 0, echop = 0;
unsigned char protodata[2] = {0, 1};
short portnumber = 3000;
struct sockaddr_x25 to;
char your_it[] = "You're it!";
char *port, *conndata, data_msg[2048];
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;
u_long len;
int handy;
while(--argc > 0) {
av++;
if (strcmp(*av,"dest")==0) {
av++;
ccitt_addr(*av, &to);
argc--;
} else if (strcmp(*av,"count")==0) {
av++;
sscanf(*av,"%ld",&count);
argc--;
} else if (strcmp(*av,"size")==0) {
av++;
sscanf(*av,"%ld",&size);
argc--;
} else if (strcmp(*av,"protolisten")==0) {
av++;
protolisten = 1;
sscanf(*av,"%ld",&handy);
argc--;
protodata[0] = handy;
} else if (strcmp(*av,"protonolisten")==0) {
av++;
protolisten = 1;
sscanf(*av,"%ld",&handy);
argc--;
protodata[0] = handy;
protodata[1] = 0;
}
}
xisrc();
}
xisrc() {
int x, s, pid, on = 1, flags = 8, n;
try(socket, (AF_CCITT, type, 0),"");
s = x;
/*try(setsockopt, (s, SOL_SOCKET, SO_DEBUG, &on, sizeof (on)), "");*/
if (protolisten) {
try(setsockopt, (s, CCITTPROTO_X25, PK_PRLISTEN,
protodata, sizeof (protodata)), "");
exit(0);
}
to.x25_opts.op_flags |= X25_MQBIT;
try(connect, (s, (struct sockaddr *) &to, to.x25_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];
struct savebuf {
struct savebuf *s_next;
struct savebuf *s_prev;
int s_n;
int s_flags;
} savebuf = {&savebuf, &savebuf};
void
savedata(n, flags)
int n;
{
register struct savebuf *s = (struct savebuf *)malloc(n + sizeof *s);
if (s == 0)
return;
insque(s, savebuf.s_prev);
s->s_n = n;
s->s_flags = flags;
if (verify)
bcopy(iov->iov_base, (char *)(s + 1), n);
}
checkback(ns)
int ns;
{
int n;
register struct savebuf *s = savebuf.s_next, *t;
while (s != &savebuf) {
iov->iov_len = s->s_n;
n = recvmsg(ns, &msg, s->s_flags);
dbprintf("echoed %d\n", n);
if (verify &&
(n != s->s_n || bcmp((char *)(s + 1), iov->iov_base, n)))
dbprintf("mismatched data or length was %d got %d\n",
s->s_n, n);
t = s; s = s->s_next; remque(t); free((char *)t);
}
}
put_record(s, flags)
int s, flags;
{
int fd, buflen;
char *buf;
int x, saved_x;
msg.msg_flags = flags;
if (verbose) {
unsigned short *zp, *zlim;
if (msg.msg_controllen) {
zp = (unsigned short *)&(cm.cm.cmhdr.cmsg_len);
printf("(CMessage Type is %x) ", cm.cm.cmhdr.cmsg_type);
printf("CMsg data: ");
x = msg.msg_controllen;
zlim = zp + ((x + 1) / 2);
while (zp < zlim) printf("%x ", *zp++);
putchar ('\n');
}
if (iov->iov_len) {
printf("sending: %s %s",
(flags & MSG_OOB ? "(OOB Data)" : ""),
(flags & MSG_EOR ? "(Record Mark)" : ""));
x = localsize;
zp = (unsigned short *)data_msg;
zlim = zp + ((x + 1) / 2);
while (zp < zlim) printf("%x ", *zp++);
putchar ('\n');
}
}
if (echop)
savedata(iov->iov_len, flags);
try(sendmsg, (s, &msg, flags), " put_record ");
saved_x = x;
if (echop && (iov->iov_base[0] & 0x80))
checkback(s);
msg.msg_control = 0;
return (saved_x);
}
int *datasize = &iov->iov_len;
char *cp, *cplim;
get_control_data(type, level)
{
datasize = (int *)&msg.msg_controllen;
cp = cm.cm.cmdata;
cplim = cp + sizeof(cm.cm.cmdata);
cm.cm.cmhdr.cmsg_level = level;
cm.cm.cmhdr.cmsg_type = type;
msg.msg_control = cm.data;
}
get_record(flags)
int *flags;
{
int factor = 1, x = 0;
char workbuf[10240];
static repeatcount, repeatsize;
if (repeatcount > 0) {
repeatcount--;
return;
}
*flags = 0;
*datasize = 0;
datasize = &iov->iov_len;
cp = data_msg + 1;
cplim = data_msg + sizeof(data_msg);
*data_msg = 0;
for(;;) {
x = scanf("%s", workbuf);
if (x == EOF)
break;
if (strcmp(workbuf, "oob") == 0)
*flags |= MSG_OOB;
else if (strcmp(workbuf, "qbit") == 0)
*data_msg |= 0x80;
else if (strcmp(workbuf, "mbit") == 0)
*data_msg |= 0x40;
else if (strcmp(workbuf, "eom") == 0)
*flags |= MSG_EOR;
else if (strcmp(workbuf, "factor") == 0) {
x = scanf("%d", &factor);
if (factor <= 0) factor = 1;
if (x == EOF)
break;
} else if (strcmp(workbuf, "repeat") == 0) {
x = scanf("%d", &repeatcount);
if (repeatcount <= 0) repeatcount = 1;
if (x == EOF)
break;
} else {
int len = strlen(workbuf);
localsize = 1;
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);
repeatsize = cm.cm.cmhdr.cmsg_len = *datasize;
}
break;
}
}
errno = 0;
return (x);
}