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.66 by tdb, Wed Apr 7 21:08:40 2004 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>
#	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(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(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
#	Line 295 | Line 301 | network_stat_t *get_network_stats(int *entries){
301		return network_stats;
302		}
303
304	<	long long transfer_diff(long long new, long long old){
304	>	static long long transfer_diff(long long new, long long old){
305		#if defined(SOL7) || defined(LINUX) || defined(FREEBSD) || defined(DFBSD) || defined(OPENBSD)
306		/* 32-bit quantities, so we must explicitly deal with wraparound. */
307		#define MAXVAL 0x100000000LL
#	Line 310 | Line 316 | long long transfer_diff(long long new, long long old){
316		#endif
317		}
318
319	<	network_stat_t *get_network_stats_diff(int *entries) {
320	<	VECTOR_DECLARE_STATIC(diff, network_stat_t, 1,
321	<	network_stat_init, network_stat_destroy);
322	<	network_stat_t *src = NULL, *dest;
319	>	sg_network_io_stats *sg_get_network_io_stats_diff(int *entries) {
320	>	VECTOR_DECLARE_STATIC(diff, sg_network_io_stats, 1,
321	>	network_stat_init, network_stat_destroy);
322	>	sg_network_io_stats *src = NULL, *dest;
323		int i, j, diff_count, new_count;
324
325		if (network_stats == NULL) {
326		/* No previous stats, so we can't calculate a difference. */
327	<	return get_network_stats(entries);
327	>	return sg_get_network_io_stats(entries);
328		}
329
330		/* Resize the results array to match the previous stats. */
#	Line 332 | Line 338 | network_stat_t *get_network_stats_diff(int *entries) {
338		src = &network_stats[i];
339		dest = &diff[i];
340
341	<	if (update_string(&dest->interface_name,
342	<	src->interface_name) == NULL) {
341	>	if (sg_update_string(&dest->interface_name,
342	>	src->interface_name) < 0) {
343		return NULL;
344		}
345		dest->rx = src->rx;
#	Line 347 | Line 353 | network_stat_t *get_network_stats_diff(int *entries) {
353		}
354
355		/* Get a new set of stats. */
356	<	if (get_network_stats(&new_count) == NULL) {
356	>	if (sg_get_network_io_stats(&new_count) == NULL) {
357		return NULL;
358		}
359
#	Line 387 | Line 393 | network_stat_t *get_network_stats_diff(int *entries) {
393
394		/* NETWORK INTERFACE STATS */
395
396	<	static void network_iface_stat_init(network_iface_stat_t *s) {
396	>	static void network_iface_stat_init(sg_network_iface_stats *s) {
397		s->interface_name = NULL;
398		s->speed = 0;
399	<	s->dup = UNKNOWN_DUPLEX;
399	>	s->dup = SG_IFACE_DUPLEX_UNKNOWN;
400		}
401
402	<	static void network_iface_stat_destroy(network_iface_stat_t *s) {
402	>	static void network_iface_stat_destroy(sg_network_iface_stats *s) {
403		free(s->interface_name);
404		}
405
406	<	network_iface_stat_t *get_network_iface_stats(int *entries){
407	<	VECTOR_DECLARE_STATIC(network_iface_stats, network_iface_stat_t, 5,
408	<	network_iface_stat_init, network_iface_stat_destroy);
409	<	network_iface_stat_t *network_iface_stat_ptr;
406	>	sg_network_iface_stats *sg_get_network_iface_stats(int *entries){
407	>	VECTOR_DECLARE_STATIC(network_iface_stats, sg_network_iface_stats, 5,
408	>	network_iface_stat_init, network_iface_stat_destroy);
409	>	sg_network_iface_stats *network_iface_stat_ptr;
410		int ifaces = 0;
411
412		#ifdef SOLARIS
413	<	kstat_ctl_t *kc;
414	<	kstat_t *ksp;
413	>	kstat_ctl_t *kc;
414	>	kstat_t *ksp;
415		kstat_named_t *knp;
416		int sock;
417		#endif
418		#ifdef ALLBSD
419	<	struct ifaddrs *net, *net_ptr;
419	>	struct ifaddrs *net, *net_ptr;
420		struct ifmediareq ifmed;
421		struct ifreq ifr;
422		int sock;
423		int x;
424		#endif
425		#ifdef LINUX
426	<	FILE *f;
427	<	/* Horrible big enough, but it should be easily big enough */
428	<	char line[8096];
426	>	FILE *f;
427	>	/* Horrible big enough, but it should be easily big enough */
428	>	char line[8096];
429		int sock;
430		#endif
431
432		#ifdef ALLBSD
433	<	if(getifaddrs(&net) != 0){
434	<	return NULL;
435	<	}
433	>	if(getifaddrs(&net) != 0){
434	>	sg_set_error(SG_ERROR_GETIFADDRS, NULL);
435	>	return NULL;
436	>	}
437
438		if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == 0) return NULL;
439
440		for(net_ptr=net; net_ptr!=NULL; net_ptr=net_ptr->ifa_next){
441	<	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
441	>	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
442
443		if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
444	<	return NULL;
445	<	}
446	<	network_iface_stat_ptr = network_iface_stats + ifaces;
444	>	return NULL;
445	>	}
446	>	network_iface_stat_ptr = network_iface_stats + ifaces;
447
448		memset(&ifr, 0, sizeof(ifr));
449		strncpy(ifr.ifr_name, net_ptr->ifa_name, sizeof(ifr.ifr_name));
#	Line 450 | Line 457 | network_iface_stat_t *get_network_iface_stats(int *ent
457		network_iface_stat_ptr->up = 0;
458		}
459
460	<	if (update_string(&network_iface_stat_ptr->interface_name,
461	<	net_ptr->ifa_name) == NULL) {
460	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
461	>	net_ptr->ifa_name) < 0) {
462		return NULL;
463		}
464
465		network_iface_stat_ptr->speed = 0;
466	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
466	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
467		ifaces++;
468
469		memset(&ifmed, 0, sizeof(struct ifmediareq));
470	<	strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
470	>	sg_strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
471		if(ioctl(sock, SIOCGIFMEDIA, (caddr_t)&ifmed) == -1){
472		/* Not all interfaces support the media ioctls. */
473		continue;
#	Line 513 | Line 520 | network_iface_stat_t *get_network_iface_stats(int *ent
520		}
521
522		if( (ifmed.ifm_active | IFM_FDX) == ifmed.ifm_active ){
523	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
523	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
524		}else if( (ifmed.ifm_active | IFM_HDX) == ifmed.ifm_active ){
525	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
525	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
526		}else{
527	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
527	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
528		}
529
530		}
#	Line 527 | Line 534 | network_iface_stat_t *get_network_iface_stats(int *ent
534
535		#ifdef SOLARIS
536		if ((kc = kstat_open()) == NULL) {
537	+	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
538		return NULL;
539		}
540
541		if ((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP)) < 0) {
542	+	sg_set_error(SG_ERROR_SOCKET, NULL);
543		return NULL;
544		}
545
#	Line 552 | Line 561 | network_iface_stat_t *get_network_iface_stats(int *ent
561		network_iface_stat_ptr = network_iface_stats + ifaces;
562		ifaces++;
563
564	<	if (update_string(&network_iface_stat_ptr->interface_name,
565	<	ksp->ks_name) == NULL) {
564	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
565	>	ksp->ks_name) < 0) {
566		return NULL;
567		}
568
#	Line 569 | Line 578 | network_iface_stat_t *get_network_iface_stats(int *ent
578		network_iface_stat_ptr->speed = 0;
579		}
580
581	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
581	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
582		if ((knp = kstat_data_lookup(ksp, "link_duplex")) != NULL) {
583		switch (knp->value.ui32) {
584		case 1:
585	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
585	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
586		break;
587		case 2:
588	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
588	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
589		break;
590		}
591		}
#	Line 588 | Line 597 | network_iface_stat_t *get_network_iface_stats(int *ent
597		#endif
598		#ifdef LINUX
599		f = fopen("/proc/net/dev", "r");
600	<	if(f == NULL){
601	<	return NULL;
602	<	}
600	>	if(f == NULL){
601	>	sg_set_error(SG_ERROR_OPEN, "/proc/net/dev");
602	>	return NULL;
603	>	}
604
605		/* Setup stuff so we can do the ioctl to get the info */
606		if((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0){
607	+	sg_set_error(SG_ERROR_SOCKET, NULL);
608		return NULL;
609		}
610
611		/* Ignore first 2 lines.. Just headings */
612	<	if((fgets(line, sizeof(line), f)) == NULL) return NULL;
613	<	if((fgets(line, sizeof(line), f)) == NULL) return NULL;
612	>	if((fgets(line, sizeof(line), f)) == NULL) {
613	>	sg_set_error(SG_ERROR_PARSE, NULL);
614	>	return NULL;
615	>	}
616	>	if((fgets(line, sizeof(line), f)) == NULL) {
617	>	sg_set_error(SG_ERROR_PARSE, NULL);
618	>	return NULL;
619	>	}
620
621	<	while((fgets(line, sizeof(line), f)) != NULL){
622	<	char *name, *ptr;
623	<	struct ifreq ifr;
624	<	struct ethtool_cmd ethcmd;
625	<	int err;
621	>	while((fgets(line, sizeof(line), f)) != NULL){
622	>	char *name, *ptr;
623	>	struct ifreq ifr;
624	>	struct ethtool_cmd ethcmd;
625	>	int err;
626
627		/* Get the interface name */
628	<	ptr = strchr(line, ':');
629	<	if (ptr == NULL) continue;
630	<	*ptr='\0';
631	<	name = line;
632	<	while(isspace(*(name))){
633	<	name++;
634	<	}
628	>	ptr = strchr(line, ':');
629	>	if (ptr == NULL) continue;
630	>	*ptr='\0';
631	>	name = line;
632	>	while(isspace(*(name))){
633	>	name++;
634	>	}
635
636	<	memset(&ifr, 0, sizeof ifr);
637	<	strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
636	>	memset(&ifr, 0, sizeof ifr);
637	>	strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
638
639		if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
640		continue;
#	Line 629 | Line 646 | network_iface_stat_t *get_network_iface_stats(int *ent
646		}
647		network_iface_stat_ptr = network_iface_stats + ifaces;
648
649	<	if (update_string(&network_iface_stat_ptr->interface_name,
650	<	name) == NULL) {
649	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
650	>	name) < 0) {
651		return NULL;
652		}
653		if ((ifr.ifr_flags & IFF_UP) != 0) {
#	Line 639 | Line 656 | network_iface_stat_t *get_network_iface_stats(int *ent
656		network_iface_stat_ptr->up = 0;
657		}
658
659	<	memset(&ethcmd, 0, sizeof ethcmd);
660	<	ethcmd.cmd = ETHTOOL_GSET;
661	<	ifr.ifr_data = (caddr_t) &ethcmd;
659	>	memset(&ethcmd, 0, sizeof ethcmd);
660	>	ethcmd.cmd = ETHTOOL_GSET;
661	>	ifr.ifr_data = (caddr_t) &ethcmd;
662
663	<	err = ioctl(sock, SIOCETHTOOL, &ifr);
664	<	if (err == 0) {
663	>	err = ioctl(sock, SIOCETHTOOL, &ifr);
664	>	if (err == 0) {
665		network_iface_stat_ptr->speed = ethcmd.speed;
666
667		switch (ethcmd.duplex) {
668		case 0x00:
669	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
669	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
670		break;
671		case 0x01:
672	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
672	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
673		break;
674		default:
675	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
675	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
676		}
677		} else {
678		/* Not all interfaces support the ethtool ioctl. */
679		network_iface_stat_ptr->speed = 0;
680	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
680	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
681		}
682
683		ifaces++;

Diff Legend

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