#include <config.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <sys/types.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <isc/interfaceiter.h>
#include <isc/mem.h>
#include <isc/result.h>
#include <isc/string.h>
#include <isc/strerror.h>
#include <isc/types.h>
#include <isc/util.h>
#define IFITER_MAGIC 0x49464954U
#define VALID_IFITER(t) ((t) != NULL && (t)->magic == IFITER_MAGIC)
struct isc_interfaceiter {
unsigned int magic;
isc_mem_t *mctx;
int socket;
INTERFACE_INFO IFData;
int numIF;
int totalIF;
INTERFACE_INFO *buf;
unsigned int bufsize;
INTERFACE_INFO *pos;
isc_interface_t current;
isc_result_t result;
};
#define IFCONF_SIZE_INITIAL 16
#define IFCONF_SIZE_INCREMENT 64
#define IFCONF_SIZE_MAX 1040
static void
get_addr(unsigned int family, isc_netaddr_t *dst, struct sockaddr *src) {
dst->family = family;
switch (family) {
case AF_INET:
memcpy(&dst->type.in,
&((struct sockaddr_in *) src)->sin_addr,
sizeof(struct in_addr));
break;
case AF_INET6:
memcpy(&dst->type.in6,
&((struct sockaddr_in6 *) src)->sin6_addr,
sizeof(struct in6_addr));
break;
default:
INSIST(0);
break;
}
}
static void
get_broadcastaddr(isc_netaddr_t *bcastaddr, isc_netaddr_t *addr, isc_netaddr_t *netmask) {
unsigned char *p, *a, *n;
int i;
p = (unsigned char *) &bcastaddr->type.in;
a = (unsigned char *) &addr->type.in;
n = (unsigned char *) &netmask->type.in;
for (i=0; i < 4; i++) {
p[i] = (unsigned char) (a[i] | ~n[i]);
}
}
isc_result_t
isc_interfaceiter_create(isc_mem_t *mctx, isc_interfaceiter_t **iterp) {
char strbuf[ISC_STRERRORSIZE];
isc_interfaceiter_t *iter;
isc_result_t result;
int error;
unsigned long bytesReturned = 0;
REQUIRE(iterp != NULL);
REQUIRE(*iterp == NULL);
iter = isc_mem_get(mctx, sizeof(*iter));
if (iter == NULL)
return (ISC_R_NOMEMORY);
iter->mctx = mctx;
iter->buf = NULL;
if ((iter->socket = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
error = WSAGetLastError();
isc__strerror(error, strbuf, sizeof(strbuf));
UNEXPECTED_ERROR(__FILE__, __LINE__,
"making interface scan socket: %s",
strbuf);
result = ISC_R_UNEXPECTED;
goto socket_failure;
}
iter->bufsize = IFCONF_SIZE_INITIAL*sizeof(INTERFACE_INFO);
for (;;) {
iter->buf = isc_mem_get(mctx, iter->bufsize);
if (iter->buf == NULL) {
result = ISC_R_NOMEMORY;
goto alloc_failure;
}
if (WSAIoctl(iter->socket, SIO_GET_INTERFACE_LIST,
0, 0, iter->buf, iter->bufsize,
&bytesReturned, 0, 0) == SOCKET_ERROR)
{
error = WSAGetLastError();
if (error != WSAEFAULT && error != WSAENOBUFS) {
errno = error;
isc__strerror(error, strbuf, sizeof(strbuf));
UNEXPECTED_ERROR(__FILE__, __LINE__,
"get interface configuration: %s",
strbuf);
result = ISC_R_UNEXPECTED;
goto ioctl_failure;
}
} else {
if (bytesReturned > 0 &&
(bytesReturned < iter->bufsize))
break;
}
if (iter->bufsize >= IFCONF_SIZE_MAX*sizeof(INTERFACE_INFO)) {
UNEXPECTED_ERROR(__FILE__, __LINE__,
"get interface configuration: "
"maximum buffer size exceeded");
result = ISC_R_UNEXPECTED;
goto ioctl_failure;
}
isc_mem_put(mctx, iter->buf, iter->bufsize);
iter->bufsize += IFCONF_SIZE_INCREMENT *
sizeof(INTERFACE_INFO);
}
iter->pos = NULL;
iter->result = ISC_R_FAILURE;
iter->numIF = 0;
iter->totalIF = bytesReturned/sizeof(INTERFACE_INFO);
iter->magic = IFITER_MAGIC;
*iterp = iter;
closesocket(iter->socket);
return (ISC_R_SUCCESS);
ioctl_failure:
isc_mem_put(mctx, iter->buf, iter->bufsize);
alloc_failure:
(void) closesocket(iter->socket);
socket_failure:
isc_mem_put(mctx, iter, sizeof(*iter));
return (result);
}
static isc_result_t
internal_current(isc_interfaceiter_t *iter, int family) {
BOOL ifNamed = FALSE;
unsigned long flags;
REQUIRE(VALID_IFITER(iter));
REQUIRE(iter->numIF >= 0);
memset(&iter->current, 0, sizeof(iter->current));
iter->current.af = family;
get_addr(family, &iter->current.address,
(struct sockaddr *)&(iter->IFData.iiAddress));
iter->current.ifindex = 0;
iter->current.scopeid = 0;
iter->current.flags = 0;
flags = iter->IFData.iiFlags;
if ((flags & IFF_UP) != 0)
iter->current.flags |= INTERFACE_F_UP;
if ((flags & IFF_POINTTOPOINT) != 0) {
iter->current.flags |= INTERFACE_F_POINTTOPOINT;
sprintf(iter->current.name, "PPP Interface %d", iter->numIF);
ifNamed = TRUE;
}
if ((flags & IFF_LOOPBACK) != 0) {
iter->current.flags |= INTERFACE_F_LOOPBACK;
sprintf(iter->current.name, "Loopback Interface %d",
iter->numIF);
ifNamed = TRUE;
}
if ((flags & IFF_BROADCAST) != 0) {
iter->current.flags |= INTERFACE_F_BROADCAST;
}
if ((flags & IFF_MULTICAST) != 0) {
iter->current.flags |= INTERFACE_F_MULTICAST;
}
switch (family) {
case AF_INET:
get_addr(family, &iter->current.netmask,
(struct sockaddr *)&(iter->IFData.iiNetmask));
break;
case AF_INET6:
break;
}
if ((iter->current.flags & INTERFACE_F_POINTTOPOINT) != 0) {
get_addr(family, &iter->current.dstaddress,
(struct sockaddr *)&(iter->IFData.iiBroadcastAddress));
}
if ((iter->current.flags & INTERFACE_F_BROADCAST) != 0) {
get_addr(family, &iter->current.broadcast,
(struct sockaddr *)&(iter->IFData.iiBroadcastAddress));
get_broadcastaddr(&iter->current.broadcast, &iter->current.address,
&iter->current.netmask);
}
if (ifNamed == FALSE)
sprintf(iter->current.name,
"IP Interface %d", iter->numIF);
return (ISC_R_SUCCESS);
}
static isc_result_t
internal_next(isc_interfaceiter_t *iter) {
if (iter->numIF >= iter->totalIF)
return (ISC_R_NOMORE);
if (iter->numIF == 0)
iter->pos = (INTERFACE_INFO *)(iter->buf + (iter->totalIF));
iter->pos--;
if (&(iter->pos) < &(iter->buf))
return (ISC_R_NOMORE);
memset(&(iter->IFData), 0, sizeof(INTERFACE_INFO));
memcpy(&(iter->IFData), iter->pos, sizeof(INTERFACE_INFO));
iter->numIF++;
return (ISC_R_SUCCESS);
}
isc_result_t
isc_interfaceiter_current(isc_interfaceiter_t *iter,
isc_interface_t *ifdata) {
REQUIRE(iter->result == ISC_R_SUCCESS);
memcpy(ifdata, &iter->current, sizeof(*ifdata));
return (ISC_R_SUCCESS);
}
isc_result_t
isc_interfaceiter_first(isc_interfaceiter_t *iter) {
isc_result_t result;
REQUIRE(VALID_IFITER(iter));
iter->numIF = 0;
for (;;) {
result = internal_next(iter);
if (result != ISC_R_SUCCESS)
break;
result = internal_current(iter, AF_INET);
if (result != ISC_R_IGNORE)
break;
}
iter->result = result;
return (result);
}
isc_result_t
isc_interfaceiter_next(isc_interfaceiter_t *iter) {
isc_result_t result;
REQUIRE(VALID_IFITER(iter));
REQUIRE(iter->result == ISC_R_SUCCESS);
for (;;) {
result = internal_next(iter);
if (result != ISC_R_SUCCESS)
break;
result = internal_current(iter,AF_INET);
if (result != ISC_R_IGNORE)
break;
}
iter->result = result;
return (result);
}
void
isc_interfaceiter_destroy(isc_interfaceiter_t **iterp) {
isc_interfaceiter_t *iter;
REQUIRE(iterp != NULL);
iter = *iterp;
REQUIRE(VALID_IFITER(iter));
isc_mem_put(iter->mctx, iter->buf, iter->bufsize);
iter->magic = 0;
isc_mem_put(iter->mctx, iter, sizeof(*iter));
*iterp = NULL;
}