]> git.proxmox.com Git - mirror_iproute2.git/blame - misc/ifstat.c
Merge branch 'master' of git://git.kernel.org/pub/scm/linux/kernel/git/shemminger...
[mirror_iproute2.git] / misc / ifstat.c
CommitLineData
aba5acdf
SH
1/*
2 * ifstat.c handy utility to read net interface statistics
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version
7 * 2 of the License, or (at your option) any later version.
8 *
9 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
10 */
11
12#include <stdio.h>
13#include <stdlib.h>
14#include <unistd.h>
15#include <fcntl.h>
16#include <string.h>
17#include <errno.h>
18#include <time.h>
19#include <sys/time.h>
20#include <fnmatch.h>
21#include <sys/file.h>
22#include <sys/socket.h>
23#include <sys/un.h>
24#include <sys/poll.h>
25#include <sys/wait.h>
26#include <sys/stat.h>
27#include <signal.h>
28#include <math.h>
9f3ea250 29#include <getopt.h>
aba5acdf 30
daf7bd5c
SH
31#include <linux/if.h>
32#include <linux/if_link.h>
aba5acdf 33
3d8048dc
NF
34#include "libnetlink.h"
35#include "json_writer.h"
36#include "SNAPSHOT.h"
5a52102b 37#include "utils.h"
aba5acdf 38
acd1e437
SH
39int dump_zeros;
40int reset_history;
41int ignore_history;
42int no_output;
43int json_output;
44int no_update;
45int scan_interval;
46int time_constant;
47int show_errors;
fcc16c22 48int pretty;
aba5acdf
SH
49double W;
50char **patterns;
51int npatterns;
5a52102b
NF
52bool is_extended;
53int filter_type;
54int sub_type;
aba5acdf
SH
55
56char info_source[128];
57int source_mismatch;
58
a571587d 59#define MAXS (sizeof(struct rtnl_link_stats)/sizeof(__u32))
5a52102b 60#define NO_SUB_TYPE 0xffff
aba5acdf 61
acd1e437 62struct ifstat_ent {
aba5acdf
SH
63 struct ifstat_ent *next;
64 char *name;
65 int ifindex;
5a52102b 66 __u64 val[MAXS];
aba5acdf 67 double rate[MAXS];
a571587d 68 __u32 ival[MAXS];
aba5acdf
SH
69};
70
ec3e625c
SH
71static const char *stats[MAXS] = {
72 "rx_packets",
73 "tx_packets",
74 "rx_bytes",
75 "tx_bytes",
76 "rx_errors",
77 "tx_errors",
78 "rx_dropped",
79 "tx_dropped",
80 "multicast",
81 "collisions",
82 "rx_length_errors",
83 "rx_over_errors",
84 "rx_crc_errors",
85 "rx_frame_errors",
86 "rx_fifo_errors",
87 "rx_missed_errors",
88 "tx_aborted_errors",
89 "tx_carrier_errors",
90 "tx_fifo_errors",
91 "tx_heartbeat_errors",
92 "tx_window_errors",
93 "rx_compressed",
94 "tx_compressed"
95};
96
aba5acdf
SH
97struct ifstat_ent *kern_db;
98struct ifstat_ent *hist_db;
99
50772dc5 100static int match(const char *id)
aba5acdf
SH
101{
102 int i;
103
104 if (npatterns == 0)
105 return 1;
106
acd1e437 107 for (i = 0; i < npatterns; i++) {
aba5acdf
SH
108 if (!fnmatch(patterns[i], id, 0))
109 return 1;
110 }
111 return 0;
112}
113
5a52102b
NF
114static int get_nlmsg_extended(const struct sockaddr_nl *who,
115 struct nlmsghdr *m, void *arg)
116{
117 struct if_stats_msg *ifsm = NLMSG_DATA(m);
118 struct rtattr *tb[IFLA_STATS_MAX+1];
119 int len = m->nlmsg_len;
120 struct ifstat_ent *n;
121
122 if (m->nlmsg_type != RTM_NEWSTATS)
123 return 0;
124
125 len -= NLMSG_LENGTH(sizeof(*ifsm));
126 if (len < 0)
127 return -1;
128
129 parse_rtattr(tb, IFLA_STATS_MAX, IFLA_STATS_RTA(ifsm), len);
130 if (tb[filter_type] == NULL)
131 return 0;
132
133 n = malloc(sizeof(*n));
134 if (!n)
135 abort();
136
137 n->ifindex = ifsm->ifindex;
138 n->name = strdup(ll_index_to_name(ifsm->ifindex));
139
140 if (sub_type == NO_SUB_TYPE) {
141 memcpy(&n->val, RTA_DATA(tb[filter_type]), sizeof(n->val));
142 } else {
143 struct rtattr *attr;
144
145 attr = parse_rtattr_one_nested(sub_type, tb[filter_type]);
35f6adef
PS
146 if (attr == NULL) {
147 free(n);
5a52102b 148 return 0;
35f6adef 149 }
5a52102b
NF
150 memcpy(&n->val, RTA_DATA(attr), sizeof(n->val));
151 }
152 memset(&n->rate, 0, sizeof(n->rate));
153 n->next = kern_db;
154 kern_db = n;
155 return 0;
156}
157
ae665a52 158static int get_nlmsg(const struct sockaddr_nl *who,
50772dc5 159 struct nlmsghdr *m, void *arg)
aba5acdf
SH
160{
161 struct ifinfomsg *ifi = NLMSG_DATA(m);
acd1e437 162 struct rtattr *tb[IFLA_MAX+1];
aba5acdf
SH
163 int len = m->nlmsg_len;
164 struct ifstat_ent *n;
165 int i;
166
167 if (m->nlmsg_type != RTM_NEWLINK)
168 return 0;
169
170 len -= NLMSG_LENGTH(sizeof(*ifi));
171 if (len < 0)
172 return -1;
173
174 if (!(ifi->ifi_flags&IFF_UP))
175 return 0;
176
aba5acdf
SH
177 parse_rtattr(tb, IFLA_MAX, IFLA_RTA(ifi), len);
178 if (tb[IFLA_IFNAME] == NULL || tb[IFLA_STATS] == NULL)
179 return 0;
180
181 n = malloc(sizeof(*n));
182 if (!n)
183 abort();
184 n->ifindex = ifi->ifi_index;
185 n->name = strdup(RTA_DATA(tb[IFLA_IFNAME]));
186 memcpy(&n->ival, RTA_DATA(tb[IFLA_STATS]), sizeof(n->ival));
187 memset(&n->rate, 0, sizeof(n->rate));
acd1e437 188 for (i = 0; i < MAXS; i++)
aba5acdf
SH
189 n->val[i] = n->ival[i];
190 n->next = kern_db;
191 kern_db = n;
192 return 0;
193}
194
d1f28cf1 195static void load_info(void)
aba5acdf
SH
196{
197 struct ifstat_ent *db, *n;
198 struct rtnl_handle rth;
5a52102b 199 __u32 filter_mask;
aba5acdf
SH
200
201 if (rtnl_open(&rth, 0) < 0)
202 exit(1);
203
5a52102b
NF
204 if (is_extended) {
205 ll_init_map(&rth);
206 filter_mask = IFLA_STATS_FILTER_BIT(filter_type);
207 if (rtnl_wilddump_stats_req_filter(&rth, AF_UNSPEC, RTM_GETSTATS,
208 filter_mask) < 0) {
209 perror("Cannot send dump request");
210 exit(1);
211 }
aba5acdf 212
5a52102b
NF
213 if (rtnl_dump_filter(&rth, get_nlmsg_extended, NULL) < 0) {
214 fprintf(stderr, "Dump terminated\n");
215 exit(1);
216 }
217 } else {
218 if (rtnl_wilddump_request(&rth, AF_INET, RTM_GETLINK) < 0) {
219 perror("Cannot send dump request");
220 exit(1);
221 }
222
223 if (rtnl_dump_filter(&rth, get_nlmsg, NULL) < 0) {
224 fprintf(stderr, "Dump terminated\n");
225 exit(1);
226 }
aba5acdf
SH
227 }
228
229 rtnl_close(&rth);
230
231 db = kern_db;
232 kern_db = NULL;
233
234 while (db) {
235 n = db;
236 db = db->next;
237 n->next = kern_db;
238 kern_db = n;
239 }
240}
241
d1f28cf1 242static void load_raw_table(FILE *fp)
aba5acdf
SH
243{
244 char buf[4096];
245 struct ifstat_ent *db = NULL;
246 struct ifstat_ent *n;
247
248 while (fgets(buf, sizeof(buf), fp) != NULL) {
249 char *p;
250 char *next;
251 int i;
252
253 if (buf[0] == '#') {
254 buf[strlen(buf)-1] = 0;
255 if (info_source[0] && strcmp(info_source, buf+1))
256 source_mismatch = 1;
257 strncpy(info_source, buf+1, sizeof(info_source)-1);
258 continue;
259 }
260 if ((n = malloc(sizeof(*n))) == NULL)
261 abort();
262
263 if (!(p = strchr(buf, ' ')))
264 abort();
265 *p++ = 0;
266
267 if (sscanf(buf, "%d", &n->ifindex) != 1)
268 abort();
269 if (!(next = strchr(p, ' ')))
270 abort();
271 *next++ = 0;
272
273 n->name = strdup(p);
274 p = next;
275
acd1e437
SH
276 for (i = 0; i < MAXS; i++) {
277 unsigned int rate;
278
aba5acdf
SH
279 if (!(next = strchr(p, ' ')))
280 abort();
281 *next++ = 0;
282 if (sscanf(p, "%llu", n->val+i) != 1)
283 abort();
a571587d 284 n->ival[i] = (__u32)n->val[i];
aba5acdf
SH
285 p = next;
286 if (!(next = strchr(p, ' ')))
287 abort();
288 *next++ = 0;
289 if (sscanf(p, "%u", &rate) != 1)
290 abort();
291 n->rate[i] = rate;
292 p = next;
293 }
294 n->next = db;
295 db = n;
296 }
297
298 while (db) {
299 n = db;
300 db = db->next;
301 n->next = kern_db;
302 kern_db = n;
303 }
304}
305
d1f28cf1 306static void dump_raw_db(FILE *fp, int to_hist)
aba5acdf 307{
fcc16c22 308 json_writer_t *jw = json_output ? jsonw_new(fp) : NULL;
aba5acdf 309 struct ifstat_ent *n, *h;
ec3e625c 310
aba5acdf 311 h = hist_db;
fcc16c22 312 if (jw) {
d721a145 313 jsonw_start_object(jw);
fcc16c22
SH
314 jsonw_pretty(jw, pretty);
315 jsonw_name(jw, info_source);
316 jsonw_start_object(jw);
317 } else
ec3e625c 318 fprintf(fp, "#%s\n", info_source);
aba5acdf 319
acd1e437 320 for (n = kern_db; n; n = n->next) {
aba5acdf
SH
321 int i;
322 unsigned long long *vals = n->val;
323 double *rates = n->rate;
acd1e437 324
aba5acdf
SH
325 if (!match(n->name)) {
326 struct ifstat_ent *h1;
acd1e437 327
aba5acdf
SH
328 if (!to_hist)
329 continue;
330 for (h1 = h; h1; h1 = h1->next) {
331 if (h1->ifindex == n->ifindex) {
332 vals = h1->val;
333 rates = h1->rate;
334 h = h1->next;
335 break;
336 }
337 }
338 }
ec3e625c 339
fcc16c22
SH
340 if (jw) {
341 jsonw_name(jw, n->name);
342 jsonw_start_object(jw);
1473bda9 343
acd1e437 344 for (i = 0; i < MAXS && stats[i]; i++)
fcc16c22
SH
345 jsonw_uint_field(jw, stats[i], vals[i]);
346 jsonw_end_object(jw);
ec3e625c
SH
347 } else {
348 fprintf(fp, "%d %s ", n->ifindex, n->name);
acd1e437 349 for (i = 0; i < MAXS; i++)
3d0b7439 350 fprintf(fp, "%llu %u ", vals[i],
acd1e437 351 (unsigned int)rates[i]);
ec3e625c
SH
352 fprintf(fp, "\n");
353 }
aba5acdf 354 }
fcc16c22 355 if (jw) {
d721a145
AK
356 jsonw_end_object(jw);
357
fcc16c22
SH
358 jsonw_end_object(jw);
359 jsonw_destroy(&jw);
360 }
aba5acdf
SH
361}
362
9f3ea250
SH
363/* use communication definitions of meg/kilo etc */
364static const unsigned long long giga = 1000000000ull;
bb6a21a4
SH
365static const unsigned long long mega = 1000000;
366static const unsigned long long kilo = 1000;
aba5acdf 367
ec3e625c
SH
368static void format_rate(FILE *fp, const unsigned long long *vals,
369 const double *rates, int i)
aba5acdf
SH
370{
371 char temp[64];
ec3e625c 372
9f3ea250
SH
373 if (vals[i] > giga)
374 fprintf(fp, "%7lluM ", vals[i]/mega);
375 else if (vals[i] > mega)
376 fprintf(fp, "%7lluK ", vals[i]/kilo);
aba5acdf
SH
377 else
378 fprintf(fp, "%8llu ", vals[i]);
379
9f3ea250 380 if (rates[i] > mega) {
acd1e437 381 sprintf(temp, "%uM", (unsigned int)(rates[i]/mega));
aba5acdf 382 fprintf(fp, "%-6s ", temp);
9f3ea250 383 } else if (rates[i] > kilo) {
acd1e437 384 sprintf(temp, "%uK", (unsigned int)(rates[i]/kilo));
aba5acdf
SH
385 fprintf(fp, "%-6s ", temp);
386 } else
acd1e437 387 fprintf(fp, "%-6u ", (unsigned int)rates[i]);
aba5acdf
SH
388}
389
ec3e625c 390static void format_pair(FILE *fp, const unsigned long long *vals, int i, int k)
aba5acdf
SH
391{
392 char temp[64];
acd1e437 393
9f3ea250
SH
394 if (vals[i] > giga)
395 fprintf(fp, "%7lluM ", vals[i]/mega);
396 else if (vals[i] > mega)
397 fprintf(fp, "%7lluK ", vals[i]/kilo);
aba5acdf
SH
398 else
399 fprintf(fp, "%8llu ", vals[i]);
400
9f3ea250 401 if (vals[k] > giga) {
acd1e437 402 sprintf(temp, "%uM", (unsigned int)(vals[k]/mega));
aba5acdf 403 fprintf(fp, "%-6s ", temp);
9f3ea250 404 } else if (vals[k] > mega) {
acd1e437 405 sprintf(temp, "%uK", (unsigned int)(vals[k]/kilo));
aba5acdf
SH
406 fprintf(fp, "%-6s ", temp);
407 } else
acd1e437 408 fprintf(fp, "%-6u ", (unsigned int)vals[k]);
aba5acdf
SH
409}
410
d1f28cf1 411static void print_head(FILE *fp)
aba5acdf
SH
412{
413 fprintf(fp, "#%s\n", info_source);
414 fprintf(fp, "%-15s ", "Interface");
415
416 fprintf(fp, "%8s/%-6s ", "RX Pkts", "Rate");
417 fprintf(fp, "%8s/%-6s ", "TX Pkts", "Rate");
418 fprintf(fp, "%8s/%-6s ", "RX Data", "Rate");
acd1e437 419 fprintf(fp, "%8s/%-6s\n", "TX Data", "Rate");
aba5acdf
SH
420
421 if (!show_errors) {
422 fprintf(fp, "%-15s ", "");
423 fprintf(fp, "%8s/%-6s ", "RX Errs", "Drop");
424 fprintf(fp, "%8s/%-6s ", "TX Errs", "Drop");
425 fprintf(fp, "%8s/%-6s ", "RX Over", "Rate");
acd1e437 426 fprintf(fp, "%8s/%-6s\n", "TX Coll", "Rate");
aba5acdf
SH
427 } else {
428 fprintf(fp, "%-15s ", "");
429 fprintf(fp, "%8s/%-6s ", "RX Errs", "Rate");
430 fprintf(fp, "%8s/%-6s ", "RX Drop", "Rate");
431 fprintf(fp, "%8s/%-6s ", "RX Over", "Rate");
acd1e437 432 fprintf(fp, "%8s/%-6s\n", "RX Leng", "Rate");
aba5acdf
SH
433
434 fprintf(fp, "%-15s ", "");
435 fprintf(fp, "%8s/%-6s ", "RX Crc", "Rate");
436 fprintf(fp, "%8s/%-6s ", "RX Frm", "Rate");
437 fprintf(fp, "%8s/%-6s ", "RX Fifo", "Rate");
acd1e437 438 fprintf(fp, "%8s/%-6s\n", "RX Miss", "Rate");
aba5acdf
SH
439
440 fprintf(fp, "%-15s ", "");
441 fprintf(fp, "%8s/%-6s ", "TX Errs", "Rate");
442 fprintf(fp, "%8s/%-6s ", "TX Drop", "Rate");
443 fprintf(fp, "%8s/%-6s ", "TX Coll", "Rate");
acd1e437 444 fprintf(fp, "%8s/%-6s\n", "TX Carr", "Rate");
aba5acdf
SH
445
446 fprintf(fp, "%-15s ", "");
447 fprintf(fp, "%8s/%-6s ", "TX Abrt", "Rate");
448 fprintf(fp, "%8s/%-6s ", "TX Fifo", "Rate");
449 fprintf(fp, "%8s/%-6s ", "TX Hear", "Rate");
acd1e437 450 fprintf(fp, "%8s/%-6s\n", "TX Wind", "Rate");
aba5acdf
SH
451 }
452}
453
fcc16c22 454static void print_one_json(json_writer_t *jw, const struct ifstat_ent *n,
ec3e625c
SH
455 const unsigned long long *vals)
456{
fcc16c22
SH
457 int i, m = show_errors ? 20 : 10;
458
459 jsonw_name(jw, n->name);
460 jsonw_start_object(jw);
461
acd1e437 462 for (i = 0; i < m && stats[i]; i++)
fcc16c22
SH
463 jsonw_uint_field(jw, stats[i], vals[i]);
464
465 jsonw_end_object(jw);
ec3e625c
SH
466}
467
468static void print_one_if(FILE *fp, const struct ifstat_ent *n,
469 const unsigned long long *vals)
aba5acdf
SH
470{
471 int i;
ec3e625c 472
aba5acdf 473 fprintf(fp, "%-15s ", n->name);
acd1e437 474 for (i = 0; i < 4; i++)
aba5acdf
SH
475 format_rate(fp, vals, n->rate, i);
476 fprintf(fp, "\n");
477
478 if (!show_errors) {
479 fprintf(fp, "%-15s ", "");
480 format_pair(fp, vals, 4, 6);
481 format_pair(fp, vals, 5, 7);
482 format_rate(fp, vals, n->rate, 11);
483 format_rate(fp, vals, n->rate, 9);
484 fprintf(fp, "\n");
485 } else {
486 fprintf(fp, "%-15s ", "");
487 format_rate(fp, vals, n->rate, 4);
488 format_rate(fp, vals, n->rate, 6);
489 format_rate(fp, vals, n->rate, 11);
490 format_rate(fp, vals, n->rate, 10);
491 fprintf(fp, "\n");
492
493 fprintf(fp, "%-15s ", "");
494 format_rate(fp, vals, n->rate, 12);
495 format_rate(fp, vals, n->rate, 13);
496 format_rate(fp, vals, n->rate, 14);
497 format_rate(fp, vals, n->rate, 15);
498 fprintf(fp, "\n");
499
500 fprintf(fp, "%-15s ", "");
501 format_rate(fp, vals, n->rate, 5);
502 format_rate(fp, vals, n->rate, 7);
503 format_rate(fp, vals, n->rate, 9);
504 format_rate(fp, vals, n->rate, 17);
505 fprintf(fp, "\n");
506
507 fprintf(fp, "%-15s ", "");
508 format_rate(fp, vals, n->rate, 16);
509 format_rate(fp, vals, n->rate, 18);
510 format_rate(fp, vals, n->rate, 19);
511 format_rate(fp, vals, n->rate, 20);
512 fprintf(fp, "\n");
513 }
514}
515
d1f28cf1 516static void dump_kern_db(FILE *fp)
aba5acdf 517{
fcc16c22 518 json_writer_t *jw = json_output ? jsonw_new(fp) : NULL;
dc484542 519 struct ifstat_ent *n;
aba5acdf 520
fcc16c22 521 if (jw) {
d721a145 522 jsonw_start_object(jw);
fcc16c22
SH
523 jsonw_pretty(jw, pretty);
524 jsonw_name(jw, info_source);
525 jsonw_start_object(jw);
526 } else
ec3e625c 527 print_head(fp);
aba5acdf 528
acd1e437 529 for (n = kern_db; n; n = n->next) {
aba5acdf
SH
530 if (!match(n->name))
531 continue;
ec3e625c 532
fcc16c22
SH
533 if (jw)
534 print_one_json(jw, n, n->val);
535 else
ec3e625c 536 print_one_if(fp, n, n->val);
aba5acdf 537 }
b530cef0
PS
538 if (jw) {
539 jsonw_end_object(jw);
540
541 jsonw_end_object(jw);
542 jsonw_destroy(&jw);
543 }
aba5acdf
SH
544}
545
d1f28cf1 546static void dump_incr_db(FILE *fp)
aba5acdf
SH
547{
548 struct ifstat_ent *n, *h;
fcc16c22 549 json_writer_t *jw = json_output ? jsonw_new(fp) : NULL;
aba5acdf 550
ec3e625c 551 h = hist_db;
fcc16c22 552 if (jw) {
d721a145 553 jsonw_start_object(jw);
fcc16c22
SH
554 jsonw_pretty(jw, pretty);
555 jsonw_name(jw, info_source);
556 jsonw_start_object(jw);
557 } else
ec3e625c 558 print_head(fp);
aba5acdf 559
acd1e437 560 for (n = kern_db; n; n = n->next) {
aba5acdf
SH
561 int i;
562 unsigned long long vals[MAXS];
563 struct ifstat_ent *h1;
564
565 memcpy(vals, n->val, sizeof(vals));
566
567 for (h1 = h; h1; h1 = h1->next) {
568 if (h1->ifindex == n->ifindex) {
569 for (i = 0; i < MAXS; i++)
570 vals[i] -= h1->val[i];
571 h = h1->next;
572 break;
573 }
574 }
575 if (!match(n->name))
576 continue;
ec3e625c 577
fcc16c22
SH
578 if (jw)
579 print_one_json(jw, n, n->val);
580 else
ec3e625c 581 print_one_if(fp, n, vals);
aba5acdf 582 }
aba5acdf 583
fcc16c22 584 if (jw) {
d721a145
AK
585 jsonw_end_object(jw);
586
fcc16c22
SH
587 jsonw_end_object(jw);
588 jsonw_destroy(&jw);
589 }
590}
aba5acdf
SH
591
592static int children;
593
d1f28cf1 594static void sigchild(int signo)
aba5acdf
SH
595{
596}
597
d1f28cf1 598static void update_db(int interval)
aba5acdf
SH
599{
600 struct ifstat_ent *n, *h;
601
602 n = kern_db;
603 kern_db = NULL;
604
605 load_info();
606
607 h = kern_db;
608 kern_db = n;
609
610 for (n = kern_db; n; n = n->next) {
611 struct ifstat_ent *h1;
acd1e437 612
aba5acdf
SH
613 for (h1 = h; h1; h1 = h1->next) {
614 if (h1->ifindex == n->ifindex) {
615 int i;
acd1e437 616
aba5acdf
SH
617 for (i = 0; i < MAXS; i++) {
618 if ((long)(h1->ival[i] - n->ival[i]) < 0) {
ae665a52 619 memset(n->ival, 0, sizeof(n->ival));
aba5acdf
SH
620 break;
621 }
622 }
ae665a52 623 for (i = 0; i < MAXS; i++) {
aba5acdf 624 double sample;
5a52102b
NF
625 __u64 incr;
626
627 if (is_extended) {
628 incr = h1->val[i] - n->val[i];
629 n->val[i] = h1->val[i];
630 } else {
631 incr = (__u32) (h1->ival[i] - n->ival[i]);
632 n->val[i] += incr;
633 n->ival[i] = h1->ival[i];
634 }
acd1e437 635
aba5acdf
SH
636 sample = (double)(incr*1000)/interval;
637 if (interval >= scan_interval) {
638 n->rate[i] += W*(sample-n->rate[i]);
639 } else if (interval >= 1000) {
640 if (interval >= time_constant) {
641 n->rate[i] = sample;
642 } else {
643 double w = W*(double)interval/scan_interval;
acd1e437 644
aba5acdf
SH
645 n->rate[i] += w*(sample-n->rate[i]);
646 }
647 }
648 }
649
650 while (h != h1) {
651 struct ifstat_ent *tmp = h;
acd1e437 652
aba5acdf
SH
653 h = h->next;
654 free(tmp->name);
655 free(tmp);
656 };
657 h = h1->next;
658 free(h1->name);
659 free(h1);
660 break;
661 }
662 }
663 }
664}
665
acd1e437 666#define T_DIFF(a, b) (((a).tv_sec-(b).tv_sec)*1000 + ((a).tv_usec-(b).tv_usec)/1000)
aba5acdf
SH
667
668
d1f28cf1 669static void server_loop(int fd)
aba5acdf 670{
737f15f6 671 struct timeval snaptime = { 0 };
aba5acdf 672 struct pollfd p;
acd1e437 673
aba5acdf
SH
674 p.fd = fd;
675 p.events = p.revents = POLLIN;
676
677 sprintf(info_source, "%d.%lu sampling_interval=%d time_const=%d",
678 getpid(), (unsigned long)random(), scan_interval/1000, time_constant/1000);
679
680 load_info();
681
682 for (;;) {
683 int status;
dd81ee04 684 time_t tdiff;
aba5acdf 685 struct timeval now;
737f15f6 686
aba5acdf
SH
687 gettimeofday(&now, NULL);
688 tdiff = T_DIFF(now, snaptime);
689 if (tdiff >= scan_interval) {
690 update_db(tdiff);
691 snaptime = now;
692 tdiff = 0;
693 }
737f15f6 694
dd81ee04 695 if (poll(&p, 1, scan_interval - tdiff) > 0
aba5acdf
SH
696 && (p.revents&POLLIN)) {
697 int clnt = accept(fd, NULL, NULL);
acd1e437 698
aba5acdf
SH
699 if (clnt >= 0) {
700 pid_t pid;
acd1e437 701
aba5acdf
SH
702 if (children >= 5) {
703 close(clnt);
704 } else if ((pid = fork()) != 0) {
acd1e437 705 if (pid > 0)
aba5acdf
SH
706 children++;
707 close(clnt);
708 } else {
709 FILE *fp = fdopen(clnt, "w");
acd1e437 710
dd81ee04 711 if (fp)
aba5acdf 712 dump_raw_db(fp, 0);
aba5acdf
SH
713 exit(0);
714 }
715 }
716 }
717 while (children && waitpid(-1, &status, WNOHANG) > 0)
718 children--;
719 }
720}
721
d1f28cf1 722static int verify_forging(int fd)
aba5acdf
SH
723{
724 struct ucred cred;
737f15f6
SH
725 socklen_t olen = sizeof(cred);
726
acd1e437 727 if (getsockopt(fd, SOL_SOCKET, SO_PEERCRED, (void *)&cred, &olen) ||
aba5acdf
SH
728 olen < sizeof(cred))
729 return -1;
730 if (cred.uid == getuid() || cred.uid == 0)
731 return 0;
732 return -1;
733}
734
5a52102b
NF
735static void xstat_usage(void)
736{
737 fprintf(stderr,
1c2df613
NF
738"Usage: ifstat supported xstats:\n"
739" cpu_hits Counts only packets that went via the CPU.\n");
5a52102b
NF
740}
741
742struct extended_stats_options_t {
743 char *name;
744 int id;
745 int sub_type;
746};
747
748/* Note: if one xstat name is subset of another, it should be before it in this
749 * list.
750 * Name length must be under 64 chars.
751 */
752static const struct extended_stats_options_t extended_stats_options[] = {
1c2df613 753 {"cpu_hits", IFLA_STATS_LINK_OFFLOAD_XSTATS, IFLA_OFFLOAD_XSTATS_CPU_HIT},
5a52102b
NF
754};
755
756static const char *get_filter_type(const char *name)
757{
758 int name_len;
759 int i;
760
761 name_len = strlen(name);
762 for (i = 0; i < ARRAY_SIZE(extended_stats_options); i++) {
763 const struct extended_stats_options_t *xstat;
764
765 xstat = &extended_stats_options[i];
766 if (strncmp(name, xstat->name, name_len) == 0) {
767 filter_type = xstat->id;
768 sub_type = xstat->sub_type;
769 return xstat->name;
770 }
771 }
772
773 fprintf(stderr, "invalid ifstat extension %s\n", name);
774 xstat_usage();
775 return NULL;
776}
777
d7e0809e
SH
778static void usage(void) __attribute__((noreturn));
779
aba5acdf
SH
780static void usage(void)
781{
d7e0809e
SH
782 fprintf(stderr,
783"Usage: ifstat [OPTION] [ PATTERN [ PATTERN ] ]\n"
eca7a742
MF
784" -h, --help this message\n"
785" -a, --ignore ignore history\n"
786" -d, --scan=SECS sample every statistics every SECS\n"
787" -e, --errors show errors\n"
788" -j, --json format output in JSON\n"
789" -n, --nooutput do history only\n"
790" -p, --pretty pretty print\n"
791" -r, --reset reset history\n"
792" -s, --noupdate don't update history\n"
793" -t, --interval=SECS report average over the last SECS\n"
794" -V, --version output version information\n"
5a52102b
NF
795" -z, --zeros show entries with zero activity\n"
796" -x, --extended=TYPE show extended stats of TYPE\n");
d7e0809e
SH
797
798 exit(-1);
aba5acdf
SH
799}
800
9f3ea250
SH
801static const struct option longopts[] = {
802 { "help", 0, 0, 'h' },
803 { "ignore", 0, 0, 'a' },
804 { "scan", 1, 0, 'd'},
805 { "errors", 0, 0, 'e' },
806 { "nooutput", 0, 0, 'n' },
ec3e625c 807 { "json", 0, 0, 'j' },
9f3ea250 808 { "reset", 0, 0, 'r' },
fcc16c22 809 { "pretty", 0, 0, 'p' },
9f3ea250
SH
810 { "noupdate", 0, 0, 's' },
811 { "interval", 1, 0, 't' },
812 { "version", 0, 0, 'V' },
813 { "zeros", 0, 0, 'z' },
5a52102b 814 { "extended", 1, 0, 'x'},
9f3ea250
SH
815 { 0 }
816};
aba5acdf
SH
817
818int main(int argc, char *argv[])
819{
820 char hist_name[128];
821 struct sockaddr_un sun;
822 FILE *hist_fp = NULL;
5a52102b 823 const char *stats_type = NULL;
aba5acdf
SH
824 int ch;
825 int fd;
826
5a52102b
NF
827 is_extended = false;
828 while ((ch = getopt_long(argc, argv, "hjpvVzrnasd:t:ex:",
9f3ea250 829 longopts, NULL)) != EOF) {
acd1e437 830 switch (ch) {
aba5acdf
SH
831 case 'z':
832 dump_zeros = 1;
833 break;
834 case 'r':
835 reset_history = 1;
836 break;
837 case 'a':
838 ignore_history = 1;
839 break;
840 case 's':
841 no_update = 1;
842 break;
843 case 'n':
844 no_output = 1;
845 break;
846 case 'e':
847 show_errors = 1;
848 break;
ec3e625c
SH
849 case 'j':
850 json_output = 1;
851 break;
fcc16c22
SH
852 case 'p':
853 pretty = 1;
854 break;
aba5acdf 855 case 'd':
9f3ea250
SH
856 scan_interval = atoi(optarg) * 1000;
857 if (scan_interval <= 0) {
858 fprintf(stderr, "ifstat: invalid scan interval\n");
859 exit(-1);
860 }
aba5acdf
SH
861 break;
862 case 't':
9f3ea250
SH
863 time_constant = atoi(optarg);
864 if (time_constant <= 0) {
aba5acdf
SH
865 fprintf(stderr, "ifstat: invalid time constant divisor\n");
866 exit(-1);
867 }
868 break;
5a52102b
NF
869 case 'x':
870 stats_type = optarg;
871 is_extended = true;
872 break;
aba5acdf
SH
873 case 'v':
874 case 'V':
875 printf("ifstat utility, iproute2-ss%s\n", SNAPSHOT);
876 exit(0);
877 case 'h':
878 case '?':
879 default:
880 usage();
881 }
882 }
883
884 argc -= optind;
885 argv += optind;
886
5a52102b
NF
887 if (stats_type) {
888 stats_type = get_filter_type(stats_type);
889 if (!stats_type)
890 exit(-1);
891 }
892
aba5acdf
SH
893 sun.sun_family = AF_UNIX;
894 sun.sun_path[0] = 0;
895 sprintf(sun.sun_path+1, "ifstat%d", getuid());
896
897 if (scan_interval > 0) {
898 if (time_constant == 0)
899 time_constant = 60;
900 time_constant *= 1000;
901 W = 1 - 1/exp(log(10)*(double)scan_interval/time_constant);
902 if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
903 perror("ifstat: socket");
904 exit(-1);
905 }
acd1e437 906 if (bind(fd, (struct sockaddr *)&sun, 2+1+strlen(sun.sun_path+1)) < 0) {
aba5acdf
SH
907 perror("ifstat: bind");
908 exit(-1);
909 }
910 if (listen(fd, 5) < 0) {
911 perror("ifstat: listen");
912 exit(-1);
913 }
a7a9ddbb
MF
914 if (daemon(0, 0)) {
915 perror("ifstat: daemon");
916 exit(-1);
917 }
aba5acdf
SH
918 signal(SIGPIPE, SIG_IGN);
919 signal(SIGCHLD, sigchild);
920 server_loop(fd);
921 exit(0);
922 }
923
924 patterns = argv;
925 npatterns = argc;
926
927 if (getenv("IFSTAT_HISTORY"))
daf7bd5c
SH
928 snprintf(hist_name, sizeof(hist_name),
929 "%s", getenv("IFSTAT_HISTORY"));
aba5acdf 930 else
5a52102b
NF
931 if (!stats_type)
932 snprintf(hist_name, sizeof(hist_name),
933 "%s/.ifstat.u%d", P_tmpdir, getuid());
934 else
935 snprintf(hist_name, sizeof(hist_name),
936 "%s/.%s_ifstat.u%d", P_tmpdir, stats_type,
937 getuid());
aba5acdf
SH
938
939 if (reset_history)
940 unlink(hist_name);
941
942 if (!ignore_history || !no_update) {
943 struct stat stb;
944
945 fd = open(hist_name, O_RDWR|O_CREAT|O_NOFOLLOW, 0600);
946 if (fd < 0) {
947 perror("ifstat: open history file");
948 exit(-1);
949 }
950 if ((hist_fp = fdopen(fd, "r+")) == NULL) {
951 perror("ifstat: fdopen history file");
952 exit(-1);
953 }
954 if (flock(fileno(hist_fp), LOCK_EX)) {
955 perror("ifstat: flock history file");
956 exit(-1);
957 }
958 if (fstat(fileno(hist_fp), &stb) != 0) {
959 perror("ifstat: fstat history file");
960 exit(-1);
961 }
962 if (stb.st_nlink != 1 || stb.st_uid != getuid()) {
963 fprintf(stderr, "ifstat: something is so wrong with history file, that I prefer not to proceed.\n");
964 exit(-1);
965 }
966 if (!ignore_history) {
967 FILE *tfp;
9a230771 968 long uptime = -1;
acd1e437 969
aba5acdf
SH
970 if ((tfp = fopen("/proc/uptime", "r")) != NULL) {
971 if (fscanf(tfp, "%ld", &uptime) != 1)
972 uptime = -1;
973 fclose(tfp);
974 }
975 if (uptime >= 0 && time(NULL) >= stb.st_mtime+uptime) {
976 fprintf(stderr, "ifstat: history is aged out, resetting\n");
d572ed4d
PS
977 if (ftruncate(fileno(hist_fp), 0))
978 perror("ifstat: ftruncate");
aba5acdf
SH
979 }
980 }
981
982 load_raw_table(hist_fp);
983
984 hist_db = kern_db;
985 kern_db = NULL;
986 }
987
988 if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) >= 0 &&
acd1e437 989 (connect(fd, (struct sockaddr *)&sun, 2+1+strlen(sun.sun_path+1)) == 0
aba5acdf 990 || (strcpy(sun.sun_path+1, "ifstat0"),
acd1e437 991 connect(fd, (struct sockaddr *)&sun, 2+1+strlen(sun.sun_path+1)) == 0))
aba5acdf
SH
992 && verify_forging(fd) == 0) {
993 FILE *sfp = fdopen(fd, "r");
acd1e437 994
6d02518f
PS
995 if (!sfp) {
996 fprintf(stderr, "ifstat: fdopen failed: %s\n",
997 strerror(errno));
998 close(fd);
999 } else {
1000 load_raw_table(sfp);
1001 if (hist_db && source_mismatch) {
1002 fprintf(stderr, "ifstat: history is stale, ignoring it.\n");
1003 hist_db = NULL;
1004 }
1005 fclose(sfp);
aba5acdf 1006 }
aba5acdf
SH
1007 } else {
1008 if (fd >= 0)
1009 close(fd);
1010 if (hist_db && info_source[0] && strcmp(info_source, "kernel")) {
1011 fprintf(stderr, "ifstat: history is stale, ignoring it.\n");
1012 hist_db = NULL;
1013 info_source[0] = 0;
1014 }
1015 load_info();
1016 if (info_source[0] == 0)
1017 strcpy(info_source, "kernel");
1018 }
1019
1020 if (!no_output) {
1021 if (ignore_history || hist_db == NULL)
1022 dump_kern_db(stdout);
1023 else
1024 dump_incr_db(stdout);
1025 }
ec3e625c 1026
aba5acdf 1027 if (!no_update) {
d572ed4d
PS
1028 if (ftruncate(fileno(hist_fp), 0))
1029 perror("ifstat: ftruncate");
aba5acdf 1030 rewind(hist_fp);
ec3e625c
SH
1031
1032 json_output = 0;
aba5acdf 1033 dump_raw_db(hist_fp, 1);
ec3e625c 1034 fclose(hist_fp);
aba5acdf
SH
1035 }
1036 exit(0);
1037}