miniupnp/miniupnpd/mac/getifstats.c

95 lines
2.5 KiB
C

/* $Id: getifstats.c,v 1.6 2012/02/11 13:10:57 nanard Exp $ */
/*
* MiniUPnP project
* http://miniupnp.free.fr/ or http://miniupnp.tuxfamily.org/
* (c) 2009 Jardel Weyrich
* This software is subject to the conditions detailed
* in the LICENCE file provided within the distribution
*/
#include <syslog.h>
#include <sys/sysctl.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <net/if.h>
#include <net/if_types.h>
#include <net/route.h>
#include <nlist.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "../getifstats.h"
#include "../config.h"
int getifstats(const char * ifname, struct ifdata * data) {
int mib[] = { CTL_NET, PF_ROUTE, 0, AF_INET, NET_RT_IFLIST, if_nametoindex(ifname) };
const size_t mib_len = sizeof(mib) / sizeof(mib[0]);
size_t needed;
char *buf, *end, *p;
struct if_msghdr *ifm;
struct if_data ifdata;
#ifdef ENABLE_GETIFSTATS_CACHING
static time_t cache_timestamp = 0;
static struct ifdata cache_data;
time_t current_time;
#endif
if (data == NULL || ifname == NULL || ifname[0] == '\0')
return -1; /* error */
data->baudrate = 4200000;
data->opackets = 0;
data->ipackets = 0;
data->obytes = 0;
data->ibytes = 0;
#ifdef ENABLE_GETIFSTATS_CACHING
current_time = time(NULL);
if (current_time == ((time_t)-1)) {
syslog(LOG_ERR, "getifstats() : time() error : %m");
} else {
if (current_time < cache_timestamp + GETIFSTATS_CACHING_DURATION) {
memcpy(data, &cache_data, sizeof(struct ifdata));
return 0;
}
}
#endif
if (sysctl(mib, mib_len, NULL, &needed, NULL, 0) == -1) {
syslog(LOG_ERR, "sysctl(): %m");
return -1;
}
buf = (char *) malloc(needed);
if (buf == NULL)
return -1; /* error */
if (sysctl(mib, mib_len, buf, &needed, NULL, 0) == -1) {
syslog(LOG_ERR, "sysctl(): %m");
free(buf);
return -1; /* error */
} else {
for (end = buf + needed, p = buf; p < end; p += ifm->ifm_msglen) {
ifm = (struct if_msghdr *) p;
if (ifm->ifm_type == RTM_IFINFO && ifm->ifm_data.ifi_type == IFT_ETHER) {
ifdata = ifm->ifm_data;
data->opackets = ifdata.ifi_opackets;
data->ipackets = ifdata.ifi_ipackets;
data->obytes = ifdata.ifi_obytes;
data->ibytes = ifdata.ifi_ibytes;
data->baudrate = ifdata.ifi_baudrate;
free(buf);
#ifdef ENABLE_GETIFSTATS_CACHING
if (current_time!=((time_t)-1)) {
cache_timestamp = current_time;
memcpy(&cache_data, data, sizeof(struct ifdata));
}
#endif
return 0; /* found, ok */
}
}
}
free(buf);
return -1; /* not found or error */
}