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 * 4. Neither the name of the University nor the names of its contributors 14 * may be used to endorse or promote products derived from this software 15 * without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27 * SUCH DAMAGE. 28 */ 29 30 #include <sys/cdefs.h> 31 32 __FBSDID("$FreeBSD$"); 33 34 #ifdef lint 35 static const char sccsid[] = "@(#)netcmds.c 8.1 (Berkeley) 6/6/93"; 36 #endif 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/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 #include <arpa/inet.h> 53 54 #include <ctype.h> 55 #include <netdb.h> 56 #include <stdlib.h> 57 #include <string.h> 58 59 #include "systat.h" 60 #include "extern.h" 61 62 #define streq(a,b) (strcmp(a,b)==0) 63 64 static struct hitem { 65 struct in_addr addr; 66 int onoff; 67 } *hosts; 68 69 int nports, nhosts, protos; 70 71 static void changeitems(const char *, int); 72 static int selectproto(const char *); 73 static void showprotos(void); 74 static int selectport(long, int); 75 static void showports(void); 76 static int selecthost(struct in_addr *, int); 77 static void showhosts(void); 78 79 int 80 netcmd(const char *cmd, const char *args) 81 { 82 83 if (prefix(cmd, "proto")) { 84 if (*args == '\0') { 85 move(CMDLINE, 0); 86 clrtoeol(); 87 addstr("which proto?"); 88 } else if (!selectproto(args)) { 89 error("%s: Unknown protocol.", args); 90 } 91 return (1); 92 } 93 if (prefix(cmd, "ignore") || prefix(cmd, "display")) { 94 changeitems(args, prefix(cmd, "display")); 95 return (1); 96 } 97 if (prefix(cmd, "reset")) { 98 selectproto(0); 99 selecthost(0, 0); 100 selectport(-1, 0); 101 return (1); 102 } 103 if (prefix(cmd, "show")) { 104 move(CMDLINE, 0); clrtoeol(); 105 if (*args == '\0') { 106 showprotos(); 107 showhosts(); 108 showports(); 109 return (1); 110 } 111 if (prefix(args, "protos")) 112 showprotos(); 113 else if (prefix(args, "hosts")) 114 showhosts(); 115 else if (prefix(args, "ports")) 116 showports(); 117 else 118 addstr("show what?"); 119 return (1); 120 } 121 return (0); 122 } 123 124 125 static void 126 changeitems(const char *args, int onoff) 127 { 128 char *cp, *tmpstr, *tmpstr1; 129 struct servent *sp; 130 struct hostent *hp; 131 struct in_addr in; 132 133 tmpstr = tmpstr1 = strdup(args); 134 cp = index(tmpstr1, '\n'); 135 if (cp) 136 *cp = '\0'; 137 for (;;tmpstr1 = cp) { 138 for (cp = tmpstr1; *cp && isspace(*cp); cp++) 139 ; 140 tmpstr1 = cp; 141 for (; *cp && !isspace(*cp); cp++) 142 ; 143 if (*cp) 144 *cp++ = '\0'; 145 if (cp - tmpstr1 == 0) 146 break; 147 sp = getservbyname(tmpstr1, 148 protos == TCP ? "tcp" : protos == UDP ? "udp" : 0); 149 if (sp) { 150 selectport(sp->s_port, onoff); 151 continue; 152 } 153 hp = gethostbyname(tmpstr1); 154 if (hp == 0) { 155 in.s_addr = inet_addr(tmpstr1); 156 if ((int)in.s_addr == -1) { 157 error("%s: unknown host or port", tmpstr1); 158 continue; 159 } 160 } else 161 in = *(struct in_addr *)hp->h_addr; 162 selecthost(&in, onoff); 163 } 164 free(tmpstr); 165 } 166 167 static int 168 selectproto(const char *proto) 169 { 170 171 if (proto == 0 || streq(proto, "all")) 172 protos = TCP | UDP; 173 else if (streq(proto, "tcp")) 174 protos = TCP; 175 else if (streq(proto, "udp")) 176 protos = UDP; 177 else 178 return (0); 179 180 return (protos); 181 } 182 183 static void 184 showprotos(void) 185 { 186 187 if ((protos&TCP) == 0) 188 addch('!'); 189 addstr("tcp "); 190 if ((protos&UDP) == 0) 191 addch('!'); 192 addstr("udp "); 193 } 194 195 static struct pitem { 196 long port; 197 int onoff; 198 } *ports; 199 200 static int 201 selectport(long port, int onoff) 202 { 203 struct pitem *p; 204 205 if (port == -1) { 206 if (ports == 0) 207 return (0); 208 free((char *)ports), ports = 0; 209 nports = 0; 210 return (1); 211 } 212 for (p = ports; p < ports+nports; p++) 213 if (p->port == port) { 214 p->onoff = onoff; 215 return (0); 216 } 217 if (nports == 0) 218 ports = (struct pitem *)malloc(sizeof (*p)); 219 else 220 ports = (struct pitem *)realloc(ports, (nports+1)*sizeof (*p)); 221 p = &ports[nports++]; 222 p->port = port; 223 p->onoff = onoff; 224 return (1); 225 } 226 227 int 228 checkport(struct inpcb *inp) 229 { 230 struct pitem *p; 231 232 if (ports) 233 for (p = ports; p < ports+nports; p++) 234 if (p->port == inp->inp_lport || p->port == inp->inp_fport) 235 return (p->onoff); 236 return (1); 237 } 238 239 static void 240 showports(void) 241 { 242 struct pitem *p; 243 struct servent *sp; 244 245 for (p = ports; p < ports+nports; p++) { 246 sp = getservbyport(p->port, 247 protos == (TCP|UDP) ? 0 : protos == TCP ? "tcp" : "udp"); 248 if (!p->onoff) 249 addch('!'); 250 if (sp) 251 printw("%s ", sp->s_name); 252 else 253 printw("%d ", p->port); 254 } 255 } 256 257 static int 258 selecthost(struct in_addr *in, int onoff) 259 { 260 struct hitem *p; 261 262 if (in == 0) { 263 if (hosts == 0) 264 return (0); 265 free((char *)hosts), hosts = 0; 266 nhosts = 0; 267 return (1); 268 } 269 for (p = hosts; p < hosts+nhosts; p++) 270 if (p->addr.s_addr == in->s_addr) { 271 p->onoff = onoff; 272 return (0); 273 } 274 if (nhosts == 0) 275 hosts = (struct hitem *)malloc(sizeof (*p)); 276 else 277 hosts = (struct hitem *)realloc(hosts, (nhosts+1)*sizeof (*p)); 278 p = &hosts[nhosts++]; 279 p->addr = *in; 280 p->onoff = onoff; 281 return (1); 282 } 283 284 int 285 checkhost(struct inpcb *inp) 286 { 287 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(void) 299 { 300 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