1 | /*************************************************************************** |
2 | * _ _ ____ _ |
3 | * Project ___| | | | _ \| | |
4 | * / __| | | | |_) | | |
5 | * | (__| |_| | _ <| |___ |
6 | * \___|\___/|_| \_\_____| |
7 | * |
8 | * Copyright (C) 1998 - 2017, Daniel Stenberg, <daniel@haxx.se>, et al. |
9 | * |
10 | * This software is licensed as described in the file COPYING, which |
11 | * you should have received as part of this distribution. The terms |
12 | * are also available at https://curl.haxx.se/docs/copyright.html. |
13 | * |
14 | * You may opt to use, copy, modify, merge, publish, distribute and/or sell |
15 | * copies of the Software, and permit persons to whom the Software is |
16 | * furnished to do so, under the terms of the COPYING file. |
17 | * |
18 | * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY |
19 | * KIND, either express or implied. |
20 | * |
21 | ***************************************************************************/ |
22 | #include "server_setup.h" |
23 | |
24 | /* Purpose |
25 | * |
26 | * Resolve the given name, using system name resolve functions (NOT any |
27 | * function provided by libcurl). Used to see if the name exists and thus if |
28 | * we can allow a test case to use it for testing. |
29 | * |
30 | * Like if 'localhost' actual exists etc. |
31 | * |
32 | */ |
33 | |
34 | #ifdef HAVE_SIGNAL_H |
35 | #include <signal.h> |
36 | #endif |
37 | #ifdef HAVE_NETINET_IN_H |
38 | #include <netinet/in.h> |
39 | #endif |
40 | #ifdef _XOPEN_SOURCE_EXTENDED |
41 | /* This define is "almost" required to build on HPUX 11 */ |
42 | #include <arpa/inet.h> |
43 | #endif |
44 | #ifdef HAVE_NETDB_H |
45 | #include <netdb.h> |
46 | #endif |
47 | |
48 | #define ENABLE_CURLX_PRINTF |
49 | /* make the curlx header define all printf() functions to use the curlx_* |
50 | versions instead */ |
51 | #include "curlx.h" /* from the private lib dir */ |
52 | #include "util.h" |
53 | |
54 | /* include memdebug.h last */ |
55 | #include "memdebug.h" |
56 | |
57 | static bool use_ipv6 = FALSE; |
58 | static const char *ipv_inuse = "IPv4" ; |
59 | |
60 | const char *serverlogfile = "" ; /* for a util.c function we don't use */ |
61 | |
62 | int main(int argc, char *argv[]) |
63 | { |
64 | int arg = 1; |
65 | const char *host = NULL; |
66 | int rc = 0; |
67 | |
68 | while(argc>arg) { |
69 | if(!strcmp("--version" , argv[arg])) { |
70 | printf("resolve IPv4%s\n" , |
71 | #ifdef ENABLE_IPV6 |
72 | "/IPv6" |
73 | #else |
74 | "" |
75 | #endif |
76 | ); |
77 | return 0; |
78 | } |
79 | else if(!strcmp("--ipv6" , argv[arg])) { |
80 | ipv_inuse = "IPv6" ; |
81 | use_ipv6 = TRUE; |
82 | arg++; |
83 | } |
84 | else if(!strcmp("--ipv4" , argv[arg])) { |
85 | /* for completeness, we support this option as well */ |
86 | ipv_inuse = "IPv4" ; |
87 | use_ipv6 = FALSE; |
88 | arg++; |
89 | } |
90 | else { |
91 | host = argv[arg++]; |
92 | } |
93 | } |
94 | if(!host) { |
95 | puts("Usage: resolve [option] <host>\n" |
96 | " --version\n" |
97 | " --ipv4" |
98 | #ifdef ENABLE_IPV6 |
99 | "\n --ipv6" |
100 | #endif |
101 | ); |
102 | return 1; |
103 | } |
104 | |
105 | #ifdef WIN32 |
106 | win32_init(); |
107 | atexit(win32_cleanup); |
108 | #endif |
109 | |
110 | #ifdef ENABLE_IPV6 |
111 | if(use_ipv6) { |
112 | /* Check that the system has IPv6 enabled before checking the resolver */ |
113 | curl_socket_t s = socket(PF_INET6, SOCK_DGRAM, 0); |
114 | if(s == CURL_SOCKET_BAD) |
115 | /* an IPv6 address was requested and we can't get/use one */ |
116 | rc = -1; |
117 | else { |
118 | sclose(s); |
119 | } |
120 | } |
121 | |
122 | if(rc == 0) { |
123 | /* getaddrinfo() resolve */ |
124 | struct addrinfo *ai; |
125 | struct addrinfo hints; |
126 | |
127 | memset(&hints, 0, sizeof(hints)); |
128 | hints.ai_family = use_ipv6 ? PF_INET6 : PF_INET; |
129 | hints.ai_socktype = SOCK_STREAM; |
130 | hints.ai_flags = AI_CANONNAME; |
131 | /* Use parenthesis around functions to stop them from being replaced by |
132 | the macro in memdebug.h */ |
133 | rc = (getaddrinfo)(host, "80" , &hints, &ai); |
134 | if(rc == 0) |
135 | (freeaddrinfo)(ai); |
136 | } |
137 | #else |
138 | if(use_ipv6) { |
139 | puts("IPv6 support has been disabled in this program" ); |
140 | return 1; |
141 | } |
142 | else { |
143 | /* gethostbyname() resolve */ |
144 | struct hostent *he; |
145 | |
146 | he = gethostbyname(host); |
147 | |
148 | rc = !he; |
149 | } |
150 | #endif |
151 | |
152 | if(rc) |
153 | printf("Resolving %s '%s' didn't work\n" , ipv_inuse, host); |
154 | |
155 | return !!rc; |
156 | } |
157 | |