fxrequestcontext.cpp

Go to the documentation of this file.

/*++
 
Copyright (c) Microsoft Corporation
 
Module Name:
 
    FxRequestContext.cpp
 
Abstract:
 
    This module implements FxRequest object
 
Author:
 
 
 
Environment:
 
    Both kernel and user mode
 
Revision History:
 
 
 
--*/
 
#include "coreprivshared.hpp"
 
// Tracing support
extern "C" {
// #include "FxRequestContext.tmh"
}
 
FxRequestContext::FxRequestContext(
    __in FX_REQUEST_CONTEXT_TYPE Type
    ) :
    m_RequestType(Type),
    m_RequestMemory(NULL)
 
/*++
 
Routine Description:
    Constructs an FxRequestContext and initialized the m_RequestType field
 
Arguments:
    Type - The type of this request.
 
 
 
 
 
Return Value:
    None.
 
  --*/
{
    InitCompletionParams();
}
 
FxRequestContext::~FxRequestContext()
/*++
 
Routine Description:
    Destruct for an FxRequestContext.  Releases all outstanding references.
 
Arguments:
    None
 
Return Value:
    None
 
  --*/
{
    ASSERT(m_RequestMemory == NULL);
}
 
VOID
FxRequestContext::StoreAndReferenceMemory(
    __in FxRequestBuffer* Buffer
    )
{
    _StoreAndReferenceMemoryWorker(this, &m_RequestMemory, Buffer);
}
 
VOID
FxRequestContext::ReleaseAndRestore(
    __in FxRequestBase* Request
    )
/*++
 
Routine Description:
    This routine releases any outstanding references taken on the previous
    format call and restores any fields in the PIRP that were overwritten
    when the formatting occurred.
 
Arguments:
    Irp
 
Return Value:
 
 
  --*/
 
{
#if (FX_CORE_MODE == FX_CORE_KERNEL_MODE)
    Request->FreeMdls();
#else
    UNREFERENCED_PARAMETER(Request);
#endif
 
    if (m_RequestMemory != NULL) {
        m_RequestMemory->RELEASE(this);
        m_RequestMemory = NULL;
    }
 
    InitCompletionParams();
}
 
VOID
FxRequestContext::_StoreAndReferenceMemoryWorker(
    __in PVOID Tag,
    __deref_out_opt IFxMemory** PPMemory,
    __in FxRequestBuffer* Buffer
    )
{
    ASSERT(*PPMemory == NULL);
 
    switch (Buffer->DataType) {
    case FxRequestBufferMemory:
        Buffer->u.Memory.Memory->ADDREF(Tag);
        *PPMemory = Buffer->u.Memory.Memory;
        break;
 
    case FxRequestBufferReferencedMdl:
        Buffer->u.RefMdl.Memory->ADDREF(Tag);
        *PPMemory = Buffer->u.RefMdl.Memory;
        break;
 
    default:
        *PPMemory = NULL;
    }
}
 
VOID
FxRequestContext::FormatWriteParams(
    __in_opt IFxMemory* WriteMemory,
    __in_opt PWDFMEMORY_OFFSET WriteOffsets
    )
{
    m_CompletionParams.Type = WdfRequestTypeWrite;
 
    if (WriteMemory != NULL) {
        m_CompletionParams.Parameters.Write.Buffer = WriteMemory->GetHandle();
    }
 
    if (WriteOffsets != NULL) {
        m_CompletionParams.Parameters.Write.Offset =
            WriteOffsets->BufferOffset;
    }
    else {
        m_CompletionParams.Parameters.Write.Offset = 0;
    }
}
 
VOID
FxRequestContext::FormatReadParams(
    __in_opt IFxMemory* ReadMemory,
    __in_opt PWDFMEMORY_OFFSET ReadOffsets
    )
{
    m_CompletionParams.Type = WdfRequestTypeRead;
 
    if (ReadMemory != NULL) {
        m_CompletionParams.Parameters.Read.Buffer = ReadMemory->GetHandle();
    }
 
    if (ReadOffsets != NULL) {
        m_CompletionParams.Parameters.Read.Offset =
            ReadOffsets->BufferOffset;
    }
    else {
        m_CompletionParams.Parameters.Read.Offset = 0;
    }
}
 
VOID
FxRequestContext::FormatOtherParams(
    __in FxInternalIoctlParams *InternalIoctlParams
    )
{
    m_CompletionParams.Type = WdfRequestTypeOther;
    m_CompletionParams.Parameters.Others.Argument1.Ptr = InternalIoctlParams->Argument1;
    m_CompletionParams.Parameters.Others.Argument2.Ptr = InternalIoctlParams->Argument2;
    m_CompletionParams.Parameters.Others.Argument4.Ptr = InternalIoctlParams->Argument4;
}