#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/param.h>
#include <stdio.h>
#include <string.h>
#include <netinfo/ni.h>
#include <string.h>
#include <sys/socket.h>
#include <net/if.h>
#include <sys/ioctl.h>
extern char *inet_ntoa();
static const char NAME_NAME[] = "name";
static const char NAME_MACHINES[] = "machines";
static const char NAME_IP_ADDRESS[] = "ip_address";
static const char NAME_SERVES[] = "serves";
static const char NAME_UNKNOWN[] = "###UNKNOWN###";
typedef struct
{
char name[IFNAMSIZ];
short flags;
struct in_addr addr;
struct in_addr mask;
struct in_addr netaddr;
struct in_addr bcast;
} interface_t;
typedef struct
{
unsigned int count;
interface_t *interface;
} interface_list_t;
static interface_list_t *my_interfaces = NULL;
static interface_list_t *
sys_interfaces(void)
{
struct ifconf ifc;
struct ifreq *ifr;
char buf[1024];
int offset, addrlen, extra, delta;
int sock;
interface_t *iface;
if (my_interfaces != NULL) return my_interfaces;
sock = socket(AF_INET, SOCK_DGRAM, 0);
if (sock < 0) return NULL;
ifc.ifc_len = sizeof(buf);
ifc.ifc_buf = buf;
if (ioctl(sock, SIOCGIFCONF, (char *)&ifc) < 0)
{
close(sock);
return NULL;
}
my_interfaces = (interface_list_t *)malloc(sizeof(interface_list_t));
my_interfaces->count = 0;
my_interfaces->interface = NULL;
delta = sizeof(struct ifreq);
addrlen = delta - IFNAMSIZ;
extra = 0;
offset = 0;
while (offset <= ifc.ifc_len)
{
ifr = (struct ifreq *)(ifc.ifc_buf + offset);
#ifndef _NO_SOCKADDR_LENGTH_
extra = ifr->ifr_addr.sa_len - addrlen;
if (extra < 0) extra = 0;
#endif
offset = offset + delta + extra;
if (ifr->ifr_addr.sa_family != AF_INET) continue;
if (ioctl(sock, SIOCGIFFLAGS, (char *)ifr) < 0) continue;
my_interfaces->count++;
if (my_interfaces->count == 1)
{
my_interfaces->interface = (interface_t *)malloc(sizeof(interface_t));
}
else
{
my_interfaces->interface = (interface_t *)realloc(my_interfaces->interface, my_interfaces->count * sizeof(interface_t));
}
iface = &(my_interfaces->interface[my_interfaces->count - 1]);
memset(iface, 0, sizeof(interface_t));
memmove(iface->name, ifr->ifr_name, IFNAMSIZ);
iface->flags = ifr->ifr_ifru.ifru_flags;
iface->addr.s_addr = ((struct sockaddr_in *)&(ifr->ifr_addr))->sin_addr.s_addr;
ioctl(sock, SIOCGIFNETMASK, (char *)ifr);
iface->mask.s_addr = ((struct sockaddr_in *)&(ifr->ifr_addr))->sin_addr.s_addr;
iface->netaddr.s_addr = iface->addr.s_addr & iface->mask.s_addr;
iface->bcast.s_addr = iface->netaddr.s_addr | (~iface->mask.s_addr);
}
close(sock);
return my_interfaces;
}
static ni_name
escape_domain(ni_name name)
{
int extra;
char *p;
char *s;
ni_name newname;
extra = 0;
for (p = name; *p; p++)
{
if ((*p == '/') || (*p == '\\')) extra++;
}
newname = malloc(strlen(name) + extra + 1);
s = newname;
for (p = name; *p; p++)
{
if ((*p == '/') || (*p == '\\')) *s++ = '\\';
*s++ = *p;
}
*s = 0;
return newname;
}
static char *
finddomain(void *ni, struct in_addr addr, ni_name tag)
{
ni_id nid;
ni_idlist idl;
ni_namelist nl;
ni_index i;
ni_name slash;
ni_name domain;
ni_status status;
status = ni_root(ni, &nid);
if (status != NI_OK) return NULL;
status = ni_lookup(ni, &nid, NAME_NAME, NAME_MACHINES, &idl);
if (status != NI_OK) return NULL;
nid.nii_object = idl.niil_val[0];
ni_idlist_free(&idl);
status = ni_lookup(ni, &nid, NAME_IP_ADDRESS, inet_ntoa(addr), &idl);
if (status != NI_OK) return NULL;
nid.nii_object = idl.niil_val[0];
ni_idlist_free(&idl);
status = ni_lookupprop(ni, &nid, NAME_SERVES, &nl);
if (status != NI_OK) return NULL;
for (i = 0; i < nl.ninl_len; i++)
{
slash = rindex(nl.ninl_val[i], '/');
if (slash == NULL) continue;
if (ni_name_match(slash + 1, tag))
{
*slash = 0;
domain = escape_domain(nl.ninl_val[i]);
ni_namelist_free(&nl);
return domain;
}
}
ni_namelist_free(&nl);
return NULL;
}
static int
sys_is_my_address(struct in_addr *a)
{
int i;
interface_list_t *l;
l = sys_interfaces();
if (l == NULL) return 0;
for (i = 0; i < l->count; i++)
{
if (a->s_addr == l->interface[i].addr.s_addr) return 1;
}
return 0;
}
static char *
ni_domainof(void *ni, void *parent)
{
struct sockaddr_in addr;
ni_name tag;
ni_name dom;
ni_status status;
interface_list_t *ilist;
int i;
status = ni_addrtag(ni, &addr, &tag);
if (status != NI_OK) return ni_name_dup(NAME_UNKNOWN);
dom = finddomain(parent, addr.sin_addr, tag);
if (dom != NULL)
{
ni_name_free(&tag);
return dom;
}
if (sys_is_my_address(&(addr.sin_addr)))
{
ilist = sys_interfaces();
if (ilist == NULL) return ni_name_dup(NAME_UNKNOWN);
for (i = 0; i < ilist->count; i++)
{
if (ilist->interface[i].addr.s_addr == htonl(INADDR_LOOPBACK)) continue;
addr.sin_addr.s_addr = ilist->interface[i].addr.s_addr;
dom = finddomain(parent, addr.sin_addr, tag);
if (dom != NULL)
{
ni_name_free(&tag);
return dom;
}
}
}
dom = malloc(strlen(tag) + 256);
sprintf(dom, "%s@%s", tag, inet_ntoa(addr.sin_addr.s_addr));
ni_name_free(&tag);
return dom;
}
static ni_status
_ni_pwdomain(void *ni, ni_name *buf)
{
void *nip;
ni_status status;
int len;
char *dom;
nip = ni_new(ni, "..");
if (nip == NULL)
{
(*buf) = malloc(2);
(*buf)[0] = 0;
return NI_OK;
}
status = _ni_pwdomain(nip, buf);
if (status != NI_OK) return status;
dom = ni_domainof(ni, nip);
len = strlen(*buf);
*buf = realloc(*buf, len + 1 + strlen(dom) + 1);
(*buf)[len] = '/';
strcpy(&(*buf)[len + 1], dom);
ni_name_free(&dom);
ni_free(nip);
return NI_OK;
}
ni_status
ni_pwdomain(void *ni, ni_name *buf)
{
ni_status status;
*buf = NULL;
status = _ni_pwdomain(ni, buf);
if (status != NI_OK)
{
if (*buf != NULL) ni_name_free(buf);
return status;
}
if ((*buf)[0] == 0)
{
(*buf)[0] = '/';
(*buf)[1] = 0;
}
return NI_OK;
}