[pschweitzer] 60566: [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 GetDeviceDriverF...
Author: pschweitzer Date: Sun Oct 6 22:16:42 2013 New Revision: 60566
URL: http://svn.reactos.org/svn/reactos?rev=60566&view=rev Log: [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 ''?)
Added: trunk/rostests/apitests/psapi/ trunk/rostests/apitests/psapi/CMakeLists.txt (with props) trunk/rostests/apitests/psapi/psapi.c (with props) trunk/rostests/apitests/psapi/testlist.c (with props) Modified: trunk/rostests/apitests/CMakeLists.txt
Modified: trunk/rostests/apitests/CMakeLists.txt URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/CMakeLists.txt?re... ============================================================================== --- trunk/rostests/apitests/CMakeLists.txt [iso-8859-1] (original) +++ trunk/rostests/apitests/CMakeLists.txt [iso-8859-1] Sun Oct 6 22:16:42 2013 @@ -13,6 +13,7 @@ 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)
Added: trunk/rostests/apitests/psapi/CMakeLists.txt URL: http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/psapi/CMakeLists.... ============================================================================== --- trunk/rostests/apitests/psapi/CMakeLists.txt (added) +++ trunk/rostests/apitests/psapi/CMakeLists.txt [iso-8859-1] Sun Oct 6 22:16:42 2013 @@ -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)
Propchange: trunk/rostests/apitests/psapi/CMakeLists.txt ------------------------------------------------------------------------------ svn:eol-style = native
Added: 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 (added) +++ trunk/rostests/apitests/psapi/psapi.c [iso-8859-1] Sun Oct 6 22:16:42 2013 @@ -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 <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"); + } +}
Propchange: trunk/rostests/apitests/psapi/psapi.c ------------------------------------------------------------------------------ svn:eol-style = native
Added: 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 (added) +++ trunk/rostests/apitests/psapi/testlist.c [iso-8859-1] Sun Oct 6 22:16:42 2013 @@ -0,0 +1,14 @@ +#define __ROS_LONG64__ + +#define STANDALONE +#include <wine/test.h> + +extern void func_GetDeviceDriverFileName(void); + +const struct test winetest_testlist[] = +{ + { "GetDeviceDriverFileName", func_GetDeviceDriverFileName }, + + { 0, 0 } +}; +
Propchange: trunk/rostests/apitests/psapi/testlist.c ------------------------------------------------------------------------------ svn:eol-style = native
-
pschweitzer＠svn.reactos.org