ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/statgrab/statgrab.c
(Generate patch)

Comparing projects/libstatgrab/src/statgrab/statgrab.c (file contents):
Revision 1.5 by ats, Thu Aug 28 21:21:32 2003 UTC vs.
Revision 1.18 by ats, Fri Feb 13 12:52:40 2004 UTC

# Line 1 | Line 1
1   /*
2   * i-scream central monitoring system
3   * http://www.i-scream.org
4 < * Copyright (C) 2000-2003 i-scream
4 > * Copyright (C) 2000-2004 i-scream
5   *
6   * This program is free software; you can redistribute it and/or
7   * modify it under the terms of the GNU General Public License
# Line 16 | Line 16
16   * You should have received a copy of the GNU General Public License
17   * along with this program; if not, write to the Free Software
18   * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19 + *
20 + * $Id$
21   */
22  
23   #ifdef HAVE_CONFIG_H
# Line 35 | Line 37 | typedef enum {
37          FLOAT,
38          DOUBLE,
39          STRING,
40 <        INT
40 >        INT,
41 >        DUPLEX
42   } stat_type;
43  
44   typedef enum {
# Line 66 | Line 69 | display_mode_type display_mode = DISPLAY_LINUX;
69   repeat_mode_type repeat_mode = REPEAT_NONE;
70   int repeat_time = 1;
71   int use_cpu_percent = 0;
72 + int use_diffs = 0;
73  
74   /* Exit with an error message. */
75   void die(const char *s) {
# Line 139 | Line 143 | void add_stat(stat_type type, void *stat, ...) {
143          ++num_stats;
144   }
145  
146 < /* Compare two stats by name, for sorting purposes. */
146 > /* Compare two stats by name for qsort and bsearch. */
147   int stats_compare(const void *a, const void *b) {
148          return strcmp(((stat *)a)->name, ((stat *)b)->name);
149   }
150  
151 < /* Clear and rebuild the stats array. */
152 < void get_stats(int use_diffs) {
153 <        cpu_states_t *cpu_s;
154 <        cpu_percent_t *cpu_p;
151 <        mem_stat_t *mem;
152 <        load_stat_t *load;
153 <        user_stat_t *user;
154 <        swap_stat_t *swap;
155 <        general_stat_t *gen;
156 <        disk_stat_t *disk;
157 <        diskio_stat_t *diskio;
158 <        process_stat_t *proc;
159 <        network_stat_t *net;
160 <        page_stat_t *page;
161 <        static int zero = 0;
162 <        int n, i;
151 > /* Compare up to the length of the key for bsearch. */
152 > int stats_compare_prefix(const void *key, const void *item) {
153 >        const char *kn = ((stat *)key)->name;
154 >        const char *in = ((stat *)item)->name;
155  
156 <        clear_stats();
156 >        return strncmp(kn, in, strlen(kn));
157 > }
158  
159 + void populate_const() {
160 +        static int zero = 0;
161 +
162          /* Constants, for use with MRTG mode. */
163          add_stat(INT, &zero, "const", "0", NULL);
164 + }
165  
166 <        /* FIXME when only fetching some stats, it'd be more efficient to only
170 <           do the libstatgrab calls needed, rather than fetching everything. */
171 <
166 > void populate_cpu() {
167          if (use_cpu_percent) {
168 <                cpu_p = cpu_percent_usage();
168 >                cpu_percent_t *cpu_p = cpu_percent_usage();
169 >
170                  if (cpu_p != NULL) {
171                          add_stat(FLOAT, &cpu_p->user,
172                                   "cpu", "user", NULL);
# Line 184 | Line 180 | void get_stats(int use_diffs) {
180                                   "cpu", "swap", NULL);
181                          add_stat(FLOAT, &cpu_p->nice,
182                                   "cpu", "nice", NULL);
183 <                        add_stat(TIME_T, &cpu_s->systime,
183 >                        add_stat(TIME_T, &cpu_p->time_taken,
184                                   "cpu", "time_taken", NULL);
185                  }
186          } else {
187 +                cpu_states_t *cpu_s;
188 +
189                  cpu_s = use_diffs ? get_cpu_diff() : get_cpu_totals();
190                  if (cpu_s != NULL) {
191                          add_stat(LONG_LONG, &cpu_s->user,
# Line 208 | Line 206 | void get_stats(int use_diffs) {
206                                   "cpu", "systime", NULL);
207                  }
208          }
209 + }
210  
211 <        mem = get_memory_stats();
211 > void populate_mem() {
212 >        mem_stat_t *mem = get_memory_stats();
213 >
214          if (mem != NULL) {
215                  add_stat(LONG_LONG, &mem->total, "mem", "total", NULL);
216                  add_stat(LONG_LONG, &mem->free, "mem", "free", NULL);
217                  add_stat(LONG_LONG, &mem->used, "mem", "used", NULL);
218                  add_stat(LONG_LONG, &mem->cache, "mem", "cache", NULL);
219          }
220 + }
221  
222 <        load = get_load_stats();
222 > void populate_load() {
223 >        load_stat_t *load = get_load_stats();
224 >
225          if (load != NULL) {
226                  add_stat(DOUBLE, &load->min1, "load", "min1", NULL);
227                  add_stat(DOUBLE, &load->min5, "load", "min5", NULL);
228                  add_stat(DOUBLE, &load->min15, "load", "min15", NULL);
229          }
230 + }
231  
232 <        user = get_user_stats();
232 > void populate_user() {
233 >        user_stat_t *user = get_user_stats();
234 >
235          if (user != NULL) {
236                  add_stat(INT, &user->num_entries, "user", "num", NULL);
237                  add_stat(STRING, &user->name_list, "user", "names", NULL);
238          }
239 + }
240  
241 <        swap = get_swap_stats();
241 > void populate_swap() {
242 >        swap_stat_t *swap = get_swap_stats();
243 >
244          if (swap != NULL) {
245                  add_stat(LONG_LONG, &swap->total, "swap", "total", NULL);
246                  add_stat(LONG_LONG, &swap->used, "swap", "used", NULL);
247                  add_stat(LONG_LONG, &swap->free, "swap", "free", NULL);
248          }
249 + }
250  
251 <        gen = get_general_stats();
251 > void populate_general() {
252 >        general_stat_t *gen = get_general_stats();
253 >
254          if (gen != NULL) {
255                  add_stat(STRING, &gen->os_name,
256                           "general", "os_name", NULL);
# Line 249 | Line 262 | void get_stats(int use_diffs) {
262                  add_stat(STRING, &gen->hostname, "general", "hostname", NULL);
263                  add_stat(TIME_T, &gen->uptime, "general", "uptime", NULL);
264          }
265 + }
266  
267 <        disk = get_disk_stats(&n);
267 > void populate_fs() {
268 >        int n, i;
269 >        disk_stat_t *disk = get_disk_stats(&n);
270 >
271          if (disk != NULL) {
272                  for (i = 0; i < n; i++) {
273                          /* FIXME it'd be nicer if libstatgrab did this */
274 <                        const char *name = disk[i].device_name,
275 <                                   *p = strrchr(name, '/');
276 <                        if (p != NULL)
277 <                                name = p + 1;
278 <                        if (*name == '\0')
279 <                                name = "root";
280 <        
274 >                        char *buf, *name, *p;
275 >                        const char *device = disk[i].device_name;
276 >
277 >                        if (strcmp(device, "/") == 0)
278 >                                device = "root";
279 >
280 >                        buf = strdup(device);
281 >                        if (buf == NULL)
282 >                                die("out of memory");
283 >
284 >                        name = buf;
285 >                        if (strlen(name) == 2 && name[1] == ':')
286 >                                name[1] = '\0';
287 >                        if (strncmp(name, "/dev/", 5) == 0)
288 >                                name += 5;
289 >                        while ((p = strchr(name, '/')) != NULL)
290 >                                *p = '_';
291 >
292                          add_stat(STRING, &disk[i].device_name,
293                                   "fs", name, "device_name", NULL);
294                          add_stat(STRING, &disk[i].fs_type,
# Line 279 | Line 307 | void get_stats(int use_diffs) {
307                                   "fs", name, "used_inodes", NULL);
308                          add_stat(LONG_LONG, &disk[i].free_inodes,
309                                   "fs", name, "free_inodes", NULL);
310 +
311 +                        free(buf);
312                  }
313          }
314 + }
315  
316 + void populate_disk() {
317 +        int n, i;
318 +        diskio_stat_t *diskio;
319 +
320          diskio = use_diffs ? get_diskio_stats_diff(&n) : get_diskio_stats(&n);
321          if (diskio != NULL) {
322                  for (i = 0; i < n; i++) {
# Line 297 | Line 332 | void get_stats(int use_diffs) {
332                                   "disk", name, "systime", NULL);
333                  }
334          }
335 + }
336  
337 <        proc = get_process_stats();
337 > void populate_proc() {
338 >        process_stat_t *proc = get_process_stats();
339 >
340          if (proc != NULL) {
341                  add_stat(INT, &proc->total, "proc", "total", NULL);
342                  add_stat(INT, &proc->running, "proc", "running", NULL);
# Line 306 | Line 344 | void get_stats(int use_diffs) {
344                  add_stat(INT, &proc->stopped, "proc", "stopped", NULL);
345                  add_stat(INT, &proc->zombie, "proc", "zombie", NULL);
346          }
347 + }
348  
349 + void populate_net() {
350 +        int n, i;
351 +        network_stat_t *net;
352 +        network_iface_stat_t *iface;
353 +
354          net = use_diffs ? get_network_stats_diff(&n) : get_network_stats(&n);
355          if (net != NULL) {
356                  for (i = 0; i < n; i++) {
# Line 323 | Line 367 | void get_stats(int use_diffs) {
367                  }
368          }
369  
370 +        iface = get_network_iface_stats(&n);
371 +        if (iface != NULL) {
372 +                for (i = 0; i < n; i++) {
373 +                        const char *name = iface[i].interface_name;
374 +
375 +                        add_stat(INT, &iface[i].speed,
376 +                                 "net", name, "speed", NULL);
377 +                        add_stat(DUPLEX, &iface[i].dup,
378 +                                 "net", name, "duplex", NULL);
379 +                }
380 +        }
381 + }
382 +
383 + void populate_page() {
384 +        page_stat_t *page;
385 +
386          page = use_diffs ? get_page_stats_diff() : get_page_stats();
387          if (page != NULL) {
388                  add_stat(LONG_LONG, &page->pages_pagein, "page", "in", NULL);
389                  add_stat(LONG_LONG, &page->pages_pageout, "page", "out", NULL);
390 <                add_stat(LONG_LONG, &page->systime, "page", "systime", NULL);
390 >                add_stat(TIME_T, &page->systime, "page", "systime", NULL);
391          }
392 + }
393  
394 + typedef struct {
395 +        const char *name;
396 +        void (*populate)();
397 +        int interesting;
398 + } toplevel;
399 + toplevel toplevels[] = {
400 +        {"const.", populate_const, 0},
401 +        {"cpu.", populate_cpu, 0},
402 +        {"mem.", populate_mem, 0},
403 +        {"load.", populate_load, 0},
404 +        {"user.", populate_user, 0},
405 +        {"swap.", populate_swap, 0},
406 +        {"general.", populate_general, 0},
407 +        {"fs.", populate_fs, 0},
408 +        {"disk.", populate_disk, 0},
409 +        {"proc.", populate_proc, 0},
410 +        {"net.", populate_net, 0},
411 +        {"page.", populate_page, 0},
412 +        {NULL, NULL, 0}
413 + };
414 +
415 + /* Set the "interesting" flag on the sections that we actually need to
416 +   fetch. */
417 + void select_interesting(int argc, char **argv) {
418 +        toplevel *t;
419 +
420 +        if (argc == 0) {
421 +                for (t = &toplevels[0]; t->name != NULL; t++)
422 +                        t->interesting = 1;
423 +        } else {
424 +                int i;
425 +
426 +                for (i = 0; i < argc; i++) {
427 +                        for (t = &toplevels[0]; t->name != NULL; t++) {
428 +                                if (strncmp(argv[i], t->name,
429 +                                            strlen(t->name)) == 0) {
430 +                                        t->interesting = 1;
431 +                                        break;
432 +                                }
433 +                        }
434 +                }
435 +        }
436 + }
437 +
438 + /* Clear and rebuild the stats array. */
439 + void get_stats() {
440 +        toplevel *t;
441 +
442 +        clear_stats();
443 +
444 +        for (t = &toplevels[0]; t->name != NULL; t++) {
445 +                if (t->interesting)
446 +                        t->populate();
447 +        }
448 +
449          qsort(stats, num_stats, sizeof *stats, stats_compare);
450   }
451  
# Line 358 | Line 474 | void print_stat_value(const stat *s) {
474          case INT:
475                  printf("%d", *(int *)v);
476                  break;
477 +        case DUPLEX:
478 +                switch (*(duplex *) v) {
479 +                case FULL_DUPLEX:
480 +                        printf("full");
481 +                        break;
482 +                case HALF_DUPLEX:
483 +                        printf("half");
484 +                        break;
485 +                default:
486 +                        printf("unknown");
487 +                        break;
488 +                }
489 +                break;
490          }
491   }
492  
# Line 389 | Line 518 | void print_stats(int argc, char **argv) {
518          } else {
519                  /* Print selected stats. */
520                  for (i = optind; i < argc; i++) {
521 +                        char *name = argv[i];
522                          stat key;
523 <                        const stat *s;
523 >                        const stat *s, *end;
524 >                        int (*compare)(const void *, const void *);
525  
526 <                        key.name = argv[i];
526 >                        key.name = name;
527 >                        if (name[strlen(name) - 1] == '.')
528 >                                compare = stats_compare_prefix;
529 >                        else
530 >                                compare = stats_compare;
531 >
532                          s = (const stat *)bsearch(&key, stats, num_stats,
533 <                                                  sizeof *stats,
534 <                                                  stats_compare);
535 <                        if (s != NULL) {
533 >                                                  sizeof *stats, compare);
534 >                        if (s == NULL) {
535 >                                printf("Unknown stat %s\n", name);
536 >                                continue;
537 >                        }
538 >
539 >                        /* Find the range of stats the user wanted. */
540 >                        for (; s >= stats; s--) {
541 >                                if (compare(&key, s) != 0)
542 >                                        break;
543 >                        }
544 >                        s++;
545 >                        for (end = s; end < &stats[num_stats]; end++) {
546 >                                if (compare(&key, end) != 0)
547 >                                        break;
548 >                        }
549 >
550 >                        /* And print them. */
551 >                        for (; s < end; s++) {
552                                  print_stat(s);
553                          }
554                  }
# Line 405 | Line 557 | void print_stats(int argc, char **argv) {
557  
558   void usage() {
559          printf("Usage: statgrab [OPTION]... [STAT]...\n"
560 <               "Display system statistics (all statistics by default).\n"
560 >               "Display system statistics.\n"
561 >               "\n"
562 >               "If no STATs are given, all will be displayed. Specify 'STAT.' to display all\n"
563 >               "statistics starting with that prefix.\n"
564                 "\n");
565          printf("  -l         Linux sysctl-style output (default)\n"
566                 "  -b         BSD sysctl-style output\n"
# Line 472 | Line 627 | int main(int argc, char **argv) {
627          if (use_cpu_percent && repeat_mode == REPEAT_NONE)
628                  die("CPU percentage usage display requires stat differences");
629  
630 +        if (repeat_mode == REPEAT_NONE)
631 +                use_diffs = 0;
632 +        else
633 +                use_diffs = 1;
634 +
635 +        select_interesting(argc - optind, &argv[optind]);
636 +
637 +        /* We don't care if statgrab_init fails, because we can just display
638 +           the statistics that can be read as non-root. */
639 +        statgrab_init();
640 +        if (statgrab_drop_privileges() != 0)
641 +                die("Failed to drop setuid/setgid privileges");
642 +
643          switch (repeat_mode) {
644          case REPEAT_NONE:
645 <                get_stats(0);
645 >                get_stats();
646                  print_stats(argc, argv);
647                  break;
648          case REPEAT_ONCE:
649 <                get_stats(1);
649 >                get_stats();
650                  sleep(repeat_time);
651 <                get_stats(1);
651 >                get_stats();
652                  print_stats(argc, argv);
653                  break;
654          case REPEAT_FOREVER:
655                  while (1) {
656 <                        get_stats(1);
656 >                        get_stats();
657                          print_stats(argc, argv);
658                          printf("\n");
659                          sleep(repeat_time);

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines