1 /* 2 * Copyright (c) 2003-2006 Niels Provos <provos (at) 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 } 377
