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