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
main(int argc,char * argv[])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