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