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.60 by ats, Mon Apr 5 00:16:24 2004 UTC vs.
Revision 1.75 by ats, Sat Jul 30 13:23:46 2005 UTC

#	Line 1 | Line 1
1		/*
2	<	* i-scream central monitoring system
2	>	* i-scream libstatgrab
3		* http://www.i-scream.org
4		* Copyright (C) 2000-2004 i-scream
5		*
#	Line 40 | Line 40
40		#include <net/if.h>
41		#include <netinet/in.h>
42		#include <sys/sockio.h>
43	+	#include <unistd.h>
44		#endif
45		#ifdef LINUX
46		#include <stdio.h>
47		#include <sys/types.h>
47	–	#include <regex.h>
48		#include <sys/ioctl.h>
49		#include <sys/socket.h>
50		#include <net/if.h>
51		#include <ctype.h>
52	<	/* Stuff which could be defined by defining KERNEL, but
53	<	* that would be a bad idea, so we'll just declare it here
54	<	*/
52	>	#include <linux/version.h>
53	>	#include <asm/types.h>
54	>	/* These aren't defined by asm/types.h unless the kernel is being
55	>	compiled, but some versions of ethtool.h need them. */
56		typedef __uint8_t u8;
57		typedef __uint16_t u16;
58		typedef __uint32_t u32;
#	Line 70 | Line 71 | typedef __uint64_t u64;
71		#include <unistd.h>
72		#endif
73
74	<	static void network_stat_init(network_stat_t *s) {
74	>	static void network_stat_init(sg_network_io_stats *s) {
75		s->interface_name = NULL;
76		s->tx = 0;
77		s->rx = 0;
#	Line 81 | Line 82 | static void network_stat_init(network_stat_t *s) {
82		s->collisions = 0;
83		}
84
85	<	static void network_stat_destroy(network_stat_t *s) {
85	>	static void network_stat_destroy(sg_network_io_stats *s) {
86		free(s->interface_name);
87		}
88
89	<	VECTOR_DECLARE_STATIC(network_stats, network_stat_t, 5,
90	<	network_stat_init, network_stat_destroy);
89	>	VECTOR_DECLARE_STATIC(network_stats, sg_network_io_stats, 5,
90	>	network_stat_init, network_stat_destroy);
91
92	<	network_stat_t *get_network_stats(int *entries){
92	>	sg_network_io_stats *sg_get_network_io_stats(int *entries){
93		int interfaces;
94	<	network_stat_t *network_stat_ptr;
94	>	sg_network_io_stats *network_stat_ptr;
95
96		#ifdef SOLARIS
97	<	kstat_ctl_t *kc;
98	<	kstat_t *ksp;
97	>	kstat_ctl_t *kc;
98	>	kstat_t *ksp;
99		kstat_named_t *knp;
100		#endif
101
#	Line 112 | Line 113 | network_stat_t *get_network_stats(int *entries){
113
114		#ifdef ALLBSD
115		if(getifaddrs(&net) != 0){
116	+	sg_set_error_with_errno(SG_ERROR_GETIFADDRS, NULL);
117		return NULL;
118		}
119
#	Line 125 | Line 127 | network_stat_t *get_network_stats(int *entries){
127		}
128		network_stat_ptr=network_stats+interfaces;
129
130	<	if (update_string(&network_stat_ptr->interface_name,
131	<	net_ptr->ifa_name) == NULL) {
130	>	if (sg_update_string(&network_stat_ptr->interface_name,
131	>	net_ptr->ifa_name) < 0) {
132		return NULL;
133		}
134		net_data=(struct if_data *)net_ptr->ifa_data;
#	Line 144 | Line 146 | network_stat_t *get_network_stats(int *entries){
146		#endif
147
148		#ifdef SOLARIS
149	<	if ((kc = kstat_open()) == NULL) {
150	<	return NULL;
151	<	}
149	>	if ((kc = kstat_open()) == NULL) {
150	>	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
151	>	return NULL;
152	>	}
153
154		interfaces=0;
155
156	<	for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
157	<	if (!strcmp(ksp->ks_class, "net")) {
158	<	kstat_read(kc, ksp, NULL);
156	>	for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
157	>	if (!strcmp(ksp->ks_class, "net")) {
158	>	kstat_read(kc, ksp, NULL);
159
160		#ifdef SOL7
161		#define LRX "rbytes"
#	Line 225 | Line 228 | network_stat_t *get_network_stats(int *entries){
228		network_stat_ptr->collisions=knp->value.ui32;
229
230		/* Read interface name */
231	<	if (update_string(&network_stat_ptr->interface_name,
232	<	ksp->ks_name) == NULL) {
231	>	if (sg_update_string(&network_stat_ptr->interface_name,
232	>	ksp->ks_name) < 0) {
233		return NULL;
234		}
235
#	Line 242 | Line 245 | network_stat_t *get_network_stats(int *entries){
245		#ifdef LINUX
246		f=fopen("/proc/net/dev", "r");
247		if(f==NULL){
248	+	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/net/dev");
249		return NULL;
250		}
251		/* read the 2 lines.. Its the title, so we dont care :) */
#	Line 250 | Line 254 | network_stat_t *get_network_stats(int *entries){
254
255
256		if((regcomp(&regex, "^ *([^:]+): *([0-9]+) +([0-9]+) +([0-9]+) +[0-9]+ +[0-9]+ +[0-9]+ +[0-9]+ +[0-9]+ +([0-9]+) +([0-9]+) +([0-9]+) +[0-9]+ +[0-9]+ +([0-9]+)", REG_EXTENDED))!=0){
257	+	sg_set_error(SG_ERROR_PARSE, NULL);
258		return NULL;
259		}
260
#	Line 263 | Line 268 | network_stat_t *get_network_stats(int *entries){
268		if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
269		return NULL;
270		}
271	<	network_stat_ptr=network_stats+interfaces;
271	>	network_stat_ptr=network_stats+interfaces;
272
273		if(network_stat_ptr->interface_name!=NULL){
274		free(network_stat_ptr->interface_name);
275		}
276
277	<	network_stat_ptr->interface_name=get_string_match(line, &line_match[1]);
278	<	network_stat_ptr->rx=get_ll_match(line, &line_match[2]);
279	<	network_stat_ptr->tx=get_ll_match(line, &line_match[5]);
280	<	network_stat_ptr->ipackets=get_ll_match(line, &line_match[3]);
281	<	network_stat_ptr->opackets=get_ll_match(line, &line_match[6]);
282	<	network_stat_ptr->ierrors=get_ll_match(line, &line_match[4]);
283	<	network_stat_ptr->oerrors=get_ll_match(line, &line_match[7]);
284	<	network_stat_ptr->collisions=get_ll_match(line, &line_match[8]);
277	>	network_stat_ptr->interface_name=sg_get_string_match(line, &line_match[1]);
278	>	network_stat_ptr->rx=sg_get_ll_match(line, &line_match[2]);
279	>	network_stat_ptr->tx=sg_get_ll_match(line, &line_match[5]);
280	>	network_stat_ptr->ipackets=sg_get_ll_match(line, &line_match[3]);
281	>	network_stat_ptr->opackets=sg_get_ll_match(line, &line_match[6]);
282	>	network_stat_ptr->ierrors=sg_get_ll_match(line, &line_match[4]);
283	>	network_stat_ptr->oerrors=sg_get_ll_match(line, &line_match[7]);
284	>	network_stat_ptr->collisions=sg_get_ll_match(line, &line_match[8]);
285		network_stat_ptr->systime=time(NULL);
286
287		interfaces++;
#	Line 287 | Line 292 | network_stat_t *get_network_stats(int *entries){
292		#endif
293
294		#ifdef CYGWIN
295	+	sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
296		return NULL;
297		#endif
298	+	#ifdef HPUX
299	+	sg_set_error(SG_ERROR_UNSUPPORTED, "HP-UX");
300	+	return NULL;
301	+	#endif
302
303		*entries=interfaces;
304
305		return network_stats;
306		}
307
308	<	long long transfer_diff(long long new, long long old){
308	>	static long long transfer_diff(long long new, long long old){
309		#if defined(SOL7) || defined(LINUX) || defined(FREEBSD) || defined(DFBSD) || defined(OPENBSD)
310		/* 32-bit quantities, so we must explicitly deal with wraparound. */
311		#define MAXVAL 0x100000000LL
#	Line 310 | Line 320 | long long transfer_diff(long long new, long long old){
320		#endif
321		}
322
323	<	network_stat_t *get_network_stats_diff(int *entries) {
324	<	VECTOR_DECLARE_STATIC(diff, network_stat_t, 1,
325	<	network_stat_init, network_stat_destroy);
326	<	network_stat_t *src = NULL, *dest;
323	>	sg_network_io_stats *sg_get_network_io_stats_diff(int *entries) {
324	>	VECTOR_DECLARE_STATIC(diff, sg_network_io_stats, 1,
325	>	network_stat_init, network_stat_destroy);
326	>	sg_network_io_stats *src = NULL, *dest;
327		int i, j, diff_count, new_count;
328
329		if (network_stats == NULL) {
330		/* No previous stats, so we can't calculate a difference. */
331	<	return get_network_stats(entries);
331	>	return sg_get_network_io_stats(entries);
332		}
333
334		/* Resize the results array to match the previous stats. */
#	Line 332 | Line 342 | network_stat_t *get_network_stats_diff(int *entries) {
342		src = &network_stats[i];
343		dest = &diff[i];
344
345	<	if (update_string(&dest->interface_name,
346	<	src->interface_name) == NULL) {
345	>	if (sg_update_string(&dest->interface_name,
346	>	src->interface_name) < 0) {
347		return NULL;
348		}
349		dest->rx = src->rx;
#	Line 347 | Line 357 | network_stat_t *get_network_stats_diff(int *entries) {
357		}
358
359		/* Get a new set of stats. */
360	<	if (get_network_stats(&new_count) == NULL) {
360	>	if (sg_get_network_io_stats(&new_count) == NULL) {
361		return NULL;
362		}
363
#	Line 385 | Line 395 | network_stat_t *get_network_stats_diff(int *entries) {
395		return diff;
396		}
397
398	+	int sg_network_io_compare_name(const void *va, const void *vb) {
399	+	const sg_network_io_stats *a = (const sg_network_io_stats *)va;
400	+	const sg_network_io_stats *b = (const sg_network_io_stats *)vb;
401	+
402	+	return strcmp(a->interface_name, b->interface_name);
403	+	}
404	+
405		/* NETWORK INTERFACE STATS */
406
407	<	static void network_iface_stat_init(network_iface_stat_t *s) {
407	>	static void network_iface_stat_init(sg_network_iface_stats *s) {
408		s->interface_name = NULL;
409		s->speed = 0;
410	<	s->dup = UNKNOWN_DUPLEX;
410	>	s->duplex = SG_IFACE_DUPLEX_UNKNOWN;
411		}
412
413	<	static void network_iface_stat_destroy(network_iface_stat_t *s) {
413	>	static void network_iface_stat_destroy(sg_network_iface_stats *s) {
414		free(s->interface_name);
415		}
416
417	<	network_iface_stat_t *get_network_iface_stats(int *entries){
418	<	VECTOR_DECLARE_STATIC(network_iface_stats, network_iface_stat_t, 5,
419	<	network_iface_stat_init, network_iface_stat_destroy);
420	<	network_iface_stat_t *network_iface_stat_ptr;
417	>	sg_network_iface_stats *sg_get_network_iface_stats(int *entries){
418	>	VECTOR_DECLARE_STATIC(network_iface_stats, sg_network_iface_stats, 5,
419	>	network_iface_stat_init, network_iface_stat_destroy);
420	>	sg_network_iface_stats *network_iface_stat_ptr;
421		int ifaces = 0;
422
423		#ifdef SOLARIS
424	<	kstat_ctl_t *kc;
425	<	kstat_t *ksp;
424	>	kstat_ctl_t *kc;
425	>	kstat_t *ksp;
426		kstat_named_t *knp;
427		int sock;
428		#endif
429		#ifdef ALLBSD
430	<	struct ifaddrs *net, *net_ptr;
430	>	struct ifaddrs *net, *net_ptr;
431		struct ifmediareq ifmed;
432		struct ifreq ifr;
433		int sock;
434		int x;
435		#endif
436		#ifdef LINUX
437	<	FILE *f;
438	<	/* Horrible big enough, but it should be easily big enough */
439	<	char line[8096];
437	>	FILE *f;
438	>	/* Horrible big enough, but it should be easily big enough */
439	>	char line[8096];
440		int sock;
441		#endif
442
443		#ifdef ALLBSD
444	<	if(getifaddrs(&net) != 0){
445	<	return NULL;
446	<	}
444	>	if(getifaddrs(&net) != 0){
445	>	sg_set_error_with_errno(SG_ERROR_GETIFADDRS, NULL);
446	>	return NULL;
447	>	}
448
449		if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == 0) return NULL;
450
451		for(net_ptr=net; net_ptr!=NULL; net_ptr=net_ptr->ifa_next){
452	<	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
452	>	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
453
454		if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
455	<	return NULL;
456	<	}
457	<	network_iface_stat_ptr = network_iface_stats + ifaces;
455	>	return NULL;
456	>	}
457	>	network_iface_stat_ptr = network_iface_stats + ifaces;
458
459		memset(&ifr, 0, sizeof(ifr));
460		strncpy(ifr.ifr_name, net_ptr->ifa_name, sizeof(ifr.ifr_name));
#	Line 450 | Line 468 | network_iface_stat_t *get_network_iface_stats(int *ent
468		network_iface_stat_ptr->up = 0;
469		}
470
471	<	if (update_string(&network_iface_stat_ptr->interface_name,
472	<	net_ptr->ifa_name) == NULL) {
471	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
472	>	net_ptr->ifa_name) < 0) {
473		return NULL;
474		}
475
476		network_iface_stat_ptr->speed = 0;
477	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
477	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
478		ifaces++;
479
480		memset(&ifmed, 0, sizeof(struct ifmediareq));
481	<	strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
481	>	sg_strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
482		if(ioctl(sock, SIOCGIFMEDIA, (caddr_t)&ifmed) == -1){
483		/* Not all interfaces support the media ioctls. */
484		continue;
#	Line 513 | Line 531 | network_iface_stat_t *get_network_iface_stats(int *ent
531		}
532
533		if( (ifmed.ifm_active | IFM_FDX) == ifmed.ifm_active ){
534	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
534	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_FULL;
535		}else if( (ifmed.ifm_active | IFM_HDX) == ifmed.ifm_active ){
536	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
536	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_HALF;
537		}else{
538	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
538	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
539		}
540
541		}
#	Line 527 | Line 545 | network_iface_stat_t *get_network_iface_stats(int *ent
545
546		#ifdef SOLARIS
547		if ((kc = kstat_open()) == NULL) {
548	+	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
549		return NULL;
550		}
551
552		if ((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP)) < 0) {
553	+	sg_set_error_with_errno(SG_ERROR_SOCKET, NULL);
554		return NULL;
555		}
556
#	Line 552 | Line 572 | network_iface_stat_t *get_network_iface_stats(int *ent
572		network_iface_stat_ptr = network_iface_stats + ifaces;
573		ifaces++;
574
575	<	if (update_string(&network_iface_stat_ptr->interface_name,
576	<	ksp->ks_name) == NULL) {
575	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
576	>	ksp->ks_name) < 0) {
577		return NULL;
578		}
579
#	Line 569 | Line 589 | network_iface_stat_t *get_network_iface_stats(int *ent
589		network_iface_stat_ptr->speed = 0;
590		}
591
592	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
592	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
593		if ((knp = kstat_data_lookup(ksp, "link_duplex")) != NULL) {
594		switch (knp->value.ui32) {
595		case 1:
596	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
596	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_HALF;
597		break;
598		case 2:
599	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
599	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_FULL;
600		break;
601		}
602		}
#	Line 588 | Line 608 | network_iface_stat_t *get_network_iface_stats(int *ent
608		#endif
609		#ifdef LINUX
610		f = fopen("/proc/net/dev", "r");
611	<	if(f == NULL){
612	<	return NULL;
613	<	}
611	>	if(f == NULL){
612	>	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/net/dev");
613	>	return NULL;
614	>	}
615
616		/* Setup stuff so we can do the ioctl to get the info */
617		if((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0){
618	+	sg_set_error_with_errno(SG_ERROR_SOCKET, NULL);
619		return NULL;
620		}
621
622		/* Ignore first 2 lines.. Just headings */
623	<	if((fgets(line, sizeof(line), f)) == NULL) return NULL;
624	<	if((fgets(line, sizeof(line), f)) == NULL) return NULL;
623	>	if((fgets(line, sizeof(line), f)) == NULL) {
624	>	sg_set_error(SG_ERROR_PARSE, NULL);
625	>	return NULL;
626	>	}
627	>	if((fgets(line, sizeof(line), f)) == NULL) {
628	>	sg_set_error(SG_ERROR_PARSE, NULL);
629	>	return NULL;
630	>	}
631
632	<	while((fgets(line, sizeof(line), f)) != NULL){
633	<	char *name, *ptr;
634	<	struct ifreq ifr;
635	<	struct ethtool_cmd ethcmd;
636	<	int err;
632	>	while((fgets(line, sizeof(line), f)) != NULL){
633	>	char *name, *ptr;
634	>	struct ifreq ifr;
635	>	struct ethtool_cmd ethcmd;
636	>	int err;
637
638		/* Get the interface name */
639	<	ptr = strchr(line, ':');
640	<	if (ptr == NULL) continue;
641	<	*ptr='\0';
642	<	name = line;
643	<	while(isspace(*(name))){
644	<	name++;
645	<	}
639	>	ptr = strchr(line, ':');
640	>	if (ptr == NULL) continue;
641	>	*ptr='\0';
642	>	name = line;
643	>	while(isspace(*(name))){
644	>	name++;
645	>	}
646
647	<	memset(&ifr, 0, sizeof ifr);
648	<	strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
647	>	memset(&ifr, 0, sizeof ifr);
648	>	strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
649
650		if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
651		continue;
#	Line 629 | Line 657 | network_iface_stat_t *get_network_iface_stats(int *ent
657		}
658		network_iface_stat_ptr = network_iface_stats + ifaces;
659
660	<	if (update_string(&network_iface_stat_ptr->interface_name,
661	<	name) == NULL) {
660	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
661	>	name) < 0) {
662		return NULL;
663		}
664		if ((ifr.ifr_flags & IFF_UP) != 0) {
#	Line 639 | Line 667 | network_iface_stat_t *get_network_iface_stats(int *ent
667		network_iface_stat_ptr->up = 0;
668		}
669
670	<	memset(&ethcmd, 0, sizeof ethcmd);
671	<	ethcmd.cmd = ETHTOOL_GSET;
672	<	ifr.ifr_data = (caddr_t) &ethcmd;
670	>	memset(&ethcmd, 0, sizeof ethcmd);
671	>	ethcmd.cmd = ETHTOOL_GSET;
672	>	ifr.ifr_data = (caddr_t) &ethcmd;
673
674	<	err = ioctl(sock, SIOCETHTOOL, &ifr);
675	<	if (err == 0) {
674	>	err = ioctl(sock, SIOCETHTOOL, &ifr);
675	>	if (err == 0) {
676		network_iface_stat_ptr->speed = ethcmd.speed;
677
678		switch (ethcmd.duplex) {
679	<	case 0x00:
680	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
679	>	case DUPLEX_FULL:
680	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_FULL;
681		break;
682	<	case 0x01:
683	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
682	>	case DUPLEX_HALF:
683	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_HALF;
684		break;
685		default:
686	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
686	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
687		}
688		} else {
689		/* Not all interfaces support the ethtool ioctl. */
690		network_iface_stat_ptr->speed = 0;
691	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
691	>	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
692		}
693
694		ifaces++;
#	Line 668 | Line 696 | network_iface_stat_t *get_network_iface_stats(int *ent
696		close(sock);
697		fclose(f);
698		#endif
699	+	#ifdef CYGWIN
700	+	sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
701	+	return NULL;
702	+	#endif
703	+	#ifdef HPUX
704	+	sg_set_error(SG_ERROR_UNSUPPORTED, "HP-UX");
705	+	return NULL;
706	+	#endif
707	+
708	+	#ifdef SG_ENABLE_DEPRECATED
709	+	network_iface_stat_ptr->dup = network_iface_stat_ptr->duplex;
710	+	#endif
711	+
712		*entries = ifaces;
713		return network_iface_stats;
714	+	}
715	+
716	+	int sg_network_iface_compare_name(const void *va, const void *vb) {
717	+	const sg_network_iface_stats *a = (const sg_network_iface_stats *)va;
718	+	const sg_network_iface_stats *b = (const sg_network_iface_stats *)vb;
719	+
720	+	return strcmp(a->interface_name, b->interface_name);
721		}
722

Diff Legend

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