ref: 540373b9f603825f27f853a441c6c046a04a0972
dir: /sys/src/cmd/ip/dhcpd/testlook.c/
#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); }