Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-11-02 04:37:32 +00:00
Code Issues Packages Projects Releases Wiki Activity
reactos/modules/rostests/apitests/apphelp/env.c

1340 lines
51 KiB
C
Raw Blame History

/*
* PROJECT: apphelp_apitest
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
* PURPOSE: Tests showing shim artifacts in the environment
* COPYRIGHT: Copyright 2016,2017 Mark Jansen (mark.jansen@reactos.org)
*/
#include <ntstatus.h>
#define WIN32_NO_STATUS
#include <windows.h>
#include <shlwapi.h>
#include <winnt.h>
#include <userenv.h>
#ifdef __REACTOS__
#include <ntndk.h>
#else
#include <winternl.h>
#endif
#include <winerror.h>
#include <stdio.h>
#include "wine/test.h"
#include <pseh/pseh2.h>
#include "apphelp_apitest.h"
typedef void* HSDB;
typedef void* PDB;
typedef DWORD TAGREF;
typedef WORD TAG;
static HMODULE hdll;
BOOL(WINAPI *pSdbGetMatchingExe)(HSDB hsdb, LPCWSTR szPath, LPCWSTR szModuleName, LPCWSTR pszEnvironment, DWORD dwFlags, PSDBQUERYRESULT_VISTA pQueryResult);
HSDB (WINAPI *pSdbInitDatabase)(DWORD dwFlags, LPCWSTR pszDatabasePath);
void (WINAPI *pSdbReleaseDatabase)(HSDB hsdb);
BOOL (WINAPI *pSdbTagRefToTagID)(HSDB hsdb, TAGREF trWhich, PDB* ppdb, TAGID* ptiWhich);
TAG (WINAPI *pSdbGetTagFromTagID)(PDB pdb, TAGID tiWhich);
TAGREF (WINAPI *pSdbGetLayerTagRef)(HSDB hsdb, LPCWSTR layerName);
/* TODO: Investigate ApphelpCheckRunApp, for some reason there is not AppCompatData generated... */
BOOL (WINAPI *pApphelpCheckRunAppEx_w7)(HANDLE FileHandle, PVOID Unk1, PVOID Unk2, PWCHAR ApplicationName, PVOID Environment, USHORT ExeType, PULONG Reason,
PVOID* SdbQueryAppCompatData, PULONG SdbQueryAppCompatDataSize, PVOID* SxsData, PULONG SxsDataSize,
PULONG FusionFlags, PULONG64 SomeFlag1, PULONG SomeFlag2);
BOOL (WINAPI *pApphelpCheckRunAppEx_w10)(HANDLE FileHandle, PVOID Unk1, PVOID Unk2, PWCHAR ApplicationName, PVOID Environment, PVOID Unk3, USHORT ExeType, PULONG Reason,
PVOID* SdbQueryAppCompatData, PULONG SdbQueryAppCompatDataSize, PVOID* SxsData, PULONG SxsDataSize,
PULONG FusionFlags, PULONG64 SomeFlag1, PULONG SomeFlag2);
BOOL (WINAPI *pSdbPackAppCompatData)(HSDB hsdb, PSDBQUERYRESULT_VISTA pQueryResult, PVOID* ppData, DWORD *dwSize);
BOOL (WINAPI *pSdbUnpackAppCompatData)(HSDB hsdb, LPCWSTR pszImageName, PVOID pData, PSDBQUERYRESULT_VISTA pQueryResult);
DWORD (WINAPI *pSdbGetAppCompatDataSize)(PVOID pData);
static HSDB g_LayerDB;
static DWORD g_ShimDataSize;
static DWORD g_ModuleVersion;
static const SDBQUERYRESULT_VISTA empty_result = { { 0 } };
static const SDBQUERYRESULT_VISTA almost_empty = { { 0 }, { 0 }, { 0 }, 0, 0, 0, 0, { 0 }, SHIMREG_DISABLE_LAYER, 0 };
#define SHIMDATA_MAGIC 0xAC0DEDAB
#define MAX_LAYER_LENGTH 256
typedef struct ShimData_Win2k3
{
WCHAR szModule[34];
DWORD dwSize;
DWORD dwMagic;
TAGREF atrExes[SDB_MAX_EXES_2k3];
TAGREF atrLayers[SDB_MAX_LAYERS];
DWORD dwUnk0;
DWORD dwUnk1;
DWORD dwCustomSDBMap;
GUID rgGuidDB[SDB_MAX_SDBS];
} ShimData_Win2k3;
typedef struct ShimData_Win7
{
WCHAR szModule[260];
DWORD dwSize;
DWORD dwMagic;
SDBQUERYRESULT_VISTA Query;
WCHAR szLayer[MAX_LAYER_LENGTH];
DWORD unknown; // 0x14c
} ShimData_Win7;
typedef struct ShimData_Win10_v1
{
WCHAR szModule[260];
DWORD dwSize;
DWORD dwMagic;
DWORD unk1;
SDBQUERYRESULT_VISTA Query;
WCHAR szLayer[MAX_LAYER_LENGTH];
char padding1[0x200];
char padding2[0x404]; // Contains some data at the start
DWORD unk2;
DWORD unk3;
WCHAR processname[MAX_PATH];
WCHAR szLayerEnv[MAX_LAYER_LENGTH];
WCHAR unk4[MAX_LAYER_LENGTH];
char padding4[120];
} ShimData_Win10_v1;
typedef struct ShimData_Win10_v2
{
DWORD dwSize;
DWORD dwMagic;
DWORD unk1;
DWORD unk2;
SDBQUERYRESULT_VISTA Query;
WCHAR szLayer[MAX_LAYER_LENGTH];
char padding1[0x200];
char padding2[0x2ae + 0x54 + 0x2a + 0x16 + 0x16];
DWORD unk3;
DWORD unk4;
WCHAR processname[MAX_PATH-2];
WCHAR szLayerEnv[MAX_LAYER_LENGTH];
WCHAR unk5[MAX_LAYER_LENGTH];
char padding4[76];
} ShimData_Win10_v2;
typedef struct ShimData_QueryOffset
{
DWORD dwSize_10_v2;
DWORD dwMagic_10_v2;
char spacing1[60];
DWORD dwSize_2k3;
DWORD dwMagic_2k3;
char spacing2[444];
DWORD dwSize_7_10;
DWORD dwMagic_7_10;
} ShimData_QueryOffset;
C_ASSERT(sizeof(ShimData_Win2k3) == 392);
C_ASSERT(sizeof(ShimData_Win7) == 1500);
C_ASSERT(sizeof(ShimData_Win10_v1) == 4712);
C_ASSERT(sizeof(ShimData_Win10_v2) == 3976);
C_ASSERT(offsetof(ShimData_Win10_v2, dwSize) == 0);
C_ASSERT(offsetof(ShimData_Win2k3, dwSize) == 68);
C_ASSERT(offsetof(ShimData_Win7, dwSize) == 520);
C_ASSERT(offsetof(ShimData_Win10_v1, dwSize) == 520);
C_ASSERT(offsetof(ShimData_Win10_v2, dwMagic) == 4);
C_ASSERT(offsetof(ShimData_Win2k3, dwMagic) == 72);
C_ASSERT(offsetof(ShimData_Win7, dwMagic) == 524);
C_ASSERT(offsetof(ShimData_Win10_v1, dwMagic) == 524);
C_ASSERT(offsetof(ShimData_QueryOffset, dwSize_10_v2) == 0);
C_ASSERT(offsetof(ShimData_QueryOffset, dwSize_2k3) == 68);
C_ASSERT(offsetof(ShimData_QueryOffset, dwSize_7_10) == 520);
C_ASSERT(offsetof(ShimData_QueryOffset, dwMagic_10_v2) == 4);
C_ASSERT(offsetof(ShimData_QueryOffset, dwMagic_2k3) == 72);
C_ASSERT(offsetof(ShimData_QueryOffset, dwMagic_7_10) == 524);
#define SDB_DATABASE_MAIN_SHIM 0x80030000
#define SDBGMEF_IGNORE_ENVIRONMENT 0x1
typedef struct test_RemoteShimInfo
{
ULARGE_INTEGER AppCompatFlags;
ULARGE_INTEGER AppCompatFlagsUser;
PVOID pShimData;
DWORD ShimDataSize;
PVOID AppCompatInfo;
} test_RemoteShimInfo;
static BOOL readproc(HANDLE proc, LPVOID address, PVOID target, DWORD size)
{
SIZE_T dwRead;
if (ReadProcessMemory(proc, address, target, size, &dwRead))
{
ok(dwRead == size, "Expected to read %u bytes, got %lu\n", size, dwRead);
return dwRead == size;
}
ok(0, "RPM failed with %u\n", GetLastError());
return FALSE;
}
static BOOL get_shiminfo(HANDLE proc, test_RemoteShimInfo* info)
{
PROCESS_BASIC_INFORMATION pbi = { 0 };
ULONG sizeOut = 0;
NTSTATUS status = NtQueryInformationProcess(proc, ProcessBasicInformation, &pbi, sizeof(pbi), &sizeOut);
ok(NT_SUCCESS(status), "Expected NtQI to succeed, but failed with: %x\n", status);
memset(info, 0, sizeof(*info));
if (NT_SUCCESS(status))
{
PEB peb = { 0 };
if (readproc(proc, pbi.PebBaseAddress, &peb, sizeof(peb)))
{
MEMORY_BASIC_INFORMATION mbi = { 0 };
SIZE_T dwRead;
info->AppCompatFlags = peb.AppCompatFlags;
info->AppCompatFlagsUser = peb.AppCompatFlagsUser;
info->AppCompatInfo = peb.AppCompatInfo;
if (peb.pShimData == NULL)
return TRUE;
dwRead = VirtualQueryEx(proc, (LPCVOID)peb.pShimData, &mbi, sizeof(mbi));
ok(dwRead == sizeof(mbi), "Expected VQE to return %u, got %lu\n", sizeof(mbi), dwRead);
if (dwRead == sizeof(mbi) || peb.pShimData == NULL)
{
info->ShimDataSize = mbi.RegionSize;
info->pShimData = malloc(mbi.RegionSize);
if (readproc(proc, peb.pShimData, info->pShimData, mbi.RegionSize))
return TRUE;
free(info->pShimData);
info->pShimData = NULL;
}
}
}
return FALSE;
}
static HANDLE create_proc(BOOL suspended)
{
static char proc_name[MAX_PATH] = { 0 };
STARTUPINFOA si = {sizeof(si)};
PROCESS_INFORMATION pi;
BOOL res;
if (!proc_name[0])
{
GetModuleFileNameA(NULL, proc_name, MAX_PATH);
}
res = CreateProcessA(NULL, proc_name, NULL, NULL, TRUE, suspended ? CREATE_SUSPENDED : 0, NULL, NULL, &si, &pi);
if (!res)
return NULL;
CloseHandle(pi.hThread);
return pi.hProcess;
}
static void create_environ(const char* layers[], size_t num)
{
char buf[256] = { 0 };
size_t n;
for (n = 0; n < num; ++n)
{
if (n)
strcat(buf, " ");
strcat(buf, layers[n]);
}
SetEnvironmentVariableA("__COMPAT_LAYER", buf);
}
static void ValidateShim(TAGREF trLayer, const char* name)
{
HSDB hsdb = pSdbInitDatabase(SDB_DATABASE_MAIN_SHIM, NULL);
ok(hsdb != NULL, "Expected a valid handle\n");
if (hsdb)
{
PDB pdb = NULL;
TAGID tagid = 0xdeadbeef;
WCHAR nameW[256] = { 0 };
BOOL ret;
mbstowcs(nameW, name, strlen(name));
ret = pSdbTagRefToTagID(hsdb, trLayer, &pdb, &tagid);
ok(ret == TRUE, "Expected pSdbTagRefToTagID to succeed\n");
if (ret)
{
TAG tag;
ok(pdb != NULL, "Expected pdb to be a valid pointer\n");
ok(tagid != 0 && tagid != 0xdeadbeef, "Expected tagid to be a valid tag id, was: 0x%x\n", tagid);
tag = pSdbGetTagFromTagID(pdb, tagid);
ok(tag == 0x700b, "Expected tag to be 0x700b, was 0x%x\n", (DWORD)tag);
}
pSdbReleaseDatabase(hsdb);
}
}
static void Validate_ShimData_Win2k3(PVOID data, size_t count, const char* layers[])
{
//size_t n;
ShimData_Win2k3* pShimData = (ShimData_Win2k3*)data;
ok(!lstrcmpW(pShimData->szModule, L"ShimEng.dll"), "Expected pShimData->Module to be %s, was %s\n", wine_dbgstr_w(L"ShimEng.dll"), wine_dbgstr_w(pShimData->szModule));
ok(pShimData->dwMagic == SHIMDATA_MAGIC, "Expected pShimData->dwMagic to be 0x%x, was 0x%x\n", SHIMDATA_MAGIC, pShimData->dwMagic);
ok(pShimData->dwSize == sizeof(ShimData_Win2k3), "Expected pShimData->dwSize to be %u, was %u\n", sizeof(ShimData_Win2k3), pShimData->dwSize);
ok(pShimData->dwCustomSDBMap == 1, "Expected pShimData->dwCustomSDBMap to be 1, was %u\n", pShimData->dwCustomSDBMap);
}
static void Validate_ShimData_Win7(PVOID data, WCHAR szApphelp[256], size_t count, const char* layers[])
{
size_t n;
ShimData_Win7* pShimData = (ShimData_Win7*)data;
ok(!lstrcmpW(pShimData->szModule, szApphelp), "Expected pShimData->Module to be %s, was %s\n",
wine_dbgstr_w(szApphelp), wine_dbgstr_w(pShimData->szModule));
ok(pShimData->dwMagic == SHIMDATA_MAGIC, "Expected pShimData->dwMagic to be 0x%x, was 0x%x\n",
SHIMDATA_MAGIC, pShimData->dwMagic);
ok(pShimData->dwSize == sizeof(ShimData_Win7), "Expected pShimData->dwSize to be %u, was %u\n",
sizeof(ShimData_Win7), pShimData->dwSize);
if (pShimData->Query.dwLayerCount != min(count, SDB_MAX_LAYERS))
{
char buf[250] = {0};
GetEnvironmentVariableA("__COMPAT_LAYER", buf, _countof(buf));
trace("At test: %s\n", buf);
}
ok(pShimData->Query.dwLayerCount == min(count, SDB_MAX_LAYERS),
"Expected LayerCount to be %u, was %u\n", min(count, SDB_MAX_LAYERS), pShimData->Query.dwLayerCount);
for (n = 0; n < SDB_MAX_LAYERS; ++n)
{
if (n < count)
{
ok(pShimData->Query.atrLayers[n] != 0, "Expected to find a valid layer in index %u / %u\n", n, count);
ValidateShim(pShimData->Query.atrLayers[n], layers[n]);
}
else
ok(pShimData->Query.atrLayers[n] == 0, "Expected to find an empty layer in index %u / %u\n", n, count);
}
ok(pShimData->unknown == 0x14c, "Expected pShimData->unknown to be 0x14c, was 0x%x\n", pShimData->unknown);
}
static void Validate_ShimData_Win10_v2(PVOID data, WCHAR szApphelp[256], size_t count, const char* layers[])
{
size_t n;
ShimData_Win10_v2* pShimData = (ShimData_Win10_v2*)data;
if (pShimData->dwMagic != SHIMDATA_MAGIC)
{
skip("Yet another unknown shimdata variant...\n");
return;
}
ok(pShimData->dwSize == sizeof(ShimData_Win10_v2), "Expected pShimData->dwSize to be %u, was %u\n",
sizeof(ShimData_Win10_v2), pShimData->dwSize);
if (pShimData->Query.dwLayerCount != min(count, SDB_MAX_LAYERS))
{
char buf[250] = {0};
GetEnvironmentVariableA("__COMPAT_LAYER", buf, _countof(buf));
trace("At test: %s\n", buf);
}
ok(pShimData->Query.dwLayerCount == min(count, SDB_MAX_LAYERS),
"Expected LayerCount to be %u, was %u\n", min(count, SDB_MAX_LAYERS), pShimData->Query.dwLayerCount);
for (n = 0; n < SDB_MAX_LAYERS; ++n)
{
if (n < count)
{
ok(pShimData->Query.atrLayers[n] != 0, "Expected to find a valid layer in index %u / %u\n", n, count);
ValidateShim(pShimData->Query.atrLayers[n], layers[n]);
}
else
ok(pShimData->Query.atrLayers[n] == 0, "Expected to find an empty layer in index %u / %u\n", n, count);
}
}
static void Validate_ShimData_Win10(PVOID data, WCHAR szApphelp[256], size_t count, const char* layers[])
{
size_t n;
ShimData_Win10_v1* pShimData = (ShimData_Win10_v1*)data;
if (pShimData->dwMagic != SHIMDATA_MAGIC)
{
Validate_ShimData_Win10_v2(data, szApphelp, count, layers);
return;
}
ok(!lstrcmpiW(pShimData->szModule, szApphelp), "Expected pShimData->Module to be %s, was %s\n",
wine_dbgstr_w(szApphelp), wine_dbgstr_w(pShimData->szModule));
ok(pShimData->dwSize == sizeof(ShimData_Win10_v1), "Expected pShimData->dwSize to be %u, was %u\n",
sizeof(ShimData_Win10_v1), pShimData->dwSize);
if (pShimData->Query.dwLayerCount != min(count, SDB_MAX_LAYERS))
{
char buf[250] = {0};
GetEnvironmentVariableA("__COMPAT_LAYER", buf, _countof(buf));
trace("At test: %s\n", buf);
}
ok(pShimData->Query.dwLayerCount == min(count, SDB_MAX_LAYERS),
"Expected LayerCount to be %u, was %u\n", min(count, SDB_MAX_LAYERS), pShimData->Query.dwLayerCount);
for (n = 0; n < SDB_MAX_LAYERS; ++n)
{
if (n < count)
{
ok(pShimData->Query.atrLayers[n] != 0, "Expected to find a valid layer in index %u / %u\n", n, count);
ValidateShim(pShimData->Query.atrLayers[n], layers[n]);
}
else
ok(pShimData->Query.atrLayers[n] == 0, "Expected to find an empty layer in index %u / %u\n", n, count);
}
}
static void Validate_EmptyShimData_Win10(PVOID data)
{
ShimData_Win10_v1* pShimData = (ShimData_Win10_v1*)data;
ok(pShimData != NULL, "Expected pShimData\n");
if (!pShimData)
return;
if (pShimData->dwMagic != SHIMDATA_MAGIC)
{
ShimData_Win10_v2* pShimData2 = (ShimData_Win10_v2*)data;
if (pShimData2->dwMagic != SHIMDATA_MAGIC)
{
skip("Unknown shimdata (win10)\n");
return;
}
ok(!lstrcmpiW(pShimData2->szLayer, L""), "Expected pShimData->szLayer to be '', was %s\n", wine_dbgstr_w(pShimData2->szLayer));
ok(pShimData2->dwSize == sizeof(ShimData_Win10_v2), "Expected pShimData->dwSize to be %u, was %u\n", sizeof(ShimData_Win10_v2), pShimData2->dwSize);
ok(!memcmp(&pShimData2->Query, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
else
{
ok(!lstrcmpiW(pShimData->szModule, L""), "Expected pShimData->Module to be '', was %s\n", wine_dbgstr_w(pShimData->szModule));
ok(!lstrcmpiW(pShimData->szLayer, L""), "Expected pShimData->szLayer to be '', was %s\n", wine_dbgstr_w(pShimData->szLayer));
ok(pShimData->dwSize == sizeof(ShimData_Win10_v1), "Expected pShimData->dwSize to be %u, was %u\n", sizeof(ShimData_Win10_v1), pShimData->dwSize);
ok(!memcmp(&pShimData->Query, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
}
static void Test_layers(WCHAR szApphelp[256])
{
static const char* layers[] = {
"256Color", "NT4SP5", "DisableNXHideUI", "DisableNXShowUI",
"WIN2000SP3", "640X480", /*"DISABLEDWM",*/ "HIGHDPIAWARE",
/*"RUNASADMIN",*/ "DISABLETHEMES" /*, "Layer_Win95VersionLie"*/ };
size_t n;
HANDLE proc;
test_RemoteShimInfo info;
BOOL res;
for (n = 0; n <= (sizeof(layers) / sizeof(layers[0])); ++n)
{
create_environ(layers, n);
proc = create_proc(TRUE);
res = get_shiminfo(proc, &info);
TerminateProcess(proc, 0);
CloseHandle(proc);
if (!res)
{
ok(0, "Unable to get process info (%u)!\n", n);
continue;
}
if (n == 0)
{
ok(info.AppCompatFlags.QuadPart == 0, "Expected AppCompatFlags to be 0, was: %s\n", wine_dbgstr_longlong(info.AppCompatFlags.QuadPart));
ok(info.AppCompatFlagsUser.QuadPart == 0, "Expected AppCompatFlagsUser to be 0, was: %s\n", wine_dbgstr_longlong(info.AppCompatFlagsUser.QuadPart));
ok(info.AppCompatInfo == NULL, "Expected AppCompatInfo to be NULL, was: %p\n", info.AppCompatInfo);
if (g_WinVersion < WINVER_WIN10)
{
ok(info.pShimData == NULL, "Expected pShimData to be NULL, was: %p\n", info.pShimData);
}
else
{
Validate_EmptyShimData_Win10(info.pShimData);
}
}
else
{
ok(info.AppCompatFlags.QuadPart == 0, "Expected AppCompatFlags to be 0, was: %s\n", wine_dbgstr_longlong(info.AppCompatFlags.QuadPart));
ok(info.AppCompatFlagsUser.QuadPart == 0, "Expected AppCompatFlagsUser to be 0, was: %s\n", wine_dbgstr_longlong(info.AppCompatFlagsUser.QuadPart));
ok(info.AppCompatInfo == NULL, "Expected AppCompatInfo to be NULL, was: %p\n", info.AppCompatInfo);
ok(info.pShimData != NULL, "Expected pShimData to be valid, was NULL\n");
ok(info.ShimDataSize == g_ShimDataSize, "Expected ShimDataSize to be %u, was: %u\n", g_ShimDataSize, info.ShimDataSize);
if (info.pShimData)
{
if (g_WinVersion < WINVER_VISTA)
Validate_ShimData_Win2k3(info.pShimData, n, layers);
else if (g_WinVersion < WINVER_WIN10)
Validate_ShimData_Win7(info.pShimData, szApphelp, n, layers);
else
Validate_ShimData_Win10(info.pShimData, szApphelp, n, layers);
}
}
free(info.pShimData);
}
}
/*
[Warn][SdbGetMatchingExe ] __COMPAT_LAYER name cannot exceed 256 characters.
[Info][SdbpGetPermLayersInternal] Failed to read value info from Key "\Registry\Machine\Software\Microsoft\Windows NT\CurrentVersion\AppCompatFlags\Layers" Status 0xc0000034
[Info][SdbpGetPermLayersInternal] Failed to read value info from Key "\REGISTRY\USER\S-1-5-21-4051718696-421402927-393408651-2638\Software\Microsoft\Windows NT\CurrentVersion\AppCompatFlags\Layers" Status 0xc0000034
[Warn][SdbpEnumUserSdb ] Failed to open key "\Registry\Machine\Software\Microsoft\Windows NT\CurrentVersion\AppCompatFlags\Custom\NotepadReplacer.exe" Status 0xc0000034
*/
static void Test_repeatlayer(WCHAR szApphelp[256])
{
static const char* layers[] = {
"256Color", "256Color", "256Color", "256Color",
"256Color", "256Color", "256Color", "256Color" };
HANDLE proc;
test_RemoteShimInfo info;
BOOL res;
SetEnvironmentVariableA("__COMPAT_LAYER", "256Color 256Color 256Color 256Color 256Color 256Color 256Color 256Color");
proc = create_proc(TRUE);
res = get_shiminfo(proc, &info);
TerminateProcess(proc, 0);
CloseHandle(proc);
if (res)
{
ok(info.AppCompatFlags.QuadPart == 0, "Expected AppCompatFlags to be 0, was: %s\n", wine_dbgstr_longlong(info.AppCompatFlags.QuadPart));
ok(info.AppCompatFlagsUser.QuadPart == 0, "Expected AppCompatFlagsUser to be 0, was: %s\n", wine_dbgstr_longlong(info.AppCompatFlagsUser.QuadPart));
ok(info.AppCompatInfo == 0, "Expected AppCompatInfo to be 0, was: %p\n", info.AppCompatInfo);
ok(info.pShimData != NULL, "Expected pShimData to be valid, was NULL\n");
ok(info.ShimDataSize == g_ShimDataSize, "Expected ShimDataSize to be %u, was: %u\n", g_ShimDataSize, info.ShimDataSize);
if (info.pShimData)
{
/* Win10 only 'loads' one layer */
if (g_WinVersion < WINVER_VISTA)
Validate_ShimData_Win2k3(info.pShimData, SDB_MAX_LAYERS, layers);
else if (g_WinVersion < WINVER_WIN10)
Validate_ShimData_Win7(info.pShimData, szApphelp, SDB_MAX_LAYERS, layers);
else
Validate_ShimData_Win10(info.pShimData, szApphelp, 1, layers);
}
}
else
{
ok(0, "Unable to get process info!\n");
}
}
TAGREF find_layer(const char* szLayerStart, const char* szLayerEnd)
{
char layer[100] = { 0 };
WCHAR layerW[100] = { 0 };
strncpy(layer, szLayerStart, szLayerEnd - szLayerStart);
if (!g_LayerDB)
{
g_LayerDB = pSdbInitDatabase(SDB_DATABASE_MAIN_SHIM, 0);
}
mbstowcs(layerW, layer, strlen(layer));
return pSdbGetLayerTagRef(g_LayerDB, layerW);
}
static void expect_layeronly_imp(SDBQUERYRESULT_VISTA* result, const char* layers, DWORD flags)
{
DWORD dwLayerCount = 0, n;
TAGREF atrLayers[SDB_MAX_LAYERS] = { 0 };
const char* layer = layers, *nextlayer;
while (layer && *layer)
{
nextlayer = strchr(layer, ' ');
atrLayers[dwLayerCount++] = find_layer(layer, nextlayer ? nextlayer : (layer + strlen(layer)));
layer = nextlayer ? (nextlayer+1) : NULL;
}
if (g_ModuleVersion < WINVER_VISTA)
{
SDBQUERYRESULT_2k3* result2 = (SDBQUERYRESULT_2k3*)result;
result = NULL;
winetest_ok(!memcmp(&result2->atrExes, &empty_result.atrExes, sizeof(result2->atrExes)), "Expected atrExes to be empty\n");
winetest_ok(!memcmp(&result2->atrLayers[dwLayerCount], &empty_result.atrLayers[dwLayerCount], sizeof(result2->atrLayers) - dwLayerCount * sizeof(result2->atrLayers[0])), "Expected atrLayers[+1] to be empty\n");
for (n = 0; n < dwLayerCount; ++n)
{
winetest_ok(result2->atrLayers[n] == atrLayers[n], "Expected atrLayers[%u] to be %x, was %x\n",
n, atrLayers[n], result2->atrLayers[n]);
}
winetest_ok(result2->dwLayerFlags == 0, "Expected dwLayerFlags to be 0, was %u\n", result2->dwLayerFlags);
winetest_ok(result2->trApphelp == 0, "Expected trApphelp to be 0, was %u\n", result2->trApphelp);
winetest_ok(result2->dwExeCount == 0, "Expected dwExeCount to be 0, was %u\n", result2->dwExeCount);
winetest_ok(result2->dwLayerCount == dwLayerCount, "Expected dwLayerCount to be %u, was %u\n", dwLayerCount, result2->dwLayerCount);
winetest_ok(!memcmp(&result2->guidID, &empty_result.guidID, sizeof(result2->guidID)), "Expected guidID to be empty\n");
winetest_ok(result2->dwFlags == flags, "Expected dwFlags to be 0x%x, was 0x%x\n", flags, result2->dwFlags);
winetest_ok(result2->dwCustomSDBMap == 1, "Expected dwCustomSDBMap to be 1, was %u\n", result2->dwCustomSDBMap);
winetest_ok(!memcmp(&result2->rgGuidDB[1], &empty_result.rgGuidDB[1], sizeof(result2->rgGuidDB) - sizeof(result2->rgGuidDB[0])), "Expected rgGuidDB[+1] to be empty\n");
}
else
{
winetest_ok(!memcmp(&result->atrExes, &empty_result.atrExes, sizeof(empty_result.atrExes)), "Expected atrExes to be empty\n");
winetest_ok(!memcmp(&result->adwExeFlags, &empty_result.adwExeFlags, sizeof(empty_result.adwExeFlags)), "Expected adwExeFlags to be empty\n");
winetest_ok(!memcmp(&result->atrLayers[dwLayerCount], &empty_result.atrLayers[dwLayerCount], sizeof(empty_result.atrLayers) - dwLayerCount * sizeof(empty_result.atrLayers[0])), "Expected atrLayers[+1] to be empty\n");
for (n = 0; n < dwLayerCount; ++n)
{
winetest_ok(result->atrLayers[n] == atrLayers[n], "Expected atrLayers[%u] to be %x, was %x\n",
n, atrLayers[n], result->atrLayers[n]);
}
winetest_ok(result->dwLayerFlags == 0, "Expected dwLayerFlags to be 0, was %u\n", result->dwLayerFlags);
winetest_ok(result->trApphelp == 0, "Expected trApphelp to be 0, was %u\n", result->trApphelp);
winetest_ok(result->dwExeCount == 0, "Expected dwExeCount to be 0, was %u\n", result->dwExeCount);
winetest_ok(result->dwLayerCount == dwLayerCount, "Expected dwLayerCount to be %u, was %u\n", dwLayerCount, result->dwLayerCount);
winetest_ok(!memcmp(&result->guidID, &empty_result.guidID, sizeof(empty_result.guidID)), "Expected guidID to be empty\n");
winetest_ok(result->dwFlags == flags, "Expected dwFlags to be 0x%x, was 0x%x\n", flags, result->dwFlags);
winetest_ok(result->dwCustomSDBMap == 1, "Expected dwCustomSDBMap to be 1, was %u\n", result->dwCustomSDBMap);
winetest_ok(!memcmp(&result->rgGuidDB[1], &empty_result.rgGuidDB[1], sizeof(empty_result.rgGuidDB) - sizeof(empty_result.rgGuidDB[0])), "Expected rgGuidDB[+1] to be empty\n");
}
}
static void Test_Shimdata(SDBQUERYRESULT_VISTA* result, const WCHAR* szLayer)
{
BOOL ret;
PVOID pData;
DWORD dwSize;
pData = NULL;
dwSize = 0;
ret = pSdbPackAppCompatData(g_LayerDB, result, &pData, &dwSize);
ok(ret == TRUE, "Expected ret to be TRUE\n");
if (pData)
{
ShimData_Win2k3* pWin2k3;
ShimData_Win7* pWin7;
ShimData_Win10_v1* pWin10;
ShimData_Win10_v2* pWin10_v2;
SDBQUERYRESULT_VISTA result2 = { { 0 } };
DWORD res = pSdbGetAppCompatDataSize(pData);
ok_int(dwSize, res);
switch(dwSize)
{
case sizeof(ShimData_Win2k3):
pWin2k3 = (ShimData_Win2k3*)pData;
ok_hex(pWin2k3->dwMagic, SHIMDATA_MAGIC);
ok_int(pWin2k3->dwSize, dwSize);
ok(pWin2k3->dwCustomSDBMap == 1, "Expected pWin2k3->dwCustomSDBMap to equal 1, was %u for %s\n", pWin2k3->dwCustomSDBMap, wine_dbgstr_w(szLayer));
//ok(!memcmp(&pWin2k3->Query, result, sizeof(SDBQUERYRESULT_2k3)), "Expected pWin2k3->Query to equal result\n");
//ok_wstr(pWin7->szLayer, szLayer);
break;
case sizeof(ShimData_Win7):
pWin7 = (ShimData_Win7*)pData;
ok_hex(pWin7->dwMagic, SHIMDATA_MAGIC);
ok_int(pWin7->dwSize, dwSize);
ok(!memcmp(&pWin7->Query, result, sizeof(*result)), "Expected pWin7->Query to equal result\n");
ok_wstr(pWin7->szLayer, szLayer);
break;
case sizeof(ShimData_Win10_v1):
pWin10 = (ShimData_Win10_v1*)pData;
ok_hex(pWin10->dwMagic, SHIMDATA_MAGIC);
ok_int(pWin10->dwSize, dwSize);
ok(!memcmp(&pWin10->Query, result, sizeof(*result)), "Expected pWin10->Query to equal result\n");
ok_wstr(pWin10->szLayerEnv, szLayer);
ok_wstr(pWin10->szLayer, L"");
break;
case sizeof(ShimData_Win10_v2):
pWin10_v2 = (ShimData_Win10_v2*)pData;
ok_hex(pWin10_v2->dwMagic, SHIMDATA_MAGIC);
ok_int(pWin10_v2->dwSize, dwSize);
ok(!memcmp(&pWin10_v2->Query, result, sizeof(*result)), "Expected pWin10->Query to equal result\n");
ok_wstr(pWin10_v2->szLayerEnv, szLayer);
ok_wstr(pWin10_v2->szLayer, L"");
break;
default:
skip("Unknown size %d\n", dwSize);
break;
}
ret = pSdbUnpackAppCompatData(g_LayerDB, NULL, pData, &result2);
ok(ret == TRUE, "Expected ret to be TRUE\n");
/* TODO: For some reason 2k3 does not seem to output the database GUIDs,
investigate when we have support for multible db's! */
if (dwSize == sizeof(ShimData_Win2k3))
{
SDBQUERYRESULT_2k3* input = (SDBQUERYRESULT_2k3*)result;
SDBQUERYRESULT_2k3* output = (SDBQUERYRESULT_2k3*)&result2;
const GUID rgGuidDB0 = {0x11111111, 0x1111, 0x1111, {0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}};
// Check expected data.
ok(input->dwLayerCount == 1,
"Expected input->dwLayerCount to be 1, was %u for %s\n",
input->dwLayerCount, wine_dbgstr_w(szLayer));
ok(input->dwCustomSDBMap == 1,
"Expected input->dwCustomSDBMap to be 1, was %u for %s\n",
input->dwCustomSDBMap, wine_dbgstr_w(szLayer));
ok(IsEqualGUID(&input->rgGuidDB[0], &rgGuidDB0),
"Expected input->rgGuidDB[0] to be %s, was %s for %s\n",
wine_dbgstr_guid(&rgGuidDB0), wine_dbgstr_guid(&input->rgGuidDB[0]), wine_dbgstr_w(szLayer));
// Check missing data.
ok(output->dwLayerCount == 0,
"Expected output->dwLayerCount to be 0, was %u for %s\n",
output->dwLayerCount, wine_dbgstr_w(szLayer));
ok(output->dwCustomSDBMap == 0,
"Expected output->dwCustomSDBMap to be 0, was %u for %s\n",
output->dwCustomSDBMap, wine_dbgstr_w(szLayer));
ok(IsEqualGUID(&output->rgGuidDB[0], &empty_result.rgGuidDB[0]),
"Expected output->rgGuidDB[0] to be empty, was %s for %s\n",
wine_dbgstr_guid(&output->rgGuidDB[0]), wine_dbgstr_w(szLayer));
// Fake it for now, so the memcmp works.
output->dwLayerCount = input->dwLayerCount;
output->dwCustomSDBMap = input->dwCustomSDBMap;
output->rgGuidDB[0] = input->rgGuidDB[0];
}
ok(!memcmp(&result2, result, sizeof(*result)), "Expected result2 to equal result for %s\n", wine_dbgstr_w(szLayer));
RtlFreeHeap(GetProcessHeap(), 0, pData);
}
}
#define expect_layeronly (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : expect_layeronly_imp
static void Test_GetMatchingExe(void)
{
BOOL ret;
SDBQUERYRESULT_VISTA result = { { 0 } };
WCHAR self[MAX_PATH];
DWORD flags = (g_WinVersion < WINVER_VISTA) ? 0 : ((g_WinVersion < WINVER_WIN10) ? 1 : 0x21);
GetModuleFileNameW(NULL, self, MAX_PATH);
SetEnvironmentVariableA("__COMPAT_LAYER", NULL);
/* szPath cannot be NULL! */
ret = pSdbGetMatchingExe(NULL, L"", NULL, NULL, 0, &result);
ok(ret == FALSE, "Expected ret to be FALSE\n");
ok(!memcmp(&result, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
result = empty_result;
ret = pSdbGetMatchingExe(NULL, self, NULL, NULL, 0, &result);
ok(ret == FALSE, "Expected ret to be FALSE\n");
ok(!memcmp(&result, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
result = empty_result;
SetEnvironmentVariableA("__COMPAT_LAYER", "Some_invalid_layer_name");
ret = pSdbGetMatchingExe(NULL, self, NULL, NULL, 0, &result);
ok(ret == FALSE, "Expected ret to be FALSE\n");
if (g_WinVersion < WINVER_WIN10)
ok(!memcmp(&result, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
else
ok(!memcmp(&result, &almost_empty, sizeof(almost_empty)), "Expected result to be almost empty\n");
result = empty_result;
SetEnvironmentVariableA("__COMPAT_LAYER", "256Color");
ret = pSdbGetMatchingExe(NULL, self, NULL, NULL, 0, &result);
ok(ret == TRUE, "Expected ret to be TRUE\n");
expect_layeronly(&result, "256Color", flags);
Test_Shimdata(&result, L"256Color");
result = empty_result;
SetEnvironmentVariableA("__COMPAT_LAYER", "640X480");
ret = pSdbGetMatchingExe(NULL, self, NULL, NULL, 0, &result);
ok(ret == TRUE, "Expected ret to be TRUE\n");
expect_layeronly(&result, "640X480", flags);
Test_Shimdata(&result, L"640X480");
/* HIGHDPIAWARE does not exist in 2k3 */
if (g_WinVersion > WINVER_2003)
{
result = empty_result;
SetEnvironmentVariableA("__COMPAT_LAYER", "HIGHDPIAWARE");
ret = pSdbGetMatchingExe(NULL, self, NULL, NULL, 0, &result);
ok(ret == TRUE, "Expected ret to be TRUE\n");
expect_layeronly(&result, "HIGHDPIAWARE", flags);
Test_Shimdata(&result, L"HIGHDPIAWARE");
result = empty_result;
SetEnvironmentVariableA("__COMPAT_LAYER", "256Color HIGHDPIAWARE 640X480");
ret = pSdbGetMatchingExe(NULL, self, NULL, NULL, 0, &result);
ok(ret == TRUE, "Expected ret to be TRUE\n");
expect_layeronly(&result, "256Color HIGHDPIAWARE 640X480", flags);
Test_Shimdata(&result, L"256Color HIGHDPIAWARE 640X480");
}
}
HANDLE xOpenFile(WCHAR* ApplicationName)
{
UNICODE_STRING FileName;
OBJECT_ATTRIBUTES ObjectAttributes;
IO_STATUS_BLOCK IoStatusBlock;
NTSTATUS Status;
HANDLE FileHandle;
if (!RtlDosPathNameToNtPathName_U(ApplicationName, &FileName, NULL, NULL))
{
skip("Unable to translate %s to Nt path\n", wine_dbgstr_w(ApplicationName));
return NULL;
}
InitializeObjectAttributes(&ObjectAttributes, &FileName, OBJ_CASE_INSENSITIVE, NULL, NULL);
Status = NtOpenFile(&FileHandle,
SYNCHRONIZE |
FILE_READ_ATTRIBUTES |
FILE_READ_DATA |
FILE_EXECUTE,
&ObjectAttributes,
&IoStatusBlock,
FILE_SHARE_READ | FILE_SHARE_DELETE,
FILE_SYNCHRONOUS_IO_NONALERT |
FILE_NON_DIRECTORY_FILE);
RtlFreeUnicodeString(&FileName);
if (!NT_SUCCESS(Status))
return NULL;
return FileHandle;
}
#define RESET_CHECKRUNAPP_VARS()\
do { \
if (AppCompatData && AppCompatData != &Query) { RtlFreeHeap(RtlGetProcessHeap(), 0, AppCompatData); } \
ExeType = IMAGE_FILE_MACHINE_I386; \
SxsDataSize = FusionFlags = Reason = 0; \
SxsData = NULL; \
memset(&Query, 0, sizeof(Query)); \
AppCompatData = &Query; \
AppCompatDataSize = 123456; \
} while (0)
#define CHECK_BASICS()\
do { \
ok_hex(ret, TRUE); \
ok(AppCompatData != NULL && AppCompatData != &Query, "Expected the pointer to be valid\n"); \
ok_hex(AppCompatDataSize, sizeof(SDBQUERYRESULT_VISTA)); \
ok(SxsData == NULL, "Expected the pointer to be NULL\n"); \
ok_hex(SxsDataSize, 0); \
ok_hex(FusionFlags, 0); \
} while (0)
/* W10 does not seem to use the flags at all, so with this macro we can still test it below 10. */
#define CHECKREASON(value, w10dum) (g_ModuleVersion < WINVER_WIN10 ? value : w10dum)
static BOOL call_ApphelpCheckRunApp(HANDLE FileHandle, PWCHAR ApplicationName, PVOID Environment, USHORT ExeType,
PULONG Reason, PVOID* SdbQueryAppCompatData, PULONG SdbQueryAppCompatDataSize,
PVOID* SxsData, PULONG SxsDataSize, PULONG FusionFlags)
{
ULONG64 SomeFlag1 = 0;
ULONG SomeFlag2 = 0;
if (pApphelpCheckRunAppEx_w7)
{
return pApphelpCheckRunAppEx_w7(FileHandle, NULL, NULL, ApplicationName, Environment, ExeType, Reason,
SdbQueryAppCompatData, SdbQueryAppCompatDataSize, SxsData, SxsDataSize,
FusionFlags, &SomeFlag1, &SomeFlag2);
}
if (pApphelpCheckRunAppEx_w10)
{
return pApphelpCheckRunAppEx_w10(FileHandle, NULL, NULL, ApplicationName, Environment, NULL, ExeType, Reason,
SdbQueryAppCompatData, SdbQueryAppCompatDataSize, SxsData, SxsDataSize,
FusionFlags, &SomeFlag1, &SomeFlag2);
}
return FALSE;
}
static void Test_ApphelpCheckRunApp(WCHAR szApphelp[256])
{
BOOL ret;
HANDLE FileHandle = NULL;
WCHAR ApplicationName[MAX_PATH], EmptyName[1] = { 0 };
DWORD expect_flags = (g_WinVersion < WINVER_WIN10) ? 1 : 0x21;
USHORT ExeType;
PVOID AppCompatData = NULL, SxsData, DuplicatedEnv, Environment;
ULONG AppCompatDataSize, SxsDataSize, FusionFlags;
ULONG Reason;
SDBQUERYRESULT_VISTA Query;
int n;
/* this are the only interesting bits (with the exception of '1', which is there to invoke the 'default' case) */
const int kTestBits = 0x70503;
if (!pApphelpCheckRunAppEx_w7 && !pApphelpCheckRunAppEx_w10)
{
skip("No usable ApphelpCheckRunAppEx\n");
return;
}
GetModuleFileNameW(NULL, ApplicationName, MAX_PATH);
FileHandle = xOpenFile(ApplicationName);
SetEnvironmentVariableA("__COMPAT_LAYER", NULL);
if (!CreateEnvironmentBlock(&DuplicatedEnv, NULL, TRUE))
DuplicatedEnv = NULL;
ok(DuplicatedEnv != NULL, "Invalid env (%u)\n", GetLastError());
/* First with the environment without __COMPAT_LAYER */
RESET_CHECKRUNAPP_VARS();
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, DuplicatedEnv, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, CHECKREASON(0x30000, 0));
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
/* We need this to be set for tests later on. */
SetEnvironmentVariableA("__COMPAT_LAYER", "256Color");
if (g_ModuleVersion < WINVER_WIN10)
{
/* Showing that when an environment is passed in, that is used instead of the current.
In Win10 this behavior is no longer observed */
RESET_CHECKRUNAPP_VARS();
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, DuplicatedEnv, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, CHECKREASON(0x30000, 0));
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
for (n = 0; n < 32; ++n)
{
ULONG ExpectedReason;
if (!(kTestBits & (1<<n)))
continue;
RESET_CHECKRUNAPP_VARS();
ExpectedReason = Reason = (1 << n);
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, DuplicatedEnv, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
if (ExpectedReason == 2)
ExpectedReason = 2;
else if (ExpectedReason == 0x100)
ExpectedReason = 0x30000;
else
ExpectedReason |= 0x30000;
ok_hex(Reason, CHECKREASON(ExpectedReason, (1 << n)));
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
if (g_ModuleVersion < WINVER_WIN10)
{
/* Now, using a NULL env, showing that the current environment is used.
W10 does no longer do this, so we skip this test here. */
Environment = NULL;
RESET_CHECKRUNAPP_VARS();
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, CHECKREASON(0x50000, 0));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
expect_layeronly(AppCompatData, "256Color", expect_flags);
for (n = 0; n < 32; ++n)
{
ULONG ExpectedReason;
RESET_CHECKRUNAPP_VARS();
if (!(kTestBits & (1<<n)))
continue;
ExpectedReason = Reason = (1 << n);
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
if (ExpectedReason == 2)
ExpectedReason = 2;
else if (ExpectedReason == 0x100)
ExpectedReason = 0x50000;
else if (ExpectedReason == 0x400)
ExpectedReason = 0x30400;
else
ExpectedReason |= 0x50000;
ok_hex(Reason, CHECKREASON(ExpectedReason, (1 << n)));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
{
if (ExpectedReason != 0x30400)
expect_layeronly(AppCompatData, "256Color", expect_flags);
else
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
}
}
/* Passing in an environment with __COMPAT_LAYER set */
Environment = GetEnvironmentStringsW();
RESET_CHECKRUNAPP_VARS();
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, CHECKREASON(0x50000, 0));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
expect_layeronly(AppCompatData, "256Color", expect_flags);
for (n = 0; n < 32; ++n)
{
ULONG ExpectedReason;
if (!(kTestBits & (1<<n)))
continue;
RESET_CHECKRUNAPP_VARS();
ExpectedReason = Reason = (1 << n);
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
if (ExpectedReason == 2)
ExpectedReason = 2;
else if (ExpectedReason == 0x100)
ExpectedReason = 0x50000;
else if (ExpectedReason == 0x400)
ExpectedReason = 0x30400;
else
ExpectedReason |= 0x50000;
ok_hex(Reason, CHECKREASON(ExpectedReason, (1 << n)));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
{
if (ExpectedReason != 0x30400 || g_ModuleVersion >= WINVER_WIN10)
expect_layeronly(AppCompatData, "256Color", expect_flags);
else
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
}
/* NULL file handle still works */
RESET_CHECKRUNAPP_VARS();
ret = call_ApphelpCheckRunApp(NULL, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, CHECKREASON(0x50000, 0));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
expect_layeronly(AppCompatData, "256Color", expect_flags);
for (n = 0; n < 32; ++n)
{
ULONG ExpectedReason;
RESET_CHECKRUNAPP_VARS();
if (!(kTestBits & (1<<n)))
continue;
ExpectedReason = Reason = (1 << n);
ret = call_ApphelpCheckRunApp(NULL, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
if (ExpectedReason == 2)
ExpectedReason = 2;
else if (ExpectedReason == 0x100)
ExpectedReason = 0x50000;
else if (ExpectedReason == 0x400)
ExpectedReason = 0x30400;
else
ExpectedReason |= 0x50000;
ok_hex(Reason, CHECKREASON(ExpectedReason, (1 << n)));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
{
/* W10 does not use the flags anymore? */
if (ExpectedReason != 0x30400 || g_ModuleVersion >= WINVER_WIN10)
expect_layeronly(AppCompatData, "256Color", expect_flags);
else
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
}
/* INVALID_HANDLE_VALUE file handle results in failure (according to flags!), but still results in AppCompatData */
RESET_CHECKRUNAPP_VARS();
ret = call_ApphelpCheckRunApp(INVALID_HANDLE_VALUE, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, 0);
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
expect_layeronly(AppCompatData, "256Color", expect_flags);
for (n = 0; n < 32; ++n)
{
ULONG ExpectedReason;
if (!(kTestBits & (1<<n)))
continue;
RESET_CHECKRUNAPP_VARS();
ExpectedReason = Reason = (1 << n);
ret = call_ApphelpCheckRunApp(INVALID_HANDLE_VALUE, NULL, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
if (ExpectedReason == 0x100)
ExpectedReason = 0;
ok_hex(Reason, CHECKREASON(ExpectedReason, (1 << n)));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
{
if (ExpectedReason != 0x400 && g_ModuleVersion < WINVER_WIN7)
expect_layeronly(AppCompatData, "256Color", expect_flags);
else
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
}
/* NULL filename crashes, showing this in the log before going down:
[Err ][SdbpGetLongFileName ] Failed to get NT path name for ""
[Err ][SdbpCreateSearchDBContext] Unable to parse executable path for "".
[Err ][SdbGetMatchingExe ] Failed to create search DB context.
*/
RESET_CHECKRUNAPP_VARS();
ret = call_ApphelpCheckRunApp(FileHandle, EmptyName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, CHECKREASON(0x30000, 0));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
/* 0 ExeType = don't care? */
RESET_CHECKRUNAPP_VARS();
ExeType = 0;
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, CHECKREASON(0x50000, 0));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
expect_layeronly(AppCompatData, "256Color", expect_flags);
for (n = 0; n < 32; ++n)
{
ULONG ExpectedReason;
if (!(kTestBits & (1<<n)))
continue;
RESET_CHECKRUNAPP_VARS();
ExeType = 0;
ExpectedReason = Reason = (1 << n);
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
if (ExpectedReason == 2)
ExpectedReason = 2;
else if (ExpectedReason == 0x100)
ExpectedReason = 0x50000;
else if (ExpectedReason == 0x400)
ExpectedReason = 0x30400;
else
ExpectedReason |= 0x50000;
ok_hex(Reason, CHECKREASON(ExpectedReason, (1 << n)));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
{
if (ExpectedReason != 0x30400 || g_ModuleVersion >= WINVER_WIN10)
expect_layeronly(AppCompatData, "256Color", expect_flags);
else
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
}
RESET_CHECKRUNAPP_VARS();
ExeType = IMAGE_FILE_MACHINE_POWERPCFP;
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
ok_hex(Reason, CHECKREASON(0x50000, 0));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
expect_layeronly(AppCompatData, "256Color", expect_flags);
for (n = 0; n < 32; ++n)
{
ULONG ExpectedReason;
if (!(kTestBits & (1<<n)))
continue;
RESET_CHECKRUNAPP_VARS();
ExeType = IMAGE_FILE_MACHINE_POWERPCFP;
ExpectedReason = Reason = (1 << n);
ret = call_ApphelpCheckRunApp(FileHandle, ApplicationName, Environment, ExeType, &Reason,
&AppCompatData, &AppCompatDataSize, &SxsData, &SxsDataSize, &FusionFlags);
CHECK_BASICS();
if (ExpectedReason == 2)
ExpectedReason = 2;
else if (ExpectedReason == 0x100)
ExpectedReason = 0x50000;
else if (ExpectedReason == 0x400)
ExpectedReason = 0x30400;
else
ExpectedReason |= 0x50000;
ok_hex(Reason, CHECKREASON(ExpectedReason, (1 << n)));
if (AppCompatData && AppCompatDataSize == sizeof(SDBQUERYRESULT_VISTA))
{
if (ExpectedReason != 0x30400 || g_ModuleVersion >= WINVER_WIN10)
expect_layeronly(AppCompatData, "256Color", expect_flags);
else
ok(!memcmp(AppCompatData, &empty_result, sizeof(empty_result)), "Expected result to be empty\n");
}
}
if (AppCompatData && AppCompatData != &Query)
RtlFreeHeap(RtlGetProcessHeap(), 0, AppCompatData);
FreeEnvironmentStringsW(Environment);
DestroyEnvironmentBlock(DuplicatedEnv);
NtClose(FileHandle);
}
START_TEST(env)
{
WCHAR szApphelp[MAX_PATH];
ShimData_QueryOffset QueryOffset;
DWORD ShimDataType;
NTSTATUS ExceptionStatus = STATUS_SUCCESS;
//SetEnvironmentVariable("SHIM_DEBUG_LEVEL", "127");
//SetEnvironmentVariable("SHIMENG_DEBUG_LEVEL", "127");
silence_debug_output();
hdll = LoadLibraryA("apphelp.dll");
g_WinVersion = get_host_winver();
g_ModuleVersion = get_module_version(hdll);
trace("Detected host: 0x%x, module: 0x%x\n", g_WinVersion, g_ModuleVersion);
GetModuleFileNameW(hdll, szApphelp, _countof(szApphelp));
pSdbGetMatchingExe = (void*)GetProcAddress(hdll, "SdbGetMatchingExe");
pSdbInitDatabase = (void*)GetProcAddress(hdll, "SdbInitDatabase");
pSdbReleaseDatabase = (void*)GetProcAddress(hdll, "SdbReleaseDatabase");
pSdbTagRefToTagID = (void*)GetProcAddress(hdll, "SdbTagRefToTagID");
pSdbGetTagFromTagID = (void*)GetProcAddress(hdll, "SdbGetTagFromTagID");
pSdbGetLayerTagRef = (void*)GetProcAddress(hdll, "SdbGetLayerTagRef");
switch (g_ModuleVersion)
{
case WINVER_WIN7:
pApphelpCheckRunAppEx_w7 = (void*)GetProcAddress(hdll, "ApphelpCheckRunAppEx");
break;
case WINVER_WIN10:
pApphelpCheckRunAppEx_w10 = (void*)GetProcAddress(hdll, "ApphelpCheckRunAppEx");
break;
default:
skip("Unknown apphelp.dll version %x, cannot determine which ApphelpCheckRunApp(Ex) function to use\n", g_ModuleVersion);
break;
}
pSdbPackAppCompatData = (void*)GetProcAddress(hdll, "SdbPackAppCompatData");
pSdbUnpackAppCompatData = (void*)GetProcAddress(hdll, "SdbUnpackAppCompatData");
pSdbGetAppCompatDataSize = (void*)GetProcAddress(hdll, "SdbGetAppCompatDataSize");
memset(&QueryOffset, 0, sizeof(QueryOffset));
QueryOffset.dwMagic_2k3 = QueryOffset.dwMagic_7_10 = QueryOffset.dwMagic_10_v2 = SHIMDATA_MAGIC;
QueryOffset.dwSize_2k3 = 1;
QueryOffset.dwSize_7_10 = 2;
QueryOffset.dwSize_10_v2 = 3;
g_ShimDataSize = g_WinVersion < WINVER_WIN10 ? 4096 : 8192;
_SEH2_TRY
{
ShimDataType = pSdbGetAppCompatDataSize(&QueryOffset);
}
_SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
{
ExceptionStatus = _SEH2_GetExceptionCode();
}
_SEH2_END;
ok(ExceptionStatus == STATUS_SUCCESS, "Exception 0x%08x, expected 0x%08x\n", ExceptionStatus, STATUS_SUCCESS);
if (ExceptionStatus != STATUS_SUCCESS)
{
skip("SdbGetAppCompatDataSize not functional\n");
return;
}
/* New version of Win10.. */
if (g_WinVersion == WINVER_WIN10 && ShimDataType == 3)
g_ShimDataSize = 4096;
if (g_WinVersion == g_ModuleVersion)
{
Test_layers(szApphelp);
Test_repeatlayer(szApphelp);
}
else
{
skip("Tests requiring process launch, reported OS version (0x%x) does not match apphelp version (0x%x)\n", g_WinVersion, g_ModuleVersion);
}
{
Test_GetMatchingExe();
}
Test_ApphelpCheckRunApp(szApphelp);
if (g_LayerDB)
pSdbReleaseDatabase(g_LayerDB);
}
Reference in a new issue View git blame Copy permalink