/src/FreeRDP/channels/remdesk/client/remdesk_main.c

Source (jump to first uncovered line)
/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * Remote Assistance Virtual Channel
 *
 * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 * Copyright 2015 Thincast Technologies GmbH
 * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.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/assert.h>
#include <winpr/print.h>

#include <freerdp/freerdp.h>
#include <freerdp/assistance.h>

#include <freerdp/channels/log.h>
#include <freerdp/client/remdesk.h>

#include "remdesk_main.h"

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_virtual_channel_write(remdeskPlugin* remdesk, wStream* s)
{
  UINT32 status = 0;

  if (!remdesk)
  {
    WLog_ERR(TAG, "remdesk was null!");
    Stream_Free(s, TRUE);
    return CHANNEL_RC_INVALID_INSTANCE;
  }

  WINPR_ASSERT(remdesk->channelEntryPoints.pVirtualChannelWriteEx);
  status = remdesk->channelEntryPoints.pVirtualChannelWriteEx(
      remdesk->InitHandle, remdesk->OpenHandle, Stream_Buffer(s), (UINT32)Stream_Length(s), s);

  if (status != CHANNEL_RC_OK)
  {
    Stream_Free(s, TRUE);
    WLog_ERR(TAG, "pVirtualChannelWriteEx failed with %s [%08" PRIX32 "]",
             WTSErrorToString(status), status);
  }
  return status;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_generate_expert_blob(remdeskPlugin* remdesk)
{
  const char* name = NULL;
  char* pass = NULL;
  const char* password = NULL;
  rdpSettings* settings = NULL;

  WINPR_ASSERT(remdesk);

  WINPR_ASSERT(remdesk->rdpcontext);
  settings = remdesk->rdpcontext->settings;
  WINPR_ASSERT(settings);

  if (remdesk->ExpertBlob)
    return CHANNEL_RC_OK;

  password = freerdp_settings_get_string(settings, FreeRDP_RemoteAssistancePassword);
  if (!password)
    password = freerdp_settings_get_string(settings, FreeRDP_Password);

  if (!password)
  {
    WLog_ERR(TAG, "password was not set!");
    return ERROR_INTERNAL_ERROR;
  }

  name = freerdp_settings_get_string(settings, FreeRDP_Username);

  if (!name)
    name = "Expert";

  const char* stub = freerdp_settings_get_string(settings, FreeRDP_RemoteAssistancePassStub);
  remdesk->EncryptedPassStub =
      freerdp_assistance_encrypt_pass_stub(password, stub, &(remdesk->EncryptedPassStubSize));

  if (!remdesk->EncryptedPassStub)
  {
    WLog_ERR(TAG, "freerdp_assistance_encrypt_pass_stub failed!");
    return ERROR_INTERNAL_ERROR;
  }

  pass = freerdp_assistance_bin_to_hex_string(remdesk->EncryptedPassStub,
                                              remdesk->EncryptedPassStubSize);

  if (!pass)
  {
    WLog_ERR(TAG, "freerdp_assistance_bin_to_hex_string failed!");
    return ERROR_INTERNAL_ERROR;
  }

  remdesk->ExpertBlob = freerdp_assistance_construct_expert_blob(name, pass);
  free(pass);

  if (!remdesk->ExpertBlob)
  {
    WLog_ERR(TAG, "freerdp_assistance_construct_expert_blob failed!");
    return ERROR_INTERNAL_ERROR;
  }

  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_read_channel_header(wStream* s, REMDESK_CHANNEL_HEADER* header)
{
  UINT32 ChannelNameLen = 0;

  WINPR_ASSERT(s);
  WINPR_ASSERT(header);

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 8))
    return ERROR_INVALID_DATA;

  Stream_Read_UINT32(s, ChannelNameLen);     /* ChannelNameLen (4 bytes) */
  Stream_Read_UINT32(s, header->DataLength); /* DataLen (4 bytes) */

  if (ChannelNameLen > 64)
  {
    WLog_ERR(TAG, "ChannelNameLen > 64!");
    return ERROR_INVALID_DATA;
  }

  if ((ChannelNameLen % 2) != 0)
  {
    WLog_ERR(TAG, "ChannelNameLen %% 2) != 0 ");
    return ERROR_INVALID_DATA;
  }

  if (Stream_Read_UTF16_String_As_UTF8_Buffer(s, ChannelNameLen / sizeof(WCHAR),
                                              header->ChannelName,
                                              ARRAYSIZE(header->ChannelName)) < 0)
    return ERROR_INTERNAL_ERROR;

  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_write_channel_header(wStream* s, REMDESK_CHANNEL_HEADER* header)
{
  UINT32 ChannelNameLen = 0;
  WCHAR ChannelNameW[32] = { 0 };

  WINPR_ASSERT(s);
  WINPR_ASSERT(header);

  for (size_t index = 0; index < 32; index++)
  {
    ChannelNameW[index] = (WCHAR)header->ChannelName[index];
  }

  ChannelNameLen = (strnlen(header->ChannelName, sizeof(header->ChannelName)) + 1) * 2;
  Stream_Write_UINT32(s, ChannelNameLen);        /* ChannelNameLen (4 bytes) */
  Stream_Write_UINT32(s, header->DataLength);    /* DataLen (4 bytes) */
  Stream_Write(s, ChannelNameW, ChannelNameLen); /* ChannelName (variable) */
  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_write_ctl_header(wStream* s, REMDESK_CTL_HEADER* ctlHeader)
{
  WINPR_ASSERT(s);
  WINPR_ASSERT(ctlHeader);

  remdesk_write_channel_header(s, &ctlHeader->ch);
  Stream_Write_UINT32(s, ctlHeader->msgType); /* msgType (4 bytes) */
  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_prepare_ctl_header(REMDESK_CTL_HEADER* ctlHeader, UINT32 msgType,
                                       UINT32 msgSize)
{
  WINPR_ASSERT(ctlHeader);

  ctlHeader->msgType = msgType;
  (void)sprintf_s(ctlHeader->ch.ChannelName, ARRAYSIZE(ctlHeader->ch.ChannelName),
                  REMDESK_CHANNEL_CTL_NAME);
  ctlHeader->ch.DataLength = 4 + msgSize;
  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_recv_ctl_server_announce_pdu(remdeskPlugin* remdesk, wStream* s,
                                                 REMDESK_CHANNEL_HEADER* header)
{
  WINPR_ASSERT(remdesk);
  WINPR_ASSERT(s);
  WINPR_ASSERT(header);

  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_recv_ctl_version_info_pdu(remdeskPlugin* remdesk, wStream* s,
                                              REMDESK_CHANNEL_HEADER* header)
{
  UINT32 versionMajor = 0;
  UINT32 versionMinor = 0;

  WINPR_ASSERT(remdesk);
  WINPR_ASSERT(s);
  WINPR_ASSERT(header);

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 8))
    return ERROR_INVALID_DATA;

  Stream_Read_UINT32(s, versionMajor); /* versionMajor (4 bytes) */
  Stream_Read_UINT32(s, versionMinor); /* versionMinor (4 bytes) */

  if ((versionMajor != 1) || (versionMinor > 2) || (versionMinor == 0))
  {
    WLog_ERR(TAG, "Unsupported protocol version %" PRId32 ".%" PRId32, versionMajor,
             versionMinor);
  }

  remdesk->Version = versionMinor;
  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_send_ctl_version_info_pdu(remdeskPlugin* remdesk)
{
  wStream* s = NULL;
  REMDESK_CTL_VERSION_INFO_PDU pdu;
  UINT error = 0;

  WINPR_ASSERT(remdesk);

  remdesk_prepare_ctl_header(&(pdu.ctlHeader), REMDESK_CTL_VERSIONINFO, 8);
  pdu.versionMajor = 1;
  pdu.versionMinor = 2;
  s = Stream_New(NULL, REMDESK_CHANNEL_CTL_SIZE + pdu.ctlHeader.ch.DataLength);

  if (!s)
  {
    WLog_ERR(TAG, "Stream_New failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  remdesk_write_ctl_header(s, &(pdu.ctlHeader));
  Stream_Write_UINT32(s, pdu.versionMajor); /* versionMajor (4 bytes) */
  Stream_Write_UINT32(s, pdu.versionMinor); /* versionMinor (4 bytes) */
  Stream_SealLength(s);

  if ((error = remdesk_virtual_channel_write(remdesk, s)))
    WLog_ERR(TAG, "remdesk_virtual_channel_write failed with error %" PRIu32 "!", error);

  return error;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_recv_ctl_result_pdu(remdeskPlugin* remdesk, wStream* s,
                                        REMDESK_CHANNEL_HEADER* header, UINT32* pResult)
{
  UINT32 result = 0;

  WINPR_ASSERT(remdesk);
  WINPR_ASSERT(s);
  WINPR_ASSERT(header);
  WINPR_ASSERT(pResult);

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 4))
    return ERROR_INVALID_DATA;

  Stream_Read_UINT32(s, result); /* result (4 bytes) */
  *pResult = result;
  // WLog_DBG(TAG, "RemdeskRecvResult: 0x%08"PRIX32"", result);
  switch (result)
  {
    case REMDESK_ERROR_HELPEESAIDNO:
      WLog_DBG(TAG, "remote assistance connection request was denied");
      return ERROR_CONNECTION_REFUSED;

    default:
      break;
  }

  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_send_ctl_authenticate_pdu(remdeskPlugin* remdesk)
{
  UINT error = ERROR_INTERNAL_ERROR;
  wStream* s = NULL;
  size_t cbExpertBlobW = 0;
  WCHAR* expertBlobW = NULL;
  size_t cbRaConnectionStringW = 0;
  REMDESK_CTL_HEADER ctlHeader = { 0 };

  WINPR_ASSERT(remdesk);

  if ((error = remdesk_generate_expert_blob(remdesk)))
  {
    WLog_ERR(TAG, "remdesk_generate_expert_blob failed with error %" PRIu32 "", error);
    return error;
  }

  const char* expertBlob = remdesk->ExpertBlob;
  WINPR_ASSERT(remdesk->rdpcontext);
  rdpSettings* settings = remdesk->rdpcontext->settings;
  WINPR_ASSERT(settings);

  const char* raConnectionString =
      freerdp_settings_get_string(settings, FreeRDP_RemoteAssistanceRCTicket);
  WCHAR* raConnectionStringW =
      ConvertUtf8ToWCharAlloc(raConnectionString, &cbRaConnectionStringW);

  if (!raConnectionStringW || (cbRaConnectionStringW > UINT32_MAX / sizeof(WCHAR)))
    goto out;

  cbRaConnectionStringW = cbRaConnectionStringW * sizeof(WCHAR);

  expertBlobW = ConvertUtf8ToWCharAlloc(expertBlob, &cbExpertBlobW);

  if (!expertBlobW || (cbExpertBlobW > UINT32_MAX / sizeof(WCHAR)))
    goto out;

  cbExpertBlobW = cbExpertBlobW * sizeof(WCHAR);
  remdesk_prepare_ctl_header(&(ctlHeader), REMDESK_CTL_AUTHENTICATE,
                             cbRaConnectionStringW + cbExpertBlobW);
  s = Stream_New(NULL, REMDESK_CHANNEL_CTL_SIZE + ctlHeader.ch.DataLength);

  if (!s)
  {
    WLog_ERR(TAG, "Stream_New failed!");
    error = CHANNEL_RC_NO_MEMORY;
    goto out;
  }

  remdesk_write_ctl_header(s, &(ctlHeader));
  Stream_Write(s, raConnectionStringW, cbRaConnectionStringW);
  Stream_Write(s, expertBlobW, cbExpertBlobW);
  Stream_SealLength(s);

  if ((error = remdesk_virtual_channel_write(remdesk, s)))
    WLog_ERR(TAG, "remdesk_virtual_channel_write failed with error %" PRIu32 "!", error);

out:
  free(raConnectionStringW);
  free(expertBlobW);

  return error;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_send_ctl_remote_control_desktop_pdu(remdeskPlugin* remdesk)
{
  UINT error = 0;
  size_t length = 0;

  WINPR_ASSERT(remdesk);
  WINPR_ASSERT(remdesk->rdpcontext);
  rdpSettings* settings = remdesk->rdpcontext->settings;
  WINPR_ASSERT(settings);

  const char* raConnectionString =
      freerdp_settings_get_string(settings, FreeRDP_RemoteAssistanceRCTicket);
  WCHAR* raConnectionStringW = ConvertUtf8ToWCharAlloc(raConnectionString, &length);
  size_t cbRaConnectionStringW = length * sizeof(WCHAR);

  if (!raConnectionStringW)
    return ERROR_INTERNAL_ERROR;

  REMDESK_CTL_HEADER ctlHeader = { 0 };
  remdesk_prepare_ctl_header(&ctlHeader, REMDESK_CTL_REMOTE_CONTROL_DESKTOP,
                             cbRaConnectionStringW);
  wStream* s = Stream_New(NULL, REMDESK_CHANNEL_CTL_SIZE + ctlHeader.ch.DataLength);

  if (!s)
  {
    WLog_ERR(TAG, "Stream_New failed!");
    error = CHANNEL_RC_NO_MEMORY;
    goto out;
  }

  remdesk_write_ctl_header(s, &ctlHeader);
  Stream_Write(s, raConnectionStringW, cbRaConnectionStringW);
  Stream_SealLength(s);

  if ((error = remdesk_virtual_channel_write(remdesk, s)))
    WLog_ERR(TAG, "remdesk_virtual_channel_write failed with error %" PRIu32 "!", error);

out:
  free(raConnectionStringW);

  return error;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_send_ctl_verify_password_pdu(remdeskPlugin* remdesk)
{
  UINT error = ERROR_INTERNAL_ERROR;
  wStream* s = NULL;
  size_t cbExpertBlobW = 0;
  WCHAR* expertBlobW = NULL;
  REMDESK_CTL_VERIFY_PASSWORD_PDU pdu = { 0 };

  WINPR_ASSERT(remdesk);

  if ((error = remdesk_generate_expert_blob(remdesk)))
  {
    WLog_ERR(TAG, "remdesk_generate_expert_blob failed with error %" PRIu32 "!", error);
    return error;
  }

  pdu.expertBlob = remdesk->ExpertBlob;
  expertBlobW = ConvertUtf8ToWCharAlloc(pdu.expertBlob, &cbExpertBlobW);

  if (!expertBlobW || (cbExpertBlobW > UINT32_MAX / sizeof(WCHAR)))
    goto out;

  cbExpertBlobW = cbExpertBlobW * sizeof(WCHAR);
  remdesk_prepare_ctl_header(&(pdu.ctlHeader), REMDESK_CTL_VERIFY_PASSWORD, cbExpertBlobW);
  s = Stream_New(NULL, REMDESK_CHANNEL_CTL_SIZE + pdu.ctlHeader.ch.DataLength);

  if (!s)
  {
    WLog_ERR(TAG, "Stream_New failed!");
    error = CHANNEL_RC_NO_MEMORY;
    goto out;
  }

  remdesk_write_ctl_header(s, &(pdu.ctlHeader));
  Stream_Write(s, (BYTE*)expertBlobW, cbExpertBlobW);
  Stream_SealLength(s);

  if ((error = remdesk_virtual_channel_write(remdesk, s)))
    WLog_ERR(TAG, "remdesk_virtual_channel_write failed with error %" PRIu32 "!", error);

out:
  free(expertBlobW);

  return error;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_send_ctl_expert_on_vista_pdu(remdeskPlugin* remdesk)
{
  UINT error = 0;
  wStream* s = NULL;
  REMDESK_CTL_EXPERT_ON_VISTA_PDU pdu;

  WINPR_ASSERT(remdesk);

  if ((error = remdesk_generate_expert_blob(remdesk)))
  {
    WLog_ERR(TAG, "remdesk_generate_expert_blob failed with error %" PRIu32 "!", error);
    return error;
  }

  pdu.EncryptedPasswordLength = remdesk->EncryptedPassStubSize;
  pdu.EncryptedPassword = remdesk->EncryptedPassStub;
  remdesk_prepare_ctl_header(&(pdu.ctlHeader), REMDESK_CTL_EXPERT_ON_VISTA,
                             pdu.EncryptedPasswordLength);
  s = Stream_New(NULL, REMDESK_CHANNEL_CTL_SIZE + pdu.ctlHeader.ch.DataLength);

  if (!s)
  {
    WLog_ERR(TAG, "Stream_New failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  remdesk_write_ctl_header(s, &(pdu.ctlHeader));
  Stream_Write(s, pdu.EncryptedPassword, pdu.EncryptedPasswordLength);
  Stream_SealLength(s);
  return remdesk_virtual_channel_write(remdesk, s);
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_recv_ctl_pdu(remdeskPlugin* remdesk, wStream* s, REMDESK_CHANNEL_HEADER* header)
{
  UINT error = CHANNEL_RC_OK;
  UINT32 msgType = 0;
  UINT32 result = 0;

  WINPR_ASSERT(remdesk);
  WINPR_ASSERT(s);
  WINPR_ASSERT(header);

  if (!Stream_CheckAndLogRequiredLength(TAG, s, 4))
    return ERROR_INVALID_DATA;

  Stream_Read_UINT32(s, msgType); /* msgType (4 bytes) */

  // WLog_DBG(TAG, "msgType: %"PRIu32"", msgType);

  switch (msgType)
  {
    case REMDESK_CTL_REMOTE_CONTROL_DESKTOP:
      break;

    case REMDESK_CTL_RESULT:
      if ((error = remdesk_recv_ctl_result_pdu(remdesk, s, header, &result)))
        WLog_ERR(TAG, "remdesk_recv_ctl_result_pdu failed with error %" PRIu32 "", error);

      break;

    case REMDESK_CTL_AUTHENTICATE:
      break;

    case REMDESK_CTL_SERVER_ANNOUNCE:
      if ((error = remdesk_recv_ctl_server_announce_pdu(remdesk, s, header)))
        WLog_ERR(TAG, "remdesk_recv_ctl_server_announce_pdu failed with error %" PRIu32 "",
                 error);

      break;

    case REMDESK_CTL_DISCONNECT:
      break;

    case REMDESK_CTL_VERSIONINFO:
      if ((error = remdesk_recv_ctl_version_info_pdu(remdesk, s, header)))
      {
        WLog_ERR(TAG, "remdesk_recv_ctl_version_info_pdu failed with error %" PRIu32 "",
                 error);
        break;
      }

      if (remdesk->Version == 1)
      {
        if ((error = remdesk_send_ctl_version_info_pdu(remdesk)))
        {
          WLog_ERR(TAG, "remdesk_send_ctl_version_info_pdu failed with error %" PRIu32 "",
                   error);
          break;
        }

        if ((error = remdesk_send_ctl_authenticate_pdu(remdesk)))
        {
          WLog_ERR(TAG, "remdesk_send_ctl_authenticate_pdu failed with error %" PRIu32 "",
                   error);
          break;
        }

        if ((error = remdesk_send_ctl_remote_control_desktop_pdu(remdesk)))
        {
          WLog_ERR(
              TAG,
              "remdesk_send_ctl_remote_control_desktop_pdu failed with error %" PRIu32 "",
              error);
          break;
        }
      }
      else if (remdesk->Version == 2)
      {
        if ((error = remdesk_send_ctl_expert_on_vista_pdu(remdesk)))
        {
          WLog_ERR(TAG,
                   "remdesk_send_ctl_expert_on_vista_pdu failed with error %" PRIu32 "",
                   error);
          break;
        }

        if ((error = remdesk_send_ctl_verify_password_pdu(remdesk)))
        {
          WLog_ERR(TAG,
                   "remdesk_send_ctl_verify_password_pdu failed with error %" PRIu32 "",
                   error);
          break;
        }
      }

      break;

    case REMDESK_CTL_ISCONNECTED:
      break;

    case REMDESK_CTL_VERIFY_PASSWORD:
      break;

    case REMDESK_CTL_EXPERT_ON_VISTA:
      break;

    case REMDESK_CTL_RANOVICE_NAME:
      break;

    case REMDESK_CTL_RAEXPERT_NAME:
      break;

    case REMDESK_CTL_TOKEN:
      break;

    default:
      WLog_ERR(TAG, "unknown msgType: %" PRIu32 "", msgType);
      error = ERROR_INVALID_DATA;
      break;
  }

  return error;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_process_receive(remdeskPlugin* remdesk, wStream* s)
{
  UINT status = 0;
  REMDESK_CHANNEL_HEADER header;

  WINPR_ASSERT(remdesk);
  WINPR_ASSERT(s);

#if 0
  WLog_DBG(TAG, "RemdeskReceive: %"PRIuz"", Stream_GetRemainingLength(s));
  winpr_HexDump(Stream_ConstPointer(s), Stream_GetRemainingLength(s));
#endif

  if ((status = remdesk_read_channel_header(s, &header)))
  {
    WLog_ERR(TAG, "remdesk_read_channel_header failed with error %" PRIu32 "", status);
    return status;
  }

  if (strcmp(header.ChannelName, "RC_CTL") == 0)
  {
    status = remdesk_recv_ctl_pdu(remdesk, s, &header);
  }
  else if (strcmp(header.ChannelName, "70") == 0)
  {
  }
  else if (strcmp(header.ChannelName, "71") == 0)
  {
  }
  else if (strcmp(header.ChannelName, ".") == 0)
  {
  }
  else if (strcmp(header.ChannelName, "1000.") == 0)
  {
  }
  else if (strcmp(header.ChannelName, "RA_FX") == 0)
  {
  }
  else
  {
  }

  return status;
}

static void remdesk_process_connect(remdeskPlugin* remdesk)
{
  WINPR_ASSERT(remdesk);
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_virtual_channel_event_data_received(remdeskPlugin* remdesk, const void* pData,
                                                        UINT32 dataLength, UINT32 totalLength,
                                                        UINT32 dataFlags)
{
  wStream* data_in = NULL;

  WINPR_ASSERT(remdesk);

  if ((dataFlags & CHANNEL_FLAG_SUSPEND) || (dataFlags & CHANNEL_FLAG_RESUME))
  {
    return CHANNEL_RC_OK;
  }

  if (dataFlags & CHANNEL_FLAG_FIRST)
  {
    if (remdesk->data_in)
      Stream_Free(remdesk->data_in, TRUE);

    remdesk->data_in = Stream_New(NULL, totalLength);

    if (!remdesk->data_in)
    {
      WLog_ERR(TAG, "Stream_New failed!");
      return CHANNEL_RC_NO_MEMORY;
    }
  }

  data_in = remdesk->data_in;

  if (!Stream_EnsureRemainingCapacity(data_in, dataLength))
  {
    WLog_ERR(TAG, "Stream_EnsureRemainingCapacity failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  Stream_Write(data_in, pData, dataLength);

  if (dataFlags & CHANNEL_FLAG_LAST)
  {
    if (Stream_Capacity(data_in) != Stream_GetPosition(data_in))
    {
      WLog_ERR(TAG, "read error");
      return ERROR_INTERNAL_ERROR;
    }

    remdesk->data_in = NULL;
    Stream_SealLength(data_in);
    Stream_SetPosition(data_in, 0);

    if (!MessageQueue_Post(remdesk->queue, NULL, 0, (void*)data_in, NULL))
    {
      WLog_ERR(TAG, "MessageQueue_Post failed!");
      return ERROR_INTERNAL_ERROR;
    }
  }

  return CHANNEL_RC_OK;
}

static VOID VCAPITYPE remdesk_virtual_channel_open_event_ex(LPVOID lpUserParam, DWORD openHandle,
                                                            UINT event, LPVOID pData,
                                                            UINT32 dataLength, UINT32 totalLength,
                                                            UINT32 dataFlags)
{
  UINT error = CHANNEL_RC_OK;
  remdeskPlugin* remdesk = (remdeskPlugin*)lpUserParam;

  switch (event)
  {
    case CHANNEL_EVENT_INITIALIZED:
      break;

    case CHANNEL_EVENT_DATA_RECEIVED:
      if (!remdesk || (remdesk->OpenHandle != openHandle))
      {
        WLog_ERR(TAG, "error no match");
        return;
      }
      if ((error = remdesk_virtual_channel_event_data_received(remdesk, pData, dataLength,
                                                               totalLength, dataFlags)))
        WLog_ERR(TAG,
                 "remdesk_virtual_channel_event_data_received failed with error %" PRIu32
                 "!",
                 error);

      break;

    case CHANNEL_EVENT_WRITE_CANCELLED:
    case CHANNEL_EVENT_WRITE_COMPLETE:
    {
      wStream* s = (wStream*)pData;
      Stream_Free(s, TRUE);
    }
    break;

    case CHANNEL_EVENT_USER:
      break;

    default:
      WLog_ERR(TAG, "unhandled event %" PRIu32 "!", event);
      error = ERROR_INTERNAL_ERROR;
      break;
  }

  if (error && remdesk && remdesk->rdpcontext)
    setChannelError(remdesk->rdpcontext, error,
                    "remdesk_virtual_channel_open_event_ex reported an error");
}

static DWORD WINAPI remdesk_virtual_channel_client_thread(LPVOID arg)
{
  wStream* data = NULL;
  wMessage message = { 0 };
  remdeskPlugin* remdesk = (remdeskPlugin*)arg;
  UINT error = CHANNEL_RC_OK;

  WINPR_ASSERT(remdesk);

  remdesk_process_connect(remdesk);

  while (1)
  {
    if (!MessageQueue_Wait(remdesk->queue))
    {
      WLog_ERR(TAG, "MessageQueue_Wait failed!");
      error = ERROR_INTERNAL_ERROR;
      break;
    }

    if (!MessageQueue_Peek(remdesk->queue, &message, TRUE))
    {
      WLog_ERR(TAG, "MessageQueue_Peek failed!");
      error = ERROR_INTERNAL_ERROR;
      break;
    }

    if (message.id == WMQ_QUIT)
      break;

    if (message.id == 0)
    {
      data = (wStream*)message.wParam;

      if ((error = remdesk_process_receive(remdesk, data)))
      {
        WLog_ERR(TAG, "remdesk_process_receive failed with error %" PRIu32 "!", error);
        Stream_Free(data, TRUE);
        break;
      }

      Stream_Free(data, TRUE);
    }
  }

  if (error && remdesk->rdpcontext)
    setChannelError(remdesk->rdpcontext, error,
                    "remdesk_virtual_channel_client_thread reported an error");

  ExitThread(error);
  return error;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_virtual_channel_event_connected(remdeskPlugin* remdesk, LPVOID pData,
                                                    UINT32 dataLength)
{
  UINT error = 0;

  WINPR_ASSERT(remdesk);

  remdesk->queue = MessageQueue_New(NULL);

  if (!remdesk->queue)
  {
    WLog_ERR(TAG, "MessageQueue_New failed!");
    error = CHANNEL_RC_NO_MEMORY;
    goto error_out;
  }

  remdesk->thread =
      CreateThread(NULL, 0, remdesk_virtual_channel_client_thread, (void*)remdesk, 0, NULL);

  if (!remdesk->thread)
  {
    WLog_ERR(TAG, "CreateThread failed");
    error = ERROR_INTERNAL_ERROR;
    goto error_out;
  }

  return remdesk->channelEntryPoints.pVirtualChannelOpenEx(
      remdesk->InitHandle, &remdesk->OpenHandle, remdesk->channelDef.name,
      remdesk_virtual_channel_open_event_ex);
error_out:
  MessageQueue_Free(remdesk->queue);
  remdesk->queue = NULL;
  return error;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT remdesk_virtual_channel_event_disconnected(remdeskPlugin* remdesk)
{
  UINT rc = CHANNEL_RC_OK;

  WINPR_ASSERT(remdesk);

  if (remdesk->queue && remdesk->thread)
  {
    if (MessageQueue_PostQuit(remdesk->queue, 0) &&
        (WaitForSingleObject(remdesk->thread, INFINITE) == WAIT_FAILED))
    {
      rc = GetLastError();
      WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "", rc);
      return rc;
    }
  }

  if (remdesk->OpenHandle != 0)
  {
    WINPR_ASSERT(remdesk->channelEntryPoints.pVirtualChannelCloseEx);
    rc = remdesk->channelEntryPoints.pVirtualChannelCloseEx(remdesk->InitHandle,
                                                            remdesk->OpenHandle);

    if (CHANNEL_RC_OK != rc)
    {
      WLog_ERR(TAG, "pVirtualChannelCloseEx failed with %s [%08" PRIX32 "]",
               WTSErrorToString(rc), rc);
    }

    remdesk->OpenHandle = 0;
  }
  MessageQueue_Free(remdesk->queue);
  CloseHandle(remdesk->thread);
  Stream_Free(remdesk->data_in, TRUE);
  remdesk->data_in = NULL;
  remdesk->queue = NULL;
  remdesk->thread = NULL;
  return rc;
}

static void remdesk_virtual_channel_event_terminated(remdeskPlugin* remdesk)
{
  WINPR_ASSERT(remdesk);

  remdesk->InitHandle = 0;
  free(remdesk->context);
  free(remdesk);
}

static VOID VCAPITYPE remdesk_virtual_channel_init_event_ex(LPVOID lpUserParam, LPVOID pInitHandle,
                                                            UINT event, LPVOID pData,
                                                            UINT dataLength)
{
  UINT error = CHANNEL_RC_OK;
  remdeskPlugin* remdesk = (remdeskPlugin*)lpUserParam;

  if (!remdesk || (remdesk->InitHandle != pInitHandle))
  {
    WLog_ERR(TAG, "error no match");
    return;
  }

  switch (event)
  {
    case CHANNEL_EVENT_CONNECTED:
      if ((error = remdesk_virtual_channel_event_connected(remdesk, pData, dataLength)))
        WLog_ERR(TAG,
                 "remdesk_virtual_channel_event_connected failed with error %" PRIu32 "",
                 error);

      break;

    case CHANNEL_EVENT_DISCONNECTED:
      if ((error = remdesk_virtual_channel_event_disconnected(remdesk)))
        WLog_ERR(TAG,
                 "remdesk_virtual_channel_event_disconnected failed with error %" PRIu32 "",
                 error);

      break;

    case CHANNEL_EVENT_TERMINATED:
      remdesk_virtual_channel_event_terminated(remdesk);
      break;

    case CHANNEL_EVENT_ATTACHED:
    case CHANNEL_EVENT_DETACHED:
    default:
      break;
  }

  if (error && remdesk->rdpcontext)
    setChannelError(remdesk->rdpcontext, error,
                    "remdesk_virtual_channel_init_event reported an error");
}

/* remdesk is always built-in */
#define VirtualChannelEntryEx remdesk_VirtualChannelEntryEx

FREERDP_ENTRY_POINT(BOOL VCAPITYPE VirtualChannelEntryEx(PCHANNEL_ENTRY_POINTS pEntryPoints,
                                                         PVOID pInitHandle))
{
  UINT rc = 0;
  remdeskPlugin* remdesk = NULL;
  RemdeskClientContext* context = NULL;
  CHANNEL_ENTRY_POINTS_FREERDP_EX* pEntryPointsEx = NULL;

  if (!pEntryPoints)
  {
    return FALSE;
  }

  remdesk = (remdeskPlugin*)calloc(1, sizeof(remdeskPlugin));

  if (!remdesk)
  {
    WLog_ERR(TAG, "calloc failed!");
    return FALSE;
  }

  remdesk->channelDef.options = CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP |
                                CHANNEL_OPTION_COMPRESS_RDP | CHANNEL_OPTION_SHOW_PROTOCOL;
  (void)sprintf_s(remdesk->channelDef.name, ARRAYSIZE(remdesk->channelDef.name),
                  REMDESK_SVC_CHANNEL_NAME);
  remdesk->Version = 2;
  pEntryPointsEx = (CHANNEL_ENTRY_POINTS_FREERDP_EX*)pEntryPoints;

  if ((pEntryPointsEx->cbSize >= sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX)) &&
      (pEntryPointsEx->MagicNumber == FREERDP_CHANNEL_MAGIC_NUMBER))
  {
    context = (RemdeskClientContext*)calloc(1, sizeof(RemdeskClientContext));

    if (!context)
    {
      WLog_ERR(TAG, "calloc failed!");
      goto error_out;
    }

    context->handle = (void*)remdesk;
    remdesk->context = context;
    remdesk->rdpcontext = pEntryPointsEx->context;
  }

  CopyMemory(&(remdesk->channelEntryPoints), pEntryPoints,
             sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX));
  remdesk->InitHandle = pInitHandle;
  rc = remdesk->channelEntryPoints.pVirtualChannelInitEx(
      remdesk, context, pInitHandle, &remdesk->channelDef, 1, VIRTUAL_CHANNEL_VERSION_WIN2000,
      remdesk_virtual_channel_init_event_ex);

  if (CHANNEL_RC_OK != rc)
  {
    WLog_ERR(TAG, "pVirtualChannelInitEx failed with %s [%08" PRIX32 "]", WTSErrorToString(rc),
             rc);
    goto error_out;
  }

  remdesk->channelEntryPoints.pInterface = context;
  return TRUE;
error_out:
  free(remdesk);
  free(context);
  return FALSE;
}

Coverage Report

Created: 2024-09-08 06:20