sys/src/libndb/ndbipinfo.c

   1 #include <u.h>
   2 #include <libc.h>
   3 #include <bio.h>
   4 #include <ndb.h>
   5 #include <ip.h>
   6
   7 enum
   8 {
   9         Ffound= 1<<0,
  10         Fignore=1<<1,
  11         Faddr=  1<<2,
  12 };
  13
  14 static Ndbtuple*        filter(Ndb *db, Ndbtuple *t, Ndbtuple *f);
  15 static Ndbtuple*        mkfilter(int argc, char **argv);
  16 static int              filtercomplete(Ndbtuple *f);
  17 static int              prefixlen(uchar *ip);
  18 static Ndbtuple*        subnet(Ndb *db, uchar *net, Ndbtuple *f, int prefix);
  19
  20 /* make a filter to be used in filter */
  21 static Ndbtuple*
  22 mkfilter(int argc, char **argv)
  23 {
  24         Ndbtuple *t, *first, *last;
  25         char *p;
  26
  27         last = first = nil;
  28         while(argc-- > 0){
  29                 t = ndbnew(0, 0);
  30                 if(first)
  31                         last->entry = t;
  32                 else
  33                         first = t;
  34                 last = t;
  35                 p = *argv++;
  36                 if(*p == '@'){                  /* @attr=val ? */
  37                         t->ptr |= Faddr;        /* return resolved address(es) */
  38                         p++;
  39                 }
  40                 strncpy(t->attr, p, sizeof(t->attr)-1);
  41         }
  42         ndbsetmalloctag(first, getcallerpc(&argc));
  43         return first;
  44 }
  45
  46 /* return true if every pair of filter has been used */
  47 static int
  48 filtercomplete(Ndbtuple *f)
  49 {
  50         for(; f; f = f->entry)
  51                 if((f->ptr & Fignore) == 0)
  52                         return 0;
  53         return 1;
  54 }
  55
  56 /* set the attribute of all entries in a tuple */
  57 static Ndbtuple*
  58 setattr(Ndbtuple *t, char *attr)
  59 {
  60         Ndbtuple *nt;
  61
  62         for(nt = t; nt; nt = nt->entry)
  63                 strcpy(nt->attr, attr);
  64         return t;
  65 }
  66
  67 /*
  68  *  return only the attr/value pairs in t maching the filter, f.
  69  *  others are freed.  line structure is preserved.
  70  */
  71 static Ndbtuple*
  72 filter(Ndb *db, Ndbtuple *t, Ndbtuple *f)
  73 {
  74         Ndbtuple *nt, *nf, *next;
  75
  76         /* filter out what we don't want */
  77         for(nt = t; nt; nt = next){
  78                 next = nt->entry;
  79
  80                 /* look through filter */
  81                 for(nf = f; nf != nil; nf = nf->entry){
  82                         if(!(nf->ptr&Fignore) && strcmp(nt->attr, nf->attr) == 0)
  83                                 break;
  84                 }
  85                 if(nf == nil){
  86                         /* remove nt from t */
  87                         t = ndbdiscard(t, nt);
  88                 } else {
  89                         if(nf->ptr & Faddr)
  90                                 t = ndbsubstitute(t, nt, setattr(ndbgetipaddr(db, nt->val), nt->attr));
  91                         nf->ptr |= Ffound;
  92                 }
  93         }
  94
  95         /* remember filter etnries that matched */
  96         for(nf = f; nf != nil; nf = nf->entry)
  97                 if(nf->ptr & Ffound)
  98                         nf->ptr = (nf->ptr & ~Ffound) | Fignore;
  99
 100         ndbsetmalloctag(t, getcallerpc(&db));
 101         return t;
 102 }
 103
 104 static int
 105 prefixlen(uchar *ip)
 106 {
 107         int y, i;
 108
 109         for(y = IPaddrlen-1; y >= 0; y--)
 110                 for(i = 8; i > 0; i--)
 111                         if(ip[y] & (1<<(8-i)))
 112                                 return y*8 + i;
 113         return 0;
 114 }
 115
 116 /*
 117  *  look through a containing subset
 118  */
 119 static Ndbtuple*
 120 subnet(Ndb *db, uchar *net, Ndbtuple *f, int prefix)
 121 {
 122         Ndbs s;
 123         char netstr[64];
 124         uchar mask[IPaddrlen];
 125         Ndbtuple *t, *nt, *xt;
 126
 127         t = nil;
 128         snprint(netstr, sizeof(netstr), "%I", net);
 129         nt = ndbsearch(db, &s, "ip", netstr);
 130         while(nt != nil){
 131                 xt = ndbfindattr(nt, nt, "ipnet");
 132                 if(xt != nil){
 133                         xt = ndbfindattr(nt, nt, "ipmask");
 134                         if(xt == nil || parseipmask(mask, xt->val, isv4(net)) == -1)
 135                                 ipmove(mask, defmask(net));
 136                         if(prefixlen(mask) <= prefix){
 137                                 t = ndbconcatenate(t, filter(db, nt, f));
 138                                 nt = nil;
 139                         }
 140                 }
 141                 ndbfree(nt);
 142                 nt = ndbsnext(&s, "ip", netstr);
 143         }
 144         ndbsetmalloctag(t, getcallerpc(&db));
 145         return t;
 146 }
 147
 148 static Ndbtuple*
 149 netinfo(Ndb *db, Ndbtuple *t, char **alist, int n)
 150 {
 151         uchar ip[IPaddrlen], net[IPaddrlen];
 152         int prefix, smallestprefix, force;
 153         Ndbtuple *f, *nt;
 154
 155         nt = ndbfindattr(t, t, "ip");
 156         if(nt == nil || parseip(ip, nt->val) == -1){
 157                 ndbfree(t);
 158                 return nil;
 159         }
 160
 161         /* get needed attributes */
 162         f = mkfilter(n, alist);
 163
 164         t = filter(db, t, f);
 165
 166         /*
 167          *  now go through subnets to fill in any missing attributes
 168          */
 169         ipmove(net, ip);
 170         if(isv4(ip)){
 171                 prefix = 127;
 172                 smallestprefix = 100;
 173                 force = 0;
 174         } else {
 175                 /* in v6, the last 8 bytes have no structure (we hope) */
 176                 prefix = 64;
 177                 smallestprefix = 2;
 178                 memset(net+8, 0, 8);
 179                 force = 1;
 180         }
 181
 182         /*
 183          *  to find a containing network, keep turning off
 184          *  the lower bit and look for a network with
 185          *  that address and a shorter mask.  tedius but
 186          *  complete, we may need to find a trick to speed this up.
 187          */
 188         for(; prefix >= smallestprefix; prefix--){
 189                 if(filtercomplete(f))
 190                         break;
 191                 if(!force && (net[prefix/8] & (1<<(7-(prefix%8)))) == 0)
 192                         continue;
 193                 force = 0;
 194                 net[prefix/8] &= ~(1<<(7-(prefix%8)));
 195                 t = ndbconcatenate(t, subnet(db, net, f, prefix));
 196         }
 197
 198         /*
 199          *  if there's an unfulfilled ipmask, make one up
 200          */
 201         nt = ndbfindattr(f, f, "ipmask");
 202         if(nt != nil && !(nt->ptr & Fignore)){
 203                 char x[64];
 204
 205                 snprint(x, sizeof(x), "%M", defmask(ip));
 206                 nt = ndbnew("ipmask", x);
 207                 nt->line = nt;
 208                 nt->entry = nil;
 209                 t = ndbconcatenate(t, nt);
 210         }
 211
 212         ndbfree(f);
 213         ndbsetmalloctag(t, getcallerpc(&db));
 214         return t;
 215 }
 216
 217 /*
 218  *  fill in all the requested attributes for a system.
 219  *  if the system's entry doesn't have all required,
 220  *  walk through successively more inclusive networks
 221  *  for inherited attributes.
 222  */
 223 Ndbtuple*
 224 ndbipinfo(Ndb *db, char *attr, char *val, char **alist, int n)
 225 {
 226         Ndbtuple *t, *nt;
 227         char *ipstr;
 228         Ndbs s;
 229
 230         /* just in case */
 231         fmtinstall('I', eipfmt);
 232         fmtinstall('M', eipfmt);
 233
 234         /*
 235          *  first look for a matching entry with an ip address
 236          */
 237         ipstr = ndbgetvalue(db, &s, attr, val, "ip", &nt);
 238         if(ipstr == nil){
 239                 /* none found, make one up */
 240                 if(strcmp(attr, "ip") != 0)
 241                         return nil;
 242                 nt = ndbnew("ip", val);
 243                 nt->line = nt;
 244                 nt->entry = nil;
 245                 t = netinfo(db, nt, alist, n);
 246         } else {
 247                 /* found one */
 248                 free(ipstr);
 249                 t = nil;
 250                 do {
 251                         nt = ndbreorder(nt, s.t);
 252                         t = ndbconcatenate(t, netinfo(db, nt, alist, n));
 253                 } while((nt = ndbsnext(&s, attr, val)) != nil);
 254         }
 255         return ndbdedup(t);
 256 }