libstatgrab/examples/network_traffic.c

/*
 * i-scream central monitoring system
 * http://www.i-scream.org
 * Copyright (C) 2000-2003 i-scream
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

/* A very basic example of how to get the network statistics from the system 
 * and diaply them. Also it adds up all the traffic to create and print out
 * a total 
 * Takes several arguments :
 * -d <number>  Takes the number of seconds to wait to get traffic sent since last call
 *              Note, this is not network traffic per second. Its the traffic since last
 *              call. If you want it traffic per second averaged over time since last call
 *              You need to divide against the time since last time this was called. The time
 *              between 2 calls is stored in systime in the network_stat_t structure.
 * -b           Display in bytes
 * -k           Display in kilobytes
 * -m           Display in megabytes
 */

#include <stdio.h>
#include <statgrab.h>
#include <stdlib.h>
#include <unistd.h>

int main(int argc, char **argv){

        extern char *optarg;
        extern int optind;
        int c;

        /* We default to 1 second updates and displaying in bytes*/
        int delay = 1;
        char units = 'b';

        network_stat_t *network_stats;
        int num_network_stats;

        /* Parse command line options */
        while ((c = getopt(argc, argv, "d:bkm")) != EOF){
                switch (c){
                        case 'd':
                                delay = atoi(optarg);
                                break;
                        case 'b':
                                units = 'b';
                                break;
                        case 'k':
                                units = 'k';    
                                break;
                        case 'm':
                                units = 'm';
                                break;
                }
        }

        /* We are not intrested in the amount of traffic ever transmitted, just differences. 
         * Because of this, we do nothing for the very first call.
         */
        
        network_stats = get_network_stats_diff(&num_network_stats);
        if (network_stats == NULL){
                perror("Error. Failed to get network stats");
                return 1;
        }

        /* Clear the screen ready for display the network stats */
        printf("\033[2J");
        
        /* Keep getting the network stats */
        while ( (network_stats = get_network_stats_diff(&num_network_stats)) != NULL){
                int x;
                int line_number = 2;

                long long total_tx=0;
                long long total_rx=0;

                for(x = 0; x < num_network_stats; x++){ 
                        /* Print at location 2, linenumber the interface name */
                        printf("\033[%d;2H%-25s : %-10s", line_number++, "Network Interface Name", network_stats->interface_name);
                        /* Print out at the correct location the traffic in the requsted units passed at command time */
                        switch(units){
                                case 'b':
                                        printf("\033[%d;2H%-25s : %8lld b", line_number++, "Network Interface Rx", network_stats->rx);
                                        printf("\033[%d;2H%-25s : %8lld b", line_number++, "Network Interface Tx", network_stats->tx);
                                        break;
                                case 'k':
                                        printf("\033[%d;2H%-25s : %5lld k", line_number++, "Network Interface Rx", (network_stats->rx / 1024));
                                        printf("\033[%d;2H%-25s : %5lld", line_number++, "Network Interface Tx", (network_stats->tx / 1024));
                                        break;
                                case 'm':
                                        printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Network Interface Rx", network_stats->rx / (1024.00*1024.00));
                                        printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Network Interface Tx", network_stats->tx / (1024.00*1024.00));
                        }
                        printf("\033[%d;2H%-25s : %ld ", line_number++, "Network Interface systime", (long) network_stats->systime);
                
                        /* Add a blank line between interfaces */       
                        line_number++;

                        /* Add up this interface to the total so we can display a "total" network io" */
                        total_tx+=network_stats->tx;
                        total_rx+=network_stats->rx;

                        /* Move the pointer onto the next interface. Since this returns a static buffer, we dont need
                         * to keep track of the orginal pointer to free later */
                        network_stats++;
                }
                
                printf("\033[%d;2H%-25s : %-10s", line_number++, "Network Interface Name", "Total Network IO");
                switch(units){
                        case 'b':
                                printf("\033[%d;2H%-25s : %8lld b", line_number++, "Network Total Rx", total_rx);
                                printf("\033[%d;2H%-25s : %8lld b", line_number++, "Network Total Tx", total_tx);
                                break;
                        case 'k':
                                printf("\033[%d;2H%-25s : %5lld k", line_number++, "Network Total Rx", (total_rx / 1024));
                                printf("\033[%d;2H%-25s : %5lld k", line_number++, "Network Total Tx", (total_tx / 1024));
                                break;
                        case 'm':
                                printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Network Total Rx", (total_rx  / (1024.00*1024.00)));
                                printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Network Total Tx", (total_tx  / (1024.00*1024.00)));
                                break;
                }
                
                fflush(stdout);

                sleep(delay);

        }

        return 0;

}
Revision:	1.3
Committed:	Sun Aug 24 20:24:09 2003 UTC (21 years, 3 months ago) by tdb
Content type:	text/plain
Branch:	MAIN
CVS Tags:	LIBSTATGRAB_0_7, LIBSTATGRAB_0_6_1, LIBSTATGRAB_0_6, LIBSTATGRAB_0_5_1, LIBSTATGRAB_0_5
Changes since 1.2:	+2 -2 lines
Log Message:	Tidy up of lots of little things. :)
#	User	Rev	Content
1	pajs	1.1	/*
2			* i-scream central monitoring system
3	tdb	1.3	* http://www.i-scream.org
4			* Copyright (C) 2000-2003 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			*/
20
21			/* A very basic example of how to get the network statistics from the system
22			* and diaply them. Also it adds up all the traffic to create and print out
23			* a total
24			* Takes several arguments :
25			* -d <number> Takes the number of seconds to wait to get traffic sent since last call
26			* Note, this is not network traffic per second. Its the traffic since last
27			* call. If you want it traffic per second averaged over time since last call
28			* You need to divide against the time since last time this was called. The time
29			* between 2 calls is stored in systime in the network_stat_t structure.
30			* -b Display in bytes
31			* -k Display in kilobytes
32			* -m Display in megabytes
33			*/
34
35			#include <stdio.h>
36			#include <statgrab.h>
37			#include <stdlib.h>
38			#include <unistd.h>
39
40			int main(int argc, char **argv){
41
42			extern char *optarg;
43			extern int optind;
44			int c;
45
46			/* We default to 1 second updates and displaying in bytes*/
47			int delay = 1;
48			char units = 'b';
49
50			network_stat_t *network_stats;
51			int num_network_stats;
52
53			/* Parse command line options */
54			while ((c = getopt(argc, argv, "d:bkm")) != EOF){
55			switch (c){
56			case 'd':
57			delay = atoi(optarg);
58			break;
59			case 'b':
60			units = 'b';
61			break;
62			case 'k':
63			units = 'k';
64			break;
65			case 'm':
66			units = 'm';
67			break;
68			}
69			}
70
71			/* We are not intrested in the amount of traffic ever transmitted, just differences.
72			* Because of this, we do nothing for the very first call.
73			*/
74
75			network_stats = get_network_stats_diff(&num_network_stats);
76			if (network_stats == NULL){
77			perror("Error. Failed to get network stats");
78			return 1;
79			}
80
81			/* Clear the screen ready for display the network stats */
82			printf("\033[2J");
83
84			/* Keep getting the network stats */
85			while ( (network_stats = get_network_stats_diff(&num_network_stats)) != NULL){
86			int x;
87			int line_number = 2;
88
89			long long total_tx=0;
90			long long total_rx=0;
91
92			for(x = 0; x < num_network_stats; x++){
93			/* Print at location 2, linenumber the interface name */
94			printf("\033[%d;2H%-25s : %-10s", line_number++, "Network Interface Name", network_stats->interface_name);
95			/* Print out at the correct location the traffic in the requsted units passed at command time */
96			switch(units){
97			case 'b':
98			printf("\033[%d;2H%-25s : %8lld b", line_number++, "Network Interface Rx", network_stats->rx);
99			printf("\033[%d;2H%-25s : %8lld b", line_number++, "Network Interface Tx", network_stats->tx);
100			break;
101			case 'k':
102			printf("\033[%d;2H%-25s : %5lld k", line_number++, "Network Interface Rx", (network_stats->rx / 1024));
103			printf("\033[%d;2H%-25s : %5lld", line_number++, "Network Interface Tx", (network_stats->tx / 1024));
104			break;
105			case 'm':
106	pajs	1.2	printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Network Interface Rx", network_stats->rx / (1024.00*1024.00));
107	pajs	1.1	printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Network Interface Tx", network_stats->tx / (1024.00*1024.00));
108			}
109			printf("\033[%d;2H%-25s : %ld ", line_number++, "Network Interface systime", (long) network_stats->systime);
110
111			/* Add a blank line between interfaces */
112			line_number++;
113
114			/* Add up this interface to the total so we can display a "total" network io" */
115			total_tx+=network_stats->tx;
116			total_rx+=network_stats->rx;
117
118			/* Move the pointer onto the next interface. Since this returns a static buffer, we dont need
119			* to keep track of the orginal pointer to free later */
120			network_stats++;
121			}
122
123			printf("\033[%d;2H%-25s : %-10s", line_number++, "Network Interface Name", "Total Network IO");
124			switch(units){
125			case 'b':
126			printf("\033[%d;2H%-25s : %8lld b", line_number++, "Network Total Rx", total_rx);
127			printf("\033[%d;2H%-25s : %8lld b", line_number++, "Network Total Tx", total_tx);
128			break;
129			case 'k':
130			printf("\033[%d;2H%-25s : %5lld k", line_number++, "Network Total Rx", (total_rx / 1024));
131			printf("\033[%d;2H%-25s : %5lld k", line_number++, "Network Total Tx", (total_tx / 1024));
132			break;
133			case 'm':
134			printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Network Total Rx", (total_rx / (1024.00*1024.00)));
135			printf("\033[%d;2H%-25s : %5.2f m", line_number++, "Network Total Tx", (total_tx / (1024.00*1024.00)));
136			break;
137			}
138
139			fflush(stdout);
140
141			sleep(delay);
142
143			}
144
145			return 0;
146
147			}