tst: fscrash: WIP

tst/fscrash/fscrash-main.c

@@ -15,18 +15,139 @@
  * software.
  */
 
+#include <winfsp/winfsp.h>
+#include <sddl.h>
 #include <stdio.h>
+#include <stdlib.h>
 
 #include "fscrash.h"
 #include "memfs.h"
 
 #define fail(format, ...)               fprintf(stderr, format, __VA_ARGS__)
+#define ASSERT(expr)                    \
+    (!(expr) ?                          \
+        (fail("ASSERT(%s) failed at %s:%d:%s\n", #expr, __FILE__, __LINE__, __func__), abort()) :\
+        (void)0)
+
+VOID Test(PWSTR Prefix)
+{
+    static WCHAR *Sddl = L"D:P(A;;GA;;;WD)";
+    static const GUID ReparseGuid =
+        { 0x2cf25cfa, 0x41af, 0x4796, { 0xb5, 0xef, 0xac, 0xa3, 0x85, 0x3, 0xe2, 0xd8 } };
+    WCHAR FileName[1024];
+    PSECURITY_DESCRIPTOR SecurityDescriptor;
+    HANDLE Handle;
+    BOOL Success;
+    UINT8 RdBuffer[512], WrBuffer[] = "Buffer";
+    REPARSE_GUID_DATA_BUFFER ReparseDataBuf;
+    DWORD BytesTransferred;
+    WIN32_FIND_DATAW FindData;
+    WIN32_FIND_STREAM_DATA FindStreamData;
+
+    Success = ConvertStringSecurityDescriptorToSecurityDescriptorW(
+        Sddl, SDDL_REVISION_1, &SecurityDescriptor, 0);
+    ASSERT(Success);
+
+    wsprintfW(FileName, L"%s\\fscrash", Prefix);
+    Success = CreateDirectoryW(FileName, 0);
+    ASSERT(Success);
+
+    wsprintfW(FileName, L"%s\\fscrash\\file0", Prefix);
+    Handle = CreateFileW(FileName,
+        GENERIC_READ | GENERIC_WRITE, 0, 0,
+        CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_NO_BUFFERING, 0);
+    ASSERT(INVALID_HANDLE_VALUE != Handle);
+
+    Success = WriteFile(Handle, WrBuffer, sizeof WrBuffer, &BytesTransferred, 0);
+    ASSERT(Success);
+    ASSERT(0 != BytesTransferred);
+
+    Success = FlushFileBuffers(Handle);
+    ASSERT(Success);
+
+    Success = SetFilePointer(Handle, 0, 0, FILE_BEGIN);
+    ASSERT(Success);
+
+    Success = ReadFile(Handle, RdBuffer, sizeof RdBuffer, &BytesTransferred, 0);
+    ASSERT(Success);
+    ASSERT(sizeof WrBuffer == BytesTransferred);
+
+    Success = SetFilePointer(Handle, 42, 0, FILE_BEGIN);
+    ASSERT(Success);
+
+    Success = SetEndOfFile(Handle);
+    ASSERT(Success);
+
+    BytesTransferred = GetFileSize(Handle, 0);
+    ASSERT(42 == BytesTransferred);
+
+    Success = LockFile(Handle, 0, 0, 1, 0);
+    ASSERT(Success);
+
+    Success = UnlockFile(Handle, 0, 0, 1, 0);
+    ASSERT(Success);
+
+    Success = SetKernelObjectSecurity(Handle, DACL_SECURITY_INFORMATION, SecurityDescriptor);
+    ASSERT(Success);
+
+    Success = GetKernelObjectSecurity(Handle, DACL_SECURITY_INFORMATION, 0, 0, &BytesTransferred);
+    ASSERT(!Success);
+    ASSERT(ERROR_INSUFFICIENT_BUFFER == GetLastError());
+
+    Success = CloseHandle(Handle);
+    ASSERT(Success);
+
+    ReparseDataBuf.ReparseTag = 0x1234;
+    ReparseDataBuf.ReparseDataLength = 0;
+    ReparseDataBuf.Reserved = 0;
+    memcpy(&ReparseDataBuf.ReparseGuid, &ReparseGuid, sizeof ReparseGuid);
+
+    Success = DeviceIoControl(Handle, FSCTL_SET_REPARSE_POINT,
+        &ReparseDataBuf, REPARSE_GUID_DATA_BUFFER_HEADER_SIZE + ReparseDataBuf.ReparseDataLength,
+        0, 0,
+        &BytesTransferred, 0);
+    ASSERT(Success);
+
+    wsprintfW(FileName, L"%s\\fscrash\\*", Prefix);
+    Handle = FindFirstFileW(FileName, &FindData);
+    ASSERT(INVALID_HANDLE_VALUE != Handle);
+    do
+    {
+    } while (FindNextFileW(Handle, &FindData));
+    ASSERT(ERROR_NO_MORE_FILES == GetLastError());
+
+    wsprintfW(FileName, L"%s\\fscrash\\file0", Prefix);
+    Handle = FindFirstStreamW(FileName, FindStreamInfoStandard, &FindStreamData, 0);
+    ASSERT(INVALID_HANDLE_VALUE != Handle);
+    do
+    {
+    } while (FindNextStreamW(Handle, &FindStreamData));
+    ASSERT(ERROR_HANDLE_EOF == GetLastError());
+
+    wsprintfW(FileName, L"%s\\fscrash\\file0", Prefix);
+    Success = DeleteFileW(FileName);
+    ASSERT(Success);
+
+    wsprintfW(FileName, L"%s\\fscrash", Prefix);
+    Success = RemoveDirectoryW(FileName);
+    ASSERT(Success);
+
+    LocalFree(SecurityDescriptor);
+}
 
 ULONG OptCrashMask = -1, OptCrashFlags = FspCrashInterceptAccessViolation, OptCrashPercent = 10;
 ULONG OptMemfsFlags = MemfsDisk;
+ULONG OptIterations = -1;
 
 int wmain(int argc, wchar_t **argv)
 {
+#if 0
+    WCHAR CurrentDirectory[MAX_PATH];
+    GetCurrentDirectoryW(MAX_PATH, CurrentDirectory);
+    Test(CurrentDirectory);
+    exit(0);
+#endif
+
     for (int argi = 1; argc > argi; argi++)
     {
         const wchar_t *a = argv[argi];
@@ -34,8 +155,6 @@ int wmain(int argc, wchar_t **argv)
         {
             if (0 == wcsncmp(L"--mask=", a, sizeof "--mask=" - 1))
                 OptCrashMask = wcstoul(a + sizeof "--mask=" - 1, 0, 0);
-            else if (0 == wcsncmp(L"--percent=", a, sizeof "--percent=" - 1))
-                OptCrashPercent = wcstoul(a + sizeof "--percent=" - 1, 0, 10);
             else if (0 == wcscmp(L"--crash", a))
             {
                 OptCrashFlags &= ~FspCrashInterceptTerminate;
@@ -50,19 +169,24 @@ int wmain(int argc, wchar_t **argv)
                 OptCrashFlags |= FspCrashInterceptEnter;
             else if (0 == wcscmp(L"--leave", a))
                 OptCrashFlags |= FspCrashInterceptLeave;
+            else if (0 == wcsncmp(L"--percent=", a, sizeof "--percent=" - 1))
+                OptCrashPercent = wcstoul(a + sizeof "--percent=" - 1, 0, 10);
             else if (0 == wcscmp(L"--disk", a))
                 OptMemfsFlags = MemfsDisk;
             else if (0 == wcscmp(L"--n", a))
                 OptMemfsFlags = MemfsDisk;
+            else if (0 == wcsncmp(L"--iterations=", a, sizeof "--iterations=" - 1))
+                OptIterations = wcstoul(a + sizeof "--iterations=" - 1, 0, 10);
         }
     }
 
     MEMFS *Memfs;
     NTSTATUS Result;
+    WCHAR Prefix[1024];
 
     Result = MemfsCreate(
         OptMemfsFlags,
-        -1,
+        0,
         1024,
         1024 * 1024,
         (MemfsNet & OptMemfsFlags) ? L"\\memfs\\share" : 0,
@@ -83,6 +207,17 @@ int wmain(int argc, wchar_t **argv)
         exit(1);
     }
 
+    if (0 == (MemfsNet & OptMemfsFlags))
+        wsprintfW(Prefix, L"\\\\?\\GLOBALROOT%s", MemfsFileSystem(Memfs)->VolumeName);
+    else
+        wsprintfW(Prefix, L"\\memfs\\share");
+
+    for (ULONG Iterations = OptIterations; -1 == Iterations || 0 != Iterations; Iterations--)
+    {
+        Test(Prefix);
+        FspCrash(MemfsFileSystem(Memfs)); /* after one time through, go ahead and crash! */
+    }
+
     MemfsStop(Memfs);
     MemfsDelete(Memfs);