--- projects/libstatgrab/examples/network_traffic.c 2003/11/07 18:43:56 1.4 +++ projects/libstatgrab/examples/network_traffic.c 2004/04/05 16:23:08 1.13 @@ -1,7 +1,7 @@ /* * i-scream central monitoring system * http://www.i-scream.org - * Copyright (C) 2000-2003 i-scream + * 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 @@ -16,6 +16,8 @@ * 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.13 2004/04/05 16:23:08 tdb Exp $ */ /* A very basic example of how to get the network statistics from the system @@ -23,13 +25,13 @@ * a total * Takes several arguments : * -d 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 + * 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 + * -m Display in megabytes */ #include @@ -40,14 +42,13 @@ 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; + sg_network_io_stats *network_stats; int num_network_stats; /* Parse command line options */ @@ -68,11 +69,20 @@ int main(int argc, char **argv){ } } - /* We are not intrested in the amount of traffic ever transmitted, just differences. + /* Initialise statgrab */ + sg_init(); + + /* Drop setuid/setgid privileges. */ + if (sg_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); + + network_stats = sg_get_network_io_stats_diff(&num_network_stats); if (network_stats == NULL){ perror("Error. Failed to get network stats"); return 1; @@ -80,62 +90,82 @@ int main(int argc, char **argv){ /* 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){ + while ( (network_stats = sg_get_network_io_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%-25s : %-10s", line_number++, "Network Interface Name", network_stats->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%-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); + 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%-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; + 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%-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%-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%-25s : %ld ", line_number++, "Network Interface systime", (long) network_stats->systime); - + 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%-25s : %-10s", line_number++, "Network Interface Name", "Total Network IO"); + + printf("\033[%d;2H%-30s : %-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); + 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%-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)); + 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%-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))); + 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);