Home | History | Annotate | Line # | Download | only in altqstat
qdisc_wfq.c revision 1.2 
      1 /*	$KAME: qdisc_wfq.c,v 1.3 2001/08/15 12:51:59 kjc Exp $	*/
      2 /*
      3  * Copyright (C) 1999-2000
      4  *	Sony Computer Science Laboratories, Inc.  All rights reserved.
      5  *
      6  * Redistribution and use in source and binary forms, with or without
      7  * modification, are permitted provided that the following conditions
      8  * are met:
      9  * 1. Redistributions of source code must retain the above copyright
     10  *    notice, this list of conditions and the following disclaimer.
     11  * 2. Redistributions in binary form must reproduce the above copyright
     12  *    notice, this list of conditions and the following disclaimer in the
     13  *    documentation and/or other materials provided with the distribution.
     14  *
     15  * THIS SOFTWARE IS PROVIDED BY SONY CSL AND CONTRIBUTORS ``AS IS'' AND
     16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     18  * ARE DISCLAIMED.  IN NO EVENT SHALL SONY CSL OR CONTRIBUTORS BE LIABLE
     19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     25  * SUCH DAMAGE.
     26  */
     27 
     28 #include <sys/param.h>
     29 #include <sys/ioctl.h>
     30 #include <sys/time.h>
     31 #include <sys/socket.h>
     32 #include <net/if.h>
     33 #include <netinet/in.h>
     34 #include <altq/altq.h>
     35 #include <altq/altq_wfq.h>
     36 
     37 #include <stdio.h>
     38 #include <stdlib.h>
     39 #include <unistd.h>
     40 #include <string.h>
     41 #include <math.h>
     42 #include <errno.h>
     43 #include <err.h>
     44 #ifndef NO_CURSES
     45 #include <curses.h>
     46 #endif
     47 
     48 #include "altqstat.h"
     49 
     50 struct wfqinfo {
     51 	int qid;
     52 	queue_stats stats;
     53 	u_quad_t last_bytes;
     54 	double bps;
     55 };
     56 
     57 #define NTOP		10
     58 static int ntop = NTOP;
     59 
     60 void
     61 wfq_stat_loop(int fd, const char *ifname, int count, int interval)
     62 {
     63 	struct wfq_getstats wfq_stats;
     64 	struct timeval cur_time, last_time;
     65 	int i, j, k, nqueues;
     66 	double sec;
     67 	struct wfqinfo *qinfo, **top;
     68 	int cnt = count;
     69 
     70 	strlcpy(wfq_stats.iface.wfq_ifacename, ifname,
     71 		sizeof(wfq_stats.iface.wfq_ifacename));
     72 
     73 	/*
     74 	 * first, find out how many queues are available
     75 	 */
     76 	for (i = 0; i < MAX_QSIZE; i++) {
     77 		wfq_stats.qid = i;
     78 		if (ioctl(fd, WFQ_GET_STATS, &wfq_stats) < 0)
     79 			break;
     80 	}
     81 	nqueues = i;
     82 	printf("wfq on %s: %d queues are used\n", ifname, nqueues);
     83 
     84 	if ((qinfo = malloc(nqueues * sizeof(struct wfqinfo))) == NULL)
     85 		err(1, "malloc failed!");
     86 	if ((top = malloc(ntop * sizeof(struct wfqinfo *))) == NULL)
     87 		err(1, "malloc failed!");
     88 
     89 #ifndef NO_CURSES
     90 	sleep(2);  /* wait a bit before clearing the screen */
     91 
     92 	initscr();
     93 #endif
     94 
     95 	gettimeofday(&last_time, NULL);
     96 	last_time.tv_sec -= interval;
     97 
     98 	while (count == 0 || cnt-- > 0) {
     99 
    100 		for (j = 0; j < ntop; j++)
    101 			top[j] = NULL;
    102 
    103 		for (i = 0; i < nqueues; i++) {
    104 			wfq_stats.qid = i;
    105 			if (ioctl(fd, WFQ_GET_STATS, &wfq_stats) < 0)
    106 				err(1, "ioctl WFQ_GET_STATS");
    107 
    108 			qinfo[i].qid = i;
    109 			qinfo[i].stats = wfq_stats.stats;
    110 		}
    111 
    112 		gettimeofday(&cur_time, NULL);
    113 		sec = calc_interval(&cur_time, &last_time);
    114 
    115 		/*
    116 		 * calculate the throughput of each queue
    117 		 */
    118 		for (i = 0; i < nqueues; i++) {
    119 			qinfo[i].bps = calc_rate(qinfo[i].stats.xmit_cnt.bytes,
    120 						 qinfo[i].last_bytes, sec);
    121 			qinfo[i].last_bytes = qinfo[i].stats.xmit_cnt.bytes;
    122 
    123 			for (j = 0; j < ntop; j++) {
    124 				if (top[j] == NULL) {
    125 					top[j] = &qinfo[i];
    126 					break;
    127 				}
    128 				if (top[j]->bps < qinfo[i].bps ||
    129 				    (top[j]->bps == qinfo[i].bps &&
    130 				     top[j]->stats.xmit_cnt.packets <
    131 				     qinfo[i].stats.xmit_cnt.packets)) {
    132 					for (k = ntop-1; k > j; k--)
    133 						top[k] = top[k-1];
    134 					top[j] = &qinfo[i];
    135 					break;
    136 				}
    137 			}
    138 		}
    139 
    140 		/*
    141 		 * display top
    142 		 */
    143 		printf("[QID] WEIGHT QSIZE(KB) SENT(pkts)     (KB)       DROP(pkts)     (KB)     bps\n\r");
    144 
    145 		for (j = 0; j < ntop; j++) {
    146 			if (top[j] != NULL)
    147 				printf("[%4d] %4d %4d %10llu %14llu %10llu %14llu %9s\n\r",
    148 				       top[j]->qid,
    149 				       top[j]->stats.weight,
    150 				       top[j]->stats.bytes / 1024,
    151 				       (ull)top[j]->stats.xmit_cnt.packets,
    152 				       (ull)top[j]->stats.xmit_cnt.bytes /1024,
    153 				       (ull)top[j]->stats.drop_cnt.packets,
    154 				       (ull)top[j]->stats.drop_cnt.bytes /1024,
    155 				       rate2str(top[j]->bps));
    156 			else
    157 				printf("\n");
    158 		}
    159 #ifndef NO_CURSES
    160 		refresh();
    161 		mvcur(ntop+1, 0, 0, 0);
    162 #endif
    163 
    164 		last_time = cur_time;
    165 		sleep(interval);
    166 	}
    167 
    168 }
    169