mirror of
https://github.com/reactos/reactos.git
synced 2025-08-06 10:33:11 +00:00
cb01eb2af4
- Remove debug prints - Modify error code translations [lwIP] - Fix race conditions caused by checking the SocketContexts of connections from outside the lwIP main thread context - Don't explicitly remove callbacks from pcbs (no reason to do apperently) - Use pbuf_free_callback (which can be safely called from outside the lwIP main thread context) instead of pbuf_free svn path=/branches/GSoC_2011/TcpIpDriver/; revision=53033
348 lines
10 KiB
C
348 lines
10 KiB
C
/*
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
* PROJECT: ReactOS TCP/IP protocol driver
|
|
* FILE: transport/tcp/event.c
|
|
* PURPOSE: Transmission Control Protocol
|
|
* PROGRAMMERS: Cameron Gutman (cameron.gutman@reactos.org)
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
#include "lwip/err.h"
|
|
#include "lwip/sys.h"
|
|
#include "lwip/pbuf.h"
|
|
#include "lwip/tcp.h"
|
|
#include "lwip/api.h"
|
|
|
|
#include "rosip.h"
|
|
|
|
static const char * const tcp_state_str[] = {
|
|
"CLOSED",
|
|
"LISTEN",
|
|
"SYN_SENT",
|
|
"SYN_RCVD",
|
|
"ESTABLISHED",
|
|
"FIN_WAIT_1",
|
|
"FIN_WAIT_2",
|
|
"CLOSE_WAIT",
|
|
"CLOSING",
|
|
"LAST_ACK",
|
|
"TIME_WAIT"
|
|
};
|
|
|
|
static
|
|
VOID
|
|
BucketCompletionWorker(PVOID Context)
|
|
{
|
|
PTDI_BUCKET Bucket = (PTDI_BUCKET)Context;
|
|
PTCP_COMPLETION_ROUTINE Complete;
|
|
|
|
Complete = (PTCP_COMPLETION_ROUTINE)Bucket->Request.RequestNotifyObject;
|
|
|
|
Complete(Bucket->Request.RequestContext, Bucket->Status, Bucket->Information);
|
|
|
|
DereferenceObject(Bucket->AssociatedEndpoint);
|
|
|
|
ExFreePoolWithTag(Bucket, TDI_BUCKET_TAG);
|
|
}
|
|
|
|
static
|
|
VOID
|
|
CompleteBucket(PCONNECTION_ENDPOINT Connection, PTDI_BUCKET Bucket, const BOOLEAN Synchronous)
|
|
{
|
|
ReferenceObject(Connection);
|
|
Bucket->AssociatedEndpoint = Connection;
|
|
if (Synchronous)
|
|
{
|
|
BucketCompletionWorker(Bucket);
|
|
}
|
|
else
|
|
{
|
|
ChewCreate(BucketCompletionWorker, Bucket);
|
|
}
|
|
}
|
|
|
|
VOID
|
|
FlushAllQueues(PCONNECTION_ENDPOINT Connection, const NTSTATUS Status)
|
|
{
|
|
PTDI_BUCKET Bucket;
|
|
PLIST_ENTRY Entry;
|
|
|
|
ReferenceObject(Connection);
|
|
|
|
while ((Entry = ExInterlockedRemoveHeadList(&Connection->ReceiveRequest, &Connection->Lock)))
|
|
{
|
|
Bucket = CONTAINING_RECORD( Entry, TDI_BUCKET, Entry );
|
|
|
|
TI_DbgPrint(DEBUG_TCP,
|
|
("Completing Receive request: %x %x\n",
|
|
Bucket->Request, Status));
|
|
|
|
Bucket->Status = Status;
|
|
Bucket->Information = 0;
|
|
|
|
CompleteBucket(Connection, Bucket, FALSE);
|
|
}
|
|
|
|
// Calling with Status == STATUS_SUCCESS means that we got a graceful closure
|
|
// so we don't want to kill everything else since send is still valid in this state
|
|
//
|
|
if (Status == STATUS_SUCCESS)
|
|
{
|
|
DereferenceObject(Connection);
|
|
return;
|
|
}
|
|
|
|
while ((Entry = ExInterlockedRemoveHeadList(&Connection->ListenRequest, &Connection->Lock)))
|
|
{
|
|
Bucket = CONTAINING_RECORD( Entry, TDI_BUCKET, Entry );
|
|
|
|
Bucket->Status = Status;
|
|
Bucket->Information = 0;
|
|
|
|
DereferenceObject(Bucket->AssociatedEndpoint);
|
|
CompleteBucket(Connection, Bucket, FALSE);
|
|
}
|
|
|
|
while ((Entry = ExInterlockedRemoveHeadList(&Connection->SendRequest, &Connection->Lock)))
|
|
{
|
|
Bucket = CONTAINING_RECORD( Entry, TDI_BUCKET, Entry );
|
|
|
|
TI_DbgPrint(DEBUG_TCP,
|
|
("Completing Send request: %x %x\n",
|
|
Bucket->Request, Status));
|
|
|
|
Bucket->Status = Status;
|
|
Bucket->Information = 0;
|
|
|
|
CompleteBucket(Connection, Bucket, FALSE);
|
|
}
|
|
|
|
while ((Entry = ExInterlockedRemoveHeadList(&Connection->ConnectRequest, &Connection->Lock)))
|
|
{
|
|
Bucket = CONTAINING_RECORD( Entry, TDI_BUCKET, Entry );
|
|
|
|
Bucket->Status = Status;
|
|
Bucket->Information = 0;
|
|
|
|
CompleteBucket(Connection, Bucket, FALSE);
|
|
}
|
|
|
|
DereferenceObject(Connection);
|
|
}
|
|
|
|
VOID
|
|
TCPFinEventHandler(void *arg, const err_t err)
|
|
{
|
|
PCONNECTION_ENDPOINT Connection = (PCONNECTION_ENDPOINT)arg;
|
|
const NTSTATUS status = TCPTranslateError(err);
|
|
|
|
/* Only clear the pointer if the shutdown was caused by an error */
|
|
if (err != ERR_OK)
|
|
{
|
|
/* We're got closed by the error so remove the PCB pointer */
|
|
Connection->SocketContext = NULL;
|
|
}
|
|
|
|
FlushAllQueues(Connection, status);
|
|
}
|
|
|
|
VOID
|
|
TCPAcceptEventHandler(void *arg, PTCP_PCB newpcb)
|
|
{
|
|
PCONNECTION_ENDPOINT Connection = (PCONNECTION_ENDPOINT)arg;
|
|
PTDI_BUCKET Bucket;
|
|
PLIST_ENTRY Entry;
|
|
PIRP Irp;
|
|
NTSTATUS Status;
|
|
KIRQL OldIrql;
|
|
|
|
ReferenceObject(Connection);
|
|
|
|
while ((Entry = ExInterlockedRemoveHeadList(&Connection->ListenRequest, &Connection->Lock)))
|
|
{
|
|
PIO_STACK_LOCATION IrpSp;
|
|
|
|
Bucket = CONTAINING_RECORD( Entry, TDI_BUCKET, Entry );
|
|
|
|
Irp = Bucket->Request.RequestContext;
|
|
IrpSp = IoGetCurrentIrpStackLocation( Irp );
|
|
|
|
TI_DbgPrint(DEBUG_TCP,("[IP, TCPAcceptEventHandler] Getting the socket\n"));
|
|
|
|
Status = TCPCheckPeerForAccept(newpcb,
|
|
(PTDI_REQUEST_KERNEL)&IrpSp->Parameters);
|
|
|
|
TI_DbgPrint(DEBUG_TCP,("Socket: Status: %x\n", Status));
|
|
|
|
Bucket->Status = Status;
|
|
Bucket->Information = 0;
|
|
|
|
if (Status == STATUS_SUCCESS)
|
|
{
|
|
LockObject(Bucket->AssociatedEndpoint, &OldIrql);
|
|
|
|
/* sanity assert...this should never be in anything else but a CLOSED state */
|
|
ASSERT( ((PTCP_PCB)Bucket->AssociatedEndpoint->SocketContext)->state == CLOSED );
|
|
|
|
/* free socket context created in FileOpenConnection, as we're using a new one */
|
|
LibTCPClose(Bucket->AssociatedEndpoint, TRUE);
|
|
|
|
/* free previously created socket context (we don't use it, we use newpcb) */
|
|
Bucket->AssociatedEndpoint->SocketContext = newpcb;
|
|
|
|
LibTCPAccept(newpcb, (PTCP_PCB)Connection->SocketContext, Bucket->AssociatedEndpoint);
|
|
|
|
UnlockObject(Bucket->AssociatedEndpoint, OldIrql);
|
|
}
|
|
|
|
DereferenceObject(Bucket->AssociatedEndpoint);
|
|
|
|
CompleteBucket(Connection, Bucket, FALSE);
|
|
}
|
|
|
|
DereferenceObject(Connection);
|
|
}
|
|
|
|
VOID
|
|
TCPSendEventHandler(void *arg, u16_t space)
|
|
{
|
|
PCONNECTION_ENDPOINT Connection = (PCONNECTION_ENDPOINT)arg;
|
|
PTDI_BUCKET Bucket;
|
|
PLIST_ENTRY Entry;
|
|
PIRP Irp;
|
|
NTSTATUS Status;
|
|
PMDL Mdl;
|
|
|
|
ReferenceObject(Connection);
|
|
|
|
while ((Entry = ExInterlockedRemoveHeadList(&Connection->SendRequest, &Connection->Lock)))
|
|
{
|
|
UINT SendLen = 0;
|
|
PVOID SendBuffer = 0;
|
|
|
|
Bucket = CONTAINING_RECORD( Entry, TDI_BUCKET, Entry );
|
|
|
|
Irp = Bucket->Request.RequestContext;
|
|
Mdl = Irp->MdlAddress;
|
|
|
|
TI_DbgPrint(DEBUG_TCP,
|
|
("Getting the user buffer from %x\n", Mdl));
|
|
|
|
NdisQueryBuffer( Mdl, &SendBuffer, &SendLen );
|
|
|
|
TI_DbgPrint(DEBUG_TCP,
|
|
("Writing %d bytes to %x\n", SendLen, SendBuffer));
|
|
|
|
TI_DbgPrint(DEBUG_TCP, ("Connection: %x\n", Connection));
|
|
TI_DbgPrint
|
|
(DEBUG_TCP,
|
|
("Connection->SocketContext: %x\n",
|
|
Connection->SocketContext));
|
|
|
|
Status = TCPTranslateError(LibTCPSend(Connection,
|
|
SendBuffer,
|
|
SendLen, TRUE));
|
|
|
|
TI_DbgPrint(DEBUG_TCP,("TCP Bytes: %d\n", SendLen));
|
|
|
|
if( Status == STATUS_PENDING )
|
|
{
|
|
ExInterlockedInsertHeadList(&Connection->SendRequest,
|
|
&Bucket->Entry,
|
|
&Connection->Lock);
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
TI_DbgPrint(DEBUG_TCP,
|
|
("Completing Send request: %x %x\n",
|
|
Bucket->Request, Status));
|
|
|
|
Bucket->Status = Status;
|
|
Bucket->Information = (Bucket->Status == STATUS_SUCCESS) ? SendLen : 0;
|
|
|
|
CompleteBucket(Connection, Bucket, FALSE);
|
|
}
|
|
}
|
|
|
|
DereferenceObject(Connection);
|
|
}
|
|
|
|
u32_t
|
|
TCPRecvEventHandler(void *arg, struct pbuf *p)
|
|
{
|
|
PCONNECTION_ENDPOINT Connection = (PCONNECTION_ENDPOINT)arg;
|
|
PTDI_BUCKET Bucket;
|
|
PLIST_ENTRY Entry;
|
|
PIRP Irp;
|
|
PMDL Mdl;
|
|
UINT Received = 0;
|
|
UINT RecvLen;
|
|
PUCHAR RecvBuffer;
|
|
|
|
ASSERT(p);
|
|
|
|
ReferenceObject(Connection);
|
|
|
|
if ((Entry = ExInterlockedRemoveHeadList(&Connection->ReceiveRequest, &Connection->Lock)))
|
|
{
|
|
Bucket = CONTAINING_RECORD( Entry, TDI_BUCKET, Entry );
|
|
|
|
Irp = Bucket->Request.RequestContext;
|
|
Mdl = Irp->MdlAddress;
|
|
|
|
TI_DbgPrint(DEBUG_TCP,
|
|
("[IP, TCPRecvEventHandler] Getting the user buffer from %x\n", Mdl));
|
|
|
|
NdisQueryBuffer( Mdl, &RecvBuffer, &RecvLen );
|
|
|
|
TI_DbgPrint(DEBUG_TCP,
|
|
("[IP, TCPRecvEventHandler] Reading %d bytes to %x\n", RecvLen, RecvBuffer));
|
|
|
|
TI_DbgPrint(DEBUG_TCP, ("Connection: %x\n", Connection));
|
|
TI_DbgPrint(DEBUG_TCP, ("[IP, TCPRecvEventHandler] Connection->SocketContext: %x\n", Connection->SocketContext));
|
|
TI_DbgPrint(DEBUG_TCP, ("[IP, TCPRecvEventHandler] RecvBuffer: %x\n", RecvBuffer));
|
|
|
|
RecvLen = MIN(p->tot_len, RecvLen);
|
|
|
|
for (Received = 0; Received < RecvLen; Received += p->len, p = p->next)
|
|
{
|
|
RtlCopyMemory(RecvBuffer + Received, p->payload, p->len);
|
|
}
|
|
|
|
TI_DbgPrint(DEBUG_TCP,("TCP Bytes: %d\n", Received));
|
|
|
|
Bucket->Status = STATUS_SUCCESS;
|
|
Bucket->Information = Received;
|
|
|
|
CompleteBucket(Connection, Bucket, FALSE);
|
|
}
|
|
|
|
DereferenceObject(Connection);
|
|
|
|
return Received;
|
|
}
|
|
|
|
VOID
|
|
TCPConnectEventHandler(void *arg, err_t err)
|
|
{
|
|
PCONNECTION_ENDPOINT Connection = (PCONNECTION_ENDPOINT)arg;
|
|
PTDI_BUCKET Bucket;
|
|
PLIST_ENTRY Entry;
|
|
|
|
ReferenceObject(Connection);
|
|
|
|
while ((Entry = ExInterlockedRemoveHeadList(&Connection->ConnectRequest, &Connection->Lock)))
|
|
{
|
|
|
|
Bucket = CONTAINING_RECORD( Entry, TDI_BUCKET, Entry );
|
|
|
|
Bucket->Status = TCPTranslateError(err);
|
|
Bucket->Information = 0;
|
|
|
|
CompleteBucket(Connection, Bucket, FALSE);
|
|
}
|
|
|
|
DereferenceObject(Connection);
|
|
}
|