1 /*- 2 * Copyright (c) 2009 Sam Leffler, Errno Consulting 3 * 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 * without modification. 11 * 2. Redistributions in binary form must reproduce at minimum a disclaimer 12 * similar to the "NO WARRANTY" disclaimer below ("Disclaimer") and any 13 * redistribution must be conditioned upon including a substantially 14 * similar Disclaimer requirement for further binary redistribution. 15 * 16 * NO WARRANTY 17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 * LIMITED TO, THE IMPLIED WARRANTIES OF NONINFRINGEMENT, MERCHANTIBILITY 20 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL 21 * THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR SPECIAL, EXEMPLARY, 22 * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 23 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER 25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 26 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 27 * THE POSSIBILITY OF SUCH DAMAGES. 28 * 29 * $FreeBSD$ 30 */ 31 32 #include <sys/param.h> 33 34 #include <err.h> 35 #include <signal.h> 36 #include <stdio.h> 37 #include <stdlib.h> 38 #include <strings.h> 39 #include <unistd.h> 40 41 #include "npestats.h" 42 43 static struct { 44 const char *tag; 45 const char *fmt; 46 } tags[] = { 47 { "default", 48 "align,fcs,macrx,overrun,learn,large,stp,badsrc,underflow,collision1,collisionM,deferred,late,excessive,mactx,carrier,toobig" 49 }, 50 }; 51 52 static const char * 53 getfmt(const char *tag) 54 { 55 int i; 56 for (i = 0; i < nitems(tags); i++) 57 if (strcasecmp(tags[i].tag, tag) == 0) 58 return tags[i].fmt; 59 return tag; 60 } 61 62 static int signalled; 63 64 static void 65 catchalarm(int signo __unused) 66 { 67 signalled = 1; 68 } 69 70 int 71 main(int argc, char *argv[]) 72 { 73 struct npestatfoo *wf; 74 const char *ifname; 75 int c, banner = 1; 76 77 ifname = getenv("NPE"); 78 if (ifname == NULL) 79 ifname = "npe0"; 80 wf = npestats_new(ifname, getfmt("default")); 81 while ((c = getopt(argc, argv, "bi:lo:z")) != -1) { 82 switch (c) { 83 case 'b': 84 banner = 0; 85 break; 86 case 'i': 87 wf->setifname(wf, optarg); 88 break; 89 case 'l': 90 wf->print_fields(wf, stdout); 91 return 0; 92 case 'o': 93 wf->setfmt(wf, getfmt(optarg)); 94 break; 95 default: 96 errx(-1, "usage: %s [-a] [-i ifname] [-l] [-o fmt] [interval]\n", argv[0]); 97 /*NOTREACHED*/ 98 } 99 } 100 argc -= optind; 101 argv += optind; 102 103 if (argc > 0) { 104 u_long interval = strtoul(argv[0], NULL, 0); 105 int line, omask; 106 107 if (interval < 1) 108 interval = 1; 109 signal(SIGALRM, catchalarm); 110 signalled = 0; 111 alarm(interval); 112 banner: 113 if (banner) 114 wf->print_header(wf, stdout); 115 line = 0; 116 loop: 117 if (line != 0) { 118 wf->collect_cur(wf); 119 wf->print_current(wf, stdout); 120 wf->update_tot(wf); 121 } else { 122 wf->collect_tot(wf); 123 wf->print_total(wf, stdout); 124 } 125 fflush(stdout); 126 omask = sigblock(sigmask(SIGALRM)); 127 if (!signalled) 128 sigpause(0); 129 sigsetmask(omask); 130 signalled = 0; 131 alarm(interval); 132 line++; 133 if (line == 21) /* XXX tty line count */ 134 goto banner; 135 else 136 goto loop; 137 /*NOTREACHED*/ 138 } else { 139 wf->collect_tot(wf); 140 wf->print_verbose(wf, stdout); 141 } 142 return 0; 143 } 144