mirror of
https://github.com/reactos/reactos.git
synced 2025-01-01 03:54:02 +00:00
[APPHELP_APITEST]
- Add tests for the apphelp SDB compatibility layer. Patch by Mark Jansen. CORE-10367 svn path=/trunk/; revision=71077
This commit is contained in:
parent
df2d6a0e8a
commit
001dc229d5
4 changed files with 951 additions and 0 deletions
|
@ -1,6 +1,7 @@
|
|||
include_directories(include)
|
||||
|
||||
add_subdirectory(advapi32)
|
||||
add_subdirectory(apphelp)
|
||||
add_subdirectory(atl)
|
||||
add_subdirectory(browseui)
|
||||
add_subdirectory(com)
|
||||
|
|
11
rostests/apitests/apphelp/CMakeLists.txt
Normal file
11
rostests/apitests/apphelp/CMakeLists.txt
Normal file
|
@ -0,0 +1,11 @@
|
|||
|
||||
add_definitions(-D__ROS_LONG64__)
|
||||
|
||||
list(APPEND SOURCE
|
||||
layerapi.c
|
||||
testlist.c)
|
||||
|
||||
add_executable(apphelp_apitest ${SOURCE})
|
||||
set_module_type(apphelp_apitest win32cui)
|
||||
add_importlibs(apphelp_apitest advapi32 shlwapi msvcrt kernel32 ntdll)
|
||||
add_cd_file(TARGET apphelp_apitest DESTINATION reactos/bin FOR all)
|
927
rostests/apitests/apphelp/layerapi.c
Normal file
927
rostests/apitests/apphelp/layerapi.c
Normal file
|
@ -0,0 +1,927 @@
|
|||
/*
|
||||
* Copyright 2015 Mark Jansen
|
||||
*
|
||||
* This library is free software; you can redistribute it and/or
|
||||
* modify it under the terms of the GNU Lesser General Public
|
||||
* License as published by the Free Software Foundation; either
|
||||
* version 2.1 of the License, or (at your option) any later version.
|
||||
*
|
||||
* This library is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public
|
||||
* License along with this library; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
||||
*/
|
||||
|
||||
#include <ntstatus.h>
|
||||
#define WIN32_NO_STATUS
|
||||
#include <windows.h>
|
||||
#include <shlwapi.h>
|
||||
#include <winnt.h>
|
||||
#ifdef __REACTOS__
|
||||
#include <ntndk.h>
|
||||
#else
|
||||
#include <winternl.h>
|
||||
#endif
|
||||
#include <winerror.h>
|
||||
#include <stdio.h>
|
||||
|
||||
#include "wine/test.h"
|
||||
|
||||
|
||||
#define GPLK_USER 1
|
||||
#define GPLK_MACHINE 2
|
||||
#define MAX_LAYER_LENGTH 256
|
||||
#define LAYER_APPLY_TO_SYSTEM_EXES 1
|
||||
|
||||
|
||||
static HMODULE hdll;
|
||||
static BOOL(WINAPI *pAllowPermLayer)(PCWSTR path);
|
||||
static BOOL(WINAPI *pSdbSetPermLayerKeys)(PCWSTR wszPath, PCWSTR wszLayers, BOOL bMachine);
|
||||
static BOOL(WINAPI *pSdbGetPermLayerKeys)(PCWSTR wszPath, PWSTR pwszLayers, PDWORD pdwBytes, DWORD dwFlags);
|
||||
static BOOL(WINAPI *pSetPermLayerState)(PCWSTR wszPath, PCWSTR wszLayer, DWORD dwFlags, BOOL bMachine, BOOL bEnable);
|
||||
|
||||
|
||||
static DWORD g_Version;
|
||||
#define APPHELP_VISTA 0x0600
|
||||
#define APPHELP_WIN8 0x0602
|
||||
#define APPHELP_WIN10 0x1000
|
||||
|
||||
|
||||
/* Helper function to disable Wow64 redirection on an os that reports it being enabled. */
|
||||
static DWORD g_QueryFlag = 0xffffffff;
|
||||
static DWORD QueryFlag(void)
|
||||
{
|
||||
if (g_QueryFlag == 0xffffffff)
|
||||
{
|
||||
ULONG_PTR wow64_ptr = 0;
|
||||
NTSTATUS status = NtQueryInformationProcess(NtCurrentProcess(), ProcessWow64Information, &wow64_ptr, sizeof(wow64_ptr), NULL);
|
||||
g_QueryFlag = (NT_SUCCESS(status) && wow64_ptr != 0) ? KEY_WOW64_64KEY : 0;
|
||||
}
|
||||
return g_QueryFlag;
|
||||
}
|
||||
|
||||
/* Helper function to prepare the registry key with a value. */
|
||||
static BOOL setLayerValue(BOOL bMachine, const char* valueName, const char* value)
|
||||
{
|
||||
HKEY key = NULL;
|
||||
LSTATUS lstatus = RegCreateKeyExA(bMachine ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER,
|
||||
"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\Layers", 0, NULL, 0, QueryFlag() | KEY_SET_VALUE, NULL, &key, NULL);
|
||||
if (lstatus == ERROR_SUCCESS)
|
||||
{
|
||||
if (value)
|
||||
lstatus = RegSetValueExA(key, valueName, 0, REG_SZ, (const BYTE*)value, strlen(value)+1);
|
||||
else
|
||||
{
|
||||
lstatus = RegDeleteValueA(key, valueName);
|
||||
lstatus = (lstatus == ERROR_FILE_NOT_FOUND ? ERROR_SUCCESS : lstatus);
|
||||
}
|
||||
RegCloseKey(key);
|
||||
}
|
||||
return lstatus == ERROR_SUCCESS;
|
||||
}
|
||||
|
||||
|
||||
static void expect_LayerValue_imp(BOOL bMachine, const char* valueName, const char* value)
|
||||
{
|
||||
HKEY key = NULL;
|
||||
LSTATUS lstatus = RegCreateKeyExA(bMachine ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER,
|
||||
"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\Layers", 0, NULL, 0, QueryFlag() | KEY_QUERY_VALUE, NULL, &key, NULL);
|
||||
winetest_ok(lstatus == ERROR_SUCCESS, "Expected to be able to open a registry key\n");
|
||||
if (lstatus == ERROR_SUCCESS)
|
||||
{
|
||||
char data[512] = { 0 };
|
||||
DWORD dwType = 0;
|
||||
DWORD dwDataLen = sizeof(data);
|
||||
lstatus = RegQueryValueExA(key, valueName, NULL, &dwType, (LPBYTE)data, &dwDataLen);
|
||||
if (value)
|
||||
{
|
||||
winetest_ok(lstatus == ERROR_SUCCESS, "Expected to get a valid value, err: %u\n", lstatus);
|
||||
if (lstatus == ERROR_SUCCESS)
|
||||
{
|
||||
winetest_ok(dwType == REG_SZ, "Expected the type to be REG_SZ, was: %u\n", dwType);
|
||||
winetest_ok(!strcmp(data, value), "Expected the data to be: '%s', was: '%s'\n", value, data);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
winetest_ok(lstatus == ERROR_FILE_NOT_FOUND, "Expected not to find the value %s\n", valueName);
|
||||
}
|
||||
RegCloseKey(key);
|
||||
}
|
||||
}
|
||||
|
||||
static void expect_LayerValue_imp2(BOOL bMachine, const char* valueName, const char* value, int use_alt, const char* alt_value)
|
||||
{
|
||||
expect_LayerValue_imp(bMachine, valueName, use_alt ? alt_value : value);
|
||||
}
|
||||
|
||||
|
||||
void expect_Sdb_imp(PCSTR path, DWORD type, BOOL result, DWORD lenResult, PCSTR stringResult)
|
||||
{
|
||||
WCHAR pathW[MAX_PATH], buffer[MAX_LAYER_LENGTH] = { 0 };
|
||||
char resultBuffer[MAX_LAYER_LENGTH] = { 0 };
|
||||
DWORD dwBufSize = sizeof(buffer);
|
||||
|
||||
/* In case of a failure, the buffer size is sometimes set to 0, and sometimes not touched,
|
||||
depending on the version. Either case is fine, since the function returns FALSE anyway. */
|
||||
|
||||
MultiByteToWideChar(CP_ACP, 0, path, -1, pathW, MAX_PATH);
|
||||
|
||||
winetest_ok(pSdbGetPermLayerKeys(pathW, buffer, &dwBufSize, type) == result, "Expected pSdbGetPermLayerKeys to %s\n", (result ? "succeed" : "fail"));
|
||||
if (!result && lenResult == 0xffffffff)
|
||||
winetest_ok(dwBufSize == 0 || dwBufSize == sizeof(buffer), "Expected dwBufSize to be 0 or %u, was %u\n", sizeof(buffer), dwBufSize);
|
||||
else
|
||||
winetest_ok(dwBufSize == lenResult ||
|
||||
/* W2k3 is off by 2 when concatenating user / machine */
|
||||
broken(g_Version < APPHELP_VISTA && type == (GPLK_MACHINE|GPLK_USER) && (lenResult + 2) == dwBufSize),
|
||||
"Expected dwBufSize to be %u, was %u\n", lenResult, dwBufSize);
|
||||
if (result)
|
||||
{
|
||||
winetest_ok(lstrlenW(buffer) * sizeof(WCHAR) + sizeof(WCHAR) == lenResult, "Expected lstrlenW(buffer)*2+2 to be %u, was %u\n",
|
||||
lenResult, lstrlenW(buffer) * sizeof(WCHAR) + sizeof(WCHAR));
|
||||
}
|
||||
WideCharToMultiByte(CP_ACP, 0, buffer, -1, resultBuffer, sizeof(resultBuffer), NULL, NULL);
|
||||
winetest_ok(!strcmp(stringResult, resultBuffer), "Expected the result to be '%s', was '%s'\n", stringResult, resultBuffer);
|
||||
}
|
||||
|
||||
|
||||
/* In case of a failure, let the location be from where the function was invoked, not inside the function itself. */
|
||||
#define expect_Sdb (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : expect_Sdb_imp
|
||||
#define expect_LayerValue (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : expect_LayerValue_imp
|
||||
#define expect_LayerValue2 (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : expect_LayerValue_imp2
|
||||
|
||||
|
||||
BOOL wrapAllowPermLayer(const char* str)
|
||||
{
|
||||
WCHAR buf[100];
|
||||
MultiByteToWideChar(CP_ACP, 0, str, -1, buf, 100);
|
||||
return pAllowPermLayer(buf);
|
||||
}
|
||||
|
||||
/* Brute forcing all ascii chars in the first 2 places seems to indicate that all it cares for is:
|
||||
- Second char has to be a ':'
|
||||
if it's not a ':', display a diagnostic message (and a different one for '\\').
|
||||
- First char does not really matter, as long as it's not on a DRIVE_REMOTE (but, according to the logging this is meant to check for a CDROM drive...)
|
||||
*/
|
||||
static void test_AllowPermLayer(void)
|
||||
{
|
||||
char buf[20];
|
||||
char drive_letter;
|
||||
UINT drivetype = 0;
|
||||
ok(pAllowPermLayer(NULL) == FALSE, "Expected AllowPermLayer to fail for NULL\n");
|
||||
if (g_Version < APPHELP_WIN8)
|
||||
{
|
||||
ok(wrapAllowPermLayer("-:"), "Expected AllowPermLayer to succeed\n");
|
||||
ok(wrapAllowPermLayer("@:"), "Expected AllowPermLayer to succeed\n");
|
||||
ok(wrapAllowPermLayer("4:"), "Expected AllowPermLayer to succeed\n");
|
||||
ok(wrapAllowPermLayer("*:"), "Expected AllowPermLayer to succeed\n");
|
||||
}
|
||||
ok(wrapAllowPermLayer("*a") == FALSE, "Expected AllowPermLayer to fail\n");
|
||||
ok(wrapAllowPermLayer("*\\") == FALSE, "Expected AllowPermLayer to fail\n");
|
||||
for (drive_letter = 'a'; drive_letter <= 'z'; ++drive_letter)
|
||||
{
|
||||
sprintf(buf, "%c:\\", drive_letter);
|
||||
drivetype = GetDriveTypeA(buf);
|
||||
ok(wrapAllowPermLayer(buf) == (drivetype != DRIVE_REMOTE), "Expected AllowPermLayer to be %d for %c:\\\n", (drivetype != DRIVE_REMOTE), drive_letter);
|
||||
}
|
||||
}
|
||||
|
||||
static BOOL wrapSdbSetPermLayerKeys(PCWSTR wszPath, PCSTR szLayers, BOOL bMachine)
|
||||
{
|
||||
WCHAR wszLayers[MAX_LAYER_LENGTH];
|
||||
MultiByteToWideChar(CP_ACP, 0, szLayers, -1, wszLayers, MAX_LAYER_LENGTH);
|
||||
return pSdbSetPermLayerKeys(wszPath, wszLayers, bMachine);
|
||||
}
|
||||
|
||||
static void test_SdbSetPermLayerKeysLevel(BOOL bMachine, const char* file)
|
||||
{
|
||||
WCHAR fileW[MAX_PATH+20];
|
||||
WCHAR emptyString[1] = { 0 };
|
||||
|
||||
MultiByteToWideChar(CP_ACP, 0, file, -1, fileW, MAX_PATH+20);
|
||||
|
||||
/* Test some parameter validation. */
|
||||
ok(pSdbSetPermLayerKeys(NULL, NULL, bMachine) == FALSE, "Expected SdbSetPermLayerKeys to fail\n");
|
||||
ok(pSdbSetPermLayerKeys(NULL, emptyString, bMachine) == FALSE, "Expected SdbSetPermLayerKeys to fail\n");
|
||||
ok(pSdbSetPermLayerKeys(emptyString, emptyString, bMachine) == FALSE, "Expected SdbSetPermLayerKeys to fail\n");
|
||||
ok(pSdbSetPermLayerKeys(fileW, NULL, bMachine) == TRUE, "Expected SdbSetPermLayerKeys to succeed\n");
|
||||
ok(pSdbSetPermLayerKeys(fileW, emptyString, bMachine) == TRUE, "Expected SdbSetPermLayerKeys to fail\n");
|
||||
|
||||
/* Basic tests */
|
||||
ok(wrapSdbSetPermLayerKeys(fileW, "TEST1", bMachine), "Expected SdbSetPermLayerKeys to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "TEST1");
|
||||
|
||||
ok(wrapSdbSetPermLayerKeys(fileW, "TEST1 TEST2", bMachine), "Expected SdbSetPermLayerKeys to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "TEST1 TEST2");
|
||||
|
||||
/* SdbSetPermLayerKeys does not do any validation of the value passed in. */
|
||||
ok(wrapSdbSetPermLayerKeys(fileW, "!#$% TEST1 TEST2", bMachine), "Expected SdbSetPermLayerKeys to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "!#$% TEST1 TEST2");
|
||||
|
||||
ok(wrapSdbSetPermLayerKeys(fileW, "!#$% TEST1 TEST2", bMachine), "Expected SdbSetPermLayerKeys to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "!#$% TEST1 TEST2");
|
||||
|
||||
ok(pSdbSetPermLayerKeys(fileW, NULL, bMachine) == TRUE, "Expected SdbSetPermLayerKeys to succeed\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
ok(wrapSdbSetPermLayerKeys(fileW, " ", bMachine), "Expected SdbSetPermLayerKeys to succeed\n");
|
||||
expect_LayerValue(bMachine, file, " ");
|
||||
|
||||
ok(pSdbSetPermLayerKeys(fileW, NULL, bMachine) == TRUE, "Expected SdbSetPermLayerKeys to fail\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
}
|
||||
|
||||
static void test_SdbGetPermLayerKeys(void)
|
||||
{
|
||||
WCHAR pathW[MAX_PATH], buffer[MAX_LAYER_LENGTH] = { 0 };
|
||||
char file[MAX_PATH + 20], tmp[MAX_PATH + 20];
|
||||
BOOL bUser, bMachine;
|
||||
HANDLE hfile;
|
||||
DWORD dwBufSize = sizeof(buffer);
|
||||
|
||||
GetTempPathA(MAX_PATH, tmp);
|
||||
GetLongPathNameA(tmp, file, sizeof(file));
|
||||
PathCombineA(tmp, file, "notexist.exe");
|
||||
PathAppendA(file, "test_file.exe");
|
||||
|
||||
/* Check that we can access the keys */
|
||||
bUser = setLayerValue(FALSE, file, "RUNASADMIN WINXPSP3");
|
||||
expect_LayerValue(FALSE, file, "RUNASADMIN WINXPSP3");
|
||||
ok(bUser, "Expected to be able to set atleast the flags for the user\n");
|
||||
if (!bUser)
|
||||
{
|
||||
skip("Cannot do any tests if I cannot set some values\n");
|
||||
return;
|
||||
}
|
||||
bMachine = setLayerValue(TRUE, file, "WINXPSP3 WINXPSP2");
|
||||
if (bMachine)
|
||||
{
|
||||
expect_LayerValue(TRUE, file, "WINXPSP3 WINXPSP2");
|
||||
}
|
||||
|
||||
|
||||
hfile = CreateFileA(file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
||||
ok(hfile != INVALID_HANDLE_VALUE, "CreateFile failed on '%s'..\n", file);
|
||||
if (hfile == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
skip("Running these tests is useless without a file present\n");
|
||||
return;
|
||||
}
|
||||
CloseHandle(hfile);
|
||||
|
||||
MultiByteToWideChar(CP_ACP, 0, file, -1, pathW, MAX_PATH);
|
||||
|
||||
/* Parameter validation */
|
||||
ok(pSdbGetPermLayerKeys(NULL, NULL, NULL, 0) == FALSE, "Expected pSdbGetPermLayerKeys to fail\n");
|
||||
ok(pSdbGetPermLayerKeys(pathW, NULL, NULL, 0) == FALSE, "Expected pSdbGetPermLayerKeys to fail\n");
|
||||
ok(pSdbGetPermLayerKeys(pathW, buffer, NULL, 0) == FALSE, "Expected pSdbGetPermLayerKeys to fail\n");
|
||||
ok(pSdbGetPermLayerKeys(pathW, buffer, &dwBufSize, 0) == FALSE, "Expected pSdbGetPermLayerKeys to fail\n");
|
||||
ok(dwBufSize == 0, "Expected dwBufSize to be %u, was %u\n", 0, dwBufSize);
|
||||
|
||||
/* It fails on a nonexisting file */
|
||||
expect_Sdb(tmp, GPLK_USER | GPLK_MACHINE, FALSE, 0xffffffff, "");
|
||||
expect_Sdb(file, GPLK_USER, TRUE, 40, "RUNASADMIN WINXPSP3");
|
||||
GetShortPathNameA(file, tmp, sizeof(tmp));
|
||||
expect_Sdb(tmp, GPLK_USER, TRUE, 40, "RUNASADMIN WINXPSP3");
|
||||
|
||||
if (bMachine)
|
||||
{
|
||||
/* Query from HKLM */
|
||||
expect_Sdb(file, GPLK_MACHINE, TRUE, 36, "WINXPSP3 WINXPSP2");
|
||||
/* Query from both, showing that duplicates are not removed */
|
||||
expect_Sdb(file, GPLK_USER | GPLK_MACHINE, TRUE, 76, "WINXPSP3 WINXPSP2 RUNASADMIN WINXPSP3");
|
||||
|
||||
/* Showing that no validation is done on the value read. */
|
||||
ok(setLayerValue(TRUE, file, "!#!# WINXPSP3 WINXPSP3 !# WINXPSP2 "), "Expected setLayerValue not to fail\n");
|
||||
expect_Sdb(file, GPLK_MACHINE, TRUE, 82, "!#!# WINXPSP3 WINXPSP3 !# WINXPSP2 ");
|
||||
/* Showing that a space is inserted, even if the last char was already a space. */
|
||||
expect_Sdb(file, GPLK_USER | GPLK_MACHINE, TRUE, 122, "!#!# WINXPSP3 WINXPSP3 !# WINXPSP2 RUNASADMIN WINXPSP3");
|
||||
/* Now clear the user key */
|
||||
setLayerValue(FALSE, file, NULL);
|
||||
/* Request both, to show that the last space (from the key) is not cut off. */
|
||||
expect_Sdb(file, GPLK_USER | GPLK_MACHINE, TRUE, 82, "!#!# WINXPSP3 WINXPSP3 !# WINXPSP2 ");
|
||||
setLayerValue(FALSE, file, "RUNASADMIN WINXPSP3");
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("Skipping tests for HKLM, cannot alter the registry\n");
|
||||
}
|
||||
/* Fail from these paths */
|
||||
sprintf(tmp, "\\?\\%s", file);
|
||||
expect_Sdb(tmp, GPLK_USER, FALSE, 0xffffffff, "");
|
||||
sprintf(tmp, "\\??\\%s", file);
|
||||
expect_Sdb(tmp, GPLK_USER, FALSE, 0xffffffff, "");
|
||||
|
||||
ok(setLayerValue(FALSE, file, "!#!# RUNASADMIN RUNASADMIN !# WINXPSP3 "), "Expected setLayerValue not to fail\n");
|
||||
/* There is no validation on information read back. */
|
||||
expect_Sdb(file, GPLK_USER, TRUE, 90, "!#!# RUNASADMIN RUNASADMIN !# WINXPSP3 ");
|
||||
|
||||
|
||||
/* Cleanup */
|
||||
ok(DeleteFileA(file), "DeleteFile failed....\n");
|
||||
setLayerValue(FALSE, file, NULL);
|
||||
setLayerValue(TRUE, file, NULL);
|
||||
}
|
||||
|
||||
|
||||
static BOOL wrapSetPermLayerState(PCWSTR wszPath, PCSTR szLayer, DWORD dwFlags, BOOL bMachine, BOOL bEnable)
|
||||
{
|
||||
WCHAR wszLayer[MAX_LAYER_LENGTH];
|
||||
MultiByteToWideChar(CP_ACP, 0, szLayer, -1, wszLayer, MAX_LAYER_LENGTH);
|
||||
return pSetPermLayerState(wszPath, wszLayer, dwFlags, bMachine, bEnable);
|
||||
}
|
||||
|
||||
static void test_SetPermLayerStateLevel(BOOL bMachine, const char* file)
|
||||
{
|
||||
WCHAR fileW[MAX_PATH+20];
|
||||
WCHAR emptyString[1] = { 0 };
|
||||
DWORD dwFlag;
|
||||
|
||||
MultiByteToWideChar(CP_ACP, 0, file, -1, fileW, MAX_PATH+20);
|
||||
|
||||
/* Test some parameter validation. */
|
||||
ok(pSetPermLayerState(fileW, NULL, 0, bMachine, 0) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
ok(pSetPermLayerState(fileW, NULL, 0, bMachine, 1) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "", 0, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(NULL, NULL, 0, bMachine, 0) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, NULL, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(NULL, NULL, 0, bMachine, 1) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, NULL, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(emptyString, "", 0, bMachine, 0) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, NULL, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(emptyString, "", 0, bMachine, 1) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, NULL, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(emptyString, "TEST", 0, bMachine, 0) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, NULL, NULL);
|
||||
|
||||
if (g_Version <= APPHELP_WIN8)
|
||||
{
|
||||
ok(wrapSetPermLayerState(emptyString, "TEST", 0, bMachine, 1) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, NULL, NULL);
|
||||
}
|
||||
|
||||
|
||||
/* Now, on to the actual tests. */
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", 0, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "test", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "test");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", 0, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST1", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "TEST TEST1", g_Version >= APPHELP_WIN8, "TEST1 TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "TEST TEST1 TEST2", g_Version >= APPHELP_WIN8, "TEST2 TEST1 TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST1", 0, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "TEST TEST2", g_Version >= APPHELP_WIN8, "TEST2 TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", 0, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "TEST2");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 0, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
/* Valid flags until win8: !# */
|
||||
/* Key is empty, now play around with the flags. */
|
||||
for (dwFlag = ((g_Version >= APPHELP_WIN8) ? 6 : 2); dwFlag < 32; ++dwFlag)
|
||||
{
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", (1<<dwFlag), bMachine, 1) == FALSE, "Expected SetPermLayerState to fail on 0x%x\n", (1<<dwFlag));
|
||||
}
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
/* Add layer flags */
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "# TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 2, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# TEST TEST2", g_Version >= APPHELP_WIN8, "!# TEST2 TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# TEST2 TEST", g_Version >= APPHELP_WIN8, "!# TEST TEST2");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST3", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# TEST2 TEST TEST3", g_Version >= APPHELP_WIN8, "!# TEST3 TEST TEST2");
|
||||
|
||||
/* Remove on a flag removes that flag from the start. */
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 2, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "# TEST TEST3", g_Version >= APPHELP_WIN8, "# TEST3 TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "TEST TEST3", g_Version >= APPHELP_WIN8, "TEST3 TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "", LAYER_APPLY_TO_SYSTEM_EXES | 2, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# TEST TEST3", g_Version >= APPHELP_WIN8, "!# TEST3 TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST3", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "! TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", 2, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
/* Try adding multiple layers: */
|
||||
ok(wrapSetPermLayerState(fileW, "TEST TEST2", LAYER_APPLY_TO_SYSTEM_EXES | 2, bMachine, 1) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "TEST2");
|
||||
|
||||
/* Try adding flags in via layer string */
|
||||
ok(wrapSetPermLayerState(fileW, "#", 0, bMachine, 1) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, file, "TEST2");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "!", 0, bMachine, 1) == FALSE, "Expected SetPermLayerState to fail\n");
|
||||
expect_LayerValue(bMachine, file, "TEST2");
|
||||
|
||||
/* Now we prepare the registry with some crap to see how data is validated. */
|
||||
setLayerValue(bMachine, file, "!#!# TEST2 TEST2 !# TEST ");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "TEST1", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# TEST2 TEST2 !# TEST TEST1", g_Version >= APPHELP_WIN8, "!# TEST1 TEST2 TEST2 !# TEST");
|
||||
|
||||
/* Removing a duplicate entry will remove all instances of it */
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 0, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# !# TEST TEST1", g_Version >= APPHELP_WIN8, "!# TEST1 !# TEST");
|
||||
|
||||
/* Adding a flag cleans other flags (from the start) */
|
||||
ok(wrapSetPermLayerState(fileW, "", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# TEST TEST1", g_Version >= APPHELP_WIN8, "!# TEST1 !# TEST");
|
||||
|
||||
if(g_Version < APPHELP_WIN8)
|
||||
{
|
||||
ok(wrapSetPermLayerState(fileW, "$%$%^^", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "!# TEST TEST1 $%$%^^");
|
||||
}
|
||||
|
||||
setLayerValue(bMachine, file, "!#!# TEST2 !# TEST ");
|
||||
ok(wrapSetPermLayerState(fileW, "", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "! TEST2 !# TEST");
|
||||
|
||||
/* Tabs are treated as spaces */
|
||||
setLayerValue(bMachine, file, "!#!# TEST2 \t TEST2 !# \t TEST ");
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# !# TEST TEST2", g_Version >= APPHELP_WIN8, "!# TEST2 !# TEST");
|
||||
|
||||
/* Newlines are left as-is */
|
||||
setLayerValue(bMachine, file, "!#!# TEST2 \n TEST2 !# \r\n TEST ");
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 0, bMachine, 1) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue2(bMachine, file, "!# \n !# \r\n TEST TEST2", g_Version >= APPHELP_WIN8, "!# TEST2 \n !# \r\n TEST");
|
||||
|
||||
/* Whitespace and duplicate flags are eaten from the start */
|
||||
setLayerValue(bMachine, file, " !#!# TEST2 \t TEST2 !# \t TEST ");
|
||||
ok(wrapSetPermLayerState(fileW, "", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "! TEST2 TEST2 !# TEST");
|
||||
|
||||
setLayerValue(bMachine, file, "!# !# TEST2 !# TEST ");
|
||||
ok(wrapSetPermLayerState(fileW, "", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "! TEST2 !# TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "! TEST2 !# TEST");
|
||||
|
||||
ok(wrapSetPermLayerState(fileW, "", 2, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "TEST2 !# TEST");
|
||||
|
||||
/* First flags are cleaned, then a layer is removed. */
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 2, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "!# TEST");
|
||||
|
||||
/* Nothing is changed, still it succeeds. */
|
||||
ok(wrapSetPermLayerState(fileW, "TEST2", 2, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, "# TEST");
|
||||
|
||||
/* And remove the last bits. */
|
||||
ok(wrapSetPermLayerState(fileW, "TEST", LAYER_APPLY_TO_SYSTEM_EXES, bMachine, 0) == TRUE, "Expected SetPermLayerState to succeed\n");
|
||||
expect_LayerValue(bMachine, file, NULL);
|
||||
}
|
||||
|
||||
static void test_SetPermLayer(void)
|
||||
{
|
||||
char file[MAX_PATH + 20], tmp[MAX_PATH + 20];
|
||||
HANDLE hfile;
|
||||
|
||||
GetTempPathA(MAX_PATH, tmp);
|
||||
GetLongPathNameA(tmp, file, sizeof(file));
|
||||
PathAppendA(file, "test_file.exe");
|
||||
|
||||
hfile = CreateFileA(file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
||||
ok(hfile != INVALID_HANDLE_VALUE, "CreateFile failed for '%s'\n", file);
|
||||
if (hfile == INVALID_HANDLE_VALUE)
|
||||
{
|
||||
skip("Running these tests is useless without a file present\n");
|
||||
return;
|
||||
}
|
||||
CloseHandle(hfile);
|
||||
|
||||
if (setLayerValue(FALSE, file, NULL))
|
||||
{
|
||||
test_SdbSetPermLayerKeysLevel(FALSE, file);
|
||||
test_SetPermLayerStateLevel(FALSE, file);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("Skipping SetPermLayerStateLevel tests for User, because I cannot prepare the environment\n");
|
||||
}
|
||||
if (setLayerValue(TRUE, file, NULL))
|
||||
{
|
||||
test_SdbSetPermLayerKeysLevel(TRUE, file);
|
||||
test_SetPermLayerStateLevel(TRUE, file);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("Skipping SetPermLayerStateLevel tests for Machine (HKLM), because I cannot prepare the environment\n");
|
||||
}
|
||||
ok(DeleteFileA(file), "DeleteFile failed....\n");
|
||||
}
|
||||
|
||||
static BOOL create_file(LPCSTR dir, LPCSTR name, int filler, size_t size)
|
||||
{
|
||||
char target[MAX_PATH], *tmp;
|
||||
HANDLE file;
|
||||
PathCombineA(target, dir, name);
|
||||
|
||||
tmp = malloc(size);
|
||||
memset(tmp, filler, size);
|
||||
|
||||
file = CreateFileA(target, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
||||
if(file == INVALID_HANDLE_VALUE)
|
||||
return FALSE;
|
||||
|
||||
WriteFile(file, tmp, size, &size, NULL);
|
||||
CloseHandle(file);
|
||||
free(tmp);
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
static BOOL delete_file(LPCSTR dir, LPCSTR name)
|
||||
{
|
||||
char target[MAX_PATH];
|
||||
PathCombineA(target, dir, name);
|
||||
return DeleteFileA(target);
|
||||
}
|
||||
|
||||
static char g_FakeDrive = 0;
|
||||
|
||||
UINT (WINAPI *pGetDriveTypeW)(LPCWSTR target) = NULL;
|
||||
UINT WINAPI mGetDriveTypeW(LPCWSTR target)
|
||||
{
|
||||
UINT uRet = pGetDriveTypeW(target);
|
||||
if(g_FakeDrive && target && (char)*target == g_FakeDrive)
|
||||
return DRIVE_CDROM;
|
||||
return uRet;
|
||||
}
|
||||
|
||||
|
||||
static PIMAGE_IMPORT_DESCRIPTOR FindImportDescriptor(PBYTE DllBase, PCSTR DllName)
|
||||
{
|
||||
ULONG Size;
|
||||
PIMAGE_IMPORT_DESCRIPTOR ImportDescriptor = RtlImageDirectoryEntryToData((HMODULE)DllBase, TRUE, IMAGE_DIRECTORY_ENTRY_IMPORT, &Size);
|
||||
while (ImportDescriptor->Name && ImportDescriptor->OriginalFirstThunk)
|
||||
{
|
||||
PCHAR Name = (PCHAR)(DllBase + ImportDescriptor->Name);
|
||||
if (!lstrcmpiA(Name, DllName))
|
||||
{
|
||||
return ImportDescriptor;
|
||||
}
|
||||
ImportDescriptor++;
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static BOOL RedirectIat(PCSTR TargetDllName, PCSTR DllName, PCSTR FunctionName, ULONG_PTR NewFunction, ULONG_PTR* OriginalFunction)
|
||||
{
|
||||
PBYTE DllBase = (PBYTE)GetModuleHandleA(TargetDllName);
|
||||
if (DllBase)
|
||||
{
|
||||
PIMAGE_IMPORT_DESCRIPTOR ImportDescriptor = FindImportDescriptor(DllBase, DllName);
|
||||
if (ImportDescriptor)
|
||||
{
|
||||
// On loaded images, OriginalFirstThunk points to the name / ordinal of the function
|
||||
PIMAGE_THUNK_DATA OriginalThunk = (PIMAGE_THUNK_DATA)(DllBase + ImportDescriptor->OriginalFirstThunk);
|
||||
// FirstThunk points to the resolved address.
|
||||
PIMAGE_THUNK_DATA FirstThunk = (PIMAGE_THUNK_DATA)(DllBase + ImportDescriptor->FirstThunk);
|
||||
while (OriginalThunk->u1.AddressOfData && FirstThunk->u1.Function)
|
||||
{
|
||||
if (!IMAGE_SNAP_BY_ORDINAL32(OriginalThunk->u1.AddressOfData))
|
||||
{
|
||||
PIMAGE_IMPORT_BY_NAME ImportName = (PIMAGE_IMPORT_BY_NAME)(DllBase + OriginalThunk->u1.AddressOfData);
|
||||
if (!lstrcmpiA((PCSTR)ImportName->Name, FunctionName))
|
||||
{
|
||||
DWORD dwOld;
|
||||
VirtualProtect(&FirstThunk->u1.Function, sizeof(ULONG_PTR), PAGE_EXECUTE_READWRITE, &dwOld);
|
||||
*OriginalFunction = FirstThunk->u1.Function;
|
||||
FirstThunk->u1.Function = NewFunction;
|
||||
VirtualProtect(&FirstThunk->u1.Function, sizeof(ULONG_PTR), dwOld, &dwOld);
|
||||
return TRUE;
|
||||
}
|
||||
}
|
||||
OriginalThunk++;
|
||||
FirstThunk++;
|
||||
}
|
||||
skip("Unable to find the Import '%s' from '%s' in %s'\n", FunctionName, DllName, TargetDllName);
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("Unable to find the ImportDescriptor for '%s' in '%s'\n", DllName, TargetDllName);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
skip("Unable to find the loaded module '%s'\n", TargetDllName);
|
||||
}
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
static BOOL RestoreIat(PCSTR target, PCSTR DllName, PCSTR FunctionName, ULONG_PTR OriginalFunction)
|
||||
{
|
||||
ULONG_PTR old = 0;
|
||||
return RedirectIat(target, DllName, FunctionName, OriginalFunction, &old);
|
||||
}
|
||||
|
||||
static BOOL wrapSdbSetPermLayerKeys2(LPCSTR dir, LPCSTR name, PCSTR szLayers, BOOL bMachine)
|
||||
{
|
||||
char szPath[MAX_PATH];
|
||||
WCHAR wszPath[MAX_PATH], wszLayers[MAX_LAYER_LENGTH];
|
||||
PathCombineA(szPath, dir, name);
|
||||
MultiByteToWideChar(CP_ACP, 0, szLayers, -1, wszLayers, MAX_LAYER_LENGTH);
|
||||
MultiByteToWideChar(CP_ACP, 0, szPath, -1, wszPath, MAX_PATH);
|
||||
return pSdbSetPermLayerKeys(wszPath, wszLayers, bMachine);
|
||||
}
|
||||
|
||||
|
||||
BOOL expect_files(const char* dir, int num, ...)
|
||||
{
|
||||
char finddir[MAX_PATH + 20];
|
||||
va_list args;
|
||||
WIN32_FIND_DATAA find = { 0 };
|
||||
HANDLE hFind;
|
||||
int cmp = 0;
|
||||
|
||||
va_start(args, num);
|
||||
|
||||
PathCombineA(finddir, dir, "*");
|
||||
hFind = FindFirstFileA(finddir, &find);
|
||||
if (hFind != INVALID_HANDLE_VALUE)
|
||||
{
|
||||
const char* file;
|
||||
do
|
||||
{
|
||||
if (!(find.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
||||
{
|
||||
if (--num < 0)
|
||||
break;
|
||||
file = va_arg(args, const char*);
|
||||
cmp = strcmp(file, find.cFileName);
|
||||
}
|
||||
} while (cmp == 0 && FindNextFileA(hFind, &find));
|
||||
FindClose(hFind);
|
||||
}
|
||||
va_end(args);
|
||||
return cmp == 0 && num == 0;
|
||||
}
|
||||
|
||||
|
||||
static void test_Sign_Media(void)
|
||||
{
|
||||
char workdir[MAX_PATH], subdir[MAX_PATH], drive[5] = "Z:";
|
||||
BOOL ret;
|
||||
|
||||
DWORD logical_drives = GetLogicalDrives();
|
||||
g_FakeDrive = 0;
|
||||
for (drive[0] = 'D'; drive[0] <= 'Z'; drive[0]++)
|
||||
{
|
||||
DWORD idx = 1 << (drive[0] - 'D' + 3);
|
||||
if (!(logical_drives & idx))
|
||||
{
|
||||
g_FakeDrive = drive[0];
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!g_FakeDrive)
|
||||
{
|
||||
skip("Unable to find a free drive\n");
|
||||
return;
|
||||
}
|
||||
|
||||
ret = GetTempPathA(MAX_PATH, workdir);
|
||||
ok(ret, "GetTempPathA error: %d\n", GetLastError());
|
||||
PathAppendA(workdir, "apphelp_test");
|
||||
|
||||
ret = CreateDirectoryA(workdir, NULL);
|
||||
ok(ret, "CreateDirectoryA error: %d\n", GetLastError());
|
||||
|
||||
PathCombineA(subdir, workdir, "sub");
|
||||
ret = CreateDirectoryA(subdir, NULL);
|
||||
ok(ret, "CreateDirectoryA error: %d\n", GetLastError());
|
||||
|
||||
ret = DefineDosDeviceA(DDD_NO_BROADCAST_SYSTEM, drive, workdir);
|
||||
ok(ret, "DefineDosDeviceA error: %d\n", GetLastError());
|
||||
if(ret)
|
||||
{
|
||||
ret = RedirectIat("apphelp.dll", "kernel32.dll", "GetDriveTypeW", (ULONG_PTR)mGetDriveTypeW, (ULONG_PTR*)&pGetDriveTypeW);
|
||||
if (g_Version < APPHELP_WIN8)
|
||||
ok(ret, "Expected redirect_iat to succeed\n");
|
||||
if(ret)
|
||||
{
|
||||
ok(create_file(workdir, "test.exe", 'a', 4), "create_file error: %d\n", GetLastError());
|
||||
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "test.exe", "TEST", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
/* 4 */
|
||||
/* test.exe */
|
||||
expect_LayerValue(0, "SIGN.MEDIA=4 test.exe", "TEST");
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "test.exe", "", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
|
||||
ok(create_file(workdir, "test.txt", 'a', 1), "create_file error: %d\n", GetLastError());
|
||||
|
||||
if (!expect_files(workdir, 2, "test.exe", "test.txt"))
|
||||
{
|
||||
skip("Skipping test, files are not returned in the expected order by the FS\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "test.exe", "TEST", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
/* (4 << 1) ^ 1 */
|
||||
/* test.exe test.txt */
|
||||
expect_LayerValue(0, "SIGN.MEDIA=9 test.exe", "TEST");
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "test.exe", "", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
}
|
||||
|
||||
ok(create_file(workdir, "test.zz", 'a', 0x1000), "create_file error: %d\n", GetLastError());
|
||||
|
||||
if (!expect_files(workdir, 3, "test.exe", "test.txt", "test.zz"))
|
||||
{
|
||||
skip("Skipping test, files are not returned in the expected order by the FS\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "test.exe", "TEST", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
/* (((4 << 1) ^ 1) << 1) ^ 0x1000 */
|
||||
/* test.exe test.txt test.zz */
|
||||
expect_LayerValue(0, "SIGN.MEDIA=1012 test.exe", "TEST");
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "test.exe", "", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
}
|
||||
|
||||
ok(create_file(subdir, "test.exe", 'a', 0x10203), "create_file error: %d\n", GetLastError());
|
||||
|
||||
if (!expect_files(subdir, 1, "test.exe"))
|
||||
{
|
||||
skip("Skipping test, files are not returned in the expected order by the FS\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "sub\\test.exe", "TEST", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
/* 0x10203 */
|
||||
/* test.exe */
|
||||
expect_LayerValue(0, "SIGN.MEDIA=10203 sub\\test.exe", "TEST");
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "sub\\test.exe", "", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
}
|
||||
|
||||
ok(create_file(subdir, "test.bbb", 'a', 0), "create_file error: %d\n", GetLastError());
|
||||
|
||||
if (!expect_files(subdir, 2, "test.bbb", "test.exe"))
|
||||
{
|
||||
skip("Skipping test, files are not returned in the expected order by the FS\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "sub\\test.exe", "TEST", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
/* 0x10203 */
|
||||
/* test.exe */
|
||||
expect_LayerValue(0, "SIGN.MEDIA=10203 sub\\test.exe", "TEST");
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "sub\\test.exe", "", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
}
|
||||
|
||||
ok(create_file(subdir, "TEST.txt", 'a', 0x30201), "create_file error: %d\n", GetLastError());
|
||||
|
||||
if (!expect_files(subdir, 3, "test.bbb", "test.exe", "TEST.txt"))
|
||||
{
|
||||
skip("Skipping test, files are not returned in the expected order by the FS\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "sub\\test.exe", "TEST", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
/* (0x10203 << 1) ^ 0x30201 */
|
||||
/* test.exe TEST.txt */
|
||||
expect_LayerValue(0, "SIGN.MEDIA=10607 sub\\test.exe", "TEST");
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "sub\\test.exe", "", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
}
|
||||
|
||||
ok(create_file(subdir, "TEST.aaa", 'a', 0x3a2a1), "create_file error: %d\n", GetLastError());
|
||||
|
||||
if (!expect_files(subdir, 4, "TEST.aaa", "test.bbb", "test.exe", "TEST.txt"))
|
||||
{
|
||||
skip("Skipping test, files are not returned in the expected order by the FS\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "sub\\test.exe", "TEST", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
/* (((0x3a2a1 << 1) ^ 0x10203) << 1) ^ 0x30201 */
|
||||
/* TEST.aaa test.exe TEST.txt */
|
||||
expect_LayerValue(0, "SIGN.MEDIA=F8C83 sub\\test.exe", "TEST");
|
||||
ok(wrapSdbSetPermLayerKeys2(drive, "sub\\test.exe", "", 0), "Expected wrapSdbSetPermLayerKeys2 to succeed\n");
|
||||
}
|
||||
|
||||
ret = RestoreIat("apphelp.dll", "kernel32.dll", "GetDriveTypeW", (ULONG_PTR)pGetDriveTypeW);
|
||||
ok(ret, "Expected restore_iat to succeed\n");
|
||||
|
||||
ok(delete_file(subdir, "test.bbb"), "delete_file error: %d\n", GetLastError());
|
||||
ok(delete_file(subdir, "TEST.aaa"), "delete_file error: %d\n", GetLastError());
|
||||
ok(delete_file(subdir, "TEST.txt"), "delete_file error: %d\n", GetLastError());
|
||||
ok(delete_file(subdir, "test.exe"), "delete_file error: %d\n", GetLastError());
|
||||
ok(delete_file(workdir, "test.zz"), "delete_file error: %d\n", GetLastError());
|
||||
ok(delete_file(workdir, "test.txt"), "delete_file error: %d\n", GetLastError());
|
||||
ok(delete_file(workdir, "test.exe"), "delete_file error: %d\n", GetLastError());
|
||||
}
|
||||
ret = DefineDosDeviceA(DDD_REMOVE_DEFINITION | DDD_NO_BROADCAST_SYSTEM, drive, NULL);
|
||||
ok(ret, "DefineDosDeviceA error: %d\n", GetLastError());
|
||||
}
|
||||
ret = RemoveDirectoryA(subdir);
|
||||
ok(ret, "RemoveDirectoryA error: %d\n", GetLastError());
|
||||
ret = RemoveDirectoryA(workdir);
|
||||
ok(ret, "RemoveDirectoryA error: %d\n", GetLastError());
|
||||
}
|
||||
|
||||
|
||||
START_TEST(layerapi)
|
||||
{
|
||||
RTL_OSVERSIONINFOEXW rtlinfo;
|
||||
/*SetEnvironmentVariable("SHIM_DEBUG_LEVEL", "4");*/
|
||||
hdll = LoadLibraryA("apphelp.dll");
|
||||
pAllowPermLayer = (void *)GetProcAddress(hdll, "AllowPermLayer");
|
||||
pSdbSetPermLayerKeys = (void *)GetProcAddress(hdll, "SdbSetPermLayerKeys");
|
||||
pSdbGetPermLayerKeys = (void *)GetProcAddress(hdll, "SdbGetPermLayerKeys");
|
||||
pSetPermLayerState = (void *)GetProcAddress(hdll, "SetPermLayerState");
|
||||
rtlinfo.dwOSVersionInfoSize = sizeof(rtlinfo);
|
||||
#ifdef __REACTOS__
|
||||
RtlGetVersion((PRTL_OSVERSIONINFOW)&rtlinfo);
|
||||
#else
|
||||
RtlGetVersion(&rtlinfo);
|
||||
#endif
|
||||
g_Version = (rtlinfo.dwMajorVersion << 8) | rtlinfo.dwMinorVersion;
|
||||
|
||||
if (!pAllowPermLayer)
|
||||
{
|
||||
skip("Skipping tests with AllowPermLayer, function not found\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
test_AllowPermLayer();
|
||||
}
|
||||
|
||||
if (!pSdbSetPermLayerKeys)
|
||||
{
|
||||
skip("Skipping tests with SdbSetPermLayerKeys, function not found\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
if (!pSdbGetPermLayerKeys)
|
||||
{
|
||||
skip("Skipping tests with SdbGetPermLayerKeys, function not found\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
test_SdbGetPermLayerKeys();
|
||||
}
|
||||
|
||||
if (!pSetPermLayerState)
|
||||
{
|
||||
skip("Skipping tests with SetPermLayerState, function not found\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
test_SetPermLayer();
|
||||
test_Sign_Media();
|
||||
}
|
||||
}
|
||||
}
|
12
rostests/apitests/apphelp/testlist.c
Normal file
12
rostests/apitests/apphelp/testlist.c
Normal file
|
@ -0,0 +1,12 @@
|
|||
/* Automatically generated file; DO NOT EDIT!! */
|
||||
|
||||
#define STANDALONE
|
||||
#include <wine/test.h>
|
||||
|
||||
extern void func_layerapi(void);
|
||||
|
||||
const struct test winetest_testlist[] =
|
||||
{
|
||||
{ "layerapi", func_layerapi },
|
||||
{ 0, 0 }
|
||||
};
|
Loading…
Reference in a new issue