#ifndef __TCP_H
#define __TCP_H
+typedef VOID
+(*PTCP_COMPLETION_ROUTINE)( PVOID Context, NTSTATUS Status, ULONG Count );
+
/* TCPv4 header structure */
typedef struct TCPv4_HEADER {
USHORT SourcePort; /* Source port */
#define TCPOPTLEN_MAX_SEG_SIZE 0x4
+#define TCP_CANCEL_DISCONNECT 0
+#define TCP_CANCEL_CLOSE 1
+
/* Data offset; 32-bit words (leftmost 4 bits); convert to bytes */
#define TCP_DATA_OFFSET(DataOffset)(((DataOffset) & 0xF0) >> (4-2))
USHORT TCPLength; /* Size of TCP segment */
} __attribute__((packed)) TCPv4_PSEUDO_HEADER, *PTCPv4_PSEUDO_HEADER;
+typedef struct _SLEEPING_THREAD {
+ LIST_ENTRY Entry;
+ PVOID SleepToken;
+ KEVENT Event;
+} SLEEPING_THREAD, *PSLEEPING_THREAD;
+
+typedef struct _CANCEL_REQUEST {
+ LIST_ENTRY Entry;
+ PVOID Context;
+ PIRP Irp;
+ UINT Flags, Type;
+} CANCEL_REQUEST, *PCANCEL_REQUEST;
/* Retransmission timeout constants */
#define SRF_SYN TCP_SYN
#define SRF_FIN TCP_FIN
+extern LONG TCP_IPIdentification;
+extern LIST_ENTRY SignalledConnections;
+extern LIST_ENTRY SleepingThreadsList;
+extern FAST_MUTEX SleepingThreadsLock;
+extern RECURSIVE_MUTEX TCPLock;
+
+/* accept.c */
+NTSTATUS TCPServiceListeningSocket( PCONNECTION_ENDPOINT Listener,
+ PCONNECTION_ENDPOINT Connection,
+ PTDI_REQUEST_KERNEL Request );
+NTSTATUS TCPListen( PCONNECTION_ENDPOINT Connection, UINT Backlog );
+VOID TCPAbortListenForSocket( PCONNECTION_ENDPOINT Listener,
+ PCONNECTION_ENDPOINT Connection );
+NTSTATUS TCPAccept
+( PTDI_REQUEST Request,
+ PCONNECTION_ENDPOINT Listener,
+ PCONNECTION_ENDPOINT Connection,
+ PTCP_COMPLETION_ROUTINE Complete,
+ PVOID Context );
+
+/* tcp.c */
+PCONNECTION_ENDPOINT TCPAllocateConnectionEndpoint( PVOID ClientContext );
+VOID TCPFreeConnectionEndpoint( PCONNECTION_ENDPOINT Connection );
+
+NTSTATUS TCPSocket( PCONNECTION_ENDPOINT Connection,
+ UINT Family, UINT Type, UINT Proto );
+
PTCP_SEGMENT TCPCreateSegment(
PIP_PACKET IPPacket,
PTCPv4_HEADER TCPHeader,
PULONG Acknowledged);
NTSTATUS TCPConnect(
- PTDI_REQUEST Request,
+ PCONNECTION_ENDPOINT Connection,
PTDI_CONNECTION_INFORMATION ConnInfo,
- PTDI_CONNECTION_INFORMATION ReturnInfo);
+ PTDI_CONNECTION_INFORMATION ReturnInfo,
+ PTCP_COMPLETION_ROUTINE Complete,
+ PVOID Context);
-NTSTATUS TCPListen(
- PTDI_REQUEST Request,
- UINT Backlog );
+NTSTATUS TCPDisconnect(
+ PCONNECTION_ENDPOINT Connection,
+ UINT Flags,
+ PTDI_CONNECTION_INFORMATION ConnInfo,
+ PTDI_CONNECTION_INFORMATION ReturnInfo,
+ PTCP_COMPLETION_ROUTINE Complete,
+ PVOID Context);
NTSTATUS TCPReceiveData(
- PTDI_REQUEST Request,
+ PCONNECTION_ENDPOINT Connection,
PNDIS_BUFFER Buffer,
ULONG ReceiveLength,
+ PULONG BytesReceived,
ULONG ReceiveFlags,
- PULONG BytesReceived);
+ PTCP_COMPLETION_ROUTINE Complete,
+ PVOID Context);
NTSTATUS TCPSendData(
- PTDI_REQUEST Request,
- PTDI_CONNECTION_INFORMATION ConnInfo,
- PNDIS_BUFFER Buffer,
- ULONG DataSize);
+ PCONNECTION_ENDPOINT Connection,
+ PCHAR Buffer,
+ ULONG DataSize,
+ PULONG DataUsed,
+ ULONG Flags);
+
+NTSTATUS TCPClose( PCONNECTION_ENDPOINT Connection );
+
+PVOID TCPPrepareInterface( PIP_INTERFACE IF );
+
+VOID TCPDisposeInterfaceData( PVOID Data );
+
+NTSTATUS TCPTranslateError( int OskitError );
+
+VOID TCPTimeout();
+
+UINT TCPAllocatePort( UINT HintPort );
+
+VOID TCPFreePort( UINT Port );
+
+NTSTATUS TCPGetPeerAddress
+( PCONNECTION_ENDPOINT Connection,
+ PTRANSPORT_ADDRESS TransportAddress );
NTSTATUS TCPStartup(
VOID);
NTSTATUS TCPShutdown(
VOID);
+void TCPMarkForDisconnect(
+ PCONNECTION_ENDPOINT Connection, BOOLEAN Marked );
+
#endif /* __TCP_H */