/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * RDPEDISP Virtual Channel Extension

 *

 * Copyright 2019 Kobi Mizrachi <kmizrachi18@gmail.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 <winpr/crt.h>

#include <winpr/stream.h>

#include <freerdp/channels/log.h>

#define TAG CHANNELS_TAG("disp.common")

#include "disp_common.h"

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

UINT disp_read_header(wStream* s, DISPLAY_CONTROL_HEADER* header)

{

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 8))

    return ERROR_INVALID_DATA;

  Stream_Read_UINT32(s, header->type);

  Stream_Read_UINT32(s, header->length);

  if (header->length < 8)

  {

    WLog_ERR(TAG, "Invalid header length %" PRIu32 ", require at least 8", header->length);

    return ERROR_INVALID_DATA;

  }

  return CHANNEL_RC_OK;

}

/**

 * Function description

 *

 * @return 0 on success, otherwise a Win32 error code

 */

UINT disp_write_header(wStream* s, const DISPLAY_CONTROL_HEADER* header)

{

  Stream_Write_UINT32(s, header->type);

  Stream_Write_UINT32(s, header->length);

  return CHANNEL_RC_OK;

}