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.2 by pajs, Tue Feb 18 23:23:36 2003 UTC vs.
Revision 1.13 by pajs, Mon Mar 31 13:24:12 2003 UTC

#	Line 22 | Line 22
22		#include "config.h"
23		#endif
24
25	–	#include <stdio.h>
26	–	#include "statgrab.h"
25		#include <stdlib.h>
26	+	#include <string.h>
27	+	#include "statgrab.h"
28	+	#include <time.h>
29		#ifdef SOLARIS
30		#include <kstat.h>
31		#include <sys/sysinfo.h>
31	–	#include <string.h>
32		#endif
33	+	#ifdef LINUX
34	+	#include <stdio.h>
35	+	#include <sys/types.h>
36	+	#include <regex.h>
37	+	#include "tools.h"
38	+	#endif
39
40	<	#define START_VAL 1
40	>	static network_stat_t *network_stats=NULL;
41	>	static int interfaces=0;
42
43	<	network_stat_t *network_stat_init(int num_iface, int *wmark, network_stat_t *net_stats){
37	<	int x;
38	<	network_stat_t *net_stats_ptr;
43	>	void network_stat_init(int start, int end, network_stat_t *net_stats){
44
45	<	if(*wmark==-1){
46	<	printf("new malloc\n");
47	<	if((net_stats=malloc(START_VAL * sizeof(network_stat_t)))==NULL){
45	>	for(net_stats+=start; start<end; start++){
46	>	net_stats->interface_name=NULL;
47	>	net_stats->tx=0;
48	>	net_stats->rx=0;
49	>	net_stats++;
50	>	}
51	>	}
52	>
53	>	network_stat_t *network_stat_malloc(int needed_entries, int *cur_entries, network_stat_t *net_stats){
54	>
55	>	if(net_stats==NULL){
56	>
57	>	if((net_stats=malloc(needed_entries * sizeof(network_stat_t)))==NULL){
58		return NULL;
59		}
60	<	*wmark=START_VAL;
61	<	net_stats_ptr=net_stats;
62	<	for(x=0;x<(*wmark);x++){
48	<	net_stats_ptr->interface_name=NULL;
49	<	net_stats_ptr++;
50	<	}
60	>	network_stat_init(0, needed_entries, net_stats);
61	>	*cur_entries=needed_entries;
62	>
63		return net_stats;
64		}
65	<	if(num_iface>(*wmark-1)){
66	<	if((net_stats=realloc(net_stats, (*wmark)*2 * sizeof(network_stat_t)))==NULL){
65	>
66	>
67	>	if(*cur_entries<needed_entries){
68	>	net_stats=realloc(net_stats, (sizeof(network_stat_t)*needed_entries));
69	>	if(net_stats==NULL){
70		return NULL;
71		}
72	<
73	<	*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;
72	>	network_stat_init(*cur_entries, needed_entries, net_stats);
73	>	*cur_entries=needed_entries;
74		}
75
76		return net_stats;
77		}
78
79	+
80		network_stat_t *get_network_stats(int *entries){
81	+
82	+	static int sizeof_network_stats=0;
83	+	network_stat_t *network_stat_ptr;
84	+
85	+	#ifdef SOLARIS
86		kstat_ctl_t *kc;
87		kstat_t *ksp;
88		kstat_named_t *knp;
89	<
75	<	int interfaces=0;
76	<	network_stat_t *network_stat_ptr;
77	<	static network_stat_t *network_stats=NULL;
78	<	static int watermark=-1;
89	>	#endif
90
91	+	#ifdef LINUX
92	+	FILE *f;
93	+	/* Horrible big enough, but it should be easily big enough */
94	+	char line[8096];
95	+	regex_t regex;
96	+	regmatch_t line_match[4];
97	+	#endif
98	+
99	+	#ifdef SOLARIS
100		if ((kc = kstat_open()) == NULL) {
101		return NULL;
102		}
103
104	+	interfaces=0;
105	+
106		for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
107		if (!strcmp(ksp->ks_class, "net")) {
108		kstat_read(kc, ksp, NULL);
109
110	<	if((knp=kstat_data_lookup(ksp, "rbytes64"))==NULL){
110	>	#ifdef SOL7
111	>	#define RLOOKUP "rbytes"
112	>	#define WLOOKUP "obytes"
113	>	#define VALTYPE value.ui32
114	>	#else
115	>	#define RLOOKUP "rbytes64"
116	>	#define WLOOKUP "obytes64"
117	>	#define VALTYPE value.ui64
118	>	#endif
119	>
120	>	if((knp=kstat_data_lookup(ksp, RLOOKUP))==NULL){
121		/* Not a network interface, so skip to the next entry */
122		continue;
123		}
124	<	network_stats=network_stat_init(interfaces, &watermark, network_stats);
124	>
125	>	network_stats=network_stat_malloc((interfaces+1), &sizeof_network_stats, network_stats);
126		if(network_stats==NULL){
127		return NULL;
128		}
129		network_stat_ptr=network_stats+interfaces;
130	<	network_stat_ptr->rx=knp->value.ui64;
130	>	network_stat_ptr->rx=knp->VALTYPE;
131
132	<	if((knp=kstat_data_lookup(ksp, "obytes64"))==NULL){
132	>	if((knp=kstat_data_lookup(ksp, WLOOKUP))==NULL){
133		/* Not a network interface, so skip to the next entry */
134		continue;
135		}
136	<	network_stat_ptr->tx=knp->value.ui64;
136	>	network_stat_ptr->tx=knp->VALTYPE;
137		if(network_stat_ptr->interface_name!=NULL){
138		free(network_stat_ptr->interface_name);
139		}
140		network_stat_ptr->interface_name=strdup(ksp->ks_name);
141	+
142	+	network_stat_ptr->systime=time(NULL);
143		interfaces++;
144		}
145		}
146
147		kstat_close(kc);
148	+	#endif
149	+	#ifdef LINUX
150	+	f=fopen("/proc/net/dev", "r");
151	+	if(f==NULL){
152	+	return NULL;
153	+	}
154	+	/* read the 2 lines.. Its the title, so we dont care :) */
155	+	fgets(line, sizeof(line), f);
156	+	fgets(line, sizeof(line), f);
157
158	+
159	+	if((regcomp(&regex, "^[[:space:]]*([^:]+):[[:space:]]*([[:digit:]]+)[[:space:]]+[[:digit:]]+[[:space:]]+[[:digit:]]+[[:space:]]+[[:digit:]]+[[:space:]]+[[:digit:]]+[[:space:]]+[[:digit:]]+[[:space:]]+[[:digit:]]+[[:space:]]+[[:digit:]]+[[:space:]]+([[:digit:]]+)", REG_EXTENDED))!=0){
160	+	return NULL;
161	+	}
162	+
163	+	interfaces=0;
164	+
165	+	while((fgets(line, sizeof(line), f)) != NULL){
166	+	if((regexec(&regex, line, 4, line_match, 0))!=0){
167	+	continue;
168	+	}
169	+	network_stats=network_stat_malloc((interfaces+1), &sizeof_network_stats, network_stats);
170	+	if(network_stats==NULL){
171	+	return NULL;
172	+	}
173	+	network_stat_ptr=network_stats+interfaces;
174	+
175	+	if(network_stat_ptr->interface_name!=NULL){
176	+	free(network_stat_ptr->interface_name);
177	+	}
178	+
179	+	network_stat_ptr->interface_name=get_string_match(line, &line_match[1]);
180	+	network_stat_ptr->rx=get_ll_match(line, &line_match[2]);
181	+	network_stat_ptr->tx=get_ll_match(line, &line_match[3]);
182	+	network_stat_ptr->systime=time(NULL);
183	+
184	+	interfaces++;
185	+	}
186	+	fclose(f);
187	+	regfree(&regex);
188	+
189	+	#endif
190		*entries=interfaces;
191
192		return network_stats;
193	+	}
194
195	+	long long transfer_diff(long long new, long long old){
196	+	#ifdef SOL7
197	+	#define MAXVAL 4294967296
198	+	#else
199	+	#define MAXVAL 18446744073709551616
200	+	#endif
201	+	long long result;
202	+	if(new>=old){
203	+	result = (new-old);
204	+	}else{
205	+	result = (MAXVAL+(new-old));
206	+	}
207	+
208	+	return result;
209	+
210		}
211	+
212	+	network_stat_t *get_network_stats_diff(int *entries){
213	+	static network_stat_t *network_stats_diff=NULL;
214	+	static int sizeof_net_stats_diff=0;
215	+	network_stat_t *network_stats_ptr, *network_stats_diff_ptr;
216	+	int ifaces, x, y;
217	+
218	+	if(network_stats==NULL){
219	+	network_stats_ptr=get_network_stats(&ifaces);
220	+	*entries=ifaces;
221	+	return network_stats_ptr;
222	+	}
223	+
224	+	network_stats_diff=network_stat_malloc(interfaces, &sizeof_net_stats_diff, network_stats_diff);
225	+	if(network_stats_diff==NULL){
226	+	return NULL;
227	+	}
228	+
229	+	network_stats_ptr=network_stats;
230	+	network_stats_diff_ptr=network_stats_diff;
231	+
232	+	for(ifaces=0;ifaces<interfaces;ifaces++){
233	+	if(network_stats_diff_ptr->interface_name!=NULL){
234	+	free(network_stats_diff_ptr->interface_name);
235	+	}
236	+	network_stats_diff_ptr->interface_name=strdup(network_stats_ptr->interface_name);
237	+	network_stats_diff_ptr->tx=network_stats_ptr->tx;
238	+	network_stats_diff_ptr->rx=network_stats_ptr->rx;
239	+	network_stats_diff_ptr->systime=network_stats->systime;
240	+
241	+	network_stats_ptr++;
242	+	network_stats_diff_ptr++;
243	+	}
244	+	network_stats_ptr=get_network_stats(&ifaces);
245	+	network_stats_diff_ptr=network_stats_diff;
246	+
247	+	for(x=0;x<sizeof_net_stats_diff;x++){
248	+
249	+	if((strcmp(network_stats_diff_ptr->interface_name, network_stats_ptr->interface_name))==0){
250	+	network_stats_diff_ptr->tx = transfer_diff(network_stats_ptr->tx, network_stats_diff_ptr->tx);
251	+	network_stats_diff_ptr->rx = transfer_diff(network_stats_ptr->rx, network_stats_diff_ptr->rx);
252	+	network_stats_diff_ptr->systime = network_stats_ptr->systime - network_stats_diff_ptr->systime;
253	+	}else{
254	+
255	+	network_stats_ptr=network_stats;
256	+	for(y=0;y<ifaces;y++){
257	+	if((strcmp(network_stats_diff_ptr->interface_name, network_stats_ptr->interface_name))==0){
258	+	network_stats_diff_ptr->tx = transfer_diff(network_stats_ptr->tx, network_stats_diff_ptr->tx);
259	+	network_stats_diff_ptr->rx = transfer_diff(network_stats_ptr->rx, network_stats_diff_ptr->rx);
260	+	network_stats_diff_ptr->systime = network_stats_ptr->systime - network_stats_diff_ptr->systime;
261	+	break;
262	+	}
263	+
264	+	network_stats_ptr++;
265	+	}
266	+	}
267	+
268	+	network_stats_ptr++;
269	+	network_stats_diff_ptr++;
270	+	}
271	+
272	+	*entries=sizeof_net_stats_diff;
273	+	return network_stats_diff;
274	+	}
275

Diff Legend

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