From 938c036387ea682d1a6bcc3e4f6fa8c5df5ac3b1 Mon Sep 17 00:00:00 2001
From: Bill Zissimopoulos <billziss@navimatics.com>
Date: Mon, 3 Oct 2016 18:27:09 -0700
Subject: [PATCH] tst: winfsp-tests: stream testing

---
 tst/winfsp-tests/stream-tests.c | 117 ++++++++++++++++++++++++++++++++
 1 file changed, 117 insertions(+)

diff --git a/tst/winfsp-tests/stream-tests.c b/tst/winfsp-tests/stream-tests.c
index 5fb1fd70..908050ba 100644
--- a/tst/winfsp-tests/stream-tests.c
+++ b/tst/winfsp-tests/stream-tests.c
@@ -1009,6 +1009,122 @@ static void stream_delete_test(void)
     }
 }
 
+static void stream_delete_pending_dotest(ULONG Flags, PWSTR Prefix, ULONG FileInfoTimeout)
+{
+    void *memfs = memfs_start_ex(Flags, FileInfoTimeout);
+
+    HANDLE Handle, StreamHandle;
+    BOOL Success;
+    WCHAR Dir1Path[MAX_PATH], Dir1StreamPath[MAX_PATH];
+    WCHAR FilePath[MAX_PATH], FileStreamPath[MAX_PATH];
+    FILE_DISPOSITION_INFO DispositionInfo;
+
+    StringCbPrintfW(Dir1Path, sizeof Dir1Path, L"%s%s\\dir1",
+        Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : memfs_volumename(memfs));
+
+    StringCbPrintfW(Dir1StreamPath, sizeof Dir1StreamPath, L"%s%s\\dir1:foo1",
+        Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : memfs_volumename(memfs));
+
+    StringCbPrintfW(FilePath, sizeof FilePath, L"%s%s\\dir1\\file0",
+        Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : memfs_volumename(memfs));
+
+    StringCbPrintfW(FileStreamPath, sizeof FileStreamPath, L"%s%s\\dir1\\file0:foo0",
+        Prefix ? L"" : L"\\\\?\\GLOBALROOT", Prefix ? Prefix : memfs_volumename(memfs));
+
+    Success = CreateDirectoryW(Dir1Path, 0);
+    ASSERT(Success);
+
+    Handle = CreateFileW(Dir1StreamPath,
+        GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+        CREATE_NEW, FILE_ATTRIBUTE_NORMAL, 0);
+    ASSERT(INVALID_HANDLE_VALUE != Handle);
+    CloseHandle(Handle);
+
+    Handle = CreateFileW(FilePath,
+        GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+        CREATE_NEW, FILE_ATTRIBUTE_NORMAL, 0);
+    ASSERT(INVALID_HANDLE_VALUE != Handle);
+    CloseHandle(Handle);
+
+    Handle = CreateFileW(FileStreamPath,
+        GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, 0,
+        CREATE_NEW, FILE_ATTRIBUTE_NORMAL, 0);
+    ASSERT(INVALID_HANDLE_VALUE != Handle);
+    CloseHandle(Handle);
+
+    Success = RemoveDirectoryW(Dir1Path);
+    ASSERT(!Success);
+    ASSERT(ERROR_DIR_NOT_EMPTY == GetLastError());
+
+    {
+        Handle = CreateFileW(FilePath,
+            DELETE, FILE_SHARE_DELETE, 0,
+            OPEN_EXISTING, 0, 0);
+        ASSERT(INVALID_HANDLE_VALUE != Handle);
+        DispositionInfo.DeleteFile = TRUE;
+        Success = SetFileInformationByHandle(Handle,
+            FileDispositionInfo, &DispositionInfo, sizeof DispositionInfo);
+        ASSERT(Success);
+
+        StreamHandle = CreateFileW(FileStreamPath,
+            FILE_READ_ATTRIBUTES, 0, 0,
+            OPEN_EXISTING, 0, 0);
+        ASSERT(INVALID_HANDLE_VALUE == StreamHandle);
+        ASSERT(ERROR_ACCESS_DENIED == GetLastError());
+
+        CloseHandle(Handle);
+    }
+
+    Success = DeleteFileW(FilePath);
+    ASSERT(!Success);
+    ASSERT(ERROR_FILE_NOT_FOUND == GetLastError());
+
+    {
+        Handle = CreateFileW(Dir1Path,
+            DELETE, FILE_SHARE_DELETE, 0,
+            OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, 0);
+        ASSERT(INVALID_HANDLE_VALUE != Handle);
+        DispositionInfo.DeleteFile = TRUE;
+        Success = SetFileInformationByHandle(Handle,
+            FileDispositionInfo, &DispositionInfo, sizeof DispositionInfo);
+        ASSERT(Success);
+
+        StreamHandle = CreateFileW(Dir1StreamPath,
+            FILE_READ_ATTRIBUTES, 0, 0,
+            OPEN_EXISTING, 0, 0);
+        ASSERT(INVALID_HANDLE_VALUE == StreamHandle);
+        ASSERT(ERROR_ACCESS_DENIED == GetLastError());
+
+        CloseHandle(Handle);
+    }
+
+    Success = RemoveDirectoryW(Dir1Path);
+    ASSERT(!Success);
+    ASSERT(ERROR_FILE_NOT_FOUND == GetLastError());
+
+    memfs_stop(memfs);
+}
+
+static void stream_delete_pending_test(void)
+{
+    if (NtfsTests)
+    {
+        WCHAR DirBuf[MAX_PATH] = L"\\\\?\\";
+        GetCurrentDirectoryW(MAX_PATH - 4, DirBuf + 4);
+        stream_delete_pending_dotest(-1, DirBuf, 0);
+    }
+    if (WinFspDiskTests)
+    {
+        stream_delete_pending_dotest(MemfsDisk, 0, 0);
+        stream_delete_pending_dotest(MemfsDisk, 0, 1000);
+    }
+    if (WinFspNetTests)
+    {
+        stream_delete_pending_dotest(MemfsNet, L"\\\\memfs\\share", 0);
+        stream_delete_pending_dotest(MemfsNet, L"\\\\memfs\\share", 1000);
+    }
+}
+
 static void stream_getsecurity_dotest(ULONG Flags, PWSTR Prefix, ULONG FileInfoTimeout)
 {
     void *memfs = memfs_start_ex(Flags, FileInfoTimeout);
@@ -1258,6 +1374,7 @@ void stream_tests(void)
     TEST(stream_getfileinfo_test);
     TEST(stream_setfileinfo_test);
     TEST(stream_delete_test);
+    TEST(stream_delete_pending_test);
     TEST(stream_getsecurity_test);
     TEST(stream_setsecurity_test);
 }