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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
|
/*++
Copyright (c) 1989-1993 Microsoft Corporation
Module Name:
devctx.c
Abstract:
This module contains code which implements the DEVICE_CONTEXT object.
Routines are provided to reference, and dereference transport device
context objects.
The transport device context object is a structure which contains a
system-defined DEVICE_OBJECT followed by information which is maintained
by the transport provider, called the context.
Environment:
Kernel mode
Revision History:
--*/
#include "st.h"
VOID
StRefDeviceContext(
IN PDEVICE_CONTEXT DeviceContext
)
/*++
Routine Description:
This routine increments the reference count on a device context.
Arguments:
DeviceContext - Pointer to a transport device context object.
Return Value:
none.
--*/
{
ASSERT (DeviceContext->ReferenceCount > 0); // not perfect, but...
(VOID)InterlockedIncrement (&DeviceContext->ReferenceCount);
} /* StRefDeviceContext */
VOID
StDerefDeviceContext(
IN PDEVICE_CONTEXT DeviceContext
)
/*++
Routine Description:
This routine dereferences a device context by decrementing the
reference count contained in the structure. Currently, we don't
do anything special when the reference count drops to zero, but
we could dynamically unload stuff then.
Arguments:
DeviceContext - Pointer to a transport device context object.
Return Value:
none.
--*/
{
LONG result;
result = InterlockedDecrement (&DeviceContext->ReferenceCount);
ASSERT (result >= 0);
if (result == 0) {
StDestroyDeviceContext (DeviceContext);
}
} /* StDerefDeviceContext */
NTSTATUS
StCreateDeviceContext(
IN PDRIVER_OBJECT DriverObject,
IN PUNICODE_STRING DeviceName,
IN OUT PDEVICE_CONTEXT *DeviceContext
)
/*++
Routine Description:
This routine creates and initializes a device context structure.
Arguments:
DriverObject - pointer to the IO subsystem supplied driver object.
DeviceContext - Pointer to a pointer to a transport device context object.
DeviceName - pointer to the name of the device this device object points to.
Return Value:
STATUS_SUCCESS if all is well; STATUS_INSUFFICIENT_RESOURCES otherwise.
--*/
{
NTSTATUS status;
PDEVICE_OBJECT deviceObject;
PDEVICE_CONTEXT deviceContext;
USHORT i;
//
// Create the device object for the sample transport, allowing
// room at the end for the device name to be stored (for use
// in logging errors).
//
status = IoCreateDevice(
DriverObject,
sizeof (DEVICE_CONTEXT) - sizeof (DEVICE_OBJECT) +
(DeviceName->Length + sizeof(UNICODE_NULL)),
DeviceName,
FILE_DEVICE_TRANSPORT,
0,
FALSE,
&deviceObject);
if (!NT_SUCCESS(status)) {
return status;
}
deviceObject->Flags |= DO_DIRECT_IO;
deviceContext = (PDEVICE_CONTEXT)deviceObject;
//
// Initialize our part of the device context.
//
RtlZeroMemory(
((PUCHAR)deviceContext) + sizeof(DEVICE_OBJECT),
sizeof(DEVICE_CONTEXT) - sizeof(DEVICE_OBJECT));
//
// Copy over the device name.
//
deviceContext->DeviceNameLength = DeviceName->Length + sizeof(WCHAR);
deviceContext->DeviceName = (PWCHAR)(deviceContext+1);
RtlCopyMemory(
deviceContext->DeviceName,
DeviceName->Buffer,
DeviceName->Length);
deviceContext->DeviceName[DeviceName->Length/sizeof(WCHAR)] = UNICODE_NULL;
//
// Initialize the reference count.
//
deviceContext->ReferenceCount = 1;
//
// initialize the various fields in the device context
//
KeInitializeSpinLock (&deviceContext->Interlock);
KeInitializeSpinLock (&deviceContext->SpinLock);
deviceContext->ControlChannelIdentifier = 1;
InitializeListHead (&deviceContext->ConnectionPool);
InitializeListHead (&deviceContext->AddressPool);
InitializeListHead (&deviceContext->AddressFilePool);
InitializeListHead (&deviceContext->AddressDatabase);
InitializeListHead (&deviceContext->PacketWaitQueue);
InitializeListHead (&deviceContext->PacketizeQueue);
InitializeListHead (&deviceContext->RequestPool);
deviceContext->PacketPool.Next = NULL;
deviceContext->ReceivePacketPool.Next = NULL;
deviceContext->ReceiveBufferPool.Next = NULL;
InitializeListHead (&deviceContext->ReceiveInProgress);
InitializeListHead (&deviceContext->IrpCompletionQueue);
deviceContext->State = DEVICECONTEXT_STATE_CLOSED;
//
// Initialize the resource that guards address ACLs.
//
ExInitializeResource (&deviceContext->AddressResource);
//
// set the netbios multicast address for this network type
//
for (i=0; i<HARDWARE_ADDRESS_LENGTH; i++) {
deviceContext->LocalAddress.Address [i] = 0; // set later
deviceContext->MulticastAddress.Address [i] = 0;
}
deviceContext->Type = ST_DEVICE_CONTEXT_SIGNATURE;
deviceContext->Size - sizeof (DEVICE_CONTEXT);
*DeviceContext = deviceContext;
return STATUS_SUCCESS;
}
VOID
StDestroyDeviceContext(
IN PDEVICE_CONTEXT DeviceContext
)
/*++
Routine Description:
This routine destroys a device context structure.
Arguments:
DeviceContext - Pointer to a pointer to a transport device context object.
Return Value:
None.
--*/
{
ExDeleteResource (&DeviceContext->AddressResource);
IoDeleteDevice ((PDEVICE_OBJECT)DeviceContext);
return;
}
|
