rostests/drivers/tcpip/InterfaceInfo.c

   1 /*
   2  * PROJECT:         ReactOS kernel-mode tests
   3  * LICENSE:         GPLv2+ - See COPYING in the top level directory
   4  * PURPOSE:         User mode part of the TcpIp.sys test suite
   5  * PROGRAMMER:      Jérôme Gardou <jerome.gardou@reactos.org>
   6  */
   7
   8 #include <apitest.h>
   9
  10 #include <ipexport.h>
  11 #include <winioctl.h>
  12 #include <tcpioctl.h>
  13 #include <tcpip_undoc.h>
  14
  15 START_TEST(InterfaceInfo)
  16 {
  17     IP_INTERFACE_INFO* pInterfaceInfo;
  18     IP_INTERFACE_INFO InterfaceInfo;
  19     HANDLE FileHandle;
  20     DWORD BufferSize;
  21     BOOL Result;
  22     DWORD Error;
  23     ULONG i;
  24
  25     /* Open a control channel file for TCP */
  26     FileHandle = CreateFileW(
  27         L"\\\\.\\Tcp",
  28         FILE_READ_DATA | FILE_WRITE_DATA,
  29         FILE_SHARE_READ | FILE_SHARE_WRITE,
  30         NULL,
  31         OPEN_EXISTING,
  32         FILE_FLAG_OVERLAPPED,
  33         NULL);
  34     ok(FileHandle != INVALID_HANDLE_VALUE, "CreateFile failed, GLE %lu\n", GetLastError());
  35
  36     /* Try the IOCTL */
  37     BufferSize = 0;
  38     pInterfaceInfo = &InterfaceInfo;
  39     Result = DeviceIoControl(
  40         FileHandle,
  41         IOCTL_IP_INTERFACE_INFO,
  42         NULL,
  43         0,
  44         pInterfaceInfo,
  45         sizeof(InterfaceInfo),
  46         &BufferSize,
  47         NULL);
  48     Error = GetLastError();
  49     ok(!Result, "DeviceIoControl succeeded.\n");
  50     ok_long(Error, ERROR_INVALID_FUNCTION);
  51     ok_long(BufferSize, 0);
  52
  53     CloseHandle(FileHandle);
  54
  55     /* This IOCTL only works with \Device\Ip */
  56     FileHandle = CreateFileW(
  57         L"\\\\.\\Ip",
  58         FILE_READ_DATA | FILE_WRITE_DATA,
  59         FILE_SHARE_READ | FILE_SHARE_WRITE,
  60         NULL,
  61         OPEN_EXISTING,
  62         FILE_FLAG_OVERLAPPED,
  63         NULL);
  64     ok(FileHandle != INVALID_HANDLE_VALUE, "CreateFile failed, GLE %lu\n", GetLastError());
  65
  66     /* Get the buffer size. */
  67     BufferSize = 0;
  68     pInterfaceInfo = &InterfaceInfo;
  69     Result = DeviceIoControl(
  70         FileHandle,
  71         IOCTL_IP_INTERFACE_INFO,
  72         NULL,
  73         0,
  74         pInterfaceInfo,
  75         sizeof(InterfaceInfo),
  76         &BufferSize,
  77         NULL);
  78     ok(Result, "DeviceIoControl failed, GLE %lu.\n", GetLastError());
  79     ok(BufferSize != 0, "Buffer size is zero.\n");
  80     trace("Buffer size is %lu.\n", BufferSize);
  81
  82     if (BufferSize > sizeof(InterfaceInfo))
  83     {
  84         pInterfaceInfo = HeapAlloc(GetProcessHeap(), 0, BufferSize);
  85         ok(pInterfaceInfo != NULL, "HeapAlloc failed.\n");
  86
  87         /* Send IOCTL to fill the buffer in. */
  88         Result = DeviceIoControl(
  89                 FileHandle,
  90                 IOCTL_IP_INTERFACE_INFO,
  91                 NULL,
  92                 0,
  93                 pInterfaceInfo,
  94                 BufferSize,
  95                 &BufferSize,
  96                 NULL);
  97         ok(Result, "DeviceIoControl failed, GLE %lu.\n", GetLastError());
  98     }
  99
 100     if (Result && BufferSize >= RTL_SIZEOF_THROUGH_FIELD(IP_INTERFACE_INFO, NumAdapters))
 101     {
 102         /* Nothing much to test. Just print out the adapters we got */
 103         trace("We got %ld adapters.\n", pInterfaceInfo->NumAdapters);
 104         for (i = 0; i < pInterfaceInfo->NumAdapters; i++)
 105         {
 106             trace("\tIndex %lu, name %S\n", pInterfaceInfo->Adapter[i].Index, pInterfaceInfo->Adapter[i].Name);
 107         }
 108     }
 109
 110     if (pInterfaceInfo != &InterfaceInfo)
 111         HeapFree(GetProcessHeap(), 0, pInterfaceInfo);
 112     CloseHandle(FileHandle);
 113 }