fltmgr_register.c File Reference

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

Include dependency graph for fltmgr_register.c:

Go to the source code of this file.

Macros
#define	RESET_REGISTRATION(basic)
#define	RESET_UNLOAD(DO)   DO->DriverUnload = NULL;

Functions
NTSTATUS FLTAPI	TestRegFilterUnload (_In_ FLT_FILTER_UNLOAD_FLAGS Flags)
BOOLEAN	TestFltRegisterFilter (_In_ PDRIVER_OBJECT DriverObject)
NTSTATUS	TestEntry (IN PDRIVER_OBJECT DriverObject, IN PCUNICODE_STRING RegistryPath, OUT PCWSTR *DeviceName, IN OUT INT *Flags)
VOID	TestFilterUnload (IN ULONG Flags)
NTSTATUS	TestInstanceSetup (_In_ PCFLT_RELATED_OBJECTS FltObjects, _In_ FLT_INSTANCE_SETUP_FLAGS Flags, _In_ DEVICE_TYPE VolumeDeviceType, _In_ FLT_FILESYSTEM_TYPE VolumeFilesystemType, _In_ PUNICODE_STRING VolumeName, _In_ ULONG SectorSize, _In_ ULONG ReportedSectorSize)
VOID	TestQueryTeardown (_In_ PCFLT_RELATED_OBJECTS FltObjects, _In_ FLT_INSTANCE_QUERY_TEARDOWN_FLAGS Flags)

Variables
static PDRIVER_OBJECT	TestDriverObject
static FLT_REGISTRATION	FilterRegistration
static PFLT_FILTER	TestFilter = NULL

Macro Definition Documentation

RESET_REGISTRATION

#define RESET_REGISTRATION

(

basic

)

Value:

    do { \
        RtlZeroMemory(&FilterRegistration, sizeof(FLT_REGISTRATION)); \
        if (basic) { \
            FilterRegistration.Size = sizeof(FLT_REGISTRATION); \
            FilterRegistration.Version = FLT_REGISTRATION_VERSION; \
        } \
    } while (0)

Definition at line 19 of file fltmgr_register.c.

RESET_UNLOAD

#define RESET_UNLOAD

(

DO

)

DO->DriverUnload = NULL;

Definition at line 28 of file fltmgr_register.c.

Function Documentation

TestEntry()

NTSTATUS TestEntry	(	IN PDRIVER_OBJECT	DriverObject,
		IN PCUNICODE_STRING	RegistryPath,
		OUT PCWSTR *	DeviceName,
		IN OUT INT *	Flags
	)

Definition at line 191 of file fltmgr_register.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    UNREFERENCED_PARAMETER(RegistryPath);
 
    DPRINT("FltMgrReg Entry!\n");
    trace("Entered FltMgrReg tests\n");
 
    /* We'll do the work ourselves in this test */
    *Flags = TESTENTRY_NO_ALL;
 
    ok_irql(PASSIVE_LEVEL);
    TestDriverObject = DriverObject;
 
 
    /* Run the tests */
    (VOID)TestFltRegisterFilter(DriverObject);
 
    return Status;
}

TestFilterUnload()

VOID TestFilterUnload

(

IN ULONG

Flags

)

Definition at line 220 of file fltmgr_register.c.

{
    PAGED_CODE();
    ok_irql(PASSIVE_LEVEL);
}

TestFltRegisterFilter()

BOOLEAN TestFltRegisterFilter

(

_In_ PDRIVER_OBJECT

DriverObject

)

Definition at line 46 of file fltmgr_register.c.

{
    UNICODE_STRING Altitude;
    UNICODE_STRING Name;
    PFLT_FILTER Filter = NULL;
    PFLT_FILTER Temp = NULL;
    NTSTATUS Status;
 
    RESET_REGISTRATION(FALSE);
#if 0
    KmtStartSeh()
        Status = FltRegisterFilter(NULL, &FilterRegistration, &Filter);
    KmtEndSeh(STATUS_INVALID_PARAMETER);
 
    KmtStartSeh()
        Status = FltRegisterFilter(DriverObject, NULL, &Filter);
    KmtEndSeh(STATUS_INVALID_PARAMETER);
 
    KmtStartSeh()
        Status = FltRegisterFilter(DriverObject, &FilterRegistration, NULL);
    KmtEndSeh(STATUS_INVALID_PARAMETER)
#endif
 
    RESET_REGISTRATION(TRUE);
    FilterRegistration.Version = 0x0100;
    Status = FltRegisterFilter(DriverObject, &FilterRegistration, &Filter);
    ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
 
    RESET_REGISTRATION(TRUE);
    FilterRegistration.Version = 0x0300;
    Status = FltRegisterFilter(DriverObject, &FilterRegistration, &Filter);
    ok_eq_hex(Status, STATUS_INVALID_PARAMETER);
 
    RESET_REGISTRATION(TRUE);
    FilterRegistration.Version = 0x0200;
    Status = FltRegisterFilter(DriverObject, &FilterRegistration, &Filter);
    ok_eq_hex(Status, STATUS_SUCCESS);
    FltUnregisterFilter(Filter);
 
 
    /* Test invalid sizes. MSDN says this is required, but it doesn't appear to be */
    RESET_REGISTRATION(TRUE);
    FilterRegistration.Size = 0;
    Status = FltRegisterFilter(DriverObject, &FilterRegistration, &Filter);
    ok_eq_hex(Status, STATUS_SUCCESS);
    FltUnregisterFilter(Filter);
 
    RESET_REGISTRATION(TRUE);
    FilterRegistration.Size = 0xFFFF;
    Status = FltRegisterFilter(DriverObject, &FilterRegistration, &Filter);
    ok_eq_hex(Status, STATUS_SUCCESS);
    FltUnregisterFilter(Filter);
 
 
    /* Now make a valid registration */
    RESET_REGISTRATION(TRUE);
    Status = FltRegisterFilter(DriverObject, &FilterRegistration, &Filter);
    ok_eq_hex(Status, STATUS_SUCCESS);
 
    /* Try to register again */
    Status = FltRegisterFilter(DriverObject, &FilterRegistration, &Temp);
    ok_eq_hex(Status, STATUS_FLT_INSTANCE_ALTITUDE_COLLISION);
 
 
    ok_eq_hex(Filter->Base.Flags, FLT_OBFL_TYPE_FILTER);
 
    /* Check we have the right filter name */
    RtlInitUnicodeString(&Name, L"Kmtest-FltMgrReg");
    ok_eq_long(RtlCompareUnicodeString(&Filter->Name, &Name, FALSE), 0);
 
    /* And the altitude is corect */
    RtlInitUnicodeString(&Altitude, L"123456");
    ok_eq_long(RtlCompareUnicodeString(&Filter->DefaultAltitude, &Altitude, FALSE), 0);
 
    //
    // FIXME: More checks
    //
 
    /* Cleanup the valid registration */
    FltUnregisterFilter(Filter);
 
    /*
     * The last thing we'll do before we exit is to properly register with the filter manager
     * and set an unload routine. This'll let us test the FltUnregisterFilter routine
     */
    RESET_REGISTRATION(TRUE);
 
    /* Set a fake unload routine we'll use to test */
    DriverObject->DriverUnload = (PDRIVER_UNLOAD)0x1234FFFF;
 
    FilterRegistration.FilterUnloadCallback = TestRegFilterUnload;
    Status = FltRegisterFilter(DriverObject, &FilterRegistration, &TestFilter);
    ok_eq_hex(Status, STATUS_SUCCESS);
 
    /* Test all the unlod routines */
    ok_eq_pointer(TestFilter->FilterUnload, TestRegFilterUnload);
    ok_eq_pointer(TestFilter->OldDriverUnload, (PFLT_FILTER_UNLOAD_CALLBACK)0x1234FFFF);
 
    // This should equal the fltmgr's private unload routine, but there's no easy way of testing it...
    //ok_eq_pointer(DriverObject->DriverUnload, FltpMiniFilterDriverUnload);
 
    /* Make sure our test address is never actually called */
    TestFilter->OldDriverUnload = (PFLT_FILTER_UNLOAD_CALLBACK)NULL;
 
    return TRUE;
}

Referenced by TestEntry().

TestInstanceSetup()

NTSTATUS TestInstanceSetup	(	_In_ PCFLT_RELATED_OBJECTS	FltObjects,
		_In_ FLT_INSTANCE_SETUP_FLAGS	Flags,
		_In_ DEVICE_TYPE	VolumeDeviceType,
		_In_ FLT_FILESYSTEM_TYPE	VolumeFilesystemType,
		_In_ PUNICODE_STRING	VolumeName,
		_In_ ULONG	SectorSize,
		_In_ ULONG	ReportedSectorSize
	)

Definition at line 228 of file fltmgr_register.c.

{
    return STATUS_FLT_DO_NOT_ATTACH;
}

TestQueryTeardown()

VOID TestQueryTeardown	(	_In_ PCFLT_RELATED_OBJECTS	FltObjects,
		_In_ FLT_INSTANCE_QUERY_TEARDOWN_FLAGS	Flags
	)

Definition at line 242 of file fltmgr_register.c.

{
    UNREFERENCED_PARAMETER(FltObjects);
    UNREFERENCED_PARAMETER(Flags);
}

TestRegFilterUnload()

NTSTATUS FLTAPI TestRegFilterUnload

(

_In_ FLT_FILTER_UNLOAD_FLAGS

Flags

)

Definition at line 156 of file fltmgr_register.c.

{
    //__debugbreak();
 
    ok_irql(PASSIVE_LEVEL);
    ok(TestFilter != NULL, "Buffer is NULL\n");
 
    //
    // FIXME: Add tests
    //
 
    FltUnregisterFilter(TestFilter);
 
    //
    // FIXME: Add tests
    //
 
    return STATUS_SUCCESS;
}

Referenced by TestFltRegisterFilter().

Variable Documentation

FilterRegistration

FLT_REGISTRATION FilterRegistration

static

Definition at line 39 of file fltmgr_register.c.

Referenced by TestFltRegisterFilter().

TestDriverObject

PDRIVER_OBJECT TestDriverObject

static

Definition at line 38 of file fltmgr_register.c.

Referenced by TestEntry().

TestFilter

PFLT_FILTER TestFilter = NULL

static

Definition at line 40 of file fltmgr_register.c.

Referenced by TestFltRegisterFilter(), and TestRegFilterUnload().