Fix various whitespace and indent issues (no functional change).
[dragonfly.git] / usr.bin / systat / netcmds.c
CommitLineData
b638a263
SW
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 * @(#)netcmds.c 8.1 (Berkeley) 6/6/93
34 * $FreeBSD: src/usr.bin/systat/netcmds.c,v 1.9 1999/08/28 01:06:04 peter Exp $
b638a263
SW
35 */
36
37/*
38 * Common network command support routines.
39 */
40#include <sys/param.h>
41#include <sys/queue.h>
42#include <sys/socket.h>
43#include <sys/socketvar.h>
44#include <sys/protosw.h>
45
46#include <net/route.h>
47#include <netinet/in.h>
48#include <netinet/in_systm.h>
49#include <netinet/ip.h>
50#include <netinet/in_pcb.h>
51#include <arpa/inet.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(char *, int);
70static int selectproto(char *);
71static void showprotos(void);
72static int selectport(long, int);
73static void showports(void);
74static int selecthost(struct in_addr *, int);
75static void showhosts(void);
76
77int
78netcmd(const char *cmd, char *args)
79{
80
81 if (prefix(cmd, "proto")) {
82 if (*args == '\0') {
83 move(CMDLINE, 0);
84 clrtoeol();
85 addstr("which proto?");
86 } else if (!selectproto(args)) {
87 error("%s: Unknown protocol.", args);
88 }
89 return (1);
90 }
91 if (prefix(cmd, "ignore") || prefix(cmd, "display")) {
92 changeitems(args, prefix(cmd, "display"));
93 return (1);
94 }
95 if (prefix(cmd, "reset")) {
96 selectproto(0);
97 selecthost(0, 0);
98 selectport(-1, 0);
99 return (1);
100 }
101 if (prefix(cmd, "show")) {
102 move(CMDLINE, 0); clrtoeol();
103 if (*args == '\0') {
104 showprotos();
105 showhosts();
106 showports();
107 return (1);
108 }
109 if (prefix(args, "protos"))
110 showprotos();
111 else if (prefix(args, "hosts"))
112 showhosts();
113 else if (prefix(args, "ports"))
114 showports();
115 else
116 addstr("show what?");
117 return (1);
118 }
119 return (0);
120}
121
122
123static void
124changeitems(char *args, int onoff)
125{
126 char *cp;
127 struct servent *sp;
128 struct hostent *hp;
129 struct in_addr in;
130
131 cp = strchr(args, '\n');
132 if (cp)
133 *cp = '\0';
134 for (;;args = cp) {
135 for (cp = args; *cp && isspace(*cp); cp++)
136 ;
137 args = cp;
138 for (; *cp && !isspace(*cp); cp++)
139 ;
140 if (*cp)
141 *cp++ = '\0';
142 if (cp - args == 0)
143 break;
144 sp = getservbyname(args,
145 protos == TCP ? "tcp" : protos == UDP ? "udp" : 0);
146 if (sp) {
147 selectport(sp->s_port, onoff);
148 continue;
149 }
150 hp = gethostbyname(args);
151 if (hp == 0) {
152 in.s_addr = inet_addr(args);
153 if ((int)in.s_addr == -1) {
154 error("%s: unknown host or port", args);
155 continue;
156 }
157 } else
158 in = *(struct in_addr *)hp->h_addr;
159 selecthost(&in, onoff);
160 }
161}
162
163static int
164selectproto(char *proto)
165{
166
167 if (proto == 0 || streq(proto, "all"))
168 protos = TCP | UDP;
169 else if (streq(proto, "tcp"))
170 protos = TCP;
171 else if (streq(proto, "udp"))
172 protos = UDP;
173 else
174 return (0);
175
176 return (protos);
177}
178
179static void
180showprotos(void)
181{
182
183 if ((protos&TCP) == 0)
184 addch('!');
185 addstr("tcp ");
186 if ((protos&UDP) == 0)
187 addch('!');
188 addstr("udp ");
189}
190
191static struct pitem {
192 long port;
193 int onoff;
194} *ports;
195
196static int
197selectport(long port, int onoff)
198{
199 struct pitem *p;
200
201 if (port == -1) {
202 if (ports == 0)
203 return (0);
204 free((char *)ports), ports = 0;
205 nports = 0;
206 return (1);
207 }
208 for (p = ports; p < ports+nports; p++)
209 if (p->port == port) {
210 p->onoff = onoff;
211 return (0);
212 }
213 if (nports == 0)
214 ports = (struct pitem *)malloc(sizeof (*p));
215 else
216 ports = (struct pitem *)realloc(ports, (nports+1)*sizeof (*p));
217 p = &ports[nports++];
218 p->port = port;
219 p->onoff = onoff;
220 return (1);
221}
222
223int
224checkport(struct inpcb *inp)
225{
226 struct pitem *p;
227
228 if (ports)
6e278935
SW
229 for (p = ports; p < ports+nports; p++) {
230 if (p->port == inp->inp_lport ||
231 p->port == inp->inp_fport)
232 return (p->onoff);
233 }
b638a263
SW
234 return (1);
235}
236
237static void
238showports(void)
239{
240 struct pitem *p;
241 struct servent *sp;
242
243 for (p = ports; p < ports+nports; p++) {
244 sp = getservbyport(p->port,
245 protos == (TCP|UDP) ? 0 : protos == TCP ? "tcp" : "udp");
246 if (!p->onoff)
247 addch('!');
248 if (sp)
249 printw("%s ", sp->s_name);
250 else
251 printw("%d ", p->port);
252 }
253}
254
255static int
256selecthost(struct in_addr *in, int onoff)
257{
258 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(struct inpcb *inp)
284{
285 struct hitem *p;
286
287 if (hosts)
6e278935
SW
288 for (p = hosts; p < hosts+nhosts; p++) {
289 if (p->addr.s_addr == inp->inp_laddr.s_addr ||
290 p->addr.s_addr == inp->inp_faddr.s_addr)
291 return (p->onoff);
292 }
b638a263
SW
293 return (1);
294}
295
296static void
297showhosts(void)
298{
299 struct hitem *p;
300 struct hostent *hp;
301
302 for (p = hosts; p < hosts+nhosts; p++) {
303 hp = gethostbyaddr(&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}