349 lines
8.1 KiB
C
349 lines
8.1 KiB
C
|
|
/*************************************************************************
|
|
*
|
|
* watchdog.c
|
|
*
|
|
* IPX Transport Driver - Watch Dog Rouines
|
|
*
|
|
* Copyright 1998, Microsoft
|
|
*
|
|
*************************************************************************/
|
|
|
|
#include <ntddk.h>
|
|
#include <tdi.h>
|
|
|
|
/*
|
|
* The following defines are necessary since they are referenced in
|
|
* afd.h but defined in sdk/inc/winsock2.h, which we can't include here.
|
|
*/
|
|
#define SG_UNCONSTRAINED_GROUP 0x01
|
|
#define SG_CONSTRAINED_GROUP 0x02
|
|
#include <afd.h>
|
|
|
|
#include <isnkrnl.h>
|
|
#include <ndis.h>
|
|
#include <wsnwlink.h>
|
|
|
|
#include <winstaw.h>
|
|
#define _DEFCHARINFO_
|
|
#include <icadd.h>
|
|
#include <ctxdd.h>
|
|
#include <icaipx.h>
|
|
//#include <cxstatus.h>
|
|
#include <sdapi.h>
|
|
#include <td.h>
|
|
|
|
#include "tdtdi.h"
|
|
#include "tdipx.h"
|
|
|
|
|
|
/*=============================================================================
|
|
== External Functions Defined
|
|
=============================================================================*/
|
|
|
|
NTSTATUS InitializeWatchDog( PTD );
|
|
VOID WatchDogTimer( PTD, PVOID );
|
|
|
|
|
|
/*=============================================================================
|
|
== Internel Functions Defined
|
|
=============================================================================*/
|
|
|
|
VOID _GetTimeoutValue( PTD, ULONG, PULONG );
|
|
|
|
|
|
/*=============================================================================
|
|
== External Functions used
|
|
=============================================================================*/
|
|
|
|
NTSTATUS _DoTdiAction( PFILE_OBJECT, ULONG, PCHAR, ULONG );
|
|
|
|
NTSTATUS
|
|
_TdiSendDatagram(
|
|
IN PTD pTd,
|
|
IN PIRP Irp OPTIONAL,
|
|
IN PFILE_OBJECT FileObject,
|
|
IN PDEVICE_OBJECT DeviceObject,
|
|
IN PTRANSPORT_ADDRESS pRemoteAddress,
|
|
IN ULONG RemoteAddressLength,
|
|
IN PVOID pBuffer,
|
|
IN ULONG BufferLength
|
|
);
|
|
|
|
|
|
/*******************************************************************************
|
|
*
|
|
* InitializeWatchDog
|
|
*
|
|
* Initialize watch dog
|
|
*
|
|
* ENTRY:
|
|
* pTd (input)
|
|
* Pointer to td data structure
|
|
*
|
|
* EXIT:
|
|
* STATUS_SUCCESS - no error, non-error match found
|
|
*
|
|
******************************************************************************/
|
|
|
|
NTSTATUS
|
|
InitializeWatchDog( PTD pTd )
|
|
{
|
|
ULONG Timeout;
|
|
NTSTATUS Status;
|
|
PTDIPX pTdIpx;
|
|
|
|
TRACE(( pTd->pContext, TC_TD, TT_API1, "TDIPX: InitializeWatchDog\n" ));
|
|
|
|
/*
|
|
* Get pointer to IPX structure
|
|
*/
|
|
pTdIpx = (PTDIPX) pTd->pPrivate;
|
|
|
|
/*
|
|
* If we really want a timer.
|
|
*/
|
|
if ( !pTdIpx->AliveTime )
|
|
return( STATUS_SUCCESS );
|
|
|
|
/*
|
|
* Create watch dog
|
|
*/
|
|
ASSERT( pTdIpx->pAliveTimer == NULL );
|
|
Status = IcaTimerCreate( pTd->pContext, &pTdIpx->pAliveTimer );
|
|
if ( !NT_SUCCESS(Status) )
|
|
goto badtimer;
|
|
|
|
/*
|
|
* Arm timer
|
|
*/
|
|
_GetTimeoutValue( pTd, pTdIpx->AliveTime, &Timeout );
|
|
IcaTimerStart( pTdIpx->pAliveTimer,
|
|
WatchDogTimer,
|
|
NULL,
|
|
Timeout,
|
|
ICALOCK_DRIVER );
|
|
|
|
badtimer:
|
|
TRACE(( pTd->pContext, TC_TD, TT_API1, "TDIPX: InitializeWatchDog %u, Status=0x%x\n", Timeout, Status ));
|
|
return( Status );
|
|
}
|
|
|
|
|
|
/*******************************************************************************
|
|
*
|
|
* _GetTimeoutValue
|
|
*
|
|
* algorithm to calculate the poll timeout value
|
|
*
|
|
*
|
|
* ENTRY:
|
|
* pTd (input)
|
|
* pointer to TD data structure
|
|
* PollTime (input)
|
|
* time gap between two pollings (msec)
|
|
* pTimeOut (output)
|
|
* timeout value for next poll (msec)
|
|
*
|
|
******************************************************************************/
|
|
|
|
VOID
|
|
_GetTimeoutValue( PTD pTd, ULONG PollTime, PULONG pTimeOut )
|
|
{
|
|
|
|
LARGE_INTEGER CurrentTime;
|
|
LARGE_INTEGER Ticks;
|
|
ULONG Time;
|
|
ULONG Remain;
|
|
|
|
KeQuerySystemTime( &CurrentTime );
|
|
|
|
/* The currentTime is in 100nsec units and need convert to msec units. */
|
|
|
|
Ticks = RtlExtendedLargeIntegerDivide( CurrentTime, 10000, &Remain );
|
|
|
|
RtlExtendedLargeIntegerDivide( Ticks, PollTime, &Remain );
|
|
|
|
*pTimeOut = PollTime - Remain;
|
|
|
|
TRACE(( pTd->pContext, TC_TD, TT_API3, "TDIPX: _GetTimeoutValue %u -> %u\n",
|
|
PollTime, *pTimeOut));
|
|
}
|
|
|
|
|
|
/*******************************************************************************
|
|
*
|
|
* WatchDogTimer
|
|
*
|
|
*
|
|
* ENTRY:
|
|
* pTd (input)
|
|
* pointer to TD data structure
|
|
* pParam (input)
|
|
* not used
|
|
*
|
|
* EXIT:
|
|
* nothing
|
|
*
|
|
******************************************************************************/
|
|
|
|
VOID
|
|
WatchDogTimer( PTD pTd, PVOID pParam )
|
|
{
|
|
ICA_CHANNEL_COMMAND Command;
|
|
CHAR ptype;
|
|
ULONG Timeout;
|
|
BYTE Buffer[1];
|
|
NTSTATUS Status;
|
|
PTDIPX pTdIpx;
|
|
PTDTDI pTdTdi;
|
|
PTD_ENDPOINT pEndpoint;
|
|
|
|
/*
|
|
* Get the connection endpoint for the remote system.
|
|
* If there is none, then the endpoint must have been closed
|
|
* so there is nothing to do.
|
|
*/
|
|
pTdTdi = (PTDTDI)pTd->pAfd;
|
|
if ( (pEndpoint = pTdTdi->pConnectionEndpoint) == NULL )
|
|
return;
|
|
ASSERT( pEndpoint->EndpointType == TdiConnectionDatagram );
|
|
|
|
/*
|
|
* Get pointer to IPX structure
|
|
*/
|
|
pTdIpx = (PTDIPX) pTd->pPrivate;
|
|
ASSERT( pTdIpx );
|
|
|
|
TRACE(( pTd->pContext, TC_TD, TT_API4, "TDIPX: WatchDogTimer\n" ));
|
|
|
|
/*
|
|
* Get time for next watchdog event
|
|
*/
|
|
_GetTimeoutValue( pTd, pTdIpx->AliveTime, &Timeout );
|
|
|
|
/*
|
|
* Check if client has sent data recently
|
|
*/
|
|
if ( pTdIpx->fClientAlive )
|
|
goto clientok;
|
|
|
|
/*
|
|
* Check the poll count
|
|
*/
|
|
if ( pTdIpx->AlivePoll ) {
|
|
|
|
TRACE(( pTd->pContext, TC_TD, TT_ERROR, "TDIPX: no response for %u pings\n",
|
|
pTdIpx->AlivePoll ));
|
|
|
|
/*
|
|
* Check timeout retry count
|
|
*/
|
|
if ( pTdIpx->AlivePoll > 6 ) {
|
|
|
|
TRACE(( pTd->pContext, TC_TD, TT_ERROR, "TDIPX: client declared dead\n" ));
|
|
|
|
/*
|
|
* Report broken connection
|
|
*/
|
|
Command.Header.Command = ICA_COMMAND_BROKEN_CONNECTION;
|
|
Command.BrokenConnection.Reason = Broken_Unexpected;
|
|
(void) IcaChannelInput( pTd->pContext,
|
|
Channel_Command,
|
|
0,
|
|
NULL,
|
|
(PCHAR) &Command,
|
|
sizeof(Command) );
|
|
|
|
/*
|
|
* Do not re-arm timer
|
|
*/
|
|
return;
|
|
}
|
|
|
|
/*
|
|
* Make timeout 10 seconds
|
|
*/
|
|
Timeout = 10 * 1000;
|
|
}
|
|
|
|
/*
|
|
* Check for valid file object
|
|
*/
|
|
if ( pTd->pFileObject == NULL )
|
|
goto badopen;
|
|
|
|
ObReferenceObject(pTd->pFileObject); //Fix 416142
|
|
|
|
/*
|
|
* Set up the default packet send type to be control
|
|
*/
|
|
ptype = IPX_TYPE_CONTROL;
|
|
Status = _DoTdiAction( pTd->pFileObject, MIPX_SETSENDPTYPE, &ptype, 1 );
|
|
ASSERT( Status == STATUS_SUCCESS );
|
|
if ( !NT_SUCCESS( Status ) )
|
|
goto badtype;
|
|
|
|
/*
|
|
* Write ping to network
|
|
*/
|
|
Buffer[0] = IPX_CTRL_PACKET_PING;
|
|
|
|
Status = _TdiSendDatagram(
|
|
pTd,
|
|
NULL, // Irp
|
|
pTd->pFileObject,
|
|
pTd->pDeviceObject,
|
|
pEndpoint->SendInfo.RemoteAddress,
|
|
pEndpoint->SendInfo.RemoteAddressLength,
|
|
Buffer,
|
|
1
|
|
);
|
|
ASSERT( Status == STATUS_SUCCESS || Status == STATUS_CTX_CLOSE_PENDING );
|
|
|
|
TRACEBUF(( pTd->pContext, TC_TD, TT_ORAW, Buffer, 1 ));
|
|
|
|
/*
|
|
* Reset address endpoint to send data packets
|
|
*/
|
|
ptype = IPX_TYPE_DATA;
|
|
Status = _DoTdiAction( pTd->pFileObject, MIPX_SETSENDPTYPE, &ptype, 1 );
|
|
ASSERT( Status == STATUS_SUCCESS );
|
|
|
|
/*
|
|
* Dereference the TDI file object and close the handle
|
|
*/
|
|
|
|
ObDereferenceObject(pTd->pFileObject); //Fix 416142
|
|
|
|
badtype:
|
|
|
|
/*
|
|
* Increment counter
|
|
*/
|
|
pTdIpx->AlivePoll++;
|
|
|
|
/*
|
|
* Increment number of byte written
|
|
*/
|
|
pTd->pStatus->Output.Bytes++;
|
|
|
|
/*
|
|
* Reset alive bit
|
|
*/
|
|
clientok:
|
|
pTdIpx->fClientAlive = FALSE;
|
|
|
|
/*
|
|
* Re-arm timer
|
|
*/
|
|
badopen:
|
|
IcaTimerStart( pTdIpx->pAliveTimer,
|
|
WatchDogTimer,
|
|
NULL,
|
|
Timeout,
|
|
ICALOCK_DRIVER );
|
|
}
|
|
|
|
|