fxwmiprovider.hpp

Go to the documentation of this file.

/*++
 
Copyright (c) Microsoft Corporation
 
Module Name:
 
    FxWmiDataBlock.hpp
 
Abstract:
 
    This module implements the WMI data block object
 
Author:
 
 
 
 
Environment:
 
    Both kernel and user mode
 
Revision History:
 
 
--*/
 
#ifndef _FXWMIPROVIDER_H_
#define _FXWMIPROVIDER_H_
 
 
struct FxWmiProviderFunctionControlCallback : public FxCallback {
 
    PFN_WDF_WMI_PROVIDER_FUNCTION_CONTROL m_Method;
 
    FxWmiProviderFunctionControlCallback(
        PFX_DRIVER_GLOBALS FxDriverGlobals
        ) :
        FxCallback(FxDriverGlobals),
        m_Method(NULL)
    {
    }
 
    ~FxWmiProviderFunctionControlCallback()
    {
    }
 
    _Must_inspect_result_
    NTSTATUS
    Invoke(
        __in WDFDEVICE Device,
        __in WDFWMIPROVIDER WmiProvider,
        __in WDF_WMI_PROVIDER_CONTROL Control,
        __in BOOLEAN Enable
        )
    {
        NTSTATUS status;
 
        UNREFERENCED_PARAMETER(Device);
 
        if (m_Method != NULL) {
            CallbackStart();
            status = m_Method(WmiProvider, Control, Enable);
            CallbackEnd();
        }
        else {
            status = STATUS_SUCCESS;
        }
 
        return status;
    }
};
 
class FxWmiProvider : public FxNonPagedObject {
 
    friend FxWmiIrpHandler;
 
public:
    FxWmiProvider(
        __in PFX_DRIVER_GLOBALS FxDriverGlobals,
        __in PWDF_WMI_PROVIDER_CONFIG Config,
        __in CfxDevice* Device
        );
 
    ~FxWmiProvider();
 
    static
    _Must_inspect_result_
    NTSTATUS
    _Create(
        __in PFX_DRIVER_GLOBALS CallersGlobals,
        __in WDFDEVICE Device,
        __in_opt PWDF_OBJECT_ATTRIBUTES ProviderAttributes,
        __in PWDF_WMI_PROVIDER_CONFIG WmiProviderConfig,
        __out WDFWMIPROVIDER* WmiProvider,
        __out FxWmiProvider** Provider
        );
 
    CfxDevice*
    GetDevice(
        VOID
        )
    {
        return m_Parent->GetDevice();
    }
 
    GUID*
    GetGUID(
        VOID
        )
    {
        return &m_Guid;
    }
 
    ULONG
    GetFlags(
        VOID
        )
    {
        return m_Flags;
    }
 
    ULONG
    GetMinInstanceBufferSize(
        VOID
        )
    {
        return m_MinInstanceBufferSize;
    }
 
    ULONG
    GetRegistrationFlagsLocked(
        VOID
        );
 
    BOOLEAN
    IsEventOnly(
        VOID
        )
    {
        return FLAG_TO_BOOL(m_Flags, WdfWmiProviderEventOnly);
    }
 
    BOOLEAN
    IsEnabled(
        __in WDF_WMI_PROVIDER_CONTROL Control
        )
    {
        switch (Control) {
        case WdfWmiEventControl:    return m_EventControlEnabled;
        case WdfWmiInstanceControl: return m_DataBlockControlEnabled;
        default:                        ASSERT(FALSE); return FALSE;
        }
    }
 
    ULONGLONG
    GetTracingHandle(
        VOID
        )
    {
        return m_TracingHandle;
    }
 
    VOID
    SetTracingHandle(
        __in ULONGLONG TracingHandle
        )
    {
        m_TracingHandle = TracingHandle;
    }
 
    WDFWMIPROVIDER
    GetHandle(
        VOID
        )
    {
        return (WDFWMIPROVIDER) GetObjectHandle();
    }
 
    BOOLEAN
    IsFunctionControlSupported(
        VOID
        )
    {
        return m_FunctionControl.m_Method != NULL;
    }
 
    _Must_inspect_result_
    NTSTATUS
    FunctionControl(
        __in WDF_WMI_PROVIDER_CONTROL Control,
        __in BOOLEAN Enable
        );
 
    _Must_inspect_result_
    NTSTATUS
    AddInstance(
         __in FxWmiInstance* Instance,
         __in BOOLEAN NoErrorIfPresent = FALSE
         );
 
    VOID
    RemoveInstance(
        __in FxWmiInstance* Instance
        );
 
    ULONG
    GetInstanceIndex(
        __in FxWmiInstance* Instance
        );
 
     _Must_inspect_result_
    FxWmiInstance*
    GetInstanceReferenced(
        __in ULONG Index,
        __in PVOID Tag
        );
 
    _Must_inspect_result_
    FxWmiInstance*
    GetInstanceReferencedLocked(
        __in ULONG Index,
        __in PVOID Tag
        );
 
    // begin FxObject overrides
    virtual
    BOOLEAN
    Dispose(
        VOID
        );
    // end FxObject overrides
 
protected:
    enum AddInstanceAction {
        AddInstanceToTail,
        AddInstanceToHead
    };
 
    _Must_inspect_result_
    NTSTATUS
    AddInstanceLocked(
        __in  FxWmiInstance* Instance,
        __in  BOOLEAN NoErrorIfPresent,
        __out PBOOLEAN Update,
        __in  AddInstanceAction Action = AddInstanceToTail
        );
 
protected:
    //
    // List entry used by FxWmiIrpHandler to hold the list of WMI providers
    //
    LIST_ENTRY m_ListEntry;
 
    LIST_ENTRY m_InstanceListHead;
 
    ULONG m_NumInstances;
 
    FxWmiIrpHandler* m_Parent;
 
    GUID m_Guid;
 
 
 
 
 
    ULONGLONG m_TracingHandle;
 
    ULONG m_MinInstanceBufferSize;
 
    union {
        //
        // Set with values from WDF_WMI_PROVIDER_FLAGS
        //
        ULONG m_Flags;
 
        //
        // Not used by the code, but by the debug extension
        //
        struct {
            ULONG EventOnly : 1;
            ULONG Expensive : 1;
            ULONG Tracing : 1;
        } m_FlagsByName;
    };
 
    FxWmiProviderFunctionControlCallback m_FunctionControl;
 
    BOOLEAN m_EventControlEnabled;
 
    BOOLEAN m_DataBlockControlEnabled;
 
    BOOLEAN m_RemoveGuid;
};
 
#endif // _FXWMIPROVIDER_H_