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.24 by ats, Wed Jan 21 23:46:54 2004 UTC vs.
Revision 1.32 by ats, Fri Feb 13 15:24:16 2004 UTC

# Line 37 | Line 37
37   #include <stdio.h>
38   #include <sys/types.h>
39   #include <regex.h>
40 + #include <sys/ioctl.h>
41 + #include <sys/socket.h>
42 + #include <net/if.h>
43 + #include <ctype.h>
44   #include "tools.h"
45 + /* Stuff which could be defined by defining KERNEL, but
46 + * that would be a bad idea, so we'll just declare it here
47 + */
48 + typedef __uint8_t u8;
49 + typedef __uint16_t u16;
50 + typedef __uint32_t u32;
51 + #include <linux/ethtool.h>
52 + #include <linux/sockios.h>
53 + #include <unistd.h>
54   #endif
55   #ifdef ALLBSD
56   #include <sys/types.h>
57   #include <sys/socket.h>
58   #include <ifaddrs.h>
59   #include <net/if.h>
60 + #include <net/if_media.h>
61 + #include <sys/ioctl.h>
62   #endif
63  
64   static network_stat_t *network_stats=NULL;
# Line 318 | Line 333 | network_stat_t *get_network_stats_diff(int *entries) {
333          *entries = diff_count;
334          return diff;
335   }
336 + /* NETWORK INTERFACE STATS */
337 +
338 + void network_iface_stat_init(int start, int end, network_iface_stat_t *net_stats){
339 +
340 +        for(net_stats+=start; start<end; start++){
341 +                net_stats->interface_name=NULL;
342 +                net_stats->speed=0;
343 +                net_stats->dup=UNKNOWN_DUPLEX;
344 +                net_stats++;
345 +        }
346 + }
347 +
348 + network_iface_stat_t *network_iface_stat_malloc(int needed_entries, int *cur_entries, network_iface_stat_t *net_stats){
349 +
350 +        if(net_stats==NULL){
351 +
352 +                if((net_stats=malloc(needed_entries * sizeof(network_iface_stat_t)))==NULL){
353 +                        return NULL;
354 +                }
355 +                network_iface_stat_init(0, needed_entries, net_stats);
356 +                *cur_entries=needed_entries;
357 +
358 +                return net_stats;
359 +        }
360 +
361 +
362 +        if(*cur_entries<needed_entries){
363 +                net_stats=realloc(net_stats, (sizeof(network_iface_stat_t)*needed_entries));
364 +                if(net_stats==NULL){
365 +                        return NULL;
366 +                }
367 +                network_iface_stat_init(*cur_entries, needed_entries, net_stats);
368 +                *cur_entries=needed_entries;
369 +        }
370 +
371 +        return net_stats;
372 + }
373 +
374 + network_iface_stat_t *get_network_iface_stats(int *entries){
375 +        static network_iface_stat_t *network_iface_stats;
376 +        network_iface_stat_t *network_iface_stat_ptr;
377 +        static int sizeof_network_iface_stats=0;        
378 +        static int ifaces;
379 +
380 + #ifdef SOLARIS
381 +        kstat_ctl_t *kc;
382 +        kstat_t *ksp;
383 +        kstat_named_t *knp;
384 + #endif
385 + #ifdef ALLBSD
386 +        struct ifaddrs *net, *net_ptr;
387 +        struct ifmediareq ifmed;
388 +        int s;
389 +        int x;
390 + #endif
391 + #ifdef LINUX
392 +        FILE *f;
393 +        /* Horrible big enough, but it should be easily big enough */
394 +        char line[8096];
395 +        void *eth_tool_cmd_buf;
396 +        int buf_size;
397 +        int sock;
398 + #endif
399 +        ifaces = 0;
400 + #ifdef ALLBSD
401 +        if(getifaddrs(&net) != 0){
402 +                return NULL;
403 +        }
404 +
405 +        if ((s = socket(AF_INET, SOCK_DGRAM, 0)) == NULL) return NULL;
406 +
407 +        for(net_ptr=net; net_ptr!=NULL; net_ptr=net_ptr->ifa_next){
408 +                if(net_ptr->ifa_addr->sa_family != AF_LINK) continue;
409 +                network_iface_stats=network_iface_stat_malloc((ifaces+1), &sizeof_network_iface_stats, network_iface_stats);
410 +                if(network_iface_stats==NULL){
411 +                        return NULL;
412 +                }
413 +                network_iface_stat_ptr = network_iface_stats + ifaces;
414 +
415 +                memset(&ifmed, 0, sizeof(struct ifmediareq));
416 +                strlcpy(ifmed.ifm_name, net_ptr->ifa_name, sizeof(ifmed.ifm_name));
417 +                if(ioctl(s, SIOCGIFMEDIA, (caddr_t)&ifmed) == -1){
418 +                        continue;
419 +                }
420 +
421 +                /* We may need to change this if we start doing wireless devices too */
422 +                if( (ifmed.ifm_active | IFM_ETHER) != ifmed.ifm_active ){
423 +                        /* Not a ETHER device */
424 +                        continue;
425 +                }
426 +
427 +                if(network_iface_stat_ptr->interface_name != NULL) free(network_iface_stat_ptr->interface_name);
428 +                network_iface_stat_ptr->interface_name = strdup(net_ptr->ifa_name);
429 +                if(network_iface_stat_ptr->interface_name == NULL) return NULL;
430 +
431 +                /* Only intrested in the first 4 bits)  - Assuming only ETHER devices */
432 +                x = ifmed.ifm_active & 0x0f;    
433 +                switch(x){
434 +                        /* 10 Mbit connections. Speedy :) */
435 +                        case(IFM_10_T):
436 +                        case(IFM_10_2):
437 +                        case(IFM_10_5):
438 +                        case(IFM_10_STP):
439 +                        case(IFM_10_FL):
440 +                                network_iface_stat_ptr->speed = 10;
441 +                                break;
442 +                        /* 100 Mbit conneections */
443 +                        case(IFM_100_TX):
444 +                        case(IFM_100_FX):
445 +                        case(IFM_100_T4):
446 +                        case(IFM_100_VG):
447 +                        case(IFM_100_T2):
448 +                                network_iface_stat_ptr->speed = 100;
449 +                                break;
450 +                        /* 1000 Mbit connections */
451 +                        case(IFM_1000_SX):
452 +                        case(IFM_1000_LX):
453 +                        case(IFM_1000_CX):
454 + #if defined(FREEBSD) && !defined(FREEBSD5)
455 +                        case(IFM_1000_TX):
456 +                        case(IFM_1000_FX):
457 + #else
458 +                        case(IFM_1000_T):
459 + #endif
460 +                                network_iface_stat_ptr->speed = 1000;
461 +                                break;
462 +                        /* We don't know what it is */
463 +                        default:
464 +                                network_iface_stat_ptr->speed = 0;
465 +                                break;
466 +                }
467 +
468 +                if( (ifmed.ifm_active | IFM_FDX) == ifmed.ifm_active ){
469 +                        network_iface_stat_ptr->dup = FULL_DUPLEX;
470 +                }else if( (ifmed.ifm_active | IFM_HDX) == ifmed.ifm_active ){
471 +                        network_iface_stat_ptr->dup = HALF_DUPLEX;
472 +                }else{
473 +                        network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
474 +                }
475 +                ifaces++;
476 +        }      
477 +        freeifaddrs(net);
478 +        close(s);
479 + #endif
480 +
481 + #ifdef SOLARIS
482 +        if ((kc = kstat_open()) == NULL) {
483 +                return NULL;
484 +        }
485 +
486 +        for (ksp = kc->kc_chain; ksp; ksp = ksp->ks_next) {
487 +                if (!strcmp(ksp->ks_class, "net")) {
488 +                        kstat_read(kc, ksp, NULL);
489 +                        if((knp=kstat_data_lookup(ksp, "ifspeed"))==NULL){
490 +                                /* Not a network interface, so skip to the next entry */
491 +                                continue;
492 +                        }
493 +                        network_iface_stats=network_iface_stat_malloc((ifaces+1), &sizeof_network_iface_stats, network_iface_stats);
494 +                        if(network_iface_stats==NULL){
495 +                                return NULL;
496 +                        }
497 +                        network_iface_stat_ptr = network_iface_stats + ifaces;
498 +                        network_iface_stat_ptr->speed = knp->value.ui64 / (1000*1000);
499 +
500 +                        if((knp=kstat_data_lookup(ksp, "link_duplex"))==NULL){
501 +                                /* Not a network interface, so skip to the next entry */
502 +                                continue;
503 +                        }
504 +
505 +                        if(knp->value.ui64 == 0){
506 +                                network_iface_stat_ptr->dup = FULL_DUPLEX;
507 +                        }else{
508 +                                network_iface_stat_ptr->dup = HALF_DUPLEX;
509 +                        }
510 +
511 +                        if(network_iface_stat_ptr->interface_name != NULL) free(network_iface_stat_ptr->interface_name);
512 +                        network_iface_stat_ptr->interface_name = strdup(ksp->ks_name);
513 +                        if(network_iface_stat_ptr->interface_name == NULL) return NULL;
514 +                        ifaces++;
515 +                }
516 +        }
517 +        kstat_close(kc);
518 +        
519 + #endif  
520 + #ifdef LINUX
521 +
522 +        f = fopen("/proc/net/dev", "r");
523 +        if(f == NULL){
524 +                return NULL;
525 +        }
526 +
527 +        /* Setup stuff so we can do the ioctl to get the info */
528 +        if((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0){
529 +                return NULL;
530 +        }
531 +
532 +        buf_size = sizeof(struct ethtool_cmd);
533 +        eth_tool_cmd_buf = malloc(buf_size);
534 +        if(eth_tool_cmd_buf == NULL) return NULL;
535 +
536 +        /* Ignore first 2 lines.. Just headings */
537 +        if((fgets(line, sizeof(line), f)) == NULL) return NULL;
538 +        if((fgets(line, sizeof(line), f)) == NULL) return NULL;
539 +
540 +        while((fgets(line, sizeof(line), f)) != NULL){
541 +                char *name, *ptr;
542 +                struct ifreq ifr;
543 +                struct ethtool_cmd *ethcmd;
544 +                int err;
545 +
546 +                /* Get the interface name */
547 +                ptr = strchr(line, ':');
548 +                if (ptr == NULL) continue;
549 +                *ptr='\0';
550 +                name = line;
551 +                while(isspace(*(name))){
552 +                        name++;
553 +                }
554 +
555 +                memset(&ifr, 0, sizeof(ifr));
556 +                memset(eth_tool_cmd_buf, 0, buf_size);
557 +                ifr.ifr_data = (caddr_t) eth_tool_cmd_buf;
558 +                strncpy(ifr.ifr_name, name, sizeof(ifr.ifr_name));
559 +
560 +                ethcmd = (struct ethtool_cmd *) ifr.ifr_data;
561 +                ethcmd->cmd = ETHTOOL_GSET;
562 +
563 +                err = ioctl(sock, SIOCETHTOOL, &ifr);
564 +                if(err < 0){
565 +                        /* This could fail if the interface doesn't support the command. Carry
566 +                         * on to the next :)
567 +                         */
568 +                        continue;
569 +                }
570 +
571 +                /* We have a good interface to add */
572 +                network_iface_stats=network_iface_stat_malloc((ifaces+1), &sizeof_network_iface_stats, network_iface_stats);
573 +                if(network_iface_stats==NULL){
574 +                        return NULL;
575 +                }
576 +                network_iface_stat_ptr = network_iface_stats + ifaces;
577 +                network_iface_stat_ptr->interface_name = strdup(name);
578 +                network_iface_stat_ptr->speed = ethcmd->speed;
579 +                network_iface_stat_ptr->dup = UNKNOWN_DUPLEX;
580 +                if(ethcmd->duplex == 0x00){
581 +                        network_iface_stat_ptr->dup = FULL_DUPLEX;
582 +                }
583 +                if(ethcmd->duplex == 0x01){
584 +                        network_iface_stat_ptr->dup = HALF_DUPLEX;
585 +                }
586 +                ifaces++;
587 +        }
588 +        close(sock);
589 +        free(eth_tool_cmd_buf);
590 + #endif
591 +        *entries = ifaces;
592 +        return network_iface_stats;
593 + }
594 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines