| 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 "curlcheck.h" | 
|---|
| 23 |  | 
|---|
| 24 | #ifdef HAVE_NETINET_IN_H | 
|---|
| 25 | #  include <netinet/in.h> | 
|---|
| 26 | #endif | 
|---|
| 27 | #ifdef HAVE_NETDB_H | 
|---|
| 28 | #  include <netdb.h> | 
|---|
| 29 | #endif | 
|---|
| 30 | #ifdef HAVE_ARPA_INET_H | 
|---|
| 31 | #  include <arpa/inet.h> | 
|---|
| 32 | #endif | 
|---|
| 33 |  | 
|---|
| 34 | #define ENABLE_CURLX_PRINTF | 
|---|
| 35 | #include "curlx.h" | 
|---|
| 36 |  | 
|---|
| 37 | #include "hash.h" | 
|---|
| 38 | #include "hostip.h" | 
|---|
| 39 |  | 
|---|
| 40 | #include "memdebug.h" /* LAST include file */ | 
|---|
| 41 |  | 
|---|
| 42 | static struct Curl_easy *data; | 
|---|
| 43 | static struct curl_hash hp; | 
|---|
| 44 | static char *data_key; | 
|---|
| 45 | static struct Curl_dns_entry *data_node; | 
|---|
| 46 |  | 
|---|
| 47 | static CURLcode unit_setup(void) | 
|---|
| 48 | { | 
|---|
| 49 | int rc; | 
|---|
| 50 | data = curl_easy_init(); | 
|---|
| 51 | if(!data) | 
|---|
| 52 | return CURLE_OUT_OF_MEMORY; | 
|---|
| 53 |  | 
|---|
| 54 | rc = Curl_mk_dnscache(&hp); | 
|---|
| 55 | if(rc) { | 
|---|
| 56 | curl_easy_cleanup(data); | 
|---|
| 57 | curl_global_cleanup(); | 
|---|
| 58 | return CURLE_OUT_OF_MEMORY; | 
|---|
| 59 | } | 
|---|
| 60 | return CURLE_OK; | 
|---|
| 61 | } | 
|---|
| 62 |  | 
|---|
| 63 | static void unit_stop(void) | 
|---|
| 64 | { | 
|---|
| 65 | if(data_node) { | 
|---|
| 66 | Curl_freeaddrinfo(data_node->addr); | 
|---|
| 67 | free(data_node); | 
|---|
| 68 | } | 
|---|
| 69 | free(data_key); | 
|---|
| 70 | Curl_hash_destroy(&hp); | 
|---|
| 71 |  | 
|---|
| 72 | curl_easy_cleanup(data); | 
|---|
| 73 | curl_global_cleanup(); | 
|---|
| 74 | } | 
|---|
| 75 |  | 
|---|
| 76 | static Curl_addrinfo *fake_ai(void) | 
|---|
| 77 | { | 
|---|
| 78 | static Curl_addrinfo *ai; | 
|---|
| 79 |  | 
|---|
| 80 | ai = calloc(1, sizeof(Curl_addrinfo)); | 
|---|
| 81 | if(!ai) | 
|---|
| 82 | return NULL; | 
|---|
| 83 |  | 
|---|
| 84 | ai->ai_canonname = strdup( "dummy"); | 
|---|
| 85 | if(!ai->ai_canonname) { | 
|---|
| 86 | free(ai); | 
|---|
| 87 | return NULL; | 
|---|
| 88 | } | 
|---|
| 89 |  | 
|---|
| 90 | ai->ai_addr = calloc(1, sizeof(struct sockaddr_in)); | 
|---|
| 91 | if(!ai->ai_addr) { | 
|---|
| 92 | free(ai->ai_canonname); | 
|---|
| 93 | free(ai); | 
|---|
| 94 | return NULL; | 
|---|
| 95 | } | 
|---|
| 96 |  | 
|---|
| 97 | ai->ai_family = AF_INET; | 
|---|
| 98 | ai->ai_addrlen = sizeof(struct sockaddr_in); | 
|---|
| 99 |  | 
|---|
| 100 | return ai; | 
|---|
| 101 | } | 
|---|
| 102 |  | 
|---|
| 103 | static CURLcode create_node(void) | 
|---|
| 104 | { | 
|---|
| 105 | data_key = aprintf( "%s:%d", "dummy", 0); | 
|---|
| 106 | if(!data_key) | 
|---|
| 107 | return CURLE_OUT_OF_MEMORY; | 
|---|
| 108 |  | 
|---|
| 109 | data_node = calloc(1, sizeof(struct Curl_dns_entry)); | 
|---|
| 110 | if(!data_node) | 
|---|
| 111 | return CURLE_OUT_OF_MEMORY; | 
|---|
| 112 |  | 
|---|
| 113 | data_node->addr = fake_ai(); | 
|---|
| 114 | if(!data_node->addr) | 
|---|
| 115 | return CURLE_OUT_OF_MEMORY; | 
|---|
| 116 |  | 
|---|
| 117 | return CURLE_OK; | 
|---|
| 118 | } | 
|---|
| 119 |  | 
|---|
| 120 |  | 
|---|
| 121 | UNITTEST_START | 
|---|
| 122 |  | 
|---|
| 123 | struct Curl_dns_entry *nodep; | 
|---|
| 124 | size_t key_len; | 
|---|
| 125 |  | 
|---|
| 126 | /* Test 1305 exits without adding anything to the hash */ | 
|---|
| 127 | if(strcmp(arg, "1305") != 0) { | 
|---|
| 128 | CURLcode rc = create_node(); | 
|---|
| 129 | abort_unless(rc == CURLE_OK, "data node creation failed"); | 
|---|
| 130 | key_len = strlen(data_key); | 
|---|
| 131 |  | 
|---|
| 132 | data_node->inuse = 1; /* hash will hold the reference */ | 
|---|
| 133 | nodep = Curl_hash_add(&hp, data_key, key_len + 1, data_node); | 
|---|
| 134 | abort_unless(nodep, "insertion into hash failed"); | 
|---|
| 135 | /* Freeing will now be done by Curl_hash_destroy */ | 
|---|
| 136 | data_node = NULL; | 
|---|
| 137 | } | 
|---|
| 138 |  | 
|---|
| 139 | UNITTEST_STOP | 
|---|
| 140 |  | 
|---|