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.1 by pajs, Tue Feb 18 19:28:30 2003 UTC vs.
Revision 1.77 by tdb, Sun Jan 22 18:10:39 2006 UTC

#	Line 1 | Line 1
1	<	/*
2	<	* i-scream central monitoring system
3	<	* http://www.i-scream.org.uk
4	<	* Copyright (C) 2000-2002 i-scream
1	>	/*
2	>	* i-scream libstatgrab
3	>	* http://www.i-scream.org
4	>	* Copyright (C) 2000-2004 i-scream
5		*
6	<	* This program is free software; you can redistribute it and/or
7	<	* modify it under the terms of the GNU General Public License
8	<	* as published by the Free Software Foundation; either version 2
9	<	* of the License, or (at your option) any later version.
6	>	* This library is free software; you can redistribute it and/or
7	>	* modify it under the terms of the GNU Lesser General Public
8	>	* License as published by the Free Software Foundation; either
9	>	* version 2.1 of the License, or (at your option) any later version.
10		*
11	<	* This program is distributed in the hope that it will be useful,
11	>	* This library is distributed in the hope that it will be useful,
12		* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	<	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14	<	* GNU General Public License for more details.
13	>	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14	>	* Lesser General Public License for more details.
15		*
16	<	* You should have received a copy of the GNU General Public License
17	<	* along with this program; if not, write to the Free Software
18	<	* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
16	>	* You should have received a copy of the GNU Lesser General Public
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
25		#include "config.h"
26		#endif
27
25	–	#include <stdio.h>
26	–	#include "statgrab.h"
28		#include <stdlib.h>
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>
36		#include <sys/sysinfo.h>
37	<	#include <string.h>
37	>	#include <sys/types.h>
38	>	#include <sys/socket.h>
39	>	#include <sys/ioctl.h>
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>
48	+	#include <sys/ioctl.h>
49	+	#include <sys/socket.h>
50	+	#include <net/if.h>
51	+	#include <ctype.h>
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;
59	+	typedef __uint64_t u64;
60	+	#include <linux/ethtool.h>
61	+	#include <linux/sockios.h>
62	+	#include <unistd.h>
63	+	#endif
64	+	#ifdef ALLBSD
65	+	#include <sys/types.h>
66	+	#include <sys/socket.h>
67	+	#include <ifaddrs.h>
68	+	#include <net/if.h>
69	+	#include <net/if_media.h>
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	<	#define START_VAL 1
79	>	static void network_stat_init(sg_network_io_stats *s) {
80	>	s->interface_name = NULL;
81	>	s->tx = 0;
82	>	s->rx = 0;
83	>	s->ipackets = 0;
84	>	s->opackets = 0;
85	>	s->ierrors = 0;
86	>	s->oerrors = 0;
87	>	s->collisions = 0;
88	>	}
89
90	<	network_stat_t *network_stat_init(int num_iface, int *wmark, network_stat_t *net_stats){
91	<	int x;
92	<	network_stat_t *net_stats_ptr;
90	>	static void network_stat_destroy(sg_network_io_stats *s) {
91	>	free(s->interface_name);
92	>	}
93
94	<	if(*wmark==-1){
95	<	printf("new malloc\n");
96	<	if((net_stats=malloc(START_VAL * sizeof(network_stat_t)))==NULL){
97	<	return NULL;
98	<	}
99	<	*wmark=START_VAL;
100	<	net_stats_ptr=net_stats;
101	<	for(x=0;x<(*wmark);x++){
102	<	net_stats_ptr->interface_name=NULL;
103	<	net_stats_ptr++;
104	<	}
105	<	return net_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	<	if(num_iface>(*wmark-1)){
110	<	if((net_stats=realloc(net_stats, (*wmark)*2 * sizeof(network_stat_t)))==NULL){
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	<
58	<	*wmark=(*wmark)*2;
59	<	net_stats_ptr=net_stats+num_iface;
60	<	for(x=num_iface;x<(*wmark);x++){
61	<	net_stats_ptr->interface_name=NULL;
62	<	net_stats_ptr++;
63	<	}
64	<	return net_stats;
117	>	if_table = tmp;
118		}
119
120	<	return net_stats;
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	<	network_stat_t *get_network_stats(int *entries){
130	<	kstat_ctl_t *kc;
131	<	kstat_t *ksp;
129	>	sg_network_io_stats *sg_get_network_io_stats(int *entries){
130	>	int interfaces;
131	>	sg_network_io_stats *network_stat_ptr;
132	>
133	>	#ifdef SOLARIS
134	>	kstat_ctl_t *kc;
135	>	kstat_t *ksp;
136		kstat_named_t *knp;
137	+	#endif
138	+
139	+	#ifdef LINUX
140	+	FILE *f;
141	+	/* Horrible big enough, but it should be easily big enough */
142	+	char line[8096];
143	+	regex_t regex;
144	+	regmatch_t line_match[9];
145	+	#endif
146	+	#ifdef ALLBSD
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);
163	+	return NULL;
164	+	}
165	+
166	+	interfaces=0;
167
168	<	int interfaces=0;
169	<	network_stat_t *network_stat_ptr;
77	<	static network_stat_t *network_stats=NULL;
78	<	static int watermark=-1;
168	>	for(net_ptr=net;net_ptr!=NULL;net_ptr=net_ptr->ifa_next){
169	>	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
170
171	<	if ((kc = kstat_open()) == NULL) {
172	<	return NULL;
173	<	}
171	>	if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
172	>	return NULL;
173	>	}
174	>	network_stat_ptr=network_stats+interfaces;
175	>
176	>	if (sg_update_string(&network_stat_ptr->interface_name,
177	>	net_ptr->ifa_name) < 0) {
178	>	return NULL;
179	>	}
180	>	net_data=(struct if_data *)net_ptr->ifa_data;
181	>	network_stat_ptr->rx=net_data->ifi_ibytes;
182	>	network_stat_ptr->tx=net_data->ifi_obytes;
183	>	network_stat_ptr->ipackets=net_data->ifi_ipackets;
184	>	network_stat_ptr->opackets=net_data->ifi_opackets;
185	>	network_stat_ptr->ierrors=net_data->ifi_ierrors;
186	>	network_stat_ptr->oerrors=net_data->ifi_oerrors;
187	>	network_stat_ptr->collisions=net_data->ifi_collisions;
188	>	network_stat_ptr->systime=time(NULL);
189	>	interfaces++;
190	>	}
191	>	freeifaddrs(net);
192	>	#endif
193
194	<	for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
195	<	if (!strcmp(ksp->ks_class, "net")) {
196	<	kstat_read(kc, ksp, NULL);
194	>	#ifdef SOLARIS
195	>	if ((kc = kstat_open()) == NULL) {
196	>	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
197	>	return NULL;
198	>	}
199
200	<	if((knp=kstat_data_lookup(ksp, "rbytes64"))==NULL){
201	<	/* Not a network interface, so skip to the next entry */
200	>	interfaces=0;
201	>
202	>	for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
203	>	if (strcmp(ksp->ks_class, "net") == 0) {
204	>	kstat_read(kc, ksp, NULL);
205	>
206	>	#ifdef SOL7
207	>	#define LRX "rbytes"
208	>	#define LTX "obytes"
209	>	#define LIPACKETS "ipackets"
210	>	#define LOPACKETS "opackets"
211	>	#define VALTYPE value.ui32
212	>	#else
213	>	#define LRX "rbytes64"
214	>	#define LTX "obytes64"
215	>	#define LIPACKETS "ipackets64"
216	>	#define LOPACKETS "opackets64"
217	>	#define VALTYPE value.ui64
218	>	#endif
219	>
220	>	/* Read rx */
221	>	if((knp=kstat_data_lookup(ksp, LRX))==NULL){
222	>	/* This is a network interface, but it doesn't
223	>	* have the rbytes/obytes values; for instance,
224	>	* the loopback devices have this behaviour
225	>	* (although they do track packets in/out). */
226	>	/* FIXME: Show packet counts when byte counts
227	>	* not available. */
228		continue;
229		}
230	<	network_stats=network_stat_init(interfaces, &watermark, network_stats);
231	<	if(network_stats==NULL){
230	>
231	>	/* Create new network_stats */
232	>	if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
233		return NULL;
234		}
235		network_stat_ptr=network_stats+interfaces;
97	–	network_stat_ptr->rx=knp->value.ui64;
236
237	<	if((knp=kstat_data_lookup(ksp, "obytes64"))==NULL){
238	<	/* Not a network interface, so skip to the next entry */
237	>	/* Finish reading rx */
238	>	network_stat_ptr->rx=knp->VALTYPE;
239	>
240	>	/* Read tx */
241	>	if((knp=kstat_data_lookup(ksp, LTX))==NULL){
242		continue;
243		}
244	<	network_stat_ptr->tx=knp->value.ui64;
245	<	if(network_stat_ptr->interface_name!=NULL){
246	<	free(network_stat_ptr->interface_name);
244	>	network_stat_ptr->tx=knp->VALTYPE;
245	>
246	>	/* Read ipackets */
247	>	if((knp=kstat_data_lookup(ksp, LIPACKETS))==NULL){
248	>	continue;
249		}
250	<	network_stat_ptr->interface_name=strdup(ksp->ks_name);
250	>	network_stat_ptr->ipackets=knp->VALTYPE;
251	>
252	>	/* Read opackets */
253	>	if((knp=kstat_data_lookup(ksp, LOPACKETS))==NULL){
254	>	continue;
255	>	}
256	>	network_stat_ptr->opackets=knp->VALTYPE;
257	>
258	>	/* Read ierrors */
259	>	if((knp=kstat_data_lookup(ksp, "ierrors"))==NULL){
260	>	continue;
261	>	}
262	>	network_stat_ptr->ierrors=knp->value.ui32;
263	>
264	>	/* Read oerrors */
265	>	if((knp=kstat_data_lookup(ksp, "oerrors"))==NULL){
266	>	continue;
267	>	}
268	>	network_stat_ptr->oerrors=knp->value.ui32;
269	>
270	>	/* Read collisions */
271	>	if((knp=kstat_data_lookup(ksp, "collisions"))==NULL){
272	>	continue;
273	>	}
274	>	network_stat_ptr->collisions=knp->value.ui32;
275	>
276	>	/* Read interface name */
277	>	if (sg_update_string(&network_stat_ptr->interface_name,
278	>	ksp->ks_name) < 0) {
279	>	return NULL;
280	>	}
281	>
282	>	/* Store systime */
283	>	network_stat_ptr->systime=time(NULL);
284	>
285		interfaces++;
286		}
287		}
288	<
288	>
289	>	kstat_close(kc);
290	>	#endif
291	>	#ifdef LINUX
292	>	f=fopen("/proc/net/dev", "r");
293	>	if(f==NULL){
294	>	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/net/dev");
295	>	return NULL;
296	>	}
297	>	/* read the 2 lines.. Its the title, so we dont care :) */
298	>	fgets(line, sizeof(line), f);
299	>	fgets(line, sizeof(line), f);
300
301	+
302	+	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){
303	+	sg_set_error(SG_ERROR_PARSE, NULL);
304	+	return NULL;
305	+	}
306	+
307	+	interfaces=0;
308	+
309	+	while((fgets(line, sizeof(line), f)) != NULL){
310	+	if((regexec(&regex, line, 9, line_match, 0))!=0){
311	+	continue;
312	+	}
313	+
314	+	if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
315	+	return NULL;
316	+	}
317	+	network_stat_ptr=network_stats+interfaces;
318	+
319	+	if(network_stat_ptr->interface_name!=NULL){
320	+	free(network_stat_ptr->interface_name);
321	+	}
322	+
323	+	network_stat_ptr->interface_name=sg_get_string_match(line, &line_match[1]);
324	+	network_stat_ptr->rx=sg_get_ll_match(line, &line_match[2]);
325	+	network_stat_ptr->tx=sg_get_ll_match(line, &line_match[5]);
326	+	network_stat_ptr->ipackets=sg_get_ll_match(line, &line_match[3]);
327	+	network_stat_ptr->opackets=sg_get_ll_match(line, &line_match[6]);
328	+	network_stat_ptr->ierrors=sg_get_ll_match(line, &line_match[4]);
329	+	network_stat_ptr->oerrors=sg_get_ll_match(line, &line_match[7]);
330	+	network_stat_ptr->collisions=sg_get_ll_match(line, &line_match[8]);
331	+	network_stat_ptr->systime=time(NULL);
332	+
333	+	interfaces++;
334	+	}
335	+	fclose(f);
336	+	regfree(&regex);
337	+
338	+	#endif
339	+
340	+	#ifdef CYGWIN
341	+	sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
342	+	return NULL;
343	+	#endif
344	+	#ifdef HPUX
345	+	sg_set_error(SG_ERROR_UNSUPPORTED, "HP-UX");
346	+	return NULL;
347	+	#endif
348	+
349	+	#ifdef WIN32
350	+	interfaces = 0;
351	+
352	+	if((if_table = win32_get_devices()) == NULL) {
353	+	sg_set_error(SG_ERROR_DEVICES, "network");
354	+	return NULL;
355	+	}
356	+
357	+	if(VECTOR_RESIZE(network_stats, if_table->dwNumEntries) < 0) {
358	+	free(if_table);
359	+	return NULL;
360	+	}
361	+
362	+	for (i=0; i<if_table->dwNumEntries; i++) {
363	+	network_stat_ptr=network_stats+i;
364	+	if_row = if_table->table[i];
365	+
366	+	if(sg_update_string(&network_stat_ptr->interface_name,
367	+	if_row.bDescr) < 0) {
368	+	free(if_table);
369	+	return NULL;
370	+	}
371	+	network_stat_ptr->tx = if_row.dwOutOctets;
372	+	network_stat_ptr->rx = if_row.dwInOctets;
373	+	network_stat_ptr->ipackets = if_row.dwInUcastPkts + if_row.dwInNUcastPkts;
374	+	network_stat_ptr->opackets = if_row.dwOutUcastPkts + if_row.dwOutNUcastPkts;
375	+	network_stat_ptr->ierrors = if_row.dwInErrors;
376	+	network_stat_ptr->oerrors = if_row.dwOutErrors;
377	+	network_stat_ptr->collisions = 0; /* can't do that */
378	+	network_stat_ptr->systime = time(NULL);
379	+
380	+	interfaces++;
381	+	}
382	+	free(if_table);
383	+
384	+	/* Please say there's a nicer way to do this...  If windows has two (or
385	+	* more) identical network cards, GetIfTable returns them with the same
386	+	* name, not like in Device Manager where the other has a #2 etc after
387	+	* it. So, add the #number here. Should we be doing this? Or should the
388	+	* end programs be dealing with duplicate names? Currently breaks
389	+	* watch.pl in rrdgraphing. But Unix does not have the issue of
390	+	* duplicate net device names.
391	+	*/
392	+	for (i=0; i<interfaces; i++) {
393	+	no = 2;
394	+	for(j=i+1; j<interfaces; j++) {
395	+	network_stat_ptr=network_stats+j;
396	+	if(strcmp(network_stats[i].interface_name,
397	+	network_stat_ptr->interface_name) == 0) {
398	+	if(snprintf(buf, sizeof(buf), " #%d", no) < 0) {
399	+	break;
400	+	}
401	+	if(sg_concat_string(&network_stat_ptr->interface_name, buf) != 0) {
402	+	return NULL;
403	+	}
404	+
405	+	no++;
406	+	}
407	+	}
408	+	}
409	+	#endif
410	+
411		*entries=interfaces;
412
413		return network_stats;
414	+	}
415
416	+	static long long transfer_diff(long long new, long long old){
417	+	#if defined(SOL7) || defined(LINUX) || defined(FREEBSD) || defined(DFBSD) || defined(OPENBSD) || defined(WIN32)
418	+	/* 32-bit quantities, so we must explicitly deal with wraparound. */
419	+	#define MAXVAL 0x100000000LL
420	+	if (new >= old) {
421	+	return new - old;
422	+	} else {
423	+	return MAXVAL + new - old;
424	+	}
425	+	#else
426	+	/* 64-bit quantities, so plain subtraction works. */
427	+	return new - old;
428	+	#endif
429	+	}
430	+
431	+	sg_network_io_stats *sg_get_network_io_stats_diff(int *entries) {
432	+	VECTOR_DECLARE_STATIC(diff, sg_network_io_stats, 1,
433	+	network_stat_init, network_stat_destroy);
434	+	sg_network_io_stats *src = NULL, *dest;
435	+	int i, j, diff_count, new_count;
436	+
437	+	if (network_stats == NULL) {
438	+	/* No previous stats, so we can't calculate a difference. */
439	+	return sg_get_network_io_stats(entries);
440	+	}
441	+
442	+	/* Resize the results array to match the previous stats. */
443	+	diff_count = VECTOR_SIZE(network_stats);
444	+	if (VECTOR_RESIZE(diff, diff_count) < 0) {
445	+	return NULL;
446	+	}
447	+
448	+	/* Copy the previous stats into the result. */
449	+	for (i = 0; i < diff_count; i++) {
450	+	src = &network_stats[i];
451	+	dest = &diff[i];
452	+
453	+	if (sg_update_string(&dest->interface_name,
454	+	src->interface_name) < 0) {
455	+	return NULL;
456	+	}
457	+	dest->rx = src->rx;
458	+	dest->tx = src->tx;
459	+	dest->ipackets = src->ipackets;
460	+	dest->opackets = src->opackets;
461	+	dest->ierrors = src->ierrors;
462	+	dest->oerrors = src->oerrors;
463	+	dest->collisions = src->collisions;
464	+	dest->systime = src->systime;
465	+	}
466	+
467	+	/* Get a new set of stats. */
468	+	if (sg_get_network_io_stats(&new_count) == NULL) {
469	+	return NULL;
470	+	}
471	+
472	+	/* For each previous stat... */
473	+	for (i = 0; i < diff_count; i++) {
474	+	dest = &diff[i];
475	+
476	+	/* ... find the corresponding new stat ... */
477	+	for (j = 0; j < new_count; j++) {
478	+	/* Try the new stat in the same position first,
479	+	since that's most likely to be it. */
480	+	src = &network_stats[(i + j) % new_count];
481	+	if (strcmp(src->interface_name, dest->interface_name) == 0) {
482	+	break;
483	+	}
484	+	}
485	+	if (j == new_count) {
486	+	/* No match found. */
487	+	continue;
488	+	}
489	+
490	+	/* ... and subtract the previous stat from it to get the
491	+	difference. */
492	+	dest->rx = transfer_diff(src->rx, dest->rx);
493	+	dest->tx = transfer_diff(src->tx, dest->tx);
494	+	dest->ipackets = transfer_diff(src->ipackets, dest->ipackets);
495	+	dest->opackets = transfer_diff(src->opackets, dest->opackets);
496	+	dest->ierrors = transfer_diff(src->ierrors, dest->ierrors);
497	+	dest->oerrors = transfer_diff(src->oerrors, dest->oerrors);
498	+	dest->collisions = transfer_diff(src->collisions, dest->collisions);
499	+	dest->systime = src->systime - dest->systime;
500	+	}
501	+
502	+	*entries = diff_count;
503	+	return diff;
504	+	}
505	+
506	+	int sg_network_io_compare_name(const void *va, const void *vb) {
507	+	const sg_network_io_stats *a = (const sg_network_io_stats *)va;
508	+	const sg_network_io_stats *b = (const sg_network_io_stats *)vb;
509	+
510	+	return strcmp(a->interface_name, b->interface_name);
511	+	}
512	+
513	+	/* NETWORK INTERFACE STATS */
514	+
515	+	static void network_iface_stat_init(sg_network_iface_stats *s) {
516	+	s->interface_name = NULL;
517	+	s->speed = 0;
518	+	s->duplex = SG_IFACE_DUPLEX_UNKNOWN;
519	+	}
520	+
521	+	static void network_iface_stat_destroy(sg_network_iface_stats *s) {
522	+	free(s->interface_name);
523	+	}
524	+
525	+	sg_network_iface_stats *sg_get_network_iface_stats(int *entries){
526	+	VECTOR_DECLARE_STATIC(network_iface_stats, sg_network_iface_stats, 5,
527	+	network_iface_stat_init, network_iface_stat_destroy);
528	+	sg_network_iface_stats *network_iface_stat_ptr;
529	+	int ifaces = 0;
530	+
531	+	#ifdef SOLARIS
532	+	kstat_ctl_t *kc;
533	+	kstat_t *ksp;
534	+	kstat_named_t *knp;
535	+	int sock;
536	+	#endif
537	+	#ifdef ALLBSD
538	+	struct ifaddrs *net, *net_ptr;
539	+	struct ifmediareq ifmed;
540	+	struct ifreq ifr;
541	+	int sock;
542	+	int x;
543	+	#endif
544	+	#ifdef LINUX
545	+	FILE *f;
546	+	/* Horrible big enough, but it should be easily big enough */
547	+	char line[8096];
548	+	int sock;
549	+	#endif
550	+	#ifdef WIN32
551	+	PMIB_IFTABLE if_table;
552	+	MIB_IFROW if_row;
553	+	int i,j,no;
554	+	char buf[5];
555	+	#endif
556	+
557	+	#ifdef ALLBSD
558	+	if(getifaddrs(&net) != 0){
559	+	sg_set_error_with_errno(SG_ERROR_GETIFADDRS, NULL);
560	+	return NULL;
561	+	}
562	+
563	+	if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == 0) return NULL;
564	+
565	+	for(net_ptr=net; net_ptr!=NULL; net_ptr=net_ptr->ifa_next){
566	+	if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
567	+
568	+	if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
569	+	return NULL;
570	+	}
571	+	network_iface_stat_ptr = network_iface_stats + ifaces;
572	+
573	+	memset(&ifr, 0, sizeof(ifr));
574	+	strncpy(ifr.ifr_name, net_ptr->ifa_name, sizeof(ifr.ifr_name));
575	+
576	+	if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0){
577	+	continue;
578	+	}
579	+	if((ifr.ifr_flags & IFF_UP) != 0){
580	+	network_iface_stat_ptr->up = 1;
581	+	}else{
582	+	network_iface_stat_ptr->up = 0;
583	+	}
584	+
585	+	if (sg_update_string(&network_iface_stat_ptr->interface_name,
586	+	net_ptr->ifa_name) < 0) {
587	+	return NULL;
588	+	}
589	+
590	+	network_iface_stat_ptr->speed = 0;
591	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
592	+	ifaces++;
593	+
594	+	memset(&ifmed, 0, sizeof(struct ifmediareq));
595	+	sg_strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
596	+	if(ioctl(sock, SIOCGIFMEDIA, (caddr_t)&ifmed) == -1){
597	+	/* Not all interfaces support the media ioctls. */
598	+	continue;
599	+	}
600	+
601	+	/* We may need to change this if we start doing wireless devices too */
602	+	if( (ifmed.ifm_active | IFM_ETHER) != ifmed.ifm_active ){
603	+	/* Not a ETHER device */
604	+	continue;
605	+	}
606	+
607	+	/* Only intrested in the first 4 bits)  - Assuming only ETHER devices */
608	+	x = ifmed.ifm_active & 0x0f;
609	+	switch(x){
610	+	/* 10 Mbit connections. Speedy :) */
611	+	case(IFM_10_T):
612	+	case(IFM_10_2):
613	+	case(IFM_10_5):
614	+	case(IFM_10_STP):
615	+	case(IFM_10_FL):
616	+	network_iface_stat_ptr->speed = 10;
617	+	break;
618	+	/* 100 Mbit conneections */
619	+	case(IFM_100_TX):
620	+	case(IFM_100_FX):
621	+	case(IFM_100_T4):
622	+	case(IFM_100_VG):
623	+	case(IFM_100_T2):
624	+	network_iface_stat_ptr->speed = 100;
625	+	break;
626	+	/* 1000 Mbit connections */
627	+	case(IFM_1000_SX):
628	+	case(IFM_1000_LX):
629	+	case(IFM_1000_CX):
630	+	#if defined(IFM_1000_TX) && !defined(OPENBSD)
631	+	case(IFM_1000_TX): /* FreeBSD 4 and others (but NOT OpenBSD)? */
632	+	#endif
633	+	#ifdef IFM_1000_FX
634	+	case(IFM_1000_FX): /* FreeBSD 4 */
635	+	#endif
636	+	#ifdef IFM_1000_T
637	+	case(IFM_1000_T): /* FreeBSD 5 */
638	+	#endif
639	+	network_iface_stat_ptr->speed = 1000;
640	+	break;
641	+	/* We don't know what it is */
642	+	default:
643	+	network_iface_stat_ptr->speed = 0;
644	+	break;
645	+	}
646	+
647	+	if( (ifmed.ifm_active | IFM_FDX) == ifmed.ifm_active ){
648	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_FULL;
649	+	}else if( (ifmed.ifm_active | IFM_HDX) == ifmed.ifm_active ){
650	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_HALF;
651	+	}else{
652	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
653	+	}
654	+
655	+	}
656	+	freeifaddrs(net);
657	+	close(sock);
658	+	#endif
659	+
660	+	#ifdef SOLARIS
661	+	if ((kc = kstat_open()) == NULL) {
662	+	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
663	+	return NULL;
664	+	}
665	+
666	+	if ((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP)) < 0) {
667	+	sg_set_error_with_errno(SG_ERROR_SOCKET, NULL);
668	+	return NULL;
669	+	}
670	+
671	+	for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
672	+	if (strcmp(ksp->ks_class, "net") == 0) {
673	+	struct ifreq ifr;
674	+
675	+	kstat_read(kc, ksp, NULL);
676	+
677	+	strncpy(ifr.ifr_name, ksp->ks_name, sizeof ifr.ifr_name);
678	+	if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
679	+	/* Not a network interface. */
680	+	continue;
681	+	}
682	+
683	+	if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
684	+	return NULL;
685	+	}
686	+	network_iface_stat_ptr = network_iface_stats + ifaces;
687	+	ifaces++;
688	+
689	+	if (sg_update_string(&network_iface_stat_ptr->interface_name,
690	+	ksp->ks_name) < 0) {
691	+	return NULL;
692	+	}
693	+
694	+	if ((ifr.ifr_flags & IFF_UP) != 0) {
695	+	network_iface_stat_ptr->up = 1;
696	+	} else {
697	+	network_iface_stat_ptr->up = 0;
698	+	}
699	+
700	+	if ((knp = kstat_data_lookup(ksp, "ifspeed")) != NULL) {
701	+	network_iface_stat_ptr->speed = knp->value.ui64 / (1000 * 1000);
702	+	} else {
703	+	network_iface_stat_ptr->speed = 0;
704	+	}
705	+
706	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
707	+	if ((knp = kstat_data_lookup(ksp, "link_duplex")) != NULL) {
708	+	switch (knp->value.ui32) {
709	+	case 1:
710	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_HALF;
711	+	break;
712	+	case 2:
713	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_FULL;
714	+	break;
715	+	}
716	+	}
717	+	}
718	+	}
719	+
720	+	close(sock);
721	+	kstat_close(kc);
722	+	#endif
723	+	#ifdef LINUX
724	+	f = fopen("/proc/net/dev", "r");
725	+	if(f == NULL){
726	+	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/net/dev");
727	+	return NULL;
728	+	}
729	+
730	+	/* Setup stuff so we can do the ioctl to get the info */
731	+	if((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0){
732	+	sg_set_error_with_errno(SG_ERROR_SOCKET, NULL);
733	+	return NULL;
734	+	}
735	+
736	+	/* Ignore first 2 lines.. Just headings */
737	+	if((fgets(line, sizeof(line), f)) == NULL) {
738	+	sg_set_error(SG_ERROR_PARSE, NULL);
739	+	return NULL;
740	+	}
741	+	if((fgets(line, sizeof(line), f)) == NULL) {
742	+	sg_set_error(SG_ERROR_PARSE, NULL);
743	+	return NULL;
744	+	}
745	+
746	+	while((fgets(line, sizeof(line), f)) != NULL){
747	+	char *name, *ptr;
748	+	struct ifreq ifr;
749	+	struct ethtool_cmd ethcmd;
750	+	int err;
751	+
752	+	/* Get the interface name */
753	+	ptr = strchr(line, ':');
754	+	if (ptr == NULL) continue;
755	+	*ptr='\0';
756	+	name = line;
757	+	while(isspace(*(name))){
758	+	name++;
759	+	}
760	+
761	+	memset(&ifr, 0, sizeof ifr);
762	+	strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
763	+
764	+	if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
765	+	continue;
766	+	}
767	+
768	+	/* We have a good interface to add */
769	+	if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
770	+	return NULL;
771	+	}
772	+	network_iface_stat_ptr = network_iface_stats + ifaces;
773	+
774	+	if (sg_update_string(&network_iface_stat_ptr->interface_name,
775	+	name) < 0) {
776	+	return NULL;
777	+	}
778	+	if ((ifr.ifr_flags & IFF_UP) != 0) {
779	+	network_iface_stat_ptr->up = 1;
780	+	} else {
781	+	network_iface_stat_ptr->up = 0;
782	+	}
783	+
784	+	memset(&ethcmd, 0, sizeof ethcmd);
785	+	ethcmd.cmd = ETHTOOL_GSET;
786	+	ifr.ifr_data = (caddr_t) &ethcmd;
787	+
788	+	err = ioctl(sock, SIOCETHTOOL, &ifr);
789	+	if (err == 0) {
790	+	network_iface_stat_ptr->speed = ethcmd.speed;
791	+
792	+	switch (ethcmd.duplex) {
793	+	case DUPLEX_FULL:
794	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_FULL;
795	+	break;
796	+	case DUPLEX_HALF:
797	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_HALF;
798	+	break;
799	+	default:
800	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
801	+	}
802	+	} else {
803	+	/* Not all interfaces support the ethtool ioctl. */
804	+	network_iface_stat_ptr->speed = 0;
805	+	network_iface_stat_ptr->duplex = SG_IFACE_DUPLEX_UNKNOWN;
806	+	}
807	+
808	+	ifaces++;
809	+	}
810	+	close(sock);
811	+	fclose(f);
812	+	#endif
813	+	#ifdef CYGWIN
814	+	sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
815	+	return NULL;
816	+	#endif
817	+	#ifdef HPUX
818	+	sg_set_error(SG_ERROR_UNSUPPORTED, "HP-UX");
819	+	return NULL;
820	+	#endif
821	+	#ifdef WIN32
822	+	ifaces = 0;
823	+
824	+	if((if_table = win32_get_devices()) == NULL) {
825	+	sg_set_error(SG_ERROR_DEVICES, "network interfaces");
826	+	return NULL;
827	+	}
828	+
829	+	if(VECTOR_RESIZE(network_iface_stats, if_table->dwNumEntries) < 0) {
830	+	free(if_table);
831	+	return NULL;
832	+	}
833	+
834	+	for(i=0; i<if_table->dwNumEntries; i++) {
835	+	network_iface_stat_ptr=network_iface_stats+i;
836	+	if_row = if_table->table[i];
837	+
838	+	if(sg_update_string(&network_iface_stat_ptr->interface_name,
839	+	if_row.bDescr) < 0) {
840	+	free(if_table);
841	+	return NULL;
842	+	}
843	+	network_iface_stat_ptr->speed = if_row.dwSpeed /1000000;
844	+
845	+	if((if_row.dwOperStatus == MIB_IF_OPER_STATUS_CONNECTED ||
846	+	if_row.dwOperStatus ==
847	+	MIB_IF_OPER_STATUS_OPERATIONAL) &&
848	+	if_row.dwAdminStatus == 1) {
849	+	network_iface_stat_ptr->up = 1;
850	+	} else {
851	+	network_iface_stat_ptr->up = 0;
852	+	}
853	+
854	+	ifaces++;
855	+	}
856	+	free(if_table);
857	+
858	+	/* again with the renumbering */
859	+	for (i=0; i<ifaces; i++) {
860	+	no = 2;
861	+	for(j=i+1; j<ifaces; j++) {
862	+	network_iface_stat_ptr=network_iface_stats+j;
863	+	if(strcmp(network_iface_stats[i].interface_name,
864	+	network_iface_stat_ptr->interface_name) == 0) {
865	+	if(snprintf(buf, sizeof(buf), " #%d", no) < 0) {
866	+	break;
867	+	}
868	+	if(sg_concat_string(&network_iface_stat_ptr->interface_name, buf) != 0) {
869	+	return NULL;
870	+	}
871	+
872	+	no++;
873	+	}
874	+	}
875	+	}
876	+	#endif
877	+
878	+	#ifdef SG_ENABLE_DEPRECATED
879	+	network_iface_stat_ptr->dup = network_iface_stat_ptr->duplex;
880	+	#endif
881	+
882	+	*entries = ifaces;
883	+	return network_iface_stats;
884	+	}
885	+
886	+	int sg_network_iface_compare_name(const void *va, const void *vb) {
887	+	const sg_network_iface_stats *a = (const sg_network_iface_stats *)va;
888	+	const sg_network_iface_stats *b = (const sg_network_iface_stats *)vb;
889	+
890	+	return strcmp(a->interface_name, b->interface_name);
891		}
892

Diff Legend

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