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 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 #include "curl_setup.h"
24
25 /***********************************************************************
26 * Only for builds using asynchronous name resolves
27 **********************************************************************/
28 #ifdef CURLRES_ASYNCH
29
30 #ifdef HAVE_NETINET_IN_H
31 #include <netinet/in.h>
32 #endif
33 #ifdef HAVE_NETDB_H
34 #include <netdb.h>
35 #endif
36 #ifdef HAVE_ARPA_INET_H
37 #include <arpa/inet.h>
38 #endif
39 #ifdef __VMS
40 #include <in.h>
41 #include <inet.h>
42 #endif
43
44 #ifdef HAVE_PROCESS_H
45 #include <process.h>
46 #endif
47
48 #include "urldata.h"
49 #include "sendf.h"
50 #include "hostip.h"
51 #include "hash.h"
52 #include "share.h"
53 #include "strerror.h"
54 #include "url.h"
55 #include "curl_memory.h"
56 /* The last #include file should be: */
57 #include "memdebug.h"
58
59 /*
60 * Curl_addrinfo_callback() gets called by ares, gethostbyname_thread()
61 * or getaddrinfo_thread() when we got the name resolved (or not!).
62 *
63 * If the status argument is CURL_ASYNC_SUCCESS, this function takes
64 * ownership of the Curl_addrinfo passed, storing the resolved data
65 * in the DNS cache.
66 *
67 * The storage operation locks and unlocks the DNS cache.
68 */
Curl_addrinfo_callback(struct connectdata * conn,int status,struct Curl_addrinfo * ai)69 CURLcode Curl_addrinfo_callback(struct connectdata *conn,
70 int status,
71 struct Curl_addrinfo *ai)
72 {
73 struct Curl_dns_entry *dns = NULL;
74 CURLcode result = CURLE_OK;
75
76 conn->async.status = status;
77
78 if(CURL_ASYNC_SUCCESS == status) {
79 if(ai) {
80 struct Curl_easy *data = conn->data;
81
82 if(data->share)
83 Curl_share_lock(data, CURL_LOCK_DATA_DNS, CURL_LOCK_ACCESS_SINGLE);
84
85 dns = Curl_cache_addr(data, ai,
86 conn->async.hostname,
87 conn->async.port);
88 if(!dns) {
89 /* failed to store, cleanup and return error */
90 Curl_freeaddrinfo(ai);
91 result = CURLE_OUT_OF_MEMORY;
92 }
93
94 if(data->share)
95 Curl_share_unlock(data, CURL_LOCK_DATA_DNS);
96 }
97 else {
98 result = CURLE_OUT_OF_MEMORY;
99 }
100 }
101
102 conn->async.dns = dns;
103
104 /* Set async.done TRUE last in this function since it may be used multi-
105 threaded and once this is TRUE the other thread may read fields from the
106 async struct */
107 conn->async.done = TRUE;
108
109 /* IPv4: The input hostent struct will be freed by ares when we return from
110 this function */
111 return result;
112 }
113
114 /* Call this function after Curl_connect() has returned async=TRUE and
115 then a successful name resolve has been received.
116
117 Note: this function disconnects and frees the conn data in case of
118 resolve failure */
Curl_async_resolved(struct connectdata * conn,bool * protocol_done)119 CURLcode Curl_async_resolved(struct connectdata *conn,
120 bool *protocol_done)
121 {
122 CURLcode result;
123
124 if(conn->async.dns) {
125 conn->dns_entry = conn->async.dns;
126 conn->async.dns = NULL;
127 }
128
129 result = Curl_setup_conn(conn, protocol_done);
130
131 if(result)
132 /* We're not allowed to return failure with memory left allocated
133 in the connectdata struct, free those here */
134 Curl_disconnect(conn, FALSE); /* close the connection */
135
136 return result;
137 }
138
139 /*
140 * Curl_getaddrinfo() is the generic low-level name resolve API within this
141 * source file. There are several versions of this function - for different
142 * name resolve layers (selected at build-time). They all take this same set
143 * of arguments
144 */
Curl_getaddrinfo(struct connectdata * conn,const char * hostname,int port,int * waitp)145 Curl_addrinfo *Curl_getaddrinfo(struct connectdata *conn,
146 const char *hostname,
147 int port,
148 int *waitp)
149 {
150 return Curl_resolver_getaddrinfo(conn, hostname, port, waitp);
151 }
152
153 #endif /* CURLRES_ASYNCH */
154