ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/network_stats.c
Revision: 1.65
Committed: Wed Apr 7 14:53:40 2004 UTC (20 years, 1 month ago) by tdb
Content type: text/plain
Branch: MAIN
Changes since 1.64: +55 -55 lines
Log Message:
Whitespace tidyup - change spaces to tabs.

File Contents

# Content
1 /*
2 * i-scream libstatgrab
3 * http://www.i-scream.org
4 * Copyright (C) 2000-2004 i-scream
5 *
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 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 GNU
14 * Lesser General Public License for more details.
15 *
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: network_stats.c,v 1.64 2004/04/06 14:52:58 tdb Exp $
22 */
23
24 #ifdef HAVE_CONFIG_H
25 #include "config.h"
26 #endif
27
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 <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 <regex.h>
49 #include <sys/ioctl.h>
50 #include <sys/socket.h>
51 #include <net/if.h>
52 #include <ctype.h>
53 /* Stuff which could be defined by defining KERNEL, but
54 * that would be a bad idea, so we'll just declare it here
55 */
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
74 static void network_stat_init(sg_network_io_stats *s) {
75 s->interface_name = NULL;
76 s->tx = 0;
77 s->rx = 0;
78 s->ipackets = 0;
79 s->opackets = 0;
80 s->ierrors = 0;
81 s->oerrors = 0;
82 s->collisions = 0;
83 }
84
85 static void network_stat_destroy(sg_network_io_stats *s) {
86 free(s->interface_name);
87 }
88
89 VECTOR_DECLARE_STATIC(network_stats, sg_network_io_stats, 5,
90 network_stat_init, network_stat_destroy);
91
92 sg_network_io_stats *sg_get_network_io_stats(int *entries){
93 int interfaces;
94 sg_network_io_stats *network_stat_ptr;
95
96 #ifdef SOLARIS
97 kstat_ctl_t *kc;
98 kstat_t *ksp;
99 kstat_named_t *knp;
100 #endif
101
102 #ifdef LINUX
103 FILE *f;
104 /* Horrible big enough, but it should be easily big enough */
105 char line[8096];
106 regex_t regex;
107 regmatch_t line_match[9];
108 #endif
109 #ifdef ALLBSD
110 struct ifaddrs *net, *net_ptr;
111 struct if_data *net_data;
112 #endif
113
114 #ifdef ALLBSD
115 if(getifaddrs(&net) != 0){
116 return NULL;
117 }
118
119 interfaces=0;
120
121 for(net_ptr=net;net_ptr!=NULL;net_ptr=net_ptr->ifa_next){
122 if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
123
124 if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
125 return NULL;
126 }
127 network_stat_ptr=network_stats+interfaces;
128
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;
136 network_stat_ptr->ipackets=net_data->ifi_ipackets;
137 network_stat_ptr->opackets=net_data->ifi_opackets;
138 network_stat_ptr->ierrors=net_data->ifi_ierrors;
139 network_stat_ptr->oerrors=net_data->ifi_oerrors;
140 network_stat_ptr->collisions=net_data->ifi_collisions;
141 network_stat_ptr->systime=time(NULL);
142 interfaces++;
143 }
144 freeifaddrs(net);
145 #endif
146
147 #ifdef SOLARIS
148 if ((kc = kstat_open()) == NULL) {
149 return NULL;
150 }
151
152 interfaces=0;
153
154 for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
155 if (!strcmp(ksp->ks_class, "net")) {
156 kstat_read(kc, ksp, NULL);
157
158 #ifdef SOL7
159 #define LRX "rbytes"
160 #define LTX "obytes"
161 #define LIPACKETS "ipackets"
162 #define LOPACKETS "opackets"
163 #define VALTYPE value.ui32
164 #else
165 #define LRX "rbytes64"
166 #define LTX "obytes64"
167 #define LIPACKETS "ipackets64"
168 #define LOPACKETS "opackets64"
169 #define VALTYPE value.ui64
170 #endif
171
172 /* Read rx */
173 if((knp=kstat_data_lookup(ksp, LRX))==NULL){
174 /* This is a network interface, but it doesn't
175 * have the rbytes/obytes values; for instance,
176 * the loopback devices have this behaviour
177 * (although they do track packets in/out). */
178 /* FIXME: Show packet counts when byte counts
179 * not available. */
180 continue;
181 }
182
183 /* Create new network_stats */
184 if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
185 return NULL;
186 }
187 network_stat_ptr=network_stats+interfaces;
188
189 /* Finish reading rx */
190 network_stat_ptr->rx=knp->VALTYPE;
191
192 /* Read tx */
193 if((knp=kstat_data_lookup(ksp, LTX))==NULL){
194 continue;
195 }
196 network_stat_ptr->tx=knp->VALTYPE;
197
198 /* Read ipackets */
199 if((knp=kstat_data_lookup(ksp, LIPACKETS))==NULL){
200 continue;
201 }
202 network_stat_ptr->ipackets=knp->VALTYPE;
203
204 /* Read opackets */
205 if((knp=kstat_data_lookup(ksp, LOPACKETS))==NULL){
206 continue;
207 }
208 network_stat_ptr->opackets=knp->VALTYPE;
209
210 /* Read ierrors */
211 if((knp=kstat_data_lookup(ksp, "ierrors"))==NULL){
212 continue;
213 }
214 network_stat_ptr->ierrors=knp->value.ui32;
215
216 /* Read oerrors */
217 if((knp=kstat_data_lookup(ksp, "oerrors"))==NULL){
218 continue;
219 }
220 network_stat_ptr->oerrors=knp->value.ui32;
221
222 /* Read collisions */
223 if((knp=kstat_data_lookup(ksp, "collisions"))==NULL){
224 continue;
225 }
226 network_stat_ptr->collisions=knp->value.ui32;
227
228 /* Read interface name */
229 if (sg_update_string(&network_stat_ptr->interface_name,
230 ksp->ks_name) < 0) {
231 return NULL;
232 }
233
234 /* Store systime */
235 network_stat_ptr->systime=time(NULL);
236
237 interfaces++;
238 }
239 }
240
241 kstat_close(kc);
242 #endif
243 #ifdef LINUX
244 f=fopen("/proc/net/dev", "r");
245 if(f==NULL){
246 return NULL;
247 }
248 /* read the 2 lines.. Its the title, so we dont care :) */
249 fgets(line, sizeof(line), f);
250 fgets(line, sizeof(line), f);
251
252
253 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){
254 return NULL;
255 }
256
257 interfaces=0;
258
259 while((fgets(line, sizeof(line), f)) != NULL){
260 if((regexec(&regex, line, 9, line_match, 0))!=0){
261 continue;
262 }
263
264 if (VECTOR_RESIZE(network_stats, interfaces + 1) < 0) {
265 return NULL;
266 }
267 network_stat_ptr=network_stats+interfaces;
268
269 if(network_stat_ptr->interface_name!=NULL){
270 free(network_stat_ptr->interface_name);
271 }
272
273 network_stat_ptr->interface_name=sg_get_string_match(line, &line_match[1]);
274 network_stat_ptr->rx=sg_get_ll_match(line, &line_match[2]);
275 network_stat_ptr->tx=sg_get_ll_match(line, &line_match[5]);
276 network_stat_ptr->ipackets=sg_get_ll_match(line, &line_match[3]);
277 network_stat_ptr->opackets=sg_get_ll_match(line, &line_match[6]);
278 network_stat_ptr->ierrors=sg_get_ll_match(line, &line_match[4]);
279 network_stat_ptr->oerrors=sg_get_ll_match(line, &line_match[7]);
280 network_stat_ptr->collisions=sg_get_ll_match(line, &line_match[8]);
281 network_stat_ptr->systime=time(NULL);
282
283 interfaces++;
284 }
285 fclose(f);
286 regfree(&regex);
287
288 #endif
289
290 #ifdef CYGWIN
291 return NULL;
292 #endif
293
294 *entries=interfaces;
295
296 return network_stats;
297 }
298
299 static long long transfer_diff(long long new, long long old){
300 #if defined(SOL7) || defined(LINUX) || defined(FREEBSD) || defined(DFBSD) || defined(OPENBSD)
301 /* 32-bit quantities, so we must explicitly deal with wraparound. */
302 #define MAXVAL 0x100000000LL
303 if (new >= old) {
304 return new - old;
305 } else {
306 return MAXVAL + new - old;
307 }
308 #else
309 /* 64-bit quantities, so plain subtraction works. */
310 return new - old;
311 #endif
312 }
313
314 sg_network_io_stats *sg_get_network_io_stats_diff(int *entries) {
315 VECTOR_DECLARE_STATIC(diff, sg_network_io_stats, 1,
316 network_stat_init, network_stat_destroy);
317 sg_network_io_stats *src = NULL, *dest;
318 int i, j, diff_count, new_count;
319
320 if (network_stats == NULL) {
321 /* No previous stats, so we can't calculate a difference. */
322 return sg_get_network_io_stats(entries);
323 }
324
325 /* Resize the results array to match the previous stats. */
326 diff_count = VECTOR_SIZE(network_stats);
327 if (VECTOR_RESIZE(diff, diff_count) < 0) {
328 return NULL;
329 }
330
331 /* Copy the previous stats into the result. */
332 for (i = 0; i < diff_count; i++) {
333 src = &network_stats[i];
334 dest = &diff[i];
335
336 if (sg_update_string(&dest->interface_name,
337 src->interface_name) < 0) {
338 return NULL;
339 }
340 dest->rx = src->rx;
341 dest->tx = src->tx;
342 dest->ipackets = src->ipackets;
343 dest->opackets = src->opackets;
344 dest->ierrors = src->ierrors;
345 dest->oerrors = src->oerrors;
346 dest->collisions = src->collisions;
347 dest->systime = src->systime;
348 }
349
350 /* Get a new set of stats. */
351 if (sg_get_network_io_stats(&new_count) == NULL) {
352 return NULL;
353 }
354
355 /* For each previous stat... */
356 for (i = 0; i < diff_count; i++) {
357 dest = &diff[i];
358
359 /* ... find the corresponding new stat ... */
360 for (j = 0; j < new_count; j++) {
361 /* Try the new stat in the same position first,
362 since that's most likely to be it. */
363 src = &network_stats[(i + j) % new_count];
364 if (strcmp(src->interface_name, dest->interface_name) == 0) {
365 break;
366 }
367 }
368 if (j == new_count) {
369 /* No match found. */
370 continue;
371 }
372
373 /* ... and subtract the previous stat from it to get the
374 difference. */
375 dest->rx = transfer_diff(src->rx, dest->rx);
376 dest->tx = transfer_diff(src->tx, dest->tx);
377 dest->ipackets = transfer_diff(src->ipackets, dest->ipackets);
378 dest->opackets = transfer_diff(src->opackets, dest->opackets);
379 dest->ierrors = transfer_diff(src->ierrors, dest->ierrors);
380 dest->oerrors = transfer_diff(src->oerrors, dest->oerrors);
381 dest->collisions = transfer_diff(src->collisions, dest->collisions);
382 dest->systime = src->systime - dest->systime;
383 }
384
385 *entries = diff_count;
386 return diff;
387 }
388
389 /* NETWORK INTERFACE STATS */
390
391 static void network_iface_stat_init(sg_network_iface_stats *s) {
392 s->interface_name = NULL;
393 s->speed = 0;
394 s->dup = SG_IFACE_DUPLEX_UNKNOWN;
395 }
396
397 static void network_iface_stat_destroy(sg_network_iface_stats *s) {
398 free(s->interface_name);
399 }
400
401 sg_network_iface_stats *sg_get_network_iface_stats(int *entries){
402 VECTOR_DECLARE_STATIC(network_iface_stats, sg_network_iface_stats, 5,
403 network_iface_stat_init, network_iface_stat_destroy);
404 sg_network_iface_stats *network_iface_stat_ptr;
405 int ifaces = 0;
406
407 #ifdef SOLARIS
408 kstat_ctl_t *kc;
409 kstat_t *ksp;
410 kstat_named_t *knp;
411 int sock;
412 #endif
413 #ifdef ALLBSD
414 struct ifaddrs *net, *net_ptr;
415 struct ifmediareq ifmed;
416 struct ifreq ifr;
417 int sock;
418 int x;
419 #endif
420 #ifdef LINUX
421 FILE *f;
422 /* Horrible big enough, but it should be easily big enough */
423 char line[8096];
424 int sock;
425 #endif
426
427 #ifdef ALLBSD
428 if(getifaddrs(&net) != 0){
429 return NULL;
430 }
431
432 if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == 0) return NULL;
433
434 for(net_ptr=net; net_ptr!=NULL; net_ptr=net_ptr->ifa_next){
435 if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
436
437 if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
438 return NULL;
439 }
440 network_iface_stat_ptr = network_iface_stats + ifaces;
441
442 memset(&ifr, 0, sizeof(ifr));
443 strncpy(ifr.ifr_name, net_ptr->ifa_name, sizeof(ifr.ifr_name));
444
445 if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0){
446 continue;
447 }
448 if((ifr.ifr_flags & IFF_UP) != 0){
449 network_iface_stat_ptr->up = 1;
450 }else{
451 network_iface_stat_ptr->up = 0;
452 }
453
454 if (sg_update_string(&network_iface_stat_ptr->interface_name,
455 net_ptr->ifa_name) < 0) {
456 return NULL;
457 }
458
459 network_iface_stat_ptr->speed = 0;
460 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
461 ifaces++;
462
463 memset(&ifmed, 0, sizeof(struct ifmediareq));
464 sg_strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
465 if(ioctl(sock, SIOCGIFMEDIA, (caddr_t)&ifmed) == -1){
466 /* Not all interfaces support the media ioctls. */
467 continue;
468 }
469
470 /* We may need to change this if we start doing wireless devices too */
471 if( (ifmed.ifm_active | IFM_ETHER) != ifmed.ifm_active ){
472 /* Not a ETHER device */
473 continue;
474 }
475
476 /* Only intrested in the first 4 bits) - Assuming only ETHER devices */
477 x = ifmed.ifm_active & 0x0f;
478 switch(x){
479 /* 10 Mbit connections. Speedy :) */
480 case(IFM_10_T):
481 case(IFM_10_2):
482 case(IFM_10_5):
483 case(IFM_10_STP):
484 case(IFM_10_FL):
485 network_iface_stat_ptr->speed = 10;
486 break;
487 /* 100 Mbit conneections */
488 case(IFM_100_TX):
489 case(IFM_100_FX):
490 case(IFM_100_T4):
491 case(IFM_100_VG):
492 case(IFM_100_T2):
493 network_iface_stat_ptr->speed = 100;
494 break;
495 /* 1000 Mbit connections */
496 case(IFM_1000_SX):
497 case(IFM_1000_LX):
498 case(IFM_1000_CX):
499 #if defined(IFM_1000_TX) && !defined(OPENBSD)
500 case(IFM_1000_TX): /* FreeBSD 4 and others (but NOT OpenBSD)? */
501 #endif
502 #ifdef IFM_1000_FX
503 case(IFM_1000_FX): /* FreeBSD 4 */
504 #endif
505 #ifdef IFM_1000_T
506 case(IFM_1000_T): /* FreeBSD 5 */
507 #endif
508 network_iface_stat_ptr->speed = 1000;
509 break;
510 /* We don't know what it is */
511 default:
512 network_iface_stat_ptr->speed = 0;
513 break;
514 }
515
516 if( (ifmed.ifm_active | IFM_FDX) == ifmed.ifm_active ){
517 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
518 }else if( (ifmed.ifm_active | IFM_HDX) == ifmed.ifm_active ){
519 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
520 }else{
521 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
522 }
523
524 }
525 freeifaddrs(net);
526 close(sock);
527 #endif
528
529 #ifdef SOLARIS
530 if ((kc = kstat_open()) == NULL) {
531 return NULL;
532 }
533
534 if ((sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP)) < 0) {
535 return NULL;
536 }
537
538 for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
539 if (!strcmp(ksp->ks_class, "net")) {
540 struct ifreq ifr;
541
542 kstat_read(kc, ksp, NULL);
543
544 strncpy(ifr.ifr_name, ksp->ks_name, sizeof ifr.ifr_name);
545 if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
546 /* Not a network interface. */
547 continue;
548 }
549
550 if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
551 return NULL;
552 }
553 network_iface_stat_ptr = network_iface_stats + ifaces;
554 ifaces++;
555
556 if (sg_update_string(&network_iface_stat_ptr->interface_name,
557 ksp->ks_name) < 0) {
558 return NULL;
559 }
560
561 if ((ifr.ifr_flags & IFF_UP) != 0) {
562 network_iface_stat_ptr->up = 1;
563 } else {
564 network_iface_stat_ptr->up = 1;
565 }
566
567 if ((knp = kstat_data_lookup(ksp, "ifspeed")) != NULL) {
568 network_iface_stat_ptr->speed = knp->value.ui64 / (1000 * 1000);
569 } else {
570 network_iface_stat_ptr->speed = 0;
571 }
572
573 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
574 if ((knp = kstat_data_lookup(ksp, "link_duplex")) != NULL) {
575 switch (knp->value.ui32) {
576 case 1:
577 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
578 break;
579 case 2:
580 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
581 break;
582 }
583 }
584 }
585 }
586
587 close(sock);
588 kstat_close(kc);
589 #endif
590 #ifdef LINUX
591 f = fopen("/proc/net/dev", "r");
592 if(f == NULL){
593 return NULL;
594 }
595
596 /* Setup stuff so we can do the ioctl to get the info */
597 if((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0){
598 return NULL;
599 }
600
601 /* Ignore first 2 lines.. Just headings */
602 if((fgets(line, sizeof(line), f)) == NULL) return NULL;
603 if((fgets(line, sizeof(line), f)) == NULL) return NULL;
604
605 while((fgets(line, sizeof(line), f)) != NULL){
606 char *name, *ptr;
607 struct ifreq ifr;
608 struct ethtool_cmd ethcmd;
609 int err;
610
611 /* Get the interface name */
612 ptr = strchr(line, ':');
613 if (ptr == NULL) continue;
614 *ptr='\0';
615 name = line;
616 while(isspace(*(name))){
617 name++;
618 }
619
620 memset(&ifr, 0, sizeof ifr);
621 strncpy(ifr.ifr_name, name, sizeof ifr.ifr_name);
622
623 if (ioctl(sock, SIOCGIFFLAGS, &ifr) < 0) {
624 continue;
625 }
626
627 /* We have a good interface to add */
628 if (VECTOR_RESIZE(network_iface_stats, ifaces + 1) < 0) {
629 return NULL;
630 }
631 network_iface_stat_ptr = network_iface_stats + ifaces;
632
633 if (sg_update_string(&network_iface_stat_ptr->interface_name,
634 name) < 0) {
635 return NULL;
636 }
637 if ((ifr.ifr_flags & IFF_UP) != 0) {
638 network_iface_stat_ptr->up = 1;
639 } else {
640 network_iface_stat_ptr->up = 0;
641 }
642
643 memset(&ethcmd, 0, sizeof ethcmd);
644 ethcmd.cmd = ETHTOOL_GSET;
645 ifr.ifr_data = (caddr_t) &ethcmd;
646
647 err = ioctl(sock, SIOCETHTOOL, &ifr);
648 if (err == 0) {
649 network_iface_stat_ptr->speed = ethcmd.speed;
650
651 switch (ethcmd.duplex) {
652 case 0x00:
653 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_FULL;
654 break;
655 case 0x01:
656 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_HALF;
657 break;
658 default:
659 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
660 }
661 } else {
662 /* Not all interfaces support the ethtool ioctl. */
663 network_iface_stat_ptr->speed = 0;
664 network_iface_stat_ptr->dup = SG_IFACE_DUPLEX_UNKNOWN;
665 }
666
667 ifaces++;
668 }
669 close(sock);
670 fclose(f);
671 #endif
672 *entries = ifaces;
673 return network_iface_stats;
674 }
675