Subversion Repositories planix.SVN

#include <u.h>

#include <libc.h>

#include <ip.h>

#include <bio.h>

#include <ndb.h>

static uchar noether[6];

/*

 *  Look for a pair with the given attribute.  look first on the same line,

 *  then in the whole entry.

 */

static Ndbtuple*

lookval(Ndbtuple *entry, Ndbtuple *line, char *attr, char *to)

{

	Ndbtuple *nt;

	/* first look on same line (closer binding) */

	for(nt = line;;){

		if(strcmp(attr, nt->attr) == 0){

			strncpy(to, nt->val, Ndbvlen);

			return nt;

		}

		nt = nt->line;

		if(nt == line)

			break;

	}

	/* search whole tuple */

	for(nt = entry; nt; nt = nt->entry)

		if(strcmp(attr, nt->attr) == 0){

			strncpy(to, nt->val, Ndbvlen);

			return nt;

		}

	return 0;

}

/*

 *  lookup an ip address

 */

static uchar*

lookupip(Ndb *db, char *name, uchar *to, Ipinfo *iip)

{

	Ndbtuple *t, *nt;

	char buf[Ndbvlen];

	uchar subnet[IPaddrlen];

	Ndbs s;

	char *attr;

	attr = ipattr(name);

	if(strcmp(attr, "ip") == 0){

		parseip(to, name);

		return to;

	}

	t = ndbgetval(db, &s, attr, name, "ip", buf);

	if(t){

		/* first look for match on same subnet */

		for(nt = t; nt; nt = nt->entry){

			if(strcmp(nt->attr, "ip") != 0)

				continue;

			parseip(to, nt->val);

			maskip(to, iip->ipmask, subnet);

			if(memcmp(subnet, iip->ipnet, sizeof(subnet)) == 0)

				return to;

		}

		/* otherwise, just take what we have */

		ndbfree(t);

		parseip(to, buf);

		return to;

	}

	return 0;

}

/*

 *  lookup a subnet and fill in anything we can

 */

static void

recursesubnet(Ndb *db, uchar *mask, Ipinfo *iip, char *fs, char *gw, char *au)

{

	Ndbs s;

	Ndbtuple *t;

	uchar submask[IPaddrlen];

	char ip[Ndbvlen];

	memmove(iip->ipmask, mask, 4);

	maskip(iip->ipaddr, iip->ipmask, iip->ipnet);

	sprint(ip, "%I", iip->ipnet);

	t = ndbsearch(db, &s, "ip", ip);

print("%s->", ip);

	if(t){

		/* look for a further subnet */

		if(lookval(t, s.t, "ipmask", ip)){

			parseip(submask, ip);

			/* recurse only if it has changed */

			if(!equivip(submask, mask))

				recursesubnet(db, submask, iip, fs, gw, au);

		}

		/* fill in what we don't have */

		if(gw[0] == 0)

			lookval(t, s.t, "ipgw", gw);

		if(fs[0] == 0)

			lookval(t, s.t, "fs", fs);

		if(au[0] == 0)

			lookval(t, s.t, "auth", au);

		ndbfree(t);

	}

}

#ifdef foo

/*

 *  find out everything we can about a system from what has been

 *  specified.

 */

int

ipinfo(Ndb *db, char *etherin, char *ipin, char *name, Ipinfo *iip)

{

	Ndbtuple *t;

	Ndbs s;

	char ether[Ndbvlen];

	char ip[Ndbvlen];

	char fsname[Ndbvlen];

	char gwname[Ndbvlen];

	char auname[Ndbvlen];

	memset(iip, 0, sizeof(Ipinfo));

	fsname[0] = 0;

	gwname[0] = 0;

	auname[0] = 0;

	/*

	 *  look for a matching entry

	 */

	t = 0;

	if(etherin)

		t = ndbgetval(db, &s, "ether", etherin, "ip", ip);

	if(t == 0 && ipin)

		t = ndbsearch(db, &s, "ip", ipin);

	if(t == 0 && name)

		t = ndbgetval(db, &s, ipattr(name), name, "ip", ip);

	if(t){

		/*

		 *  copy in addresses and name

		 */

		if(lookval(t, s.t, "ip", ip))

			parseip(iip->ipaddr, ip);

		if(lookval(t, s.t, "ether", ether))

			parseether(iip->etheraddr, ether);

		lookval(t, s.t, "dom", iip->domain);

		/*

		 *  Look for bootfile, fs, and gateway.

		 *  If necessary, search through all entries for

		 *  this ip address.

		 */

		while(t){

			if(iip->bootf[0] == 0)

				lookval(t, s.t, "bootf", iip->bootf);

			if(fsname[0] == 0)

				lookval(t, s.t, "fs", fsname);

			if(gwname[0] == 0)

				lookval(t, s.t, "ipgw", gwname);

			if(auname[0] == 0)

				lookval(t, s.t, "auth", auname);

			ndbfree(t);

			if(iip->bootf[0] && fsname[0] && gwname[0] && auname[0])

				break;

			t = ndbsnext(&s, "ether", ether);

		}

	} else if(ipin) {

		/*

		 *  copy in addresses (all we know)

		 */

		parseip(iip->ipaddr, ipin);

		if(etherin)

			parseether(iip->etheraddr, etherin);

	} else

		return -1;

	/*

	 *  Look up the client's network and find a subnet mask for it.

	 *  Fill in from the subnet (or net) entry anything we can't figure

	 *  out from the client record.

	 */

	recursesubnet(db, classmask[CLASS(iip->ipaddr)], iip, fsname, gwname, auname);

	/* lookup fs's and gw's ip addresses */

	if(fsname[0])

		lookupip(db, fsname, iip->fsip, iip);

	if(gwname[0])

		lookupip(db, gwname, iip->gwip, iip);

	if(auname[0])

		lookupip(db, auname, iip->auip, iip);

	return 0;

}

#endif

void

main(int argc, char **argv)

{

	Ipinfo ii;

	Ndb *db;

	db = ndbopen(0);

	fmtinstall('E', eipconv);

	fmtinstall('I', eipconv);

	if(argc < 2)

		exits(0);

	if(strchr(argv[1], '.')){

		if(ipinfo(db, 0, argv[1], 0, &ii) < 0)

			exits(0);

	} else {

		if(ipinfo(db, argv[1], 0, 0, &ii) < 0)

			exits(0);

	}

	fprint(2, "a %I m %I n %I f %s e %E\n", ii.ipaddr,

		ii.ipmask, ii.ipnet, ii.bootf, ii.etheraddr);

}