ViewVC Help

View File | Revision Log | Show Annotations | Revision Graph | Root Listing

root/i-scream/projects/libstatgrab/src/libstatgrab/network_stats.c

Comparing projects/libstatgrab/src/libstatgrab/network_stats.c (file contents):
Revision 1.75 by ats, Sat Jul 30 13:23:46 2005 UTC vs.
Revision 1.81 by tdb, Wed Jun 9 14:44:28 2010 UTC

#	Line 70 | Line 70 | typedef __uint64_t u64;
70		#include <sys/ioctl.h>
71		#include <unistd.h>
72		#endif
73	+	#ifdef WIN32
74	+	#include <windows.h>
75	+	#include <Iphlpapi.h>
76	+	#include "win32.h"
77	+	#endif
78
79		static void network_stat_init(sg_network_io_stats *s) {
80		s->interface_name = NULL;
#	Line 89 | Line 94 | static void network_stat_destroy(sg_network_io_stats *
94		VECTOR_DECLARE_STATIC(network_stats, sg_network_io_stats, 5,
95		network_stat_init, network_stat_destroy);
96
97	+	#ifdef WIN32
98	+	static PMIB_IFTABLE win32_get_devices()
99	+	{
100	+	PMIB_IFTABLE if_table;
101	+	PMIB_IFTABLE tmp;
102	+	unsigned long dwSize = 0;
103	+
104	+	// Allocate memory for pointers
105	+	if_table = sg_malloc(sizeof(MIB_IFTABLE));
106	+	if(if_table == NULL) {
107	+	return NULL;
108	+	}
109	+
110	+	// Get necessary size for the buffer
111	+	if(GetIfTable(if_table, &dwSize, 0) == ERROR_INSUFFICIENT_BUFFER) {
112	+	tmp = sg_realloc(if_table, dwSize);
113	+	if(tmp == NULL) {
114	+	free(if_table);
115	+	return NULL;
116	+	}
117	+	if_table = tmp;
118	+	}
119	+
120	+	// Get the data
121	+	if(GetIfTable(if_table, &dwSize, 0) != NO_ERROR) {
122	+	free(if_table);
123	+	return NULL;
124	+	}
125	+	return if_table;
126	+	}
127	+	#endif /* WIN32 */
128	+
129		sg_network_io_stats *sg_get_network_io_stats(int *entries){
130		int interfaces;
131		sg_network_io_stats *network_stat_ptr;
#	Line 110 | Line 147 | sg_network_io_stats *sg_get_network_io_stats(int *entr
147		struct ifaddrs *net, *net_ptr;
148		struct if_data *net_data;
149		#endif
150	+	#ifdef WIN32
151	+	PMIB_IFTABLE if_table;
152	+	MIB_IFROW if_row;
153	+	int i, no, j;
154
155	+	/* used for duplicate interface names. 5 for space, hash, up to two
156	+	* numbers and terminating slash */
157	+	char buf[5];
158	+	#endif
159	+
160		#ifdef ALLBSD
161		if(getifaddrs(&net) != 0){
162		sg_set_error_with_errno(SG_ERROR_GETIFADDRS, NULL);
#	Line 154 | Line 200 | sg_network_io_stats *sg_get_network_io_stats(int *entr
200		interfaces=0;
201
202		for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
203	<	if (!strcmp(ksp->ks_class, "net")) {
203	>	if (strcmp(ksp->ks_class, "net") == 0) {
204		kstat_read(kc, ksp, NULL);
205
206		#ifdef SOL7
#	Line 184 | Line 230 | sg_network_io_stats *sg_get_network_io_stats(int *entr
230
231		/* Create new network_stats */
232		if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
233	+	kstat_close(kc);
234		return NULL;
235		}
236		network_stat_ptr=network_stats+interfaces;
237
238	+	/* Read interface name */
239	+	if (sg_update_string(&network_stat_ptr->interface_name,
240	+	ksp->ks_name) < 0) {
241	+	kstat_close(kc);
242	+	return NULL;
243	+	}
244	+
245		/* Finish reading rx */
246		network_stat_ptr->rx=knp->VALTYPE;
247
#	Line 227 | Line 281 | sg_network_io_stats *sg_get_network_io_stats(int *entr
281		}
282		network_stat_ptr->collisions=knp->value.ui32;
283
230	–	/* Read interface name */
231	–	if (sg_update_string(&network_stat_ptr->interface_name,
232	–	ksp->ks_name) < 0) {
233	–	return NULL;
234	–	}
284
285		/* Store systime */
286		network_stat_ptr->systime=time(NULL);
#	Line 300 | Line 349 | sg_network_io_stats *sg_get_network_io_stats(int *entr
349		return NULL;
350		#endif
351
352	+	#ifdef WIN32
353	+	interfaces = 0;
354	+
355	+	if((if_table = win32_get_devices()) == NULL) {
356	+	sg_set_error(SG_ERROR_DEVICES, "network");
357	+	return NULL;
358	+	}
359	+
360	+	if(VECTOR_RESIZE(network_stats, if_table->dwNumEntries) < 0) {
361	+	free(if_table);
362	+	return NULL;
363	+	}
364	+
365	+	for (i=0; i<if_table->dwNumEntries; i++) {
366	+	network_stat_ptr=network_stats+i;
367	+	if_row = if_table->table[i];
368	+
369	+	if(sg_update_string(&network_stat_ptr->interface_name,
370	+	if_row.bDescr) < 0) {
371	+	free(if_table);
372	+	return NULL;
373	+	}
374	+	network_stat_ptr->tx = if_row.dwOutOctets;
375	+	network_stat_ptr->rx = if_row.dwInOctets;
376	+	network_stat_ptr->ipackets = if_row.dwInUcastPkts + if_row.dwInNUcastPkts;
377	+	network_stat_ptr->opackets = if_row.dwOutUcastPkts + if_row.dwOutNUcastPkts;
378	+	network_stat_ptr->ierrors = if_row.dwInErrors;
379	+	network_stat_ptr->oerrors = if_row.dwOutErrors;
380	+	network_stat_ptr->collisions = 0; /* can't do that */
381	+	network_stat_ptr->systime = time(NULL);
382	+
383	+	interfaces++;
384	+	}
385	+	free(if_table);
386	+
387	+	/* Please say there's a nicer way to do this...  If windows has two (or
388	+	* more) identical network cards, GetIfTable returns them with the same
389	+	* name, not like in Device Manager where the other has a #2 etc after
390	+	* it. So, add the #number here. Should we be doing this? Or should the
391	+	* end programs be dealing with duplicate names? Currently breaks
392	+	* watch.pl in rrdgraphing. But Unix does not have the issue of
393	+	* duplicate net device names.
394	+	*/
395	+	for (i=0; i<interfaces; i++) {
396	+	no = 2;
397	+	for(j=i+1; j<interfaces; j++) {
398	+	network_stat_ptr=network_stats+j;
399	+	if(strcmp(network_stats[i].interface_name,
400	+	network_stat_ptr->interface_name) == 0) {
401	+	if(snprintf(buf, sizeof(buf), " #%d", no) < 0) {
402	+	break;
403	+	}
404	+	if(sg_concat_string(&network_stat_ptr->interface_name, buf) != 0) {
405	+	return NULL;
406	+	}
407	+
408	+	no++;
409	+	}
410	+	}
411	+	}
412	+	#endif
413	+
414		*entries=interfaces;
415
416		return network_stats;
417		}
418
419		static long long transfer_diff(long long new, long long old){
420	<	#if defined(SOL7) || defined(LINUX) || defined(FREEBSD) || defined(DFBSD) || defined(OPENBSD)
420	>	#if defined(SOL7) || defined(LINUX) || defined(FREEBSD) || defined(DFBSD) || defined(OPENBSD) || defined(WIN32)
421		/* 32-bit quantities, so we must explicitly deal with wraparound. */
422		#define MAXVAL 0x100000000LL
423		if (new >= old) {
#	Line 439 | Line 550 | sg_network_iface_stats *sg_get_network_iface_stats(int
550		char line[8096];
551		int sock;
552		#endif
553	+	#ifdef WIN32
554	+	PMIB_IFTABLE if_table;
555	+	MIB_IFROW if_row;
556	+	int i,j,no;
557	+	char buf[5];
558	+	#endif
559
560		#ifdef ALLBSD
561		if(getifaddrs(&net) != 0){
#	Line 490 | Line 607 | sg_network_iface_stats *sg_get_network_iface_stats(int
607		continue;
608		}
609
610	<	/* Only intrested in the first 4 bits)  - Assuming only ETHER devices */
611	<	x = ifmed.ifm_active & 0x0f;
610	>	/* Assuming only ETHER devices */
611	>	x = IFM_SUBTYPE(ifmed.ifm_active);
612		switch(x){
613		/* 10 Mbit connections. Speedy :) */
614		case(IFM_10_T):
#	Line 501 | Line 618 | sg_network_iface_stats *sg_get_network_iface_stats(int
618		case(IFM_10_FL):
619		network_iface_stat_ptr->speed = 10;
620		break;
621	<	/* 100 Mbit conneections */
621	>	/* 100 Mbit connections */
622		case(IFM_100_TX):
623		case(IFM_100_FX):
624		case(IFM_100_T4):
#	Line 551 | Line 668 | sg_network_iface_stats *sg_get_network_iface_stats(int
668
669		if ((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP)) < 0) {
670		sg_set_error_with_errno(SG_ERROR_SOCKET, NULL);
671	+	kstat_close(kc);
672		return NULL;
673		}
674
675		for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
676	<	if (!strcmp(ksp->ks_class, "net")) {
676	>	if (strcmp(ksp->ks_class, "net") == 0) {
677		struct ifreq ifr;
678
679		kstat_read(kc, ksp, NULL);
#	Line 567 | Line 685 | sg_network_iface_stats *sg_get_network_iface_stats(int
685		}
686
687		if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
688	+	kstat_close(kc);
689		return NULL;
690		}
691		network_iface_stat_ptr = network_iface_stats + ifaces;
#	Line 574 | Line 693 | sg_network_iface_stats *sg_get_network_iface_stats(int
693
694		if (sg_update_string(&network_iface_stat_ptr->interface_name,
695		ksp->ks_name) < 0) {
696	+	kstat_close(kc);
697		return NULL;
698		}
699
700		if ((ifr.ifr_flags & IFF_UP) != 0) {
701	<	network_iface_stat_ptr->up = 1;
701	>	if ((knp = kstat_data_lookup(ksp, "link_up")) != NULL) {
702	>	/* take in to account if link
703	>	* is up as well as interface */
704	>	if (knp->value.ui32 != 0u) {
705	>	network_iface_stat_ptr->up = 1;
706	>	} else {
707	>	network_iface_stat_ptr->up = 0;
708	>	}
709	>	}
710	>	else {
711	>	/* maintain compatibility */
712	>	network_iface_stat_ptr->up = 1;
713	>	}
714		} else {
715	<	network_iface_stat_ptr->up = 1;
715	>	network_iface_stat_ptr->up = 0;
716		}
717
718		if ((knp = kstat_data_lookup(ksp, "ifspeed")) != NULL) {
#	Line 605 | Line 737 | sg_network_iface_stats *sg_get_network_iface_stats(int
737
738		close(sock);
739		kstat_close(kc);
740	<	#endif
740	>	#endif
741		#ifdef LINUX
742		f = fopen("/proc/net/dev", "r");
743		if(f == NULL){
#	Line 703 | Line 835 | sg_network_iface_stats *sg_get_network_iface_stats(int
835		#ifdef HPUX
836		sg_set_error(SG_ERROR_UNSUPPORTED, "HP-UX");
837		return NULL;
838	+	#endif
839	+	#ifdef WIN32
840	+	ifaces = 0;
841	+
842	+	if((if_table = win32_get_devices()) == NULL) {
843	+	sg_set_error(SG_ERROR_DEVICES, "network interfaces");
844	+	return NULL;
845	+	}
846	+
847	+	if(VECTOR_RESIZE(network_iface_stats, if_table->dwNumEntries) < 0) {
848	+	free(if_table);
849	+	return NULL;
850	+	}
851	+
852	+	for(i=0; i<if_table->dwNumEntries; i++) {
853	+	network_iface_stat_ptr=network_iface_stats+i;
854	+	if_row = if_table->table[i];
855	+
856	+	if(sg_update_string(&network_iface_stat_ptr->interface_name,
857	+	if_row.bDescr) < 0) {
858	+	free(if_table);
859	+	return NULL;
860	+	}
861	+	network_iface_stat_ptr->speed = if_row.dwSpeed /1000000;
862	+
863	+	if((if_row.dwOperStatus == MIB_IF_OPER_STATUS_CONNECTED ||
864	+	if_row.dwOperStatus ==
865	+	MIB_IF_OPER_STATUS_OPERATIONAL) &&
866	+	if_row.dwAdminStatus == 1) {
867	+	network_iface_stat_ptr->up = 1;
868	+	} else {
869	+	network_iface_stat_ptr->up = 0;
870	+	}
871	+
872	+	ifaces++;
873	+	}
874	+	free(if_table);
875	+
876	+	/* again with the renumbering */
877	+	for (i=0; i<ifaces; i++) {
878	+	no = 2;
879	+	for(j=i+1; j<ifaces; j++) {
880	+	network_iface_stat_ptr=network_iface_stats+j;
881	+	if(strcmp(network_iface_stats[i].interface_name,
882	+	network_iface_stat_ptr->interface_name) == 0) {
883	+	if(snprintf(buf, sizeof(buf), " #%d", no) < 0) {
884	+	break;
885	+	}
886	+	if(sg_concat_string(&network_iface_stat_ptr->interface_name, buf) != 0) {
887	+	return NULL;
888	+	}
889	+
890	+	no++;
891	+	}
892	+	}
893	+	}
894		#endif
895
896		#ifdef SG_ENABLE_DEPRECATED

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines