mirror of
https://github.com/reactos/reactos.git
synced 2024-08-20 18:35:32 +00:00
b7674bb164
Add an apitest for psapi GetDeviceDriverFileName. I'm looking for help to bring it even farther. Ideally, it would be interesting to be able to GetDeviceDriverFileName on ntoskrnl base address. The whole point is about getting it dynamically. The day we can do it properly, I can predict that it will fail on ReactOS, we're not having correct paths for KDCOM, HAL, and NTOSKRNL modules in the kernel (thank you FreeLdr? - Where are you starting '\'?) svn path=/trunk/; revision=60566
129 lines
3.2 KiB
C
129 lines
3.2 KiB
C
/*
|
|
* PROJECT: ReactOS api tests
|
|
* LICENSE: GPLv2+ - See COPYING in the top level directory
|
|
* PURPOSE: Test for GetDeviceDriverFileName
|
|
* PROGRAMMER: Pierre Schweitzer
|
|
*/
|
|
|
|
#include <apitest.h>
|
|
#include <psapi.h>
|
|
#include <tlhelp32.h>
|
|
#include <stdio.h>
|
|
|
|
typedef struct
|
|
{
|
|
LPVOID ImageBase;
|
|
CHAR Path[255];
|
|
DWORD Len;
|
|
} TEST_MODULE_INFO;
|
|
|
|
static LPVOID IntGetImageBase(LPCSTR Image)
|
|
{
|
|
HANDLE Snap;
|
|
MODULEENTRY32 Module;
|
|
|
|
Snap = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, 0);
|
|
if (Snap == INVALID_HANDLE_VALUE)
|
|
{
|
|
return (LPVOID)0x00000000;
|
|
}
|
|
|
|
Module.dwSize = sizeof(MODULEENTRY32);
|
|
if(!Module32First(Snap, &Module))
|
|
{
|
|
CloseHandle(Snap);
|
|
return (LPVOID)0x00000000;
|
|
}
|
|
|
|
do
|
|
{
|
|
if (lstrcmpiA(Module.szExePath, Image) == 0)
|
|
{
|
|
CloseHandle(Snap);
|
|
return (LPVOID)Module.modBaseAddr;
|
|
}
|
|
} while(Module32Next(Snap, &Module));
|
|
|
|
CloseHandle(Snap);
|
|
return (LPVOID)0x00000000;
|
|
}
|
|
|
|
static BOOLEAN IntGetModuleInformation(LPCSTR Module, BOOLEAN IsDriver, BOOLEAN IsProcMod, TEST_MODULE_INFO * Info)
|
|
{
|
|
CHAR System[255];
|
|
UINT Len;
|
|
|
|
memset(Info, 0, sizeof(TEST_MODULE_INFO));
|
|
|
|
/* Get system path */
|
|
Len = GetSystemWindowsDirectory(System, 255);
|
|
if (Len > 255 || Len == 0)
|
|
{
|
|
printf("GetSystemWindowsDirectory failed\n");
|
|
return FALSE;
|
|
}
|
|
|
|
/* Make path to module */
|
|
strcat(System, "\\system32\\");
|
|
if (IsDriver) strcat(System, "drivers\\");
|
|
strcat(System, Module);
|
|
|
|
/* Get base address */
|
|
if (IsProcMod)
|
|
{
|
|
Info->ImageBase = IntGetImageBase(System);
|
|
if (!Info->ImageBase)
|
|
{
|
|
printf("IntGetImageBase failed\n");
|
|
return FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* FIXME */
|
|
printf("Not supported yet!\n");
|
|
return FALSE;
|
|
}
|
|
|
|
/* Skip disk */
|
|
strcpy(Info->Path, System + 2);
|
|
Info->Len = strlen(Info->Path);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
START_TEST(GetDeviceDriverFileName)
|
|
{
|
|
DWORD Len;
|
|
CHAR FileName[255];
|
|
TEST_MODULE_INFO ModInfo;
|
|
|
|
SetLastError(0xDEADBEEF);
|
|
Len = GetDeviceDriverFileNameA(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, &ModInfo))
|
|
{
|
|
SetLastError(0xDEADBEEF);
|
|
Len = GetDeviceDriverFileNameA(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 = GetDeviceDriverFileNameA(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");
|
|
}
|
|
}
|