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; |
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 |
+ |
|