RtlSplayTree.c File Reference

#include <kmt_test.h>
#include <debug.h>

Include dependency graph for RtlSplayTree.c:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros
#define	KMT_EMULATE_KERNEL
#define	NDEBUG

Functions
static RTL_GENERIC_COMPARE_RESULTS NTAPI	CompareCharTable (PRTL_GENERIC_TABLE Table, PVOID A, PVOID B)
static PVOID NTAPI	AllocRoutine (PRTL_GENERIC_TABLE Table, CLONG ByteSize)
static VOID NTAPI	FreeRoutine (PRTL_GENERIC_TABLE Table, PVOID Buffer)
static void	RtlSplayTreeTest ()
	START_TEST (RtlSplayTree)

Variables
static LIST_ENTRY	Allocations

Macro Definition Documentation

KMT_EMULATE_KERNEL

#define KMT_EMULATE_KERNEL

Definition at line 8 of file RtlSplayTree.c.

NDEBUG

#define NDEBUG

Definition at line 11 of file RtlSplayTree.c.

Function Documentation

AllocRoutine()

static PVOID NTAPI AllocRoutine ( PRTL_GENERIC_TABLE  Table, CLONG  ByteSize  )

static

Definition at line 26 of file RtlSplayTree.c.

{
    PLIST_ENTRY Entry = ExAllocatePool
        (NonPagedPool, sizeof(LIST_ENTRY) + ByteSize);
    InsertTailList(&Allocations, Entry);
    return &Entry[1];
}

Referenced by RtlSplayTreeTest().

CompareCharTable()

static RTL_GENERIC_COMPARE_RESULTS NTAPI CompareCharTable ( PRTL_GENERIC_TABLE  Table, PVOID  A, PVOID  B  )

static

Definition at line 17 of file RtlSplayTree.c.

{
    RTL_GENERIC_COMPARE_RESULTS Result = (*((PCHAR)A) < *((PCHAR)B)) ? GenericLessThan :
        (*((PCHAR)A) > *((PCHAR)B)) ? GenericGreaterThan :
        GenericEqual;
    return Result;
}

Referenced by RtlSplayTreeTest().

FreeRoutine()

static VOID NTAPI FreeRoutine ( PRTL_GENERIC_TABLE  Table, PVOID  Buffer  )

static

Definition at line 35 of file RtlSplayTree.c.

{
    PLIST_ENTRY Entry = (PLIST_ENTRY)(((PCHAR)Buffer) - sizeof(LIST_ENTRY));
    RemoveEntryList(Entry);
    ExFreePool(Entry);
}

RtlSplayTreeTest()

static void RtlSplayTreeTest ( )

static

Definition at line 42 of file RtlSplayTree.c.

{
    ULONG i, del;
    PCHAR Ch;
    CHAR Text[] = "the quick_brown!fOx-jUmp3d/0vER+THe^lazy.D@g";
    CHAR NewE[] = "11111111111111111111111111111111110111111111";
    RTL_GENERIC_TABLE Table;
    RtlInitializeGenericTable
        (&Table,
         CompareCharTable,
         AllocRoutine,
         FreeRoutine,
         NULL);
    for (i = 0; Text[i]; i++) {
        BOOLEAN WasNew;
        Ch = (PCHAR)RtlInsertElementGenericTable
            (&Table,
             &Text[i],
             sizeof(Text[i]),
             &WasNew);
        ok(Ch && *Ch == Text[i], "Copy character into node\n");
        ok(WasNew == (NewE[i] == '1'),
           "Character newness didn't match for char %u: '%c'\n",
           i, Text[i]);
    }
    for (Ch = (PCHAR)RtlEnumerateGenericTable(&Table, TRUE), i = 0;
         Ch;
         Ch = (PCHAR)RtlEnumerateGenericTable(&Table, FALSE), i++) {
        ok(strchr(Text, *Ch) != NULL, "Nonexistent character\n");
    }
    ok(RtlNumberGenericTableElements(&Table) == strlen(Text) - 1, "Not the right number of elements\n");
    ok(RtlLookupElementGenericTable(&Table, "q") != NULL, "Could not lookup q\n");
    ok(!RtlLookupElementGenericTable(&Table, "#"), "Found a character that shouldn't appear\n");
    ok(strlen(Text) == i + 1, "Didn't enumerate enough characters\n");
    del = 0;
    for (i = 0; Text[i]; i++) {
        if (NewE[i] == '1') {
            BOOLEAN WasDeleted;
            WasDeleted = RtlDeleteElementGenericTable(&Table, &Text[i]);
            del += WasDeleted;
        }
    }
    ok(!RtlNumberGenericTableElements(&Table), "Not zero elements\n");
    ok(!RtlGetElementGenericTable(&Table, 0), "Elements left when we removed them all\n");
    ok(strlen(Text) == del + 1, "Deleted too many times\n");
    ok(IsListEmpty(&Allocations), "Didn't free all memory\n");
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

RtlSplayTree

)

Definition at line 90 of file RtlSplayTree.c.

{
    InitializeListHead(&Allocations);
    RtlSplayTreeTest();
}

Variable Documentation

Allocations

LIST_ENTRY Allocations

static

Definition at line 14 of file RtlSplayTree.c.

Referenced by AllocRoutine(), RtlSplayTreeTest(), and START_TEST().