Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-04-05 21:21:33 +00:00
Code Issues Projects Releases Packages Wiki Activity

[PSAPI_APITEST]

Browse source 
Add tests for GetDeviceDriverBaseName
Replace 0x00000000 -> NULL. Dedicated to Thomas ;-)

svn path=/trunk/; revision=60601
This commit is contained in:
Pierre Schweitzer 2013-10-10 19:18:43 +00:00
parent 196ba3d618
commit 9b29c6967f
2 changed files with 80 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

89
rostests/apitests/psapi/psapi.c
View file

@ -1,7 +1,7 @@
/* /*
* PROJECT: ReactOS api tests * PROJECT: ReactOS api tests
* LICENSE: GPLv2+ - See COPYING in the top level directory * LICENSE: GPLv2+ - See COPYING in the top level directory
* PURPOSE: Test for GetDeviceDriverFileName * PURPOSE: Test for GetDeviceDriverFileName & GetDeviceDriverBaseName
* PROGRAMMER: Pierre Schweitzer * PROGRAMMER: Pierre Schweitzer
*/ */
@ -25,14 +25,14 @@ static LPVOID IntGetImageBase(LPCSTR Image)
Snap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, 0); Snap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, 0);
if (Snap == INVALID_HANDLE_VALUE) if (Snap == INVALID_HANDLE_VALUE)
{ {
return (LPVOID)0x00000000; return NULL;
} }
Module.dwSize = sizeof(MODULEENTRY32); Module.dwSize = sizeof(MODULEENTRY32);
if(!Module32First(Snap, &Module)) if(!Module32First(Snap, &Module))
{ {
CloseHandle(Snap); CloseHandle(Snap);
return (LPVOID)0x00000000; return NULL;
} }
do do
@ -45,10 +45,10 @@ static LPVOID IntGetImageBase(LPCSTR Image)
} while(Module32Next(Snap, &Module)); } while(Module32Next(Snap, &Module));
CloseHandle(Snap); CloseHandle(Snap);
return (LPVOID)0x00000000; return NULL;
} }
static BOOLEAN IntGetModuleInformation(LPCSTR Module, BOOLEAN IsDriver, BOOLEAN IsProcMod, TEST_MODULE_INFO * Info) static BOOLEAN IntGetModuleInformation(LPCSTR Module, BOOLEAN IsDriver, BOOLEAN IsProcMod, BOOLEAN BaseName, TEST_MODULE_INFO * Info)
{ {
CHAR System[255]; CHAR System[255];
UINT Len; UINT Len;
@ -85,9 +85,17 @@ static BOOLEAN IntGetModuleInformation(LPCSTR Module, BOOLEAN IsDriver, BOOLEAN
return FALSE; return FALSE;
} }
/* Skip disk */ if (BaseName)
strcpy(Info->Path, System + 2); {
Info->Len = strlen(Info->Path); strcpy(Info->Path, Module);
Info->Len = strlen(Info->Path);
}
else
{
/* Skip disk */
strcpy(Info->Path, System + 2);
Info->Len = strlen(Info->Path);
}
return TRUE; return TRUE;
} }
@ -103,7 +111,7 @@ START_TEST(GetDeviceDriverFileName)
ok(Len == 0, "Len: %lu\n", Len); ok(Len == 0, "Len: %lu\n", Len);
ok(GetLastError() == ERROR_INVALID_HANDLE, "Error: %lx\n", GetLastError()); 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); SetLastError(0xDEADBEEF);
Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255); Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
@ -126,7 +134,7 @@ START_TEST(GetDeviceDriverFileName)
skip("Couldn't find info about ntdll.dll\n"); 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); SetLastError(0xDEADBEEF);
Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255); Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
@ -138,7 +146,7 @@ START_TEST(GetDeviceDriverFileName)
skip("Couldn't find info about msvcrt.dll\n"); 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); SetLastError(0xDEADBEEF);
Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255); Len = GetDeviceDriverFileNameA(ModInfo.ImageBase, FileName, 255);
@ -150,3 +158,62 @@ START_TEST(GetDeviceDriverFileName)
skip("Couldn't find info about psapi.dll\n"); 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");
}
}

2
rostests/apitests/psapi/testlist.c
View file

@ -4,10 +4,12 @@
#include <wine/test.h> #include <wine/test.h>
extern void func_GetDeviceDriverFileName(void); extern void func_GetDeviceDriverFileName(void);
extern void func_GetDeviceDriverBaseName(void);
const struct test winetest_testlist[] = const struct test winetest_testlist[] =
{ {
{ "GetDeviceDriverFileName", func_GetDeviceDriverFileName }, { "GetDeviceDriverFileName", func_GetDeviceDriverFileName },
{ "GetDeviceDriverBaseName", func_GetDeviceDriverBaseName },
{ 0, 0 } { 0, 0 }
}; };