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: HookRemoveDirectoryW

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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

5
tst/winfsp-tests/winfsp-tests.c
View File

@ -21,12 +21,9 @@
#include <tlib/testsuite.h> #include <tlib/testsuite.h>
#include <time.h> #include <time.h>
#define WINFSP_TESTS_NO_HOOKS
#include "winfsp-tests.h" #include "winfsp-tests.h"
#undef CreateFileW
#undef CloseHandle
#undef DeleteFileW
int NtfsTests = 0; int NtfsTests = 0;
int WinFspDiskTests = 1; int WinFspDiskTests = 1;
int WinFspNetTests = 1; int WinFspNetTests = 1;

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

@ -28,9 +28,12 @@
#define testalpha(c) ('a' <= ((c) | 0x20) && ((c) | 0x20) <= 'z') #define testalpha(c) ('a' <= ((c) | 0x20) && ((c) | 0x20) <= 'z')
#define togglealpha(c) ((c) ^ 0x20) #define togglealpha(c) ((c) ^ 0x20)
#if !defined(WINFSP_TESTS_NO_HOOKS)
#define CreateFileW HookCreateFileW #define CreateFileW HookCreateFileW
#define CloseHandle HookCloseHandle #define CloseHandle HookCloseHandle
#define DeleteFileW HookDeleteFileW #define DeleteFileW HookDeleteFileW
#define RemoveDirectoryW HookRemoveDirectoryW
#endif
HANDLE HookCreateFileW( HANDLE HookCreateFileW(
LPCWSTR lpFileName, LPCWSTR lpFileName,
DWORD dwDesiredAccess, DWORD dwDesiredAccess,
@ -43,6 +46,8 @@ BOOL HookCloseHandle(
HANDLE hObject); HANDLE hObject);
BOOL HookDeleteFileW( BOOL HookDeleteFileW(
LPCWSTR lpFileName); LPCWSTR lpFileName);
BOOL HookRemoveDirectoryW(
LPCWSTR lpPathName);
HANDLE ResilientCreateFileW( HANDLE ResilientCreateFileW(
LPCWSTR lpFileName, LPCWSTR lpFileName,