]>
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 <laforge@gnumonks.org>\n");
60 fprintf(stderr
, "This program is free software licensed under GNU GPLv2\nwith ABSOLUTELY NO WARRANTY.\n\n");
61 fprintf(stderr
, "Parameters:\n");
62 fprintf(stderr
, "\t-V --version\t\tPrint Version of Program\n");
63 fprintf(stderr
, "\t-c --count <count>\t"
64 "Print <count> number of intervals\n");
65 fprintf(stderr
, "\t-d --dump\t\t"
66 "Dump list of available files/keys\n");
67 fprintf(stderr
, "\t-j --json\t\t"
68 "Display in JSON format\n");
69 fprintf(stderr
, "\t-f --file <file>\tStatistics file to use\n");
70 fprintf(stderr
, "\t-h --help\t\tThis help message\n");
71 fprintf(stderr
, "\t-i --interval <intv>\t"
72 "Set interval to 'intv' seconds\n");
73 fprintf(stderr
, "\t-k --keys k,k,k,...\tDisplay only keys specified\n");
74 fprintf(stderr
, "\t-s --subject [0-2]\tControl header printing:\n");
75 fprintf(stderr
, "\t\t\t\t0 = never\n");
76 fprintf(stderr
, "\t\t\t\t1 = once\n");
77 fprintf(stderr
, "\t\t\t\t2 = every 20 lines (default))\n");
78 fprintf(stderr
, "\t-w --width n,n,n,...\tWidth for each field\n");
79 fprintf(stderr
, "\n");
86 struct lnstat_field
*lf
;
94 struct field_param params
[MAX_FIELDS
];
97 static void print_line(FILE *of
, const struct lnstat_file
*lnstat_files
,
98 const struct field_params
*fp
)
102 for (i
= 0; i
< fp
->num
; i
++) {
103 const struct lnstat_field
*lf
= fp
->params
[i
].lf
;
105 fprintf(of
, "%*lu|", fp
->params
[i
].print
.width
, lf
->result
);
110 static void print_json(FILE *of
, const struct lnstat_file
*lnstat_files
,
111 const struct field_params
*fp
)
113 json_writer_t
*jw
= jsonw_new(of
);
116 jsonw_start_object(jw
);
117 for (i
= 0; i
< fp
->num
; i
++) {
118 const struct lnstat_field
*lf
= fp
->params
[i
].lf
;
120 jsonw_uint_field(jw
, lf
->name
, lf
->result
);
122 jsonw_end_object(jw
);
126 /* find lnstat_field according to user specification */
127 static int map_field_params(struct lnstat_file
*lnstat_files
,
128 struct field_params
*fps
, int interval
)
131 struct lnstat_file
*lf
;
133 /* no field specification on commandline, need to build default */
135 for (lf
= lnstat_files
; lf
; lf
= lf
->next
) {
136 for (i
= 0; i
< lf
->num_fields
; i
++) {
137 fps
->params
[j
].lf
= &lf
->fields
[i
];
138 fps
->params
[j
].lf
->file
->interval
.tv_sec
=
140 if (!fps
->params
[j
].print
.width
)
141 fps
->params
[j
].print
.width
=
144 if (++j
>= MAX_FIELDS
- 1) {
146 "WARN: MAX_FIELDS (%d) reached, 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
] = calloc(1, HDR_LINE_LENGTH
);
188 for (i
= 0; i
< fps
->num
; i
++) {
189 char *cname
, *fname
= fps
->params
[i
].lf
->name
;
190 unsigned int width
= fps
->params
[i
].print
.width
;
192 snprintf(th
.hdr
[0]+ofs
, width
+2, "%*.*s|", width
, width
,
193 fps
->params
[i
].lf
->file
->basename
);
196 for (h
= 1; h
< HDR_LINES
; h
++) {
197 if (cname
- fname
>= strlen(fname
))
198 snprintf(th
.hdr
[h
]+ofs
, width
+2,
199 "%*.*s|", width
, width
, "");
202 snprintf(th
.hdr
[h
]+ofs
, width
+2,
203 "%*.*s|", width
, width
, cname
);
210 for (h
= 1; h
<= th
.num_lines
; h
++) {
211 for (i
= 0; i
< ofs
; i
++) {
212 if (th
.hdr
[h
][i
] == '\0')
220 static int print_hdr(FILE *of
, struct table_hdr
*th
)
224 for (i
= 0; i
< th
->num_lines
; i
++) {
225 fputs(th
->hdr
[i
], of
);
232 int main(int argc
, char **argv
)
234 struct lnstat_file
*lnstat_files
;
235 const char *basename
;
237 int interval
= DEFAULT_INTERVAL
;
243 } mode
= MODE_NORMAL
;
244 unsigned long count
= 0;
245 struct table_hdr
*header
;
246 static struct field_params fp
;
247 int num_req_files
= 0;
248 char *req_files
[LNSTAT_MAX_FILES
];
250 /* backwards compatibility mode for old tools */
251 basename
= strrchr(argv
[0], '/');
253 basename
+= 1; /* name after slash */
255 basename
= argv
[0]; /* no slash */
257 if (!strcmp(basename
, "rtstat")) {
258 /* rtstat compatibility mode */
259 req_files
[0] = "rt_cache";
261 } else if (!strcmp(basename
, "ctstat")) {
262 /* ctstat compatibility mode */
263 req_files
[0] = "ip_conntrack";
267 while ((c
= getopt_long(argc
, argv
, "Vc:djpf:h?i:k:s:w:",
268 opts
, NULL
)) != -1) {
274 count
= strtoul(optarg
, NULL
, 0);
283 req_files
[num_req_files
++] = strdup(optarg
);
290 sscanf(optarg
, "%u", &interval
);
293 tmp
= strdup(optarg
);
296 for (tok
= strtok(tmp
, ",");
298 tok
= strtok(NULL
, ",")) {
299 if (fp
.num
>= MAX_FIELDS
) {
301 "WARN: too many keys requested: (%d max)\n",
305 fp
.params
[fp
.num
++].name
= tok
;
309 sscanf(optarg
, "%u", &hdr
);
312 tmp
= strdup(optarg
);
316 for (tok
= strtok(tmp
, ",");
318 tok
= strtok(NULL
, ",")) {
319 len
= strtoul(tok
, NULL
, 0);
320 if (len
> FIELD_WIDTH_MAX
)
321 len
= FIELD_WIDTH_MAX
;
322 fp
.params
[i
].print
.width
= len
;
326 for (i
= 0; i
< MAX_FIELDS
; i
++)
327 fp
.params
[i
].print
.width
= len
;
336 lnstat_files
= lnstat_scan_dir(PROC_NET_STAT
, num_req_files
,
337 (const char **) req_files
);
341 lnstat_dump(stdout
, lnstat_files
);
346 if (!map_field_params(lnstat_files
, &fp
, interval
))
349 header
= build_hdr_string(lnstat_files
, &fp
, 80);
356 for (i
= 0; i
< count
|| !count
; i
++) {
357 lnstat_update(lnstat_files
);
358 if (mode
== MODE_JSON
)
359 print_json(stdout
, lnstat_files
, &fp
);
361 if ((hdr
> 1 && !(i
% 20)) ||
362 (hdr
== 1 && i
== 0))
363 print_hdr(stdout
, header
);
364 print_line(stdout
, lnstat_files
, &fp
);
367 if (i
< count
- 1 || !count
)