ReactOS 0.4.16-dev-1946-g52006dd
Functions
info.c File Reference
#include "afd.h"
Include dependency graph for info.c:

Go to the source code of this file.

Functions

NTSTATUS NTAPI AfdGetInfo (PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
 
NTSTATUS NTAPI AfdSetInfo (PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
 
NTSTATUS NTAPI AfdGetSockName (PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
 
NTSTATUS NTAPI AfdGetPeerName (PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
 

Function Documentation

◆ AfdGetInfo()

NTSTATUS NTAPI AfdGetInfo ( PDEVICE_OBJECT  DeviceObject,
PIRP  Irp,
PIO_STACK_LOCATION  IrpSp 
)

Definition at line 14 of file info.c.

15 {
16 NTSTATUS Status = STATUS_SUCCESS;
17 PAFD_INFO InfoReq = LockRequest(Irp, IrpSp, TRUE, NULL);
18 PFILE_OBJECT FileObject = IrpSp->FileObject;
19 PAFD_FCB FCB = FileObject->FsContext;
20 PLIST_ENTRY CurrentEntry;
21
22 UNREFERENCED_PARAMETER(DeviceObject);
23
24 AFD_DbgPrint(MID_TRACE,("Called %p %x\n", InfoReq,
25 InfoReq ? InfoReq->InformationClass : 0));
26
27 if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
28
29 if (!InfoReq)
30 return UnlockAndMaybeComplete(FCB, STATUS_NO_MEMORY, Irp, 0);
31
32 _SEH2_TRY {
33 switch( InfoReq->InformationClass ) {
34 case AFD_INFO_RECEIVE_WINDOW_SIZE:
35 InfoReq->Information.Ulong = FCB->Recv.Size;
36 break;
37
38 case AFD_INFO_SEND_WINDOW_SIZE:
39 InfoReq->Information.Ulong = FCB->Send.Size;
40 AFD_DbgPrint(MID_TRACE,("Send window size %u\n", FCB->Send.Size));
41 break;
42
43 case AFD_INFO_GROUP_ID_TYPE:
44 InfoReq->Information.LargeInteger.u.HighPart = FCB->GroupType;
45 InfoReq->Information.LargeInteger.u.LowPart = FCB->GroupID;
46 AFD_DbgPrint(MID_TRACE, ("Group ID: %u Group Type: %u\n", FCB->GroupID, FCB->GroupType));
47 break;
48
49 case AFD_INFO_BLOCKING_MODE:
50 InfoReq->Information.Boolean = FCB->NonBlocking;
51 break;
52
53 case AFD_INFO_INLINING_MODE:
54 InfoReq->Information.Boolean = FCB->OobInline;
55 break;
56
57 case AFD_INFO_RECEIVE_CONTENT_SIZE:
58 InfoReq->Information.Ulong = FCB->Recv.Content - FCB->Recv.BytesUsed;
59 break;
60
61 case AFD_INFO_SENDS_IN_PROGRESS:
62 InfoReq->Information.Ulong = 0;
63
64 /* Count the queued sends */
65 CurrentEntry = FCB->PendingIrpList[FUNCTION_SEND].Flink;
66 while (CurrentEntry != &FCB->PendingIrpList[FUNCTION_SEND])
67 {
68 InfoReq->Information.Ulong++;
69 CurrentEntry = CurrentEntry->Flink;
70 }
71
72 /* This needs to count too because when this is dispatched
73 * the user-mode IRP has already been completed and therefore
74 * will NOT be in our pending IRP list. We count this as one send
75 * outstanding although it could be multiple since we batch sends
76 * when waiting for the in flight request to return, so this number
77 * may not be accurate but it really doesn't matter that much since
78 * it's more or less a zero/non-zero comparison to determine whether
79 * we can shutdown the socket
80 */
81 if (FCB->SendIrp.InFlightRequest)
82 InfoReq->Information.Ulong++;
83 break;
84
85 default:
86 AFD_DbgPrint(MIN_TRACE,("Unknown info id %x\n",
87 InfoReq->InformationClass));
88 Status = STATUS_INVALID_PARAMETER;
89 break;
90 }
91 } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
92 AFD_DbgPrint(MIN_TRACE,("Exception executing GetInfo\n"));
93 Status = STATUS_INVALID_PARAMETER;
94 } _SEH2_END;
95
96 AFD_DbgPrint(MID_TRACE,("Returning %x\n", Status));
97
98 return UnlockAndMaybeComplete( FCB, Status, Irp, 0 );
99}
FUNCTION_SEND
#define FUNCTION_SEND
Definition: afd.h:86
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
MIN_TRACE
#define MIN_TRACE
Definition: debug.h:14
MID_TRACE
#define MID_TRACE
Definition: debug.h:15
Irp
_In_ PIRP Irp
Definition: csq.h:116
STATUS_NO_MEMORY
#define STATUS_NO_MEMORY
Definition: d3dkmdt.h:51
NULL
#define NULL
Definition: types.h:112
TRUE
#define TRUE
Definition: types.h:120
IrpSp
_In_ PIO_STACK_LOCATION IrpSp
Definition: create.c:4138
LockRequest
PVOID LockRequest(PIRP Irp, PIO_STACK_LOCATION IrpSp, BOOLEAN Output, KPROCESSOR_MODE *LockMode)
Definition: lock.c:24
LostSocket
NTSTATUS LostSocket(PIRP Irp)
Definition: lock.c:387
UnlockAndMaybeComplete
NTSTATUS NTAPI UnlockAndMaybeComplete(PAFD_FCB FCB, NTSTATUS Status, PIRP Irp, UINT Information)
Definition: lock.c:375
SocketAcquireStateLock
BOOLEAN SocketAcquireStateLock(PAFD_FCB FCB)
Definition: lock.c:360
AFD_DbgPrint
#define AFD_DbgPrint(_t_, _x_)
Definition: debug.h:60
Status
Status
Definition: gdiplustypes.h:25
EXCEPTION_EXECUTE_HANDLER
#define EXCEPTION_EXECUTE_HANDLER
Definition: excpt.h:90
UNREFERENCED_PARAMETER
#define UNREFERENCED_PARAMETER(P)
Definition: ntbasedef.h:329
_SEH2_EXCEPT
#define _SEH2_EXCEPT(...)
Definition: pseh2_64.h:82
_SEH2_END
#define _SEH2_END
Definition: pseh2_64.h:171
_SEH2_TRY
#define _SEH2_TRY
Definition: pseh2_64.h:71
AFD_INFO_SEND_WINDOW_SIZE
#define AFD_INFO_SEND_WINDOW_SIZE
Definition: shared.h:186
AFD_INFO_RECEIVE_WINDOW_SIZE
#define AFD_INFO_RECEIVE_WINDOW_SIZE
Definition: shared.h:185
AFD_INFO_RECEIVE_CONTENT_SIZE
#define AFD_INFO_RECEIVE_CONTENT_SIZE
Definition: shared.h:188
AFD_INFO_BLOCKING_MODE
#define AFD_INFO_BLOCKING_MODE
Definition: shared.h:183
AFD_INFO_SENDS_IN_PROGRESS
#define AFD_INFO_SENDS_IN_PROGRESS
Definition: shared.h:184
AFD_INFO_INLINING_MODE
#define AFD_INFO_INLINING_MODE
Definition: shared.h:182
AFD_INFO_GROUP_ID_TYPE
#define AFD_INFO_GROUP_ID_TYPE
Definition: shared.h:187
STATUS_SUCCESS
#define STATUS_SUCCESS
Definition: shellext.h:65
_AFD_FCB
Definition: afd.h:159
_AFD_INFO
Definition: shared.h:28
_AFD_INFO::InformationClass
ULONG InformationClass
Definition: shared.h:29
_AFD_INFO::Information
union _AFD_INFO::@3592 Information
_AFD_INFO::Ulong
ULONG Ulong
Definition: shared.h:31
_AFD_INFO::Boolean
BOOLEAN Boolean
Definition: shared.h:33
_AFD_INFO::LargeInteger
LARGE_INTEGER LargeInteger
Definition: shared.h:32
_FCB
Definition: cdstruc.h:902
_IO_STACK_LOCATION::FileObject
PFILE_OBJECT FileObject
Definition: iotypes.h:3171
_LIST_ENTRY
Definition: typedefs.h:120
_LIST_ENTRY::Flink
struct _LIST_ENTRY * Flink
Definition: typedefs.h:121
STATUS_INVALID_PARAMETER
#define STATUS_INVALID_PARAMETER
Definition: udferr_usr.h:135
_LARGE_INTEGER::u
struct _LARGE_INTEGER::@2479 u
DeviceObject
_In_ PDEVICE_OBJECT DeviceObject
Definition: wdfdevice.h:2061
FileObject
_In_ WDFREQUEST _In_ WDFFILEOBJECT FileObject
Definition: wdfdevice.h:550
PFILE_OBJECT
* PFILE_OBJECT
Definition: iotypes.h:1998

Referenced by AfdDispatch().

◆ AfdGetPeerName()

NTSTATUS NTAPI AfdGetPeerName ( PDEVICE_OBJECT  DeviceObject,
PIRP  Irp,
PIO_STACK_LOCATION  IrpSp 
)

Definition at line 281 of file info.c.

282 {
283 NTSTATUS Status;
284 PFILE_OBJECT FileObject = IrpSp->FileObject;
285 PAFD_FCB FCB = FileObject->FsContext;
286
287 UNREFERENCED_PARAMETER(DeviceObject);
288
289 if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
290
291 if (FCB->RemoteAddress == NULL) {
292 AFD_DbgPrint(MIN_TRACE,("Invalid parameter\n"));
293 return UnlockAndMaybeComplete( FCB, STATUS_INVALID_PARAMETER, Irp, 0 );
294 }
295
296 if (IrpSp->Parameters.DeviceIoControl.OutputBufferLength >= TaLengthOfTransportAddress(FCB->RemoteAddress))
297 {
298 RtlCopyMemory(Irp->UserBuffer, FCB->RemoteAddress, TaLengthOfTransportAddress(FCB->RemoteAddress));
299 Status = STATUS_SUCCESS;
300 }
301 else
302 {
303 AFD_DbgPrint(MIN_TRACE,("Buffer too small\n"));
304 Status = STATUS_BUFFER_TOO_SMALL;
305 }
306
307 return UnlockAndMaybeComplete( FCB, Status, Irp, 0 );
308}
STATUS_BUFFER_TOO_SMALL
#define STATUS_BUFFER_TOO_SMALL
Definition: shellext.h:69
_IO_STACK_LOCATION::DeviceIoControl
struct _IO_STACK_LOCATION::@1694::@1695 DeviceIoControl
_IO_STACK_LOCATION::Parameters
union _IO_STACK_LOCATION::@1694 Parameters
TaLengthOfTransportAddress
UINT TaLengthOfTransportAddress(PTRANSPORT_ADDRESS Addr)
Definition: tdiconn.c:56
RtlCopyMemory
#define RtlCopyMemory(Destination, Source, Length)
Definition: typedefs.h:263

Referenced by AfdDispatch().

◆ AfdGetSockName()

NTSTATUS NTAPI AfdGetSockName ( PDEVICE_OBJECT  DeviceObject,
PIRP  Irp,
PIO_STACK_LOCATION  IrpSp 
)

Definition at line 232 of file info.c.

233 {
234 NTSTATUS Status = STATUS_SUCCESS;
235 PFILE_OBJECT FileObject = IrpSp->FileObject;
236 PAFD_FCB FCB = FileObject->FsContext;
237 PMDL Mdl = NULL;
238
239 UNREFERENCED_PARAMETER(DeviceObject);
240 ASSERT(Irp->MdlAddress == NULL);
241
242 if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
243
244 if( FCB->AddressFile.Object == NULL && FCB->Connection.Object == NULL ) {
245 return UnlockAndMaybeComplete( FCB, STATUS_INVALID_PARAMETER, Irp, 0 );
246 }
247
248 Mdl = IoAllocateMdl( Irp->UserBuffer,
249 IrpSp->Parameters.DeviceIoControl.OutputBufferLength,
250 FALSE,
251 FALSE,
252 NULL );
253
254 if( Mdl != NULL ) {
255 _SEH2_TRY {
256 MmProbeAndLockPages( Mdl, Irp->RequestorMode, IoModifyAccess );
257 } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
258 AFD_DbgPrint(MIN_TRACE, ("MmProbeAndLockPages() failed.\n"));
259 Status = _SEH2_GetExceptionCode();
260 } _SEH2_END;
261
262 if( NT_SUCCESS(Status) ) {
263 Status = TdiQueryInformation( FCB->Connection.Object
264 ? FCB->Connection.Object
265 : FCB->AddressFile.Object,
266 TDI_QUERY_ADDRESS_INFO,
267 Mdl );
268 }
269
270 /* Check if MmProbeAndLockPages or TdiQueryInformation failed and
271 * clean up Mdl */
272 if (!NT_SUCCESS(Status) && Irp->MdlAddress != Mdl)
273 IoFreeMdl(Mdl);
274 } else
275 Status = STATUS_INSUFFICIENT_RESOURCES;
276
277 return UnlockAndMaybeComplete( FCB, Status, Irp, 0 );
278}
FALSE
#define FALSE
Definition: types.h:117
NT_SUCCESS
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:33
IoFreeMdl
#define IoFreeMdl
Definition: fxmdl.h:89
IoAllocateMdl
#define IoAllocateMdl
Definition: fxmdl.h:88
MmProbeAndLockPages
VOID NTAPI MmProbeAndLockPages(IN PMDL Mdl, IN KPROCESSOR_MODE AccessMode, IN LOCK_OPERATION Operation)
Definition: mdlsup.c:931
ASSERT
#define ASSERT(a)
Definition: mode.c:44
_SEH2_GetExceptionCode
#define _SEH2_GetExceptionCode()
Definition: pseh2_64.h:181
TDI_QUERY_ADDRESS_INFO
#define TDI_QUERY_ADDRESS_INFO
Definition: tdi.h:181
TdiQueryInformation
NTSTATUS TdiQueryInformation(PFILE_OBJECT FileObject, LONG QueryType, PMDL MdlBuffer)
Query for information.
Definition: tdi.c:693
STATUS_INSUFFICIENT_RESOURCES
#define STATUS_INSUFFICIENT_RESOURCES
Definition: udferr_usr.h:158
Mdl
_In_ WDFDEVICE _In_ PVOID _In_opt_ PMDL Mdl
Definition: wdfdmatransaction.h:89
IoModifyAccess
@ IoModifyAccess
Definition: ketypes.h:917

Referenced by AfdDispatch().

◆ AfdSetInfo()

NTSTATUS NTAPI AfdSetInfo ( PDEVICE_OBJECT  DeviceObject,
PIRP  Irp,
PIO_STACK_LOCATION  IrpSp 
)

Definition at line 102 of file info.c.

103 {
104 NTSTATUS Status = STATUS_SUCCESS;
105 PAFD_INFO InfoReq = LockRequest(Irp, IrpSp, FALSE, NULL);
106 PFILE_OBJECT FileObject = IrpSp->FileObject;
107 PAFD_FCB FCB = FileObject->FsContext;
108 PCHAR NewBuffer;
109
110 UNREFERENCED_PARAMETER(DeviceObject);
111
112 if (!SocketAcquireStateLock(FCB)) return LostSocket(Irp);
113
114 if (!InfoReq)
115 return UnlockAndMaybeComplete(FCB, STATUS_NO_MEMORY, Irp, 0);
116
117 _SEH2_TRY {
118 switch (InfoReq->InformationClass) {
119 case AFD_INFO_BLOCKING_MODE:
120 AFD_DbgPrint(MID_TRACE,("Blocking mode set to %u\n", InfoReq->Information.Boolean));
121 FCB->NonBlocking = InfoReq->Information.Boolean;
122 break;
123 case AFD_INFO_INLINING_MODE:
124 FCB->OobInline = InfoReq->Information.Boolean;
125 break;
126 case AFD_INFO_RECEIVE_WINDOW_SIZE:
127 if (FCB->State == SOCKET_STATE_CONNECTED ||
128 FCB->Flags & AFD_ENDPOINT_CONNECTIONLESS)
129 {
130 /* FIXME: likely not right, check tcpip.sys for TDI_QUERY_MAX_DATAGRAM_INFO */
131 if (InfoReq->Information.Ulong > 0 && InfoReq->Information.Ulong < 0xFFFF &&
132 InfoReq->Information.Ulong != FCB->Recv.Size)
133 {
134 NewBuffer = ExAllocatePoolWithTag(PagedPool,
135 InfoReq->Information.Ulong,
136 TAG_AFD_DATA_BUFFER);
137
138 if (NewBuffer)
139 {
140 if (FCB->Recv.Content > InfoReq->Information.Ulong)
141 FCB->Recv.Content = InfoReq->Information.Ulong;
142
143 if (FCB->Recv.Window)
144 {
145 RtlCopyMemory(NewBuffer,
146 FCB->Recv.Window,
147 FCB->Recv.Content);
148
149 ExFreePoolWithTag(FCB->Recv.Window, TAG_AFD_DATA_BUFFER);
150 }
151
152 FCB->Recv.Size = InfoReq->Information.Ulong;
153 FCB->Recv.Window = NewBuffer;
154
155 Status = STATUS_SUCCESS;
156 }
157 else
158 {
159 Status = STATUS_NO_MEMORY;
160 }
161 }
162 else
163 {
164 Status = STATUS_SUCCESS;
165 }
166 }
167 else
168 {
169 Status = STATUS_INVALID_PARAMETER;
170 }
171 break;
172 case AFD_INFO_SEND_WINDOW_SIZE:
173 if (FCB->State == SOCKET_STATE_CONNECTED ||
174 FCB->Flags & AFD_ENDPOINT_CONNECTIONLESS)
175 {
176 if (InfoReq->Information.Ulong > 0 && InfoReq->Information.Ulong < 0xFFFF &&
177 InfoReq->Information.Ulong != FCB->Send.Size)
178 {
179 NewBuffer = ExAllocatePoolWithTag(PagedPool,
180 InfoReq->Information.Ulong,
181 TAG_AFD_DATA_BUFFER);
182
183 if (NewBuffer)
184 {
185 if (FCB->Send.BytesUsed > InfoReq->Information.Ulong)
186 FCB->Send.BytesUsed = InfoReq->Information.Ulong;
187
188 if (FCB->Send.Window)
189 {
190 RtlCopyMemory(NewBuffer,
191 FCB->Send.Window,
192 FCB->Send.BytesUsed);
193
194 ExFreePoolWithTag(FCB->Send.Window, TAG_AFD_DATA_BUFFER);
195 }
196
197 FCB->Send.Size = InfoReq->Information.Ulong;
198 FCB->Send.Window = NewBuffer;
199
200 Status = STATUS_SUCCESS;
201 }
202 else
203 {
204 Status = STATUS_NO_MEMORY;
205 }
206 }
207 else
208 {
209 Status = STATUS_SUCCESS;
210 }
211 }
212 else
213 {
214 Status = STATUS_INVALID_PARAMETER;
215 }
216 break;
217 default:
218 AFD_DbgPrint(MIN_TRACE,("Unknown request %u\n", InfoReq->InformationClass));
219 break;
220 }
221 } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
222 AFD_DbgPrint(MIN_TRACE,("Exception executing SetInfo\n"));
223 Status = STATUS_INVALID_PARAMETER;
224 } _SEH2_END;
225
226 AFD_DbgPrint(MID_TRACE,("Returning %x\n", Status));
227
228 return UnlockAndMaybeComplete(FCB, Status, Irp, 0);
229}
SOCKET_STATE_CONNECTED
#define SOCKET_STATE_CONNECTED
Definition: afd.h:76
TAG_AFD_DATA_BUFFER
#define TAG_AFD_DATA_BUFFER
Definition: afd.h:38
ExAllocatePoolWithTag
#define ExAllocatePoolWithTag(hernya, size, tag)
Definition: env_spec_w32.h:350
PagedPool
#define PagedPool
Definition: env_spec_w32.h:308
ExFreePoolWithTag
#define ExFreePoolWithTag(_P, _T)
Definition: module.h:1109
AFD_ENDPOINT_CONNECTIONLESS
#define AFD_ENDPOINT_CONNECTIONLESS
Definition: shared.h:153
_FCB::Flags
ULONG Flags
Definition: ntfs.h:536
PCHAR
char * PCHAR
Definition: typedefs.h:51

Referenced by AfdDispatch().