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
sig_handler(int sig)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
alrm_handler(int sig)79 alrm_handler(int sig)
80 {
81 /* nothing */
82 }
83
84 static void
usage(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
main(int argc,char ** argv)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
calc_interval(struct timeval * cur_time,struct timeval * last_time)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
calc_rate(u_int64_t new_bytes,u_int64_t last_bytes,double interval)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
calc_pps(u_int64_t new_pkts,u_int64_t last_pkts,double interval)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 *
rate2str(double rate)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