/**

 * 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.

 */

#include <freerdp/config.h>

#include "heartbeat.h"

state_run_t rdp_recv_heartbeat_packet(rdpRdp* rdp, wStream* s)

{

  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_Seek_UINT8(s);           /* 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_VRB(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;

  UINT16 sec_flags = 0;

  wStream* s = rdp_message_channel_pdu_init(rdp, &sec_flags);

  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_flags | 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);

}