ANSC_STATUS
BbhmDiagipOpen
(
ANSC_HANDLE hThisObject
)
{
ANSC_STATUS returnStatus = ANSC_STATUS_SUCCESS;
PBBHM_DIAG_IP_PING_OBJECT pMyObject = (PBBHM_DIAG_IP_PING_OBJECT )hThisObject;
PBBHM_IP_PING_PROPERTY pProperty = (PBBHM_IP_PING_PROPERTY )&pMyObject->Property;
PBBHM_IP_PING_TDO_OBJECT pStateTimer = (PBBHM_IP_PING_TDO_OBJECT )pMyObject->hStateTimer;
PBBHM_IP_PING_SINK_OBJECT pSink = NULL;
PANSC_XSOCKET_OBJECT pSocket = NULL;
ULONG i = 0;
if ( pMyObject->bActive )
{
return ANSC_STATUS_SUCCESS;
}
if ( pProperty->Status == BBHM_IP_PING_STATUS_RUNNING )
{
return ANSC_STATUS_FAILURE;
}
pSink = (PBBHM_IP_PING_SINK_OBJECT)BbhmDiagipSinkCreate((ANSC_HANDLE)pMyObject);
if ( !pSink )
{
return ANSC_STATUS_FAILURE;
}
pSocket =
(PANSC_XSOCKET_OBJECT)AnscCreateXsocket
(
pMyObject->hContainerContext,
(ANSC_HANDLE)pMyObject,
(ANSC_HANDLE)NULL
);
if ( !pSocket )
{
pSink->Remove((ANSC_HANDLE)pSink);
return ANSC_STATUS_FAILURE;
}
else
{
pSink->SetXsocket((ANSC_HANDLE)pSink, (ANSC_HANDLE)pSocket);
}
pMyObject->hSinkObject = pSink;
pMyObject->ResetPropertyCounter((ANSC_HANDLE)pMyObject);
pMyObject->bActive = TRUE;
return returnStatus;
}
ANSC_STATUS
BbhmDiagipStart
(
ANSC_HANDLE hThisObject
)
{
ANSC_STATUS returnStatus = ANSC_STATUS_SUCCESS;
PBBHM_DIAG_IP_PING_OBJECT pMyObject = (PBBHM_DIAG_IP_PING_OBJECT)hThisObject;
PDSLH_PING_INFO pDiagInfo = (PDSLH_PING_INFO)pMyObject->hDslhDiagInfo;
PBBHM_IP_PING_PROPERTY pProperty = (PBBHM_IP_PING_PROPERTY )&pMyObject->Property;
PBBHM_IP_PING_TDO_OBJECT pStateTimer = (PBBHM_IP_PING_TDO_OBJECT)pMyObject->hStateTimer;
PBBHM_IP_PING_SINK_OBJECT pSink = (PBBHM_IP_PING_SINK_OBJECT)pMyObject->hSinkObject;
PANSC_XSOCKET_OBJECT pSocket = (PANSC_XSOCKET_OBJECT )pSink->GetXsocket((ANSC_HANDLE)pSink);
ULONG numPkts = pProperty->NumPkts;
ULONG pktSize = pProperty->PktSize;
PCHAR pSendBuffer = pMyObject->hSendBuffer;
ULONG i = 0;
PICMPV4_ECHO_MESSAGE pIcmpHeader = NULL;
ULONG StartTime = 0;
UCHAR SrcIp[4] = {0, 0, 0, 0};
if ( !pMyObject->bActive )
{
pProperty->Status = BBHM_IP_PING_STATUS_ABORT;
pMyObject->Stop(hThisObject);
return ANSC_STATUS_FAILURE;
}
pMyObject->ResetPropertyCounter((ANSC_HANDLE)pMyObject);
pSocket->SetPeerName ((ANSC_HANDLE)pSocket, pProperty->pDstAddrName);
pSocket->SetHostName ((ANSC_HANDLE)pSocket, pProperty->pSrcAddrName);
/*
pSocket->SetHostAddress ((ANSC_HANDLE)pSocket, SrcIp );
pSocket->SetHostPort ((ANSC_HANDLE)pSocket, 0 );
*/
/*pSocket->SetTransportType((ANSC_HANDLE)pSocket, ICMP_TRANSPORT );*/
pSocket->SetType ((ANSC_HANDLE)pSocket, ANSC_XSOCKET_TYPE_RAW );
pSocket->SetMode ((ANSC_HANDLE)pSocket, 0 );
pSocket->SetXsink ((ANSC_HANDLE)pSocket, (ANSC_HANDLE)pSink );
/*
* The underlying socket wrapper may require an explicit startup() call, such is the case on
* Microsoft windows platforms. The wrapper initialization has to done for each task. On most
* real-time operating systems, this call is not required.
*/
AnscStartupXsocketWrapper((ANSC_HANDLE)pMyObject);
/* For IPv4/IPv6 compatible purpose we shall resolve the address first */
returnStatus = BbhmDiagResolvAddr(hThisObject);
if ( returnStatus != ANSC_STATUS_SUCCESS )
{
pProperty->Status = BBHM_IP_PING_STATUS_ERROR_HostName;
pMyObject->Stop(hThisObject);
return ANSC_STATUS_FAILURE;
}
pMyObject->IPProtocol = pSocket->GetIpProtocol((ANSC_HANDLE)pSocket);
if ( pMyObject->IPProtocol == XSKT_SOCKET_AF_INET )
{
pSocket->SetTransportType((ANSC_HANDLE)pSocket, IP4_PROTOCOL_ICMP);
}
else if ( pMyObject->IPProtocol == XSKT_SOCKET_AF_INET6 )
{
pSocket->SetTransportType((ANSC_HANDLE)pSocket, IP6_PROTOCOL_ICMP);
}
else
{
pProperty->Status = BBHM_IP_PING_STATUS_ABORT;
pMyObject->Stop(hThisObject);
return ANSC_STATUS_FAILURE;
}
/*
* We shall open the socket and listen on it right away. Since we're still in the context of
* initialiation, the wrapper module must be aware of the fact that the socket is opened before
* the first call returns.
*/
returnStatus = pSocket->Bind((ANSC_HANDLE)pSocket);
if ( returnStatus != ANSC_STATUS_SUCCESS )
{
pProperty->Status = BBHM_IP_PING_STATUS_ABORT;
pMyObject->Stop(hThisObject);
return ANSC_STATUS_FAILURE;
}
returnStatus = pSocket->Open((ANSC_HANDLE)pSocket); /* Create recv task */
if ( returnStatus != ANSC_STATUS_SUCCESS )
{
AnscTrace("Socket Open Failed!\n");
pProperty->Status = BBHM_IP_PING_STATUS_ABORT;
pMyObject->Stop(hThisObject);
return ANSC_STATUS_FAILURE;
}
if ( !pSendBuffer )
{
returnStatus = ANSC_STATUS_FAILURE;
pProperty->Status = BBHM_IP_PING_STATUS_ABORT;
pMyObject->Stop(hThisObject);
return ANSC_STATUS_FAILURE;
}
/* Set DSCP */
if ( pDiagInfo->DSCP != 0 )
{
pSocket->ApplyDSCP((ANSC_HANDLE)pSocket, pDiagInfo->DSCP);
}
AnscSleep(100);
if ( pMyObject->IPProtocol == XSKT_SOCKET_AF_INET )
{
pIcmpHeader = (PICMPV4_ECHO_MESSAGE)pMyObject->hSendBuffer;
AnscIcmpv4EchoSetType (pIcmpHeader, ICMP_TYPE_ECHO_REQUEST );
AnscIcmpv4EchoSetCode (pIcmpHeader, 0 );
AnscIcmpv4EchoSetId (pIcmpHeader, tempId );
AnscIcmpv4EchoSetSeqNumber (pIcmpHeader, (USHORT)pMyObject->GetPktsSent((ANSC_HANDLE)pMyObject));
for ( i = 0; i < pktSize; i++ )
{
((PUCHAR)pMyObject->hSendBuffer)[i + sizeof(ICMPV4_HEADER)] = (UCHAR)i;
}
AnscIcmpv4EchoSetChecksum (pIcmpHeader, 0 );
AnscIcmpv4CalculateChecksum (((PICMPV4_HEADER)pIcmpHeader), pktSize + sizeof(ICMPV4_HEADER));
}
else if ( pMyObject->IPProtocol == XSKT_SOCKET_AF_INET6 )
{
pIcmpHeader = (PICMPV6_ECHO_MESSAGE)pMyObject->hSendBuffer;
AnscIcmpv6EchoSetType (pIcmpHeader, ICMP6_TYPE_ECHO_REQUEST );
AnscIcmpv6EchoSetCode (pIcmpHeader, 0 );
AnscIcmpv6EchoSetId (pIcmpHeader, tempId );
AnscIcmpv6EchoSetSeqNumber (pIcmpHeader, (USHORT)pMyObject->GetPktsSent((ANSC_HANDLE)pMyObject));
for ( i = 0; i < pktSize; i++ )
{
((PUCHAR)pMyObject->hSendBuffer)[i + sizeof(ICMPV6_HEADER)] = (UCHAR)i;
}
}
pStateTimer->SetTimerType((ANSC_HANDLE)pStateTimer, ANSC_TIMER_TYPE_SPORADIC);
pStateTimer->SetInterval ((ANSC_HANDLE)pStateTimer, pProperty->TimeBetween );
pStateTimer->SetCounter ((ANSC_HANDLE)pStateTimer, pProperty->NumPkts );
/*
pSocket->SetPeerAddress ((ANSC_HANDLE)pSocket , pProperty->DstIp.Dot );
pSocket->SetPeerPort ((ANSC_HANDLE)pSocket , 0 );
*/
pMyObject->SetStatus((ANSC_HANDLE)pMyObject, BBHM_IP_PING_STATUS_RUNNING);
pStateTimer->Start ((ANSC_HANDLE)pStateTimer);
StartTime = AnscGetTickInMilliSeconds();
if ( pMyObject->IPProtocol == XSKT_SOCKET_AF_INET )
{
returnStatus =
pMyObject->AddEchoEntry
(
(ANSC_HANDLE)pMyObject,
AnscIcmpv4EchoGetSeqNumber(pIcmpHeader),
StartTime
);
returnStatus =
pMyObject->Send
(
(ANSC_HANDLE)pMyObject,
(ANSC_HANDLE)pMyObject->hSinkObject,
(PVOID)pMyObject->hSendBuffer,
pktSize + sizeof(ICMPV4_HEADER)
);
}
else if ( pMyObject->IPProtocol == XSKT_SOCKET_AF_INET6 )
{
returnStatus =
pMyObject->AddEchoEntry
(
(ANSC_HANDLE)pMyObject,
AnscIcmpv6EchoGetSeqNumber(pIcmpHeader),
StartTime
);
returnStatus =
pMyObject->Send
(
(ANSC_HANDLE)pMyObject,
(ANSC_HANDLE)pMyObject->hSinkObject,
(PVOID)pMyObject->hSendBuffer,
pktSize + sizeof(ICMPV6_HEADER)
);
}
pProperty->PktsSent++;
return returnStatus;
}