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
|
/*++
Copyright (c) 1989-1993 Microsoft Corporation
Module Name:
event.c
Abstract:
This module contains code which performs the following TDI services:
o TdiSetEventHandler
Environment:
Kernel mode
Revision History:
--*/
#include "precomp.h"
#pragma hdrstop
PVOID TdiDefaultHandlers[6] = {
TdiDefaultConnectHandler,
TdiDefaultDisconnectHandler,
TdiDefaultErrorHandler,
TdiDefaultReceiveHandler,
TdiDefaultRcvDatagramHandler,
TdiDefaultRcvExpeditedHandler
};
NTSTATUS
NbiTdiSetEventHandler(
IN PDEVICE Device,
IN PREQUEST Request
)
/*++
Routine Description:
This routine performs the TdiSetEventHandler request for the
transport provider. The caller (request dispatcher) verifies
that this routine will not be executed on behalf of a user-mode
client, as this request enables direct callouts at DISPATCH_LEVEL.
Arguments:
Device - The netbios device object.
Request - Pointer to the request.
Return Value:
NTSTATUS - status of operation.
--*/
{
NTSTATUS Status;
CTELockHandle LockHandle;
PTDI_REQUEST_KERNEL_SET_EVENT Parameters;
PADDRESS_FILE AddressFile;
UINT EventType;
UNREFERENCED_PARAMETER (Device);
//
// Get the Address this is associated with; if there is none, get out.
//
AddressFile = REQUEST_OPEN_CONTEXT(Request);
#if defined(_PNP_POWER)
Status = NbiVerifyAddressFile (AddressFile, CONFLICT_IS_OK);
#else
Status = NbiVerifyAddressFile (AddressFile);
#endif _PNP_POWER
if (!NT_SUCCESS (Status)) {
return Status;
}
NB_GET_LOCK (&AddressFile->Address->Lock, &LockHandle);
Parameters = (PTDI_REQUEST_KERNEL_SET_EVENT)REQUEST_PARAMETERS(Request);
EventType = (UINT)(Parameters->EventType);
if (Parameters->EventType > TDI_EVENT_RECEIVE_EXPEDITED) {
Status = STATUS_INVALID_PARAMETER;
} else {
if (Parameters->EventHandler == NULL) {
AddressFile->RegisteredHandler[EventType] = FALSE;
AddressFile->Handlers[EventType] = TdiDefaultHandlers[EventType];
AddressFile->HandlerContexts[EventType] = NULL;
} else {
AddressFile->Handlers[EventType] = Parameters->EventHandler;
AddressFile->HandlerContexts[EventType] = Parameters->EventContext;
AddressFile->RegisteredHandler[EventType] = TRUE;
}
}
NB_FREE_LOCK (&AddressFile->Address->Lock, LockHandle);
NbiDereferenceAddressFile (AddressFile, AFREF_VERIFY);
return Status;
} /* NbiTdiSetEventHandler */
|
