96 lines
2.4 KiB
C
96 lines
2.4 KiB
C
|
/**
|
||
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
||
|
|
* Heartbeat PDUs
|
||
|
|
*
|
||
|
|
* Copyright 2014 Dell Software <Mike.McDonald@software.dell.com>
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifdef HAVE_CONFIG_H
|
||
|
|
#include "config.h"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#define WITH_DEBUG_HEARTBEAT
|
||
|
|
|
||
|
|
#include "heartbeat.h"
|
||
|
|
|
||
|
|
int rdp_recv_heartbeat_packet(rdpRdp* rdp, wStream* s)
|
||
|
|
{
|
||
|
|
BYTE reserved;
|
||
|
|
BYTE period;
|
||
|
|
BYTE count1;
|
||
|
|
BYTE count2;
|
||
|
|
BOOL rc;
|
||
|
|
|
||
|
|
if (Stream_GetRemainingLength(s) < 4)
|
||
|
|
return -1;
|
||
|
|
|
||
|
|
Stream_Read_UINT8(s, reserved); /* reserved (1 byte) */
|
||
|
|
Stream_Read_UINT8(s, period); /* period (1 byte) */
|
||
|
|
Stream_Read_UINT8(s, count1); /* count1 (1 byte) */
|
||
|
|
Stream_Read_UINT8(s, count2); /* count2 (1 byte) */
|
||
|
|
|
||
|
|
WLog_DBG(HEARTBEAT_TAG,
|
||
|
|
"received Heartbeat PDU -> period=%" PRIu8 ", count1=%" PRIu8 ", count2=%" PRIu8 "",
|
||
|
|
period, count1, count2);
|
||
|
|
|
||
|
|
rc = IFCALLRESULT(TRUE, rdp->heartbeat->ServerHeartbeat, rdp->instance, period, count1, count2);
|
||
|
|
if (!rc)
|
||
|
|
{
|
||
|
|
WLog_ERR(HEARTBEAT_TAG, "heartbeat->ServerHeartbeat callback failed!");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL freerdp_heartbeat_send_heartbeat_pdu(freerdp_peer* peer, BYTE period, BYTE count1, BYTE count2)
|
||
|
|
{
|
||
|
|
rdpRdp* rdp = peer->context->rdp;
|
||
|
|
wStream* s = rdp_message_channel_pdu_init(rdp);
|
||
|
|
|
||
|
|
if (!s)
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
Stream_Seek_UINT8(s); /* reserved (1 byte) */
|
||
|
|
Stream_Write_UINT8(s, period); /* period (1 byte) */
|
||
|
|
Stream_Write_UINT8(s, count1); /* count1 (1 byte) */
|
||
|
|
Stream_Write_UINT8(s, count2); /* count2 (1 byte) */
|
||
|
|
|
||
|
|
WLog_DBG(HEARTBEAT_TAG,
|
||
|
|
"sending Heartbeat PDU -> period=%" PRIu8 ", count1=%" PRIu8 ", count2=%" PRIu8 "",
|
||
|
|
period, count1, count2);
|
||
|
|
|
||
|
|
if (!rdp_send_message_channel_pdu(rdp, s, SEC_HEARTBEAT))
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
rdpHeartbeat* heartbeat_new(void)
|
||
|
|
{
|
||
|
|
rdpHeartbeat* heartbeat = (rdpHeartbeat*)calloc(1, sizeof(rdpHeartbeat));
|
||
|
|
|
||
|
|
if (heartbeat)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
return heartbeat;
|
||
|
|
}
|
||
|
|
|
||
|
|
void heartbeat_free(rdpHeartbeat* heartbeat)
|
||
|
|
{
|
||
|
|
free(heartbeat);
|
||
|
|
}
|