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