[TCPIP] Store creator PID in ADDRESS_FILE
[reactos.git] / drivers / network / tcpip / include / titypes.h
1 /*
2 * COPYRIGHT: See COPYING in the top level directory
3 * PROJECT: ReactOS TCP/IP protocol driver
4 * FILE: include/titypes.h
5 * PURPOSE: TCP/IP protocol driver types
6 */
7
8 #pragma once
9
10 /*
11 * VOID ReferenceObject(
12 * PVOID Object)
13 */
14 #define ReferenceObject(Object) \
15 { \
16 ASSERT((Object)->RefCount); \
17 InterlockedIncrement(&((Object)->RefCount)); \
18 }
19
20 /*
21 * VOID DereferenceObject(
22 * PVOID Object)
23 */
24 #define DereferenceObject(Object) \
25 { \
26 ASSERT((Object)->RefCount); \
27 if (InterlockedDecrement(&((Object)->RefCount)) == 0) \
28 (((Object)->Free)(Object)); \
29 }
30
31 /*
32 * VOID LockObject(PVOID Object, PKIRQL OldIrql)
33 */
34 #define LockObject(Object, Irql) \
35 { \
36 ReferenceObject(Object); \
37 KeAcquireSpinLock(&((Object)->Lock), Irql); \
38 memcpy(&(Object)->OldIrql, Irql, sizeof(KIRQL)); \
39 }
40
41 /*
42 * VOID LockObjectAtDpcLevel(PVOID Object)
43 */
44 #define LockObjectAtDpcLevel(Object) \
45 { \
46 ReferenceObject(Object); \
47 KeAcquireSpinLockAtDpcLevel(&((Object)->Lock)); \
48 (Object)->OldIrql = DISPATCH_LEVEL; \
49 }
50
51 /*
52 * VOID UnlockObject(PVOID Object, KIRQL OldIrql)
53 */
54 #define UnlockObject(Object, OldIrql) \
55 { \
56 KeReleaseSpinLock(&((Object)->Lock), OldIrql); \
57 DereferenceObject(Object); \
58 }
59
60 /*
61 * VOID UnlockObjectFromDpcLevel(PVOID Object)
62 */
63 #define UnlockObjectFromDpcLevel(Object) \
64 { \
65 KeReleaseSpinLockFromDpcLevel(&((Object)->Lock)); \
66 DereferenceObject(Object); \
67 }
68
69
70
71 #include <ip.h>
72
73 struct _ADDRESS_FILE;
74
75 /***************************************************
76 * Connection-less communication support structures *
77 ***************************************************/
78
79 typedef NTSTATUS (*DATAGRAM_SEND_ROUTINE)(
80 struct _ADDRESS_FILE *AddrFile,
81 PTDI_CONNECTION_INFORMATION ConnInfo,
82 PCHAR Buffer,
83 ULONG DataSize,
84 PULONG DataUsed);
85
86 /* Datagram completion handler prototype */
87 typedef VOID (*DATAGRAM_COMPLETION_ROUTINE)(
88 PVOID Context,
89 NDIS_STATUS Status,
90 ULONG Count);
91
92 typedef DATAGRAM_COMPLETION_ROUTINE PDATAGRAM_COMPLETION_ROUTINE;
93
94 typedef struct _DATAGRAM_RECEIVE_REQUEST {
95 struct _ADDRESS_FILE *AddressFile; /* AddressFile on behalf of */
96 LIST_ENTRY ListEntry; /* Entry on list */
97 IP_ADDRESS RemoteAddress; /* Remote address we receive from (NULL means any) */
98 USHORT RemotePort; /* Remote port we receive from (0 means any) */
99 PTDI_CONNECTION_INFORMATION ReturnInfo;/* Return information */
100 PCHAR Buffer; /* Pointer to receive buffer */
101 ULONG BufferSize; /* Size of Buffer */
102 DATAGRAM_COMPLETION_ROUTINE Complete; /* Completion routine */
103 PVOID Context; /* Pointer to context information */
104 DATAGRAM_COMPLETION_ROUTINE UserComplete; /* Completion routine */
105 PVOID UserContext; /* Pointer to context information */
106 PIRP Irp; /* IRP on behalf of */
107 } DATAGRAM_RECEIVE_REQUEST, *PDATAGRAM_RECEIVE_REQUEST;
108
109 /* Datagram build routine prototype */
110 typedef NTSTATUS (*DATAGRAM_BUILD_ROUTINE)(
111 PVOID Context,
112 PIP_ADDRESS LocalAddress,
113 USHORT LocalPort,
114 PIP_PACKET *IPPacket);
115
116 typedef struct _DATAGRAM_SEND_REQUEST {
117 LIST_ENTRY ListEntry;
118 PNDIS_PACKET PacketToSend;
119 DATAGRAM_COMPLETION_ROUTINE Complete; /* Completion routine */
120 PVOID Context; /* Pointer to context information */
121 IP_PACKET Packet;
122 UINT BufferSize;
123 IP_ADDRESS RemoteAddress;
124 USHORT RemotePort;
125 ULONG Flags; /* Protocol specific flags */
126 } DATAGRAM_SEND_REQUEST, *PDATAGRAM_SEND_REQUEST;
127
128 /* Transport address file context structure. The FileObject->FsContext2
129 field holds a pointer to this structure */
130 typedef struct _ADDRESS_FILE {
131 LIST_ENTRY ListEntry; /* Entry on list */
132 LONG RefCount; /* Reference count */
133 OBJECT_FREE_ROUTINE Free; /* Routine to use to free resources for the object */
134 KSPIN_LOCK Lock; /* Spin lock to manipulate this structure */
135 KIRQL OldIrql; /* Currently not used */
136 IP_ADDRESS Address; /* Address of this address file */
137 USHORT Family; /* Address family */
138 USHORT Protocol; /* Protocol number */
139 USHORT Port; /* Network port (network byte order) */
140 LONG Sharers; /* Number of file objects with this addr file */
141 UCHAR TTL; /* Time to live stored in packets sent from this address file */
142 UINT DF; /* Don't fragment */
143 UINT BCast; /* Receive broadcast packets */
144 UINT HeaderIncl; /* Include header in RawIP packets */
145 WORK_QUEUE_ITEM WorkItem; /* Work queue item handle */
146 DATAGRAM_COMPLETION_ROUTINE Complete; /* Completion routine for delete request */
147 PVOID Context; /* Delete request context */
148 DATAGRAM_SEND_ROUTINE Send; /* Routine to send a datagram */
149 LIST_ENTRY ReceiveQueue; /* List of outstanding receive requests */
150 LIST_ENTRY TransmitQueue; /* List of outstanding transmit requests */
151 struct _CONNECTION_ENDPOINT *Connection;
152 /* Associated connection or NULL if no associated connection exist */
153 struct _CONNECTION_ENDPOINT *Listener;
154 /* Associated listener (see transport/tcp/accept.c) */
155 IP_ADDRESS AddrCache; /* One entry address cache (destination
156 address of last packet transmitted) */
157 HANDLE ProcessId; /* Creator process ID */
158
159 /* The following members are used to control event notification */
160
161 /* Connection indication handler */
162 PTDI_IND_CONNECT ConnectHandler;
163 PVOID ConnectHandlerContext;
164 BOOLEAN RegisteredConnectHandler;
165 /* Disconnect indication handler */
166 PTDI_IND_DISCONNECT DisconnectHandler;
167 PVOID DisconnectHandlerContext;
168 BOOLEAN RegisteredDisconnectHandler;
169 /* Error indication handler */
170 PTDI_IND_ERROR ErrorHandler;
171 PVOID ErrorHandlerContext;
172 PVOID ErrorHandlerOwner;
173 BOOLEAN RegisteredErrorHandler;
174 /* Receive indication handler */
175 PTDI_IND_RECEIVE ReceiveHandler;
176 PVOID ReceiveHandlerContext;
177 BOOLEAN RegisteredReceiveHandler;
178 /* Receive datagram indication handler */
179 PTDI_IND_RECEIVE_DATAGRAM ReceiveDatagramHandler;
180 PVOID ReceiveDatagramHandlerContext;
181 BOOLEAN RegisteredReceiveDatagramHandler;
182 /* Expedited receive indication handler */
183 PTDI_IND_RECEIVE_EXPEDITED ExpeditedReceiveHandler;
184 PVOID ExpeditedReceiveHandlerContext;
185 BOOLEAN RegisteredExpeditedReceiveHandler;
186 /* Chained receive indication handler */
187 PTDI_IND_CHAINED_RECEIVE ChainedReceiveHandler;
188 PVOID ChainedReceiveHandlerContext;
189 BOOLEAN RegisteredChainedReceiveHandler;
190 /* Chained receive datagram indication handler */
191 PTDI_IND_CHAINED_RECEIVE_DATAGRAM ChainedReceiveDatagramHandler;
192 PVOID ChainedReceiveDatagramHandlerContext;
193 BOOLEAN RegisteredChainedReceiveDatagramHandler;
194 /* Chained expedited receive indication handler */
195 PTDI_IND_CHAINED_RECEIVE_EXPEDITED ChainedReceiveExpeditedHandler;
196 PVOID ChainedReceiveExpeditedHandlerContext;
197 BOOLEAN RegisteredChainedReceiveExpeditedHandler;
198 } ADDRESS_FILE, *PADDRESS_FILE;
199
200 /* Structure used to search through Address Files */
201 typedef struct _AF_SEARCH {
202 PLIST_ENTRY Next; /* Next address file to check */
203 PIP_ADDRESS Address; /* Pointer to address to be found */
204 USHORT Port; /* Network port */
205 USHORT Protocol; /* Protocol number */
206 } AF_SEARCH, *PAF_SEARCH;
207
208 /*******************************************************
209 * Connection-oriented communication support structures *
210 *******************************************************/
211
212 typedef struct _TCP_RECEIVE_REQUEST {
213 LIST_ENTRY ListEntry; /* Entry on list */
214 PNDIS_BUFFER Buffer; /* Pointer to receive buffer */
215 ULONG BufferSize; /* Size of Buffer */
216 DATAGRAM_COMPLETION_ROUTINE Complete; /* Completion routine */
217 PVOID Context; /* Pointer to context information */
218 } TCP_RECEIVE_REQUEST, *PTCP_RECEIVE_REQUEST;
219
220 /* Connection states */
221 typedef enum {
222 ctListen = 0, /* Waiting for incoming connection requests */
223 ctSynSent, /* Waiting for matching connection request */
224 ctSynReceived, /* Waiting for connection request acknowledgment */
225 ctEstablished, /* Connection is open for data transfer */
226 ctFinWait1, /* Waiting for termination request or ack. for same */
227 ctFinWait2, /* Waiting for termination request from remote TCP */
228 ctCloseWait, /* Waiting for termination request from local user */
229 ctClosing, /* Waiting for termination ack. from remote TCP */
230 ctLastAck, /* Waiting for termination request ack. from remote TCP */
231 ctTimeWait, /* Waiting for enough time to pass to be sure the remote TCP
232 received the ack. of its connection termination request */
233 ctClosed /* Represents a closed connection */
234 } CONNECTION_STATE, *PCONNECTION_STATE;
235
236
237 /* Structure for an TCP segment */
238 typedef struct _TCP_SEGMENT {
239 LIST_ENTRY ListEntry;
240 PIP_PACKET IPPacket; /* Pointer to IP packet */
241 PVOID SegmentData; /* Pointer to segment data */
242 ULONG SequenceNumber; /* Sequence number of first byte in segment */
243 ULONG Length; /* Number of bytes in segment */
244 ULONG BytesDelivered; /* Number of bytes already delivered to the client */
245 } TCP_SEGMENT, *PTCP_SEGMENT;
246
247 typedef struct _TDI_BUCKET {
248 LIST_ENTRY Entry;
249 struct _CONNECTION_ENDPOINT *AssociatedEndpoint;
250 TDI_REQUEST Request;
251 NTSTATUS Status;
252 ULONG Information;
253 } TDI_BUCKET, *PTDI_BUCKET;
254
255 /* Transport connection context structure A.K.A. Transmission Control Block
256 (TCB) in TCP terminology. The FileObject->FsContext2 field holds a pointer
257 to this structure */
258 typedef struct _CONNECTION_ENDPOINT {
259 PVOID SocketContext; /* Context for lower layer (MUST be first member in struct) */
260 LIST_ENTRY ListEntry; /* Entry on list */
261 LONG RefCount; /* Reference count */
262 OBJECT_FREE_ROUTINE Free; /* Routine to use to free resources for the object */
263 KSPIN_LOCK Lock; /* Spin lock to protect this structure */
264 KIRQL OldIrql; /* The old irql is stored here for use in HandleSignalledConnection */
265 PVOID ClientContext; /* Pointer to client context information */
266 PADDRESS_FILE AddressFile; /* Associated address file object (NULL if none) */
267
268 /* Requests */
269 LIST_ENTRY ConnectRequest; /* Queued connect requests */
270 LIST_ENTRY ListenRequest; /* Queued listen requests */
271 LIST_ENTRY ReceiveRequest; /* Queued receive requests */
272 LIST_ENTRY SendRequest; /* Queued send requests */
273 LIST_ENTRY ShutdownRequest;/* Queued shutdown requests */
274
275 LIST_ENTRY PacketQueue; /* Queued received packets waiting to be processed */
276
277 /* Disconnect Timer */
278 KTIMER DisconnectTimer;
279 KDPC DisconnectDpc;
280
281 /* Socket state */
282 BOOLEAN SendShutdown;
283 BOOLEAN ReceiveShutdown;
284 NTSTATUS ReceiveShutdownStatus;
285 BOOLEAN Closing;
286
287 struct _CONNECTION_ENDPOINT *Next; /* Next connection in address file list */
288 } CONNECTION_ENDPOINT, *PCONNECTION_ENDPOINT;
289
290
291
292 /*************************
293 * TDI support structures *
294 *************************/
295
296 /* Transport control channel context structure. The FileObject->FsContext2
297 field holds a pointer to this structure */
298 typedef struct _CONTROL_CHANNEL {
299 LIST_ENTRY ListEntry; /* Entry on list */
300 LONG RefCount; /* Reference count */
301 OBJECT_FREE_ROUTINE Free; /* Routine to use to free resources for the object */
302 KSPIN_LOCK Lock; /* Spin lock to protect this structure */
303 } CONTROL_CHANNEL, *PCONTROL_CHANNEL;
304
305 /* Transport (TCP/UDP) endpoint context structure. The FileObject->FsContext
306 field holds a pointer to this structure */
307 typedef struct _TRANSPORT_CONTEXT {
308 union {
309 HANDLE AddressHandle;
310 CONNECTION_CONTEXT ConnectionContext;
311 HANDLE ControlChannel;
312 } Handle;
313 BOOLEAN CancelIrps;
314 KEVENT CleanupEvent;
315 } TRANSPORT_CONTEXT, *PTRANSPORT_CONTEXT;
316
317 typedef struct _TI_QUERY_CONTEXT {
318 PIRP Irp;
319 PMDL InputMdl;
320 PMDL OutputMdl;
321 TCP_REQUEST_QUERY_INFORMATION_EX QueryInfo;
322 } TI_QUERY_CONTEXT, *PTI_QUERY_CONTEXT;
323
324 /* EOF */