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.59 by ats, Sun Apr 4 22:52:16 2004 UTC vs.
Revision 1.70 by ats, Sun Nov 7 12:30:54 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 29 | Line 29
29		#include <string.h>
30		#include "statgrab.h"
31		#include "vector.h"
32	+	#include "tools.h"
33		#include <time.h>
34		#ifdef SOLARIS
35		#include <kstat.h>
#	Line 39 | 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>
46	–	#include <regex.h>
48		#include <sys/ioctl.h>
49		#include <sys/socket.h>
50		#include <net/if.h>
51		#include <ctype.h>
51	–	#include "tools.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		*/
#	Line 70 | Line 70 | typedef __uint64_t u64;
70		#include <unistd.h>
71		#endif
72
73	<	static void network_stat_init(network_stat_t *s) {
73	>	static void network_stat_init(sg_network_io_stats *s) {
74		s->interface_name = NULL;
75		s->tx = 0;
76		s->rx = 0;
#	Line 81 | Line 81 | static void network_stat_init(network_stat_t *s) {
81		s->collisions = 0;
82		}
83
84	<	static void network_stat_destroy(network_stat_t *s) {
84	>	static void network_stat_destroy(sg_network_io_stats *s) {
85		free(s->interface_name);
86		}
87
88	<	VECTOR_DECLARE_STATIC(network_stats, network_stat_t, 5,
89	<	network_stat_init, network_stat_destroy);
88	>	VECTOR_DECLARE_STATIC(network_stats, sg_network_io_stats, 5,
89	>	network_stat_init, network_stat_destroy);
90
91	<	network_stat_t *get_network_stats(int *entries){
91	>	sg_network_io_stats *sg_get_network_io_stats(int *entries){
92		int interfaces;
93	<	network_stat_t *network_stat_ptr;
93	>	sg_network_io_stats *network_stat_ptr;
94
95		#ifdef SOLARIS
96	<	kstat_ctl_t *kc;
97	<	kstat_t *ksp;
96	>	kstat_ctl_t *kc;
97	>	kstat_t *ksp;
98		kstat_named_t *knp;
99		#endif
100
#	Line 112 | Line 112 | network_stat_t *get_network_stats(int *entries){
112
113		#ifdef ALLBSD
114		if(getifaddrs(&net) != 0){
115	+	sg_set_error_with_errno(SG_ERROR_GETIFADDRS, NULL);
116		return NULL;
117		}
118
#	Line 125 | Line 126 | network_stat_t *get_network_stats(int *entries){
126		}
127		network_stat_ptr=network_stats+interfaces;
128
129	<	if(network_stat_ptr->interface_name!=NULL) free(network_stat_ptr->interface_name);
130	<	network_stat_ptr->interface_name=strdup(net_ptr->ifa_name);
131	<	if(network_stat_ptr->interface_name==NULL) return NULL;
129	>	if (sg_update_string(&network_stat_ptr->interface_name,
130	>	net_ptr->ifa_name) < 0) {
131	>	return NULL;
132	>	}
133		net_data=(struct if_data *)net_ptr->ifa_data;
134		network_stat_ptr->rx=net_data->ifi_ibytes;
135		network_stat_ptr->tx=net_data->ifi_obytes;
#	Line 143 | Line 145 | network_stat_t *get_network_stats(int *entries){
145		#endif
146
147		#ifdef SOLARIS
148	<	if ((kc = kstat_open()) == NULL) {
149	<	return NULL;
150	<	}
148	>	if ((kc = kstat_open()) == NULL) {
149	>	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
150	>	return NULL;
151	>	}
152
153		interfaces=0;
154
155	<	for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
156	<	if (!strcmp(ksp->ks_class, "net")) {
157	<	kstat_read(kc, ksp, NULL);
155	>	for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
156	>	if (!strcmp(ksp->ks_class, "net")) {
157	>	kstat_read(kc, ksp, NULL);
158
159		#ifdef SOL7
160		#define LRX "rbytes"
#	Line 224 | Line 227 | network_stat_t *get_network_stats(int *entries){
227		network_stat_ptr->collisions=knp->value.ui32;
228
229		/* Read interface name */
230	<	if(network_stat_ptr->interface_name!=NULL){
231	<	free(network_stat_ptr->interface_name);
230	>	if (sg_update_string(&network_stat_ptr->interface_name,
231	>	ksp->ks_name) < 0) {
232	>	return NULL;
233		}
230	–	network_stat_ptr->interface_name=strdup(ksp->ks_name);
234
235		/* Store systime */
236		network_stat_ptr->systime=time(NULL);
#	Line 241 | Line 244 | network_stat_t *get_network_stats(int *entries){
244		#ifdef LINUX
245		f=fopen("/proc/net/dev", "r");
246		if(f==NULL){
247	+	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/net/dev");
248		return NULL;
249		}
250		/* read the 2 lines.. Its the title, so we dont care :) */
#	Line 249 | Line 253 | network_stat_t *get_network_stats(int *entries){
253
254
255		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){
256	+	sg_set_error(SG_ERROR_PARSE, NULL);
257		return NULL;
258		}
259
#	Line 262 | Line 267 | network_stat_t *get_network_stats(int *entries){
267		if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
268		return NULL;
269		}
270	<	network_stat_ptr=network_stats+interfaces;
270	>	network_stat_ptr=network_stats+interfaces;
271
272		if(network_stat_ptr->interface_name!=NULL){
273		free(network_stat_ptr->interface_name);
274		}
275
276	<	network_stat_ptr->interface_name=get_string_match(line, &line_match[1]);
277	<	network_stat_ptr->rx=get_ll_match(line, &line_match[2]);
278	<	network_stat_ptr->tx=get_ll_match(line, &line_match[5]);
279	<	network_stat_ptr->ipackets=get_ll_match(line, &line_match[3]);
280	<	network_stat_ptr->opackets=get_ll_match(line, &line_match[6]);
281	<	network_stat_ptr->ierrors=get_ll_match(line, &line_match[4]);
282	<	network_stat_ptr->oerrors=get_ll_match(line, &line_match[7]);
283	<	network_stat_ptr->collisions=get_ll_match(line, &line_match[8]);
276	>	network_stat_ptr->interface_name=sg_get_string_match(line, &line_match[1]);
277	>	network_stat_ptr->rx=sg_get_ll_match(line, &line_match[2]);
278	>	network_stat_ptr->tx=sg_get_ll_match(line, &line_match[5]);
279	>	network_stat_ptr->ipackets=sg_get_ll_match(line, &line_match[3]);
280	>	network_stat_ptr->opackets=sg_get_ll_match(line, &line_match[6]);
281	>	network_stat_ptr->ierrors=sg_get_ll_match(line, &line_match[4]);
282	>	network_stat_ptr->oerrors=sg_get_ll_match(line, &line_match[7]);
283	>	network_stat_ptr->collisions=sg_get_ll_match(line, &line_match[8]);
284		network_stat_ptr->systime=time(NULL);
285
286		interfaces++;
#	Line 286 | Line 291 | network_stat_t *get_network_stats(int *entries){
291		#endif
292
293		#ifdef CYGWIN
294	+	sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
295		return NULL;
296		#endif
297	+	#ifdef HPUX
298	+	sg_set_error(SG_ERROR_UNSUPPORTED, "HP-UX");
299	+	return NULL;
300	+	#endif
301
302		*entries=interfaces;
303
304		return network_stats;
305		}
306
307	<	long long transfer_diff(long long new, long long old){
307	>	static long long transfer_diff(long long new, long long old){
308		#if defined(SOL7) || defined(LINUX) || defined(FREEBSD) || defined(DFBSD) || defined(OPENBSD)
309		/* 32-bit quantities, so we must explicitly deal with wraparound. */
310		#define MAXVAL 0x100000000LL
#	Line 309 | Line 319 | long long transfer_diff(long long new, long long old){
319		#endif
320		}
321
322	<	network_stat_t *get_network_stats_diff(int *entries) {
323	<	VECTOR_DECLARE_STATIC(diff, network_stat_t, 1,
324	<	network_stat_init, network_stat_destroy);
325	<	network_stat_t *src = NULL, *dest;
322	>	sg_network_io_stats *sg_get_network_io_stats_diff(int *entries) {
323	>	VECTOR_DECLARE_STATIC(diff, sg_network_io_stats, 1,
324	>	network_stat_init, network_stat_destroy);
325	>	sg_network_io_stats *src = NULL, *dest;
326		int i, j, diff_count, new_count;
327
328		if (network_stats == NULL) {
329		/* No previous stats, so we can't calculate a difference. */
330	<	return get_network_stats(entries);
330	>	return sg_get_network_io_stats(entries);
331		}
332
333		/* Resize the results array to match the previous stats. */
#	Line 331 | Line 341 | network_stat_t *get_network_stats_diff(int *entries) {
341		src = &network_stats[i];
342		dest = &diff[i];
343
344	<	if (dest->interface_name != NULL) {
345	<	free(dest->interface_name);
344	>	if (sg_update_string(&dest->interface_name,
345	>	src->interface_name) < 0) {
346	>	return NULL;
347		}
337	–	dest->interface_name = strdup(src->interface_name);
348		dest->rx = src->rx;
349		dest->tx = src->tx;
350		dest->ipackets = src->ipackets;
#	Line 346 | Line 356 | network_stat_t *get_network_stats_diff(int *entries) {
356		}
357
358		/* Get a new set of stats. */
359	<	if (get_network_stats(&new_count) == NULL) {
359	>	if (sg_get_network_io_stats(&new_count) == NULL) {
360		return NULL;
361		}
362
#	Line 384 | Line 394 | network_stat_t *get_network_stats_diff(int *entries) {
394		return diff;
395		}
396
397	+	int sg_network_io_compare_name(const void *va, const void *vb) {
398	+	const sg_network_io_stats *a = (const sg_network_io_stats *)va;
399	+	const sg_network_io_stats *b = (const sg_network_io_stats *)vb;
400	+
401	+	return strcmp(a->interface_name, b->interface_name);
402	+	}
403	+
404		/* NETWORK INTERFACE STATS */
405
406	<	static void network_iface_stat_init(network_iface_stat_t *s) {
406	>	static void network_iface_stat_init(sg_network_iface_stats *s) {
407		s->interface_name = NULL;
408		s->speed = 0;
409	<	s->dup = UNKNOWN_DUPLEX;
409	>	s->dup = SG_IFACE_DUPLEX_UNKNOWN;
410		}
411
412	<	static void network_iface_stat_destroy(network_iface_stat_t *s) {
412	>	static void network_iface_stat_destroy(sg_network_iface_stats *s) {
413		free(s->interface_name);
414		}
415
416	<	network_iface_stat_t *get_network_iface_stats(int *entries){
417	<	VECTOR_DECLARE_STATIC(network_iface_stats, network_iface_stat_t, 5,
418	<	network_iface_stat_init, network_iface_stat_destroy);
419	<	network_iface_stat_t *network_iface_stat_ptr;
416	>	sg_network_iface_stats *sg_get_network_iface_stats(int *entries){
417	>	VECTOR_DECLARE_STATIC(network_iface_stats, sg_network_iface_stats, 5,
418	>	network_iface_stat_init, network_iface_stat_destroy);
419	>	sg_network_iface_stats *network_iface_stat_ptr;
420		int ifaces = 0;
421
422		#ifdef SOLARIS
423	<	kstat_ctl_t *kc;
424	<	kstat_t *ksp;
423	>	kstat_ctl_t *kc;
424	>	kstat_t *ksp;
425		kstat_named_t *knp;
426		int sock;
427		#endif
428		#ifdef ALLBSD
429	<	struct ifaddrs *net, *net_ptr;
429	>	struct ifaddrs *net, *net_ptr;
430		struct ifmediareq ifmed;
431		struct ifreq ifr;
432		int sock;
433		int x;
434		#endif
435		#ifdef LINUX
436	<	FILE *f;
437	<	/* Horrible big enough, but it should be easily big enough */
438	<	char line[8096];
436	>	FILE *f;
437	>	/* Horrible big enough, but it should be easily big enough */
438	>	char line[8096];
439		int sock;
440		#endif
441
442		#ifdef ALLBSD
443	<	if(getifaddrs(&net) != 0){
444	<	return NULL;
445	<	}
443	>	if(getifaddrs(&net) != 0){
444	>	sg_set_error_with_errno(SG_ERROR_GETIFADDRS, NULL);
445	>	return NULL;
446	>	}
447
448		if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == 0) return NULL;
449
450		for(net_ptr=net; net_ptr!=NULL; net_ptr=net_ptr->ifa_next){
451	<	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
451	>	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
452
453		if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
454	<	return NULL;
455	<	}
456	<	network_iface_stat_ptr = network_iface_stats + ifaces;
454	>	return NULL;
455	>	}
456	>	network_iface_stat_ptr = network_iface_stats + ifaces;
457
458		memset(&ifr, 0, sizeof(ifr));
459		strncpy(ifr.ifr_name, net_ptr->ifa_name, sizeof(ifr.ifr_name));
#	Line 449 | Line 467 | network_iface_stat_t *get_network_iface_stats(int *ent
467		network_iface_stat_ptr->up = 0;
468		}
469
470	<	if (network_iface_stat_ptr->interface_name != NULL) free(network_iface_stat_ptr->interface_name);
471	<	network_iface_stat_ptr->interface_name = strdup(net_ptr->ifa_name);
472	<	if (network_iface_stat_ptr->interface_name == NULL) return NULL;
470	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
471	>	net_ptr->ifa_name) < 0) {
472	>	return NULL;
473	>	}
474
475		network_iface_stat_ptr->speed = 0;
476	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
476	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
477		ifaces++;
478
479		memset(&ifmed, 0, sizeof(struct ifmediareq));
480	<	strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
480	>	sg_strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
481		if(ioctl(sock, SIOCGIFMEDIA, (caddr_t)&ifmed) == -1){
482		/* Not all interfaces support the media ioctls. */
483		continue;
#	Line 511 | Line 530 | network_iface_stat_t *get_network_iface_stats(int *ent
530		}
531
532		if( (ifmed.ifm_active | IFM_FDX) == ifmed.ifm_active ){
533	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
533	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
534		}else if( (ifmed.ifm_active | IFM_HDX) == ifmed.ifm_active ){
535	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
535	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
536		}else{
537	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
537	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
538		}
539
540		}
#	Line 525 | Line 544 | network_iface_stat_t *get_network_iface_stats(int *ent
544
545		#ifdef SOLARIS
546		if ((kc = kstat_open()) == NULL) {
547	+	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
548		return NULL;
549		}
550
551		if ((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP)) < 0) {
552	+	sg_set_error_with_errno(SG_ERROR_SOCKET, NULL);
553		return NULL;
554		}
555
#	Line 550 | Line 571 | network_iface_stat_t *get_network_iface_stats(int *ent
571		network_iface_stat_ptr = network_iface_stats + ifaces;
572		ifaces++;
573
574	<	if (network_iface_stat_ptr->interface_name != NULL) free(network_iface_stat_ptr->interface_name);
575	<	network_iface_stat_ptr->interface_name = strdup(ksp->ks_name);
576	<	if (network_iface_stat_ptr->interface_name == NULL) return NULL;
574	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
575	>	ksp->ks_name) < 0) {
576	>	return NULL;
577	>	}
578
579		if ((ifr.ifr_flags & IFF_UP) != 0) {
580		network_iface_stat_ptr->up = 1;
#	Line 566 | Line 588 | network_iface_stat_t *get_network_iface_stats(int *ent
588		network_iface_stat_ptr->speed = 0;
589		}
590
591	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
591	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
592		if ((knp = kstat_data_lookup(ksp, "link_duplex")) != NULL) {
593		switch (knp->value.ui32) {
594		case 1:
595	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
595	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
596		break;
597		case 2:
598	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
598	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
599		break;
600		}
601		}
#	Line 585 | Line 607 | network_iface_stat_t *get_network_iface_stats(int *ent
607		#endif
608		#ifdef LINUX
609		f = fopen("/proc/net/dev", "r");
610	<	if(f == NULL){
611	<	return NULL;
612	<	}
610	>	if(f == NULL){
611	>	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/net/dev");
612	>	return NULL;
613	>	}
614
615		/* Setup stuff so we can do the ioctl to get the info */
616		if((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0){
617	+	sg_set_error_with_errno(SG_ERROR_SOCKET, NULL);
618		return NULL;
619		}
620
621		/* Ignore first 2 lines.. Just headings */
622	<	if((fgets(line, sizeof(line), f)) == NULL) return NULL;
623	<	if((fgets(line, sizeof(line), f)) == NULL) return NULL;
622	>	if((fgets(line, sizeof(line), f)) == NULL) {
623	>	sg_set_error(SG_ERROR_PARSE, NULL);
624	>	return NULL;
625	>	}
626	>	if((fgets(line, sizeof(line), f)) == NULL) {
627	>	sg_set_error(SG_ERROR_PARSE, NULL);
628	>	return NULL;
629	>	}
630
631	<	while((fgets(line, sizeof(line), f)) != NULL){
632	<	char *name, *ptr;
633	<	struct ifreq ifr;
634	<	struct ethtool_cmd ethcmd;
635	<	int err;
631	>	while((fgets(line, sizeof(line), f)) != NULL){
632	>	char *name, *ptr;
633	>	struct ifreq ifr;
634	>	struct ethtool_cmd ethcmd;
635	>	int err;
636
637		/* Get the interface name */
638	<	ptr = strchr(line, ':');
639	<	if (ptr == NULL) continue;
640	<	*ptr='\0';
641	<	name = line;
642	<	while(isspace(*(name))){
643	<	name++;
644	<	}
638	>	ptr = strchr(line, ':');
639	>	if (ptr == NULL) continue;
640	>	*ptr='\0';
641	>	name = line;
642	>	while(isspace(*(name))){
643	>	name++;
644	>	}
645
646	<	memset(&ifr, 0, sizeof ifr);
647	<	strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
646	>	memset(&ifr, 0, sizeof ifr);
647	>	strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
648
649		if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
650		continue;
#	Line 625 | Line 655 | network_iface_stat_t *get_network_iface_stats(int *ent
655		return NULL;
656		}
657		network_iface_stat_ptr = network_iface_stats + ifaces;
658	<	network_iface_stat_ptr->interface_name = strdup(name);
658	>
659	>	if (sg_update_string(&network_iface_stat_ptr->interface_name,
660	>	name) < 0) {
661	>	return NULL;
662	>	}
663		if ((ifr.ifr_flags & IFF_UP) != 0) {
664		network_iface_stat_ptr->up = 1;
665		} else {
666		network_iface_stat_ptr->up = 0;
667		}
668
669	<	memset(&ethcmd, 0, sizeof ethcmd);
670	<	ethcmd.cmd = ETHTOOL_GSET;
671	<	ifr.ifr_data = (caddr_t) &ethcmd;
669	>	memset(&ethcmd, 0, sizeof ethcmd);
670	>	ethcmd.cmd = ETHTOOL_GSET;
671	>	ifr.ifr_data = (caddr_t) &ethcmd;
672
673	<	err = ioctl(sock, SIOCETHTOOL, &ifr);
674	<	if (err == 0) {
673	>	err = ioctl(sock, SIOCETHTOOL, &ifr);
674	>	if (err == 0) {
675		network_iface_stat_ptr->speed = ethcmd.speed;
676
677		switch (ethcmd.duplex) {
678		case 0x00:
679	<	network_iface_stat_ptr->dup = FULL_DUPLEX;
679	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
680		break;
681		case 0x01:
682	<	network_iface_stat_ptr->dup = HALF_DUPLEX;
682	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
683		break;
684		default:
685	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
685	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
686		}
687		} else {
688		/* Not all interfaces support the ethtool ioctl. */
689		network_iface_stat_ptr->speed = 0;
690	<	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
690	>	network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
691		}
692
693		ifaces++;
#	Line 663 | Line 697 | network_iface_stat_t *get_network_iface_stats(int *ent
697		#endif
698		*entries = ifaces;
699		return network_iface_stats;
700	+	}
701	+
702	+	int sg_network_iface_compare_name(const void *va, const void *vb) {
703	+	const sg_network_iface_stats *a = (const sg_network_iface_stats *)va;
704	+	const sg_network_iface_stats *b = (const sg_network_iface_stats *)vb;
705	+
706	+	return strcmp(a->interface_name, b->interface_name);
707		}
708

Diff Legend

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