ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/network_stats.c
(Generate patch)

Comparing projects/libstatgrab/src/libstatgrab/network_stats.c (file contents):
Revision 1.25 by pajs, Fri Jan 23 23:23:33 2004 UTC vs.
Revision 1.47 by tdb, Sat Mar 6 21:49:13 2004 UTC

# Line 32 | 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 55 | 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 129 | 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 158 | 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 170 | 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){
173                                /* Not a network interface, so skip to the next entry */
208                                  continue;
209                          }
210                          network_stat_ptr->tx=knp->VALTYPE;
# Line 238 | 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 282 | Line 316 | network_stat_t *get_network_stats_diff(int *entries) {
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  
# Line 312 | Line 351 | network_stat_t *get_network_stats_diff(int *entries) {
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  
# Line 325 | Line 369 | void network_iface_stat_init(int start, int end, netwo
369          for(net_stats+=start; start<end; start++){
370                  net_stats->interface_name=NULL;
371                  net_stats->speed=0;
372 <                net_stats->dup=NO_DUPLEX;
372 >                net_stats->dup=UNKNOWN_DUPLEX;
373                  net_stats++;
374          }
375   }
# Line 360 | Line 404 | network_iface_stat_t *get_network_iface_stats(int *ent
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;
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 SOLARIS
430 <        if ((kc = kstat_open()) == NULL) {
429 > #ifdef ALLBSD
430 >        if(getifaddrs(&net) != 0){
431                  return NULL;
432          }
433  
434 <        ifaces=0;
434 >        if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == 0) return NULL;
435  
436 <        for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
437 <                if (!strcmp(ksp->ks_class, "net")) {
438 <                        kstat_read(kc, ksp, NULL);
439 <                        if((knp=kstat_data_lookup(ksp, "ifspeed"))==NULL){
440 <                                /* Not a network interface, so skip to the next entry */
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 >                        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 <                        network_iface_stats=network_iface_stat_malloc((ifaces+1), &sizeof_network_iface_stats, network_iface_stats);
551 <                        if(network_iface_stats==NULL){
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 + interfaces;
556 <                        network_iface_stat_ptr->speed = knp->value.ui64 / (1000*1000);
555 >                        network_iface_stat_ptr = network_iface_stats + ifaces;
556 >                        ifaces++;
557  
558 <                        if((knp=kstat_data_lookup(ksp, "link_duplex"))==NULL){
559 <                                /* Not a network interface, so skip to the next entry */
560 <                                continue;
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->value.ui64 == 0){
569 <                                network_iface_stat_ptr->dup = FULL_DUPLEX;
570 <                        }else{
571 <                                network_iface_stat_ptr->dup = HALF_DUPLEX;
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->interface_name = strdup(ksp->ks_name);
575 <                        interfaces++;
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 >                        }
585                  }
586          }
587 +
588 +        close(sock);
589          kstat_close(kc);
408        
590   #endif  
591 <        *entries = interfaces;
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 >        /* 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