1/*************************************************************************** 2 * _ _ ____ _ 3 * Project ___| | | | _ \| | 4 * / __| | | | |_) | | 5 * | (__| |_| | _ <| |___ 6 * \___|\___/|_| \_\_____| 7 * 8 * Copyright (C) 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.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 * SPDX-License-Identifier: curl 22 * 23 ***************************************************************************/ 24#include "curlcheck.h" 25 26#include "urldata.h" 27#include "connect.h" 28#include "share.h" 29 30#include "memdebug.h" /* LAST include file */ 31 32static void unit_stop(void) 33{ 34 curl_global_cleanup(); 35} 36 37static CURLcode unit_setup(void) 38{ 39 CURLcode res = CURLE_OK; 40 41 global_init(CURL_GLOBAL_ALL); 42 43 return res; 44} 45 46struct testcase { 47 /* host:port:address[,address]... */ 48 const char *optval; 49 50 /* lowercase host and port to retrieve the addresses from hostcache */ 51 const char *host; 52 int port; 53 54 /* 0 to 9 addresses expected from hostcache */ 55 const char *address[10]; 56}; 57 58 59/* CURLOPT_RESOLVE address parsing test - to test the following defect fix: 60 61 1) if there is already existing host:port pair in the DNS cache and 62 we call CURLOPT_RESOLVE, it should also replace addresses. 63 for example, if there is "test.com:80" with address "1.1.1.1" 64 and we called CURLOPT_RESOLVE with address "2.2.2.2", then DNS entry needs to 65 reflect that. 66 67 2) when cached address is already there and close to expire, then by the 68 time request is made, it can get expired. This happens because, when 69 we set address using CURLOPT_RESOLVE, 70 it usually marks as permanent (by setting timestamp to zero). However, 71 if address already exists 72in the cache, then it does not mark it, but just leaves it as it is. 73 So we fixing this by timestamp to zero if address already exists too. 74 75Test: 76 77 - insert new entry 78 - verify that timestamp is not zero 79 - call set options with CURLOPT_RESOLVE 80 - then, call Curl_loadhostpairs 81 82 expected result: cached address has zero timestamp. 83 84 - call set options with CURLOPT_RESOLVE with same host:port pair, 85 different address. 86 - then, call Curl_loadhostpairs 87 88 expected result: cached address has zero timestamp and new address 89*/ 90 91static const struct testcase tests[] = { 92 /* spaces aren't allowed, for now */ 93 { "test.com:80:127.0.0.1", 94 "test.com", 80, { "127.0.0.1", } 95 }, 96 { "test.com:80:127.0.0.2", 97 "test.com", 80, { "127.0.0.2", } 98 }, 99}; 100 101UNITTEST_START 102{ 103 int i; 104 int testnum = sizeof(tests) / sizeof(struct testcase); 105 struct Curl_multi *multi = NULL; 106 struct Curl_easy *easy = NULL; 107 struct curl_slist *list = NULL; 108 109/* important: we setup cache outside of the loop 110 and also clean cache after the loop. In contrast,for example, 111 test 1607 sets up and cleans cache on each iteration. */ 112 113 for(i = 0; i < testnum; ++i) { 114 int j; 115 int addressnum = sizeof (tests[i].address) / sizeof (*tests[i].address); 116 struct Curl_addrinfo *addr; 117 struct Curl_dns_entry *dns; 118 void *entry_id; 119 bool problem = false; 120 easy = curl_easy_init(); 121 if(!easy) { 122 curl_global_cleanup(); 123 return CURLE_OUT_OF_MEMORY; 124 } 125 /* create a multi handle and add the easy handle to it so that the 126 hostcache is setup */ 127 multi = curl_multi_init(); 128 if(!multi) 129 goto error; 130 curl_multi_add_handle(multi, easy); 131 132 list = curl_slist_append(NULL, tests[i].optval); 133 if(!list) 134 goto error; 135 136 curl_easy_setopt(easy, CURLOPT_RESOLVE, list); 137 138 if(Curl_loadhostpairs(easy)) 139 goto error; 140 141 entry_id = (void *)aprintf("%s:%d", tests[i].host, tests[i].port); 142 if(!entry_id) 143 goto error; 144 145 dns = Curl_hash_pick(easy->dns.hostcache, entry_id, strlen(entry_id) + 1); 146 free(entry_id); 147 entry_id = NULL; 148 149 addr = dns ? dns->addr : NULL; 150 151 for(j = 0; j < addressnum; ++j) { 152 int port = 0; 153 char ipaddress[MAX_IPADR_LEN] = {0}; 154 155 if(!addr && !tests[i].address[j]) 156 break; 157 158 if(addr && !Curl_addr2string(addr->ai_addr, addr->ai_addrlen, 159 ipaddress, &port)) { 160 fprintf(stderr, "%s:%d tests[%d] failed. Curl_addr2string failed.\n", 161 __FILE__, __LINE__, i); 162 problem = true; 163 break; 164 } 165 166 if(addr && !tests[i].address[j]) { 167 fprintf(stderr, "%s:%d tests[%d] failed. the retrieved addr " 168 "is %s but tests[%d].address[%d] is NULL.\n", 169 __FILE__, __LINE__, i, ipaddress, i, j); 170 problem = true; 171 break; 172 } 173 174 if(!addr && tests[i].address[j]) { 175 fprintf(stderr, "%s:%d tests[%d] failed. the retrieved addr " 176 "is NULL but tests[%d].address[%d] is %s.\n", 177 __FILE__, __LINE__, i, i, j, tests[i].address[j]); 178 problem = true; 179 break; 180 } 181 182 if(!curl_strequal(ipaddress, tests[i].address[j])) { 183 fprintf(stderr, "%s:%d tests[%d] failed. the retrieved addr " 184 "%s is not equal to tests[%d].address[%d] %s.\n", 185 __FILE__, __LINE__, i, ipaddress, i, j, tests[i].address[j]); 186 problem = true; 187 break; 188 } 189 190 if(port != tests[i].port) { 191 fprintf(stderr, "%s:%d tests[%d] failed. the retrieved port " 192 "for tests[%d].address[%d] is %d but tests[%d].port is %d.\n", 193 __FILE__, __LINE__, i, i, j, port, i, tests[i].port); 194 problem = true; 195 break; 196 } 197 198 addr = addr->ai_next; 199 } 200 201 curl_easy_cleanup(easy); 202 easy = NULL; 203 Curl_hash_destroy(&multi->hostcache); 204 curl_multi_cleanup(multi); 205 multi = NULL; 206 curl_slist_free_all(list); 207 list = NULL; 208 209 if(problem) { 210 unitfail++; 211 continue; 212 } 213 } 214 goto unit_test_abort; 215error: 216 curl_easy_cleanup(easy); 217 curl_multi_cleanup(multi); 218 curl_slist_free_all(list); 219} 220UNITTEST_STOP 221