ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/examples/disk_traffic.c
Revision: 1.7
Committed: Mon Jan 19 16:49:19 2004 UTC (20 years, 9 months ago) by tdb
Content type: text/plain
Branch: MAIN
CVS Tags: LIBSTATGRAB_0_9, LIBSTATGRAB_0_8_2, LIBSTATGRAB_0_8_1
Changes since 1.6: +2 -0 lines
Log Message:
A whole bunch of minor cosmetic changes.

File Contents

# User Rev Content
1 pajs 1.1 /*
2     * i-scream central monitoring system
3 tdb 1.2 * http://www.i-scream.org
4 tdb 1.6 * Copyright (C) 2000-2004 i-scream
5 pajs 1.1 *
6     * This program is free software; you can redistribute it and/or
7     * modify it under the terms of the GNU General Public License
8     * as published by the Free Software Foundation; either version 2
9     * of the License, or (at your option) any later version.
10     *
11     * This program is distributed in the hope that it will be useful,
12     * but WITHOUT ANY WARRANTY; without even the implied warranty of
13     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14     * GNU General Public License for more details.
15     *
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 tdb 1.7 *
20     * $Id$
21 pajs 1.1 */
22    
23     /* A very basic example of how to get the disk statistics from the system
24     * and diaply them. Also it adds up all the traffic to create and print out
25     * a total
26     * Takes several arguments :
27 tdb 1.4 * -d <number> Takes the number of seconds to wait to get traffic sent since last call
28     * Note, this is not disk traffic per second. Its the traffic since last
29     * call. If you want it traffic per second averaged over time since last call
30 pajs 1.1 * You need to divide against the time since last time this was called. The time
31     * between 2 calls is stored in systime in the disk_stat_t structure.
32     * -b Display in bytes
33     * -k Display in kilobytes
34 tdb 1.4 * -m Display in megabytes
35 pajs 1.1 */
36    
37     #include <stdio.h>
38     #include <statgrab.h>
39     #include <stdlib.h>
40     #include <unistd.h>
41    
42     int main(int argc, char **argv){
43    
44     extern char *optarg;
45     extern int optind;
46     int c;
47    
48     /* We default to 1 second updates and displaying in bytes*/
49     int delay = 1;
50     char units = 'b';
51    
52     diskio_stat_t *diskio_stats;
53     int num_diskio_stats;
54    
55     /* Parse command line options */
56 ats 1.3 while ((c = getopt(argc, argv, "d:bkm")) != -1){
57 pajs 1.1 switch (c){
58     case 'd':
59     delay = atoi(optarg);
60     break;
61     case 'b':
62     units = 'b';
63     break;
64     case 'k':
65     units = 'k';
66     break;
67     case 'm':
68     units = 'm';
69     break;
70     }
71     }
72    
73 tdb 1.4 /* Initialise statgrab */
74     statgrab_init();
75    
76 tdb 1.5 /* Drop setuid/setgid privileges. */
77     if (statgrab_drop_privileges() != 0) {
78     perror("Error. Failed to drop privileges");
79     return 1;
80     }
81    
82 tdb 1.4 /* We are not interested in the amount of traffic ever transmitted, just differences.
83 pajs 1.1 * Because of this, we do nothing for the very first call.
84     */
85 tdb 1.4
86 pajs 1.1 diskio_stats = get_diskio_stats_diff(&num_diskio_stats);
87     if (diskio_stats == NULL){
88     perror("Error. Failed to get disk stats");
89     return 1;
90     }
91    
92     /* Clear the screen ready for display the disk stats */
93     printf("\033[2J");
94 tdb 1.4
95 pajs 1.1 /* Keep getting the disk stats */
96     while ( (diskio_stats = get_diskio_stats_diff(&num_diskio_stats)) != NULL){
97     int x;
98     int line_number = 2;
99    
100     long long total_write=0;
101     long long total_read=0;
102    
103     for(x = 0; x < num_diskio_stats; x++){
104     /* Print at location 2, linenumber the interface name */
105     printf("\033[%d;2H%-25s : %-10s", line_number++, "Disk Name", diskio_stats->disk_name);
106     /* Print out at the correct location the traffic in the requsted units passed at command time */
107     switch(units){
108     case 'b':
109     printf("\033[%d;2H%-25s : %8lld b", line_number++, "Disk read", diskio_stats->read_bytes);
110     printf("\033[%d;2H%-25s : %8lld b", line_number++, "Disk write", diskio_stats->write_bytes);
111     break;
112     case 'k':
113     printf("\033[%d;2H%-25s : %5lld k", line_number++, "Disk read", (diskio_stats->read_bytes / 1024));
114 tdb 1.4 printf("\033[%d;2H%-25s : %5lld", line_number++, "Disk write", (diskio_stats->write_bytes / 1024));
115     break;
116 pajs 1.1 case 'm':
117     printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Disk read", diskio_stats->read_bytes / (1024.00*1024.00));
118     printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Disk write", diskio_stats->write_bytes / (1024.00*1024.00));
119     }
120     printf("\033[%d;2H%-25s : %ld ", line_number++, "Disk systime", (long) diskio_stats->systime);
121 tdb 1.4
122 pajs 1.1 /* Add a blank line between interfaces */
123     line_number++;
124    
125     /* Add up this interface to the total so we can display a "total" disk io" */
126     total_write+=diskio_stats->write_bytes;
127     total_read+=diskio_stats->read_bytes;
128    
129     /* Move the pointer onto the next interface. Since this returns a static buffer, we dont need
130     * to keep track of the orginal pointer to free later */
131     diskio_stats++;
132     }
133 tdb 1.4
134 pajs 1.1 printf("\033[%d;2H%-25s : %-10s", line_number++, "Disk Name", "Total Disk IO");
135     switch(units){
136     case 'b':
137     printf("\033[%d;2H%-25s : %8lld b", line_number++, "Disk Total read", total_read);
138     printf("\033[%d;2H%-25s : %8lld b", line_number++, "Disk Total write", total_write);
139     break;
140     case 'k':
141     printf("\033[%d;2H%-25s : %5lld k", line_number++, "Disk Total read", (total_read / 1024));
142     printf("\033[%d;2H%-25s : %5lld k", line_number++, "Disk Total write", (total_write / 1024));
143     break;
144     case 'm':
145     printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Disk Total read", (total_read / (1024.00*1024.00)));
146     printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Disk Total write", (total_write / (1024.00*1024.00)));
147     break;
148     }
149 tdb 1.4
150 pajs 1.1 fflush(stdout);
151    
152     sleep(delay);
153    
154     }
155    
156     return 0;
157    
158     }