Home | History | Annotate | Line # | Download | only in altqstat 
      1 /*	$NetBSD: qdisc_wfq.c,v 1.5 2006/10/28 11:43:02 peter Exp $	*/
      2 /*	$KAME: qdisc_wfq.c,v 1.5 2002/11/08 06:36:18 kjc Exp $	*/
      3 /*
      4  * Copyright (C) 1999-2000
      5  *	Sony Computer Science Laboratories, Inc.  All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY SONY CSL 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 SONY CSL 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 INTERRUPTION)
     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 
     29 #include <sys/param.h>
     30 #include <sys/ioctl.h>
     31 #include <sys/time.h>
     32 #include <sys/socket.h>
     33 #include <net/if.h>
     34 #include <netinet/in.h>
     35 #include <altq/altq.h>
     36 #include <altq/altq_wfq.h>
     37 
     38 #include <stdio.h>
     39 #include <stdlib.h>
     40 #include <unistd.h>
     41 #include <string.h>
     42 #include <signal.h>
     43 #include <errno.h>
     44 #include <err.h>
     45 #ifndef NO_CURSES
     46 #include <curses.h>
     47 #endif
     48 
     49 #include "altqstat.h"
     50 
     51 struct wfqinfo {
     52 	int qid;
     53 	queue_stats stats;
     54 	u_quad_t last_bytes;
     55 	double bps;
     56 };
     57 
     58 #define NTOP		10
     59 static int ntop = NTOP;
     60 
     61 void
     62 wfq_stat_loop(int fd, const char *ifname, int count, int interval)
     63 {
     64 	struct wfq_getstats wfq_stats;
     65 	struct timeval cur_time, last_time;
     66 	int i, j, k, nqueues;
     67 	double sec;
     68 	struct wfqinfo *qinfo, **top;
     69 	int cnt = count;
     70 	sigset_t		omask;
     71 
     72 	strlcpy(wfq_stats.iface.wfq_ifacename, ifname,
     73 		sizeof(wfq_stats.iface.wfq_ifacename));
     74 
     75 	/*
     76 	 * first, find out how many queues are available
     77 	 */
     78 	for (i = 0; i < MAX_QSIZE; i++) {
     79 		wfq_stats.qid = i;
     80 		if (ioctl(fd, WFQ_GET_STATS, &wfq_stats) < 0)
     81 			break;
     82 	}
     83 	nqueues = i;
     84 	printf("wfq on %s: %d queues are used\n", ifname, nqueues);
     85 
     86 	if ((qinfo = malloc(nqueues * sizeof(struct wfqinfo))) == NULL)
     87 		err(1, "malloc failed!");
     88 	if ((top = malloc(ntop * sizeof(struct wfqinfo *))) == NULL)
     89 		err(1, "malloc failed!");
     90 
     91 #ifndef NO_CURSES
     92 	sleep(2);  /* wait a bit before clearing the screen */
     93 
     94 	initscr();
     95 #endif
     96 
     97 	gettimeofday(&last_time, NULL);
     98 	last_time.tv_sec -= interval;
     99 
    100 	for (;;) {
    101 		for (j = 0; j < ntop; j++)
    102 			top[j] = NULL;
    103 
    104 		for (i = 0; i < nqueues; i++) {
    105 			wfq_stats.qid = i;
    106 			if (ioctl(fd, WFQ_GET_STATS, &wfq_stats) < 0)
    107 				err(1, "ioctl WFQ_GET_STATS");
    108 
    109 			qinfo[i].qid = i;
    110 			qinfo[i].stats = wfq_stats.stats;
    111 		}
    112 
    113 		gettimeofday(&cur_time, NULL);
    114 		sec = calc_interval(&cur_time, &last_time);
    115 
    116 		/*
    117 		 * calculate the throughput of each queue
    118 		 */
    119 		for (i = 0; i < nqueues; i++) {
    120 			qinfo[i].bps = calc_rate(qinfo[i].stats.xmit_cnt.bytes,
    121 						 qinfo[i].last_bytes, sec);
    122 			qinfo[i].last_bytes = qinfo[i].stats.xmit_cnt.bytes;
    123 
    124 			for (j = 0; j < ntop; j++) {
    125 				if (top[j] == NULL) {
    126 					top[j] = &qinfo[i];
    127 					break;
    128 				}
    129 				if (top[j]->bps < qinfo[i].bps ||
    130 				    (top[j]->bps == qinfo[i].bps &&
    131 				     top[j]->stats.xmit_cnt.packets <
    132 				     qinfo[i].stats.xmit_cnt.packets)) {
    133 					for (k = ntop-1; k > j; k--)
    134 						top[k] = top[k-1];
    135 					top[j] = &qinfo[i];
    136 					break;
    137 				}
    138 			}
    139 		}
    140 
    141 		/*
    142 		 * display top
    143 		 */
    144 		printf("[QID] WEIGHT QSIZE(KB) SENT(pkts)     (KB)       DROP(pkts)     (KB)     bps\n\r");
    145 
    146 		for (j = 0; j < ntop; j++) {
    147 			if (top[j] != NULL)
    148 				printf("[%4d] %4d %4d %10llu %14llu %10llu %14llu %9s\n\r",
    149 				       top[j]->qid,
    150 				       top[j]->stats.weight,
    151 				       top[j]->stats.bytes / 1024,
    152 				       (ull)top[j]->stats.xmit_cnt.packets,
    153 				       (ull)top[j]->stats.xmit_cnt.bytes /1024,
    154 				       (ull)top[j]->stats.drop_cnt.packets,
    155 				       (ull)top[j]->stats.drop_cnt.bytes /1024,
    156 				       rate2str(top[j]->bps));
    157 			else
    158 				printf("\n");
    159 		}
    160 #ifndef NO_CURSES
    161 		refresh();
    162 		mvcur(ntop+1, 0, 0, 0);
    163 #endif
    164 
    165 		last_time = cur_time;
    166 
    167 		if (count != 0 && --cnt == 0)
    168 			break;
    169 
    170 		/* wait for alarm signal */
    171 		if (sigprocmask(SIG_BLOCK, NULL, &omask) == 0)
    172 			sigsuspend(&omask);
    173 	}
    174 
    175 }
    176