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
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
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
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
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
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
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
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
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: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to regress_dns.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / libevent / test