libstatgrab/examples/network_traffic.c

/*
 * i-scream central monitoring system
 * http://www.i-scream.org
 * Copyright (C) 2000-2004 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.
 *
 * $Id: network_traffic.c,v 1.11 2004/03/08 12:30:08 tdb Exp $
 */

/* 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;
        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")) != -1){
                switch (c){
                        case 'd':
                                delay = atoi(optarg);
                                break;
                        case 'b':
                                units = 'b';
                                break;
                        case 'k':
                                units = 'k';    
                                break;
                        case 'm':
                                units = 'm';
                                break;
                }
        }

        /* Initialise statgrab */
        statgrab_init();

        /* Drop setuid/setgid privileges. */
        if (statgrab_drop_privileges() != 0) {
                perror("Error. Failed to drop privileges");
                return 1;
        }

        /* We are not interested 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;
                long long total_ipackets=0;
                long long total_opackets=0;
                long long total_ierrors=0;
                long long total_oerrors=0;
                long long total_collisions=0;

                for(x = 0; x < num_network_stats; x++){ 
                        /* Print at location 2, linenumber the interface name */
                        printf("\033[%d;2H%-30s : %-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%-30s : %8lld b", line_number++, "Network Interface Rx", network_stats->rx);
                                        printf("\033[%d;2H%-30s : %8lld b", line_number++, "Network Interface Tx", network_stats->tx);
                                        break;
                                case 'k':
                                        printf("\033[%d;2H%-30s : %5lld k", line_number++, "Network Interface Rx", (network_stats->rx / 1024));
                                        printf("\033[%d;2H%-30s : %5lld", line_number++, "Network Interface Tx", (network_stats->tx / 1024));
                                        break;
                                case 'm':
                                        printf("\033[%d;2H%-30s : %5.2f m", line_number++, "Network Interface Rx", network_stats->rx / (1024.00*1024.00));
                                        printf("\033[%d;2H%-30s : %5.2f m", line_number++, "Network Interface Tx", network_stats->tx / (1024.00*1024.00));
                        }
                        printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface packets in", network_stats->ipackets);
                        printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface packets out", network_stats->opackets);
                        printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface errors in", network_stats->ierrors);
                        printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface errors out", network_stats->oerrors);
                        printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface collisions", network_stats->collisions);
                        printf("\033[%d;2H%-30s : %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;
                        total_ipackets+=network_stats->ipackets;
                        total_opackets+=network_stats->opackets;
                        total_ierrors+=network_stats->ierrors;
                        total_oerrors+=network_stats->oerrors;
                        total_collisions+=network_stats->collisions;

                        /* 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%-30s : %-10s", line_number++, "Network Interface Name", "Total Network IO");
                switch(units){
                        case 'b':
                                printf("\033[%d;2H%-30s : %8lld b", line_number++, "Network Total Rx", total_rx);
                                printf("\033[%d;2H%-30s : %8lld b", line_number++, "Network Total Tx", total_tx);
                                break;
                        case 'k':
                                printf("\033[%d;2H%-30s : %5lld k", line_number++, "Network Total Rx", (total_rx / 1024));
                                printf("\033[%d;2H%-30s : %5lld k", line_number++, "Network Total Tx", (total_tx / 1024));
                                break;
                        case 'm':
                                printf("\033[%d;2H%-30s : %5.2f m", line_number++, "Network Total Rx", (total_rx  / (1024.00*1024.00)));
                                printf("\033[%d;2H%-30s : %5.2f m", line_number++, "Network Total Tx", (total_tx  / (1024.00*1024.00)));
                                break;
                }
                printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total packets in", total_ipackets);
                printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total packets out", total_opackets);
                printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total errors in", total_ierrors);
                printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total errors out", total_oerrors);
                printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total collisions", total_collisions);

                fflush(stdout);

                sleep(delay);

        }

        return 0;

}
Revision:	1.12
Committed:	Sun Apr 4 22:48:49 2004 UTC (21 years, 6 months ago) by tdb
Content type:	text/plain
Branch:	MAIN
Changes since 1.11:	+1 -2 lines
Log Message:	Fix a bunch of warnings on the BSD's.
#	Content
1	/*
2	* i-scream central monitoring system
3	* http://www.i-scream.org
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
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	* $Id: network_traffic.c,v 1.11 2004/03/08 12:30:08 tdb Exp $
21	*/
22
23	/* A very basic example of how to get the network 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	* -d <number> Takes the number of seconds to wait to get traffic sent since last call
28	* Note, this is not network traffic per second. Its the traffic since last
29	* call. If you want it traffic per second averaged over time since last call
30	* 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 network_stat_t structure.
32	* -b Display in bytes
33	* -k Display in kilobytes
34	* -m Display in megabytes
35	*/
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	int c;
46
47	/* We default to 1 second updates and displaying in bytes*/
48	int delay = 1;
49	char units = 'b';
50
51	network_stat_t *network_stats;
52	int num_network_stats;
53
54	/* Parse command line options */
55	while ((c = getopt(argc, argv, "d:bkm")) != -1){
56	switch (c){
57	case 'd':
58	delay = atoi(optarg);
59	break;
60	case 'b':
61	units = 'b';
62	break;
63	case 'k':
64	units = 'k';
65	break;
66	case 'm':
67	units = 'm';
68	break;
69	}
70	}
71
72	/* Initialise statgrab */
73	statgrab_init();
74
75	/* Drop setuid/setgid privileges. */
76	if (statgrab_drop_privileges() != 0) {
77	perror("Error. Failed to drop privileges");
78	return 1;
79	}
80
81	/* We are not interested in the amount of traffic ever transmitted, just differences.
82	* Because of this, we do nothing for the very first call.
83	*/
84
85	network_stats = get_network_stats_diff(&num_network_stats);
86	if (network_stats == NULL){
87	perror("Error. Failed to get network stats");
88	return 1;
89	}
90
91	/* Clear the screen ready for display the network stats */
92	printf("\033[2J");
93
94	/* Keep getting the network stats */
95	while ( (network_stats = get_network_stats_diff(&num_network_stats)) != NULL){
96	int x;
97	int line_number = 2;
98
99	long long total_tx=0;
100	long long total_rx=0;
101	long long total_ipackets=0;
102	long long total_opackets=0;
103	long long total_ierrors=0;
104	long long total_oerrors=0;
105	long long total_collisions=0;
106
107	for(x = 0; x < num_network_stats; x++){
108	/* Print at location 2, linenumber the interface name */
109	printf("\033[%d;2H%-30s : %-10s", line_number++, "Network Interface Name", network_stats->interface_name);
110	/* Print out at the correct location the traffic in the requsted units passed at command time */
111	switch(units){
112	case 'b':
113	printf("\033[%d;2H%-30s : %8lld b", line_number++, "Network Interface Rx", network_stats->rx);
114	printf("\033[%d;2H%-30s : %8lld b", line_number++, "Network Interface Tx", network_stats->tx);
115	break;
116	case 'k':
117	printf("\033[%d;2H%-30s : %5lld k", line_number++, "Network Interface Rx", (network_stats->rx / 1024));
118	printf("\033[%d;2H%-30s : %5lld", line_number++, "Network Interface Tx", (network_stats->tx / 1024));
119	break;
120	case 'm':
121	printf("\033[%d;2H%-30s : %5.2f m", line_number++, "Network Interface Rx", network_stats->rx / (1024.00*1024.00));
122	printf("\033[%d;2H%-30s : %5.2f m", line_number++, "Network Interface Tx", network_stats->tx / (1024.00*1024.00));
123	}
124	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface packets in", network_stats->ipackets);
125	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface packets out", network_stats->opackets);
126	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface errors in", network_stats->ierrors);
127	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface errors out", network_stats->oerrors);
128	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Interface collisions", network_stats->collisions);
129	printf("\033[%d;2H%-30s : %ld ", line_number++, "Network Interface systime", (long) network_stats->systime);
130
131	/* Add a blank line between interfaces */
132	line_number++;
133
134	/* Add up this interface to the total so we can display a "total" network io" */
135	total_tx+=network_stats->tx;
136	total_rx+=network_stats->rx;
137	total_ipackets+=network_stats->ipackets;
138	total_opackets+=network_stats->opackets;
139	total_ierrors+=network_stats->ierrors;
140	total_oerrors+=network_stats->oerrors;
141	total_collisions+=network_stats->collisions;
142
143	/* Move the pointer onto the next interface. Since this returns a static buffer, we dont need
144	* to keep track of the orginal pointer to free later */
145	network_stats++;
146	}
147
148	printf("\033[%d;2H%-30s : %-10s", line_number++, "Network Interface Name", "Total Network IO");
149	switch(units){
150	case 'b':
151	printf("\033[%d;2H%-30s : %8lld b", line_number++, "Network Total Rx", total_rx);
152	printf("\033[%d;2H%-30s : %8lld b", line_number++, "Network Total Tx", total_tx);
153	break;
154	case 'k':
155	printf("\033[%d;2H%-30s : %5lld k", line_number++, "Network Total Rx", (total_rx / 1024));
156	printf("\033[%d;2H%-30s : %5lld k", line_number++, "Network Total Tx", (total_tx / 1024));
157	break;
158	case 'm':
159	printf("\033[%d;2H%-30s : %5.2f m", line_number++, "Network Total Rx", (total_rx / (1024.00*1024.00)));
160	printf("\033[%d;2H%-30s : %5.2f m", line_number++, "Network Total Tx", (total_tx / (1024.00*1024.00)));
161	break;
162	}
163	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total packets in", total_ipackets);
164	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total packets out", total_opackets);
165	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total errors in", total_ierrors);
166	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total errors out", total_oerrors);
167	printf("\033[%d;2H%-30s : %lld ", line_number++, "Network Total collisions", total_collisions);
168
169	fflush(stdout);
170
171	sleep(delay);
172
173	}
174
175	return 0;
176
177	}