tst: winfsp-tests: HookRemoveDirectoryW

This commit is contained in:
Bill Zissimopoulos
2016-10-26 12:08:29 -07:00
parent 0532cee99c
commit e07ef0712e
3 changed files with 21 additions and 8 deletions

View File

@ -15,12 +15,9 @@
* software.
*/
#define WINFSP_TESTS_NO_HOOKS
#include "winfsp-tests.h"
#undef CreateFileW
#undef CloseHandle
#undef DeleteFileW
#define FILENAMEBUF_SIZE 1024
static VOID PrepareFileName(PCWSTR FileName, PWSTR FileNameBuf)
@ -187,8 +184,22 @@ BOOL HookCloseHandle(
BOOL HookDeleteFileW(
LPCWSTR lpFileName)
{
WCHAR FileNameBuf[FILENAMEBUF_SIZE];
PrepareFileName(lpFileName, FileNameBuf);
if (!OptResilient)
return DeleteFileW(lpFileName);
else
return ResilientDeleteFileW(lpFileName);
}
BOOL HookRemoveDirectoryW(
LPCWSTR lpPathName)
{
WCHAR FileNameBuf[FILENAMEBUF_SIZE];
PrepareFileName(lpPathName, FileNameBuf);
return RemoveDirectoryW(lpPathName);
}