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
66static int dns_ok = 0;
67static int dns_err = 0;
68
69void dns_suite(void);
70
71static void
72dns_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
132out:
133	event_loopexit(NULL);
134}
135
136static void
137dns_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
152static void
153dns_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
170static void
171dns_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
188static int n_server_responses = 0;
189
190static void
191dns_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
235static void
236dns_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
295static void
296dns_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
365void
366dns_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