hostasyn.c 4.5 KB

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