Subversion Repositories planix.SVN

Rev

Details | Last modification | View Log | RSS feed

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