--- /dev/null
+/*
+ * PROJECT: ReactOS kernel-mode tests
+ * LICENSE: LGPLv2+ - See COPYING.LIB in the top level directory
+ * PURPOSE: Kernel-Mode Test Suite FsRtl Test
+ * PROGRAMMER: Pierre Schweitzer <pierre.schweitzer@reactos.org>
+ */
+
+#include <kmt_test.h>
+
+#define NDEBUG
+#include <debug.h>
+
+static VOID FsRtlMcbTest()
+{
+}
+
+static VOID FsRtlLargeMcbTest()
+{
+ LARGE_MCB LargeMcb;
+ ULONG NbRuns, Index;
+ LONGLONG Vbn, Lbn, SectorCount, StartingLbn, CountFromStartingLbn;
+
+ FsRtlInitializeLargeMcb(&LargeMcb, PagedPool);
+
+ ok(FsRtlLookupLastLargeMcbEntry(&LargeMcb, &Vbn, &Lbn) == FALSE, "expected FALSE, got TRUE\n");
+ ok(FsRtlLookupLastLargeMcbEntryAndIndex(&LargeMcb, &Vbn, &Lbn, &Index) == FALSE, "expected FALSE, got TRUE\n");
+
+ ok(FsRtlAddLargeMcbEntry(&LargeMcb, 1, 1, 1024) == TRUE, "expected TRUE, got FALSE\n");
+ NbRuns = FsRtlNumberOfRunsInLargeMcb(&LargeMcb);
+ ok(NbRuns == 2, "Expected 2 runs, got: %lu\n", NbRuns);
+ ok(FsRtlLookupLastLargeMcbEntry(&LargeMcb, &Vbn, &Lbn) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Vbn == 1024, "Expected Vbn 1024, got: %I64d\n", Vbn);
+ ok(Lbn == 1024, "Expected Lbn 1024, got: %I64d\n", Lbn);
+ ok(FsRtlLookupLastLargeMcbEntryAndIndex(&LargeMcb, &Vbn, &Lbn, &Index) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Vbn == 1024, "Expected Vbn 1024, got: %I64d\n", Vbn);
+ ok(Lbn == 1024, "Expected Lbn 1024, got: %I64d\n", Lbn);
+ ok(Index == 1, "Expected Index 1, got: %lu\n", Index);
+
+ ok(FsRtlAddLargeMcbEntry(&LargeMcb, 2048, 2, 1024) == TRUE, "expected TRUE, got FALSE\n");
+ NbRuns = FsRtlNumberOfRunsInLargeMcb(&LargeMcb);
+ ok(NbRuns == 4, "Expected 4 runs, got: %lu\n", NbRuns);
+ ok(FsRtlLookupLastLargeMcbEntry(&LargeMcb, &Vbn, &Lbn) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Vbn == 3071, "Expected Vbn 3071, got: %I64d\n", Vbn);
+ ok(Lbn == 1025, "Expected Lbn 1025, got: %I64d\n", Lbn);
+ ok(FsRtlLookupLastLargeMcbEntryAndIndex(&LargeMcb, &Vbn, &Lbn, &Index) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Vbn == 3071, "Expected Vbn 3071, got: %I64d\n", Vbn);
+ ok(Lbn == 1025, "Expected Lbn 1025, got: %I64d\n", Lbn);
+ ok(Index == 3, "Expected Index 3, got: %lu\n", Index);
+
+ ok(FsRtlGetNextLargeMcbEntry(&LargeMcb, 0, &Vbn, &Lbn, &SectorCount) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Vbn == 0, "Expected Vbn 0, got: %I64d\n", Vbn);
+ ok(Lbn == -1, "Expected Lbn -1, got: %I64d\n", Lbn);
+ ok(SectorCount == 1, "Expected SectorCount 1, got: %I64d\n", SectorCount);
+
+ ok(FsRtlGetNextLargeMcbEntry(&LargeMcb, 1, &Vbn, &Lbn, &SectorCount) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Vbn == 1, "Expected Vbn 1, got: %I64d\n", Vbn);
+ ok(Lbn == 1, "Expected Lbn 1, got: %I64d\n", Lbn);
+ ok(SectorCount == 1024, "Expected SectorCount 1024, got: %I64d\n", SectorCount);
+
+ ok(FsRtlGetNextLargeMcbEntry(&LargeMcb, 2, &Vbn, &Lbn, &SectorCount) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Vbn == 1025, "Expected Vbn 1025, got: %I64d\n", Vbn);
+ ok(Lbn == -1, "Expected Lbn -1, got: %I64d\n", Lbn);
+ ok(SectorCount == 1023, "Expected SectorCount 1023, got: %I64d\n", SectorCount);
+
+ ok(FsRtlGetNextLargeMcbEntry(&LargeMcb, 3, &Vbn, &Lbn, &SectorCount) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Vbn == 2048, "Expected Vbn 2048, got: %I64d\n", Vbn);
+ ok(Lbn == 2, "Expected Lbn 2, got: %I64d\n", Lbn);
+ ok(SectorCount == 1024, "Expected SectorCount 1024, got: %I64d\n", SectorCount);
+
+ ok(FsRtlGetNextLargeMcbEntry(&LargeMcb, 4, &Vbn, &Lbn, &SectorCount) == FALSE, "expected FALSE, got TRUE\n");
+
+ ok(FsRtlLookupLargeMcbEntry(&LargeMcb, 1, &Lbn, &SectorCount, &StartingLbn, &CountFromStartingLbn, &Index) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Lbn == 1, "Expected Lbn 1, got: %I64d\n", Lbn);
+ ok(SectorCount == 1024, "Expected SectorCount 1024, got: %I64d\n", SectorCount);
+ ok(StartingLbn == 1, "Expected StartingLbn 1, got: %I64d\n", StartingLbn);
+ ok(CountFromStartingLbn == 1024, "Expected CountFromStartingLbn 1024, got: %I64d\n", CountFromStartingLbn);
+ ok(Index == 1, "Expected Index 1, got: %lu\n", Index);
+
+ ok(FsRtlLookupLargeMcbEntry(&LargeMcb, 2048, &Lbn, &SectorCount, &StartingLbn, &CountFromStartingLbn, &Index) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Lbn == 2, "Expected Lbn 2, got: %I64d\n", Lbn);
+ ok(SectorCount == 1024, "Expected SectorCount 1024, got: %I64d\n", SectorCount);
+ ok(StartingLbn == 2, "Expected StartingLbn 2, got: %I64d\n", StartingLbn);
+ ok(CountFromStartingLbn == 1024, "Expected CountFromStartingLbn 1024, got: %I64d\n", CountFromStartingLbn);
+ ok(Index == 3, "Expected Index 3, got: %lu\n", Index);
+
+ ok(FsRtlLookupLargeMcbEntry(&LargeMcb, 3073, &Lbn, &SectorCount, &StartingLbn, &CountFromStartingLbn, &Index) == FALSE, "expected FALSE, got TRUE\n");
+
+ FsRtlRemoveLargeMcbEntry(&LargeMcb, 1, 1024);
+ NbRuns = FsRtlNumberOfRunsInLargeMcb(&LargeMcb);
+ ok(NbRuns == 2, "Expected 2 runs, got: %lu\n", NbRuns);
+ ok(FsRtlLookupLargeMcbEntry(&LargeMcb, 512, &Lbn, &SectorCount, &StartingLbn, &CountFromStartingLbn, &Index) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Lbn == -1, "Expected Lbn -1, got: %I64d\n", Lbn);
+ ok(SectorCount == 1536, "Expected SectorCount 1536, got: %I64d\n", SectorCount);
+ ok(StartingLbn == -1, "Expected StartingLbn -1, got: %I64d\n", StartingLbn);
+ ok(CountFromStartingLbn == 2048, "Expected CountFromStartingLbn 2048, got: %I64d\n", CountFromStartingLbn);
+ ok(Index == 0, "Expected Index 0, got: %lu\n", Index);
+ ok(FsRtlLookupLastLargeMcbEntryAndIndex(&LargeMcb, &Vbn, &Lbn, &Index) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Vbn == 3071, "Expected Vbn 3071, got: %I64d\n", Vbn);
+ ok(Lbn == 1025, "Expected Lbn 1025, got: %I64d\n", Lbn);
+ ok(Index == 1, "Expected Index 1, got: %lu\n", Index);
+
+ ok(FsRtlSplitLargeMcb(&LargeMcb, 2048, 1024) == TRUE, "expected FALSE, got TRUE\n");
+ NbRuns = FsRtlNumberOfRunsInLargeMcb(&LargeMcb);
+ ok(NbRuns == 2, "Expected 2 runs, got: %lu\n", NbRuns);
+ ok(FsRtlLookupLastLargeMcbEntryAndIndex(&LargeMcb, &Vbn, &Lbn, &Index) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Vbn == 4095, "Expected Vbn 4095, got: %I64d\n", Vbn);
+ ok(Lbn == 1025, "Expected Lbn 1025, got: %I64d\n", Lbn);
+ ok(Index == 1, "Expected Index 1, got: %lu\n", Index);
+ ok(FsRtlLookupLargeMcbEntry(&LargeMcb, 2048, &Lbn, &SectorCount, &StartingLbn, &CountFromStartingLbn, &Index) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Lbn == -1, "Expected Lbn -1, got: %I64d\n", Lbn);
+ ok(SectorCount == 1024, "Expected SectorCount 1024, got: %I64d\n", SectorCount);
+ ok(StartingLbn == -1, "Expected StartingLbn -1, got: %I64d\n", StartingLbn);
+ ok(CountFromStartingLbn == 3072, "Expected CountFromStartingLbn 3072, got: %I64d\n", CountFromStartingLbn);
+ ok(Index == 0, "Expected Index 0, got: %lu\n", Index);
+ ok(FsRtlLookupLargeMcbEntry(&LargeMcb, 3072, &Lbn, &SectorCount, &StartingLbn, &CountFromStartingLbn, &Index) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Lbn == 2, "Expected Lbn 2, got: %I64d\n", Lbn);
+ ok(SectorCount == 1024, "Expected SectorCount 1024, got: %I64d\n", SectorCount);
+ ok(StartingLbn == 2, "Expected StartingLbn 2, got: %I64d\n", StartingLbn);
+ ok(CountFromStartingLbn == 1024, "Expected CountFromStartingLbn 1024, got: %I64d\n", CountFromStartingLbn);
+ ok(Index == 1, "Expected Index 1, got: %lu\n", Index);
+
+ ok(FsRtlAddLargeMcbEntry(&LargeMcb, 3584, 3, 1024) == FALSE, "expected FALSE, got TRUE\n");
+
+ ok(FsRtlAddLargeMcbEntry(&LargeMcb, 4095, 1025, 1024) == TRUE, "expected TRUE, got FALSE\n");
+ NbRuns = FsRtlNumberOfRunsInLargeMcb(&LargeMcb);
+ ok(NbRuns == 2, "Expected 2 runs, got: %lu\n", NbRuns);
+ ok(FsRtlLookupLastLargeMcbEntry(&LargeMcb, &Vbn, &Lbn) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Vbn == 5118, "Expected Vbn 5118, got: %I64d\n", Vbn);
+ ok(Lbn == 2048, "Expected Lbn 2048, got: %I64d\n", Lbn);
+ ok(FsRtlLookupLastLargeMcbEntryAndIndex(&LargeMcb, &Vbn, &Lbn, &Index) == TRUE, "expected FALSE, got TRUE\n");
+ ok(Vbn == 5118, "Expected Vbn 5118, got: %I64d\n", Vbn);
+ ok(Lbn == 2048, "Expected Lbn 2048, got: %I64d\n", Lbn);
+ ok(Index == 1, "Expected Index 1, got: %lu\n", Index);
+
+ FsRtlTruncateLargeMcb(&LargeMcb, 4607);
+ ok(FsRtlLookupLargeMcbEntry(&LargeMcb, 4095, &Lbn, &SectorCount, &StartingLbn, &CountFromStartingLbn, &Index) == TRUE, "expected TRUE, got FALSE\n");
+ ok(Lbn == 1025, "Expected Lbn 1025, got: %I64d\n", Lbn);
+ ok(SectorCount == 512, "Expected SectorCount 512, got: %I64d\n", SectorCount);
+ ok(StartingLbn == 2, "Expected StartingLbn 2, got: %I64d\n", StartingLbn);
+ ok(CountFromStartingLbn == 1535, "Expected CountFromStartingLbn 1535, got: %I64d\n", CountFromStartingLbn);
+ ok(Index == 1, "Expected Index 1, got: %lu\n", Index);
+
+ FsRtlUninitializeLargeMcb(&LargeMcb);
+}
+
+START_TEST(FsRtlMcb)
+{
+ FsRtlMcbTest();
+ FsRtlLargeMcbTest();
+}