diff --git a/tst/fscrash/fscrash-main.c b/tst/fscrash/fscrash-main.c index 588e4c50..e3468327 100644 --- a/tst/fscrash/fscrash-main.c +++ b/tst/fscrash/fscrash-main.c @@ -15,18 +15,139 @@ * software. */ +#include +#include #include +#include #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);