allocators.c File Reference

#include "precomp.h"
#include <debug.h>

Include dependency graph for allocators.c:

Go to the source code of this file.

Classes
struct	ALLOCATOR

Macros
#define	NDEBUG

Typedefs
typedef PVOID(*	PFNKSPAGEDPOOLALLOCATE) (IN PPAGED_LOOKASIDE_LIST Lookaside)
typedef PVOID(*	PFNKSNPAGEDPOOLALLOCATE) (IN PNPAGED_LOOKASIDE_LIST Lookaside)
typedef VOID(*	PFNKSPAGEDPOOLFREE) (IN PPAGED_LOOKASIDE_LIST Lookaside, IN PVOID Entry)
typedef VOID(*	PFNKSNPAGEDPOOLFREE) (IN PNPAGED_LOOKASIDE_LIST Lookaside, IN PVOID Entry)
typedef VOID(NTAPI *	PFNKSNPAGEDPOOLDELETE) (IN PNPAGED_LOOKASIDE_LIST Lookaside)
typedef VOID(NTAPI *	PFNKSPAGEDPOOLDELETE) (IN PPAGED_LOOKASIDE_LIST Lookaside)
typedef struct ALLOCATOR *	PALLOCATOR

Enumerations
enum	ALLOCATOR_TYPE { ALLOCATOR_NPAGED_LOOKASIDE , ALLOCATOR_PAGED_LOOKASIDE , ALLOCATOR_CUSTOM }
enum	ALLOC_REQUEST { ALLOCATOR_DEVICE_CONTROL , ALLOCATOR_DEVICE_CLOSE , ALLOCATOR_ALLOCATE , ALLOCATOR_FREE }

Functions
NTSTATUS NTAPI	IKsAllocator_Allocate (IN PFILE_OBJECT FileObject, PVOID *Frame)
VOID NTAPI	IKsAllocator_FreeFrame (IN PFILE_OBJECT FileObject, PVOID Frame)
NTSTATUS NTAPI	IKsAllocator_fnQueryInterface (IKsAllocator *iface, IN REFIID refiid, OUT PVOID *Output)
ULONG NTAPI	IKsAllocator_fnAddRef (IKsAllocator *iface)
ULONG NTAPI	IKsAllocator_fnRelease (IKsAllocator *iface)
NTSTATUS NTAPI	IKsAllocator_fnDeviceIoControl (IKsAllocator *iface, IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS NTAPI	IKsAllocator_fnClose (IKsAllocator *iface)
NTSTATUS NTAPI	IKsAllocator_fnAllocateFrame (IKsAllocator *iface, IN PVOID *OutFrame)
VOID NTAPI	IKsAllocator_fnFreeFrame (IKsAllocator *iface, IN PVOID Frame)
KSDDKAPI NTSTATUS NTAPI	KsCreateAllocator (IN HANDLE ConnectionHandle, IN PKSALLOCATOR_FRAMING AllocatorFraming, OUT PHANDLE AllocatorHandle)
KSDDKAPI NTSTATUS NTAPI	KsCreateDefaultAllocator (IN PIRP Irp)
KSDDKAPI NTSTATUS NTAPI	KsValidateAllocatorCreateRequest (IN PIRP Irp, OUT PKSALLOCATOR_FRAMING *OutAllocatorFraming)
NTSTATUS	IKsAllocator_DispatchRequest (IN PDEVICE_OBJECT DeviceObject, IN PFILE_OBJECT FileObject, IN PIRP Irp, IN PVOID Frame, IN ALLOC_REQUEST Request)
NTSTATUS NTAPI	IKsAllocator_DispatchDeviceIoControl (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS NTAPI	IKsAllocator_DispatchClose (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
KSDDKAPI NTSTATUS NTAPI	KsCreateDefaultAllocatorEx (IN PIRP Irp, IN PVOID InitializeContext OPTIONAL, IN PFNKSDEFAULTALLOCATE DefaultAllocate OPTIONAL, IN PFNKSDEFAULTFREE DefaultFree OPTIONAL, IN PFNKSINITIALIZEALLOCATOR InitializeAllocator OPTIONAL, IN PFNKSDELETEALLOCATOR DeleteAllocator OPTIONAL)
KSDDKAPI NTSTATUS NTAPI	KsValidateAllocatorFramingEx (IN PKSALLOCATOR_FRAMING_EX Framing, IN ULONG BufferSize, IN const KSALLOCATOR_FRAMING_EX *PinFraming)

Variables
const GUID	IID_IKsAllocator = {0x642F5D00L, 0x4791, 0x11D0, {0xA5, 0xD6, 0x28, 0xDB, 0x04, 0xC1, 0x00, 0x00}}
const GUID	KSPROPSETID_StreamAllocator = {0x0cf6e4342, 0xec87, 0x11cf, {0xa1, 0x30, 0x00, 0x20, 0xaf, 0xd1, 0x56, 0xe4}}
static IKsAllocatorVtbl	vt_IKsAllocator
static KSDISPATCH_TABLE	DispatchTable

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 11 of file allocators.c.

Typedef Documentation

PALLOCATOR

typedef struct ALLOCATOR * PALLOCATOR

PFNKSNPAGEDPOOLALLOCATE

typedef PVOID(* PFNKSNPAGEDPOOLALLOCATE) (IN PNPAGED_LOOKASIDE_LIST Lookaside)

Definition at line 31 of file allocators.c.

PFNKSNPAGEDPOOLDELETE

typedef VOID(NTAPI * PFNKSNPAGEDPOOLDELETE) (IN PNPAGED_LOOKASIDE_LIST Lookaside)

Definition at line 36 of file allocators.c.

PFNKSNPAGEDPOOLFREE

typedef VOID(* PFNKSNPAGEDPOOLFREE) (IN PNPAGED_LOOKASIDE_LIST Lookaside, IN PVOID Entry)

Definition at line 34 of file allocators.c.

PFNKSPAGEDPOOLALLOCATE

typedef PVOID(* PFNKSPAGEDPOOLALLOCATE) (IN PPAGED_LOOKASIDE_LIST Lookaside)

Definition at line 30 of file allocators.c.

PFNKSPAGEDPOOLDELETE

typedef VOID(NTAPI * PFNKSPAGEDPOOLDELETE) (IN PPAGED_LOOKASIDE_LIST Lookaside)

Definition at line 37 of file allocators.c.

PFNKSPAGEDPOOLFREE

typedef VOID(* PFNKSPAGEDPOOLFREE) (IN PPAGED_LOOKASIDE_LIST Lookaside, IN PVOID Entry)

Definition at line 33 of file allocators.c.

Enumeration Type Documentation

ALLOC_REQUEST

enum ALLOC_REQUEST

Enumerator
ALLOCATOR_DEVICE_CONTROL
ALLOCATOR_DEVICE_CLOSE
ALLOCATOR_ALLOCATE
ALLOCATOR_FREE

Definition at line 21 of file allocators.c.

{
    ALLOCATOR_DEVICE_CONTROL,
    ALLOCATOR_DEVICE_CLOSE,
    ALLOCATOR_ALLOCATE,
    ALLOCATOR_FREE
 
}ALLOC_REQUEST;

ALLOCATOR_TYPE

enum ALLOCATOR_TYPE

Enumerator
ALLOCATOR_NPAGED_LOOKASIDE
ALLOCATOR_PAGED_LOOKASIDE
ALLOCATOR_CUSTOM

Definition at line 14 of file allocators.c.

{
    ALLOCATOR_NPAGED_LOOKASIDE,
    ALLOCATOR_PAGED_LOOKASIDE,
    ALLOCATOR_CUSTOM
}ALLOCATOR_TYPE;

Function Documentation

IKsAllocator_Allocate()

NTSTATUS NTAPI IKsAllocator_Allocate	(	IN PFILE_OBJECT	FileObject,
		PVOID *	Frame
	)

Definition at line 535 of file allocators.c.

{
    NTSTATUS Status;
 
    /* dispatch request */
    Status = IKsAllocator_DispatchRequest(NULL, FileObject, NULL, (PVOID)Frame, ALLOCATOR_ALLOCATE);
 
    return Status;
}

Referenced by IKsAllocator_fnDeviceIoControl().

IKsAllocator_DispatchClose()

NTSTATUS NTAPI IKsAllocator_DispatchClose	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 513 of file allocators.c.

{
    PIO_STACK_LOCATION IoStack;
    NTSTATUS Status;
 
    /* get current irp stack */
    IoStack = IoGetCurrentIrpStackLocation(Irp);
 
    /* dispatch request */
    Status = IKsAllocator_DispatchRequest(DeviceObject, IoStack->FileObject, Irp, NULL, ALLOCATOR_DEVICE_CLOSE);
 
    /* complete request */
    Irp->IoStatus.Status = Status;
    CompleteRequest(Irp, IO_NO_INCREMENT);
 
    return Status;
}

IKsAllocator_DispatchDeviceIoControl()

NTSTATUS NTAPI IKsAllocator_DispatchDeviceIoControl	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 491 of file allocators.c.

{
    PIO_STACK_LOCATION IoStack;
    NTSTATUS Status;
 
    /* get current irp stack */
    IoStack = IoGetCurrentIrpStackLocation(Irp);
 
    /* dispatch request */
    Status = IKsAllocator_DispatchRequest(DeviceObject, IoStack->FileObject, Irp, NULL, ALLOCATOR_DEVICE_CONTROL);
 
    /* complete request */
    Irp->IoStatus.Status = Status;
    CompleteRequest(Irp, IO_NO_INCREMENT);
 
    return Status;
}

IKsAllocator_DispatchRequest()

NTSTATUS IKsAllocator_DispatchRequest	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PFILE_OBJECT	FileObject,
		IN PIRP	Irp,
		IN PVOID	Frame,
		IN ALLOC_REQUEST	Request
	)

Definition at line 435 of file allocators.c.

{
    PKSIOBJECT_HEADER Header;
    NTSTATUS Status;
    IKsAllocator * Allocator;
 
    /* sanity check */
    ASSERT(FileObject);
 
    /* get object header */
    Header = (PKSIOBJECT_HEADER)FileObject->FsContext2;
 
    /* get real allocator */
    Status = Header->Unknown->lpVtbl->QueryInterface(Header->Unknown, &IID_IKsAllocator, (PVOID*)&Allocator);
 
    if (!NT_SUCCESS(Status))
    {
        /* misbehaving object */
        return STATUS_UNSUCCESSFUL;
    }
 
    if (Request == ALLOCATOR_DEVICE_CONTROL)
    {
        /* dispatch request allocator */
        Status = Allocator->lpVtbl->DispatchDeviceIoControl(Allocator, DeviceObject, Irp);
    }
    else if (Request == ALLOCATOR_DEVICE_CLOSE)
    {
        /* delete allocator */
        Status = Allocator->lpVtbl->Close(Allocator);
    }
    else if (Request == ALLOCATOR_ALLOCATE)
    {
        /* allocate frame */
        Status = Allocator->lpVtbl->AllocateFrame(Allocator, (PVOID*)Frame);
 
    }else if (Request == ALLOCATOR_FREE)
    {
        /* allocate frame */
        Allocator->lpVtbl->FreeFrame(Allocator, Frame);
        Status = STATUS_SUCCESS;
    }
 
    /* release interface */
    Allocator->lpVtbl->Release(Allocator);
 
    return Status;
}

Referenced by IKsAllocator_Allocate(), IKsAllocator_DispatchClose(), IKsAllocator_DispatchDeviceIoControl(), and IKsAllocator_FreeFrame().

IKsAllocator_fnAddRef()

ULONG NTAPI IKsAllocator_fnAddRef

(

IKsAllocator *

iface

)

Definition at line 117 of file allocators.c.

{
    PALLOCATOR This = (PALLOCATOR)CONTAINING_RECORD(iface, ALLOCATOR, lpVtbl);
 
    return InterlockedIncrement(&This->ref);
}

IKsAllocator_fnAllocateFrame()

NTSTATUS NTAPI IKsAllocator_fnAllocateFrame	(	IKsAllocator *	iface,
		IN PVOID *	OutFrame
	)

Definition at line 294 of file allocators.c.

{
    PVOID Frame = NULL;
    PALLOCATOR This = (PALLOCATOR)CONTAINING_RECORD(iface, ALLOCATOR, lpVtbl);
 
    /* FIXME locks */
 
    /* now allocate frame */
    if (This->Type == ALLOCATOR_CUSTOM)
    {
        Frame = This->Allocate.DefaultAllocate(This->u.CustomList);
    }
    else if (This->Type == ALLOCATOR_NPAGED_LOOKASIDE)
    {
        Frame = This->Allocate.NPagedPool(&This->u.NPagedList);
    }
    else if (This->Type == ALLOCATOR_PAGED_LOOKASIDE)
    {
        Frame = This->Allocate.PagedPool(&This->u.PagedList);
    }
 
    if (Frame)
    {
        *OutFrame = Frame;
        InterlockedIncrement((PLONG)&This->Status.AllocatedFrames);
        return STATUS_SUCCESS;
    }
 
    return STATUS_UNSUCCESSFUL;
}

IKsAllocator_fnClose()

NTSTATUS NTAPI IKsAllocator_fnClose

(

IKsAllocator *

iface

)

Definition at line 265 of file allocators.c.

{
    PALLOCATOR This = (PALLOCATOR)CONTAINING_RECORD(iface, ALLOCATOR, lpVtbl);
 
    /* FIXME locks */
 
    /* now close allocator */
    if (This->Type == ALLOCATOR_CUSTOM)
    {
        This->Delete.DefaultDelete(This->u.CustomList);
    }
    else if (This->Type == ALLOCATOR_NPAGED_LOOKASIDE)
    {
        This->Delete.NPagedPool(&This->u.NPagedList);
    }
    else if (This->Type == ALLOCATOR_PAGED_LOOKASIDE)
    {
        This->Delete.PagedPool(&This->u.PagedList);
    }
 
    /* free object header */
    KsFreeObjectHeader(&This->Header);
 
    return STATUS_SUCCESS;
}

IKsAllocator_fnDeviceIoControl()

NTSTATUS NTAPI IKsAllocator_fnDeviceIoControl	(	IKsAllocator *	iface,
		IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 145 of file allocators.c.

{
    PALLOCATOR This = (PALLOCATOR)CONTAINING_RECORD(iface, ALLOCATOR, lpVtbl);
    PIO_STACK_LOCATION IoStack;
    PKSSTREAMALLOCATOR_FUNCTIONTABLE FunctionTable;
    PKSSTREAMALLOCATOR_STATUS State;
    PKSPROPERTY Property;
 
    /* FIXME locks */
 
    /* get current irp stack */
    IoStack = IoGetCurrentIrpStackLocation(Irp);
 
    if (IoStack->Parameters.DeviceIoControl.IoControlCode  != IOCTL_KS_PROPERTY)
    {
        /* only KSPROPERTY requests are supported */
        UNIMPLEMENTED;
 
        /* complete and forget irps */
        Irp->IoStatus.Status = STATUS_NOT_IMPLEMENTED;
        CompleteRequest(Irp, IO_NO_INCREMENT);
 
        return STATUS_NOT_IMPLEMENTED;
   }
 
    if (IoStack->Parameters.DeviceIoControl.InputBufferLength < sizeof(KSPROPERTY))
    {
        /* invalid request */
        Irp->IoStatus.Status = STATUS_INVALID_DEVICE_REQUEST;
        CompleteRequest(Irp, IO_NO_INCREMENT);
 
        return STATUS_INVALID_DEVICE_REQUEST;
    }
 
    /* check the request */
    Property = (PKSPROPERTY)IoStack->Parameters.DeviceIoControl.Type3InputBuffer;
 
    if (IsEqualGUIDAligned(&Property->Set, &KSPROPSETID_StreamAllocator))
    {
        if (Property->Id == KSPROPERTY_STREAMALLOCATOR_FUNCTIONTABLE)
        {
            if (IoStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(KSSTREAMALLOCATOR_FUNCTIONTABLE))
            {
                /* buffer too small */
                Irp->IoStatus.Status = STATUS_BUFFER_TOO_SMALL;
                Irp->IoStatus.Information = sizeof(KSSTREAMALLOCATOR_FUNCTIONTABLE);
                /* complete and forget irp */
                CompleteRequest(Irp, IO_NO_INCREMENT);
                return STATUS_BUFFER_TOO_SMALL;
            }
            if (!(Property->Flags & KSPROPERTY_TYPE_GET))
            {
                /* only support retrieving the property */
                Irp->IoStatus.Status = STATUS_UNSUCCESSFUL;
                /* complete and forget irp */
                CompleteRequest(Irp, IO_NO_INCREMENT);
                return STATUS_UNSUCCESSFUL;
            }
 
            /* get output buffer */
            FunctionTable = (PKSSTREAMALLOCATOR_FUNCTIONTABLE)Irp->UserBuffer;
 
            FunctionTable->AllocateFrame = IKsAllocator_Allocate;
            FunctionTable->FreeFrame = IKsAllocator_FreeFrame;
 
            /* save result */
            Irp->IoStatus.Status = STATUS_SUCCESS;
            Irp->IoStatus.Information = sizeof(KSSTREAMALLOCATOR_FUNCTIONTABLE);
            /* complete request */
            CompleteRequest(Irp, IO_NO_INCREMENT);
            return STATUS_SUCCESS;
        }
        else if (Property->Id == KSPROPERTY_STREAMALLOCATOR_STATUS)
        {
            if (IoStack->Parameters.DeviceIoControl.OutputBufferLength < sizeof(KSPROPERTY_STREAMALLOCATOR_STATUS))
            {
                /* buffer too small */
                Irp->IoStatus.Status = STATUS_BUFFER_TOO_SMALL;
                Irp->IoStatus.Information = sizeof(KSPROPERTY_STREAMALLOCATOR_STATUS);
                /* complete and forget irp */
                CompleteRequest(Irp, IO_NO_INCREMENT);
                return STATUS_BUFFER_TOO_SMALL;
            }
            if (!(Property->Flags & KSPROPERTY_TYPE_GET))
            {
                /* only support retrieving the property */
                Irp->IoStatus.Status = STATUS_UNSUCCESSFUL;
                /* complete and forget irp */
                CompleteRequest(Irp, IO_NO_INCREMENT);
                return STATUS_UNSUCCESSFUL;
            }
 
            /* get output buffer */
            State = (PKSSTREAMALLOCATOR_STATUS)Irp->UserBuffer;
 
            /* copy allocator status */
            RtlMoveMemory(State, &This->Status, sizeof(KSSTREAMALLOCATOR_STATUS));
 
            /* save result */
            Irp->IoStatus.Status = STATUS_SUCCESS;
            Irp->IoStatus.Information = sizeof(KSSTREAMALLOCATOR_STATUS);
 
            /* complete request */
            CompleteRequest(Irp, IO_NO_INCREMENT);
            return STATUS_SUCCESS;
        }
    }
 
    /* unhandled request */
    Irp->IoStatus.Status = STATUS_NOT_SUPPORTED;
    CompleteRequest(Irp, IO_NO_INCREMENT);
 
    return STATUS_NOT_SUPPORTED;
}

IKsAllocator_fnFreeFrame()

VOID NTAPI IKsAllocator_fnFreeFrame	(	IKsAllocator *	iface,
		IN PVOID	Frame
	)

Definition at line 329 of file allocators.c.

{
    PALLOCATOR This = (PALLOCATOR)CONTAINING_RECORD(iface, ALLOCATOR, lpVtbl);
 
    /* now allocate frame */
    if (This->Type == ALLOCATOR_CUSTOM)
    {
        This->Free.DefaultFree(This->u.CustomList, Frame);
    }
    else if (This->Type == ALLOCATOR_NPAGED_LOOKASIDE)
    {
        This->Free.NPagedPool(&This->u.NPagedList, Frame);
    }
    else if (This->Type == ALLOCATOR_PAGED_LOOKASIDE)
    {
        This->Free.PagedPool(&This->u.PagedList, Frame);
    }
}

IKsAllocator_fnQueryInterface()

NTSTATUS NTAPI IKsAllocator_fnQueryInterface	(	IKsAllocator *	iface,
		IN REFIID	refiid,
		OUT PVOID *	Output
	)

Definition at line 98 of file allocators.c.

{
    PALLOCATOR This = (PALLOCATOR)CONTAINING_RECORD(iface, ALLOCATOR, lpVtbl);
 
    if (IsEqualGUIDAligned(refiid, &IID_IUnknown) ||
        IsEqualGUIDAligned(refiid, &IID_IKsAllocator))
    {
        *Output = &This->lpVtbl;
        _InterlockedIncrement(&This->ref);
        return STATUS_SUCCESS;
    }
    return STATUS_UNSUCCESSFUL;
}

IKsAllocator_fnRelease()

ULONG NTAPI IKsAllocator_fnRelease

(

IKsAllocator *

iface

)

Definition at line 127 of file allocators.c.

{
    PALLOCATOR This = (PALLOCATOR)CONTAINING_RECORD(iface, ALLOCATOR, lpVtbl);
 
    InterlockedDecrement(&This->ref);
 
    if (This->ref == 0)
    {
        FreeItem(This);
        return 0;
    }
    /* Return new reference count */
    return This->ref;
}

IKsAllocator_FreeFrame()

VOID NTAPI IKsAllocator_FreeFrame	(	IN PFILE_OBJECT	FileObject,
		PVOID	Frame
	)

Definition at line 549 of file allocators.c.

{
    /* dispatch request */
    IKsAllocator_DispatchRequest(NULL, FileObject, NULL, Frame, ALLOCATOR_FREE);
}

Referenced by IKsAllocator_fnDeviceIoControl().

KsCreateAllocator()

KSDDKAPI NTSTATUS NTAPI KsCreateAllocator	(	IN HANDLE	ConnectionHandle,
		IN PKSALLOCATOR_FRAMING	AllocatorFraming,
		OUT PHANDLE	AllocatorHandle
	)

Definition at line 367 of file allocators.c.

{
    return KspCreateObjectType(ConnectionHandle,
                               KSSTRING_Allocator,
                               (PVOID)AllocatorFraming,
                               sizeof(KSALLOCATOR_FRAMING),
                               GENERIC_READ,
                               AllocatorHandle);
}

KsCreateDefaultAllocator()

KSDDKAPI NTSTATUS NTAPI KsCreateDefaultAllocator

(

IN PIRP

Irp

)

Definition at line 384 of file allocators.c.

{
    return KsCreateDefaultAllocatorEx(Irp, NULL, NULL, NULL, NULL, NULL);
}

Referenced by DispatchCreateKMixAllocator().

KsCreateDefaultAllocatorEx()

KSDDKAPI NTSTATUS NTAPI KsCreateDefaultAllocatorEx	(	IN PIRP	Irp,
		IN PVOID InitializeContext	OPTIONAL,
		IN PFNKSDEFAULTALLOCATE DefaultAllocate	OPTIONAL,
		IN PFNKSDEFAULTFREE DefaultFree	OPTIONAL,
		IN PFNKSINITIALIZEALLOCATOR InitializeAllocator	OPTIONAL,
		IN PFNKSDELETEALLOCATOR DeleteAllocator	OPTIONAL
	)

Definition at line 576 of file allocators.c.

{
    NTSTATUS Status;
    PKSALLOCATOR_FRAMING AllocatorFraming;
    PALLOCATOR Allocator;
    PVOID Ctx;
 
    /* first validate connect request */
    Status = KsValidateAllocatorCreateRequest(Irp, &AllocatorFraming);
    if (!NT_SUCCESS(Status))
        return STATUS_INVALID_PARAMETER;
 
    /* check the valid file alignment */
    if (AllocatorFraming->FileAlignment > (PAGE_SIZE-1))
    {
        FreeItem(AllocatorFraming);
        return STATUS_INVALID_PARAMETER;
    }
 
    /* allocate allocator struct */
    Allocator = AllocateItem(NonPagedPool, sizeof(ALLOCATOR));
    if (!Allocator)
    {
        FreeItem(AllocatorFraming);
        return STATUS_INSUFFICIENT_RESOURCES;
    }
 
    /* allocate object header */
 
    Status = KsAllocateObjectHeader((KSOBJECT_HEADER*)&Allocator->Header, 0, NULL, Irp, &DispatchTable);
    if (!NT_SUCCESS(Status))
    {
        FreeItem(AllocatorFraming);
        FreeItem(Allocator);
        return Status;
    }
 
    /* set allocator type in object header */
    Allocator->lpVtbl = &vt_IKsAllocator;
    Allocator->Header->Unknown = (PUNKNOWN)&Allocator->lpVtbl;
    Allocator->ref = 1;
 
    if (DefaultAllocate)
    {
        /* use external allocator */
        Allocator->Type  = ALLOCATOR_CUSTOM;
        Allocator->Allocate.DefaultAllocate = DefaultAllocate;
        Allocator->Free.DefaultFree = DefaultFree;
        Allocator->Delete.DefaultDelete = DeleteAllocator;
        Ctx = InitializeAllocator(InitializeContext, AllocatorFraming, &Allocator->u.CustomList);
        /* check for success */
        if (!Ctx)
        {
            KsFreeObjectHeader(Allocator->Header);
            FreeItem(Allocator);
            return Status;
        }
    }
    else if (AllocatorFraming->PoolType == NonPagedPool)
    {
        /* use non-paged pool allocator */
        Allocator->Type  = ALLOCATOR_NPAGED_LOOKASIDE;
        Allocator->Allocate.NPagedPool = ExAllocateFromNPagedLookasideList;
        Allocator->Free.NPagedPool = ExFreeToNPagedLookasideList;
        Allocator->Delete.NPagedPool = ExDeleteNPagedLookasideList;
        ExInitializeNPagedLookasideList(&Allocator->u.NPagedList, NULL, NULL, 0, AllocatorFraming->FrameSize, 0, 0);
    }
    else if (AllocatorFraming->PoolType == PagedPool)
    {
        /* use paged pool allocator */
        Allocator->Allocate.PagedPool = ExAllocateFromPagedLookasideList;
        Allocator->Free.PagedPool = ExFreeToPagedLookasideList;
        Allocator->Delete.PagedPool = ExDeletePagedLookasideList;
        Allocator->Type  = ALLOCATOR_PAGED_LOOKASIDE;
        ExInitializePagedLookasideList(&Allocator->u.PagedList, NULL, NULL, 0, AllocatorFraming->FrameSize, 0, 0);
 
    }
 
    /* backup allocator framing */
    RtlMoveMemory(&Allocator->Status.Framing, AllocatorFraming, sizeof(KSALLOCATOR_FRAMING));
    FreeItem(AllocatorFraming);
 
    return Status;
}

Referenced by KsCreateDefaultAllocator().

KsValidateAllocatorCreateRequest()

KSDDKAPI NTSTATUS NTAPI KsValidateAllocatorCreateRequest	(	IN PIRP	Irp,
		OUT PKSALLOCATOR_FRAMING *	OutAllocatorFraming
	)

Definition at line 396 of file allocators.c.

{
    PKSALLOCATOR_FRAMING AllocatorFraming;
    ULONG Size;
    NTSTATUS Status;
    ULONG SupportedFlags;
 
    /* set minimum request size */
    Size = sizeof(KSALLOCATOR_FRAMING);
 
    Status = KspCopyCreateRequest(Irp,
                                  KSSTRING_Allocator,
                                  &Size,
                                  (PVOID*)&AllocatorFraming);
 
    if (!NT_SUCCESS(Status))
        return Status;
 
    /* allowed supported flags */
    SupportedFlags = (KSALLOCATOR_OPTIONF_COMPATIBLE | KSALLOCATOR_OPTIONF_SYSTEM_MEMORY |
                      KSALLOCATOR_REQUIREMENTF_INPLACE_MODIFIER | KSALLOCATOR_REQUIREMENTF_SYSTEM_MEMORY | KSALLOCATOR_REQUIREMENTF_FRAME_INTEGRITY |
                      KSALLOCATOR_REQUIREMENTF_MUST_ALLOCATE);
 
 
    if (!AllocatorFraming->FrameSize || (AllocatorFraming->OptionsFlags & (~SupportedFlags)))
    {
        FreeItem(AllocatorFraming);
        return STATUS_INVALID_PARAMETER;
    }
 
    /* store result */
    *OutAllocatorFraming = AllocatorFraming;
 
    return Status;
}

Referenced by KsCreateDefaultAllocatorEx().

KsValidateAllocatorFramingEx()

KSDDKAPI NTSTATUS NTAPI KsValidateAllocatorFramingEx	(	IN PKSALLOCATOR_FRAMING_EX	Framing,
		IN ULONG	BufferSize,
		IN const KSALLOCATOR_FRAMING_EX *	PinFraming
	)

Definition at line 671 of file allocators.c.

{
    if (BufferSize < sizeof(KSALLOCATOR_FRAMING_EX))
       return  STATUS_INVALID_DEVICE_REQUEST;
 
    /* verify framing */
    if ((Framing->FramingItem[0].Flags & KSALLOCATOR_FLAG_PARTIAL_READ_SUPPORT) &&
         Framing->OutputCompression.RatioNumerator != MAXULONG &&
         Framing->OutputCompression.RatioDenominator != 0 &&
         Framing->OutputCompression.RatioDenominator < Framing->OutputCompression.RatioNumerator)
    {
        /* framing request is ok */
        return STATUS_SUCCESS;
    }
 
    return STATUS_INVALID_DEVICE_REQUEST;
}

Variable Documentation

DispatchTable

KSDISPATCH_TABLE DispatchTable

static

Initial value:

=
{
    IKsAllocator_DispatchDeviceIoControl,
    KsDispatchInvalidDeviceRequest,
    KsDispatchInvalidDeviceRequest,
    KsDispatchInvalidDeviceRequest,
    IKsAllocator_DispatchClose,
    KsDispatchQuerySecurity,
    KsDispatchSetSecurity,
    KsDispatchFastIoDeviceControlFailure,
    KsDispatchFastReadFailure,
    KsDispatchFastReadFailure,
}

Definition at line 558 of file allocators.c.

IID_IKsAllocator

const GUID IID_IKsAllocator = {0x642F5D00L, 0x4791, 0x11D0, {0xA5, 0xD6, 0x28, 0xDB, 0x04, 0xC1, 0x00, 0x00}}

Definition at line 79 of file allocators.c.

Referenced by IKsAllocator_DispatchRequest(), and IKsAllocator_fnQueryInterface().

KSPROPSETID_StreamAllocator

const GUID KSPROPSETID_StreamAllocator = {0x0cf6e4342, 0xec87, 0x11cf, {0xa1, 0x30, 0x00, 0x20, 0xaf, 0xd1, 0x56, 0xe4}}

Definition at line 80 of file allocators.c.

Referenced by IKsAllocator_fnDeviceIoControl().

vt_IKsAllocator

IKsAllocatorVtbl vt_IKsAllocator

static

Initial value:

=
{
    IKsAllocator_fnQueryInterface,
    IKsAllocator_fnAddRef,
    IKsAllocator_fnRelease,
    IKsAllocator_fnDeviceIoControl,
    IKsAllocator_fnClose,
    IKsAllocator_fnAllocateFrame,
    IKsAllocator_fnFreeFrame
}

Definition at line 351 of file allocators.c.

Referenced by KsCreateDefaultAllocatorEx().