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 #include <sys/cdefs.h> 35 36 __FBSDID("$FreeBSD$"); 37 38 #ifdef lint 39 static const char sccsid[] = "@(#)netcmds.c 8.1 (Berkeley) 6/6/93"; 40 #endif 41 42 /* 43 * Common network command support routines. 44 */ 45 #include <sys/param.h> 46 #include <sys/queue.h> 47 #include <sys/socket.h> 48 #include <sys/socketvar.h> 49 #include <sys/protosw.h> 50 51 #include <net/route.h> 52 #include <netinet/in.h> 53 #include <netinet/in_systm.h> 54 #include <netinet/ip.h> 55 #include <netinet/in_pcb.h> 56 #include <arpa/inet.h> 57 58 #include <ctype.h> 59 #include <netdb.h> 60 #include <stdlib.h> 61 #include <string.h> 62 63 #include "systat.h" 64 #include "extern.h" 65 66 #define streq(a,b) (strcmp(a,b)==0) 67 68 static struct hitem { 69 struct in_addr addr; 70 int onoff; 71 } *hosts; 72 73 int nports, nhosts, protos; 74 75 static void changeitems(const char *, int); 76 static int selectproto(const char *); 77 static void showprotos(void); 78 static int selectport(long, int); 79 static void showports(void); 80 static int selecthost(struct in_addr *, int); 81 static void showhosts(void); 82 83 int 84 netcmd(cmd, args) 85 const char *cmd, *args; 86 { 87 88 if (prefix(cmd, "proto")) { 89 if (*args == '\0') { 90 move(CMDLINE, 0); 91 clrtoeol(); 92 addstr("which proto?"); 93 } else if (!selectproto(args)) { 94 error("%s: Unknown protocol.", args); 95 } 96 return (1); 97 } 98 if (prefix(cmd, "ignore") || prefix(cmd, "display")) { 99 changeitems(args, prefix(cmd, "display")); 100 return (1); 101 } 102 if (prefix(cmd, "reset")) { 103 selectproto(0); 104 selecthost(0, 0); 105 selectport(-1, 0); 106 return (1); 107 } 108 if (prefix(cmd, "show")) { 109 move(CMDLINE, 0); clrtoeol(); 110 if (*args == '\0') { 111 showprotos(); 112 showhosts(); 113 showports(); 114 return (1); 115 } 116 if (prefix(args, "protos")) 117 showprotos(); 118 else if (prefix(args, "hosts")) 119 showhosts(); 120 else if (prefix(args, "ports")) 121 showports(); 122 else 123 addstr("show what?"); 124 return (1); 125 } 126 return (0); 127 } 128 129 130 static void 131 changeitems(args, onoff) 132 const char *args; 133 int onoff; 134 { 135 char *cp, *tmpstr, *tmpstr1; 136 struct servent *sp; 137 struct hostent *hp; 138 struct in_addr in; 139 140 tmpstr = tmpstr1 = strdup(args); 141 cp = index(tmpstr1, '\n'); 142 if (cp) 143 *cp = '\0'; 144 for (;;tmpstr1 = cp) { 145 for (cp = tmpstr1; *cp && isspace(*cp); cp++) 146 ; 147 tmpstr1 = cp; 148 for (; *cp && !isspace(*cp); cp++) 149 ; 150 if (*cp) 151 *cp++ = '\0'; 152 if (cp - tmpstr1 == 0) 153 break; 154 sp = getservbyname(tmpstr1, 155 protos == TCP ? "tcp" : protos == UDP ? "udp" : 0); 156 if (sp) { 157 selectport(sp->s_port, onoff); 158 continue; 159 } 160 hp = gethostbyname(tmpstr1); 161 if (hp == 0) { 162 in.s_addr = inet_addr(tmpstr1); 163 if ((int)in.s_addr == -1) { 164 error("%s: unknown host or port", tmpstr1); 165 continue; 166 } 167 } else 168 in = *(struct in_addr *)hp->h_addr; 169 selecthost(&in, onoff); 170 } 171 free(tmpstr); 172 } 173 174 static int 175 selectproto(proto) 176 const char *proto; 177 { 178 179 if (proto == 0 || streq(proto, "all")) 180 protos = TCP | UDP; 181 else if (streq(proto, "tcp")) 182 protos = TCP; 183 else if (streq(proto, "udp")) 184 protos = UDP; 185 else 186 return (0); 187 188 return (protos); 189 } 190 191 static void 192 showprotos() 193 { 194 195 if ((protos&TCP) == 0) 196 addch('!'); 197 addstr("tcp "); 198 if ((protos&UDP) == 0) 199 addch('!'); 200 addstr("udp "); 201 } 202 203 static struct pitem { 204 long port; 205 int onoff; 206 } *ports; 207 208 static int 209 selectport(port, onoff) 210 long port; 211 int onoff; 212 { 213 register struct pitem *p; 214 215 if (port == -1) { 216 if (ports == 0) 217 return (0); 218 free((char *)ports), ports = 0; 219 nports = 0; 220 return (1); 221 } 222 for (p = ports; p < ports+nports; p++) 223 if (p->port == port) { 224 p->onoff = onoff; 225 return (0); 226 } 227 if (nports == 0) 228 ports = (struct pitem *)malloc(sizeof (*p)); 229 else 230 ports = (struct pitem *)realloc(ports, (nports+1)*sizeof (*p)); 231 p = &ports[nports++]; 232 p->port = port; 233 p->onoff = onoff; 234 return (1); 235 } 236 237 int 238 checkport(inp) 239 register struct inpcb *inp; 240 { 241 register struct pitem *p; 242 243 if (ports) 244 for (p = ports; p < ports+nports; p++) 245 if (p->port == inp->inp_lport || p->port == inp->inp_fport) 246 return (p->onoff); 247 return (1); 248 } 249 250 static void 251 showports() 252 { 253 register struct pitem *p; 254 struct servent *sp; 255 256 for (p = ports; p < ports+nports; p++) { 257 sp = getservbyport(p->port, 258 protos == (TCP|UDP) ? 0 : protos == TCP ? "tcp" : "udp"); 259 if (!p->onoff) 260 addch('!'); 261 if (sp) 262 printw("%s ", sp->s_name); 263 else 264 printw("%d ", p->port); 265 } 266 } 267 268 static int 269 selecthost(in, onoff) 270 struct in_addr *in; 271 int onoff; 272 { 273 register struct hitem *p; 274 275 if (in == 0) { 276 if (hosts == 0) 277 return (0); 278 free((char *)hosts), hosts = 0; 279 nhosts = 0; 280 return (1); 281 } 282 for (p = hosts; p < hosts+nhosts; p++) 283 if (p->addr.s_addr == in->s_addr) { 284 p->onoff = onoff; 285 return (0); 286 } 287 if (nhosts == 0) 288 hosts = (struct hitem *)malloc(sizeof (*p)); 289 else 290 hosts = (struct hitem *)realloc(hosts, (nhosts+1)*sizeof (*p)); 291 p = &hosts[nhosts++]; 292 p->addr = *in; 293 p->onoff = onoff; 294 return (1); 295 } 296 297 int 298 checkhost(inp) 299 register struct inpcb *inp; 300 { 301 register struct hitem *p; 302 303 if (hosts) 304 for (p = hosts; p < hosts+nhosts; p++) 305 if (p->addr.s_addr == inp->inp_laddr.s_addr || 306 p->addr.s_addr == inp->inp_faddr.s_addr) 307 return (p->onoff); 308 return (1); 309 } 310 311 static void 312 showhosts() 313 { 314 register struct hitem *p; 315 struct hostent *hp; 316 317 for (p = hosts; p < hosts+nhosts; p++) { 318 hp = gethostbyaddr((char *)&p->addr, sizeof (p->addr), AF_INET); 319 if (!p->onoff) 320 addch('!'); 321 printw("%s ", hp ? hp->h_name : (char *)inet_ntoa(p->addr)); 322 } 323 } 324