From a9bcc81f821d7c66f623779fa5147e728eb3c388 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sat, 4 May 2024 03:24:41 +0200 Subject: Adding upstream version 3.3.0+dfsg1. Signed-off-by: Daniel Baumann --- libfreerdp/core/heartbeat.c | 98 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 98 insertions(+) create mode 100644 libfreerdp/core/heartbeat.c (limited to 'libfreerdp/core/heartbeat.c') diff --git a/libfreerdp/core/heartbeat.c b/libfreerdp/core/heartbeat.c new file mode 100644 index 0000000..1fb6089 --- /dev/null +++ b/libfreerdp/core/heartbeat.c @@ -0,0 +1,98 @@ +/** + * FreeRDP: A Remote Desktop Protocol Implementation + * Heartbeat PDUs + * + * Copyright 2014 Dell Software + * + * 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. + */ + +#include + +#define WITH_DEBUG_HEARTBEAT + +#include "heartbeat.h" + +state_run_t rdp_recv_heartbeat_packet(rdpRdp* rdp, wStream* s) +{ + BYTE reserved = 0; + BYTE period = 0; + BYTE count1 = 0; + BYTE count2 = 0; + BOOL rc = 0; + + WINPR_ASSERT(rdp); + WINPR_ASSERT(rdp->context); + WINPR_ASSERT(s); + + if (!Stream_CheckAndLogRequiredLength(AUTODETECT_TAG, s, 4)) + return STATE_RUN_FAILED; + + 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->context->instance, period, count1, + count2); + if (!rc) + { + WLog_ERR(HEARTBEAT_TAG, "heartbeat->ServerHeartbeat callback failed!"); + return STATE_RUN_FAILED; + } + + return STATE_RUN_SUCCESS; +} + +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); +} -- cgit v1.2.3