Skip to content

Commit

Permalink
Merge pull request godotengine#57379 from Faless/net/4.x_ip_cache_fixes
Browse files Browse the repository at this point in the history
[Net] Simplify IP resolution code, fix caching.
  • Loading branch information
akien-mga authored Jan 29, 2022
2 parents cb3d308 + 49297d9 commit 01f5d7c
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 31 deletions.
42 changes: 12 additions & 30 deletions core/io/ip.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -98,6 +98,11 @@ struct _IP_ResolverPrivate {
if (queue[i].status.get() != IP::RESOLVER_STATUS_WAITING) {
continue;
}
// We might be overriding another result, but we don't care as long as the result is valid.
if (response.size()) {
String key = get_cache_key(hostname, type);
cache[key] = response;
}
queue[i].response = response;
queue[i].status.set(response.is_empty() ? IP::RESOLVER_STATUS_ERROR : IP::RESOLVER_STATUS_DONE);
}
Expand All @@ -120,30 +125,8 @@ struct _IP_ResolverPrivate {
};

IPAddress IP::resolve_hostname(const String &p_hostname, IP::Type p_type) {
List<IPAddress> res;
String key = _IP_ResolverPrivate::get_cache_key(p_hostname, p_type);

resolver->mutex.lock();
if (resolver->cache.has(key)) {
res = resolver->cache[key];
} else {
// This should be run unlocked so the resolver thread can keep
// resolving other requests.
resolver->mutex.unlock();
_resolve_hostname(res, p_hostname, p_type);
resolver->mutex.lock();
// We might be overriding another result, but we don't care (they are the
// same hostname).
resolver->cache[key] = res;
}
resolver->mutex.unlock();

for (int i = 0; i < res.size(); ++i) {
if (res[i].is_valid()) {
return res[i];
}
}
return IPAddress();
const Array addresses = resolve_hostname_addresses(p_hostname, p_type);
return addresses.size() ? addresses[0].operator IPAddress() : IPAddress();
}

Array IP::resolve_hostname_addresses(const String &p_hostname, Type p_type) {
Expand All @@ -159,17 +142,16 @@ Array IP::resolve_hostname_addresses(const String &p_hostname, Type p_type) {
resolver->mutex.unlock();
_resolve_hostname(res, p_hostname, p_type);
resolver->mutex.lock();
// We might be overriding another result, but we don't care (they are the
// same hostname).
resolver->cache[key] = res;
// We might be overriding another result, but we don't care as long as the result is valid.
if (res.size()) {
resolver->cache[key] = res;
}
}
resolver->mutex.unlock();

Array result;
for (int i = 0; i < res.size(); ++i) {
if (res[i].is_valid()) {
result.push_back(String(res[i]));
}
result.push_back(String(res[i]));
}
return result;
}
Expand Down
2 changes: 1 addition & 1 deletion drivers/unix/ip_unix.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ void IPUnix::_resolve_hostname(List<IPAddress> &r_addresses, const String &p_hos
continue;
}
IPAddress ip = _sockaddr2ip(next->ai_addr);
if (!r_addresses.find(ip)) {
if (ip.is_valid() && !r_addresses.find(ip)) {
r_addresses.push_back(ip);
}
next = next->ai_next;
Expand Down

0 comments on commit 01f5d7c

Please sign in to comment.