new goodies

1 files changed, 168 insertions, 0 deletions

diff --git a/src/cmd/ip/dhcpd/testlookup.c b/src/cmd/ip/dhcpd/testlookup.c
new file mode 100755
index 00000000..c69624c5
--- /dev/null
+++ b/src/cmd/ip/dhcpd/testlookup.c
@@ -0,0 +1,168 @@
+#include <u.h>
+#include <libc.h>
+#include <ip.h>
+#include <bio.h>
+#include <ndb.h>
+
+static uchar noether[6];
+	Ndb *db;
+
+static void
+recursesubnet(Ndb *db, uchar *addr, uchar *mask, char *attr, char *name, char *name1)
+{
+	Ndbs s;
+	Ndbtuple *t, *nt;
+	uchar submask[IPaddrlen], net[IPaddrlen];
+	char ip[Ndbvlen];
+	int found;
+
+	maskip(addr, mask, net);
+	sprint(ip, "%I", net);
+	t = ndbsearch(db, &s, "ip", ip);
+	if(t == 0)
+		return;
+
+	for(nt = t; nt; nt = nt->entry){
+		if(strcmp(nt->attr, "ipmask") == 0){
+			parseip(submask, nt->val);
+			if(memcmp(submask, mask, IPaddrlen) != 0)
+				recursesubnet(db, addr, submask, attr, name, name1);
+			break;
+		}
+	}
+
+	if(name[0] == 0){
+		found = 0;
+		for(nt = t; nt; nt = nt->entry){
+			if(strcmp(nt->attr, attr) == 0){
+				if(found){
+					strcpy(name, nt->val);
+					name1[0] = 0;
+					found = 1;
+				} else {
+					strcpy(name1, nt->val);
+					break;
+				}
+			}
+		}
+	}
+
+	ndbfree(t);
+}
+
+/*
+ *  lookup an ip address
+ */
+static int
+getipaddr(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 1;
+	}
+
+	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 1;
+		}
+
+		/* otherwise, just take what we have */
+		ndbfree(t);
+		parseip(to, buf);
+		return 1;
+	}
+	return 0;
+}
+
+/*
+ *  return the ip addresses for a type of server for system ip
+ */
+int
+lookupserver(char *attr, uchar ipaddrs[2][IPaddrlen], Ipinfo *iip)
+{
+	Ndbtuple *t, *nt;
+	Ndbs s;
+	char ip[32];
+	char name[Ndbvlen];
+	char name1[Ndbvlen];
+	int i;
+
+	name[0] = name1[0] = 0;
+
+	snprint(ip, sizeof(ip), "%I", iip->ipaddr);
+	t = ndbsearch(db, &s, "ip", ip);
+	while(t){
+		for(nt = t; nt; nt = nt->entry){
+			if(strcmp(attr, nt->attr) == 0){
+				if(*name == 0)
+					strcpy(name, nt->val);
+				else {
+					strcpy(name1, nt->val);
+					break;
+				}
+			}
+		}
+		if(name[0])
+			break;
+		t = ndbsnext(&s, "ip", ip);
+	}
+
+	if(name[0] == 0)
+		recursesubnet(db, iip->ipaddr, classmask[CLASS(iip->ipaddr)], attr, name, name1);
+
+	i = 0;
+	if(name[0] && getipaddr(db, name, *ipaddrs, iip) == 1){
+		ipaddrs++;
+		i++;
+	}
+	if(name1[0] && getipaddr(db, name1, *ipaddrs, iip) == 1)
+		i++;
+	return i;
+}
+
+void
+main(int argc, char **argv)
+{
+	Ipinfo ii;
+	uchar addrs[2][IPaddrlen];
+	int i, j;
+
+	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);
+	}
+	print("a %I m %I n %I f %s e %E a %I\n", ii.ipaddr,
+		ii.ipmask, ii.ipnet, ii.bootf, ii.etheraddr, ii.auip);
+
+	i = lookupserver("auth", addrs, &ii);
+	print("lookupserver returns %d\n", i);
+	for(j = 0; j < i; j++)
+		print("%I\n", addrs[j]);
+	i = lookupserver("dns", addrs, &ii);
+	print("lookupserver returns %d\n", i);
+	for(j = 0; j < i; j++)
+		print("%I\n", addrs[j]);
+}