Zebediah Figura : ntdll: Introduce IOCTL_AFD_WINE_GET_INTERFACE_LIST.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 67959e966328fc7391b9329e69da9e63a05ed39b
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=67959e966328fc7391b9329e69da9e63a05ed39b

Author: Zebediah Figura <z.figura12 at gmail.com>
Date:   Mon Jun  7 19:53:09 2021 -0500

ntdll: Introduce IOCTL_AFD_WINE_GET_INTERFACE_LIST.

Signed-off-by: Zebediah Figura <z.figura12 at gmail.com>
Signed-off-by: Alexandre Julliard <julliard at winehq.org>

---

 dlls/ntdll/unix/socket.c | 93 ++++++++++++++++++++++++++++++++++++++++++++++++
 include/wine/afd.h       |  1 +
 2 files changed, 94 insertions(+)

diff --git a/dlls/ntdll/unix/socket.c b/dlls/ntdll/unix/socket.c
index e5f4862bf3c..bc77514669d 100644
--- a/dlls/ntdll/unix/socket.c
+++ b/dlls/ntdll/unix/socket.c
@@ -24,7 +24,14 @@
 
 #include "config.h"
 #include <errno.h>
+#include <sys/types.h>
 #include <unistd.h>
+#ifdef HAVE_IFADDRS_H
+# include <ifaddrs.h>
+#endif
+#ifdef HAVE_NET_IF_H
+# include <net/if.h>
+#endif
 #ifdef HAVE_SYS_IOCTL_H
 # include <sys/ioctl.h>
 #endif
@@ -1355,6 +1362,92 @@ NTSTATUS sock_ioctl( HANDLE handle, HANDLE event, PIO_APC_ROUTINE apc, void *apc
             break;
         }
 
+        case IOCTL_AFD_WINE_GET_INTERFACE_LIST:
+        {
+#ifdef HAVE_GETIFADDRS
+            INTERFACE_INFO *info = out_buffer;
+            struct ifaddrs *ifaddrs, *ifaddr;
+            unsigned int count = 0;
+            ULONG ret_size;
+
+            if ((status = server_get_unix_fd( handle, 0, &fd, &needs_close, NULL, NULL )))
+                return status;
+
+            if (getifaddrs( &ifaddrs ) < 0)
+            {
+                status = sock_errno_to_status( errno );
+                break;
+            }
+
+            for (ifaddr = ifaddrs; ifaddr != NULL; ifaddr = ifaddr->ifa_next)
+            {
+                if (ifaddr->ifa_addr && ifaddr->ifa_addr->sa_family == AF_INET) ++count;
+            }
+
+            ret_size = count * sizeof(*info);
+            if (out_size < ret_size)
+            {
+                status = STATUS_PENDING;
+                complete_async( handle, event, apc, apc_user, io, STATUS_BUFFER_TOO_SMALL, 0 );
+                freeifaddrs( ifaddrs );
+                break;
+            }
+
+            memset( out_buffer, 0, ret_size );
+
+            count = 0;
+            for (ifaddr = ifaddrs; ifaddr != NULL; ifaddr = ifaddr->ifa_next)
+            {
+                in_addr_t addr, mask;
+
+                if (!ifaddr->ifa_addr || ifaddr->ifa_addr->sa_family != AF_INET)
+                    continue;
+
+                addr = ((const struct sockaddr_in *)ifaddr->ifa_addr)->sin_addr.s_addr;
+                mask = ((const struct sockaddr_in *)ifaddr->ifa_netmask)->sin_addr.s_addr;
+
+                info[count].iiFlags = 0;
+                if (ifaddr->ifa_flags & IFF_BROADCAST)
+                    info[count].iiFlags |= WS_IFF_BROADCAST;
+                if (ifaddr->ifa_flags & IFF_LOOPBACK)
+                    info[count].iiFlags |= WS_IFF_LOOPBACK;
+                if (ifaddr->ifa_flags & IFF_MULTICAST)
+                    info[count].iiFlags |= WS_IFF_MULTICAST;
+#ifdef IFF_POINTTOPOINT
+                if (ifaddr->ifa_flags & IFF_POINTTOPOINT)
+                    info[count].iiFlags |= WS_IFF_POINTTOPOINT;
+#endif
+                if (ifaddr->ifa_flags & IFF_UP)
+                    info[count].iiFlags |= WS_IFF_UP;
+
+                info[count].iiAddress.AddressIn.sin_family = WS_AF_INET;
+                info[count].iiAddress.AddressIn.sin_port = 0;
+                info[count].iiAddress.AddressIn.sin_addr.WS_s_addr = addr;
+
+                info[count].iiNetmask.AddressIn.sin_family = WS_AF_INET;
+                info[count].iiNetmask.AddressIn.sin_port = 0;
+                info[count].iiNetmask.AddressIn.sin_addr.WS_s_addr = mask;
+
+                if (ifaddr->ifa_flags & IFF_BROADCAST)
+                {
+                    info[count].iiBroadcastAddress.AddressIn.sin_family = WS_AF_INET;
+                    info[count].iiBroadcastAddress.AddressIn.sin_port = 0;
+                    info[count].iiBroadcastAddress.AddressIn.sin_addr.WS_s_addr = addr | ~mask;
+                }
+
+                ++count;
+            }
+
+            freeifaddrs( ifaddrs );
+            status = STATUS_PENDING;
+            complete_async( handle, event, apc, apc_user, io, STATUS_SUCCESS, ret_size );
+#else
+            FIXME( "Interface list queries are currently not supported on this platform.\n" );
+            status = STATUS_NOT_SUPPORTED;
+#endif
+            break;
+        }
+
         default:
         {
             if ((code >> 16) == FILE_DEVICE_NETWORK)
diff --git a/include/wine/afd.h b/include/wine/afd.h
index 2b4c39e7a86..cc328f9bcae 100644
--- a/include/wine/afd.h
+++ b/include/wine/afd.h
@@ -100,6 +100,7 @@ struct afd_poll_params
 #define IOCTL_AFD_WINE_COMPLETE_ASYNC       CTL_CODE(FILE_DEVICE_NETWORK, 210, METHOD_BUFFERED, FILE_ANY_ACCESS)
 #define IOCTL_AFD_WINE_FIONREAD             CTL_CODE(FILE_DEVICE_NETWORK, 211, METHOD_BUFFERED, FILE_ANY_ACCESS)
 #define IOCTL_AFD_WINE_SIOCATMARK           CTL_CODE(FILE_DEVICE_NETWORK, 212, METHOD_BUFFERED, FILE_ANY_ACCESS)
+#define IOCTL_AFD_WINE_GET_INTERFACE_LIST   CTL_CODE(FILE_DEVICE_NETWORK, 213, METHOD_BUFFERED, FILE_ANY_ACCESS)
 
 struct afd_create_params
 {