1 /***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 1998 - 2015, 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 http://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 SessionHandle *data;
43 static struct curl_hash hp;
44 static char *data_key;
45 static struct Curl_dns_entry *data_node;
46
unit_setup(void)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
unit_stop(void)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
fake_ai(void)76 static Curl_addrinfo *fake_ai(void)
77 {
78 static Curl_addrinfo *ai;
79 int ss_size;
80
81 ss_size = sizeof (struct sockaddr_in);
82
83 if((ai = calloc(1, sizeof(Curl_addrinfo))) == NULL)
84 return NULL;
85
86 if((ai->ai_canonname = strdup("dummy")) == NULL) {
87 free(ai);
88 return NULL;
89 }
90
91 if((ai->ai_addr = calloc(1, ss_size)) == NULL) {
92 free(ai->ai_canonname);
93 free(ai);
94 return NULL;
95 }
96
97 ai->ai_family = AF_INET;
98 ai->ai_addrlen = ss_size;
99
100 return ai;
101 }
102
create_node(void)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 /* To do: test retrieval, deletion, edge conditions */
139 }
140
141 UNITTEST_STOP
142