]>
git.proxmox.com Git - mirror_iproute2.git/blob - misc/lnstat.c
1 /* lnstat - Unified linux network statistics
3 * Copyright (C) 2004 by Harald Welte <laforge@gnumonks.org>
5 * Development of this code was funded by Astaro AG, http://www.astaro.com/
7 * Based on original concept and ideas from predecessor rtstat.c:
9 * Copyright 2001 by Robert Olsson <robert.olsson@its.uu.se>
10 * Uppsala University, Sweden
12 * This program is free software; you can redistribute it and/or modify
13 * it under the terms of the GNU General Public License as published by
14 * the Free Software Foundation; either version 2 of the License, or
15 * (at your option) any later version.
19 /* Maximum number of fields that can be displayed */
20 #define MAX_FIELDS 128
22 /* Maximum number of header lines */
25 /* default field width if none specified */
26 #define FIELD_WIDTH_DEFAULT 8
27 #define FIELD_WIDTH_MAX 20
29 #define DEFAULT_INTERVAL 2
31 #define HDR_LINE_LENGTH (MAX_FIELDS*FIELD_WIDTH_MAX)
39 #include <json_writer.h>
42 static struct option opts
[] = {
43 { "version", 0, NULL
, 'V' },
44 { "count", 1, NULL
, 'c' },
45 { "dump", 0, NULL
, 'd' },
46 { "json", 0, NULL
, 'j' },
47 { "file", 1, NULL
, 'f' },
48 { "help", 0, NULL
, 'h' },
49 { "interval", 1, NULL
, 'i' },
50 { "keys", 1, NULL
, 'k' },
51 { "subject", 1, NULL
, 's' },
52 { "width", 1, NULL
, 'w' },
53 { "oneline", 0, NULL
, 0 },
56 static int usage(char *name
, int exit_code
)
58 fprintf(stderr
, "%s Version %s\n", name
, LNSTAT_VERSION
);
59 fprintf(stderr
, "Copyright (C) 2004 by Harald Welte "
60 "<laforge@gnumonks.org>\n");
61 fprintf(stderr
, "This program is free software licensed under GNU GPLv2"
62 "\nwith ABSOLUTELY NO WARRANTY.\n\n");
63 fprintf(stderr
, "Parameters:\n");
64 fprintf(stderr
, "\t-V --version\t\tPrint Version of Program\n");
65 fprintf(stderr
, "\t-c --count <count>\t"
66 "Print <count> number of intervals\n");
67 fprintf(stderr
, "\t-d --dump\t\t"
68 "Dump list of available files/keys\n");
69 fprintf(stderr
, "\t-j --json\t\t"
70 "Display in JSON format\n");
71 fprintf(stderr
, "\t-f --file <file>\tStatistics file to use\n");
72 fprintf(stderr
, "\t-h --help\t\tThis help message\n");
73 fprintf(stderr
, "\t-i --interval <intv>\t"
74 "Set interval to 'intv' seconds\n");
75 fprintf(stderr
, "\t-k --keys k,k,k,...\tDisplay only keys specified\n");
76 fprintf(stderr
, "\t-s --subject [0-2]\t?\n");
77 fprintf(stderr
, "\t-w --width n,n,n,...\tWidth for each field\n");
78 fprintf(stderr
, "\n");
85 struct lnstat_field
*lf
;
93 struct field_param params
[MAX_FIELDS
];
96 static void print_line(FILE *of
, const struct lnstat_file
*lnstat_files
,
97 const struct field_params
*fp
)
101 for (i
= 0; i
< fp
->num
; i
++) {
102 const struct lnstat_field
*lf
= fp
->params
[i
].lf
;
104 fprintf(of
, "%*lu|", fp
->params
[i
].print
.width
, lf
->result
);
109 static void print_json(FILE *of
, const struct lnstat_file
*lnstat_files
,
110 const struct field_params
*fp
)
112 json_writer_t
*jw
= jsonw_new(of
);
115 jsonw_start_object(jw
);
116 for (i
= 0; i
< fp
->num
; i
++) {
117 const struct lnstat_field
*lf
= fp
->params
[i
].lf
;
119 jsonw_uint_field(jw
, lf
->name
, lf
->result
);
121 jsonw_end_object(jw
);
125 /* find lnstat_field according to user specification */
126 static int map_field_params(struct lnstat_file
*lnstat_files
,
127 struct field_params
*fps
, int interval
)
130 struct lnstat_file
*lf
;
132 /* no field specification on commandline, need to build default */
134 for (lf
= lnstat_files
; lf
; lf
= lf
->next
) {
135 for (i
= 0; i
< lf
->num_fields
; i
++) {
136 fps
->params
[j
].lf
= &lf
->fields
[i
];
137 fps
->params
[j
].lf
->file
->interval
.tv_sec
=
139 if (!fps
->params
[j
].print
.width
)
140 fps
->params
[j
].print
.width
=
143 if (++j
>= MAX_FIELDS
- 1) {
145 "WARN: MAX_FIELDS (%d) reached,"
146 " truncating number of keys\n",
157 for (i
= 0; i
< fps
->num
; i
++) {
158 fps
->params
[i
].lf
= lnstat_find_field(lnstat_files
,
159 fps
->params
[i
].name
);
160 if (!fps
->params
[i
].lf
) {
161 fprintf(stderr
, "Field `%s' unknown\n",
162 fps
->params
[i
].name
);
165 fps
->params
[i
].lf
->file
->interval
.tv_sec
= interval
;
166 if (!fps
->params
[i
].print
.width
)
167 fps
->params
[i
].print
.width
= FIELD_WIDTH_DEFAULT
;
174 char *hdr
[HDR_LINES
];
177 static struct table_hdr
*build_hdr_string(struct lnstat_file
*lnstat_files
,
178 struct field_params
*fps
,
182 static struct table_hdr th
;
185 for (i
= 0; i
< HDR_LINES
; i
++) {
186 th
.hdr
[i
] = malloc(HDR_LINE_LENGTH
);
187 memset(th
.hdr
[i
], 0, HDR_LINE_LENGTH
);
190 for (i
= 0; i
< fps
->num
; i
++) {
191 char *cname
, *fname
= fps
->params
[i
].lf
->name
;
192 unsigned int width
= fps
->params
[i
].print
.width
;
194 snprintf(th
.hdr
[0]+ofs
, width
+2, "%*.*s|", width
, width
,
195 fps
->params
[i
].lf
->file
->basename
);
198 for (h
= 1; h
< HDR_LINES
; h
++) {
199 if (cname
- fname
>= strlen(fname
))
200 snprintf(th
.hdr
[h
]+ofs
, width
+2,
201 "%*.*s|", width
, width
, "");
204 snprintf(th
.hdr
[h
]+ofs
, width
+2,
205 "%*.*s|", width
, width
, cname
);
212 for (h
= 1; h
<= th
.num_lines
; h
++) {
213 for (i
= 0; i
< ofs
; i
++) {
214 if (th
.hdr
[h
][i
] == '\0')
222 static int print_hdr(FILE *of
, struct table_hdr
*th
)
226 for (i
= 0; i
< th
->num_lines
; i
++) {
227 fputs(th
->hdr
[i
], of
);
234 int main(int argc
, char **argv
)
236 struct lnstat_file
*lnstat_files
;
237 const char *basename
;
239 int interval
= DEFAULT_INTERVAL
;
245 } mode
= MODE_NORMAL
;
246 unsigned long count
= 0;
247 struct table_hdr
*header
;
248 static struct field_params fp
;
249 int num_req_files
= 0;
250 char *req_files
[LNSTAT_MAX_FILES
];
252 /* backwards compatibility mode for old tools */
253 basename
= strrchr(argv
[0], '/');
255 basename
+= 1; /* name after slash */
257 basename
= argv
[0]; /* no slash */
259 if (!strcmp(basename
, "rtstat")) {
260 /* rtstat compatibility mode */
261 req_files
[0] = "rt_cache";
263 } else if (!strcmp(basename
, "ctstat")) {
264 /* ctstat compatibility mode */
265 req_files
[0] = "ip_conntrack";
269 while ((c
= getopt_long(argc
, argv
,"Vc:djpf:h?i:k:s:w:",
270 opts
, NULL
)) != -1) {
276 count
= strtoul(optarg
, NULL
, 0);
285 req_files
[num_req_files
++] = strdup(optarg
);
292 sscanf(optarg
, "%u", &interval
);
295 tmp
= strdup(optarg
);
298 for (tok
= strtok(tmp
, ",");
300 tok
= strtok(NULL
, ",")) {
301 if (fp
.num
>= MAX_FIELDS
) {
303 "WARN: too many keys"
304 " requested: (%d max)\n",
308 fp
.params
[fp
.num
++].name
= tok
;
312 sscanf(optarg
, "%u", &hdr
);
315 tmp
= strdup(optarg
);
319 for (tok
= strtok(tmp
, ",");
321 tok
= strtok(NULL
, ",")) {
322 len
= strtoul(tok
, NULL
, 0);
323 if (len
> FIELD_WIDTH_MAX
)
324 len
= FIELD_WIDTH_MAX
;
325 fp
.params
[i
].print
.width
= len
;
329 for (i
= 0; i
< MAX_FIELDS
; i
++)
330 fp
.params
[i
].print
.width
= len
;
339 lnstat_files
= lnstat_scan_dir(PROC_NET_STAT
, num_req_files
,
340 (const char **) req_files
);
344 lnstat_dump(stdout
, lnstat_files
);
349 if (!map_field_params(lnstat_files
, &fp
, interval
))
352 header
= build_hdr_string(lnstat_files
, &fp
, 80);
359 for (i
= 0; i
< count
|| !count
; ) {
360 lnstat_update(lnstat_files
);
361 if (mode
== MODE_JSON
)
362 print_json(stdout
, lnstat_files
, &fp
);
365 (! (i
% 20))) || (hdr
== 1 && i
== 0))
366 print_hdr(stdout
, header
);
367 print_line(stdout
, lnstat_files
, &fp
);
370 if (i
< count
- 1 || !count
)