diff options
Diffstat (limited to 'drivers/windows/ip_windows.cpp')
| -rw-r--r-- | drivers/windows/ip_windows.cpp | 166 |
1 files changed, 166 insertions, 0 deletions
diff --git a/drivers/windows/ip_windows.cpp b/drivers/windows/ip_windows.cpp new file mode 100644 index 0000000000..c8d94bfb2a --- /dev/null +++ b/drivers/windows/ip_windows.cpp @@ -0,0 +1,166 @@ +/**************************************************************************/ +/* ip_windows.cpp */ +/**************************************************************************/ +/* This file is part of: */ +/* REDOT ENGINE */ +/* https://redotengine.org */ +/**************************************************************************/ +/* Copyright (c) 2024-present Redot Engine contributors */ +/* (see REDOT_AUTHORS.md) */ +/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */ +/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */ +/* */ +/* Permission is hereby granted, free of charge, to any person obtaining */ +/* a copy of this software and associated documentation files (the */ +/* "Software"), to deal in the Software without restriction, including */ +/* without limitation the rights to use, copy, modify, merge, publish, */ +/* distribute, sublicense, and/or sell copies of the Software, and to */ +/* permit persons to whom the Software is furnished to do so, subject to */ +/* the following conditions: */ +/* */ +/* The above copyright notice and this permission notice shall be */ +/* included in all copies or substantial portions of the Software. */ +/* */ +/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ +/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ +/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */ +/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ +/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ +/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ +/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ +/**************************************************************************/ + +#if defined(WINDOWS_ENABLED) + +#include "ip_windows.h" + +#define WIN32_LEAN_AND_MEAN +#include <windows.h> +#include <winsock2.h> +#include <ws2tcpip.h> + +#include <iphlpapi.h> + +#include <stdio.h> + +#include <string.h> + +static IPAddress _sockaddr2ip(struct sockaddr *p_addr) { + IPAddress ip; + + if (p_addr->sa_family == AF_INET) { + struct sockaddr_in *addr = (struct sockaddr_in *)p_addr; + ip.set_ipv4((uint8_t *)&(addr->sin_addr)); + } else if (p_addr->sa_family == AF_INET6) { + struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *)p_addr; + ip.set_ipv6(addr6->sin6_addr.s6_addr); + } + + return ip; +} + +void IPWindows::_resolve_hostname(List<IPAddress> &r_addresses, const String &p_hostname, Type p_type) const { + struct addrinfo hints; + struct addrinfo *result = nullptr; + + memset(&hints, 0, sizeof(struct addrinfo)); + if (p_type == TYPE_IPV4) { + hints.ai_family = AF_INET; + } else if (p_type == TYPE_IPV6) { + hints.ai_family = AF_INET6; + hints.ai_flags = 0; + } else { + hints.ai_family = AF_UNSPEC; + hints.ai_flags = AI_ADDRCONFIG; + } + hints.ai_flags &= ~AI_NUMERICHOST; + + int s = getaddrinfo(p_hostname.utf8().get_data(), nullptr, &hints, &result); + if (s != 0) { + print_verbose("getaddrinfo failed! Cannot resolve hostname."); + return; + } + + if (result == nullptr || result->ai_addr == nullptr) { + print_verbose("Invalid response from getaddrinfo."); + if (result) { + freeaddrinfo(result); + } + return; + } + + struct addrinfo *next = result; + + do { + if (next->ai_addr == nullptr) { + next = next->ai_next; + continue; + } + IPAddress ip = _sockaddr2ip(next->ai_addr); + if (ip.is_valid() && !r_addresses.find(ip)) { + r_addresses.push_back(ip); + } + next = next->ai_next; + } while (next); + + freeaddrinfo(result); +} + +void IPWindows::get_local_interfaces(HashMap<String, Interface_Info> *r_interfaces) const { + ULONG buf_size = 1024; + IP_ADAPTER_ADDRESSES *addrs; + + while (true) { + addrs = (IP_ADAPTER_ADDRESSES *)memalloc(buf_size); + int err = GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_SKIP_ANYCAST | GAA_FLAG_SKIP_MULTICAST | GAA_FLAG_SKIP_DNS_SERVER | GAA_FLAG_SKIP_FRIENDLY_NAME, + nullptr, addrs, &buf_size); + if (err == NO_ERROR) { + break; + } + memfree(addrs); + if (err == ERROR_BUFFER_OVERFLOW) { + continue; // Will go back and alloc the right size. + } + + ERR_FAIL_MSG("Call to GetAdaptersAddresses failed with error " + itos(err) + "."); + } + + IP_ADAPTER_ADDRESSES *adapter = addrs; + + while (adapter != nullptr) { + Interface_Info info; + info.name = adapter->AdapterName; + info.name_friendly = adapter->FriendlyName; + info.index = String::num_uint64(adapter->IfIndex); + + IP_ADAPTER_UNICAST_ADDRESS *address = adapter->FirstUnicastAddress; + while (address != nullptr) { + int family = address->Address.lpSockaddr->sa_family; + if (family != AF_INET && family != AF_INET6) { + continue; + } + info.ip_addresses.push_front(_sockaddr2ip(address->Address.lpSockaddr)); + address = address->Next; + } + adapter = adapter->Next; + // Only add interface if it has at least one IP. + if (info.ip_addresses.size() > 0) { + r_interfaces->insert(info.name, info); + } + } + + memfree(addrs); +} + +void IPWindows::make_default() { + _create = _create_unix; +} + +IP *IPWindows::_create_unix() { + return memnew(IPWindows); +} + +IPWindows::IPWindows() { +} + +#endif // WINDOWS_ENABLED |