Subversion Repositories planix.SVN

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