Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-23 00:43:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

tst: winfsp-tests: WIP

Browse Source
This commit is contained in:
Bill Zissimopoulos 2016-10-26 13:40:30 -07:00
parent dfe45e1be5
commit 5e71992153
2 changed files with 42 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tst/winfsp-tests/hooks.c
View File

@ -154,6 +154,21 @@ BOOL HookCloseHandle(
hObject);
}
BOOL HookSetFileAttributesW(
LPCWSTR lpFileName,
DWORD dwFileAttributes)
{
WCHAR FileNameBuf[FILENAMEBUF_SIZE];
BOOL Success;
PrepareFileName(lpFileName, FileNameBuf);
MaybeAdjustTraversePrivilege(FALSE);
Success = SetFileAttributesW(FileNameBuf, dwFileAttributes);
MaybeAdjustTraversePrivilege(TRUE);
return Success;
}
BOOL HookCreateDirectoryW(
LPCWSTR lpPathName,
LPSECURITY_ATTRIBUTES lpSecurityAttributes)
@ -230,3 +245,20 @@ HANDLE HookFindFirstFileW(
MaybeAdjustTraversePrivilege(TRUE);
return Handle;
}
HANDLE HookFindFirstStreamW(
LPCWSTR lpFileName,
STREAM_INFO_LEVELS InfoLevel,
LPVOID lpFindStreamData,
DWORD dwFlags)
{
WCHAR FileNameBuf[FILENAMEBUF_SIZE];
HANDLE Handle;
PrepareFileName(lpFileName, FileNameBuf);
MaybeAdjustTraversePrivilege(FALSE);
Handle = FindFirstStreamW(FileNameBuf, InfoLevel, lpFindStreamData, dwFlags);
MaybeAdjustTraversePrivilege(TRUE);
return Handle;
}

10
tst/winfsp-tests/winfsp-tests.h
View File

@ -31,11 +31,13 @@
#if !defined(WINFSP_TESTS_NO_HOOKS)
#define CreateFileW HookCreateFileW
#define CloseHandle HookCloseHandle
#define SetFileAttributesW HookSetFileAttributesW
#define CreateDirectoryW HookCreateDirectoryW
#define DeleteFileW HookDeleteFileW
#define RemoveDirectoryW HookRemoveDirectoryW
#define MoveFileExW HookMoveFileExW
#define FindFirstFileW HookFindFirstFileW
#define FindFirstStreamW HookFindFirstStreamW
#endif
HANDLE HookCreateFileW(
LPCWSTR lpFileName,
@ -47,6 +49,9 @@ HANDLE HookCreateFileW(
HANDLE hTemplateFile);
BOOL HookCloseHandle(
HANDLE hObject);
BOOL HookSetFileAttributesW(
LPCWSTR lpFileName,
DWORD dwFileAttributes);
BOOL HookCreateDirectoryW(
LPCWSTR lpPathName,
LPSECURITY_ATTRIBUTES lpSecurityAttributes);
@ -61,6 +66,11 @@ BOOL HookMoveFileExW(
HANDLE HookFindFirstFileW(
LPCWSTR lpFileName,
LPWIN32_FIND_DATAW lpFindFileData);
HANDLE HookFindFirstStreamW(
LPCWSTR lpFileName,
STREAM_INFO_LEVELS InfoLevel,
LPVOID lpFindStreamData,
DWORD dwFlags);
HANDLE ResilientCreateFileW(
LPCWSTR lpFileName,