Home | History | Annotate | Line # | Download | only in altqstat
      1 /*	$NetBSD: altqstat.c,v 1.9 2021/07/24 21:31:39 andvar Exp $	*/
      2 /*	$KAME: altqstat.c,v 1.8 2002/10/27 03:19:35 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/time.h>
     31 #include <sys/fcntl.h>
     32 
     33 #include <stdio.h>
     34 #include <stdlib.h>
     35 #include <unistd.h>
     36 #include <string.h>
     37 #include <signal.h>
     38 #include <errno.h>
     39 #include <err.h>
     40 #ifndef NO_CURSES
     41 #include <curses.h>
     42 #endif
     43 
     44 #include "quip_client.h"
     45 #include "altqstat.h"
     46 
     47 #define DEV_PATH	"/dev/altq"
     48 
     49 int qdiscfd = -1;
     50 int show_config = 0;
     51 double ival = 5.0;
     52 int no_server = 0;
     53 char *interface = NULL;
     54 char *qdisc_name = NULL;
     55 
     56 stat_loop_t *stat_loop;
     57 
     58 __dead static void sig_handler(int);
     59 static void alrm_handler(int);
     60 __dead static void usage(void);
     61 
     62 static void
     63 sig_handler(int sig)
     64 {
     65 	char buf[8192];
     66 
     67 	snprintf(buf, sizeof buf, "Exiting on signal %d\n", sig);
     68 	write(STDERR_FILENO, buf, strlen(buf));
     69 
     70 #ifndef NO_CURSES
     71 	/* XXX signal race */
     72 	if (qdisc_name != NULL && strcmp(qdisc_name, "wfq") == 0)
     73 		endwin();	/* wfqstat uses curses */
     74 #endif
     75 	_exit(0);
     76 }
     77 
     78 static void
     79 alrm_handler(int sig)
     80 {
     81 	/* nothing */
     82 }
     83 
     84 static void
     85 usage(void)
     86 {
     87 	fprintf(stderr, "usage: altqstat [-enrs] [-c count] [-w wait] [-i interface|-I input_interface]\n");
     88 	exit(1);
     89 }
     90 
     91 int
     92 main (int argc, char **argv)
     93 {
     94 	int ch, raw_mode = 0;
     95 	int qtype, interval;
     96 	int count = 0;
     97 	struct itimerval it;
     98 	char device[64], qname[64], input[32];
     99 
    100 	while ((ch = getopt(argc, argv, "I:c:ei:nrsw:")) != -1) {
    101 		switch (ch) {
    102 		case 'I':
    103 			snprintf(input, sizeof(input), "_%s", optarg);
    104 			interface = input;
    105 			break;
    106 		case 'c':
    107 			count = atoi(optarg);
    108 			if (count < 1)
    109 				errx(1, "Please supply a count value bigger than 0.");
    110 			break;
    111 		case 'e':
    112 			quip_echo = 1;
    113 			break;
    114 		case 'i':
    115 			interface = optarg;
    116 			break;
    117 		case 'n':
    118 			no_server = 1;
    119 			break;
    120 		case 'r':
    121 			raw_mode = 1;
    122 			quip_echo = 1;
    123 			break;
    124 		case 's':
    125 			show_config = 1;
    126 			break;
    127 		case 'w':
    128 			ival = strtod(optarg, NULL);
    129 			break;
    130 		default:
    131 			usage();
    132 			break;
    133 		}
    134 	}
    135 
    136 	signal(SIGINT, sig_handler);
    137 	signal(SIGTERM, sig_handler);
    138 	signal(SIGPIPE, sig_handler);
    139 	signal(SIGALRM, alrm_handler);
    140 
    141 	if (no_server == 0) {
    142 		if (quip_openserver() < 0 && interface == NULL)
    143 			errx(1, "you have to specify interface!");
    144 	}
    145 
    146 	if (raw_mode == 1) {
    147 		quip_rawmode();
    148 		quip_closeserver();
    149 		exit(0);
    150 	}
    151 
    152 	if (show_config) {
    153 		if (no_server)
    154 			errx(1, "no server (-n) can't be set for show config (-s)!");
    155 		quip_printconfig();
    156 		quip_closeserver();
    157 		exit(0);
    158 	}
    159 
    160 	interface = quip_selectinterface(interface);
    161 	if (interface == NULL)
    162 		errx(1, "no interface found!");
    163 
    164 	qtype = ifname2qdisc(interface, qname);
    165 	if (qtype == 0)
    166 		errx(1, "altq is not attached on %s!", interface);
    167 
    168 	qdisc_name = qname;
    169 
    170 	stat_loop = qdisc2stat_loop(qdisc_name);
    171 	if (stat_loop == NULL)
    172 		errx(1, "qdisc %s is not supported!", qdisc_name);
    173 
    174 	printf("%s: %s on interface %s\n",
    175 	       argv[0], qdisc_name, interface);
    176 
    177 	snprintf(device, sizeof(device), "%s/%s", DEV_PATH, qdisc_name);
    178 	if ((qdiscfd = open(device, O_RDONLY)) < 0)
    179 		err(1, "can't open %s", device);
    180 
    181 	interval = (int)(ival * 1000.0);
    182 	it.it_interval.tv_sec = interval / 1000;
    183 	it.it_interval.tv_usec = interval % 1000 * 1000;
    184 	it.it_value = it.it_interval;
    185 	setitimer(ITIMER_REAL, &it, NULL);
    186 
    187 	(*stat_loop)(qdiscfd, interface, count, (int)ival);
    188 
    189 	exit(0);
    190 }
    191 
    192 /* calculate interval in sec */
    193 double
    194 calc_interval(struct timeval *cur_time, struct timeval *last_time)
    195 {
    196 	double sec;
    197 
    198 	sec = (double)(cur_time->tv_sec - last_time->tv_sec) +
    199 	    (double)(cur_time->tv_usec - last_time->tv_usec) / 1000000;
    200 	return (sec);
    201 }
    202 
    203 
    204 /* calculate rate in bps */
    205 double
    206 calc_rate(u_int64_t new_bytes, u_int64_t last_bytes, double interval)
    207 {
    208 	double rate;
    209 
    210 	rate = (double)(new_bytes - last_bytes) * 8 / interval;
    211 	return (rate);
    212 }
    213 
    214 /* calculate packets in second */
    215 double
    216 calc_pps(u_int64_t new_pkts, u_int64_t last_pkts, double interval)
    217 {
    218 	double pps;
    219 
    220 	pps = (double)(new_pkts - last_pkts) / interval;
    221 	return (pps);
    222 }
    223 
    224 #define	R2S_BUFS	8
    225 #define	RATESTR_MAX	16
    226 char *
    227 rate2str(double rate)
    228 {
    229 	char *buf;
    230 	static char r2sbuf[R2S_BUFS][RATESTR_MAX];  /* ring buffer */
    231 	static int idx = 0;
    232 
    233 	buf = r2sbuf[idx++];
    234 	if (idx == R2S_BUFS)
    235 		idx = 0;
    236 
    237 	if (rate == 0.0)
    238 		snprintf(buf, RATESTR_MAX, "0");
    239 	else if (rate >= 1000000.0)
    240 		snprintf(buf, RATESTR_MAX, "%.2fM", rate / 1000000.0);
    241 	else
    242 		snprintf(buf, RATESTR_MAX, "%.2fk", rate / 1000.0);
    243 	return (buf);
    244 }
    245