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.22 by tdb, Fri Jan 16 15:54:54 2004 UTC vs.
Revision 1.47 by tdb, Sat Mar 6 21:49:13 2004 UTC

#	Line 17 | Line 17
17		* License along with this library; if not, write to the Free Software
18		* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19		* 02111-1307 USA
20	+	*
21	+	* $Id$
22		*/
23
24		#ifdef HAVE_CONFIG_H
#	Line 30 | Line 32
32		#ifdef SOLARIS
33		#include <kstat.h>
34		#include <sys/sysinfo.h>
35	+	#include <sys/types.h>
36	+	#include <sys/socket.h>
37	+	#include <sys/ioctl.h>
38	+	#include <net/if.h>
39	+	#include <netinet/in.h>
40	+	#include <sys/sockio.h>
41		#endif
42		#ifdef LINUX
43		#include <stdio.h>
44		#include <sys/types.h>
45		#include <regex.h>
46	+	#include <sys/ioctl.h>
47	+	#include <sys/socket.h>
48	+	#include <net/if.h>
49	+	#include <ctype.h>
50		#include "tools.h"
51	+	/* Stuff which could be defined by defining KERNEL, but
52	+	* that would be a bad idea, so we'll just declare it here
53	+	*/
54	+	typedef __uint8_t u8;
55	+	typedef __uint16_t u16;
56	+	typedef __uint32_t u32;
57	+	typedef __uint64_t u64;
58	+	#include <linux/ethtool.h>
59	+	#include <linux/sockios.h>
60	+	#include <unistd.h>
61		#endif
62		#ifdef ALLBSD
63		#include <sys/types.h>
64		#include <sys/socket.h>
65		#include <ifaddrs.h>
66		#include <net/if.h>
67	+	#include <net/if_media.h>
68	+	#include <sys/ioctl.h>
69		#endif
70
71		static network_stat_t *network_stats=NULL;
#	Line 53 | Line 77 | void network_stat_init(int start, int end, network_sta
77		net_stats->interface_name=NULL;
78		net_stats->tx=0;
79		net_stats->rx=0;
80	+	net_stats->ipackets=0;
81	+	net_stats->opackets=0;
82	+	net_stats->ierrors=0;
83	+	net_stats->oerrors=0;
84	+	net_stats->collisions=0;
85		net_stats++;
86		}
87		}
#	Line 127 | Line 156 | network_stat_t *get_network_stats(int *entries){
156		if(network_stat_ptr->interface_name==NULL) return NULL;
157		net_data=(struct if_data *)net_ptr->ifa_data;
158		network_stat_ptr->rx=net_data->ifi_ibytes;
159	<	network_stat_ptr->tx=net_data->ifi_obytes;
159	>	network_stat_ptr->tx=net_data->ifi_obytes;
160	>	network_stat_ptr->ipackets=net_data->ifi_ipackets;
161	>	network_stat_ptr->opackets=net_data->ifi_opackets;
162	>	network_stat_ptr->ierrors=net_data->ifi_ierrors;
163	>	network_stat_ptr->oerrors=net_data->ifi_oerrors;
164	>	network_stat_ptr->collisions=net_data->ifi_collisions;
165		network_stat_ptr->systime=time(NULL);
166		interfaces++;
167		}
#	Line 156 | Line 190 | network_stat_t *get_network_stats(int *entries){
190		#endif
191
192		if((knp=kstat_data_lookup(ksp, RLOOKUP))==NULL){
193	<	/* Not a network interface, so skip to the next entry */
193	>	/* This is a network interface, but it doesn't
194	>	* have the rbytes/obytes values; for instance,
195	>	* the loopback devices have this behaviour
196	>	* (although they do track packets in/out). */
197		continue;
198		}
199
#	Line 168 | Line 205 | network_stat_t *get_network_stats(int *entries){
205		network_stat_ptr->rx=knp->VALTYPE;
206
207		if((knp=kstat_data_lookup(ksp, WLOOKUP))==NULL){
171	–	/* Not a network interface, so skip to the next entry */
208		continue;
209		}
210		network_stat_ptr->tx=knp->VALTYPE;
#	Line 236 | Line 272 | network_stat_t *get_network_stats(int *entries){
272		}
273
274		long long transfer_diff(long long new, long long old){
275	<	#if defined(SOL7) || defined(LINUX) || defined(FREEBSD)
275	>	#if defined(SOL7) || defined(LINUX) || defined(FREEBSD) || defined(DFBSD)
276		#define MAXVAL 4294967296LL
277		#else
278		#define MAXVAL 18446744073709551616LL
#	Line 252 | Line 288 | long long transfer_diff(long long new, long long old){
288
289		}
290
291	<	network_stat_t *get_network_stats_diff(int *entries){
292	<	static network_stat_t *network_stats_diff=NULL;
293	<	static int sizeof_net_stats_diff=0;
294	<	network_stat_t *network_stats_ptr, *network_stats_diff_ptr;
295	<	int ifaces, x, y;
291	>	network_stat_t *get_network_stats_diff(int *entries) {
292	>	static network_stat_t *diff = NULL;
293	>	static int diff_count = 0;
294	>	network_stat_t *src, *dest;
295	>	int i, j, new_count;
296
297	<	if(network_stats==NULL){
298	<	network_stats_ptr=get_network_stats(&ifaces);
299	<	*entries=ifaces;
264	<	return network_stats_ptr;
297	>	if (network_stats == NULL) {
298	>	/* No previous stats, so we can't calculate a difference. */
299	>	return get_network_stats(entries);
300		}
301
302	<	network_stats_diff=network_stat_malloc(interfaces, &sizeof_net_stats_diff, network_stats_diff);
303	<	if(network_stats_diff==NULL){
302	>	/* Resize the results array to match the previous stats. */
303	>	diff = network_stat_malloc(interfaces, &diff_count, diff);
304	>	if (diff == NULL) {
305		return NULL;
306		}
307
308	<	network_stats_ptr=network_stats;
309	<	network_stats_diff_ptr=network_stats_diff;
308	>	/* Copy the previous stats into the result. */
309	>	for (i = 0; i < diff_count; i++) {
310	>	src = &network_stats[i];
311	>	dest = &diff[i];
312
313	<	for(ifaces=0;ifaces<interfaces;ifaces++){
314	<	if(network_stats_diff_ptr->interface_name!=NULL){
277	<	free(network_stats_diff_ptr->interface_name);
313	>	if (dest->interface_name != NULL) {
314	>	free(dest->interface_name);
315		}
316	<	network_stats_diff_ptr->interface_name=strdup(network_stats_ptr->interface_name);
317	<	network_stats_diff_ptr->tx=network_stats_ptr->tx;
318	<	network_stats_diff_ptr->rx=network_stats_ptr->rx;
319	<	network_stats_diff_ptr->systime=network_stats->systime;
320	<
321	<	network_stats_ptr++;
322	<	network_stats_diff_ptr++;
316	>	dest->interface_name = strdup(src->interface_name);
317	>	dest->rx = src->rx;
318	>	dest->tx = src->tx;
319	>	dest->ipackets = src->ipackets;
320	>	dest->opackets = src->opackets;
321	>	dest->ierrors = src->ierrors;
322	>	dest->oerrors = src->oerrors;
323	>	dest->collisions = src->collisions;
324	>	dest->systime = src->systime;
325		}
326	<	network_stats_ptr=get_network_stats(&ifaces);
327	<	if (network_stats_ptr == NULL) {
326	>
327	>	/* Get a new set of stats. */
328	>	if (get_network_stats(&new_count) == NULL) {
329		return NULL;
330		}
291	–	network_stats_diff_ptr=network_stats_diff;
331
332	<	for(x=0;x<sizeof_net_stats_diff;x++){
332	>	/* For each previous stat... */
333	>	for (i = 0; i < diff_count; i++) {
334	>	dest = &diff[i];
335
336	<	if((strcmp(network_stats_diff_ptr->interface_name, network_stats_ptr->interface_name))==0){
337	<	network_stats_diff_ptr->tx = transfer_diff(network_stats_ptr->tx, network_stats_diff_ptr->tx);
338	<	network_stats_diff_ptr->rx = transfer_diff(network_stats_ptr->rx, network_stats_diff_ptr->rx);
339	<	network_stats_diff_ptr->systime = network_stats_ptr->systime - network_stats_diff_ptr->systime;
336	>	/* ... find the corresponding new stat ... */
337	>	for (j = 0; j < new_count; j++) {
338	>	/* Try the new stat in the same position first,
339	>	since that's most likely to be it. */
340	>	src = &network_stats[(i + j) % new_count];
341	>	if (strcmp(src->interface_name, dest->interface_name) == 0) {
342	>	break;
343	>	}
344	>	}
345	>	if (j == new_count) {
346	>	/* No match found. */
347	>	continue;
348	>	}
349	>
350	>	/* ... and subtract the previous stat from it to get the
351	>	difference. */
352	>	dest->rx = transfer_diff(src->rx, dest->rx);
353	>	dest->tx = transfer_diff(src->tx, dest->tx);
354	>	dest->ipackets = transfer_diff(src->ipackets, dest->ipackets);
355	>	dest->opackets = transfer_diff(src->opackets, dest->opackets);
356	>	dest->ierrors = transfer_diff(src->ierrors, dest->ierrors);
357	>	dest->oerrors = transfer_diff(src->oerrors, dest->oerrors);
358	>	dest->collisions = transfer_diff(src->collisions, dest->collisions);
359	>	dest->systime = src->systime - dest->systime;
360	>	}
361	>
362	>	*entries = diff_count;
363	>	return diff;
364	>	}
365	>	/* NETWORK INTERFACE STATS */
366	>
367	>	void network_iface_stat_init(int start, int end, network_iface_stat_t *net_stats){
368	>
369	>	for(net_stats+=start; start<end; start++){
370	>	net_stats->interface_name=NULL;
371	>	net_stats->speed=0;
372	>	net_stats->dup=UNKNOWN_DUPLEX;
373	>	net_stats++;
374	>	}
375	>	}
376	>
377	>	network_iface_stat_t *network_iface_stat_malloc(int needed_entries, int *cur_entries, network_iface_stat_t *net_stats){
378	>
379	>	if(net_stats==NULL){
380	>
381	>	if((net_stats=malloc(needed_entries * sizeof(network_iface_stat_t)))==NULL){
382	>	return NULL;
383	>	}
384	>	network_iface_stat_init(0, needed_entries, net_stats);
385	>	*cur_entries=needed_entries;
386	>
387	>	return net_stats;
388	>	}
389	>
390	>
391	>	if(*cur_entries<needed_entries){
392	>	net_stats=realloc(net_stats, (sizeof(network_iface_stat_t)*needed_entries));
393	>	if(net_stats==NULL){
394	>	return NULL;
395	>	}
396	>	network_iface_stat_init(*cur_entries, needed_entries, net_stats);
397	>	*cur_entries=needed_entries;
398	>	}
399	>
400	>	return net_stats;
401	>	}
402	>
403	>	network_iface_stat_t *get_network_iface_stats(int *entries){
404	>	static network_iface_stat_t *network_iface_stats;
405	>	network_iface_stat_t *network_iface_stat_ptr;
406	>	static int sizeof_network_iface_stats=0;
407	>	int ifaces = 0;
408	>
409	>	#ifdef SOLARIS
410	>	kstat_ctl_t *kc;
411	>	kstat_t *ksp;
412	>	kstat_named_t *knp;
413	>	int sock;
414	>	#endif
415	>	#ifdef ALLBSD
416	>	struct ifaddrs *net, *net_ptr;
417	>	struct ifmediareq ifmed;
418	>	struct ifreq ifr;
419	>	int sock;
420	>	int x;
421	>	#endif
422	>	#ifdef LINUX
423	>	FILE *f;
424	>	/* Horrible big enough, but it should be easily big enough */
425	>	char line[8096];
426	>	int sock;
427	>	#endif
428	>
429	>	#ifdef ALLBSD
430	>	if(getifaddrs(&net) != 0){
431	>	return NULL;
432	>	}
433	>
434	>	if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == 0) return NULL;
435	>
436	>	for(net_ptr=net; net_ptr!=NULL; net_ptr=net_ptr->ifa_next){
437	>	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
438	>	network_iface_stats=network_iface_stat_malloc((ifaces+1), &sizeof_network_iface_stats, network_iface_stats);
439	>	if(network_iface_stats==NULL){
440	>	return NULL;
441	>	}
442	>	network_iface_stat_ptr = network_iface_stats + ifaces;
443	>
444	>	memset(&ifr, 0, sizeof(ifr));
445	>	strncpy(ifr.ifr_name, net_ptr->ifa_name, sizeof(ifr.ifr_name));
446	>
447	>	if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0){
448	>	continue;
449	>	}
450	>	if((ifr.ifr_flags & IFF_UP) != 0){
451	>	network_iface_stat_ptr->up = 1;
452		}else{
453	<
454	<	network_stats_ptr=network_stats;
455	<	for(y=0;y<ifaces;y++){
456	<	if((strcmp(network_stats_diff_ptr->interface_name, network_stats_ptr->interface_name))==0){
457	<	network_stats_diff_ptr->tx = transfer_diff(network_stats_ptr->tx, network_stats_diff_ptr->tx);
458	<	network_stats_diff_ptr->rx = transfer_diff(network_stats_ptr->rx, network_stats_diff_ptr->rx);
459	<	network_stats_diff_ptr->systime = network_stats_ptr->systime - network_stats_diff_ptr->systime;
453	>	network_iface_stat_ptr->up = 0;
454	>	}
455	>
456	>	if (network_iface_stat_ptr->interface_name != NULL) free(network_iface_stat_ptr->interface_name);
457	>	network_iface_stat_ptr->interface_name = strdup(net_ptr->ifa_name);
458	>	if (network_iface_stat_ptr->interface_name == NULL) return NULL;
459	>
460	>	network_iface_stat_ptr->speed = 0;
461	>	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
462	>	ifaces++;
463	>
464	>	memset(&ifmed, 0, sizeof(struct ifmediareq));
465	>	strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
466	>	if(ioctl(sock, SIOCGIFMEDIA, (caddr_t)&ifmed) == -1){
467	>	/* Not all interfaces support the media ioctls. */
468	>	continue;
469	>	}
470	>
471	>	/* We may need to change this if we start doing wireless devices too */
472	>	if( (ifmed.ifm_active | IFM_ETHER) != ifmed.ifm_active ){
473	>	/* Not a ETHER device */
474	>	continue;
475	>	}
476	>
477	>	/* Only intrested in the first 4 bits)  - Assuming only ETHER devices */
478	>	x = ifmed.ifm_active & 0x0f;
479	>	switch(x){
480	>	/* 10 Mbit connections. Speedy :) */
481	>	case(IFM_10_T):
482	>	case(IFM_10_2):
483	>	case(IFM_10_5):
484	>	case(IFM_10_STP):
485	>	case(IFM_10_FL):
486	>	network_iface_stat_ptr->speed = 10;
487	>	break;
488	>	/* 100 Mbit conneections */
489	>	case(IFM_100_TX):
490	>	case(IFM_100_FX):
491	>	case(IFM_100_T4):
492	>	case(IFM_100_VG):
493	>	case(IFM_100_T2):
494	>	network_iface_stat_ptr->speed = 100;
495	>	break;
496	>	/* 1000 Mbit connections */
497	>	case(IFM_1000_SX):
498	>	case(IFM_1000_LX):
499	>	case(IFM_1000_CX):
500	>	#ifdef IFM_1000_TX
501	>	case(IFM_1000_TX): /* FreeBSD 4 and others? */
502	>	#endif
503	>	#ifdef IFM_1000_FX
504	>	case(IFM_1000_FX): /* FreeBSD 4 */
505	>	#endif
506	>	#ifdef IFM_1000_T
507	>	case(IFM_1000_T): /* FreeBSD 5 */
508	>	#endif
509	>	network_iface_stat_ptr->speed = 1000;
510	>	break;
511	>	/* We don't know what it is */
512	>	default:
513	>	network_iface_stat_ptr->speed = 0;
514	>	break;
515	>	}
516	>
517	>	if( (ifmed.ifm_active | IFM_FDX) == ifmed.ifm_active ){
518	>	network_iface_stat_ptr->dup = FULL_DUPLEX;
519	>	}else if( (ifmed.ifm_active | IFM_HDX) == ifmed.ifm_active ){
520	>	network_iface_stat_ptr->dup = HALF_DUPLEX;
521	>	}else{
522	>	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
523	>	}
524	>
525	>	}
526	>	freeifaddrs(net);
527	>	close(sock);
528	>	#endif
529	>
530	>	#ifdef SOLARIS
531	>	if ((kc = kstat_open()) == NULL) {
532	>	return NULL;
533	>	}
534	>
535	>	if ((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP)) < 0) {
536	>	return NULL;
537	>	}
538	>
539	>	for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
540	>	if (!strcmp(ksp->ks_class, "net")) {
541	>	struct ifreq ifr;
542	>
543	>	kstat_read(kc, ksp, NULL);
544	>
545	>	strncpy(ifr.ifr_name, ksp->ks_name, sizeof ifr.ifr_name);
546	>	if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
547	>	/* Not a network interface. */
548	>	continue;
549	>	}
550	>
551	>	network_iface_stats = network_iface_stat_malloc(ifaces + 1, &sizeof_network_iface_stats, network_iface_stats);
552	>	if (network_iface_stats == NULL) {
553	>	return NULL;
554	>	}
555	>	network_iface_stat_ptr = network_iface_stats + ifaces;
556	>	ifaces++;
557	>
558	>	if (network_iface_stat_ptr->interface_name != NULL) free(network_iface_stat_ptr->interface_name);
559	>	network_iface_stat_ptr->interface_name = strdup(ksp->ks_name);
560	>	if (network_iface_stat_ptr->interface_name == NULL) return NULL;
561	>
562	>	if ((ifr.ifr_flags & IFF_UP) != 0) {
563	>	network_iface_stat_ptr->up = 1;
564	>	} else {
565	>	network_iface_stat_ptr->up = 1;
566	>	}
567	>
568	>	if ((knp = kstat_data_lookup(ksp, "ifspeed")) != NULL) {
569	>	network_iface_stat_ptr->speed = knp->value.ui64 / (1000 * 1000);
570	>	} else {
571	>	network_iface_stat_ptr->speed = 0;
572	>	}
573	>
574	>	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
575	>	if ((knp = kstat_data_lookup(ksp, "link_duplex")) != NULL) {
576	>	switch (knp->value.ui32) {
577	>	case 1:
578	>	network_iface_stat_ptr->dup = HALF_DUPLEX;
579		break;
580	+	case 2:
581	+	network_iface_stat_ptr->dup = FULL_DUPLEX;
582	+	break;
583		}
584	<
310	<	network_stats_ptr++;
311	<	}
584	>	}
585		}
586	+	}
587
588	<	network_stats_ptr++;
589	<	network_stats_diff_ptr++;
588	>	close(sock);
589	>	kstat_close(kc);
590	>	#endif
591	>	#ifdef LINUX
592	>	f = fopen("/proc/net/dev", "r");
593	>	if(f == NULL){
594	>	return NULL;
595	>	}
596	>
597	>	/* Setup stuff so we can do the ioctl to get the info */
598	>	if((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0){
599	>	return NULL;
600		}
601
602	<	*entries=sizeof_net_stats_diff;
603	<	return network_stats_diff;
604	<	}
602	>	/* Ignore first 2 lines.. Just headings */
603	>	if((fgets(line, sizeof(line), f)) == NULL) return NULL;
604	>	if((fgets(line, sizeof(line), f)) == NULL) return NULL;
605	>
606	>	while((fgets(line, sizeof(line), f)) != NULL){
607	>	char *name, *ptr;
608	>	struct ifreq ifr;
609	>	struct ethtool_cmd ethcmd;
610	>	int err;
611	>
612	>	/* Get the interface name */
613	>	ptr = strchr(line, ':');
614	>	if (ptr == NULL) continue;
615	>	*ptr='\0';
616	>	name = line;
617	>	while(isspace(*(name))){
618	>	name++;
619	>	}
620	>
621	>	memset(&ifr, 0, sizeof ifr);
622	>	strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
623	>
624	>	if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
625	>	continue;
626	>	}
627	>
628	>	/* We have a good interface to add */
629	>	network_iface_stats=network_iface_stat_malloc((ifaces+1), &sizeof_network_iface_stats, network_iface_stats);
630	>	if(network_iface_stats==NULL){
631	>	return NULL;
632	>	}
633	>	network_iface_stat_ptr = network_iface_stats + ifaces;
634	>	network_iface_stat_ptr->interface_name = strdup(name);
635	>	if ((ifr.ifr_flags & IFF_UP) != 0) {
636	>	network_iface_stat_ptr->up = 1;
637	>	} else {
638	>	network_iface_stat_ptr->up = 0;
639	>	}
640	>
641	>	memset(&ethcmd, 0, sizeof ethcmd);
642	>	ethcmd.cmd = ETHTOOL_GSET;
643	>	ifr.ifr_data = (caddr_t) &ethcmd;
644	>
645	>	err = ioctl(sock, SIOCETHTOOL, &ifr);
646	>	if (err == 0) {
647	>	network_iface_stat_ptr->speed = ethcmd.speed;
648	>
649	>	switch (ethcmd.duplex) {
650	>	case 0x00:
651	>	network_iface_stat_ptr->dup = FULL_DUPLEX;
652	>	break;
653	>	case 0x01:
654	>	network_iface_stat_ptr->dup = HALF_DUPLEX;
655	>	break;
656	>	default:
657	>	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
658	>	}
659	>	} else {
660	>	/* Not all interfaces support the ethtool ioctl. */
661	>	network_iface_stat_ptr->speed = 0;
662	>	network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
663	>	}
664	>
665	>	ifaces++;
666	>	}
667	>	close(sock);
668	>	fclose(f);
669	>	#endif
670	>	*entries = ifaces;
671	>	return network_iface_stats;
672	>	}
673

Diff Legend

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