From b7674bb16497c02bd50ce46acacc17bb12a9511a Mon Sep 17 00:00:00 2001 From: Pierre Schweitzer Date: Sun, 6 Oct 2013 22:16:42 +0000 Subject: [PATCH] [PSAPI_APITEST] 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 --- rostests/apitests/CMakeLists.txt | 1 + rostests/apitests/psapi/CMakeLists.txt | 10 ++ rostests/apitests/psapi/psapi.c | 128 +++++++++++++++++++++++++ rostests/apitests/psapi/testlist.c | 14 +++ 4 files changed, 153 insertions(+) create mode 100644 rostests/apitests/psapi/CMakeLists.txt create mode 100644 rostests/apitests/psapi/psapi.c create mode 100644 rostests/apitests/psapi/testlist.c diff --git a/rostests/apitests/CMakeLists.txt b/rostests/apitests/CMakeLists.txt index b7d7a29e3c4..0d5d986e76a 100644 --- a/rostests/apitests/CMakeLists.txt +++ b/rostests/apitests/CMakeLists.txt @@ -13,6 +13,7 @@ add_subdirectory(kernel32) add_subdirectory(msvcrt) add_subdirectory(ntdll) add_subdirectory(powrprof) +add_subdirectory(psapi) add_subdirectory(user32) if((NOT MSVC) AND (ARCH STREQUAL "i386")) add_subdirectory(w32kdll) diff --git a/rostests/apitests/psapi/CMakeLists.txt b/rostests/apitests/psapi/CMakeLists.txt new file mode 100644 index 00000000000..5c92c82d92b --- /dev/null +++ b/rostests/apitests/psapi/CMakeLists.txt @@ -0,0 +1,10 @@ + +list(APPEND SOURCE + psapi.c + testlist.c) + +add_executable(psapi_apitest ${SOURCE}) +target_link_libraries(psapi_apitest wine) +set_module_type(psapi_apitest win32cui) +add_importlibs(psapi_apitest psapi msvcrt kernel32 ntdll) +add_cd_file(TARGET psapi_apitest DESTINATION reactos/bin FOR all) diff --git a/rostests/apitests/psapi/psapi.c b/rostests/apitests/psapi/psapi.c new file mode 100644 index 00000000000..9cbf8ab0c1e --- /dev/null +++ b/rostests/apitests/psapi/psapi.c @@ -0,0 +1,128 @@ +/* + * PROJECT: ReactOS api tests + * LICENSE: GPLv2+ - See COPYING in the top level directory + * PURPOSE: Test for GetDeviceDriverFileName + * PROGRAMMER: Pierre Schweitzer + */ + +#include +#include +#include +#include + +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"); + } +} diff --git a/rostests/apitests/psapi/testlist.c b/rostests/apitests/psapi/testlist.c new file mode 100644 index 00000000000..170b213e0f2 --- /dev/null +++ b/rostests/apitests/psapi/testlist.c @@ -0,0 +1,14 @@ +#define __ROS_LONG64__ + +#define STANDALONE +#include + +extern void func_GetDeviceDriverFileName(void); + +const struct test winetest_testlist[] = +{ + { "GetDeviceDriverFileName", func_GetDeviceDriverFileName }, + + { 0, 0 } +}; + -- 2.17.1