[pschweitzer] 60601: [PSAPI_APITEST] Add tests for GetDeviceDriverBaseName Replace 0x00000000 -> NULL. Dedicated to Thomas ; -)

Author: pschweitzer
Date: Thu Oct 10 19:18:43 2013
New Revision: 60601

URL: http://svn.reactos.org/svn/reactos?rev=60601&view=rev
Log:
[PSAPI_APITEST]
Add tests for GetDeviceDriverBaseName
Replace 0x00000000 -> NULL. Dedicated to Thomas ;-)

Modified:
    trunk/rostests/apitests/psapi/psapi.c
    trunk/rostests/apitests/psapi/testlist.c

Modified: trunk/rostests/apitests/psapi/psapi.c
URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/psapi/psapi.c?rev...
==============================================================================
--- trunk/rostests/apitests/psapi/psapi.c	[iso-8859-1] (original)
+++ trunk/rostests/apitests/psapi/psapi.c	[iso-8859-1] Thu Oct 10 19:18:43 2013
@@ -1,7 +1,7 @@
 /*
  * PROJECT:         ReactOS api tests
  * LICENSE:         GPLv2+ - See COPYING in the top level directory
- * PURPOSE:         Test for GetDeviceDriverFileName
+ * PURPOSE:         Test for GetDeviceDriverFileName & GetDeviceDriverBaseName
  * PROGRAMMER:      Pierre Schweitzer
  */

@@ -25,14 +25,14 @@
     Snap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, 0);
     if (Snap == INVALID_HANDLE_VALUE)
     {
-        return (LPVOID)0x00000000;
+        return NULL;
     }

Module.dwSize = sizeof(MODULEENTRY32);
     if(!Module32First(Snap, &Module))
     {
         CloseHandle(Snap);
-        return (LPVOID)0x00000000;
+        return NULL;
     }

do
@@ -45,10 +45,10 @@
     } while(Module32Next(Snap, &Module));

CloseHandle(Snap);
-    return (LPVOID)0x00000000;
-}
-
-static BOOLEAN IntGetModuleInformation(LPCSTR Module, BOOLEAN IsDriver, BOOLEAN IsProcMod, TEST_MODULE_INFO * Info)
+    return NULL;
+}
+
+static BOOLEAN IntGetModuleInformation(LPCSTR Module, BOOLEAN IsDriver, BOOLEAN IsProcMod, BOOLEAN BaseName, TEST_MODULE_INFO * Info)
 {
     CHAR System[255];
     UINT Len;
@@ -85,9 +85,17 @@
         return FALSE;
     }

-    /* Skip disk */
-    strcpy(Info->Path, System + 2);
-    Info->Len = strlen(Info->Path);
+    if (BaseName)
+    {
+        strcpy(Info->Path, Module);
+        Info->Len = strlen(Info->Path);
+    }
+    else
+    {
+        /* Skip disk */
+        strcpy(Info->Path, System + 2);
+        Info->Len = strlen(Info->Path);
+    }

return TRUE;
 }
@@ -103,7 +111,7 @@
     ok(Len == 0, "Len: %lu\n", Len);
     ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());

-    if (IntGetModuleInformation("ntdll.dll", FALSE, TRUE, &ModInfo))
+    if (IntGetModuleInformation("ntdll.dll", FALSE, TRUE, FALSE, &ModInfo))
     {
         SetLastError(0xDEADBEEF);
         Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
@@ -126,7 +134,7 @@
         skip("Couldn't find info about ntdll.dll\n");
     }

-    if (IntGetModuleInformation("msvcrt.dll", FALSE, TRUE, &ModInfo))
+    if (IntGetModuleInformation("msvcrt.dll", FALSE, TRUE, FALSE, &ModInfo))
     {
         SetLastError(0xDEADBEEF);
         Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
@@ -138,7 +146,7 @@
         skip("Couldn't find info about msvcrt.dll\n");
     }

-    if (IntGetModuleInformation("psapi.dll", FALSE, TRUE, &ModInfo))
+    if (IntGetModuleInformation("psapi.dll", FALSE, TRUE, FALSE, &ModInfo))
     {
         SetLastError(0xDEADBEEF);
         Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
@@ -150,3 +158,62 @@
         skip("Couldn't find info about psapi.dll\n");
     }
 }
+
+START_TEST(GetDeviceDriverBaseName)
+{
+    DWORD Len;
+    CHAR FileName[255];
+    TEST_MODULE_INFO ModInfo;
+
+    SetLastError(0xDEADBEEF);
+    Len = GetDeviceDriverBaseNameA(0, FileName, 255);
+    ok(Len == 0, "Len: %lu\n", Len);
+    ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());
+
+    if (IntGetModuleInformation("ntdll.dll", FALSE, TRUE, TRUE, &ModInfo))
+    {
+        SetLastError(0xDEADBEEF);
+        Len = GetDeviceDriverBaseNameA(ModInfo.ImageBase, FileName, 255);
+        ok(Len == ModInfo.Len, "Len: %lu\n", Len);
+        ok(GetLastError() == 0xDEADBEEF, "Error: %lx\n", GetLastError());
+        ok(lstrcmpiA(ModInfo.Path, FileName) == 0, "File name: %s\n", FileName);
+
+        /* Test with too small buffer */
+        SetLastError(0xDEADBEEF);
+        ModInfo.Len--;
+        ModInfo.Path[ModInfo.Len] = 0;
+        FileName[ModInfo.Len] = 0;
+        Len = GetDeviceDriverBaseNameA(ModInfo.ImageBase, FileName, ModInfo.Len);
+        ok(Len == ModInfo.Len, "Len: %lu\n", Len);
+        ok(GetLastError() == 0xDEADBEEF, "Error: %lx\n", GetLastError());
+        ok(lstrcmpiA(ModInfo.Path, FileName) == 0, "File name: %s\n", FileName);
+    }
+    else
+    {
+        skip("Couldn't find info about ntdll.dll\n");
+    }
+
+    if (IntGetModuleInformation("msvcrt.dll", FALSE, TRUE, TRUE, &ModInfo))
+    {
+        SetLastError(0xDEADBEEF);
+        Len = GetDeviceDriverBaseNameA(ModInfo.ImageBase, FileName, 255);
+        ok(Len == 0, "Len: %lu\n", Len);
+        ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());
+    }
+    else
+    {
+        skip("Couldn't find info about msvcrt.dll\n");
+    }
+
+    if (IntGetModuleInformation("psapi.dll", FALSE, TRUE, TRUE, &ModInfo))
+    {
+        SetLastError(0xDEADBEEF);
+        Len = GetDeviceDriverBaseNameA(ModInfo.ImageBase, FileName, 255);
+        ok(Len == 0, "Len: %lu\n", Len);
+        ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError());
+    }
+    else
+    {
+        skip("Couldn't find info about psapi.dll\n");
+    }
+}

Modified: trunk/rostests/apitests/psapi/testlist.c
URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/psapi/testlist.c?...
==============================================================================
--- trunk/rostests/apitests/psapi/testlist.c	[iso-8859-1] (original)
+++ trunk/rostests/apitests/psapi/testlist.c	[iso-8859-1] Thu Oct 10 19:18:43 2013
@@ -4,10 +4,12 @@
 #include <wine/test.h>

extern void func_GetDeviceDriverFileName(void);
+extern void func_GetDeviceDriverBaseName(void);

const struct test winetest_testlist[] =
 {
     { "GetDeviceDriverFileName",                    func_GetDeviceDriverFileName },
+    { "GetDeviceDriverBaseName",                    func_GetDeviceDriverBaseName },

{ 0, 0 }
 };