137 lines
2.9 KiB
C++
137 lines
2.9 KiB
C++
/*++
|
|
|
|
Copyright (c) 2000 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
|
|
Abstract:
|
|
|
|
|
|
Notes:
|
|
|
|
|
|
History:
|
|
|
|
|
|
|
|
--*/
|
|
|
|
#include "precomp.h"
|
|
|
|
IMPLEMENT_SHIM_BEGIN(TSPerUserFiles)
|
|
#include "ShimHookMacro.h"
|
|
|
|
#include "TSPerUserFiles_utils.h"
|
|
|
|
APIHOOK_ENUM_BEGIN
|
|
APIHOOK_ENUM_ENTRY(CreateFileA)
|
|
APIHOOK_ENUM_ENTRY(CreateFileW)
|
|
APIHOOK_ENUM_END
|
|
|
|
|
|
CPerUserPaths* g_pPerUserPaths = NULL;
|
|
|
|
|
|
HANDLE
|
|
APIHOOK(CreateFileA)(
|
|
LPCSTR lpFileName,
|
|
DWORD dwDesiredAccess,
|
|
DWORD dwShareMode,
|
|
LPSECURITY_ATTRIBUTES lpSecurityAttributes,
|
|
DWORD dwCreationDisposition,
|
|
DWORD dwFlagsAndAttributes,
|
|
HANDLE hTemplateFile
|
|
)
|
|
{
|
|
LPCSTR strCorrect = lpFileName;
|
|
|
|
if (g_pPerUserPaths) {
|
|
strCorrect = g_pPerUserPaths->GetPerUserPathA(lpFileName);
|
|
}
|
|
|
|
return ORIGINAL_API(CreateFileA)(strCorrect,
|
|
dwDesiredAccess,
|
|
dwShareMode,
|
|
lpSecurityAttributes,
|
|
dwCreationDisposition,
|
|
dwFlagsAndAttributes,
|
|
hTemplateFile);
|
|
}
|
|
|
|
HANDLE
|
|
APIHOOK(CreateFileW)(
|
|
LPCWSTR lpFileName,
|
|
DWORD dwDesiredAccess,
|
|
DWORD dwShareMode,
|
|
LPSECURITY_ATTRIBUTES lpSecurityAttributes,
|
|
DWORD dwCreationDisposition,
|
|
DWORD dwFlagsAndAttributes,
|
|
HANDLE hTemplateFile
|
|
)
|
|
{
|
|
LPCWSTR strCorrect = lpFileName;
|
|
|
|
if (g_pPerUserPaths) {
|
|
strCorrect = g_pPerUserPaths->GetPerUserPathW(lpFileName);
|
|
}
|
|
|
|
return ORIGINAL_API(CreateFileW)(strCorrect,
|
|
dwDesiredAccess,
|
|
dwShareMode,
|
|
lpSecurityAttributes,
|
|
dwCreationDisposition,
|
|
dwFlagsAndAttributes,
|
|
hTemplateFile);
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Register hooked functions
|
|
|
|
--*/
|
|
|
|
BOOL
|
|
NOTIFY_FUNCTION(
|
|
DWORD fdwReason
|
|
)
|
|
{
|
|
if (fdwReason == DLL_PROCESS_ATTACH) {
|
|
|
|
DPF("TSPerUserFiles",
|
|
eDbgLevelInfo,
|
|
"[NOTIFY_FUNCTION] DLL_PROCESS_ATTACH\n");
|
|
|
|
g_pPerUserPaths = new CPerUserPaths;
|
|
|
|
if (g_pPerUserPaths) {
|
|
|
|
if (!g_pPerUserPaths->Init()) {
|
|
delete g_pPerUserPaths;
|
|
g_pPerUserPaths = NULL;
|
|
}
|
|
}
|
|
} else if (fdwReason == DLL_PROCESS_DETACH) {
|
|
|
|
DPF("TSPerUserFiles",
|
|
eDbgLevelInfo,
|
|
"[NOTIFY_FUNCTION] DLL_PROCESS_DETACH\n");
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
HOOK_BEGIN
|
|
|
|
APIHOOK_ENTRY(KERNEL32.DLL, CreateFileA)
|
|
APIHOOK_ENTRY(KERNEL32.DLL, CreateFileW)
|
|
|
|
CALL_NOTIFY_FUNCTION
|
|
|
|
HOOK_END
|
|
|
|
IMPLEMENT_SHIM_END
|
|
|