/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * Cliprdr common

 *

 * Copyright 2013 Marc-Andre Moreau <marcandre.moreau@gmail.com>

 * Copyright 2015 Thincast Technologies GmbH

 * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>

 * 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 <winpr/crt.h>

#include <winpr/stream.h>

#include <freerdp/channels/log.h>

#define TAG CHANNELS_TAG("cliprdr.common")

#include "cliprdr_common.h"

static BOOL cliprdr_validate_file_contents_request(const CLIPRDR_FILE_CONTENTS_REQUEST* request)

{

  /*

   * [MS-RDPECLIP] 2.2.5.3 File Contents Request PDU (CLIPRDR_FILECONTENTS_REQUEST).

   *

   * A request for the size of the file identified by the lindex field. The size MUST be

   * returned as a 64-bit, unsigned integer. The cbRequested field MUST be set to

   * 0x00000008 and both the nPositionLow and nPositionHigh fields MUST be

   * set to 0x00000000.

   */

  if (request->dwFlags & FILECONTENTS_SIZE)

  {

    if (request->cbRequested != sizeof(UINT64))

    {

      WLog_ERR(TAG, "cbRequested must be %" PRIu32 ", got %" PRIu32 "", sizeof(UINT64),

               request->cbRequested);

      return FALSE;

    }

    if (request->nPositionHigh != 0 || request->nPositionLow != 0)

    {

      WLog_ERR(TAG, "nPositionHigh and nPositionLow must be set to 0");

      return FALSE;

    }

  }

  return TRUE;

}

wStream* cliprdr_packet_new(UINT16 msgType, UINT16 msgFlags, UINT32 dataLen)

{

  wStream* s = NULL;

  s = Stream_New(NULL, dataLen + 8);

  if (!s)

  {

    WLog_ERR(TAG, "Stream_New failed!");

    return NULL;

  }

  Stream_Write_UINT16(s, msgType);

  Stream_Write_UINT16(s, msgFlags);

  /* Write actual length after the entire packet has been constructed. */

  Stream_Write_UINT32(s, 0);

  return s;

}

static void cliprdr_write_file_contents_request(wStream* s,

                                                const CLIPRDR_FILE_CONTENTS_REQUEST* request)

{

  Stream_Write_UINT32(s, request->streamId);      /* streamId (4 bytes) */

  Stream_Write_UINT32(s, request->listIndex);     /* listIndex (4 bytes) */

  Stream_Write_UINT32(s, request->dwFlags);       /* dwFlags (4 bytes) */

  Stream_Write_UINT32(s, request->nPositionLow);  /* nPositionLow (4 bytes) */

  Stream_Write_UINT32(s, request->nPositionHigh); /* nPositionHigh (4 bytes) */

  Stream_Write_UINT32(s, request->cbRequested);   /* cbRequested (4 bytes) */

  if (request->haveClipDataId)

    Stream_Write_UINT32(s, request->clipDataId); /* clipDataId (4 bytes) */

}

static INLINE void cliprdr_write_lock_unlock_clipdata(wStream* s, UINT32 clipDataId)

{

  Stream_Write_UINT32(s, clipDataId);

}

static void cliprdr_write_lock_clipdata(wStream* s,

                                        const CLIPRDR_LOCK_CLIPBOARD_DATA* lockClipboardData)

{

  cliprdr_write_lock_unlock_clipdata(s, lockClipboardData->clipDataId);

}

static void cliprdr_write_unlock_clipdata(wStream* s,

                                          const CLIPRDR_UNLOCK_CLIPBOARD_DATA* unlockClipboardData)

{

  cliprdr_write_lock_unlock_clipdata(s, unlockClipboardData->clipDataId);

}

static void cliprdr_write_file_contents_response(wStream* s,

                                                 const CLIPRDR_FILE_CONTENTS_RESPONSE* response)

{

  Stream_Write_UINT32(s, response->streamId); /* streamId (4 bytes) */

  Stream_Write(s, response->requestedData, response->cbRequested);

}

wStream* cliprdr_packet_lock_clipdata_new(const CLIPRDR_LOCK_CLIPBOARD_DATA* lockClipboardData)

{

  wStream* s = NULL;

  if (!lockClipboardData)

    return NULL;

  s = cliprdr_packet_new(CB_LOCK_CLIPDATA, 0, 4);

  if (!s)

    return NULL;

  cliprdr_write_lock_clipdata(s, lockClipboardData);

  return s;

}

wStream*

cliprdr_packet_unlock_clipdata_new(const CLIPRDR_UNLOCK_CLIPBOARD_DATA* unlockClipboardData)

{

  wStream* s = NULL;

  if (!unlockClipboardData)

    return NULL;

  s = cliprdr_packet_new(CB_UNLOCK_CLIPDATA, 0, 4);

  if (!s)

    return NULL;

  cliprdr_write_unlock_clipdata(s, unlockClipboardData);

  return s;

}

wStream* cliprdr_packet_file_contents_request_new(const CLIPRDR_FILE_CONTENTS_REQUEST* request)

{

  wStream* s = NULL;

  if (!request)

    return NULL;

  s = cliprdr_packet_new(CB_FILECONTENTS_REQUEST, 0, 28);

  if (!s)

    return NULL;

  cliprdr_write_file_contents_request(s, request);

  return s;

}

wStream* cliprdr_packet_file_contents_response_new(const CLIPRDR_FILE_CONTENTS_RESPONSE* response)

{

  wStream* s = NULL;

  if (!response)

    return NULL;

  s = cliprdr_packet_new(CB_FILECONTENTS_RESPONSE, response->common.msgFlags,

                         4 + response->cbRequested);

  if (!s)

    return NULL;

  cliprdr_write_file_contents_response(s, response);

  return s;

}

wStream* cliprdr_packet_format_list_new(const CLIPRDR_FORMAT_LIST* formatList,

                                        BOOL useLongFormatNames)

{

  wStream* s = NULL;

  size_t formatNameSize = 0;

  char* szFormatName = NULL;

  WCHAR* wszFormatName = NULL;

  BOOL asciiNames = FALSE;

  CLIPRDR_FORMAT* format = NULL;

  UINT32 length = 0;

  if (formatList->common.msgType != CB_FORMAT_LIST)

    WLog_WARN(TAG, "called with invalid type %08" PRIx32, formatList->common.msgType);

  if (!useLongFormatNames)

  {

    length = formatList->numFormats * 36;

    s = cliprdr_packet_new(CB_FORMAT_LIST, 0, length);

    if (!s)

    {

      WLog_ERR(TAG, "cliprdr_packet_new failed!");

      return NULL;

    }

    for (UINT32 index = 0; index < formatList->numFormats; index++)

    {

      size_t formatNameLength = 0;

      format = (CLIPRDR_FORMAT*)&(formatList->formats[index]);

      Stream_Write_UINT32(s, format->formatId); /* formatId (4 bytes) */

      formatNameSize = 0;

      szFormatName = format->formatName;

      if (asciiNames)

      {

        if (szFormatName)

          formatNameLength = strnlen(szFormatName, 32);

        if (formatNameLength > 31)

          formatNameLength = 31;

        Stream_Write(s, szFormatName, formatNameLength);

        Stream_Zero(s, 32 - formatNameLength);

      }

      else

      {

        wszFormatName = NULL;

        if (szFormatName)

        {

          wszFormatName = ConvertUtf8ToWCharAlloc(szFormatName, &formatNameSize);

          if (!wszFormatName)

          {

            Stream_Free(s, TRUE);

            return NULL;

          }

          formatNameSize += 1; /* append terminating '\0' */

        }

        if (formatNameSize > 15)

          formatNameSize = 15;

        /* size in bytes  instead of wchar */

        formatNameSize *= sizeof(WCHAR);

        if (wszFormatName)

          Stream_Write(s, wszFormatName, (size_t)formatNameSize);

        Stream_Zero(s, (size_t)(32 - formatNameSize));

        free(wszFormatName);

      }

    }

  }

  else

  {

    length = 0;

    for (UINT32 index = 0; index < formatList->numFormats; index++)

    {

      format = (CLIPRDR_FORMAT*)&(formatList->formats[index]);

      length += 4;

      formatNameSize = sizeof(WCHAR);

      if (format->formatName)

      {

        SSIZE_T size = ConvertUtf8ToWChar(format->formatName, NULL, 0);

        if (size < 0)

          return NULL;

        formatNameSize = (size_t)(size + 1) * sizeof(WCHAR);

      }

      length += (UINT32)formatNameSize;

    }

    s = cliprdr_packet_new(CB_FORMAT_LIST, 0, length);

    if (!s)

    {

      WLog_ERR(TAG, "cliprdr_packet_new failed!");

      return NULL;

    }

    for (UINT32 index = 0; index < formatList->numFormats; index++)

    {

      format = (CLIPRDR_FORMAT*)&(formatList->formats[index]);

      Stream_Write_UINT32(s, format->formatId); /* formatId (4 bytes) */

      if (format->formatName)

      {

        const size_t cap = Stream_Capacity(s);

        const size_t pos = Stream_GetPosition(s);

        const size_t rem = cap - pos;

        if ((cap < pos) || ((rem / 2) > INT_MAX))

        {

          Stream_Free(s, TRUE);

          return NULL;

        }

        const size_t len = strnlen(format->formatName, rem / sizeof(WCHAR));

        if (Stream_Write_UTF16_String_From_UTF8(s, len + 1, format->formatName, len, TRUE) <

            0)

        {

          Stream_Free(s, TRUE);

          return NULL;

        }

      }

      else

      {

        Stream_Write_UINT16(s, 0);

      }

    }

  }

  return s;

}

UINT cliprdr_read_unlock_clipdata(wStream* s, CLIPRDR_UNLOCK_CLIPBOARD_DATA* unlockClipboardData)

{

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

    return ERROR_INVALID_DATA;

  Stream_Read_UINT32(s, unlockClipboardData->clipDataId); /* clipDataId (4 bytes) */

  return CHANNEL_RC_OK;

}

UINT cliprdr_read_format_data_request(wStream* s, CLIPRDR_FORMAT_DATA_REQUEST* request)

{

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

    return ERROR_INVALID_DATA;

  Stream_Read_UINT32(s, request->requestedFormatId); /* requestedFormatId (4 bytes) */

  return CHANNEL_RC_OK;

}

UINT cliprdr_read_format_data_response(wStream* s, CLIPRDR_FORMAT_DATA_RESPONSE* response)

{

  response->requestedFormatData = NULL;

  if (!Stream_CheckAndLogRequiredLength(TAG, s, response->common.dataLen))

    return ERROR_INVALID_DATA;

  if (response->common.dataLen)

    response->requestedFormatData = Stream_ConstPointer(s);

  return CHANNEL_RC_OK;

}

UINT cliprdr_read_file_contents_request(wStream* s, CLIPRDR_FILE_CONTENTS_REQUEST* request)

{

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

    return ERROR_INVALID_DATA;

  request->haveClipDataId = FALSE;

  Stream_Read_UINT32(s, request->streamId);      /* streamId (4 bytes) */

  Stream_Read_UINT32(s, request->listIndex);     /* listIndex (4 bytes) */

  Stream_Read_UINT32(s, request->dwFlags);       /* dwFlags (4 bytes) */

  Stream_Read_UINT32(s, request->nPositionLow);  /* nPositionLow (4 bytes) */

  Stream_Read_UINT32(s, request->nPositionHigh); /* nPositionHigh (4 bytes) */

  Stream_Read_UINT32(s, request->cbRequested);   /* cbRequested (4 bytes) */

  if (Stream_GetRemainingLength(s) >= 4)

  {

    Stream_Read_UINT32(s, request->clipDataId); /* clipDataId (4 bytes) */

    request->haveClipDataId = TRUE;

  }

  if (!cliprdr_validate_file_contents_request(request))

    return ERROR_BAD_ARGUMENTS;

  return CHANNEL_RC_OK;

}

UINT cliprdr_read_file_contents_response(wStream* s, CLIPRDR_FILE_CONTENTS_RESPONSE* response)

{

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

    return ERROR_INVALID_DATA;

  Stream_Read_UINT32(s, response->streamId);   /* streamId (4 bytes) */

  response->requestedData = Stream_ConstPointer(s); /* requestedFileContentsData */

  WINPR_ASSERT(response->common.dataLen >= 4);

  response->cbRequested = response->common.dataLen - 4;

  return CHANNEL_RC_OK;

}

UINT cliprdr_read_format_list(wStream* s, CLIPRDR_FORMAT_LIST* formatList, BOOL useLongFormatNames)

{

  UINT32 index = 0;

  size_t formatNameLength = 0;

  const char* szFormatName = NULL;

  const WCHAR* wszFormatName = NULL;

  wStream sub1buffer = { 0 };

  CLIPRDR_FORMAT* formats = NULL;

  UINT error = ERROR_INTERNAL_ERROR;

  const BOOL asciiNames = (formatList->common.msgFlags & CB_ASCII_NAMES) ? TRUE : FALSE;

  index = 0;

  /* empty format list */

  formatList->formats = NULL;

  formatList->numFormats = 0;

  wStream* sub1 =

      Stream_StaticConstInit(&sub1buffer, Stream_ConstPointer(s), formatList->common.dataLen);

  if (!Stream_SafeSeek(s, formatList->common.dataLen))

    return ERROR_INVALID_DATA;

  if (!formatList->common.dataLen)

  {

  }

  else if (!useLongFormatNames)

  {

    const size_t cap = Stream_Capacity(sub1);

    formatList->numFormats = (cap / 36);

    if ((formatList->numFormats * 36) != cap)

    {

      WLog_ERR(TAG, "Invalid short format list length: %" PRIuz "", cap);

      return ERROR_INTERNAL_ERROR;

    }

    if (formatList->numFormats)

      formats = (CLIPRDR_FORMAT*)calloc(formatList->numFormats, sizeof(CLIPRDR_FORMAT));

    if (!formats)

    {

      WLog_ERR(TAG, "calloc failed!");

      return CHANNEL_RC_NO_MEMORY;

    }

    formatList->formats = formats;

    while (Stream_GetRemainingLength(sub1) >= 4)

    {

      CLIPRDR_FORMAT* format = &formats[index];

      Stream_Read_UINT32(sub1, format->formatId); /* formatId (4 bytes) */

      /* According to MS-RDPECLIP 2.2.3.1.1.1 formatName is "a 32-byte block containing

       * the *null-terminated* name assigned to the Clipboard Format: (32 ASCII 8 characters

       * or 16 Unicode characters)"

       * However, both Windows RDSH and mstsc violate this specs as seen in the following

       * example of a transferred short format name string: [R.i.c.h. .T.e.x.t. .F.o.r.m.a.t.]

       * These are 16 unicode charaters - *without* terminating null !

       */

      szFormatName = Stream_ConstPointer(sub1);

      wszFormatName = Stream_ConstPointer(sub1);

      if (!Stream_SafeSeek(sub1, 32))

        goto error_out;

      free(format->formatName);

      format->formatName = NULL;

      if (asciiNames)

      {

        if (szFormatName[0])

        {

          /* ensure null termination */

          format->formatName = strndup(szFormatName, 31);

          if (!format->formatName)

          {

            WLog_ERR(TAG, "malloc failed!");

            error = CHANNEL_RC_NO_MEMORY;

            goto error_out;

          }

        }

      }

      else

      {

        if (wszFormatName[0])

        {

          format->formatName = ConvertWCharNToUtf8Alloc(wszFormatName, 16, NULL);

          if (!format->formatName)

            goto error_out;

        }

      }

      index++;

    }

  }

  else

  {

    wStream sub2buffer = sub1buffer;

    wStream* sub2 = &sub2buffer;

    while (Stream_GetRemainingLength(sub1) > 0)

    {

      size_t rest = 0;

      if (!Stream_SafeSeek(sub1, 4)) /* formatId (4 bytes) */

        goto error_out;

      wszFormatName = Stream_ConstPointer(sub1);

      rest = Stream_GetRemainingLength(sub1);

      formatNameLength = _wcsnlen(wszFormatName, rest / sizeof(WCHAR));

      if (!Stream_SafeSeek(sub1, (formatNameLength + 1) * sizeof(WCHAR)))

        goto error_out;

      formatList->numFormats++;

    }

    if (formatList->numFormats)

      formats = (CLIPRDR_FORMAT*)calloc(formatList->numFormats, sizeof(CLIPRDR_FORMAT));

    if (!formats)

    {

      WLog_ERR(TAG, "calloc failed!");

      return CHANNEL_RC_NO_MEMORY;

    }

    formatList->formats = formats;

    while (Stream_GetRemainingLength(sub2) >= 4)

    {

      size_t rest = 0;

      CLIPRDR_FORMAT* format = &formats[index];

      Stream_Read_UINT32(sub2, format->formatId); /* formatId (4 bytes) */

      free(format->formatName);

      format->formatName = NULL;

      wszFormatName = Stream_ConstPointer(sub2);

      rest = Stream_GetRemainingLength(sub2);

      formatNameLength = _wcsnlen(wszFormatName, rest / sizeof(WCHAR));

      if (!Stream_SafeSeek(sub2, (formatNameLength + 1) * sizeof(WCHAR)))

        goto error_out;

      if (formatNameLength)

      {

        format->formatName =

            ConvertWCharNToUtf8Alloc(wszFormatName, formatNameLength, NULL);

        if (!format->formatName)

          goto error_out;

      }

      index++;

    }

  }

  return CHANNEL_RC_OK;

error_out:

  cliprdr_free_format_list(formatList);

  return error;

}

void cliprdr_free_format_list(CLIPRDR_FORMAT_LIST* formatList)

{

  if (formatList == NULL)

    return;

  if (formatList->formats)

  {

    for (UINT32 index = 0; index < formatList->numFormats; index++)

    {

      free(formatList->formats[index].formatName);

    }

    free(formatList->formats);

    formatList->formats = NULL;

    formatList->numFormats = 0;

  }

}