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
43static 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
52static const char *
53getfmt(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
62static int signalled;
63
64static void
65catchalarm(int signo __unused)
66{
67	signalled = 1;
68}
69
70int
71main(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