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.8 by ats, Fri Aug 29 06:56:12 2003 UTC vs.
Revision 1.17 by tdb, Mon Jan 19 16:49:23 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 140 | Line 142 | void add_stat(stat_type type, void *stat, ...) {
142          ++num_stats;
143   }
144  
145 < /* Compare two stats by name, for sorting purposes. */
145 > /* Compare two stats by name for qsort and bsearch. */
146   int stats_compare(const void *a, const void *b) {
147          return strcmp(((stat *)a)->name, ((stat *)b)->name);
148   }
149  
150 + /* Compare up to the length of the key for bsearch. */
151 + int stats_compare_prefix(const void *key, const void *item) {
152 +        const char *kn = ((stat *)key)->name;
153 +        const char *in = ((stat *)item)->name;
154 +
155 +        return strncmp(kn, in, strlen(kn));
156 + }
157 +
158   void populate_const() {
159          static int zero = 0;
160  
# Line 260 | Line 270 | void populate_fs() {
270          if (disk != NULL) {
271                  for (i = 0; i < n; i++) {
272                          /* FIXME it'd be nicer if libstatgrab did this */
273 <                        const char *name = disk[i].device_name,
274 <                                   *p = strrchr(name, '/');
275 <                        if (p != NULL)
276 <                                name = p + 1;
277 <                        if (*name == '\0')
278 <                                name = "root";
279 <        
273 >                        char *buf, *name, *p;
274 >                        const char *device = disk[i].device_name;
275 >
276 >                        if (strcmp(device, "/") == 0)
277 >                                device = "root";
278 >
279 >                        buf = strdup(device);
280 >                        if (buf == NULL)
281 >                                die("out of memory");
282 >
283 >                        name = buf;
284 >                        if (strlen(name) == 2 && name[1] == ':')
285 >                                name[1] = '\0';
286 >                        if (strncmp(name, "/dev/", 5) == 0)
287 >                                name += 5;
288 >                        while ((p = strchr(name, '/')) != NULL)
289 >                                *p = '_';
290 >
291                          add_stat(STRING, &disk[i].device_name,
292                                   "fs", name, "device_name", NULL);
293                          add_stat(STRING, &disk[i].fs_type,
# Line 285 | Line 306 | void populate_fs() {
306                                   "fs", name, "used_inodes", NULL);
307                          add_stat(LONG_LONG, &disk[i].free_inodes,
308                                   "fs", name, "free_inodes", NULL);
309 +
310 +                        free(buf);
311                  }
312          }
313   }
# Line 350 | Line 373 | void populate_page() {
373          if (page != NULL) {
374                  add_stat(LONG_LONG, &page->pages_pagein, "page", "in", NULL);
375                  add_stat(LONG_LONG, &page->pages_pageout, "page", "out", NULL);
376 <                add_stat(LONG_LONG, &page->systime, "page", "systime", NULL);
376 >                add_stat(TIME_T, &page->systime, "page", "systime", NULL);
377          }
378   }
379  
# Line 468 | Line 491 | void print_stats(int argc, char **argv) {
491          } else {
492                  /* Print selected stats. */
493                  for (i = optind; i < argc; i++) {
494 +                        char *name = argv[i];
495                          stat key;
496 <                        const stat *s;
496 >                        const stat *s, *end;
497 >                        int (*compare)(const void *, const void *);
498  
499 <                        key.name = argv[i];
499 >                        key.name = name;
500 >                        if (name[strlen(name) - 1] == '.')
501 >                                compare = stats_compare_prefix;
502 >                        else
503 >                                compare = stats_compare;
504 >
505                          s = (const stat *)bsearch(&key, stats, num_stats,
506 <                                                  sizeof *stats,
507 <                                                  stats_compare);
508 <                        if (s != NULL) {
506 >                                                  sizeof *stats, compare);
507 >                        if (s == NULL) {
508 >                                printf("Unknown stat %s\n", name);
509 >                                continue;
510 >                        }
511 >
512 >                        /* Find the range of stats the user wanted. */
513 >                        for (; s >= stats; s--) {
514 >                                if (compare(&key, s) != 0)
515 >                                        break;
516 >                        }
517 >                        s++;
518 >                        for (end = s; end < &stats[num_stats]; end++) {
519 >                                if (compare(&key, end) != 0)
520 >                                        break;
521 >                        }
522 >
523 >                        /* And print them. */
524 >                        for (; s < end; s++) {
525                                  print_stat(s);
526                          }
527                  }
# Line 484 | Line 530 | void print_stats(int argc, char **argv) {
530  
531   void usage() {
532          printf("Usage: statgrab [OPTION]... [STAT]...\n"
533 <               "Display system statistics (all statistics by default).\n"
533 >               "Display system statistics.\n"
534 >               "\n"
535 >               "If no STATs are given, all will be displayed. Specify 'STAT.' to display all\n"
536 >               "statistics starting with that prefix.\n"
537                 "\n");
538          printf("  -l         Linux sysctl-style output (default)\n"
539                 "  -b         BSD sysctl-style output\n"
# Line 557 | Line 606 | int main(int argc, char **argv) {
606                  use_diffs = 1;
607  
608          select_interesting(argc - optind, &argv[optind]);
609 +
610 +        /* We don't care if statgrab_init fails, because we can just display
611 +           the statistics that can be read as non-root. */
612 +        statgrab_init();
613 +        if (statgrab_drop_privileges() != 0)
614 +                die("Failed to drop setuid/setgid privileges");
615  
616          switch (repeat_mode) {
617          case REPEAT_NONE:

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines