123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168 |
- #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]);
- }
|