1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
/*++
Copyright (c) Microsoft Corporation. All rights reserved.
Module Name:
simbatt.h
Abstract:
This is the header file for the simulated battery driver.
N.B. This code is provided "AS IS" without any expressed or implied warranty.
--*/
//---------------------------------------------------------------------- Pragmas
#pragma once
//--------------------------------------------------------------------- Includes
#include <wdm.h>
#include <wdf.h>
#include <batclass.h>
#include <wmistr.h>
#include <wmilib.h>
#include <ntstrsafe.h>
//------------------------------------------------------------- Debug Facilities
#define SIMBATT_ERROR DPFLTR_ERROR_LEVEL // ed Kd_IHVDRIVER_Mask 0x1
#define SIMBATT_WARN DPFLTR_WARNING_LEVEL // ed Kd_IHVDRIVER_Mask 0x2
#define SIMBATT_TRACE DPFLTR_TRACE_LEVEL // ed Kd_IHVDRIVER_Mask 0x4
#define SIMBATT_INFO DPFLTR_INFO_LEVEL // ed Kd_IHVDRIVER_Mask 0x8
#if defined(DEBUGPRINT)
#define DebugPrint(_Level, _Msg, ...) \
SimBattPrint(_Level, _Msg, __VA_ARGS__)
#define DebugEnter() \
DebugPrint(SIMBATT_TRACE, "Entering " __FUNCTION__ "\n")
#define DebugExit() \
DebugPrint(SIMBATT_TRACE, "Leaving " __FUNCTION__ "\n")
#define DebugExitStatus(_status_) \
DebugPrint(SIMBATT_TRACE, \
"Leaving " __FUNCTION__ ": Status=0x%x\n", \
_status_)
#else
#define DebugPrint(l, m, ...)
#define DebugEnter()
#define DebugExit()
#define DebugExitStatus(_status_)
#endif
//--------------------------------------------------------------------- Literals
#define SIMBATT_TAG 'StaB'
#define SIMBATT_STATE_REG_NAME L"SimbattState"
//------------------------------------------------------------------ Definitions
typedef struct {
UNICODE_STRING RegistryPath;
} SIMBATT_GLOBAL_DATA, *PSIMBATT_GLOBAL_DATA;
#define SIMBATT_STATE_VERSION_1 1
#define SIMBATT_STATE_VERSION SIMBATT_STATE_VERSION_1
typedef struct {
USHORT Version;
BATTERY_MANUFACTURE_DATE ManufactureDate;
BATTERY_INFORMATION BatteryInfo;
BATTERY_STATUS BatteryStatus;
BATTERY_REPORTING_SCALE GranularityScale[4];
ULONG GranularityCount;
ULONG EstimatedTime;
ULONG Temperature;
ULONG MaxCurrentDraw;
WCHAR DeviceName[MAX_BATTERY_STRING_SIZE];
WCHAR ManufacturerName[MAX_BATTERY_STRING_SIZE];
WCHAR SerialNumber[MAX_BATTERY_STRING_SIZE];
WCHAR UniqueId[MAX_BATTERY_STRING_SIZE];
} SIMBATT_STATE, *PSIMBATT_STATE;
typedef struct {
//
// Battery class registration
//
PVOID ClassHandle;
WDFWAITLOCK ClassInitLock;
WMILIB_CONTEXT WmiLibContext;
//
// Battery state
//
WDFWAITLOCK StateLock;
ULONG BatteryTag;
SIMBATT_STATE State;
} SIMBATT_FDO_DATA, *PSIMBATT_FDO_DATA;
//------------------------------------------------------ WDF Context Declaration
WDF_DECLARE_CONTEXT_TYPE_WITH_NAME(SIMBATT_GLOBAL_DATA, GetGlobalData);
WDF_DECLARE_CONTEXT_TYPE_WITH_NAME(SIMBATT_FDO_DATA, GetDeviceExtension);
//----------------------------------------------------- Prototypes (miniclass.c)
_IRQL_requires_same_
VOID
SimBattPrepareHardware (
_In_ WDFDEVICE Device
);
EVT_WDF_IO_QUEUE_IO_DEVICE_CONTROL SimBattIoDeviceControl;
BCLASS_QUERY_TAG_CALLBACK SimBattQueryTag;
BCLASS_QUERY_INFORMATION_CALLBACK SimBattQueryInformation;
BCLASS_SET_INFORMATION_CALLBACK SimBattSetInformation;
BCLASS_QUERY_STATUS_CALLBACK SimBattQueryStatus;
BCLASS_SET_STATUS_NOTIFY_CALLBACK SimBattSetStatusNotify;
BCLASS_DISABLE_STATUS_NOTIFY_CALLBACK SimBattDisableStatusNotify;
_IRQL_requires_same_
VOID
SimBattPrint (
_In_ ULONG Level,
_In_ PCSTR Format,
...
);
|
