1 /*
2 * Copyright (c) 2003-2006 Niels Provos <provos@citi.umich.edu>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28 #ifdef WIN32
29 #include <winsock2.h>
30 #include <windows.h>
31 #endif
32
33 #ifdef HAVE_CONFIG_H
34 #include "config.h"
35 #endif
36
37 #include <sys/types.h>
38 #include <sys/stat.h>
39 #ifdef HAVE_SYS_TIME_H
40 #include <sys/time.h>
41 #endif
42 #include <sys/queue.h>
43 #ifndef WIN32
44 #include <sys/socket.h>
45 #include <signal.h>
46 #include <netinet/in.h>
47 #include <arpa/inet.h>
48 #include <unistd.h>
49 #endif
50 #ifdef HAVE_NETINET_IN6_H
51 #include <netinet/in6.h>
52 #endif
53 #ifdef HAVE_NETDB_H
54 #include <netdb.h>
55 #endif
56 #include <fcntl.h>
57 #include <stdlib.h>
58 #include <stdio.h>
59 #include <string.h>
60 #include <errno.h>
61
62 #include "event.h"
63 #include "evdns.h"
64 #include "log.h"
65
66 static int dns_ok = 0;
67 static int dns_err = 0;
68
69 void dns_suite(void);
70
71 static void
dns_gethostbyname_cb(int result,char type,int count,int ttl,void * addresses,void * arg)72 dns_gethostbyname_cb(int result, char type, int count, int ttl,
73 void *addresses, void *arg)
74 {
75 dns_ok = dns_err = 0;
76
77 if (result == DNS_ERR_TIMEOUT) {
78 fprintf(stdout, "[Timed out] ");
79 dns_err = result;
80 goto out;
81 }
82
83 if (result != DNS_ERR_NONE) {
84 fprintf(stdout, "[Error code %d] ", result);
85 goto out;
86 }
87
88 fprintf(stderr, "type: %d, count: %d, ttl: %d: ", type, count, ttl);
89
90 switch (type) {
91 case DNS_IPv6_AAAA: {
92 #if defined(HAVE_STRUCT_IN6_ADDR) && defined(HAVE_INET_NTOP) && defined(INET6_ADDRSTRLEN)
93 struct in6_addr *in6_addrs = addresses;
94 char buf[INET6_ADDRSTRLEN+1];
95 int i;
96 /* a resolution that's not valid does not help */
97 if (ttl < 0)
98 goto out;
99 for (i = 0; i < count; ++i) {
100 const char *b = inet_ntop(AF_INET6, &in6_addrs[i], buf,sizeof(buf));
101 if (b)
102 fprintf(stderr, "%s ", b);
103 else
104 fprintf(stderr, "%s ", strerror(errno));
105 }
106 #endif
107 break;
108 }
109 case DNS_IPv4_A: {
110 struct in_addr *in_addrs = addresses;
111 int i;
112 /* a resolution that's not valid does not help */
113 if (ttl < 0)
114 goto out;
115 for (i = 0; i < count; ++i)
116 fprintf(stderr, "%s ", inet_ntoa(in_addrs[i]));
117 break;
118 }
119 case DNS_PTR:
120 /* may get at most one PTR */
121 if (count != 1)
122 goto out;
123
124 fprintf(stderr, "%s ", *(char **)addresses);
125 break;
126 default:
127 goto out;
128 }
129
130 dns_ok = type;
131
132 out:
133 event_loopexit(NULL);
134 }
135
136 static void
dns_gethostbyname(void)137 dns_gethostbyname(void)
138 {
139 fprintf(stdout, "Simple DNS resolve: ");
140 dns_ok = 0;
141 evdns_resolve_ipv4("www.monkey.org", 0, dns_gethostbyname_cb, NULL);
142 event_dispatch();
143
144 if (dns_ok == DNS_IPv4_A) {
145 fprintf(stdout, "OK\n");
146 } else {
147 fprintf(stdout, "FAILED\n");
148 exit(1);
149 }
150 }
151
152 static void
dns_gethostbyname6(void)153 dns_gethostbyname6(void)
154 {
155 fprintf(stdout, "IPv6 DNS resolve: ");
156 dns_ok = 0;
157 evdns_resolve_ipv6("www.ietf.org", 0, dns_gethostbyname_cb, NULL);
158 event_dispatch();
159
160 if (dns_ok == DNS_IPv6_AAAA) {
161 fprintf(stdout, "OK\n");
162 } else if (!dns_ok && dns_err == DNS_ERR_TIMEOUT) {
163 fprintf(stdout, "SKIPPED\n");
164 } else {
165 fprintf(stdout, "FAILED (%d)\n", dns_ok);
166 exit(1);
167 }
168 }
169
170 static void
dns_gethostbyaddr(void)171 dns_gethostbyaddr(void)
172 {
173 struct in_addr in;
174 in.s_addr = htonl(0x7f000001ul); /* 127.0.0.1 */
175 fprintf(stdout, "Simple reverse DNS resolve: ");
176 dns_ok = 0;
177 evdns_resolve_reverse(&in, 0, dns_gethostbyname_cb, NULL);
178 event_dispatch();
179
180 if (dns_ok == DNS_PTR) {
181 fprintf(stdout, "OK\n");
182 } else {
183 fprintf(stdout, "FAILED\n");
184 exit(1);
185 }
186 }
187
188 static int n_server_responses = 0;
189
190 static void
dns_server_request_cb(struct evdns_server_request * req,void * data)191 dns_server_request_cb(struct evdns_server_request *req, void *data)
192 {
193 int i, r;
194 const char TEST_ARPA[] = "11.11.168.192.in-addr.arpa";
195 for (i = 0; i < req->nquestions; ++i) {
196 struct in_addr ans;
197 ans.s_addr = htonl(0xc0a80b0bUL); /* 192.168.11.11 */
198 if (req->questions[i]->type == EVDNS_TYPE_A &&
199 req->questions[i]->dns_question_class == EVDNS_CLASS_INET &&
200 !strcmp(req->questions[i]->name, "zz.example.com")) {
201 r = evdns_server_request_add_a_reply(req, "zz.example.com",
202 1, &ans.s_addr, 12345);
203 if (r<0)
204 dns_ok = 0;
205 } else if (req->questions[i]->type == EVDNS_TYPE_AAAA &&
206 req->questions[i]->dns_question_class == EVDNS_CLASS_INET &&
207 !strcmp(req->questions[i]->name, "zz.example.com")) {
208 char addr6[17] = "abcdefghijklmnop";
209 r = evdns_server_request_add_aaaa_reply(req, "zz.example.com",
210 1, addr6, 123);
211 if (r<0)
212 dns_ok = 0;
213 } else if (req->questions[i]->type == EVDNS_TYPE_PTR &&
214 req->questions[i]->dns_question_class == EVDNS_CLASS_INET &&
215 !strcmp(req->questions[i]->name, TEST_ARPA)) {
216 r = evdns_server_request_add_ptr_reply(req, NULL, TEST_ARPA,
217 "ZZ.EXAMPLE.COM", 54321);
218 if (r<0)
219 dns_ok = 0;
220 } else {
221 fprintf(stdout, "Unexpected question %d %d \"%s\" ",
222 req->questions[i]->type,
223 req->questions[i]->dns_question_class,
224 req->questions[i]->name);
225 dns_ok = 0;
226 }
227 }
228 r = evdns_server_request_respond(req, 0);
229 if (r<0) {
230 fprintf(stdout, "Couldn't send reply. ");
231 dns_ok = 0;
232 }
233 }
234
235 static void
dns_server_gethostbyname_cb(int result,char type,int count,int ttl,void * addresses,void * arg)236 dns_server_gethostbyname_cb(int result, char type, int count, int ttl,
237 void *addresses, void *arg)
238 {
239 if (result != DNS_ERR_NONE) {
240 fprintf(stdout, "Unexpected result %d. ", result);
241 dns_ok = 0;
242 goto out;
243 }
244 if (count != 1) {
245 fprintf(stdout, "Unexpected answer count %d. ", count);
246 dns_ok = 0;
247 goto out;
248 }
249 switch (type) {
250 case DNS_IPv4_A: {
251 struct in_addr *in_addrs = addresses;
252 if (in_addrs[0].s_addr != htonl(0xc0a80b0bUL) || ttl != 12345) {
253 fprintf(stdout, "Bad IPv4 response \"%s\" %d. ",
254 inet_ntoa(in_addrs[0]), ttl);
255 dns_ok = 0;
256 goto out;
257 }
258 break;
259 }
260 case DNS_IPv6_AAAA: {
261 #if defined (HAVE_STRUCT_IN6_ADDR) && defined(HAVE_INET_NTOP) && defined(INET6_ADDRSTRLEN)
262 struct in6_addr *in6_addrs = addresses;
263 char buf[INET6_ADDRSTRLEN+1];
264 if (memcmp(&in6_addrs[0].s6_addr, "abcdefghijklmnop", 16)
265 || ttl != 123) {
266 const char *b = inet_ntop(AF_INET6, &in6_addrs[0],buf,sizeof(buf));
267 fprintf(stdout, "Bad IPv6 response \"%s\" %d. ", b, ttl);
268 dns_ok = 0;
269 goto out;
270 }
271 #endif
272 break;
273 }
274 case DNS_PTR: {
275 char **addrs = addresses;
276 if (strcmp(addrs[0], "ZZ.EXAMPLE.COM") || ttl != 54321) {
277 fprintf(stdout, "Bad PTR response \"%s\" %d. ",
278 addrs[0], ttl);
279 dns_ok = 0;
280 goto out;
281 }
282 break;
283 }
284 default:
285 fprintf(stdout, "Bad response type %d. ", type);
286 dns_ok = 0;
287 }
288
289 out:
290 if (++n_server_responses == 3) {
291 event_loopexit(NULL);
292 }
293 }
294
295 static void
dns_server(void)296 dns_server(void)
297 {
298 int sock;
299 struct sockaddr_in my_addr;
300 struct evdns_server_port *port;
301 struct in_addr resolve_addr;
302
303 dns_ok = 1;
304 fprintf(stdout, "DNS server support: ");
305
306 /* Add ourself as the only nameserver, and make sure we really are
307 * the only nameserver. */
308 evdns_nameserver_ip_add("127.0.0.1:35353");
309 if (evdns_count_nameservers() != 1) {
310 fprintf(stdout, "Couldn't set up.\n");
311 exit(1);
312 }
313
314 /* Now configure a nameserver port. */
315 sock = socket(AF_INET, SOCK_DGRAM, 0);
316 if (sock == -1) {
317 perror("socket");
318 exit(1);
319 }
320 #ifdef WIN32
321 {
322 u_long nonblocking = 1;
323 ioctlsocket(sock, FIONBIO, &nonblocking);
324 }
325 #else
326 fcntl(sock, F_SETFL, O_NONBLOCK);
327 #endif
328 memset(&my_addr, 0, sizeof(my_addr));
329 my_addr.sin_family = AF_INET;
330 my_addr.sin_port = htons(35353);
331 my_addr.sin_addr.s_addr = htonl(0x7f000001UL);
332 if (bind(sock, (struct sockaddr*)&my_addr, sizeof(my_addr)) < 0) {
333 perror("bind");
334 exit (1);
335 }
336 port = evdns_add_server_port(sock, 0, dns_server_request_cb, NULL);
337
338 /* Send two queries. */
339 evdns_resolve_ipv4("zz.example.com", DNS_QUERY_NO_SEARCH,
340 dns_server_gethostbyname_cb, NULL);
341 evdns_resolve_ipv6("zz.example.com", DNS_QUERY_NO_SEARCH,
342 dns_server_gethostbyname_cb, NULL);
343 resolve_addr.s_addr = htonl(0xc0a80b0bUL); /* 192.168.11.11 */
344 evdns_resolve_reverse(&resolve_addr, 0,
345 dns_server_gethostbyname_cb, NULL);
346
347 event_dispatch();
348
349 if (dns_ok) {
350 fprintf(stdout, "OK\n");
351 } else {
352 fprintf(stdout, "FAILED\n");
353 exit(1);
354 }
355
356 evdns_close_server_port(port);
357 evdns_shutdown(0); /* remove ourself as nameserver. */
358 #ifdef WIN32
359 closesocket(sock);
360 #else
361 close(sock);
362 #endif
363 }
364
365 void
dns_suite(void)366 dns_suite(void)
367 {
368 dns_server(); /* Do this before we call evdns_init. */
369
370 evdns_init();
371 dns_gethostbyname();
372 dns_gethostbyname6();
373 dns_gethostbyaddr();
374
375 evdns_shutdown(0);
376 }
377