1/*
2 * Copyright (c) 2003, Trent Nelson, <trent@arpa.com>.
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 * 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. The name of the author may not be used to endorse or promote products
14 *    derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTIFSTAT_ERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 *
28 * $FreeBSD$
29 */
30
31#include <sys/types.h>
32#include <sys/socket.h>
33#include <sys/sysctl.h>
34#include <net/if.h>
35#include <net/if_mib.h>
36
37#include <stdlib.h>
38#include <string.h>
39#include <err.h>
40#include <errno.h>
41
42#include "systat.h"
43#include "extern.h"
44#include "convtbl.h"
45
46				/* Column numbers */
47
48#define C1	0		/*  0-19 */
49#define C2	20		/* 20-39 */
50#define C3	40		/* 40-59 */
51#define C4	60		/* 60-80 */
52#define C5	80		/* Used for label positioning. */
53
54static const int col0 = 0;
55static const int col1 = C1;
56static const int col2 = C2;
57static const int col3 = C3;
58static const int col4 = C4;
59static const int col5 = C5;
60
61
62SLIST_HEAD(, if_stat)		curlist;
63SLIST_HEAD(, if_stat_disp)	displist;
64
65struct if_stat {
66	SLIST_ENTRY(if_stat)	 link;
67	char	if_name[IF_NAMESIZE];
68	struct	ifmibdata if_mib;
69	struct	timeval tv;
70	struct	timeval tv_lastchanged;
71	u_long	if_in_curtraffic;
72	u_long	if_out_curtraffic;
73	u_long	if_in_traffic_peak;
74	u_long	if_out_traffic_peak;
75	u_int	if_row;			/* Index into ifmib sysctl */
76	u_int	if_ypos;		/* 0 if not being displayed */
77	u_int	display;
78};
79
80extern	 u_int curscale;
81
82static	 void  right_align_string(struct if_stat *);
83static	 void  getifmibdata(const int, struct ifmibdata *);
84static	 void  sort_interface_list(void);
85static	 u_int getifnum(void);
86
87#define IFSTAT_ERR(n, s)	do {					\
88	putchar('');							\
89	closeifstat(wnd);						\
90	err((n), (s));							\
91} while (0)
92
93#define TOPLINE 3
94#define TOPLABEL \
95"      Interface           Traffic               Peak                Total"
96
97#define STARTING_ROW	(TOPLINE + 1)
98#define ROW_SPACING	(3)
99
100#define CLEAR_LINE(y, x)	do {					\
101	wmove(wnd, y, x);						\
102	wclrtoeol(wnd);							\
103} while (0)
104
105#define IN_col2		(ifp->if_in_curtraffic)
106#define OUT_col2	(ifp->if_out_curtraffic)
107#define IN_col3		(ifp->if_in_traffic_peak)
108#define OUT_col3	(ifp->if_out_traffic_peak)
109#define IN_col4		(ifp->if_mib.ifmd_data.ifi_ibytes)
110#define OUT_col4	(ifp->if_mib.ifmd_data.ifi_obytes)
111
112#define EMPTY_COLUMN 	"                    "
113#define CLEAR_COLUMN(y, x)	mvprintw((y), (x), "%20s", EMPTY_COLUMN);
114
115#define DOPUTRATE(c, r, d)	do {					\
116	CLEAR_COLUMN(r, c);						\
117	mvprintw(r, (c), "%10.3f %s%s  ",				\
118		 convert(d##_##c, curscale),				\
119		 get_string(d##_##c, curscale),				\
120		 "/s");							\
121} while (0)
122
123#define DOPUTTOTAL(c, r, d)	do {					\
124	CLEAR_COLUMN((r), (c));						\
125	mvprintw((r), (c), "%12.3f %s  ",				\
126		 convert(d##_##c, SC_AUTO),				\
127		 get_string(d##_##c, SC_AUTO));				\
128} while (0)
129
130#define PUTRATE(c, r)	do {						\
131	DOPUTRATE(c, (r), IN);						\
132	DOPUTRATE(c, (r)+1, OUT);					\
133} while (0)
134
135#define PUTTOTAL(c, r)	do {						\
136	DOPUTTOTAL(c, (r), IN);						\
137	DOPUTTOTAL(c, (r)+1, OUT);					\
138} while (0)
139
140#define PUTNAME(p) do {							\
141	mvprintw(p->if_ypos, 0, "%s", p->if_name);			\
142	mvprintw(p->if_ypos, col2-3, "%s", (const char *)"in");		\
143	mvprintw(p->if_ypos+1, col2-3, "%s", (const char *)"out");	\
144} while (0)
145
146
147WINDOW *
148openifstat(void)
149{
150	return (subwin(stdscr, LINES-3-1, 0, MAINWIN_ROW, 0));
151}
152
153void
154closeifstat(WINDOW *w)
155{
156	struct if_stat	*node = NULL;
157
158	while (!SLIST_EMPTY(&curlist)) {
159		node = SLIST_FIRST(&curlist);
160		SLIST_REMOVE_HEAD(&curlist, link);
161		free(node);
162	}
163
164	if (w != NULL) {
165		wclear(w);
166		wrefresh(w);
167		delwin(w);
168	}
169
170	return;
171}
172
173
174void
175labelifstat(void)
176{
177
178	wmove(wnd, TOPLINE, 0);
179	wclrtoeol(wnd);
180	mvprintw(TOPLINE, 0, "%s", TOPLABEL);
181
182	return;
183}
184
185void
186showifstat(void)
187{
188	struct	if_stat *ifp = NULL;
189	SLIST_FOREACH(ifp, &curlist, link) {
190		if (ifp->display == 0)
191			continue;
192		PUTNAME(ifp);
193		PUTRATE(col2, ifp->if_ypos);
194		PUTRATE(col3, ifp->if_ypos);
195		PUTTOTAL(col4, ifp->if_ypos);
196	}
197
198	return;
199}
200
201int
202initifstat(void)
203{
204	struct   if_stat *p = NULL;
205	u_int	 n = 0, i = 0;
206
207	n = getifnum();
208	if (n <= 0)
209		return -1;
210
211	SLIST_INIT(&curlist);
212
213	for (i = 0; i < n; i++) {
214		p = (struct if_stat *)calloc(1, sizeof(struct if_stat));
215		if (p == NULL)
216			IFSTAT_ERR(1, "out of memory");
217		SLIST_INSERT_HEAD(&curlist, p, link);
218		p->if_row = i+1;
219		getifmibdata(p->if_row, &p->if_mib);
220		right_align_string(p);
221
222		/*
223		 * Initially, we only display interfaces that have
224		 * received some traffic.
225		 */
226		if (p->if_mib.ifmd_data.ifi_ibytes != 0)
227			p->display = 1;
228	}
229
230	sort_interface_list();
231
232	return 1;
233}
234
235void
236fetchifstat(void)
237{
238	struct	if_stat *ifp = NULL;
239	struct	timeval tv, new_tv, old_tv;
240	double	elapsed = 0.0;
241	u_int	new_inb, new_outb, old_inb, old_outb = 0;
242	u_int	we_need_to_sort_interface_list = 0;
243
244	SLIST_FOREACH(ifp, &curlist, link) {
245		/*
246		 * Grab a copy of the old input/output values before we
247		 * call getifmibdata().
248		 */
249		old_inb = ifp->if_mib.ifmd_data.ifi_ibytes;
250		old_outb = ifp->if_mib.ifmd_data.ifi_obytes;
251		ifp->tv_lastchanged = ifp->if_mib.ifmd_data.ifi_lastchange;
252
253		if (gettimeofday(&new_tv, (struct timezone *)0) != 0)
254			IFSTAT_ERR(2, "error getting time of day");
255		(void)getifmibdata(ifp->if_row, &ifp->if_mib);
256
257
258		new_inb = ifp->if_mib.ifmd_data.ifi_ibytes;
259		new_outb = ifp->if_mib.ifmd_data.ifi_obytes;
260
261		/* Display interface if it's received some traffic. */
262		if (new_inb > 0 && old_inb == 0) {
263			ifp->display = 1;
264			we_need_to_sort_interface_list++;
265		}
266
267		/*
268		 * The rest is pretty trivial.  Calculate the new values
269		 * for our current traffic rates, and while we're there,
270		 * see if we have new peak rates.
271		 */
272		old_tv = ifp->tv;
273		timersub(&new_tv, &old_tv, &tv);
274		elapsed = tv.tv_sec + (tv.tv_usec * 1e-6);
275
276		ifp->if_in_curtraffic = new_inb - old_inb;
277		ifp->if_out_curtraffic = new_outb - old_outb;
278
279		/*
280		 * Rather than divide by the time specified on the comm-
281		 * and line, we divide by ``elapsed'' as this is likely
282		 * to be more accurate.
283		 */
284		ifp->if_in_curtraffic /= elapsed;
285		ifp->if_out_curtraffic /= elapsed;
286
287		if (ifp->if_in_curtraffic > ifp->if_in_traffic_peak)
288			ifp->if_in_traffic_peak = ifp->if_in_curtraffic;
289
290		if (ifp->if_out_curtraffic > ifp->if_out_traffic_peak)
291			ifp->if_out_traffic_peak = ifp->if_out_curtraffic;
292
293		ifp->tv.tv_sec = new_tv.tv_sec;
294		ifp->tv.tv_usec = new_tv.tv_usec;
295
296	}
297
298	if (we_need_to_sort_interface_list)
299		sort_interface_list();
300
301	return;
302}
303
304/*
305 * We want to right justify our interface names against the first column
306 * (first sixteen or so characters), so we need to do some alignment.
307 */
308static void
309right_align_string(struct if_stat *ifp)
310{
311	int	 str_len = 0, pad_len = 0;
312	char	*newstr = NULL, *ptr = NULL;
313
314	if (ifp == NULL || ifp->if_mib.ifmd_name == NULL)
315		return;
316	else {
317		/* string length + '\0' */
318		str_len = strlen(ifp->if_mib.ifmd_name)+1;
319		pad_len = IF_NAMESIZE-(str_len);
320
321		newstr = ifp->if_name;
322		ptr = newstr + pad_len;
323		(void)memset((void *)newstr, (int)' ', IF_NAMESIZE);
324		(void)strncpy(ptr, (const char *)&ifp->if_mib.ifmd_name,
325			      str_len);
326	}
327
328	return;
329}
330
331/*
332 * This function iterates through our list of interfaces, identifying
333 * those that are to be displayed (ifp->display = 1).  For each interf-
334 * rface that we're displaying, we generate an appropriate position for
335 * it on the screen (ifp->if_ypos).
336 *
337 * This function is called any time a change is made to an interface's
338 * ``display'' state.
339 */
340void
341sort_interface_list(void)
342{
343	struct	if_stat	*ifp = NULL;
344	u_int	y = 0;
345
346	y = STARTING_ROW;
347	SLIST_FOREACH(ifp, &curlist, link) {
348		if (ifp->display) {
349			ifp->if_ypos = y;
350			y += ROW_SPACING;
351		}
352	}
353}
354
355static
356unsigned int
357getifnum(void)
358{
359	u_int	data    = 0;
360	size_t	datalen = 0;
361	static	int name[] = { CTL_NET,
362			       PF_LINK,
363			       NETLINK_GENERIC,
364			       IFMIB_SYSTEM,
365			       IFMIB_IFCOUNT };
366
367	datalen = sizeof(data);
368	if (sysctl(name, 5, (void *)&data, (size_t *)&datalen, (void *)NULL,
369	    (size_t)0) != 0)
370		IFSTAT_ERR(1, "sysctl error");
371	return data;
372}
373
374static void
375getifmibdata(int row, struct ifmibdata *data)
376{
377	size_t	datalen = 0;
378	static	int name[] = { CTL_NET,
379			       PF_LINK,
380			       NETLINK_GENERIC,
381			       IFMIB_IFDATA,
382			       0,
383			       IFDATA_GENERAL };
384	datalen = sizeof(*data);
385	name[4] = row;
386
387	if ((sysctl(name, 6, (void *)data, (size_t *)&datalen, (void *)NULL,
388	    (size_t)0) != 0) && (errno != ENOENT))
389		IFSTAT_ERR(2, "sysctl error getting interface data");
390}
391
392int
393cmdifstat(const char *cmd, const char *args)
394{
395	int	retval = 0;
396
397	retval = ifcmd(cmd, args);
398	/* ifcmd() returns 1 on success */
399	if (retval == 1) {
400		showifstat();
401		refresh();
402	}
403
404	return retval;
405}
406