1 87a52e04 2005-12-26 devnull #include <u.h>
  2 87a52e04 2005-12-26 devnull #include <libc.h>
  3 87a52e04 2005-12-26 devnull #include <ip.h>
  4 87a52e04 2005-12-26 devnull #include <bio.h>
  5 87a52e04 2005-12-26 devnull #include <ndb.h>
  6 87a52e04 2005-12-26 devnull 
  7 87a52e04 2005-12-26 devnull static uchar noether[6];
  8 87a52e04 2005-12-26 devnull 	Ndb *db;
  9 87a52e04 2005-12-26 devnull 
 10 87a52e04 2005-12-26 devnull static void
 11 87a52e04 2005-12-26 devnull recursesubnet(Ndb *db, uchar *addr, uchar *mask, char *attr, char *name, char *name1)
 12 87a52e04 2005-12-26 devnull {
 13 87a52e04 2005-12-26 devnull 	Ndbs s;
 14 87a52e04 2005-12-26 devnull 	Ndbtuple *t, *nt;
 15 87a52e04 2005-12-26 devnull 	uchar submask[IPaddrlen], net[IPaddrlen];
 16 87a52e04 2005-12-26 devnull 	char ip[Ndbvlen];
 17 87a52e04 2005-12-26 devnull 	int found;
 18 87a52e04 2005-12-26 devnull 
 19 87a52e04 2005-12-26 devnull 	maskip(addr, mask, net);
 20 87a52e04 2005-12-26 devnull 	sprint(ip, "%I", net);
 21 87a52e04 2005-12-26 devnull 	t = ndbsearch(db, &s, "ip", ip);
 22 87a52e04 2005-12-26 devnull 	if(t == 0)
 23 87a52e04 2005-12-26 devnull 		return;
 24 87a52e04 2005-12-26 devnull 
 25 87a52e04 2005-12-26 devnull 	for(nt = t; nt; nt = nt->entry){
 26 87a52e04 2005-12-26 devnull 		if(strcmp(nt->attr, "ipmask") == 0){
 27 87a52e04 2005-12-26 devnull 			parseip(submask, nt->val);
 28 87a52e04 2005-12-26 devnull 			if(memcmp(submask, mask, IPaddrlen) != 0)
 29 87a52e04 2005-12-26 devnull 				recursesubnet(db, addr, submask, attr, name, name1);
 30 87a52e04 2005-12-26 devnull 			break;
 31 87a52e04 2005-12-26 devnull 		}
 32 87a52e04 2005-12-26 devnull 	}
 33 87a52e04 2005-12-26 devnull 
 34 87a52e04 2005-12-26 devnull 	if(name[0] == 0){
 35 87a52e04 2005-12-26 devnull 		found = 0;
 36 87a52e04 2005-12-26 devnull 		for(nt = t; nt; nt = nt->entry){
 37 87a52e04 2005-12-26 devnull 			if(strcmp(nt->attr, attr) == 0){
 38 87a52e04 2005-12-26 devnull 				if(found){
 39 87a52e04 2005-12-26 devnull 					strcpy(name, nt->val);
 40 87a52e04 2005-12-26 devnull 					name1[0] = 0;
 41 87a52e04 2005-12-26 devnull 					found = 1;
 42 87a52e04 2005-12-26 devnull 				} else {
 43 87a52e04 2005-12-26 devnull 					strcpy(name1, nt->val);
 44 87a52e04 2005-12-26 devnull 					break;
 45 87a52e04 2005-12-26 devnull 				}
 46 87a52e04 2005-12-26 devnull 			}
 47 87a52e04 2005-12-26 devnull 		}
 48 87a52e04 2005-12-26 devnull 	}
 49 87a52e04 2005-12-26 devnull 
 50 87a52e04 2005-12-26 devnull 	ndbfree(t);
 51 87a52e04 2005-12-26 devnull }
 52 87a52e04 2005-12-26 devnull 
 53 87a52e04 2005-12-26 devnull /*
 54 87a52e04 2005-12-26 devnull  *  lookup an ip address
 55 87a52e04 2005-12-26 devnull  */
 56 87a52e04 2005-12-26 devnull static int
 57 87a52e04 2005-12-26 devnull getipaddr(Ndb *db, char *name, uchar *to, Ipinfo *iip)
 58 87a52e04 2005-12-26 devnull {
 59 87a52e04 2005-12-26 devnull 	Ndbtuple *t, *nt;
 60 87a52e04 2005-12-26 devnull 	char buf[Ndbvlen];
 61 87a52e04 2005-12-26 devnull 	uchar subnet[IPaddrlen];
 62 87a52e04 2005-12-26 devnull 	Ndbs s;
 63 87a52e04 2005-12-26 devnull 	char *attr;
 64 87a52e04 2005-12-26 devnull 
 65 87a52e04 2005-12-26 devnull 	attr = ipattr(name);
 66 87a52e04 2005-12-26 devnull 	if(strcmp(attr, "ip") == 0){
 67 87a52e04 2005-12-26 devnull 		parseip(to, name);
 68 87a52e04 2005-12-26 devnull 		return 1;
 69 87a52e04 2005-12-26 devnull 	}
 70 87a52e04 2005-12-26 devnull 
 71 87a52e04 2005-12-26 devnull 	t = ndbgetval(db, &s, attr, name, "ip", buf);
 72 87a52e04 2005-12-26 devnull 	if(t){
 73 87a52e04 2005-12-26 devnull 		/* first look for match on same subnet */
 74 87a52e04 2005-12-26 devnull 		for(nt = t; nt; nt = nt->entry){
 75 87a52e04 2005-12-26 devnull 			if(strcmp(nt->attr, "ip") != 0)
 76 87a52e04 2005-12-26 devnull 				continue;
 77 87a52e04 2005-12-26 devnull 			parseip(to, nt->val);
 78 87a52e04 2005-12-26 devnull 			maskip(to, iip->ipmask, subnet);
 79 87a52e04 2005-12-26 devnull 			if(memcmp(subnet, iip->ipnet, sizeof(subnet)) == 0)
 80 87a52e04 2005-12-26 devnull 				return 1;
 81 87a52e04 2005-12-26 devnull 		}
 82 87a52e04 2005-12-26 devnull 
 83 87a52e04 2005-12-26 devnull 		/* otherwise, just take what we have */
 84 87a52e04 2005-12-26 devnull 		ndbfree(t);
 85 87a52e04 2005-12-26 devnull 		parseip(to, buf);
 86 87a52e04 2005-12-26 devnull 		return 1;
 87 87a52e04 2005-12-26 devnull 	}
 88 87a52e04 2005-12-26 devnull 	return 0;
 89 87a52e04 2005-12-26 devnull }
 90 87a52e04 2005-12-26 devnull 
 91 87a52e04 2005-12-26 devnull /*
 92 87a52e04 2005-12-26 devnull  *  return the ip addresses for a type of server for system ip
 93 87a52e04 2005-12-26 devnull  */
 94 87a52e04 2005-12-26 devnull int
 95 87a52e04 2005-12-26 devnull lookupserver(char *attr, uchar ipaddrs[2][IPaddrlen], Ipinfo *iip)
 96 87a52e04 2005-12-26 devnull {
 97 87a52e04 2005-12-26 devnull 	Ndbtuple *t, *nt;
 98 87a52e04 2005-12-26 devnull 	Ndbs s;
 99 87a52e04 2005-12-26 devnull 	char ip[32];
100 87a52e04 2005-12-26 devnull 	char name[Ndbvlen];
101 87a52e04 2005-12-26 devnull 	char name1[Ndbvlen];
102 87a52e04 2005-12-26 devnull 	int i;
103 87a52e04 2005-12-26 devnull 
104 87a52e04 2005-12-26 devnull 	name[0] = name1[0] = 0;
105 87a52e04 2005-12-26 devnull 
106 87a52e04 2005-12-26 devnull 	snprint(ip, sizeof(ip), "%I", iip->ipaddr);
107 87a52e04 2005-12-26 devnull 	t = ndbsearch(db, &s, "ip", ip);
108 87a52e04 2005-12-26 devnull 	while(t){
109 87a52e04 2005-12-26 devnull 		for(nt = t; nt; nt = nt->entry){
110 87a52e04 2005-12-26 devnull 			if(strcmp(attr, nt->attr) == 0){
111 87a52e04 2005-12-26 devnull 				if(*name == 0)
112 87a52e04 2005-12-26 devnull 					strcpy(name, nt->val);
113 87a52e04 2005-12-26 devnull 				else {
114 87a52e04 2005-12-26 devnull 					strcpy(name1, nt->val);
115 87a52e04 2005-12-26 devnull 					break;
116 87a52e04 2005-12-26 devnull 				}
117 87a52e04 2005-12-26 devnull 			}
118 87a52e04 2005-12-26 devnull 		}
119 87a52e04 2005-12-26 devnull 		if(name[0])
120 87a52e04 2005-12-26 devnull 			break;
121 87a52e04 2005-12-26 devnull 		t = ndbsnext(&s, "ip", ip);
122 87a52e04 2005-12-26 devnull 	}
123 87a52e04 2005-12-26 devnull 
124 87a52e04 2005-12-26 devnull 	if(name[0] == 0)
125 87a52e04 2005-12-26 devnull 		recursesubnet(db, iip->ipaddr, classmask[CLASS(iip->ipaddr)], attr, name, name1);
126 87a52e04 2005-12-26 devnull 
127 87a52e04 2005-12-26 devnull 	i = 0;
128 87a52e04 2005-12-26 devnull 	if(name[0] && getipaddr(db, name, *ipaddrs, iip) == 1){
129 87a52e04 2005-12-26 devnull 		ipaddrs++;
130 87a52e04 2005-12-26 devnull 		i++;
131 87a52e04 2005-12-26 devnull 	}
132 87a52e04 2005-12-26 devnull 	if(name1[0] && getipaddr(db, name1, *ipaddrs, iip) == 1)
133 87a52e04 2005-12-26 devnull 		i++;
134 87a52e04 2005-12-26 devnull 	return i;
135 87a52e04 2005-12-26 devnull }
136 87a52e04 2005-12-26 devnull 
137 87a52e04 2005-12-26 devnull void
138 87a52e04 2005-12-26 devnull main(int argc, char **argv)
139 87a52e04 2005-12-26 devnull {
140 87a52e04 2005-12-26 devnull 	Ipinfo ii;
141 87a52e04 2005-12-26 devnull 	uchar addrs[2][IPaddrlen];
142 87a52e04 2005-12-26 devnull 	int i, j;
143 87a52e04 2005-12-26 devnull 
144 87a52e04 2005-12-26 devnull 	db = ndbopen(0);
145 87a52e04 2005-12-26 devnull 
146 87a52e04 2005-12-26 devnull 	fmtinstall('E', eipconv);
147 87a52e04 2005-12-26 devnull 	fmtinstall('I', eipconv);
148 87a52e04 2005-12-26 devnull 	if(argc < 2)
149 87a52e04 2005-12-26 devnull 		exits(0);
150 87a52e04 2005-12-26 devnull 	if(strchr(argv[1], '.')){
151 87a52e04 2005-12-26 devnull 		if(ipinfo(db, 0, argv[1], 0, &ii) < 0)
152 87a52e04 2005-12-26 devnull 			exits(0);
153 87a52e04 2005-12-26 devnull 	} else {
154 87a52e04 2005-12-26 devnull 		if(ipinfo(db, argv[1], 0, 0, &ii) < 0)
155 87a52e04 2005-12-26 devnull 			exits(0);
156 87a52e04 2005-12-26 devnull 	}
157 87a52e04 2005-12-26 devnull 	print("a %I m %I n %I f %s e %E a %I\n", ii.ipaddr,
158 87a52e04 2005-12-26 devnull 		ii.ipmask, ii.ipnet, ii.bootf, ii.etheraddr, ii.auip);
159 87a52e04 2005-12-26 devnull 
160 87a52e04 2005-12-26 devnull 	i = lookupserver("auth", addrs, &ii);
161 87a52e04 2005-12-26 devnull 	print("lookupserver returns %d\n", i);
162 87a52e04 2005-12-26 devnull 	for(j = 0; j < i; j++)
163 87a52e04 2005-12-26 devnull 		print("%I\n", addrs[j]);
164 87a52e04 2005-12-26 devnull 	i = lookupserver("dns", addrs, &ii);
165 87a52e04 2005-12-26 devnull 	print("lookupserver returns %d\n", i);
166 87a52e04 2005-12-26 devnull 	for(j = 0; j < i; j++)
167 87a52e04 2005-12-26 devnull 		print("%I\n", addrs[j]);
168 87a52e04 2005-12-26 devnull }