netcmds.c revision 1591 
1/*-
2 * Copyright (c) 1980, 1992, 1993
3 *	The Regents of the University of California.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 *    must display the following acknowledgement:
15 *	This product includes software developed by the University of
16 *	California, Berkeley and its contributors.
17 * 4. Neither the name of the University nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#ifndef lint
35static char sccsid[] = "@(#)netcmds.c	8.1 (Berkeley) 6/6/93";
36#endif /* not lint */
37
38/*
39 * Common network command support routines.
40 */
41#include <sys/param.h>
42#include <sys/socket.h>
43#include <sys/socketvar.h>
44#include <sys/mbuf.h>
45#include <sys/protosw.h>
46
47#include <net/route.h>
48#include <netinet/in.h>
49#include <netinet/in_systm.h>
50#include <netinet/ip.h>
51#include <netinet/in_pcb.h>
52
53#include <netdb.h>
54#include <stdlib.h>
55#include <string.h>
56#include <ctype.h>
57#include "systat.h"
58#include "extern.h"
59
60#define	streq(a,b)	(strcmp(a,b)==0)
61
62static	struct hitem {
63	struct	in_addr addr;
64	int	onoff;
65} *hosts;
66
67int nports, nhosts, protos;
68
69static void changeitems __P((char *, int));
70static int selectproto __P((char *));
71static void showprotos __P((void));
72static int selectport __P((long, int));
73static void showports __P((void));
74static int selecthost __P((struct in_addr *, int));
75static void showhosts __P((void));
76
77int
78netcmd(cmd, args)
79	char *cmd, *args;
80{
81
82	if (prefix(cmd, "tcp") || prefix(cmd, "udp")) {
83		selectproto(cmd);
84		return (1);
85	}
86	if (prefix(cmd, "ignore") || prefix(cmd, "display")) {
87		changeitems(args, prefix(cmd, "display"));
88		return (1);
89	}
90	if (prefix(cmd, "reset")) {
91		selectproto(0);
92		selecthost(0, 0);
93		selectport(-1, 0);
94		return (1);
95	}
96	if (prefix(cmd, "show")) {
97		move(CMDLINE, 0); clrtoeol();
98		if (*args == '\0') {
99			showprotos();
100			showhosts();
101			showports();
102			return (1);
103		}
104		if (prefix(args, "protos"))
105			showprotos();
106		else if (prefix(args, "hosts"))
107			showhosts();
108		else if (prefix(args, "ports"))
109			showports();
110		else
111			addstr("show what?");
112		return (1);
113	}
114	return (0);
115}
116
117
118static void
119changeitems(args, onoff)
120	char *args;
121	int onoff;
122{
123	register char *cp;
124	struct servent *sp;
125	struct hostent *hp;
126	struct in_addr in;
127	char *index();
128
129	cp = index(args, '\n');
130	if (cp)
131		*cp = '\0';
132	for (;;args = cp) {
133		for (cp = args; *cp && isspace(*cp); cp++)
134			;
135		args = cp;
136		for (; *cp && !isspace(*cp); cp++)
137			;
138		if (*cp)
139			*cp++ = '\0';
140		if (cp - args == 0)
141			break;
142		sp = getservbyname(args,
143		    protos == TCP ? "tcp" : protos == UDP ? "udp" : 0);
144		if (sp) {
145			selectport(sp->s_port, onoff);
146			continue;
147		}
148		hp = gethostbyname(args);
149		if (hp == 0) {
150			in.s_addr = inet_addr(args);
151			if (in.s_addr == -1) {
152				error("%s: unknown host or port", args);
153				continue;
154			}
155		} else
156			in = *(struct in_addr *)hp->h_addr;
157		selecthost(&in, onoff);
158	}
159}
160
161static int
162selectproto(proto)
163	char *proto;
164{
165	int new = protos;
166
167	if (proto == 0 || streq(proto, "all"))
168		new = TCP|UDP;
169	else if (streq(proto, "tcp"))
170		new = TCP;
171	else if (streq(proto, "udp"))
172		new = UDP;
173	return (new != protos, protos = new);
174}
175
176static void
177showprotos()
178{
179
180	if ((protos&TCP) == 0)
181		addch('!');
182	addstr("tcp ");
183	if ((protos&UDP) == 0)
184		addch('!');
185	addstr("udp ");
186}
187
188static	struct pitem {
189	long	port;
190	int	onoff;
191} *ports;
192
193static int
194selectport(port, onoff)
195	long port;
196	int onoff;
197{
198	register struct pitem *p;
199
200	if (port == -1) {
201		if (ports == 0)
202			return (0);
203		free((char *)ports), ports = 0;
204		nports = 0;
205		return (1);
206	}
207	for (p = ports; p < ports+nports; p++)
208		if (p->port == port) {
209			p->onoff = onoff;
210			return (0);
211		}
212	if (nports == 0)
213		ports = (struct pitem *)malloc(sizeof (*p));
214	else
215		ports = (struct pitem *)realloc(ports, (nports+1)*sizeof (*p));
216	p = &ports[nports++];
217	p->port = port;
218	p->onoff = onoff;
219	return (1);
220}
221
222int
223checkport(inp)
224	register struct inpcb *inp;
225{
226	register struct pitem *p;
227
228	if (ports)
229	for (p = ports; p < ports+nports; p++)
230		if (p->port == inp->inp_lport || p->port == inp->inp_fport)
231			return (p->onoff);
232	return (1);
233}
234
235static void
236showports()
237{
238	register struct pitem *p;
239	struct servent *sp;
240
241	for (p = ports; p < ports+nports; p++) {
242		sp = getservbyport(p->port,
243		    protos == TCP|UDP ? 0 : protos == TCP ? "tcp" : "udp");
244		if (!p->onoff)
245			addch('!');
246		if (sp)
247			printw("%s ", sp->s_name);
248		else
249			printw("%d ", p->port);
250	}
251}
252
253static int
254selecthost(in, onoff)
255	struct in_addr *in;
256	int onoff;
257{
258	register struct hitem *p;
259
260	if (in == 0) {
261		if (hosts == 0)
262			return (0);
263		free((char *)hosts), hosts = 0;
264		nhosts = 0;
265		return (1);
266	}
267	for (p = hosts; p < hosts+nhosts; p++)
268		if (p->addr.s_addr == in->s_addr) {
269			p->onoff = onoff;
270			return (0);
271		}
272	if (nhosts == 0)
273		hosts = (struct hitem *)malloc(sizeof (*p));
274	else
275		hosts = (struct hitem *)realloc(hosts, (nhosts+1)*sizeof (*p));
276	p = &hosts[nhosts++];
277	p->addr = *in;
278	p->onoff = onoff;
279	return (1);
280}
281
282int
283checkhost(inp)
284	register struct inpcb *inp;
285{
286	register struct hitem *p;
287
288	if (hosts)
289	for (p = hosts; p < hosts+nhosts; p++)
290		if (p->addr.s_addr == inp->inp_laddr.s_addr ||
291		    p->addr.s_addr == inp->inp_faddr.s_addr)
292			return (p->onoff);
293	return (1);
294}
295
296static void
297showhosts()
298{
299	register struct hitem *p;
300	struct hostent *hp;
301
302	for (p = hosts; p < hosts+nhosts; p++) {
303		hp = gethostbyaddr((char *)&p->addr, sizeof (p->addr), AF_INET);
304		if (!p->onoff)
305			addch('!');
306		printw("%s ", hp ? hp->h_name : (char *)inet_ntoa(p->addr));
307	}
308}
309