/*++

Copyright (c) 1990-2000 Microsoft Corporation, All Rights Reserved

Module Name:

    toastmon.h

Abstract:

Environment:

    Kernel mode

--*/

#if     !defined(__TOASTMON_H__)
#define __TOASTMON_H__

#include <wdm.h>
#include <wdf.h>

#define DRIVER_TAG              'NOMT' // "TMON"
#define READ_BUF_SIZE           100
#define WRITE_BUF_SIZE          120

typedef struct _DEVICE_EXTENSION {

    WDFDEVICE           WdfDevice;
    PVOID               NotificationHandle; // Interface notification handle
    WDFCOLLECTION       TargetDeviceCollection;
    WDFWAITLOCK         TargetDeviceCollectionLock;
    PVOID               WMIDeviceArrivalNotificationObject;

} DEVICE_EXTENSION, *PDEVICE_EXTENSION;

WDF_DECLARE_CONTEXT_TYPE_WITH_NAME(DEVICE_EXTENSION, GetDeviceExtension)


typedef struct  _TARGET_DEVICE_INFO {

    PDEVICE_EXTENSION DeviceExtension; // Our FDO device extension
    WDFREQUEST        ReadRequest;
    WDFREQUEST        WriteRequest;
    WDFTIMER          TimerForPostingRequests;

} TARGET_DEVICE_INFO, *PTARGET_DEVICE_INFO;

WDF_DECLARE_CONTEXT_TYPE_WITH_NAME(TARGET_DEVICE_INFO, GetTargetDeviceInfo)


typedef struct _TIMER_CONTEXT {

   WDFIOTARGET IoTarget;

} TIMER_CONTEXT, *PTIMER_CONTEXT ;

WDF_DECLARE_CONTEXT_TYPE_WITH_NAME(TIMER_CONTEXT, GetTimerContext)


/********************* function prototypes ***********************************/

DRIVER_INITIALIZE                       DriverEntry;
EVT_WDF_DRIVER_DEVICE_ADD               ToastMon_EvtDeviceAdd;
EVT_WDF_DEVICE_CONTEXT_CLEANUP          ToastMon_EvtDeviceContextCleanup;
EVT_WDF_DEVICE_D0_ENTRY                 ToastMon_EvtDeviceD0Entry;
EVT_WDF_DEVICE_D0_EXIT                  ToastMon_EvtDeviceD0Exit;
EVT_WDF_IO_TARGET_QUERY_REMOVE          ToastMon_EvtIoTargetQueryRemove;
EVT_WDF_IO_TARGET_REMOVE_CANCELED       ToastMon_EvtIoTargetRemoveCanceled;
EVT_WDF_IO_TARGET_REMOVE_COMPLETE       ToastMon_EvtIoTargetRemoveComplete;
EVT_WDF_TIMER                           Toastmon_EvtTimerPostRequests;
EVT_WDF_REQUEST_COMPLETION_ROUTINE      Toastmon_ReadRequestCompletionRoutine;
EVT_WDF_REQUEST_COMPLETION_ROUTINE      Toastmon_WriteRequestCompletionRoutine;
DRIVER_NOTIFICATION_CALLBACK_ROUTINE    ToastMon_PnpNotifyInterfaceChange;
FWMI_NOTIFICATION_CALLBACK              WmiNotificationCallback;

NTSTATUS
Toastmon_OpenDevice(
    _In_  WDFDEVICE         Device,
    _In_  PUNICODE_STRING   SymbolicLink,
    _Out_ WDFIOTARGET*      Target
    );

NTSTATUS
ToastMon_PostReadRequests(
    _In_ WDFIOTARGET IoTarget
    );

NTSTATUS
ToastMon_PostWriteRequests(
    _In_ WDFIOTARGET IoTarget
    );

NTSTATUS
RegisterForWMINotification(
    _Inout_ PDEVICE_EXTENSION DeviceExt
    );

VOID
UnregisterForWMINotification(
    _Inout_ PDEVICE_EXTENSION DeviceExt
    );

NTSTATUS
GetTargetFriendlyName(
    _In_  WDFIOTARGET   Target,
    _Out_ WDFMEMORY*    TargetName
    );

#endif