From d2dbd9bdac15cf270bffdca9d54e12cf39a563fe Mon Sep 17 00:00:00 2001
From: Hans Leidekker <hans@codeweavers.com>
Date: Mon, 16 Apr 2012 12:35:44 +0200
Subject: [PATCH] iphlpapi: Add tests for GetExtendedTcpTable.

---
 dlls/iphlpapi/tests/iphlpapi.c | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/dlls/iphlpapi/tests/iphlpapi.c b/dlls/iphlpapi/tests/iphlpapi.c
index 0cbfb31e0e5..ec388a36937 100644
--- a/dlls/iphlpapi/tests/iphlpapi.c
+++ b/dlls/iphlpapi/tests/iphlpapi.c
@@ -66,6 +66,7 @@ typedef DWORD (WINAPI *GetPerAdapterInfoFunc)(ULONG,PIP_PER_ADAPTER_INFO,PULONG)
 typedef DWORD (WINAPI *GetAdaptersAddressesFunc)(ULONG,ULONG,PVOID,PIP_ADAPTER_ADDRESSES,PULONG);
 typedef DWORD (WINAPI *NotifyAddrChangeFunc)(PHANDLE,LPOVERLAPPED);
 typedef BOOL  (WINAPI *CancelIPChangeNotifyFunc)(LPOVERLAPPED);
+typedef DWORD (WINAPI *GetExtendedTcpTableFunc)(PVOID,PDWORD,BOOL,ULONG,TCP_TABLE_CLASS,ULONG);
 
 static GetNumberOfInterfacesFunc gGetNumberOfInterfaces = NULL;
 static GetIpAddrTableFunc gGetIpAddrTable = NULL;
@@ -87,6 +88,7 @@ static GetPerAdapterInfoFunc gGetPerAdapterInfo = NULL;
 static GetAdaptersAddressesFunc gGetAdaptersAddresses = NULL;
 static NotifyAddrChangeFunc gNotifyAddrChange = NULL;
 static CancelIPChangeNotifyFunc gCancelIPChangeNotify = NULL;
+static GetExtendedTcpTableFunc gGetExtendedTcpTable = NULL;
 
 static void loadIPHlpApi(void)
 {
@@ -130,6 +132,8 @@ static void loadIPHlpApi(void)
      hLibrary, "NotifyAddrChange");
     gCancelIPChangeNotify = (CancelIPChangeNotifyFunc)GetProcAddress(
      hLibrary, "CancelIPChangeNotify");
+    gGetExtendedTcpTable = (GetExtendedTcpTableFunc)GetProcAddress(
+     hLibrary, "GetExtendedTcpTable");
   }
 }
 
@@ -154,6 +158,7 @@ static void freeIPHlpApi(void)
     gGetUdpTable = NULL;
     gNotifyAddrChange = NULL;
     gCancelIPChangeNotify = NULL;
+    gGetExtendedTcpTable = NULL;
     FreeLibrary(hLibrary);
     hLibrary = NULL;
   }
@@ -970,6 +975,39 @@ static void test_GetAdaptersAddresses(void)
     HeapFree(GetProcessHeap(), 0, ptr);
 }
 
+static void test_GetExtendedTcpTable(void)
+{
+    DWORD ret, size;
+    MIB_TCPTABLE *table;
+    MIB_TCPTABLE_OWNER_PID *table_pid;
+
+    if (!gGetExtendedTcpTable)
+    {
+        win_skip("GetExtendedTcpTable not available\n");
+        return;
+    }
+    ret = gGetExtendedTcpTable( NULL, NULL, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
+    ok( ret == ERROR_INVALID_PARAMETER, "got %u\n", ret );
+
+    size = 0;
+    ret = gGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
+    ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %u\n", ret );
+
+    table = HeapAlloc( GetProcessHeap(), 0, size );
+    ret = gGetExtendedTcpTable( table, &size, TRUE, AF_INET, TCP_TABLE_BASIC_ALL, 0 );
+    ok( ret == ERROR_SUCCESS, "got %u\n", ret );
+    HeapFree( GetProcessHeap(), 0, table );
+
+    size = 0;
+    ret = gGetExtendedTcpTable( NULL, &size, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0 );
+    ok( ret == ERROR_INSUFFICIENT_BUFFER, "got %u\n", ret );
+
+    table_pid = HeapAlloc( GetProcessHeap(), 0, size );
+    ret = gGetExtendedTcpTable( table_pid, &size, TRUE, AF_INET, TCP_TABLE_OWNER_PID_ALL, 0 );
+    ok( ret == ERROR_SUCCESS, "got %u\n", ret );
+    HeapFree( GetProcessHeap(), 0, table_pid );
+}
+
 START_TEST(iphlpapi)
 {
 
@@ -987,6 +1025,7 @@ START_TEST(iphlpapi)
 
     testWin2KFunctions();
     test_GetAdaptersAddresses();
+    test_GetExtendedTcpTable();
     freeIPHlpApi();
   }
 }
-- 
2.11.4.GIT