1f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch/* 2f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * libjingle 3f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * Copyright 2008, Google Inc. 4f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 5f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * Redistribution and use in source and binary forms, with or without 6f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * modification, are permitted provided that the following conditions are met: 7f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 8f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 1. Redistributions of source code must retain the above copyright notice, 9f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * this list of conditions and the following disclaimer. 10f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 2. Redistributions in binary form must reproduce the above copyright notice, 11f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * this list of conditions and the following disclaimer in the documentation 12f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * and/or other materials provided with the distribution. 13f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 3. The name of the author may not be used to endorse or promote products 14f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * derived from this software without specific prior written permission. 15f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * 16f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED 17f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO 19f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 21f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; 22f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 23f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 24f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 25f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch */ 27f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 28f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#include "talk/base/nethelpers.h" 29f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 30f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#include "talk/base/byteorder.h" 31f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#include "talk/base/signalthread.h" 32f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 33f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochnamespace talk_base { 34f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 35dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#if defined(LINUX) || defined(ANDROID) 36dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenstatic const size_t kInitHostentLen = 1024; 37dc0f95d653279beabeb9817299e2902918ba123eKristian Monsenstatic const size_t kMaxHostentLen = kInitHostentLen * 8; 38dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#endif 39dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 40f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch// AsyncResolver 41f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 42f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen MurdochAsyncResolver::AsyncResolver() : result_(NULL), error_(0) { 43f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 44f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 45f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen MurdochAsyncResolver::~AsyncResolver() { 46f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch FreeHostEnt(result_); 47f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 48f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 49f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochvoid AsyncResolver::DoWork() { 50f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result_ = SafeGetHostByName(addr_.hostname().c_str(), &error_); 51f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 52f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 53f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochvoid AsyncResolver::OnWorkDone() { 54f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (result_) { 55f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch addr_.SetIP(NetworkToHost32( 56f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch *reinterpret_cast<uint32*>(result_->h_addr_list[0]))); 57f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 58f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 59f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 60f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch// The functions below are used to do gethostbyname, but with an allocated 61f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch// instead of a static buffer. 62f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochhostent* SafeGetHostByName(const char* hostname, int* herrno) { 63dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (NULL == hostname || NULL == herrno) { 64dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return NULL; 65dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen } 66f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch hostent* result = NULL; 67dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#if defined(WIN32) 68f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // On Windows we have to allocate a buffer, and manually copy the hostent, 69f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // along with its embedded pointers. 70f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch hostent* ent = gethostbyname(hostname); 71f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch if (!ent) { 72f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch *herrno = WSAGetLastError(); 73f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return NULL; 74f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 75f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 76f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // Get the total number of bytes we need to copy, and allocate our buffer. 77f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch int num_aliases = 0, num_addrs = 0; 78f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch int total_len = sizeof(hostent); 79f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch total_len += strlen(ent->h_name) + 1; 80f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch while (ent->h_aliases[num_aliases]) { 81f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch total_len += sizeof(char*) + strlen(ent->h_aliases[num_aliases]) + 1; 82f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ++num_aliases; 83f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 84f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch total_len += sizeof(char*); 85f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch while (ent->h_addr_list[num_addrs]) { 86f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch total_len += sizeof(char*) + ent->h_length; 87f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch ++num_addrs; 88f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 89f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch total_len += sizeof(char*); 90f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 91f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result = static_cast<hostent*>(malloc(total_len)); 92dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (NULL == result) { 93dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return NULL; 94dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen } 95f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch char* p = reinterpret_cast<char*>(result) + sizeof(hostent); 96f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 97f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // Copy the hostent into it, along with its embedded pointers. 98f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_name = p; 99f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch memcpy(p, ent->h_name, strlen(ent->h_name) + 1); 100f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch p += strlen(ent->h_name) + 1; 101f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 102f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_aliases = reinterpret_cast<char**>(p); 103f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch p += (num_aliases + 1) * sizeof(char*); 104f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch for (int i = 0; i < num_aliases; ++i) { 105f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_aliases[i] = p; 106731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick memcpy(p, ent->h_aliases[i], strlen(ent->h_aliases[i]) + 1); 107f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch p += strlen(ent->h_aliases[i]) + 1; 108f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 109f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_aliases[num_aliases] = NULL; 110f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 111f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_addrtype = ent->h_addrtype; 112f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_length = ent->h_length; 113f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 114f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_addr_list = reinterpret_cast<char**>(p); 115f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch p += (num_addrs + 1) * sizeof(char*); 116f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch for (int i = 0; i < num_addrs; ++i) { 117f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_addr_list[i] = p; 118f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch memcpy(p, ent->h_addr_list[i], ent->h_length); 119f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch p += ent->h_length; 120f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch } 121f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result->h_addr_list[num_addrs] = NULL; 122f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 123f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch *herrno = 0; 124dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#elif defined(LINUX) || defined(ANDROID) 125dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen // gethostbyname() is not thread safe, so we need to call gethostbyname_r() 126dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen // which is a reentrant version of gethostbyname(). 127dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen ASSERT(kInitHostentLen > sizeof(hostent)); 128dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen size_t size = kInitHostentLen; 129dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen int ret; 130dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen void* buf = malloc(size); 131dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (NULL == buf) { 132dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return NULL; 133dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen } 134dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen char* aux = static_cast<char*>(buf) + sizeof(hostent); 135dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen size_t aux_len = size - sizeof(hostent); 136dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen while ((ret = gethostbyname_r(hostname, reinterpret_cast<hostent*>(buf), aux, 137dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen aux_len, &result, herrno)) == ERANGE) { 138dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen size *= 2; 139dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (size > kMaxHostentLen) { 140dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen break; // Just to be safe. 141dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen } 142dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen buf = realloc(buf, size); 143dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (NULL == buf) { 144dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return NULL; 145dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen } 146dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen aux = static_cast<char*>(buf) + sizeof(hostent); 147dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen aux_len = size - sizeof(hostent); 148dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen } 149dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen if (ret != 0 || buf != result) { 150dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen free(buf); 151dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen return NULL; 152dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen } 153dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen *herrno = 0; 154dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#elif defined(OSX) || defined(IOS) 155f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch // Mac OS returns an object with everything allocated. 156f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch result = getipnodebyname(hostname, AF_INET, AI_DEFAULT, herrno); 157f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#else 158f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#error "I don't know how to do gethostbyname safely on your system." 159f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#endif 160f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch return result; 161f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 162f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 163f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch// This function should mirror the above function, and free any resources 164f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch// allocated by the above. 165f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdochvoid FreeHostEnt(hostent* host) { 166dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#if defined(OSX) || defined(IOS) 167f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch freehostent(host); 168dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#elif defined(WIN32) || defined(POSIX) 169dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen free(host); 170f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#else 171f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#error "I don't know how to free a hostent on your system." 172f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch#endif 173f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} 174f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch 175f74420b3285b9fe04a7e00aa3b8c0ab07ea344bcBen Murdoch} // namespace talk_base 176