fxdrivercallbacks.hpp

Go to the documentation of this file.

/*++
 
Copyright (c) Microsoft Corporation
 
Module Name:
 
    FxDriverCallbacks.h
 
Abstract:
 
    This module implements the FxDriver object callbacks
 
Author:
 
 
 
 
Environment:
 
    Both kernel and user mode
 
Revision History:
 
--*/
 
#ifndef _FXDRIVERCALLBACKS_H_
#define _FXDRIVERCALLBACKS_H_
 
//
// These delegates are in a seperate file
//
 
//
// DrvDeviceInitialize callback delegate
//
class FxDriverDeviceAdd : public FxLockedCallback {
 
public:
    PFN_WDF_DRIVER_DEVICE_ADD Method;
 
    FxDriverDeviceAdd(
        VOID
        ) :
        FxLockedCallback()
    {
        Method = NULL;
    }
 
    _Must_inspect_result_
    NTSTATUS
    Invoke(
        __in WDFDRIVER  Driver,
        __in PWDFDEVICE_INIT DeviceInit
        )
    {
        if (Method != NULL) {
            NTSTATUS status;
            KIRQL irql = 0;
 
            CallbackStart(&irql);
            status = Method(Driver, DeviceInit);
            CallbackEnd(irql);
 
            return status;
        }
        else {
            return STATUS_UNSUCCESSFUL;
        }
    }
};
 
//
// DrvUnload callback delegate
//
class FxDriverUnload : public FxCallback {
 
public:
 
    PFN_WDF_DRIVER_UNLOAD Method;
 
    FxDriverUnload(
        VOID
        ) :
        FxCallback()
    {
        Method = NULL;
    }
 
    void
    Invoke(
        __in WDFDRIVER Driver
        )
    {
        if (Method != NULL) {
 
            CallbackStart();
            Method(Driver);
            CallbackEnd();
            return;
        }
        else {
            return;
        }
    }
};
 
#endif // _FXDRIVERCALLBACKS_H_