xref: /csrg-svn/sys/tests/netiso/tisink.c (revision 44424)
1 /*
2  * Copyright (c) 1988, 1990 Regents of the University of California.
3  * All rights reserved.
4  *
5  * %sccs.include.redist.c%
6  */
7 #ifndef lint
8 static char sccsid[] = "@(#)tisink.c	7.2 (Berkeley) 06/28/90";
9 #endif /* not lint */
10 
11 /*
12  * This is a test program to be a sink for TP4 connections.
13  */
14 #include <sys/param.h>
15 #include <sys/uio.h>
16 #include <sys/socket.h>
17 #include <sys/ioctl.h>
18 #include <net/route.h>
19 #include <net/if.h>
20 #define  TCPT_NTIMERS 4
21 #include <netiso/iso.h>
22 #include <netiso/tp_param.h>
23 #include <netiso/tp_user.h>
24 
25 #include <stdio.h>
26 #include <errno.h>
27 #include <ctype.h>
28 #include <netdb.h>
29 
30 
31 #define dbprintf if(verbose)printf
32 #define try(a,b,c) {x = (a b); dbprintf("%s%s returns %d\n",c,"a",x);\
33 		if(x<0) {perror("a"); myexit(0);}}
34 
35 
36 struct  ifreq ifr;
37 short port = 3000;
38 struct  sockaddr_iso faddr, laddr = { sizeof(laddr), AF_ISO };
39 struct  sockaddr_iso *siso = &laddr;
40 
41 long size, count = 10, forkp, confp, echop, mynamep, verbose = 1, playtag = 0;
42 long records, intercept;
43 
44 char buf[2048];
45 char your_it[] = "You're it!";
46 
47 char *Servername;
48 
49 main(argc, argv)
50 int argc;
51 char *argv[];
52 {
53 	register char **av = argv;
54 	register char *cp;
55 	struct iso_addr iso_addr();
56 
57 	while(--argc > 0) {
58 		av++;
59 		if(strcmp(*av,"Servername")==0) {
60 			av++;
61 			Servername = *av;
62 			argc--;
63 		} else if (strcmp(*av,"host")==0) {
64 			av++;
65 			laddr.siso_addr = iso_addr(*av);
66 			argc--;
67 		} else if (strcmp(*av,"count")==0) {
68 			av++;
69 			sscanf(*av,"%ld",&count);
70 			argc--;
71 		} else if (strcmp(*av,"port")==0) {
72 			av++;
73 			sscanf(*av,"%hd",&port);
74 			argc--;
75 		} else if (strcmp(*av,"size")==0) {
76 			av++;
77 			sscanf(*av,"%ld",&size);
78 			argc--;
79 		}
80 	}
81 	if (Servername) {
82 		int tlen = laddr.siso_tlen = strlen(Servername);
83 		int len =  TSEL(siso) + tlen - (caddr_t) &siso;
84 		if (len > sizeof(*siso)) {
85 			siso = (struct sockaddr_iso *)malloc(len);
86 			*siso = laddr;
87 			siso->siso_len = len;
88 		}
89 		bcopy(Servername, TSEL(siso), tlen);
90 	} else {
91 		port = htons(port);
92 		laddr.siso_tlen = sizeof(port);
93 		bcopy((char *)&port, TSEL(siso), sizeof(port));
94 	}
95 	tisink();
96 }
97 #define BIG 2048
98 #define MIDLIN 512
99 char readbuf[BIG];
100 struct iovec iov[1] = {
101 	readbuf,
102 	sizeof readbuf,
103 };
104 char name[MIDLIN];
105 union {
106     struct {
107 	    struct cmsghdr	cmhdr;
108 	    char		cmdata[128 - sizeof(struct cmsghdr)];
109     } cm;
110     char data[128];
111 } cbuf;
112 #define control cbuf.data
113 struct msghdr msghdr = {
114 	name, sizeof(name),
115 	iov, sizeof(iov)/sizeof(iov[1]),
116 	control, sizeof control,
117 	0 /* flags */
118 };
119 
120 tisink()
121 {
122 	int x, s, pid, on = 1, loop = 0, n;
123 	extern int errno;
124 
125 	try(socket, (AF_ISO, SOCK_SEQPACKET, 0),"");
126 
127 	s = x;
128 
129 	try(bind, (s, (struct sockaddr *) siso, siso->siso_len), "");
130 
131 	/*try(setsockopt, (s, SOL_SOCKET, SO_DEBUG, &on, sizeof (on)), ""); */
132 
133 	try(listen, (s, 5), "");
134 	if (intercept) {
135 	    try(setsockopt,
136 		(s, SOL_TRANSPORT, TPOPT_INTERCEPT, &on, sizeof(on)), "");
137 	}
138 	for(;;) {
139 		int child, ns;
140 		int addrlen = sizeof(faddr);
141 		char childname[50];
142 
143 		try (accept, (s, &faddr, &addrlen), "");
144 		ns = x;
145 		dumpit("connection from:", &faddr, sizeof faddr);
146 		if (mynamep) {
147 			addrlen = sizeof(faddr);
148 			try (getsockname, (ns, &faddr, &addrlen), "");
149 			dumpit("connected as:", &faddr, addrlen);
150 		}
151 		loop++;
152 		if (loop > 3) myexit(0);
153 		if (forkp) {
154 			try(fork, (), "");
155 		} else
156 			x = 0;
157 		if (x == 0)  {
158 		    long n, count = 0, cn, flags;
159 		    records = 0;
160 		    if (confp) {
161 			msghdr.msg_iovlen = 0;
162 			msghdr.msg_namelen = 0;
163 			msghdr.msg_controllen =
164 			    cbuf.cm.cmhdr.cmsg_len = sizeof (cbuf.cm.cmhdr);
165 			cbuf.cm.cmhdr.cmsg_level = SOL_TRANSPORT;
166 			cbuf.cm.cmhdr.cmsg_type = TPOPT_CFRM_DATA;
167 			n = sendmsg(ns, &msghdr, 0);
168 			if (n <= 0) {
169 				printf("confirm: errno is %d\n", errno);
170 				fflush(stdout);
171 				perror("Confirm error");
172 			} else {
173 				dbprintf("confim ok\n");
174 			}
175 			sleep(10);
176 		    }
177 		    for (;;) {
178 			msghdr.msg_iovlen = 1;
179 			msghdr.msg_controllen = sizeof(control);
180 			iov->iov_len = sizeof(readbuf);
181 			n = recvmsg(ns, &msghdr, 0);
182 			flags = msghdr.msg_flags;
183 			count++;
184 			dbprintf("recvmsg from child %d got %d ctl %d flags %x\n",
185 				    getpid(), n, (cn = msghdr.msg_controllen),
186 					flags);
187 			fflush(stdout);
188 			if (cn && verbose)
189 				dumpit("control data:\n", control, cn);
190 			if (n < 0) {
191 				fprintf(stderr, "errno is %d\n", errno);
192 				perror("recvmsg");
193 				/*sleep (10);*/
194 				break;
195 			} else {
196 				if (verbose)
197 					dumpit("data:\n", readbuf, n);
198 			}
199 			if (echop) {
200 				n = answerback(flags, n, ns);
201 			}
202 			if (flags & MSG_EOR)
203 				records++;
204 			if (playtag && n == sizeof(your_it) && (flags & MSG_EOR)
205 			    && bcmp(readbuf, your_it, n) == 0) {
206 				printf("Answering back!!!!\n");
207 				answerback(flags, n, ns);
208 				answerback(flags, n, ns);
209 			}
210 			errno = 0;
211 		    }
212 		}
213 		myexit(0);
214 	}
215 }
216 answerback(flags, n, ns)
217 {
218 	msghdr.msg_controllen = 0;
219 	msghdr.msg_iovlen = 1;
220 	iov->iov_len = n;
221 	n = sendmsg(ns, &msghdr, flags);
222 	dbprintf("echoed %d\n", n);
223 	return n;
224 }
225 
226 dumpit(what, where, n)
227 char *what; unsigned short *where; int n;
228 {
229 	unsigned short *s = where;
230 	unsigned short *z = where + (n+1)/2;
231 	int count = 0;
232 	printf(what);
233 	while(s < z) {
234 		count++;
235 		printf("%x ",*s++);
236 		if ((count & 15) == 0)
237 			putchar('\n');
238 	}
239 	if (count & 15)
240 		putchar('\n');
241 	fflush(stdout);
242 }
243 myexit(n)
244 {
245 	fflush(stderr);
246 	printf("got %d records\n", records);
247 	fflush(stdout);
248 	exit(n);
249 }
250