ReactOS 0.4.15-dev-7953-g1f49173
Functions
event.c File Reference
#include <afd.h>
Include dependency graph for event.c:

Go to the source code of this file.

Functions

NTSTATUS AfdEventError (IN PVOID TdiEventContext, IN NTSTATUS Status)
 
NTSTATUS AfdEventDisconnect (IN PVOID TdiEventContext, IN CONNECTION_CONTEXT ConnectionContext, IN LONG DisconnectDataLength, IN PVOID DisconnectData, IN LONG DisconnectInformationLength, IN PVOID DisconnectInformation, IN ULONG DisconnectFlags)
 
NTSTATUS AfdEventReceive (IN PVOID TdiEventContext, IN CONNECTION_CONTEXT ConnectionContext, IN ULONG ReceiveFlags, IN ULONG BytesIndicated, IN ULONG BytesAvailable, OUT ULONG *BytesTaken, IN PVOID Tsdu, OUT PIRP *IoRequestPacket)
 
TDI_STATUS ClientEventReceiveExpedited (IN PVOID TdiEventContext, IN CONNECTION_CONTEXT ConnectionContext, IN ULONG ReceiveFlags, IN ULONG BytesIndicated, IN ULONG BytesAvailable, OUT ULONG *BytesTaken, IN PVOID Tsdu, OUT PIRP *IoRequestPacket)
 
NTSTATUS ClientEventChainedReceive (IN PVOID TdiEventContext, IN CONNECTION_CONTEXT ConnectionContext, IN ULONG ReceiveFlags, IN ULONG ReceiveLength, IN ULONG StartingOffset, IN PMDL Tsdu, IN PVOID TsduDescriptor)
 
NTSTATUS AfdEventReceiveDatagramHandler (IN PVOID TdiEventContext, IN LONG SourceAddressLength, IN PVOID SourceAddress, IN LONG OptionsLength, IN PVOID Options, IN ULONG ReceiveDatagramFlags, IN ULONG BytesIndicated, IN ULONG BytesAvailable, OUT ULONG *BytesTaken, IN PVOID Tsdu, OUT PIRP *IoRequestPacket)
 
NTSTATUS AfdRegisterEventHandlers (PAFDFCB FCB)
 
NTSTATUS AfdDeregisterEventHandlers (PAFDFCB FCB)
 

Function Documentation

◆ AfdDeregisterEventHandlers()

NTSTATUS AfdDeregisterEventHandlers ( PAFDFCB  FCB)

Definition at line 274 of file event.c.

276{
277 NTSTATUS Status;
278
279 Status = TdiSetEventHandler(FCB->TdiAddressObject,
280 TDI_EVENT_ERROR,
281 NULL,
282 NULL);
283 if (!NT_SUCCESS(Status)) { return Status; }
284
285 switch (FCB->SocketType) {
286 case SOCK_STREAM:
287 Status = TdiSetEventHandler(FCB->TdiAddressObject,
288 TDI_EVENT_DISCONNECT,
289 NULL,
290 NULL);
291 if (!NT_SUCCESS(Status)) { return Status; }
292
293 Status = TdiSetEventHandler(FCB->TdiAddressObject,
294 TDI_EVENT_RECEIVE,
295 NULL,
296 NULL);
297 if (!NT_SUCCESS(Status)) { return Status; }
298
299 Status = TdiSetEventHandler(FCB->TdiAddressObject,
300 TDI_EVENT_RECEIVE_EXPEDITED,
301 NULL,
302 NULL);
303 if (!NT_SUCCESS(Status)) { return Status; }
304
305 Status = TdiSetEventHandler(FCB->TdiAddressObject,
306 TDI_EVENT_CHAINED_RECEIVE,
307 NULL,
308 NULL);
309 if (!NT_SUCCESS(Status)) { return Status; }
310 break;
311
312 case SOCK_DGRAM:
313 case SOCK_RAW:
314 Status = TdiSetEventHandler(FCB->TdiAddressObject,
315 TDI_EVENT_RECEIVE_DATAGRAM,
316 NULL,
317 NULL);
318 if (!NT_SUCCESS(Status)) { return Status; }
319 }
320 return STATUS_SUCCESS;
321}
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
NULL
#define NULL
Definition: types.h:112
NT_SUCCESS
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:32
TdiSetEventHandler
NTSTATUS TdiSetEventHandler(PFILE_OBJECT FileObject, LONG EventType, PVOID Handler, PVOID Context)
Definition: tdi.c:546
SOCK_STREAM
#define SOCK_STREAM
Definition: tcpip.h:118
Status
Status
Definition: gdiplustypes.h:25
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
_FCB
Definition: cdstruc.h:902
TDI_EVENT_ERROR
#define TDI_EVENT_ERROR
Definition: tdikrnl.h:148
TDI_EVENT_RECEIVE_EXPEDITED
#define TDI_EVENT_RECEIVE_EXPEDITED
Definition: tdikrnl.h:151
TDI_EVENT_CHAINED_RECEIVE
#define TDI_EVENT_CHAINED_RECEIVE
Definition: tdikrnl.h:153
TDI_EVENT_DISCONNECT
#define TDI_EVENT_DISCONNECT
Definition: tdikrnl.h:147
TDI_EVENT_RECEIVE
#define TDI_EVENT_RECEIVE
Definition: tdikrnl.h:149
TDI_EVENT_RECEIVE_DATAGRAM
#define TDI_EVENT_RECEIVE_DATAGRAM
Definition: tdikrnl.h:150
SOCK_RAW
#define SOCK_RAW
Definition: winsock.h:337
SOCK_DGRAM
#define SOCK_DGRAM
Definition: winsock.h:336

◆ AfdEventDisconnect()

NTSTATUS AfdEventDisconnect ( IN PVOID  TdiEventContext,
IN CONNECTION_CONTEXT  ConnectionContext,
IN LONG  DisconnectDataLength,
IN PVOID  DisconnectData,
IN LONG  DisconnectInformationLength,
IN PVOID  DisconnectInformation,
IN ULONG  DisconnectFlags 
)

Definition at line 23 of file event.c.

31{
32 AFD_DbgPrint(MAX_TRACE, ("Called.\n"));
33
34 return STATUS_SUCCESS;
35}
MAX_TRACE
#define MAX_TRACE
Definition: debug.h:16
AFD_DbgPrint
#define AFD_DbgPrint(_t_, _x_)
Definition: debug.h:60

Referenced by AfdRegisterEventHandlers().

◆ AfdEventError()

NTSTATUS AfdEventError ( IN PVOID  TdiEventContext,
IN NTSTATUS  Status 
)

Definition at line 13 of file event.c.

16{
17 AFD_DbgPrint(MAX_TRACE, ("Called.\n"));
18
19 return STATUS_SUCCESS;
20}

Referenced by AfdRegisterEventHandlers().

◆ AfdEventReceive()

NTSTATUS AfdEventReceive ( IN PVOID  TdiEventContext,
IN CONNECTION_CONTEXT  ConnectionContext,
IN ULONG  ReceiveFlags,
IN ULONG  BytesIndicated,
IN ULONG  BytesAvailable,
OUT ULONG *  BytesTaken,
IN PVOID  Tsdu,
OUT PIRP *  IoRequestPacket 
)

Definition at line 38 of file event.c.

47{
48 PAFDFCB FCB = (PAFDFCB)TdiEventContext;
49 PVOID ReceiveBuffer;
50 PAFD_BUFFER Buffer;
51 KIRQL OldIrql;
52
53 AFD_DbgPrint(MAX_TRACE, ("Called.\n"));
54
55 AFD_DbgPrint(MID_TRACE, ("Receiving (%d) bytes on socket\n",
56 BytesAvailable));
57
58 ReceiveBuffer = ExAllocatePoolWithTag(NonPagedPool,
59 BytesAvailable,
60 TAG_AFD_DATA_BUFFER);
61
62 if (!ReceiveBuffer)
63 return STATUS_INSUFFICIENT_RESOURCES;
64
65 /*Buffer = (PAFD_BUFFER)ExAllocateFromNPagedLookasideList(
66 &BufferLookasideList);*/
67 Buffer = (PAFD_BUFFER)ExAllocatePoolWithTag(NonPagedPool,
68 sizeof(AFD_BUFFER),
69 TAG_AFD_DATA_BUFFER);
70
71 if (!Buffer) {
72 ExFreePoolWithTag(ReceiveBuffer, TAG_AFD_DATA_BUFFER);
73 return STATUS_INSUFFICIENT_RESOURCES;
74 }
75
76 /* Copy the data to a local buffer */
77 RtlCopyMemory(ReceiveBuffer, Tsdu, BytesAvailable);
78
79 Buffer->Buffer.len = BytesAvailable;
80 Buffer->Buffer.buf = ReceiveBuffer;
81 Buffer->Offset = 0;
82
83 KeAcquireSpinLock(&FCB->ReceiveQueueLock, &OldIrql);
84
85 InsertTailList( &FCB->ReceiveQueue, &Buffer->ListEntry );
86
87 TryToSatisfyRecvRequest( FCB, TRUE );
88
89 KeReleaseSpinLock(&FCB->ReceiveQueueLock, OldIrql);
90
91 *BytesTaken = BytesAvailable;
92
93 AFD_DbgPrint(MAX_TRACE, ("Leaving.\n"));
94
95 return STATUS_SUCCESS;
96}
TAG_AFD_DATA_BUFFER
#define TAG_AFD_DATA_BUFFER
Definition: afd.h:38
MID_TRACE
#define MID_TRACE
Definition: debug.h:15
Buffer
Definition: bufpool.h:45
TRUE
#define TRUE
Definition: types.h:120
InsertTailList
#define InsertTailList(ListHead, Entry)
Definition: env_spec_w32.h:1003
ExAllocatePoolWithTag
#define ExAllocatePoolWithTag(hernya, size, tag)
Definition: env_spec_w32.h:350
KIRQL
UCHAR KIRQL
Definition: env_spec_w32.h:591
KeReleaseSpinLock
#define KeReleaseSpinLock(sl, irql)
Definition: env_spec_w32.h:627
KeAcquireSpinLock
#define KeAcquireSpinLock(sl, irql)
Definition: env_spec_w32.h:609
NonPagedPool
#define NonPagedPool
Definition: env_spec_w32.h:307
ExFreePoolWithTag
#define ExFreePoolWithTag(_P, _T)
Definition: module.h:1109
RtlCopyMemory
#define RtlCopyMemory(Destination, Source, Length)
Definition: typedefs.h:263
STATUS_INSUFFICIENT_RESOURCES
#define STATUS_INSUFFICIENT_RESOURCES
Definition: udferr_usr.h:158
OldIrql
_Requires_lock_held_ Interrupt _Releases_lock_ Interrupt _In_ _IRQL_restores_ KIRQL OldIrql
Definition: kefuncs.h:778

Referenced by AfdRegisterEventHandlers().

◆ AfdEventReceiveDatagramHandler()

NTSTATUS AfdEventReceiveDatagramHandler ( IN PVOID  TdiEventContext,
IN LONG  SourceAddressLength,
IN PVOID  SourceAddress,
IN LONG  OptionsLength,
IN PVOID  Options,
IN ULONG  ReceiveDatagramFlags,
IN ULONG  BytesIndicated,
IN ULONG  BytesAvailable,
OUT ULONG *  BytesTaken,
IN PVOID  Tsdu,
OUT PIRP *  IoRequestPacket 
)

Definition at line 130 of file event.c.

142{
143 PAFDFCB FCB = (PAFDFCB)TdiEventContext;
144 PAFD_READ_REQUEST ReadRequest;
145 PVOID ReceiveBuffer;
146 PAFD_BUFFER Buffer;
147 PLIST_ENTRY Entry;
148 NTSTATUS Status;
149 KIRQL OldIrql;
150 ULONG Count;
151 BOOLEAN CompleteIrp;
152
153 AFD_DbgPrint(MAX_TRACE, ("Called.\n"));
154
155 AFD_DbgPrint(MID_TRACE, ("Receiving (%d) bytes from (0x%X).\n",
156 BytesAvailable, *(PULONG)SourceAddress));
157
158 ReceiveBuffer = ExAllocatePoolWithTag(NonPagedPool,
159 BytesAvailable,
160 TAG_AFD_DATA_BUFFER);
161
162 if (!ReceiveBuffer)
163 return STATUS_INSUFFICIENT_RESOURCES;
164
165 /*Buffer = (PAFD_BUFFER)ExAllocateFromNPagedLookasideList(
166 &BufferLookasideList);*/
167 Buffer = (PAFD_BUFFER)ExAllocatePoolWithTag(NonPagedPool,
168 sizeof(AFD_BUFFER),
169 TAG_AFD_DATA_BUFFER);
170
171 if (!Buffer) {
172 ExFreePoolWithTag(ReceiveBuffer, TAG_AFD_DATA_BUFFER);
173 return STATUS_INSUFFICIENT_RESOURCES;
174 }
175
176 /* Copy the data to a local buffer */
177 RtlCopyMemory(ReceiveBuffer, Tsdu, BytesAvailable);
178
179 Buffer->Buffer.len = BytesAvailable;
180 Buffer->Buffer.buf = ReceiveBuffer;
181 Buffer->Offset = 0;
182
183 ExInterlockedInsertTailList(&FCB->ReceiveQueue,
184 &Buffer->ListEntry,
185 &FCB->ReceiveQueueLock);
186
187 KeAcquireSpinLock(&FCB->ReceiveQueueLock, &OldIrql);
188
189 if (CompleteIrp = !IsListEmpty(&FCB->ReadRequestQueue)) {
190 AFD_DbgPrint(MAX_TRACE, ("Satisfying read request.\n"));
191
192 Entry = RemoveHeadList(&FCB->ReceiveQueue);
193 ReadRequest = CONTAINING_RECORD(Entry, AFD_READ_REQUEST, ListEntry);
194
195 Status = FillWSABuffers(FCB,
196 ReadRequest->RecvFromRequest->Buffers,
197 ReadRequest->RecvFromRequest->BufferCount,
198 &Count,
199 FALSE); /* Packet oriented */
200 ReadRequest->RecvFromReply->NumberOfBytesRecvd = Count;
201 ReadRequest->RecvFromReply->Status = NO_ERROR;
202
203 ReadRequest->Irp->IoStatus.Information = 0;
204 ReadRequest->Irp->IoStatus.Status = Status;
205 }
206
207 KeReleaseSpinLock(&FCB->ReceiveQueueLock, OldIrql);
208
209 if (CompleteIrp) {
210 AFD_DbgPrint(MAX_TRACE, ("Completing IRP at (0x%X).\n", ReadRequest->Irp));
211 IoCompleteRequest(ReadRequest->Irp, IO_NETWORK_INCREMENT);
212 *BytesTaken = BytesAvailable;
213 }
214
215 AFD_DbgPrint(MAX_TRACE, ("Leaving.\n"));
216
217 return STATUS_SUCCESS;
218}
BOOLEAN
unsigned char BOOLEAN
Definition: ProcessorBind.h:185
NO_ERROR
#define NO_ERROR
Definition: dderror.h:5
FALSE
#define FALSE
Definition: types.h:117
IO_NETWORK_INCREMENT
#define IO_NETWORK_INCREMENT
Definition: tcpip.h:43
CompleteIrp
VOID CompleteIrp(IN PIRP Irp, IN NTSTATUS Status, IN ULONG_PTR Information)
Definition: pnp.c:12
IsListEmpty
#define IsListEmpty(ListHead)
Definition: env_spec_w32.h:954
RemoveHeadList
#define RemoveHeadList(ListHead)
Definition: env_spec_w32.h:964
ExInterlockedInsertTailList
PLIST_ENTRY NTAPI ExInterlockedInsertTailList(IN OUT PLIST_ENTRY ListHead, IN OUT PLIST_ENTRY ListEntry, IN OUT PKSPIN_LOCK Lock)
Definition: interlocked.c:140
Count
int Count
Definition: noreturn.cpp:7
IoCompleteRequest
#define IoCompleteRequest
Definition: irp.c:1240
Entry
base of all file and directory entries
Definition: entries.h:83
_LIST_ENTRY
Definition: typedefs.h:120
PULONG
uint32_t * PULONG
Definition: typedefs.h:59
CONTAINING_RECORD
#define CONTAINING_RECORD(address, type, field)
Definition: typedefs.h:260
ULONG
uint32_t ULONG
Definition: typedefs.h:59
SourceAddress
_Must_inspect_result_ typedef _In_ PHYSICAL_ADDRESS SourceAddress
Definition: iotypes.h:1127

Referenced by AfdRegisterEventHandlers().

◆ AfdRegisterEventHandlers()

NTSTATUS AfdRegisterEventHandlers ( PAFDFCB  FCB)

Definition at line 221 of file event.c.

223{
224 NTSTATUS Status;
225
226 assert(FCB->TdiAddressObject);
227
228 /* Report errors for all types of sockets */
229 Status = TdiSetEventHandler(FCB->TdiAddressObject,
230 TDI_EVENT_ERROR,
231 (PVOID)AfdEventError,
232 (PVOID)FCB);
233 if (!NT_SUCCESS(Status)) { return Status; }
234
235 switch (FCB->SocketType) {
236 case SOCK_STREAM:
237 Status = TdiSetEventHandler(FCB->TdiAddressObject,
238 TDI_EVENT_DISCONNECT,
239 (PVOID)AfdEventDisconnect,
240 (PVOID)FCB);
241 if (!NT_SUCCESS(Status)) { return Status; }
242
243 Status = TdiSetEventHandler(FCB->TdiAddressObject,
244 TDI_EVENT_RECEIVE,
245 (PVOID)AfdEventReceive,
246 (PVOID)FCB);
247 if (!NT_SUCCESS(Status)) { return Status; }
248
249 Status = TdiSetEventHandler(FCB->TdiAddressObject,
250 TDI_EVENT_RECEIVE_EXPEDITED,
251 (PVOID)ClientEventReceiveExpedited,
252 (PVOID)FCB);
253 if (!NT_SUCCESS(Status)) { return Status; }
254
255 Status = TdiSetEventHandler(FCB->TdiAddressObject,
256 TDI_EVENT_CHAINED_RECEIVE,
257 (PVOID)ClientEventChainedReceive,
258 (PVOID)FCB);
259 if (!NT_SUCCESS(Status)) { return Status; }
260 break;
261
262 case SOCK_DGRAM:
263 case SOCK_RAW:
264 Status = TdiSetEventHandler(FCB->TdiAddressObject,
265 TDI_EVENT_RECEIVE_DATAGRAM,
266 (PVOID)AfdEventReceiveDatagramHandler,
267 (PVOID)FCB);
268 if (!NT_SUCCESS(Status)) { return Status; }
269 }
270 return STATUS_SUCCESS;
271}
assert
#define assert(x)
Definition: debug.h:53
ClientEventChainedReceive
NTSTATUS ClientEventChainedReceive(IN PVOID TdiEventContext, IN CONNECTION_CONTEXT ConnectionContext, IN ULONG ReceiveFlags, IN ULONG ReceiveLength, IN ULONG StartingOffset, IN PMDL Tsdu, IN PVOID TsduDescriptor)
Definition: event.c:115
AfdEventError
NTSTATUS AfdEventError(IN PVOID TdiEventContext, IN NTSTATUS Status)
Definition: event.c:13
AfdEventDisconnect
NTSTATUS AfdEventDisconnect(IN PVOID TdiEventContext, IN CONNECTION_CONTEXT ConnectionContext, IN LONG DisconnectDataLength, IN PVOID DisconnectData, IN LONG DisconnectInformationLength, IN PVOID DisconnectInformation, IN ULONG DisconnectFlags)
Definition: event.c:23
AfdEventReceiveDatagramHandler
NTSTATUS AfdEventReceiveDatagramHandler(IN PVOID TdiEventContext, IN LONG SourceAddressLength, IN PVOID SourceAddress, IN LONG OptionsLength, IN PVOID Options, IN ULONG ReceiveDatagramFlags, IN ULONG BytesIndicated, IN ULONG BytesAvailable, OUT ULONG *BytesTaken, IN PVOID Tsdu, OUT PIRP *IoRequestPacket)
Definition: event.c:130
AfdEventReceive
NTSTATUS AfdEventReceive(IN PVOID TdiEventContext, IN CONNECTION_CONTEXT ConnectionContext, IN ULONG ReceiveFlags, IN ULONG BytesIndicated, IN ULONG BytesAvailable, OUT ULONG *BytesTaken, IN PVOID Tsdu, OUT PIRP *IoRequestPacket)
Definition: event.c:38
ClientEventReceiveExpedited
TDI_STATUS ClientEventReceiveExpedited(IN PVOID TdiEventContext, IN CONNECTION_CONTEXT ConnectionContext, IN ULONG ReceiveFlags, IN ULONG BytesIndicated, IN ULONG BytesAvailable, OUT ULONG *BytesTaken, IN PVOID Tsdu, OUT PIRP *IoRequestPacket)
Definition: event.c:99

◆ ClientEventChainedReceive()

NTSTATUS ClientEventChainedReceive ( IN PVOID  TdiEventContext,
IN CONNECTION_CONTEXT  ConnectionContext,
IN ULONG  ReceiveFlags,
IN ULONG  ReceiveLength,
IN ULONG  StartingOffset,
IN PMDL  Tsdu,
IN PVOID  TsduDescriptor 
)

Definition at line 115 of file event.c.

123{
124 AFD_DbgPrint(MAX_TRACE, ("Called.\n"));
125
126 return STATUS_SUCCESS;
127}

Referenced by AfdRegisterEventHandlers().

◆ ClientEventReceiveExpedited()

TDI_STATUS ClientEventReceiveExpedited ( IN PVOID  TdiEventContext,
IN CONNECTION_CONTEXT  ConnectionContext,
IN ULONG  ReceiveFlags,
IN ULONG  BytesIndicated,
IN ULONG  BytesAvailable,
OUT ULONG *  BytesTaken,
IN PVOID  Tsdu,
OUT PIRP *  IoRequestPacket 
)

Definition at line 99 of file event.c.

108{
109 AFD_DbgPrint(MAX_TRACE, ("Called.\n"));
110
111 return STATUS_SUCCESS;
112}

Referenced by AfdRegisterEventHandlers().