Home | History | Annotate | Download | only in server 
      1 /***************************************************************************
      2  *                                  _   _ ____  _
      3  *  Project                     ___| | | |  _ \| |
      4  *                             / __| | | | |_) | |
      5  *                            | (__| |_| |  _ <| |___
      6  *                             \___|\___/|_| \_\_____|
      7  *
      8  * Copyright (C) 1998 - 2017, Daniel Stenberg, <daniel (at) 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