1 d957951b 2005-02-11 devnull #include <u.h>
2 d957951b 2005-02-11 devnull #include <libc.h>
3 d957951b 2005-02-11 devnull #include <bio.h>
4 d957951b 2005-02-11 devnull #include <ndb.h>
5 d957951b 2005-02-11 devnull #include <ip.h>
9 d957951b 2005-02-11 devnull Ffound= 1<<0,
10 cbeb0b26 2006-04-01 devnull Fignore= 1<<1,
11 cbeb0b26 2006-04-01 devnull Faddr= 1<<2
14 d957951b 2005-02-11 devnull static Ndbtuple* filter(Ndb *db, Ndbtuple *t, Ndbtuple *f);
15 d957951b 2005-02-11 devnull static Ndbtuple* mkfilter(int argc, char **argv);
16 d957951b 2005-02-11 devnull static int filtercomplete(Ndbtuple *f);
17 d957951b 2005-02-11 devnull static int prefixlen(uchar *ip);
18 d957951b 2005-02-11 devnull static Ndbtuple* subnet(Ndb *db, uchar *net, Ndbtuple *f, int prefix);
20 d957951b 2005-02-11 devnull /* make a filter to be used in filter */
21 d957951b 2005-02-11 devnull static Ndbtuple*
22 d957951b 2005-02-11 devnull mkfilter(int argc, char **argv)
24 d957951b 2005-02-11 devnull Ndbtuple *t, *first, *last;
27 d957951b 2005-02-11 devnull last = first = nil;
28 d957951b 2005-02-11 devnull while(argc-- > 0){
29 d957951b 2005-02-11 devnull t = ndbnew(0, 0);
30 d957951b 2005-02-11 devnull if(first)
31 d957951b 2005-02-11 devnull last->entry = t;
33 d957951b 2005-02-11 devnull first = t;
34 d957951b 2005-02-11 devnull last = t;
35 d957951b 2005-02-11 devnull p = *argv++;
36 d957951b 2005-02-11 devnull if(*p == '@'){
37 d957951b 2005-02-11 devnull t->ptr |= Faddr;
40 d957951b 2005-02-11 devnull strncpy(t->attr, p, sizeof(t->attr)-1);
42 d957951b 2005-02-11 devnull return first;
45 d957951b 2005-02-11 devnull /* return true if every pair of filter has been used */
46 d957951b 2005-02-11 devnull static int
47 d957951b 2005-02-11 devnull filtercomplete(Ndbtuple *f)
49 d957951b 2005-02-11 devnull for(; f; f = f->entry)
50 d957951b 2005-02-11 devnull if((f->ptr & Fignore) == 0)
51 d957951b 2005-02-11 devnull return 0;
52 d957951b 2005-02-11 devnull return 1;
55 d957951b 2005-02-11 devnull /* set the attribute of all entries in a tuple */
56 d957951b 2005-02-11 devnull static Ndbtuple*
57 d957951b 2005-02-11 devnull setattr(Ndbtuple *t, char *attr)
59 d957951b 2005-02-11 devnull Ndbtuple *nt;
61 d957951b 2005-02-11 devnull for(nt = t; nt; nt = nt->entry)
62 d957951b 2005-02-11 devnull strcpy(nt->attr, attr);
63 d957951b 2005-02-11 devnull return t;
67 d957951b 2005-02-11 devnull * return only the attr/value pairs in t maching the filter, f.
68 d957951b 2005-02-11 devnull * others are freed. line structure is preserved.
70 d957951b 2005-02-11 devnull static Ndbtuple*
71 d957951b 2005-02-11 devnull filter(Ndb *db, Ndbtuple *t, Ndbtuple *f)
73 d957951b 2005-02-11 devnull Ndbtuple *nt, *nf, *next;
75 d957951b 2005-02-11 devnull /* filter out what we don't want */
76 d957951b 2005-02-11 devnull for(nt = t; nt; nt = next){
77 d957951b 2005-02-11 devnull next = nt->entry;
79 d957951b 2005-02-11 devnull /* look through filter */
80 d957951b 2005-02-11 devnull for(nf = f; nf != nil; nf = nf->entry){
81 d957951b 2005-02-11 devnull if(!(nf->ptr&Fignore) && strcmp(nt->attr, nf->attr) == 0)
84 d957951b 2005-02-11 devnull if(nf == nil){
85 d957951b 2005-02-11 devnull /* remove nt from t */
86 d957951b 2005-02-11 devnull t = ndbdiscard(t, nt);
88 d957951b 2005-02-11 devnull if(nf->ptr & Faddr)
89 d957951b 2005-02-11 devnull t = ndbsubstitute(t, nt, setattr(ndbgetipaddr(db, nt->val), nt->attr));
90 d957951b 2005-02-11 devnull nf->ptr |= Ffound;
94 d957951b 2005-02-11 devnull /* remember filter etnries that matched */
95 d957951b 2005-02-11 devnull for(nf = f; nf != nil; nf = nf->entry)
96 d957951b 2005-02-11 devnull if(nf->ptr & Ffound)
97 d957951b 2005-02-11 devnull nf->ptr = (nf->ptr & ~Ffound) | Fignore;
99 d957951b 2005-02-11 devnull return t;
102 d957951b 2005-02-11 devnull static int
103 d957951b 2005-02-11 devnull prefixlen(uchar *ip)
105 d957951b 2005-02-11 devnull int y, i;
107 d957951b 2005-02-11 devnull for(y = IPaddrlen-1; y >= 0; y--)
108 d957951b 2005-02-11 devnull for(i = 8; i > 0; i--)
109 d957951b 2005-02-11 devnull if(ip[y] & (1<<(8-i)))
110 d957951b 2005-02-11 devnull return y*8 + i;
111 d957951b 2005-02-11 devnull return 0;
115 d957951b 2005-02-11 devnull * look through a containing subset
117 d957951b 2005-02-11 devnull static Ndbtuple*
118 d957951b 2005-02-11 devnull subnet(Ndb *db, uchar *net, Ndbtuple *f, int prefix)
121 d957951b 2005-02-11 devnull Ndbtuple *t, *nt, *xt;
122 d957951b 2005-02-11 devnull char netstr[128];
123 d957951b 2005-02-11 devnull uchar mask[IPaddrlen];
124 d957951b 2005-02-11 devnull int masklen;
126 d957951b 2005-02-11 devnull t = nil;
127 d957951b 2005-02-11 devnull sprint(netstr, "%I", net);
128 d957951b 2005-02-11 devnull nt = ndbsearch(db, &s, "ip", netstr);
129 d957951b 2005-02-11 devnull while(nt != nil){
130 d957951b 2005-02-11 devnull xt = ndbfindattr(nt, nt, "ipnet");
132 d957951b 2005-02-11 devnull xt = ndbfindattr(nt, nt, "ipmask");
134 d957951b 2005-02-11 devnull parseipmask(mask, xt->val);
136 d957951b 2005-02-11 devnull ipmove(mask, defmask(net));
137 d957951b 2005-02-11 devnull masklen = prefixlen(mask);
138 d957951b 2005-02-11 devnull if(masklen <= prefix)
139 d957951b 2005-02-11 devnull t = ndbconcatenate(t, filter(db, nt, f));
141 d957951b 2005-02-11 devnull ndbfree(nt);
142 d957951b 2005-02-11 devnull nt = ndbsnext(&s, "ip", netstr);
144 d957951b 2005-02-11 devnull return t;
148 d957951b 2005-02-11 devnull * fill in all the requested attributes for a system.
149 d957951b 2005-02-11 devnull * if the system's entry doesn't have all required,
150 d957951b 2005-02-11 devnull * walk through successively more inclusive networks
151 d957951b 2005-02-11 devnull * for inherited attributes.
153 d957951b 2005-02-11 devnull Ndbtuple*
154 d957951b 2005-02-11 devnull ndbipinfo(Ndb *db, char *attr, char *val, char **alist, int n)
156 d957951b 2005-02-11 devnull Ndbtuple *t, *nt, *f;
158 d957951b 2005-02-11 devnull char *ipstr;
159 d957951b 2005-02-11 devnull uchar net[IPaddrlen];
160 d957951b 2005-02-11 devnull uchar ip[IPaddrlen];
161 d957951b 2005-02-11 devnull int prefix, smallestprefix;
162 d957951b 2005-02-11 devnull int force;
164 d957951b 2005-02-11 devnull /* just in case */
165 d957951b 2005-02-11 devnull fmtinstall('I', eipfmt);
166 d957951b 2005-02-11 devnull fmtinstall('M', eipfmt);
168 d957951b 2005-02-11 devnull /* get needed attributes */
169 d957951b 2005-02-11 devnull f = mkfilter(n, alist);
172 d957951b 2005-02-11 devnull * first look for a matching entry with an ip address
174 d957951b 2005-02-11 devnull t = nil;
175 d957951b 2005-02-11 devnull ipstr = ndbgetvalue(db, &s, attr, val, "ip", &nt);
176 d957951b 2005-02-11 devnull if(ipstr == nil){
177 d957951b 2005-02-11 devnull /* none found, make one up */
178 d957951b 2005-02-11 devnull if(strcmp(attr, "ip") != 0)
179 d957951b 2005-02-11 devnull return nil;
180 d957951b 2005-02-11 devnull t = ndbnew("ip", val);
181 d957951b 2005-02-11 devnull t->line = t;
182 d957951b 2005-02-11 devnull t->entry = nil;
183 d957951b 2005-02-11 devnull parseip(net, val);
184 d957951b 2005-02-11 devnull } else {
185 d957951b 2005-02-11 devnull /* found one */
186 d957951b 2005-02-11 devnull while(nt != nil){
187 d957951b 2005-02-11 devnull nt = ndbreorder(nt, s.t);
188 d957951b 2005-02-11 devnull t = ndbconcatenate(t, nt);
189 d957951b 2005-02-11 devnull nt = ndbsnext(&s, attr, val);
191 d957951b 2005-02-11 devnull parseip(net, ipstr);
192 d957951b 2005-02-11 devnull free(ipstr);
194 d957951b 2005-02-11 devnull ipmove(ip, net);
195 d957951b 2005-02-11 devnull t = filter(db, t, f);
198 d957951b 2005-02-11 devnull * now go through subnets to fill in any missing attributes
200 d957951b 2005-02-11 devnull if(isv4(net)){
201 d957951b 2005-02-11 devnull prefix = 127;
202 d957951b 2005-02-11 devnull smallestprefix = 100;
203 d957951b 2005-02-11 devnull force = 0;
204 d957951b 2005-02-11 devnull } else {
205 d957951b 2005-02-11 devnull /* in v6, the last 8 bytes have no structure (we hope) */
206 d957951b 2005-02-11 devnull prefix = 64;
207 d957951b 2005-02-11 devnull smallestprefix = 2;
208 d957951b 2005-02-11 devnull memset(net+8, 0, 8);
209 d957951b 2005-02-11 devnull force = 1;
213 d957951b 2005-02-11 devnull * to find a containing network, keep turning off
214 d957951b 2005-02-11 devnull * the lower bit and look for a network with
215 d957951b 2005-02-11 devnull * that address and a shorter mask. tedius but
216 d957951b 2005-02-11 devnull * complete, we may need to find a trick to speed this up.
218 d957951b 2005-02-11 devnull for(; prefix >= smallestprefix; prefix--){
219 d957951b 2005-02-11 devnull if(filtercomplete(f))
221 d957951b 2005-02-11 devnull if(!force && (net[prefix/8] & (1<<(7-(prefix%8)))) == 0)
222 d957951b 2005-02-11 devnull continue;
223 d957951b 2005-02-11 devnull force = 0;
224 d957951b 2005-02-11 devnull net[prefix/8] &= ~(1<<(7-(prefix%8)));
225 d957951b 2005-02-11 devnull t = ndbconcatenate(t, subnet(db, net, f, prefix));
229 d957951b 2005-02-11 devnull * if there's an unfulfilled ipmask, make one up
231 d957951b 2005-02-11 devnull nt = ndbfindattr(f, f, "ipmask");
232 d957951b 2005-02-11 devnull if(nt && !(nt->ptr & Fignore)){
233 d957951b 2005-02-11 devnull char x[64];
235 d957951b 2005-02-11 devnull snprint(x, sizeof(x), "%M", defmask(ip));
236 d957951b 2005-02-11 devnull t = ndbconcatenate(t, ndbnew("ipmask", x));
239 d957951b 2005-02-11 devnull ndbfree(f);
240 d957951b 2005-02-11 devnull return t;
