From: Paul Gofman <pgofman@codeweavers.com> --- dlls/ntdll/tests/info.c | 89 +++++++++++++++++++++++++++++++++++++++++ include/winternl.h | 5 ++- 2 files changed, 92 insertions(+), 2 deletions(-) diff --git a/dlls/ntdll/tests/info.c b/dlls/ntdll/tests/info.c index 3b133ce1c69..814ac377547 100644 --- a/dlls/ntdll/tests/info.c +++ b/dlls/ntdll/tests/info.c @@ -25,6 +25,7 @@ #define WIN32_NO_STATUS #include "windef.h" #include "winbase.h" +#include "winreg.h" #include "winternl.h" #include "winnls.h" #include "ddk/ntddk.h" @@ -4378,6 +4379,93 @@ static void test_processor_idle_cycle_time(void) ok( size == cpu_count * sizeof(*buffer), "got %#lx.\n", size ); } +static ULONG get_process_parameters_flags( HANDLE process ) +{ + RTL_USER_PROCESS_PARAMETERS *params; + PROCESS_BASIC_INFORMATION pbi; + ULONG flags = 0xdeadbeef; + NTSTATUS status; + SIZE_T len; + BOOL ret; + + status = pNtQueryInformationProcess( process, ProcessBasicInformation, &pbi, sizeof(pbi), NULL ); + ok( !status, "got %#lx.\n", status ); + ret = ReadProcessMemory( process, (char *)pbi.PebBaseAddress + offsetof(PEB, ProcessParameters), ¶ms, sizeof(params), &len ); + ok( ret, "got error %ld.\n", GetLastError() ); + ret = ReadProcessMemory( process, ¶ms->Flags, &flags, sizeof(flags), &len ); + ok( ret, "got error %ld.\n", GetLastError() ); + return flags; +} + +static void test_process_parameters_flags( int argc, char **argv ) +{ + PEB *peb = NtCurrentTeb()->Peb; + STARTUPINFOA si = { 0 }; + char cmdline[MAX_PATH]; + PROCESS_INFORMATION pi; + char keyname[MAX_PATH]; + const char *basename; + ULONG flags; + DWORD err; + HKEY hkey; + BOOL ret; + + if ((basename = strrchr( argv[0], '\\' ))) basename++; + else basename = argv[0]; + + sprintf( keyname, "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Image File Execution Options\\%s", basename ); + if (!strcmp( keyname + strlen(keyname) - 3, ".so" )) keyname[strlen(keyname) - 3] = 0; + todo_wine ok( peb->ProcessParameters->Flags & PROCESS_PARAMS_IMAGE_KEY_MISSING, "got %#lx.\n", peb->ProcessParameters->Flags ); + sprintf( cmdline, "%s %s %s", argv[0], argv[1], "check_pp_flags" ); + + si.cb = sizeof(si); + ret = CreateProcessA( NULL, cmdline, NULL, NULL, FALSE, DEBUG_PROCESS | CREATE_SUSPENDED, NULL, NULL, &si, &pi ); + ok( ret, "got error %ld.\n", GetLastError() ); + flags = get_process_parameters_flags( pi.hProcess ); + ok( !(flags & PROCESS_PARAMS_IMAGE_KEY_MISSING), "got %#lx.\n", peb->ProcessParameters->Flags ); + TerminateProcess( pi.hProcess, 0 ); + CloseHandle( pi.hThread ); + CloseHandle( pi.hProcess ); + + si.cb = sizeof(si); + ret = CreateProcessA( NULL, cmdline, NULL, NULL, FALSE, CREATE_SUSPENDED, NULL, NULL, &si, &pi ); + ok( ret, "got error %ld.\n", GetLastError() ); + flags = get_process_parameters_flags( pi.hProcess ); + todo_wine ok( flags & PROCESS_PARAMS_IMAGE_KEY_MISSING, "got %#lx.\n", peb->ProcessParameters->Flags ); + TerminateProcess( pi.hProcess, 0 ); + CloseHandle( pi.hThread ); + CloseHandle( pi.hProcess ); + + err = RegCreateKeyA( HKEY_LOCAL_MACHINE, keyname, &hkey ); + if (err == ERROR_ACCESS_DENIED) + { + skip( "Not authorized to change the image file execution options.\n" ); + return; + } + ok( !err, "got %#lx.\n", err ); + + si.cb = sizeof(si); + ret = CreateProcessA( NULL, cmdline, NULL, NULL, FALSE, CREATE_SUSPENDED, NULL, NULL, &si, &pi ); + ok( ret, "got error %ld.\n", GetLastError() ); + flags = get_process_parameters_flags( pi.hProcess ); + ok( !(flags & PROCESS_PARAMS_IMAGE_KEY_MISSING), "got %#lx.\n", peb->ProcessParameters->Flags ); + TerminateProcess( pi.hProcess, 0 ); + CloseHandle( pi.hThread ); + CloseHandle( pi.hProcess ); + + RegCloseKey( hkey ); + RegDeleteKeyA( HKEY_LOCAL_MACHINE, keyname ); + + si.cb = sizeof(si); + ret = CreateProcessA( NULL, cmdline, NULL, NULL, FALSE, CREATE_SUSPENDED, NULL, NULL, &si, &pi ); + ok( ret, "got error %ld.\n", GetLastError() ); + flags = get_process_parameters_flags( pi.hProcess ); + todo_wine ok( flags & PROCESS_PARAMS_IMAGE_KEY_MISSING, "got %#lx.\n", peb->ProcessParameters->Flags ); + TerminateProcess( pi.hProcess, 0 ); + CloseHandle( pi.hThread ); + CloseHandle( pi.hProcess ); +} + START_TEST(info) { char **argv; @@ -4459,4 +4547,5 @@ START_TEST(info) test_process_token(argc, argv); test_process_id(); test_processor_idle_cycle_time(); + test_process_parameters_flags(argc, argv); } diff --git a/include/winternl.h b/include/winternl.h index bed72e83ce4..81dc3db2b74 100644 --- a/include/winternl.h +++ b/include/winternl.h @@ -201,8 +201,9 @@ typedef struct _RTL_USER_PROCESS_PARAMETERS ULONG LoaderThreads; } RTL_USER_PROCESS_PARAMETERS, *PRTL_USER_PROCESS_PARAMETERS; -/* value for Flags field (FIXME: not the correct name) */ -#define PROCESS_PARAMS_FLAG_NORMALIZED 1 +/* value for Flags field (FIXME: not the correct names) */ +#define PROCESS_PARAMS_FLAG_NORMALIZED 0x00000001 +#define PROCESS_PARAMS_IMAGE_KEY_MISSING 0x00004000 typedef struct _PEB_LDR_DATA { -- GitLab https://gitlab.winehq.org/wine/wine/-/merge_requests/11167