1287 lines
43 KiB
C++
1287 lines
43 KiB
C++
/*==========================================================================
|
|
*
|
|
* Copyright (C) 1999 Microsoft Corporation. All Rights Reserved.
|
|
*
|
|
* File: Connect.cpp
|
|
* Content: This file contains support for the CONNECT/DISCONNECT protocol in DirectNet.
|
|
* It is organized with FrontEnd routines first (Connect, Listen),
|
|
* and Backend handlers (timeouts, frame crackers) after.
|
|
*
|
|
* History:
|
|
* Date By Reason
|
|
* ==== == ======
|
|
* 11/11/98 ejs Created
|
|
* 07/01/2000 masonb Assumed Ownership
|
|
*
|
|
****************************************************************************/
|
|
|
|
#include "dnproti.h"
|
|
|
|
|
|
/*** FRONT END ***/
|
|
|
|
/*
|
|
** Connect
|
|
**
|
|
** This function attempts to make a connection to a specified address.
|
|
** The function establishes the existance of a DirectNet entity and maps
|
|
** an EndPoint handle. Then we exchange CONNECT packets which allows each
|
|
** side to establish a baseline RTT.
|
|
**
|
|
*/
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "DNPConnect"
|
|
|
|
HRESULT DNPConnect( PProtocolData pPData,
|
|
IDirectPlay8Address *const lpaLocal,
|
|
IDirectPlay8Address *const lpaRemote,
|
|
const HANDLE hSPHandle,
|
|
const ULONG ulFlags,
|
|
void *const Context,
|
|
PHANDLE phHandle)
|
|
{
|
|
PSPD pSPD; // Service Provider to handle this connect
|
|
PMSD pMSD;
|
|
SPCONNECTDATA ConnData; // Parameter Block
|
|
HRESULT hr;
|
|
|
|
// Determine which SP will take this call
|
|
//
|
|
|
|
DPFX(DPFPREP,DPF_CALLIN_LVL, "Parameters: pPData[%p], paLocal[%p], paRemote[%p], hSPHandle[%x], ulFlags[%x], Context[%p], phHandle[%p]", pPData, lpaLocal, lpaRemote, hSPHandle, ulFlags, Context, phHandle);
|
|
|
|
pSPD = (PSPD) hSPHandle;
|
|
ASSERT_SPD(pSPD);
|
|
|
|
// Core should not call any Protocol APIs after calling DNPRemoveServiceProvider
|
|
ASSERT(!(pSPD->ulSPFlags & SPFLAGS_TERMINATING));
|
|
|
|
// We use an MSD to describe this op even though it isn't technically a message
|
|
if((pMSD = static_cast<PMSD>( MSDPool->Get(MSDPool) )) == NULL)
|
|
{
|
|
DPFX(DPFPREP,0, "Returning DPNERR_OUTOFMEMORY - failed to create new MSD");
|
|
return DPNERR_OUTOFMEMORY;
|
|
}
|
|
|
|
pMSD->CommandID = COMMAND_ID_CONNECT;
|
|
pMSD->pSPD = pSPD;
|
|
pMSD->Context = Context;
|
|
|
|
ASSERT(pMSD->pEPD == NULL); // MSD_Get/Release ensures this, and IndicateConnect requires it.
|
|
|
|
// Prepare to call SP to map the endpoint.
|
|
ConnData.pAddressDeviceInfo = lpaLocal;
|
|
ConnData.pAddressHost = lpaRemote;
|
|
ConnData.dwReserved = 0; // Never used
|
|
|
|
DNASSERT( ( ulFlags & ~( DN_CONNECTFLAGS_OKTOQUERYFORADDRESSING | DN_CONNECTFLAGS_ADDITIONALMULTIPLEXADAPTERS ) ) == 0 );
|
|
ConnData.dwFlags = 0;
|
|
if ( ( ulFlags & DN_CONNECTFLAGS_OKTOQUERYFORADDRESSING ) != 0 )
|
|
{
|
|
ConnData.dwFlags |= DPNSPF_OKTOQUERY;
|
|
}
|
|
|
|
if ( ( ulFlags & DN_CONNECTFLAGS_ADDITIONALMULTIPLEXADAPTERS ) != 0 )
|
|
{
|
|
ConnData.dwFlags |= DPNSPF_ADDITIONALMULTIPLEXADAPTERS;
|
|
}
|
|
|
|
ConnData.pvContext = pMSD;
|
|
ConnData.hCommand = 0;
|
|
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_IN_SERVICE_PROVIDER;
|
|
|
|
#ifdef DEBUG
|
|
// Hook up MSD before calling into SP
|
|
Lock(&pSPD->SPLock);
|
|
pMSD->blSPLinkage.InsertBefore( &pSPD->blMessageList); // Put this on cmd list
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_ON_GLOBAL_LIST;
|
|
Unlock(&pSPD->SPLock);
|
|
#endif
|
|
|
|
*phHandle = pMSD;
|
|
|
|
// SP Connect call is guaranteed to return immediately
|
|
|
|
LOCK_MSD(pMSD, "SP Ref"); // Add reference for call into SP
|
|
LOCK_MSD(pMSD, "Temp Ref");
|
|
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "Calling SP->Connect, pSPD[%p], pMSD[%p]", pSPD, pMSD);
|
|
/**/hr = IDP8ServiceProvider_Connect(pSPD->IISPIntf, &ConnData); /** CALL SP **/
|
|
|
|
if(hr != DPNERR_PENDING)
|
|
{
|
|
// SP Connect should always be asynchronous so if it isnt PENDING then it must have failed
|
|
DPFX(DPFPREP,1, "SP->Connect did not return DPNERR_PENDING, assuming failure, hr[%x]", hr);
|
|
|
|
ASSERT(hr != DPN_OK);
|
|
|
|
Lock(&pMSD->CommandLock); // This will be unlocked by final RELEASE_MSD
|
|
|
|
pMSD->ulMsgFlags1 &= ~(MFLAGS_ONE_IN_SERVICE_PROVIDER); // clear InSP flag
|
|
|
|
#ifdef DEBUG
|
|
Lock(&pSPD->SPLock);
|
|
pMSD->blSPLinkage.RemoveFromList(); // knock this off the pending list
|
|
pMSD->ulMsgFlags1 &= ~(MFLAGS_ONE_ON_GLOBAL_LIST);
|
|
Unlock(&pSPD->SPLock);
|
|
#endif
|
|
|
|
DECREMENT_MSD(pMSD, "Temp Ref");
|
|
DECREMENT_MSD(pMSD, "SP Ref"); // Remove one ref for SP call
|
|
RELEASE_MSD(pMSD, "Release On Fail"); // Remove one ref to free resource
|
|
|
|
DPFX(DPFPREP,1, "Returning hr[%x]", hr);
|
|
return hr;
|
|
}
|
|
|
|
Lock(&pMSD->CommandLock);
|
|
|
|
pMSD->hCommand = ConnData.hCommand; // retain SP command handle
|
|
pMSD->dwCommandDesc = ConnData.dwCommandDescriptor;
|
|
|
|
RELEASE_MSD(pMSD, "Temp Ref"); // Unlocks CommandLock
|
|
|
|
DPFX(DPFPREP,DPF_CALLIN_LVL, "Returning DPNERR_PENDING, pMSD[%p]", pMSD);
|
|
return DPNERR_PENDING;
|
|
}
|
|
|
|
/*
|
|
** Listen
|
|
**
|
|
** This command tells DN that it should start to accept connection requests.
|
|
** This command will return pending, and will continue to indicate connections
|
|
** until it is explicitly cancelled. It may be desireable to establish a limit
|
|
** mechanism of some sort, but for the time being this will do.
|
|
**
|
|
** Now it is desirable to Listen on multiple ports on a single adapter. This
|
|
** means that we need to accept multiple concurrent Listen commands on each adapter.
|
|
** Another fact of life is that we need to crack the Target address far enough to
|
|
** determine which SP to submit the Listen on.
|
|
*/
|
|
|
|
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "DNPListen"
|
|
|
|
HRESULT DNPListen( PProtocolData pPData,
|
|
IDirectPlay8Address *const lpaTarget,
|
|
// IDP8ServiceProvider* pISP,
|
|
const HANDLE hSPHandle,
|
|
ULONG ulFlags,
|
|
PVOID Context,
|
|
PHANDLE phHandle)
|
|
{
|
|
PSPD pSPD;
|
|
PMSD pMSD;
|
|
SPLISTENDATA ListenData;
|
|
HRESULT hr;
|
|
|
|
DPFX(DPFPREP,DPF_CALLIN_LVL, "Parameters: pPData[%p], paTarget[%p], hSPHandle[%x], ulFlags[%x], Context[%p], phHandle[%p]", pPData, lpaTarget, hSPHandle, ulFlags, Context, phHandle);
|
|
|
|
pSPD = (PSPD) hSPHandle;
|
|
ASSERT_SPD(pSPD);
|
|
|
|
// Core should not call any Protocol APIs after calling DNPRemoveServiceProvider
|
|
ASSERT(!(pSPD->ulSPFlags & SPFLAGS_TERMINATING));
|
|
|
|
// We use an MSD to describe this op even though it isn't technically a message
|
|
if((pMSD = static_cast<PMSD>( MSDPool->Get(MSDPool) )) == NULL)
|
|
{
|
|
DPFX(DPFPREP,0, "Returning DPNERR_OUTOFMEMORY - failed to create new MSD");
|
|
return DPNERR_OUTOFMEMORY;
|
|
}
|
|
|
|
pMSD->CommandID = COMMAND_ID_LISTEN;
|
|
pMSD->pSPD = pSPD;
|
|
pMSD->Context = Context;
|
|
|
|
ListenData.pAddressDeviceInfo = lpaTarget;
|
|
DNASSERT( ( ulFlags & ~( DN_LISTENFLAGS_OKTOQUERYFORADDRESSING ) ) == 0 );
|
|
ListenData.dwFlags = 0;
|
|
if ( ( ulFlags & DN_LISTENFLAGS_OKTOQUERYFORADDRESSING ) != 0 )
|
|
{
|
|
ListenData.dwFlags |= DPNSPF_OKTOQUERY;
|
|
}
|
|
ListenData.pvContext = pMSD;
|
|
ListenData.hCommand = 0;
|
|
|
|
*phHandle = pMSD;
|
|
|
|
// SP Listen call is guarenteed to return immediately
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_IN_SERVICE_PROVIDER;
|
|
|
|
#ifdef DEBUG
|
|
Lock(&pSPD->SPLock);
|
|
pMSD->blSPLinkage.InsertBefore( &pSPD->blMessageList); // Dont support timeouts for Listen
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_ON_GLOBAL_LIST;
|
|
Unlock(&pSPD->SPLock);
|
|
#endif
|
|
|
|
LOCK_MSD(pMSD, "SP Ref"); // AddRef for SP
|
|
LOCK_MSD(pMSD, "Temp Ref");
|
|
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "Calling SP->Listen, pSPD[%p], pMSD[%p]", pSPD, pMSD);
|
|
/**/hr = IDP8ServiceProvider_Listen(pSPD->IISPIntf, &ListenData); /** CALL SP **/
|
|
|
|
if(hr != DPNERR_PENDING)
|
|
{
|
|
// SP Listen should always be asynchronous so if it isnt PENDING then it must have failed
|
|
DPFX(DPFPREP,1, "SP->Listen did not return DPNERR_PENDING, assuming failure, hr[%x]", hr);
|
|
|
|
ASSERT(hr != DPN_OK);
|
|
|
|
Lock(&pMSD->CommandLock);
|
|
|
|
pMSD->ulMsgFlags1 &= ~(MFLAGS_ONE_IN_SERVICE_PROVIDER);
|
|
|
|
#ifdef DEBUG
|
|
Lock(&pSPD->SPLock);
|
|
pMSD->blSPLinkage.RemoveFromList(); // knock this off the pending list
|
|
pMSD->ulMsgFlags1 &= ~(MFLAGS_ONE_ON_GLOBAL_LIST);
|
|
Unlock(&pSPD->SPLock);
|
|
#endif
|
|
|
|
DECREMENT_MSD(pMSD, "Temp Ref");
|
|
DECREMENT_MSD(pMSD, "SP Ref"); // release once for SP
|
|
RELEASE_MSD(pMSD, "Release On Fail"); // release again to return resource
|
|
|
|
DPFX(DPFPREP,1, "Returning hr[%x]", hr);
|
|
return hr;
|
|
}
|
|
|
|
Lock(&pMSD->CommandLock);
|
|
|
|
pMSD->hCommand = ListenData.hCommand; // retail SP command handle
|
|
pMSD->dwCommandDesc = ListenData.dwCommandDescriptor;
|
|
|
|
RELEASE_MSD(pMSD, "Temp Ref"); // Unlocks CommandLock
|
|
|
|
DPFX(DPFPREP,DPF_CALLIN_LVL, "Returning DPNERR_PENDING, pMSD[%p]", pMSD);
|
|
return DPNERR_PENDING;
|
|
}
|
|
|
|
/*** BACKEND ROUTINES ***/
|
|
|
|
/*
|
|
** Complete Connect
|
|
**
|
|
** The user's Connect operation has completed. Clean everything up
|
|
** and signal the user.
|
|
**
|
|
** THIS IS ALWAYS CALLED WITH THE COMMAND LOCK HELD IN MSD
|
|
*/
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "CompleteConnect"
|
|
|
|
VOID CompleteConnect(PMSD pMSD, PSPD pSPD, PEPD pEPD, HRESULT hr)
|
|
{
|
|
AssertCriticalSectionIsTakenByThisThread(&pMSD->CommandLock, TRUE);
|
|
|
|
// We expect to never get here twice
|
|
ASSERT(!(pMSD->ulMsgFlags1 & MFLAGS_ONE_COMPLETE));
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_COMPLETE;
|
|
|
|
// Connects cannot have timeout timers
|
|
ASSERT(pMSD->TimeoutTimer == NULL);
|
|
|
|
#ifdef DEBUG
|
|
Lock(&pSPD->SPLock);
|
|
if(pMSD->ulMsgFlags1 & MFLAGS_ONE_ON_GLOBAL_LIST)
|
|
{
|
|
pMSD->blSPLinkage.RemoveFromList(); // Remove MSD from master command list
|
|
pMSD->ulMsgFlags1 &= ~(MFLAGS_ONE_ON_GLOBAL_LIST);
|
|
}
|
|
Unlock(&pSPD->SPLock);
|
|
|
|
ASSERT(!(pMSD->ulMsgFlags1 & MFLAGS_ONE_COMPLETED_TO_CORE));
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_COMPLETED_TO_CORE;
|
|
pMSD->CallStackCoreCompletion.NoteCurrentCallStack();
|
|
#endif
|
|
|
|
pMSD->pEPD = NULL;
|
|
Unlock(&pMSD->CommandLock);
|
|
|
|
if(pEPD)
|
|
{
|
|
ASSERT(hr == DPN_OK);
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "(%p) Calling Core->CompleteConnect, pMSD[%p], Core Context[%p], hr[%x], pEPD[%p]", pEPD, pMSD, pMSD->Context, hr, pEPD);
|
|
pSPD->pPData->pfVtbl->CompleteConnect(pSPD->pPData->Parent, pMSD->Context, hr, (PHANDLE) pEPD, &pEPD->Context);
|
|
|
|
Lock(&pEPD->EPLock);
|
|
ReceiveComplete(pEPD); // Complete any, releases EPLock
|
|
}
|
|
else
|
|
{
|
|
ASSERT(hr != DPN_OK);
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "Calling Core->CompleteConnect with NULL EPD, pMSD[%p], Core Context[%p], hr[%x]", pMSD, pMSD->Context, hr);
|
|
pSPD->pPData->pfVtbl->CompleteConnect(pSPD->pPData->Parent, pMSD->Context, hr, NULL, NULL);
|
|
}
|
|
|
|
// Release the final reference on the MSD AFTER indicating to the Core
|
|
Lock(&pMSD->CommandLock);
|
|
RELEASE_MSD(pMSD, "Final Release On Complete"); // Finished with this one, releases CommandLock
|
|
}
|
|
|
|
/*
|
|
** Complete Disconnect
|
|
**
|
|
** THIS IS ALWAYS CALLED WITH THE COMMAND LOCK HELD IN MSD
|
|
*/
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "CompleteDisconnect"
|
|
|
|
VOID CompleteDisconnect(PMSD pMSD, PSPD pSPD, PEPD pEPD)
|
|
{
|
|
AssertCriticalSectionIsTakenByThisThread(&pMSD->CommandLock, TRUE);
|
|
|
|
// We expect to never get here twice
|
|
ASSERT(!(pMSD->ulMsgFlags1 & MFLAGS_ONE_COMPLETE));
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_COMPLETE;
|
|
|
|
// Disconnects cannot have timeout timers
|
|
ASSERT(pMSD->TimeoutTimer == NULL);
|
|
|
|
#ifdef DEBUG
|
|
Lock(&pSPD->SPLock);
|
|
if(pMSD->ulMsgFlags1 & MFLAGS_ONE_ON_GLOBAL_LIST)
|
|
{
|
|
pMSD->blSPLinkage.RemoveFromList(); // Remove MSD from master command list
|
|
pMSD->ulMsgFlags1 &= ~(MFLAGS_ONE_ON_GLOBAL_LIST);
|
|
}
|
|
Unlock(&pSPD->SPLock);
|
|
|
|
ASSERT(!(pMSD->ulMsgFlags1 & MFLAGS_ONE_COMPLETED_TO_CORE));
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_COMPLETED_TO_CORE;
|
|
pMSD->CallStackCoreCompletion.NoteCurrentCallStack();
|
|
#endif
|
|
|
|
// No one else should use this
|
|
pMSD->pEPD = NULL;
|
|
|
|
if(pMSD->CommandID == COMMAND_ID_DISCONNECT)
|
|
{
|
|
Unlock(&pMSD->CommandLock);
|
|
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "(%p) Calling Core->CompleteDisconnect, DPN_OK, pMSD[%p], Core Context[%p]", pEPD, pMSD, pMSD->Context);
|
|
pSPD->pPData->pfVtbl->CompleteDisconnect(pSPD->pPData->Parent, pMSD->Context, DPN_OK);
|
|
}
|
|
else
|
|
{
|
|
Unlock(&pMSD->CommandLock);
|
|
|
|
ASSERT(pMSD->CommandID == COMMAND_ID_DISC_RESPONSE);
|
|
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "(%p) Calling Core->IndicateConnectionTerminated, DPN_OK, Core Context[%p]", pEPD, pEPD->Context);
|
|
pSPD->pPData->pfVtbl->IndicateConnectionTerminated(pSPD->pPData->Parent, pEPD->Context, DPN_OK);
|
|
}
|
|
|
|
Lock(&pMSD->CommandLock);
|
|
Lock(&pEPD->EPLock);
|
|
RELEASE_EPD(pEPD, "UNLOCK (DISC COMMAND)"); // release hold on EPD, releases EPLock
|
|
RELEASE_MSD(pMSD, "Final Release On Complete"); // Finished with this one, releases CommandLock
|
|
}
|
|
|
|
/*
|
|
** Complete SP Connect
|
|
**
|
|
** A Connect Command has completed in the Service Provider. This does not mean our
|
|
** work is done... this means we now have a mapped EndPoint so we can exchange packets
|
|
** with this partner. We will now ping this partner to get an initial RTT and make sure
|
|
** there really is a protocol over there that will talk to us.
|
|
**
|
|
** Of course, if SP does not return success then we can nip this whole thing in
|
|
** the proverbial bud.
|
|
**
|
|
** ** COMMAND LOCK is held on entry **
|
|
*/
|
|
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "CompleteSPConnect"
|
|
|
|
VOID CompleteSPConnect(PMSD pMSD, PSPD pSPD, HRESULT hr)
|
|
{
|
|
PEPD pEPD;
|
|
|
|
DPFX(DPFPREP,5, "SP Completes Connect, pMSD[%p])", pMSD);
|
|
|
|
AssertCriticalSectionIsTakenByThisThread(&pMSD->CommandLock, TRUE);
|
|
|
|
pEPD = pMSD->pEPD;
|
|
|
|
ASSERT(!(pMSD->ulMsgFlags1 & MFLAGS_ONE_COMPLETE));
|
|
|
|
if(hr != DPN_OK)
|
|
{
|
|
// This will only happen once since DoCancel will not have done it because the IN_SP flag was set, and
|
|
// ConnectRetryTimeout has never yet been set.
|
|
if (pEPD)
|
|
{
|
|
ASSERT_EPD(pEPD);
|
|
Lock(&pEPD->EPLock);
|
|
|
|
// Unlink EPD from MSD
|
|
ASSERT(pEPD->pCommand == pMSD);
|
|
pEPD->pCommand = NULL;
|
|
DECREMENT_MSD(pMSD, "EPD Ref"); // Release Reference from EPD
|
|
|
|
DropLink(pEPD); // This releases the EPLock
|
|
}
|
|
DPFX(DPFPREP,5, "SP failed Connect, completing Connect, pMSD[%p], hr[%x]", pMSD, hr);
|
|
CompleteConnect(pMSD, pSPD, NULL, hr); // SP failed the connect call
|
|
return;
|
|
}
|
|
|
|
// After a successful connect, we should have an endpoint
|
|
pEPD = pMSD->pEPD;
|
|
ASSERT_EPD(pEPD);
|
|
|
|
Lock(&pEPD->EPLock);
|
|
|
|
// The endpoint should have already been linked to this MSD
|
|
ASSERT(pEPD->pCommand == pMSD);
|
|
|
|
if(pMSD->ulMsgFlags1 & MFLAGS_ONE_CANCELLED)
|
|
{
|
|
// We get here when someone called DoCancel while we were still in the SP. As above:
|
|
// This will only happen once since DoCancel will not have done it because the IN_SP flag was set, and
|
|
// ConnectRetryTimeout has never yet been set.
|
|
|
|
// Unlink EPD from MSD
|
|
ASSERT(pEPD->pCommand == pMSD);
|
|
pEPD->pCommand = NULL;
|
|
DECREMENT_MSD(pMSD, "EPD Ref"); // Release Reference from EPD
|
|
|
|
DropLink(pEPD); // This releases the EPLock
|
|
|
|
DPFX(DPFPREP,5, "(%p) Command is cancelled or timed out, Complete Connect, pMSD[%p]", pEPD, pMSD);
|
|
CompleteConnect(pMSD, pSPD, NULL, DPNERR_USERCANCEL);
|
|
return;
|
|
}
|
|
|
|
// Set up End Point Data /////////////////////////////////////////////////////////////////
|
|
|
|
// Transition state
|
|
ASSERT(pEPD->ulEPFlags & EPFLAGS_STATE_DORMANT);
|
|
pEPD->ulEPFlags &= ~(EPFLAGS_STATE_DORMANT);
|
|
pEPD->ulEPFlags |= EPFLAGS_STATE_CONNECTING;
|
|
|
|
// Send CONNECT
|
|
|
|
pEPD->dwSessID = pSPD->pPData->dwNextSessID++;
|
|
|
|
DPFX(DPFPREP,5, "(%p) Sending CONNECT Frame", pEPD);
|
|
(void) SendCommandFrame(pEPD, FRAME_EXOPCODE_CONNECT, 0);
|
|
|
|
// Set timer for reply, then wait for reply or TO.
|
|
pEPD->uiRetryTimeout = pSPD->pPData->dwConnectTimeout;
|
|
pEPD->uiRetryCount = pSPD->pPData->dwConnectRetries;
|
|
|
|
LOCK_EPD(pEPD, "LOCK (CONN Retry Timer)"); // Create reference for timer
|
|
DPFX(DPFPREP,5, "(%p) Setting Connect Retry Timer", pEPD);
|
|
SetMyTimer(pEPD->uiRetryTimeout, 100, ConnectRetryTimeout, (PVOID) pEPD, &pEPD->ConnectTimer, &pEPD->ConnectTimerUnique);
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
Unlock(&pMSD->CommandLock);
|
|
}
|
|
|
|
/*
|
|
** Connect Retry Timeout
|
|
**
|
|
** Retry timer has expired on a Connect operation. This one function
|
|
** is shared by Calling and Listening partners. Complexity is due to the
|
|
** fact that cancel code cannot always ensure that this handler will not
|
|
** run, so there are flags to signal various edge conditions (cancel, abort,
|
|
** completion, high-level timeout).
|
|
*/
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "ConnectRetryTimeout"
|
|
|
|
VOID CALLBACK
|
|
ConnectRetryTimeout(PVOID pvHandle, UINT uiUnique, PVOID pvUser)
|
|
{
|
|
PMSD pMSD;
|
|
PEPD pEPD = (PEPD) pvUser;
|
|
|
|
DPFX(DPFPREP,5, "ENTER Connect Retry Timeout pEPD=%p", pEPD);
|
|
|
|
ASSERT_EPD(pEPD);
|
|
|
|
Lock(&pEPD->EPLock);
|
|
|
|
if((pEPD->ConnectTimer != pvHandle)||(pEPD->ConnectTimerUnique != uiUnique))
|
|
{
|
|
// Timer been reset! This is a spurious fire and should be ignored.
|
|
RELEASE_EPD(pEPD, "UNLOCK: (Spurious (ie late) firing of CONNECT timer)"); // releases EPLock
|
|
DPFX(DPFPREP,7, "(%p) Ignoring late CONNECT timer", pEPD);
|
|
return;
|
|
}
|
|
|
|
pMSD = pEPD->pCommand;
|
|
|
|
if(pMSD == NULL)
|
|
{
|
|
pEPD->ConnectTimer = 0;
|
|
RELEASE_EPD(pEPD, "UNLOCK: (Conn retry timer - after completion)"); // releases EPLock
|
|
return;
|
|
}
|
|
|
|
ASSERT_MSD(pMSD);
|
|
|
|
// Make sure this doesn't go away when we leave the lock
|
|
LOCK_MSD(pMSD, "Hold For Lock");
|
|
Unlock(&pEPD->EPLock); // Release before taking higher level lock
|
|
|
|
// Take both locks in the proper order
|
|
Lock(&pMSD->CommandLock);
|
|
Lock(&pEPD->EPLock);
|
|
|
|
pEPD->ConnectTimer = 0;
|
|
|
|
// This timer should only be running in a CONNECTING state
|
|
if( (pMSD->ulMsgFlags1 & (MFLAGS_ONE_CANCELLED | MFLAGS_ONE_COMPLETE)) ||
|
|
(!(pEPD->ulEPFlags & EPFLAGS_STATE_CONNECTING)))
|
|
{
|
|
RELEASE_MSD(pMSD, "Hold for lock"); // Remove temporary reference and release lock
|
|
RELEASE_EPD(pEPD, "UNLOCK (Conn Retry Timer)"); // Remove reference for timer, releases EPLock
|
|
return; // and thats all for now
|
|
}
|
|
|
|
// IF more retries are allowed and command is still active, send another CONNECT frame
|
|
|
|
if(pEPD->uiRetryCount-- > 0)
|
|
{
|
|
pEPD->uiRetryTimeout = min(pEPD->uiRetryTimeout * 2, 5000); // exp backoff to a max of 5000ms until we establish our first RTT
|
|
|
|
if(pMSD->CommandID == COMMAND_ID_CONNECT)
|
|
{
|
|
DPFX(DPFPREP,5, "(%p) Sending CONNECT Frame", pEPD);
|
|
(void) SendCommandFrame(pEPD, FRAME_EXOPCODE_CONNECT, 0);
|
|
}
|
|
// Listen -- retry CONNECTED frame
|
|
else
|
|
{
|
|
pEPD->ulEPFlags |= EPFLAGS_CHECKPOINT_INIT; // We will expect a reply to this frame
|
|
DPFX(DPFPREP,5, "(%p) Sending CONNECTED Frame", pEPD);
|
|
(void) SendCommandFrame(pEPD, FRAME_EXOPCODE_CONNECTED, 0);
|
|
}
|
|
|
|
// Send the next ping
|
|
DPFX(DPFPREP,7, "(%p) Setting Connect Retry Timer", pEPD);
|
|
SetMyTimer(pEPD->uiRetryTimeout, 100, ConnectRetryTimeout, (PVOID) pEPD, &pEPD->ConnectTimer, &pEPD->ConnectTimerUnique);
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
RELEASE_MSD(pMSD, "Hold for lock"); // Remove temporary reference and release lock
|
|
|
|
// Since we have re-started timer, we don't adjust refcount
|
|
}
|
|
else
|
|
{
|
|
// We got no response and timed out.
|
|
|
|
if(pMSD->CommandID == COMMAND_ID_CONNECT)
|
|
{
|
|
DECREMENT_EPD(pEPD, "UNLOCK: (Connect Timer (Failure Path))");// Dec Ref for this timer, releases EPLock
|
|
|
|
// This will only happen once since we know DoCancel has not been called due to our CANCELLED check above,
|
|
// and if it is now called it will see our CANCELLED flag set below. We also know that the success case hasn't
|
|
// happened or COMPLETE above would have been set. We also have not had two timers get here because of the
|
|
// CANCELLED check above and set here.
|
|
pMSD->ulMsgFlags1 |= MFLAGS_ONE_CANCELLED;
|
|
|
|
// Unlink EPD from MSD
|
|
ASSERT(pEPD->pCommand == pMSD);
|
|
pEPD->pCommand = NULL;
|
|
DECREMENT_MSD(pMSD, "EPD Ref"); // Release Reference from EPD
|
|
|
|
DropLink(pEPD);// Releases EPLock
|
|
|
|
DECREMENT_MSD(pMSD, "Hold for lock"); // Remove temporary reference
|
|
|
|
DPFX(DPFPREP,1, "(%p) Connect retries exhausted, completing Connect, pMSD[%p]", pEPD, pMSD);
|
|
CompleteConnect(pMSD, pMSD->pSPD, NULL, DPNERR_NORESPONSE); // releases CommandLock
|
|
}
|
|
|
|
// Listen - clean up associated state info, then blow away end point
|
|
else
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) Connect retries exhausted on Listen, Kill Connection, pMSD[%p]", pEPD, pMSD);
|
|
|
|
if(pEPD->ulEPFlags & EPFLAGS_LINKED_TO_LISTEN)
|
|
{
|
|
pEPD->ulEPFlags &= ~(EPFLAGS_LINKED_TO_LISTEN);
|
|
pEPD->blSPLinkage.RemoveFromList(); // Unlink EPD from Listen Queue
|
|
}
|
|
|
|
ASSERT(pEPD->pCommand != NULL);
|
|
pEPD->pCommand = NULL; // Unlink listen from EPD
|
|
DECREMENT_MSD(pMSD, "EPD Ref"); // release reference for link to EPD
|
|
RELEASE_MSD(pMSD, "Hold for lock"); // Remove temporary reference and release lock
|
|
|
|
DECREMENT_EPD(pEPD, "UNLOCK: (Connect Timer (Failure Path))");// Dec Ref for this timer, SPLock not already held
|
|
DropLink(pEPD);
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
** Process Connection Request
|
|
**
|
|
** Somebody wants to connect to us. If we have a listen posted we will
|
|
** fill out a checkpoint structure to correlate his response and we will fire
|
|
** off a CONNECTED frame ourselves
|
|
**
|
|
** Since our connection will not be up until we receive a CONNECTED response
|
|
** to our response we will need to set up a retry timer ourselves.
|
|
*/
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "ProcessConnectRequest"
|
|
|
|
VOID ProcessConnectRequest(PSPD pSPD, PEPD pEPD, PCFRAME pCFrame)
|
|
{
|
|
PMSD pMSD = NULL;
|
|
|
|
DPFX(DPFPREP,DPF_CALLIN_LVL, "CONNECT REQUEST RECEIVED; EPD=%p SessID=%x", pEPD, pCFrame->dwSessID);
|
|
|
|
Lock(&pEPD->EPLock);
|
|
|
|
if((pMSD = pEPD->pCommand) == NULL)
|
|
{
|
|
// There are two cases: we are a connecting endpoint or we are a listening endpoint. In the connecting
|
|
// case we will fail in the following 'if' because we do not allow connections on non-listening endpoints.
|
|
// In the listening case, the fact that pMSD is NULL means that we have received the other side's
|
|
// CONNECTED packet, which also tells us that they have seen our CONNECTED packet. The only reason we
|
|
// would now be seeing a CONNECT is if a) there was a stale, late-delivered packet on the wire, or b)
|
|
// some malicious user is spoofing packets to us. In both cases, ignoring the packet is the right
|
|
// thing to do.
|
|
// There is a third possibility for the listening endpoint case, and that is that the other side went down
|
|
// and we didn't realize it, and they are now attempting to reconnect. The best we can do is wait the full
|
|
// timeout until the link is torn down on our side, and let their retries make the connection for them at
|
|
// that time. If we cut the timeout short, we have no way to know that a malicious user can't force
|
|
// legitimate connections closed by spoofing CONNECT packets.
|
|
DPFX(DPFPREP,1, "(%p) CONNECT Frame received on CONNECTED link, ignoring", pEPD);
|
|
DNASSERTX(FALSE, 3);
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
return;
|
|
}
|
|
|
|
ASSERT_MSD(pMSD);
|
|
LOCK_MSD(pMSD, "LOCK: Hold For Lock"); // Place reference on Cmd until we can lock it
|
|
Unlock(&pEPD->EPLock);
|
|
|
|
Lock(&pMSD->CommandLock);
|
|
Lock(&pEPD->EPLock); // Serialize access to EPD (this may not really be new sess)
|
|
|
|
// Make sure this endpoint was listening for connections
|
|
// This could be a Connect in which case this is a malicious packet. It could also be a Disonnect or Disconnect Response
|
|
// if the endpoint is being disconnected. In any case we just need to ignore the connect packet.
|
|
if(pMSD->CommandID != COMMAND_ID_LISTEN)
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) PROTOCOL RECEIVED CONNECT REQUEST ON A NON-LISTENING ENDPOINT, IGNORING, pMSD[%p]", pEPD, pMSD);
|
|
DNASSERTX(FALSE, 3);
|
|
Unlock(&pEPD->EPLock);
|
|
RELEASE_MSD(pMSD, "UNLOCK: Hold For Lock");
|
|
return;
|
|
}
|
|
|
|
// Make sure we can work with this version
|
|
if((pCFrame->dwVersion >> 16) != (DNET_VERSION_NUMBER >> 16))
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) PROTOCOL RECEIVED CONNECT REQUEST FROM AN INCOMPATIBLE VERSION(theirs %x, ours %x), DROPPING LINK", pEPD, pCFrame->dwVersion, DNET_VERSION_NUMBER);
|
|
DNASSERTX(FALSE, 2);
|
|
RELEASE_MSD(pMSD, "UNLOCK: Hold For Lock");
|
|
RejectInvalidPacket(pEPD, TRUE); // This releases the EPLock
|
|
return;
|
|
}
|
|
|
|
// Make sure the listen command is still valid
|
|
if(pMSD->ulMsgFlags1 & MFLAGS_ONE_CANCELLED)
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) PROTOCOL RECEIVED CONNECT REQUEST ON A LISTEN THAT IS CANCELLED, DROPPING LINK, pMSD[%p]", pEPD, pMSD);
|
|
RELEASE_MSD(pMSD, "UNLOCK: Hold For Lock");
|
|
RejectInvalidPacket(pEPD, TRUE); // This releases the EPLock
|
|
return;
|
|
}
|
|
|
|
// We shouldn't use the pMSD past here since this will unlock it
|
|
RELEASE_MSD(pMSD, "UNLOCK: Hold For Lock");
|
|
|
|
// Are we already connected?
|
|
if(pEPD->ulEPFlags & (EPFLAGS_STATE_CONNECTED | EPFLAGS_STATE_TERMINATING))
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) CONNECT Frame received on Connected or Terminating link, ignoring", pEPD);
|
|
|
|
// If connection has been completed then we don't need to do more work
|
|
|
|
// This can happen if we failed to cancel the connect timer in ProcessConnectedResponse and a CONNECT packet
|
|
// comes in.
|
|
Unlock(&pEPD->EPLock);
|
|
return;
|
|
}
|
|
|
|
// If we are already in a CONNECTING state then this is not the first CONNECT frame we have seen. If the SessID's
|
|
// match, then the partner probably didn't hear our first response, so we will resend it. If the SessID's don't
|
|
// match, then either the partner aborted and is starting with a new SessID, or a malicious party is spoofing the
|
|
// partner's address, and sending a bogus packet. In either of these cases we will ignore the connect. A partner
|
|
// aborting a connect mid-way probably crashed anyway, and waiting for us to timeout the first connect attempt
|
|
// will be the least of their worries.
|
|
|
|
if(pEPD->ulEPFlags & EPFLAGS_STATE_CONNECTING)
|
|
{
|
|
if(pCFrame->dwSessID != pEPD->dwSessID)
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) Received non-matching SessionID, ignoring CONNECT", pEPD);
|
|
Unlock(&pEPD->EPLock);
|
|
return;
|
|
}
|
|
|
|
// Unexpected CONNECT Frame has same Session ID. Partner probably lost our response. We will
|
|
// respond again to this one.
|
|
|
|
DPFX(DPFPREP,1, "(%p) Received duplicate CONNECT request. Sending another response...", pEPD);
|
|
|
|
// Listen side must set this before sending CONNECTED
|
|
pEPD->ulEPFlags |= EPFLAGS_CHECKPOINT_INIT; // We will expect a reply to this frame
|
|
|
|
// If this fails they will be sending another CONNECT anyway, so we do nothing
|
|
(void) SendCommandFrame(pEPD, FRAME_EXOPCODE_CONNECTED, pCFrame->bMsgID);
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
return;
|
|
}
|
|
|
|
// Transition state
|
|
ASSERT(pEPD->ulEPFlags & EPFLAGS_STATE_DORMANT);
|
|
pEPD->ulEPFlags &= ~(EPFLAGS_STATE_DORMANT);
|
|
pEPD->ulEPFlags |= EPFLAGS_STATE_CONNECTING;
|
|
|
|
pEPD->dwSessID = pCFrame->dwSessID; // Use this SessID in all C-traffic
|
|
pEPD->ulEPFlags |= EPFLAGS_CHECKPOINT_INIT; // We will expect a reply to this frame
|
|
|
|
DPFX(DPFPREP,5, "(%p) Sending CONNECTED Frame", pEPD);
|
|
(void) SendCommandFrame(pEPD, FRAME_EXOPCODE_CONNECTED, pCFrame->bMsgID);
|
|
|
|
pEPD->uiRetryTimeout = pSPD->pPData->dwConnectTimeout; // Got to start somewhere
|
|
pEPD->uiRetryCount = pSPD->pPData->dwConnectRetries; // w/exponential wait
|
|
|
|
LOCK_EPD(pEPD, "LOCK: (CONNECT RETRY TIMER)"); // Create reference for timer
|
|
DPFX(DPFPREP,5, "(%p) Setting Connect Timer", pEPD);
|
|
SetMyTimer(pEPD->uiRetryTimeout, 100, ConnectRetryTimeout, (PVOID) pEPD, &pEPD->ConnectTimer, &pEPD->ConnectTimerUnique);
|
|
if (pEPD->ConnectTimer == 0)
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) Setting Connect Retry Timer failed", pEPD);
|
|
|
|
// If we can't even set timers due to low memory, then the best we can do is
|
|
// abandon this new connection and hope to give good service to our existing
|
|
// connections.
|
|
DECREMENT_EPD(pEPD, "UNLOCK: (CONNECT RETRY TIMER)");
|
|
|
|
pEPD->ulEPFlags &= ~(EPFLAGS_LINKED_TO_LISTEN);
|
|
pEPD->blSPLinkage.RemoveFromList(); // Unlink EPD from Listen Queue
|
|
|
|
// Unlink the MSD from the EPD
|
|
ASSERT(pEPD->pCommand == pMSD);
|
|
pEPD->pCommand = NULL;
|
|
|
|
DropLink(pEPD); // This releases EPLock
|
|
|
|
Lock(&pMSD->CommandLock);
|
|
RELEASE_MSD(pMSD, "EPD Ref");
|
|
|
|
return;
|
|
}
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
}
|
|
|
|
/*
|
|
** Process Connected Response
|
|
**
|
|
** A response to a connection request has arrived (or a response to
|
|
** our connection response). Now the connection is officially up (on
|
|
** our end of the circuit). Set the link-state according to our first
|
|
** RTT sample and get ready to party.
|
|
**
|
|
** If we are the originating party, we will want to send a
|
|
** CONNECTED frame to our partner, even though the connection is
|
|
** complete from our perspective. This will allow partner to establish
|
|
** his baseline RTT and clock bias as we can do here. In this case, he
|
|
** will have his POLL bit set in the frame we just received.
|
|
**
|
|
** Now, we might get additional CONNECTED frames after the first one
|
|
** where we startup the link. This would most likely be due to our CONNECTED
|
|
** response getting lost. So if we get a CONNECTED frame with POLL set
|
|
** after our link is up, we will just go ahead and respond again without
|
|
** adjusting our state.
|
|
**
|
|
** Note about Locks:
|
|
**
|
|
** This code is complicated by the precedence of CritSec ownership. To simplify
|
|
** as much as possible we will take the Listen command lock at the very start of the
|
|
** procedure (when appropriate) because it has the highest level lock. This prevents
|
|
** us from completing the whole connection process and then finding that the Listen
|
|
** went away so we can't indicate it to the user.
|
|
**
|
|
** We keep a RefCnt on the Listen so it won't go away while a new session
|
|
** is pending on it.
|
|
*/
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "ProcessConnectedResponse"
|
|
|
|
VOID ProcessConnectedResponse(PSPD pSPD, PEPD pEPD, PCFRAME pCFrame, DWORD tNow)
|
|
{
|
|
PCHKPT pCP;
|
|
PMSD pMSD;
|
|
|
|
DPFX(DPFPREP,DPF_CALLIN_LVL, "CONNECT RESPONSE RECEIVED (pEPD=0x%p)", pEPD);
|
|
|
|
Lock(&pEPD->EPLock);
|
|
|
|
// If the link has not seen a CONNECT or issued a CONNECT, we do not expect a CONNECTED.
|
|
// Since this is the only reason this EPD was created, we will tear it down by rejecting
|
|
// the connection.
|
|
if (pEPD->ulEPFlags & EPFLAGS_STATE_DORMANT)
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) CONNECTED response received on a dormant link, dropping link", pEPD);
|
|
RejectInvalidPacket(pEPD, TRUE); // This will release the EPLock
|
|
return;
|
|
}
|
|
|
|
// There is a possibility that the link is in the terminating state. If so, we don't
|
|
// care about CONNECTED packets.
|
|
if (pEPD->ulEPFlags & EPFLAGS_STATE_TERMINATING)
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) CONNECTED response received on a terminating link, ignoring", pEPD);
|
|
Unlock(&pEPD->EPLock);
|
|
return;
|
|
}
|
|
|
|
// At this point either we are connecting and this is a response, or someone has connected to us, and this is
|
|
// the reply to our response. Note that this could be a duplicate of one of these cases, so we may already
|
|
// be in a connected state. There is also the possiblity that these are malicious packets.
|
|
ASSERT(pEPD->ulEPFlags & (EPFLAGS_STATE_CONNECTING | EPFLAGS_STATE_CONNECTED));
|
|
|
|
// If the SessID does not match, ignore the CONNECTED packet. This is either a malicious attempt at messing
|
|
// up a legitimate connection, or our partner aborted and has come back in with a new session ID.
|
|
if(pEPD->dwSessID != pCFrame->dwSessID)
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) CONNECTED response has bad SessID, ignoring", pEPD);
|
|
Unlock(&pEPD->EPLock);
|
|
return;
|
|
}
|
|
|
|
// If we have completed our side of the connection then our only responsibility is to send responses
|
|
// if our partner is still POLLING us.
|
|
|
|
if(pEPD->ulEPFlags & EPFLAGS_STATE_CONNECTED)
|
|
{
|
|
// A Listen will set POLL on the CONNECTED packet, a Connect will not
|
|
if(pCFrame->bCommand & PACKET_COMMAND_POLL)
|
|
{
|
|
// This side must be a listen unless this packet is malicious
|
|
|
|
DPFX(DPFPREP,5, "(%p) Duplicate CONNECTED frame, sending another response...", pEPD);
|
|
|
|
// If this fails we will let the partner's retry catch it.
|
|
(void) SendCommandFrame(pEPD, FRAME_EXOPCODE_CONNECTED, pCFrame->bMsgID);
|
|
}
|
|
Unlock(&pEPD->EPLock);
|
|
return;
|
|
}
|
|
|
|
// Since we are not CONNECTED yet, we must be in a CONNECTING state in order to receive a
|
|
// CONNECTED response.
|
|
ASSERT(pEPD->ulEPFlags & EPFLAGS_STATE_CONNECTING);
|
|
|
|
// MSD should not be NULL if we are in the CONNECTING state
|
|
pMSD = pEPD->pCommand;
|
|
ASSERT_MSD(pMSD);
|
|
LOCK_MSD(pMSD, "Hold For Lock"); // Place reference on Cmd until we can lock it
|
|
Unlock(&pEPD->EPLock);
|
|
|
|
Lock(&pMSD->CommandLock);
|
|
Lock(&pEPD->EPLock);
|
|
|
|
// Since we left the EPLock, we must verify that we are still in the connecting state
|
|
if (!(pEPD->ulEPFlags & EPFLAGS_STATE_CONNECTING))
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) EPD left the CONNECTING state while we were out of the lock, ignoring CONNECTED frame", pEPD);
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
RELEASE_MSD(pMSD, "Hold For Lock"); // This releases the command lock
|
|
return;
|
|
}
|
|
|
|
if(pMSD->ulMsgFlags1 & (MFLAGS_ONE_CANCELLED | MFLAGS_ONE_COMPLETE))
|
|
{
|
|
DPFX(DPFPREP,1, "(%p) Connect/Listen command cancelled or complete, ignoring CONNECTED frame", pEPD);
|
|
|
|
// Whoever cancelled the Listen should be disconnecting this connection too
|
|
// so all we have to do here is bail out.
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
RELEASE_MSD(pMSD, "Hold For Lock"); // This releases the command lock
|
|
return;
|
|
}
|
|
|
|
// Next, take care of this guy's reply if we still owe him one
|
|
|
|
// A Listen will set POLL on the CONNECTED packet, a Connect will not
|
|
if(pCFrame->bCommand & PACKET_COMMAND_POLL)
|
|
{
|
|
// This side must be a listen unless this packet is malicious
|
|
|
|
DPFX(DPFPREP,5, "(%p) Sending CONNECTED Frame", pEPD);
|
|
if(SendCommandFrame(pEPD, FRAME_EXOPCODE_CONNECTED, pCFrame->bMsgID) != DPN_OK)
|
|
{
|
|
DPFX(DPFPREP,5, "(%p) Sending CONNECTED Frame Failed", pEPD);
|
|
|
|
// We cannot complete the connection... we will just let things time out
|
|
RELEASE_MSD(pMSD, "Hold For Lock");
|
|
Unlock(&pEPD->EPLock); // Protect the pCommand field
|
|
return;
|
|
}
|
|
}
|
|
|
|
// Now we can setup our new link, but only if this frame Correlates to a checkpoint we have outstanding
|
|
// so we can seed our state variables.
|
|
|
|
// Can we correlate resp?
|
|
if((pCP = LookupCheckPoint(pEPD, pCFrame->bRspID)) != NULL)
|
|
{
|
|
// We are connected, so shut off retry timer
|
|
if(pEPD->ConnectTimer != 0)
|
|
{
|
|
DPFX(DPFPREP,5, "(%p) Cancelling Connect Timer", pEPD);
|
|
if(CancelMyTimer(pEPD->ConnectTimer, pEPD->ConnectTimerUnique) == DPN_OK)
|
|
{
|
|
DECREMENT_EPD(pEPD, "UNLOCK: (Conn Retry Timer - Connect Complete)"); // remove reference for timer, SPLock not already held
|
|
}
|
|
else
|
|
{
|
|
DPFX(DPFPREP,5, "(%p) Cancelling Connect Timer Failed", pEPD);
|
|
}
|
|
pEPD->ConnectTimer = 0; // This will prevent timer from trying to do any work if it couldn't cancel
|
|
}
|
|
|
|
// Transition state
|
|
ASSERT(pEPD->ulEPFlags & EPFLAGS_STATE_CONNECTING);
|
|
pEPD->ulEPFlags &= ~(EPFLAGS_STATE_CONNECTING);
|
|
pEPD->ulEPFlags |= EPFLAGS_STATE_CONNECTED | EPFLAGS_STREAM_UNBLOCKED;// Link is open for business
|
|
|
|
DPFX(DPFPREP,1, "(%p) Partner Reported Version: %x, Our Version: %x, Initial RTT %d - %d = %d", pEPD, pCFrame->dwVersion, DNET_VERSION_NUMBER, tNow, pCP->tTimestamp, tNow - pCP->tTimestamp);
|
|
|
|
if(pEPD->ulEPFlags & EPFLAGS_LINKED_TO_LISTEN)
|
|
{
|
|
pEPD->ulEPFlags &= ~(EPFLAGS_LINKED_TO_LISTEN);
|
|
pEPD->blSPLinkage.RemoveFromList(); // Unlink EPD from Listen Queue
|
|
}
|
|
|
|
InitLinkParameters(pEPD, (tNow - pCP->tTimestamp), sizeof(CFRAME), (pCP->tTimestamp - pCFrame->tTimestamp), tNow);
|
|
ChkPtPool->Release(ChkPtPool, pCP);
|
|
|
|
DPFX(DPFPREP,5, "(%p) N(R) = 0, N(S) = 0", pEPD);
|
|
pEPD->bNextSend = 0;
|
|
pEPD->bNextReceive = 0;
|
|
|
|
FlushCheckPoints(pEPD); // Make sure we do this before the InitCheckPoint
|
|
|
|
/*
|
|
** It turns out that the first RTT measurement is a very bad one (slow) because because
|
|
** it includes overhead for opening and binding a new socket, endpoint creation, etc.
|
|
** Therefore each side will take another quick sample right away. The initial calculations
|
|
** above will still serve as an initial RTT until this better sample is available
|
|
*/
|
|
|
|
PerformCheckpoint(pEPD); // Take another RTT sample
|
|
|
|
// Cleanup connect operation
|
|
if(pMSD->CommandID == COMMAND_ID_CONNECT)
|
|
{
|
|
// There was a CONNECT Command issued that now must be completed
|
|
|
|
DECREMENT_MSD(pMSD, "Hold For Lock"); // Remove temporary reference from above.
|
|
|
|
// This will not happen twice because both COMPLETE and CANCELLED are checked above, and the
|
|
// call to CompleteConnect will set COMPLETE.
|
|
|
|
// Unlink the MSD from the EPD
|
|
ASSERT(pEPD->pCommand == pMSD);
|
|
pEPD->pCommand = NULL;
|
|
DECREMENT_MSD(pMSD, "EPD Ref"); // Release reference for EPD link
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
|
|
CompleteConnect(pMSD, pSPD, pEPD, DPN_OK); // This releases the MSD Lock
|
|
}
|
|
else
|
|
{ // LISTENING
|
|
// We were the listener. We will indicate a Connect event on the listen
|
|
// command w/o completing the Listen
|
|
|
|
ASSERT(pMSD->CommandID == COMMAND_ID_LISTEN);
|
|
|
|
// We know this will only happen once because the person who does it will transition us out
|
|
// of the CONNECTING state, and we can't get here unless we are in that state.
|
|
|
|
// Unlink the MSD from the EPD
|
|
ASSERT(pEPD->pCommand == pMSD);
|
|
pEPD->pCommand = NULL;
|
|
DECREMENT_MSD(pMSD, "EPD Ref"); // Release reference for EPD link
|
|
|
|
Unlock(&pEPD->EPLock);
|
|
|
|
Unlock(&pMSD->CommandLock);
|
|
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "(%p) Calling Core->IndicateConnect, pMSD[%p], Core Context[%p]", pEPD, pMSD, pMSD->Context);
|
|
pSPD->pPData->pfVtbl->IndicateConnect(pSPD->pPData->Parent, pMSD->Context, (PHANDLE) pEPD, &pEPD->Context);
|
|
|
|
// Complete any receives that queued while waiting for IndicateConnect
|
|
Lock(&pEPD->EPLock);
|
|
ReceiveComplete(pEPD); // releases EPLock
|
|
|
|
// Release the final reference on the MSD AFTER indicating to the Core
|
|
Lock(&pMSD->CommandLock);
|
|
RELEASE_MSD(pMSD, "Hold For Lock"); // release temp MSD (releases lock)
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/*
|
|
** Uncorrelated CONNECTED frame. How can this happen? Parter's response must
|
|
** have been dropped, so he is retrying his CONN frame. Since we are trying to
|
|
** measure an accurate RTT we dont want to use his retry against our original
|
|
** request, so he zeros out his Resp correlator. We will eventually retry with
|
|
** new correlator and hopefully that frame will get through.
|
|
*/
|
|
|
|
DPFX(DPFPREP,1, "(%p) Uncorrelated CONNECTED frame arrives", pEPD);
|
|
Unlock(&pEPD->EPLock);
|
|
RELEASE_MSD(pMSD, "Hold For Lock");
|
|
}
|
|
}
|
|
|
|
/*
|
|
** Drop Link
|
|
**
|
|
** For whatever reason we are dropping an active link. This requires us to
|
|
** Cancel any outstanding commands and give an indication to the user.
|
|
**
|
|
**
|
|
** ** CALLED WITH EPD->EPLOCK HELD; RETURNS WITH LOCK RELEASED **
|
|
*/
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "DropLink"
|
|
|
|
VOID DropLink(PEPD pEPD)
|
|
{
|
|
PRCD pRCD;
|
|
PRCD pNext;
|
|
CBilink *pLink;
|
|
PSPRECEIVEDBUFFER pRcvBuff = NULL;
|
|
PProtocolData pPData;
|
|
BOOL fIndicateDisconnect;
|
|
|
|
DPFX(DPFPREP,2, "Drop Link %p (refcnt=%d)", pEPD, pEPD->lRefCnt);
|
|
|
|
ASSERT_EPD(pEPD);
|
|
AssertCriticalSectionIsTakenByThisThread(&pEPD->EPLock, TRUE);
|
|
|
|
PSPD pSPD = pEPD->pSPD;
|
|
|
|
// First set/clear flags to prevent any new commands from issueing
|
|
|
|
// We will not indicate disconnect if the Core never knew about the connetion
|
|
fIndicateDisconnect = (pEPD->ulEPFlags & EPFLAGS_STATE_CONNECTED);
|
|
|
|
// Transition state
|
|
pEPD->ulEPFlags &= ~(EPFLAGS_STATE_CONNECTING | EPFLAGS_STATE_DORMANT | EPFLAGS_STATE_CONNECTED |
|
|
EPFLAGS_SDATA_READY | EPFLAGS_STREAM_UNBLOCKED); // Link is now down
|
|
pEPD->ulEPFlags |= EPFLAGS_STATE_TERMINATING; // Accept no new commands
|
|
|
|
// I am creating a RefCnt bump for the send pipeline, which means we will no longer pull EPDs off
|
|
// the pipeline here. The clearing of the flags above will cause the EPD to be dropped from the
|
|
// pipeline the next time it is due to be serviced. We CAN still clean up all the frames'n'stuff
|
|
// because the send loop doesnt need to actually DO anything with this EPD. This behavior allows
|
|
// the SendThread to loop through the pipeline queue, surrendering locks, without having the queue
|
|
// changing beneath it.
|
|
|
|
// Stop all timers (there are five now)
|
|
|
|
if(pEPD->RetryTimer)
|
|
{
|
|
if(CancelMyTimer(pEPD->RetryTimer, pEPD->RetryTimerUnique) == DPN_OK)
|
|
{
|
|
DECREMENT_EPD(pEPD, "UNLOCK (DROP RETRY)"); // SPLock not already held
|
|
}
|
|
pEPD->RetryTimer = 0;
|
|
}
|
|
if(pEPD->ConnectTimer)
|
|
{
|
|
if(CancelMyTimer(pEPD->ConnectTimer, pEPD->ConnectTimerUnique) == DPN_OK)
|
|
{
|
|
DECREMENT_EPD(pEPD, "UNLOCK (DROP CONNECT RETRY)"); // SPLock not already held
|
|
}
|
|
pEPD->ConnectTimer = 0;
|
|
}
|
|
if(pEPD->DelayedAckTimer)
|
|
{
|
|
if(CancelMyTimer(pEPD->DelayedAckTimer, pEPD->DelayedAckTimerUnique) == DPN_OK)
|
|
{
|
|
DECREMENT_EPD(pEPD, "UNLOCK (DROP DELAYEDACK)"); // SPLock not already held
|
|
}
|
|
pEPD->DelayedAckTimer = 0;
|
|
}
|
|
if(pEPD->DelayedMaskTimer)
|
|
{
|
|
if(CancelMyTimer(pEPD->DelayedMaskTimer, pEPD->DelayedMaskTimerUnique) == DPN_OK)
|
|
{
|
|
DECREMENT_EPD(pEPD, "UNLOCK (DROP DELAYED MASK)"); // SPLock not already held
|
|
}
|
|
pEPD->DelayedMaskTimer = 0;
|
|
}
|
|
if(pEPD->SendTimer)
|
|
{
|
|
if(CancelMyTimer(pEPD->SendTimer, pEPD->SendTimerUnique) == DPN_OK)
|
|
{
|
|
DECREMENT_EPD(pEPD, "UNLOCK (DROP SENDTIMER)"); // SPLock not already held
|
|
pEPD->SendTimer = 0;
|
|
}
|
|
}
|
|
if(pEPD->BGTimer)
|
|
{
|
|
if(CancelMyTimer(pEPD->BGTimer, pEPD->BGTimerUnique) == DPN_OK)
|
|
{
|
|
DECREMENT_EPD(pEPD, "UNLOCK (DROP BG TIMER)"); // SPLock not already held
|
|
pEPD->BGTimer = 0;
|
|
}
|
|
}
|
|
|
|
AbortSendsOnConnection(pEPD); // Cancel pending commands; releases EPLock
|
|
|
|
Lock(&pEPD->EPLock);
|
|
|
|
// Connects, Listens, and Disconnects are associated with an EPD through the pCommand member. AbortSends will
|
|
// have removed any Disconnects, and no Connects or Listens should still be hanging on when we call DropLink.
|
|
ASSERT(pEPD->pCommand == NULL);
|
|
|
|
// Now we clean up any receives in progress. We throw away any partial or mis-ordered messages.
|
|
while((pRCD = pEPD->pNewMessage) != NULL)
|
|
{
|
|
ASSERT_RCD(pRCD);
|
|
|
|
pEPD->pNewMessage = pRCD->pMsgLink;
|
|
if(pRCD->pRcvBuff == NULL)
|
|
{
|
|
ASSERT(pRCD->ulRFlags & (RFLAGS_FRAME_INDICATED_NONSEQ | RFLAGS_FRAME_LOST));
|
|
}
|
|
|
|
RELEASE_SP_BUFFER(pRCD->pRcvBuff);
|
|
|
|
RELEASE_RCD(pRCD);
|
|
}
|
|
|
|
while(!pEPD->blOddFrameList.IsEmpty())
|
|
{
|
|
pLink = pEPD->blOddFrameList.GetNext();
|
|
pRCD = CONTAINING_RECORD(pLink, RCD, blOddFrameLinkage);
|
|
ASSERT_RCD(pRCD);
|
|
|
|
pLink->RemoveFromList();
|
|
|
|
if(pRCD->pRcvBuff == NULL)
|
|
{
|
|
ASSERT(pRCD->ulRFlags & (RFLAGS_FRAME_INDICATED_NONSEQ | RFLAGS_FRAME_LOST));
|
|
}
|
|
|
|
RELEASE_SP_BUFFER(pRCD->pRcvBuff);
|
|
|
|
RELEASE_RCD(pRCD);
|
|
}
|
|
|
|
while(!pEPD->blCompleteList.IsEmpty())
|
|
{
|
|
pLink = pEPD->blCompleteList.GetNext();
|
|
pRCD = CONTAINING_RECORD(pLink, RCD, blCompleteLinkage);
|
|
ASSERT_RCD(pRCD);
|
|
|
|
pLink->RemoveFromList();
|
|
|
|
ASSERT(pEPD->uiCompleteMsgCount > 0);
|
|
pEPD->uiCompleteMsgCount--;
|
|
|
|
while(pRCD != NULL)
|
|
{
|
|
ASSERT_RCD(pRCD);
|
|
pNext = pRCD->pMsgLink;
|
|
|
|
RELEASE_SP_BUFFER(pRCD->pRcvBuff);
|
|
|
|
RELEASE_RCD(pRCD);
|
|
pRCD = pNext;
|
|
}
|
|
}
|
|
|
|
pPData = pEPD->pSPD->pPData;
|
|
IDP8ServiceProvider *pSPIntf = pEPD->pSPD->IISPIntf;
|
|
|
|
if (!(pEPD->ulEPFlags & EPFLAGS_INDICATED_DISCONNECT))
|
|
{
|
|
if (fIndicateDisconnect)
|
|
{
|
|
// Make sure we are the only one that indicates the disconnect
|
|
pEPD->ulEPFlags |= EPFLAGS_INDICATED_DISCONNECT;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Someone else beat us to it.
|
|
fIndicateDisconnect = FALSE;
|
|
}
|
|
|
|
// We need to make sure that we don't allow DNPRemoveServiceProvider to complete until we are out of the Core
|
|
// and SP. This reference will do that, and is released at the end of this function.
|
|
LOCK_EPD(pEPD, "Drop Link Temp Ref");
|
|
|
|
// Remove the base reference if it still exists
|
|
if(!(pEPD->ulEPFlags & EPFLAGS_KILLED))
|
|
{
|
|
pEPD->ulEPFlags |= EPFLAGS_KILLED;
|
|
|
|
RELEASE_EPD(pEPD, "UNLOCK (KILLCONN - Base Ref)"); // RELEASE the EPD, releases EPLock
|
|
}
|
|
else
|
|
{
|
|
Unlock(&pEPD->EPLock);
|
|
}
|
|
|
|
if(pRcvBuff)
|
|
{
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "(%p) Calling SP->ReturnReceiveBuffers, pSPD[%p], pRcvBuff[%p]", pEPD, pSPD, pRcvBuff);
|
|
IDP8ServiceProvider_ReturnReceiveBuffers(pSPIntf, pRcvBuff);
|
|
}
|
|
|
|
// Tell user that session is over
|
|
|
|
// If the Core previously knew about this endpoint, and we have not yet indicated disconnect to the
|
|
// Core, we need to do it now.
|
|
if(fIndicateDisconnect)
|
|
{
|
|
DPFX(DPFPREP,DPF_CALLOUT_LVL, "(%p) Calling Core->IndicateConnectionTerminated, DPNERR_CONNECTIONLOST, Core Context[%p]", pEPD, pEPD->Context);
|
|
pEPD->pSPD->pPData->pfVtbl->IndicateConnectionTerminated(pPData->Parent, pEPD->Context, DPNERR_CONNECTIONLOST);
|
|
}
|
|
|
|
Lock(&pEPD->EPLock);
|
|
RELEASE_EPD(pEPD, "Drop Link Temp Ref");
|
|
}
|