Context.c File Reference

#include "fltmgr.h"
#include "fltmgrint.h"
#include <debug.h>

Include dependency graph for Context.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
static BOOLEAN	IsContextTypeValid (_In_ FLT_CONTEXT_TYPE ContextType)
static NTSTATUS	SetupContextHeader (_In_ PFLT_FILTER Filter, _In_ PCFLT_CONTEXT_REGISTRATION ContextPtr, _Out_ PALLOCATE_CONTEXT_HEADER ContextHeader)
NTSTATUS	FltpRegisterContexts (_In_ PFLT_FILTER Filter, _In_ const FLT_CONTEXT_REGISTRATION *Context)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 14 of file Context.c.

Function Documentation

FltpRegisterContexts()

NTSTATUS FltpRegisterContexts	(	_In_ PFLT_FILTER	Filter,
		_In_ const FLT_CONTEXT_REGISTRATION *	Context
	)

Definition at line 43 of file Context.c.

{
    PCFLT_CONTEXT_REGISTRATION ContextPtr;
    PALLOCATE_CONTEXT_HEADER ContextHeader, Prev;
    PVOID Buffer;
    ULONG BufferSize = 0;
    USHORT i;
    NTSTATUS Status;
 
    /* Loop through all entries in the context registration array */
    ContextPtr = Context;
    while (ContextPtr)
    {
        /* Bail if we found the terminator */
        if (ContextPtr->ContextType == FLT_CONTEXT_END)
            break;
 
        /* Make sure we have a valid context  */
        if (IsContextTypeValid(ContextPtr->ContextType))
        {
            /* Each context is backed by a crtl struct. Reserve space for it */
            BufferSize += sizeof(STREAM_LIST_CTRL);
        }
 
        /* Move to the next entry */
        ContextPtr++;
    }
 
    /* Bail if we found no valid registration requests */
    if (BufferSize == 0)
    {
        return STATUS_SUCCESS;
    }
 
    /* Allocate the pool that'll hold the context crtl structs */
    Buffer = ExAllocatePoolWithTag(NonPagedPool, BufferSize, FM_TAG_CONTEXT_REGISTA);
    if (!Buffer) return STATUS_INSUFFICIENT_RESOURCES;
 
    RtlZeroMemory(Buffer, BufferSize);
 
    /* Setup our loop data */
    ContextHeader = Buffer;
    Prev = NULL;
    Status = STATUS_SUCCESS;
 
    for (i = 0; i < MAX_CONTEXT_TYPES; i++)
    {
        ContextPtr = (PFLT_CONTEXT_REGISTRATION)Context;
        while (ContextPtr)
        {
            /* We don't support variable sized contents yet */
            FLT_ASSERT(ContextPtr->Size != FLT_VARIABLE_SIZED_CONTEXTS);
 
            /* Bail if we found the terminator */
            if (ContextPtr->ContextType == FLT_CONTEXT_END)
                break;
 
            /* Size and pooltag are only checked when ContextAllocateCallback is null */
            if (ContextPtr->ContextAllocateCallback == FALSE && ContextPtr->PoolTag == FALSE)
            {
                Status = STATUS_FLT_INVALID_CONTEXT_REGISTRATION;
                goto Quit;
            }
 
            /* Make sure we have a valid context  */
            if (IsContextTypeValid(ContextPtr->ContextType))
            {
                Status = SetupContextHeader(Filter, ContextPtr, ContextHeader);
                if (NT_SUCCESS(Status))
                {
                    if (Prev)
                    {
                        Prev->Next = ContextHeader;
                    }
 
                    Filter->SupportedContexts[i] = ContextHeader;
                }
            }
 
            Prev = ContextHeader;
 
            /* Move to the next entry */
            ContextPtr++;
        }
    }
 
Quit:
    if (NT_SUCCESS(Status))
    {
        Filter->SupportedContextsListHead = Buffer;
    }
    else
    {
        ExFreePoolWithTag(Buffer, FM_TAG_CONTEXT_REGISTA);
        //FIXME: Cleanup anything that SetupContextHeader may have allocated
    }
 
    return Status;
}

Referenced by FltRegisterFilter().

IsContextTypeValid()

static BOOLEAN IsContextTypeValid ( _In_ FLT_CONTEXT_TYPE  ContextType )

static

Definition at line 149 of file Context.c.

{
    switch (ContextType)
    {
    case FLT_VOLUME_CONTEXT:
    case FLT_INSTANCE_CONTEXT:
    case FLT_FILE_CONTEXT:
    case FLT_STREAM_CONTEXT:
    case FLT_STREAMHANDLE_CONTEXT:
    case FLT_TRANSACTION_CONTEXT:
        return TRUE;
    }
 
    return FALSE;
}

Referenced by FltpRegisterContexts().

SetupContextHeader()

static NTSTATUS SetupContextHeader ( _In_ PFLT_FILTER  Filter, _In_ PCFLT_CONTEXT_REGISTRATION  ContextPtr, _Out_ PALLOCATE_CONTEXT_HEADER  ContextHeader  )

static

Definition at line 167 of file Context.c.

{
    return 0;
}

Referenced by FltpRegisterContexts().