tests/libtest/lib1515.c

   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
  23 /*
  24  * Check for bugs #1303 and #1327: libcurl should never remove DNS entries
  25  * created via CURLOPT_RESOLVE, neither after DNS_CACHE_TIMEOUT elapses
  26  * (test1515) nor a dead connection is detected (test1616).
  27  */
  28
  29 #include "test.h"
  30 #include "testutil.h"
  31 #include "warnless.h"
  32 #include "memdebug.h"
  33
  34 #define TEST_HANG_TIMEOUT 60 * 1000
  35
  36 #define DNS_TIMEOUT 1
  37
  38 #if defined(WIN32) || defined(_WIN32)
  39 #define sleep(s) Sleep(s * 1000)
  40 #endif
  41
  42 static int debug_callback(CURL *curl, curl_infotype info, char *msg,
  43                           size_t len, void *ptr)
  44 {
  45   (void)curl;
  46   (void)ptr;
  47
  48   if(info == CURLINFO_TEXT)
  49     fprintf(stderr, "debug: %.*s", (int) len, msg);
  50
  51   return 0;
  52 }
  53
  54 static int do_one_request(CURLM *m, char *URL, char *resolve)
  55 {
  56   CURL *curls;
  57   struct curl_slist *resolve_list = NULL;
  58   int still_running;
  59   int res = 0;
  60   CURLMsg *msg;
  61   int msgs_left;
  62
  63   resolve_list = curl_slist_append(resolve_list, resolve);
  64
  65   easy_init(curls);
  66
  67   easy_setopt(curls, CURLOPT_URL, URL);
  68   easy_setopt(curls, CURLOPT_RESOLVE, resolve_list);
  69   easy_setopt(curls, CURLOPT_DEBUGFUNCTION, debug_callback);
  70   easy_setopt(curls, CURLOPT_VERBOSE, 1);
  71   easy_setopt(curls, CURLOPT_DNS_CACHE_TIMEOUT, DNS_TIMEOUT);
  72
  73   multi_add_handle(m, curls);
  74   multi_perform(m, &still_running);
  75
  76   abort_on_test_timeout();
  77
  78   while(still_running) {
  79     struct timeval timeout;
  80     fd_set fdread, fdwrite, fdexcep;
  81     int maxfd = -99;
  82
  83     FD_ZERO(&fdread);
  84     FD_ZERO(&fdwrite);
  85     FD_ZERO(&fdexcep);
  86     timeout.tv_sec = 1;
  87     timeout.tv_usec = 0;
  88
  89     multi_fdset(m, &fdread, &fdwrite, &fdexcep, &maxfd);
  90     select_test(maxfd + 1, &fdread, &fdwrite, &fdexcep, &timeout);
  91
  92     abort_on_test_timeout();
  93     multi_perform(m, &still_running);
  94
  95     abort_on_test_timeout();
  96   }
  97
  98   do {
  99     msg = curl_multi_info_read(m, &msgs_left);
 100     if(msg && msg->msg == CURLMSG_DONE && msg->easy_handle == curls) {
 101       res = msg->data.result;
 102       break;
 103     }
 104   } while(msg);
 105
 106 test_cleanup:
 107
 108   curl_multi_remove_handle(m, curls);
 109   curl_easy_cleanup(curls);
 110   curl_slist_free_all(resolve_list);
 111
 112   return res;
 113 }
 114
 115 int test(char *URL)
 116 {
 117   CURLM *multi = NULL;
 118   int res = 0;
 119   char *address = libtest_arg2;
 120   char *port = libtest_arg3;
 121   char *path = URL;
 122   char dns_entry[256];
 123   int i;
 124   int count = 2;
 125
 126   snprintf(dns_entry, sizeof(dns_entry), "testserver.example.com:%s:%s",
 127            port, address);
 128
 129   start_test_timing();
 130
 131   global_init(CURL_GLOBAL_ALL);
 132   multi_init(multi);
 133
 134   for(i = 1; i <= count; i++) {
 135     char target_url[256];
 136     snprintf(target_url, sizeof(target_url),
 137              "http://testserver.example.com:%s/%s%04d", port, path, i);
 138
 139     /* second request must succeed like the first one */
 140     res = do_one_request(multi, target_url, dns_entry);
 141     if(res)
 142       goto test_cleanup;
 143
 144     if(i < count)
 145       sleep(DNS_TIMEOUT + 1);
 146   }
 147
 148 test_cleanup:
 149
 150   curl_multi_cleanup(multi);
 151   curl_global_cleanup();
 152
 153   return (int) res;
 154 }