Implementation of guest-network-get-interfaces for windows
Signed-off-by: Kenth Andersson <ke...@eastmark.net> --- configure | 2 +- qga/commands-win32.c | 247 ++++++++++++++++++++++++++++++++++++++++++++++++++- 2 files changed, 244 insertions(+), 5 deletions(-) diff --git a/configure b/configure index 681abfc..7c6c60c 100755 --- a/configure +++ b/configure @@ -717,7 +717,7 @@ EOF sysconfdir="\${prefix}" local_statedir= confsuffix="" - libs_qga="-lws2_32 -lwinmm -lpowrprof $libs_qga" + libs_qga="-lws2_32 -lwinmm -lpowrprof -liphlpapi $libs_qga" fi werror="" diff --git a/qga/commands-win32.c b/qga/commands-win32.c index 3bcbeae..fb6fdba 100644 --- a/qga/commands-win32.c +++ b/qga/commands-win32.c @@ -14,6 +14,9 @@ #include <glib.h> #include <wtypes.h> #include <powrprof.h> +#include <winsock2.h> +#include <iphlpapi.h> +#include <ws2tcpip.h> #include "qga/guest-agent-core.h" #include "qga/vss-win32.h" #include "qga-qmp-commands.h" @@ -29,6 +32,9 @@ (365 * (1970 - 1601) + \ (1970 - 1601) / 4 - 3)) +/* Defines the max length of an IPv6 address in human readable format + pad */ +#define MAX_IPv6_LEN 50 + static void acquire_privilege(const char *name, Error **errp) { HANDLE token = NULL; @@ -359,10 +365,244 @@ void qmp_guest_suspend_hybrid(Error **errp) error_set(errp, QERR_UNSUPPORTED); } +static int get_prefix_length(PIP_ADAPTER_PREFIX prefix, + SOCKET_ADDRESS *sockaddr, int socklen) +{ + PIP_ADAPTER_PREFIX p; + struct sockaddr *addr = sockaddr->lpSockaddr; + + for (p = prefix; p; p = p->Next) { + /* Compare the ip-adderss address family with the prefix + * address family */ + if (addr->sa_family == p->Address.lpSockaddr->sa_family) { + int num_bytes = p->PrefixLength / 8; + unsigned char *src = 0; + unsigned char *dst = 0; + int remaining_bits; + + if (addr->sa_family == AF_INET) { + struct sockaddr_in *sin = (struct sockaddr_in *)addr; + src = (unsigned char *)&(sin->sin_addr.s_addr); + } else if (addr->sa_family == AF_INET6) { + struct sockaddr_in6 *sin = (struct sockaddr_in6 *)addr; + src = (unsigned char *)sin->sin6_addr.s6_addr; + } + + if (p->Address.lpSockaddr->sa_family == AF_INET) { + struct sockaddr_in *sin = + (struct sockaddr_in *)p->Address.lpSockaddr; + dst = (unsigned char *)&(sin->sin_addr.s_addr); + } else if (p->Address.lpSockaddr->sa_family == AF_INET6) { + struct sockaddr_in6 *sin = + (struct sockaddr_in6 *)p->Address.lpSockaddr; + dst = (unsigned char *)sin->sin6_addr.s6_addr; + } + + /* If non of the addresses are in correct format we will continue + * to next one */ + if (src == 0 || dst == 0) { + continue; + } + + /* Check if prefix network is the same network as we are testing + * start with whole bytes */ + + if (memcmp(src, dst, num_bytes) != 0) { + continue; + } + + /* Check the remaning bits */ + remaining_bits = p->PrefixLength % 8; + + if (remaining_bits != 0) { + unsigned char mask = 0xff << (8 - remaining_bits); + int i = num_bytes; + + if ((src[i] & mask) != (dst[i] & mask)) { + continue; + } + } + + return p->PrefixLength; + } + } + return 0; +} + + + GuestNetworkInterfaceList *qmp_guest_network_get_interfaces(Error **errp) { - error_set(errp, QERR_UNSUPPORTED); - return NULL; + GuestNetworkInterfaceList *head = NULL, *curr_item = NULL; + DWORD ret_val; + + PIP_ADAPTER_ADDRESSES adpt_addrs; + PIP_ADAPTER_ADDRESSES curr_adpt_addrs; + PIP_ADAPTER_UNICAST_ADDRESS adpt_uc_addr; + + /* GetAdaptersAddresses requires ULONG */ + ULONG bufLen = sizeof(IP_ADAPTER_ADDRESSES); + + /* Startup WinSock32 */ + WORD ws_version_requested = MAKEWORD(2, 2); + WSADATA ws_data; + WSAStartup(ws_version_requested, &ws_data); + + /* Allocate adapter address list with one entry, used to + * fetch the read length needed by GetAdapterAddresses */ + adpt_addrs = g_malloc(sizeof(IP_ADAPTER_ADDRESSES)); + + /* Get adapter adresses and if it doesn't fit in adpt_addrs + * we will realloc */ + if (GetAdaptersAddresses(AF_UNSPEC, + GAA_FLAG_INCLUDE_PREFIX, + NULL, + adpt_addrs, &bufLen) == + ERROR_BUFFER_OVERFLOW) { + + g_free(adpt_addrs); + adpt_addrs = g_malloc(bufLen); + } + + /* Get adapter address list */ + ret_val = GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_INCLUDE_PREFIX, + NULL, + adpt_addrs, &bufLen); + if (ret_val == NO_ERROR) { + + /* Iterate all adapter addresses */ + curr_adpt_addrs = adpt_addrs; + + while (curr_adpt_addrs) { + /* Check if this adapter is up and running */ + + if (curr_adpt_addrs->OperStatus == IfOperStatusUp && + (curr_adpt_addrs->IfType == IF_TYPE_ETHERNET_CSMACD || + curr_adpt_addrs->IfType == IF_TYPE_IEEE80211 || + curr_adpt_addrs->IfType == IF_TYPE_SOFTWARE_LOOPBACK)) { + + int len = 0; + char *temp_description = 0; + + GuestNetworkInterfaceList *interface_list = + g_malloc0(sizeof(*interface_list)); + interface_list->value = + g_malloc0(sizeof(*interface_list->value)); + + len = wcslen(curr_adpt_addrs->Description) + 1; + temp_description = g_malloc(len); + + wcstombs(temp_description, + curr_adpt_addrs->Description, + len); + + interface_list->value->name = g_strdup(temp_description); + + g_free(temp_description); + + if (curr_item == NULL) { + head = curr_item = interface_list; + } else { + curr_item->next = interface_list; + curr_item = interface_list; + } + + /* Format MAC address */ + interface_list->value->hardware_address = + g_strdup_printf("%02x:%02x:%02x:%02x:%02x:%02x", + (unsigned int) curr_adpt_addrs->PhysicalAddress[0], + (unsigned int) curr_adpt_addrs->PhysicalAddress[1], + (unsigned int) curr_adpt_addrs->PhysicalAddress[2], + (unsigned int) curr_adpt_addrs->PhysicalAddress[3], + (unsigned int) curr_adpt_addrs->PhysicalAddress[4], + (unsigned int) curr_adpt_addrs->PhysicalAddress[5]); + + + interface_list->value->has_hardware_address = true; + + /* Iterate all unicast addresses of this adapter */ + GuestIpAddressList **address_list = + &interface_list->value->ip_addresses; + + GuestIpAddressList *address_item = NULL; + + adpt_uc_addr = curr_adpt_addrs->FirstUnicastAddress; + + while (adpt_uc_addr) { + SOCKET_ADDRESS *saddr = &adpt_uc_addr->Address; + int socklen; + char buffer[MAX_IPv6_LEN]; + DWORD len = MAX_IPv6_LEN; + + /* Allocate an address item */ + address_item = g_malloc0(sizeof(*address_item)); + + address_item->value = + g_malloc0(sizeof(*address_item->value)); + + /* Is this IPv4 or IPv6 */ + if (saddr->lpSockaddr->sa_family == AF_INET) { + address_item->value->ip_address_type = + GUEST_IP_ADDRESS_TYPE_IPV4; + socklen = sizeof(struct sockaddr_in); + } else if (saddr->lpSockaddr->sa_family == AF_INET6) { + address_item->value->ip_address_type = + GUEST_IP_ADDRESS_TYPE_IPV6; + socklen = sizeof(struct sockaddr_in6); + } else { + socklen = 0; + } + + /* Temporary buffer to hold human readable address */ + WSAAddressToString(saddr->lpSockaddr, + socklen, NULL, buffer, &len); + buffer[len] = 0; + + address_item->value->ip_address = g_strdup(buffer); + + /* Get the prefix for this address */ + address_item->value->prefix = + get_prefix_length(curr_adpt_addrs->FirstPrefix, + saddr, socklen); + + /* Add this address to the end of the address list */ + while (*address_list && (*address_list)->next) { + address_list = &(*address_list)->next; + } + + if (!*address_list) { + *address_list = address_item; + } else { + (*address_list)->next = address_item; + } + + /* Jump to next address */ + adpt_uc_addr = adpt_uc_addr->Next; + } + interface_list->value->has_ip_addresses = true; + + } + + /* Jump to next adapter */ + curr_adpt_addrs = curr_adpt_addrs->Next; + } + } + + /* Free the adapter list */ + if (adpt_addrs != NULL) { + g_free(adpt_addrs); + } + + /* Cleanup WinSock32 */ + WSACleanup(); + + if (!head) { + error_set(errp, + ERROR_CLASS_GENERIC_ERROR, + "Could not get network interface information"); + } + + return head; } int64_t qmp_guest_get_time(Error **errp) @@ -452,8 +692,7 @@ GList *ga_command_blacklist_init(GList *blacklist) const char *list_unsupported[] = { "guest-file-open", "guest-file-close", "guest-file-read", "guest-file-write", "guest-file-seek", "guest-file-flush", - "guest-suspend-hybrid", "guest-network-get-interfaces", - "guest-get-vcpus", "guest-set-vcpus", + "guest-suspend-hybrid", "guest-get-vcpus", "guest-set-vcpus", "guest-fsfreeze-freeze-list", "guest-get-fsinfo", "guest-fstrim", NULL}; char **p = (char **)list_unsupported; -- 1.9.3 (Apple Git-50)