tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Better init_host_port.

Browse Source 
GitOrigin-RevId: 04f6f90435cc7d33e18ab594658787044ec6c4f4
This commit is contained in:
levlam 2018-07-01 02:10:17 +03:00
parent 06481d89dd
commit 07f731df95
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tdutils/td/utils/port/IPAddress.cpp
View File

@ -309,8 +309,9 @@ Status IPAddress::init_host_port(CSlice host, CSlice port) {
addrinfo hints;
addrinfo *info = nullptr;
std::memset(&hints, 0, sizeof(hints));
hints.ai_family = AF_INET; // TODO AF_UNSPEC;
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
hints.ai_protocol = IPPROTO_TCP;
LOG(INFO) << "Try to init IP address of " << host << " with port " << port;
auto err = getaddrinfo(host.c_str(), port.c_str(), &hints, &info);
if (err != 0) {
@ -324,16 +325,21 @@ Status IPAddress::init_host_port(CSlice host, CSlice port) {
freeaddrinfo(info);
};
// prefer ipv4
addrinfo *best_info = info;
for (auto *ptr = info->ai_next; ptr != nullptr; ptr = ptr->ai_next) {
if (ptr->ai_socktype == AF_INET) {
addrinfo *best_info = nullptr;
for (auto *ptr = info; ptr != nullptr; ptr = ptr->ai_next) {
if (ptr->ai_family == AF_INET) {
// just use first IPv4 address
best_info = ptr;
break;
}
if (ptr->ai_family == AF_INET6 && best_info == nullptr) {
// or first IPv6 address if there is no IPv4
best_info = ptr;
}
}
if (best_info == nullptr) {
return Status::Error("Failed to find IPv4/IPv6 address");
}
// just use first address
CHECK(best_info != nullptr);
return init_sockaddr(best_info->ai_addr, narrow_cast<socklen_t>(best_info->ai_addrlen));
}