freerdp.c

 
#include <freerdp/config.h>
 
#include "settings.h"
 
#include <string.h>
#include <stdarg.h>
#include <time.h>
 
#include "rdp.h"
#include "input.h"
#include "update.h"
#include "surface.h"
#include "transport.h"
#include "connection.h"
#include "message.h"
#include <freerdp/buildflags.h>
#include "gateway/rpc_fault.h"
 
#include <winpr/assert.h>
 
#include <winpr/crt.h>
#include <winpr/string.h>
#include <winpr/stream.h>
#include <winpr/wtsapi.h>
#include <winpr/ssl.h>
#include <winpr/debug.h>
 
#include <freerdp/freerdp.h>
#include <freerdp/streamdump.h>
#include <freerdp/error.h>
#include <freerdp/event.h>
#include <freerdp/locale/keyboard.h>
#include <freerdp/locale/locale.h>
#include <freerdp/channels/channels.h>
#include <freerdp/version.h>
#include <freerdp/log.h>
#include <freerdp/utils/signal.h>
 
#include "../cache/pointer.h"
#include "utils.h"
 
#define TAG FREERDP_TAG("core")
 
static void sig_abort_connect(int signum, const char* signame, void* ctx)
{
  rdpContext* context = (rdpContext*)ctx;
 
  WLog_INFO(TAG, "Signal %s [%d], terminating session %p", signame, signum,
            WINPR_CXX_COMPAT_CAST(const void*, context));
  if (context)
    freerdp_abort_connect_context(context);
}
 
static int freerdp_connect_begin(freerdp* instance)
{
  BOOL rc = 0;
  rdpRdp* rdp = NULL;
  BOOL status = TRUE;
  rdpSettings* settings = NULL;
 
  if (!instance)
    return -1;
 
  WINPR_ASSERT(instance->context);
 
  /* We always set the return code to 0 before we start the connect sequence*/
  instance->ConnectionCallbackState = CLIENT_STATE_INITIAL;
  freerdp_set_last_error_log(instance->context, FREERDP_ERROR_SUCCESS);
  clearChannelError(instance->context);
  if (!utils_reset_abort(instance->context->rdp))
    return -1;
 
  rdp = instance->context->rdp;
  WINPR_ASSERT(rdp);
 
  settings = instance->context->settings;
  WINPR_ASSERT(settings);
 
  freerdp_channels_register_instance(instance->context->channels, instance);
 
  if (!freerdp_settings_set_default_order_support(settings))
    return -1;
 
  freerdp_add_signal_cleanup_handler(instance->context, sig_abort_connect);
 
  IFCALLRET(instance->PreConnect, status, instance);
  instance->ConnectionCallbackState = CLIENT_STATE_PRECONNECT_PASSED;
 
  freerdp_settings_print_warnings(settings);
  if (status)
    status = freerdp_settings_enforce_monitor_exists(settings);
 
  if (status)
    status = freerdp_settings_enforce_consistency(settings);
 
  if (status)
    status = freerdp_settings_check_client_after_preconnect(settings);
 
  if (status)
    status = rdp_set_backup_settings(rdp);
  if (status)
    status = utils_reload_channels(instance->context);
 
  const UINT32 cp = freerdp_settings_get_uint32(settings, FreeRDP_KeyboardCodePage);
  int64_t KeyboardLayout = freerdp_get_keyboard_default_layout_for_locale(cp);
  if (KeyboardLayout == 0)
    KeyboardLayout = freerdp_settings_get_uint32(settings, FreeRDP_KeyboardLayout);
 
  switch (KeyboardLayout)
  {
    case KBD_JAPANESE:
    case KBD_JAPANESE_INPUT_SYSTEM_MS_IME2002:
    {
      if (!freerdp_settings_set_uint32(settings, FreeRDP_KeyboardType,
                                       WINPR_KBD_TYPE_JAPANESE))
        return -1;
      if (!freerdp_settings_set_uint32(settings, FreeRDP_KeyboardSubType, 2))
        return -1;
      if (!freerdp_settings_set_uint32(settings, FreeRDP_KeyboardFunctionKey, 12))
        return -1;
    }
    break;
    default:
      break;
  }
 
  if (!status)
  {
    rdpContext* context = instance->context;
    WINPR_ASSERT(context);
    freerdp_set_last_error_if_not(context, FREERDP_ERROR_PRE_CONNECT_FAILED);
 
    WLog_Print(context->log, WLOG_ERROR, "freerdp_pre_connect failed: %s",
               rdp_client_connection_state_string(instance->ConnectionCallbackState));
    return 0;
  }
 
  rc = rdp_client_connect(rdp);
 
  /* --authonly tests the connection without a UI */
  if (freerdp_settings_get_bool(rdp->settings, FreeRDP_AuthenticationOnly))
  {
    rdpContext* context = rdp->context;
    WINPR_ASSERT(context);
    WLog_Print(context->log, WLOG_ERROR, "Authentication only, exit status %" PRId32 "", rc);
    return 0;
  }
 
  return rc ? 1 : 0;
}
 
BOOL freerdp_connect(freerdp* instance)
{
  BOOL status = FALSE;
  ConnectionResultEventArgs e = { 0 };
  const int rc = freerdp_connect_begin(instance);
  rdpRdp* rdp = NULL;
  UINT status2 = ERROR_INTERNAL_ERROR;
 
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
 
  rdp = instance->context->rdp;
  WINPR_ASSERT(rdp);
  WINPR_ASSERT(rdp->settings);
 
  if (rc > 0)
  /* Pointers might have changed in between */
  {
    rdp_update_internal* up = update_cast(rdp->update);
 
    if (freerdp_settings_get_bool(rdp->settings, FreeRDP_DumpRemoteFx))
    {
      up->pcap_rfx = pcap_open(
          freerdp_settings_get_string(rdp->settings, FreeRDP_DumpRemoteFxFile), TRUE);
 
      if (up->pcap_rfx)
        up->dump_rfx = TRUE;
    }
 
    pointer_cache_register_callbacks(instance->context->update);
    status = IFCALLRESULT(TRUE, instance->PostConnect, instance);
    instance->ConnectionCallbackState = CLIENT_STATE_POSTCONNECT_PASSED;
 
    if (status)
      status2 = freerdp_channels_post_connect(instance->context->channels, instance);
  }
  else
  {
    status2 = CHANNEL_RC_OK;
    if (freerdp_get_last_error(instance->context) == FREERDP_ERROR_CONNECT_TRANSPORT_FAILED)
      status = freerdp_reconnect(instance);
    else
      goto freerdp_connect_finally;
  }
 
  if (!status || (status2 != CHANNEL_RC_OK) || !update_post_connect(instance->context->update))
  {
    rdpContext* context = instance->context;
    WINPR_ASSERT(context);
    WLog_Print(context->log, WLOG_ERROR, "freerdp_post_connect failed");
 
    freerdp_set_last_error_if_not(context, FREERDP_ERROR_POST_CONNECT_FAILED);
 
    status = FALSE;
    goto freerdp_connect_finally;
  }
 
  if (rdp->settings->PlayRemoteFx)
  {
    wStream* s = NULL;
    rdp_update_internal* update = update_cast(instance->context->update);
    pcap_record record = { 0 };
 
    WINPR_ASSERT(update);
    update->pcap_rfx = pcap_open(rdp->settings->PlayRemoteFxFile, FALSE);
    status = FALSE;
 
    if (!update->pcap_rfx)
      goto freerdp_connect_finally;
    else
      update->play_rfx = TRUE;
 
    status = TRUE;
 
    while (pcap_has_next_record(update->pcap_rfx) && status)
    {
      pcap_get_next_record_header(update->pcap_rfx, &record);
 
      s = transport_take_from_pool(rdp->transport, record.length);
      if (!s)
        break;
 
      record.data = Stream_Buffer(s);
      pcap_get_next_record_content(update->pcap_rfx, &record);
      Stream_SetLength(s, record.length);
      Stream_SetPosition(s, 0);
 
      if (!update_begin_paint(&update->common))
        status = FALSE;
      else
      {
        if (update_recv_surfcmds(&update->common, s) < 0)
          status = FALSE;
 
        if (!update_end_paint(&update->common))
          status = FALSE;
      }
 
      Stream_Release(s);
    }
 
    pcap_close(update->pcap_rfx);
    update->pcap_rfx = NULL;
    goto freerdp_connect_finally;
  }
 
  if (rdp->errorInfo == ERRINFO_SERVER_INSUFFICIENT_PRIVILEGES)
    freerdp_set_last_error_log(instance->context, FREERDP_ERROR_INSUFFICIENT_PRIVILEGES);
 
  transport_set_connected_event(rdp->transport);
 
freerdp_connect_finally:
  EventArgsInit(&e, "freerdp");
  e.result = status ? 0 : -1;
  PubSub_OnConnectionResult(rdp->pubSub, instance->context, &e);
 
  if (!status)
    freerdp_disconnect(instance);
 
  return status;
}
 
#if !defined(WITHOUT_FREERDP_3x_DEPRECATED)
BOOL freerdp_abort_connect(freerdp* instance)
{
  if (!instance)
    return FALSE;
 
  return freerdp_abort_connect_context(instance->context);
}
#endif
 
BOOL freerdp_abort_connect_context(rdpContext* context)
{
  if (!context)
    return FALSE;
 
  freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_CANCELLED);
  return utils_abort_connect(context->rdp);
}
 
#if defined(WITH_FREERDP_DEPRECATED)
BOOL freerdp_get_fds(freerdp* instance, void** rfds, int* rcount, WINPR_ATTR_UNUSED void** wfds,
                     WINPR_ATTR_UNUSED int* wcount)
{
  rdpRdp* rdp = NULL;
 
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
 
  rdp = instance->context->rdp;
  WINPR_ASSERT(rdp);
 
  transport_get_fds(rdp->transport, rfds, rcount);
  return TRUE;
}
#endif
 
BOOL freerdp_check_fds(freerdp* instance)
{
  int status = 0;
  rdpRdp* rdp = NULL;
 
  if (!instance)
    return FALSE;
 
  if (!instance->context)
    return FALSE;
 
  if (!instance->context->rdp)
    return FALSE;
 
  rdp = instance->context->rdp;
  status = rdp_check_fds(rdp);
 
  if (status < 0)
  {
    TerminateEventArgs e;
    rdpContext* context = instance->context;
    WINPR_ASSERT(context);
 
    WLog_Print(context->log, WLOG_DEBUG, "rdp_check_fds() - %i", status);
    EventArgsInit(&e, "freerdp");
    e.code = 0;
    PubSub_OnTerminate(rdp->pubSub, context, &e);
    return FALSE;
  }
 
  return TRUE;
}
 
DWORD freerdp_get_event_handles(rdpContext* context, HANDLE* events, DWORD count)
{
  DWORD nCount = 0;
 
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->rdp);
  WINPR_ASSERT(events || (count == 0));
 
  const size_t rrc = rdp_get_event_handles(context->rdp, &events[nCount], count - nCount);
  if (rrc == 0)
    return 0;
 
  nCount += WINPR_ASSERTING_INT_CAST(uint32_t, rrc);
 
  if (events && (nCount < count + 2))
  {
    events[nCount++] = freerdp_channels_get_event_handle(context->instance);
    events[nCount++] = getChannelErrorEventHandle(context);
  }
  else
    return 0;
 
  const SSIZE_T rc = freerdp_client_channel_get_registered_event_handles(
      context->channels, &events[nCount], count - nCount);
  if (rc < 0)
    return 0;
  return nCount + (DWORD)rc;
}
 
/* Resend mouse cursor position to prevent session lock in prevent-session-lock mode */
static BOOL freerdp_prevent_session_lock(rdpContext* context)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->input);
 
  rdp_input_internal* in = input_cast(context->input);
 
  UINT32 FakeMouseMotionInterval =
      freerdp_settings_get_uint32(context->settings, FreeRDP_FakeMouseMotionInterval);
  if (FakeMouseMotionInterval && in->lastInputTimestamp)
  {
    const time_t now = time(NULL);
    if (WINPR_ASSERTING_INT_CAST(size_t, now) - in->lastInputTimestamp >
        FakeMouseMotionInterval)
    {
      WLog_Print(context->log, WLOG_DEBUG,
                 "fake mouse move: x=%d y=%d lastInputTimestamp=%" PRIu64 " "
                 "FakeMouseMotionInterval=%" PRIu32,
                 in->lastX, in->lastY, in->lastInputTimestamp, FakeMouseMotionInterval);
 
      BOOL status = freerdp_input_send_mouse_event(context->input, PTR_FLAGS_MOVE, in->lastX,
                                                   in->lastY);
      if (!status)
      {
        if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
          WLog_Print(context->log, WLOG_ERROR,
                     "freerdp_prevent_session_lock() failed - %" PRIi32 "", status);
 
        return FALSE;
      }
 
      return status;
    }
  }
 
  return TRUE;
}
 
BOOL freerdp_check_event_handles(rdpContext* context)
{
  WINPR_ASSERT(context);
 
  BOOL status = freerdp_check_fds(context->instance);
 
  if (!status)
  {
    if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
      WLog_Print(context->log, WLOG_ERROR, "freerdp_check_fds() failed - %" PRIi32 "",
                 status);
 
    return FALSE;
  }
 
  status = freerdp_channels_check_fds(context->channels, context->instance);
 
  if (!status)
  {
    if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
      WLog_Print(context->log, WLOG_ERROR,
                 "freerdp_channels_check_fds() failed - %" PRIi32 "", status);
 
    return FALSE;
  }
 
  status = checkChannelErrorEvent(context);
 
  if (!status)
  {
    if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
      WLog_Print(context->log, WLOG_ERROR, "checkChannelErrorEvent() failed - %" PRIi32 "",
                 status);
 
    return FALSE;
  }
 
  status = freerdp_prevent_session_lock(context);
 
  return status;
}
 
wMessageQueue* freerdp_get_message_queue(freerdp* instance, DWORD id)
{
  wMessageQueue* queue = NULL;
 
  WINPR_ASSERT(instance);
 
  rdpContext* context = instance->context;
  WINPR_ASSERT(context);
 
  switch (id)
  {
    case FREERDP_UPDATE_MESSAGE_QUEUE:
    {
      rdp_update_internal* update = update_cast(context->update);
      queue = update->queue;
    }
    break;
 
    case FREERDP_INPUT_MESSAGE_QUEUE:
    {
      rdp_input_internal* input = input_cast(context->input);
      queue = input->queue;
    }
    break;
    default:
      break;
  }
 
  return queue;
}
 
HANDLE freerdp_get_message_queue_event_handle(freerdp* instance, DWORD id)
{
  HANDLE event = NULL;
  wMessageQueue* queue = freerdp_get_message_queue(instance, id);
 
  if (queue)
    event = MessageQueue_Event(queue);
 
  return event;
}
 
int freerdp_message_queue_process_message(freerdp* instance, DWORD id, wMessage* message)
{
  int status = -1;
  rdpContext* context = NULL;
 
  WINPR_ASSERT(instance);
 
  context = instance->context;
  WINPR_ASSERT(context);
 
  switch (id)
  {
    case FREERDP_UPDATE_MESSAGE_QUEUE:
      status = update_message_queue_process_message(context->update, message);
      break;
 
    case FREERDP_INPUT_MESSAGE_QUEUE:
      status = input_message_queue_process_message(context->input, message);
      break;
    default:
      break;
  }
 
  return status;
}
 
int freerdp_message_queue_process_pending_messages(freerdp* instance, DWORD id)
{
  int status = -1;
  rdpContext* context = NULL;
 
  WINPR_ASSERT(instance);
 
  context = instance->context;
  WINPR_ASSERT(context);
 
  switch (id)
  {
    case FREERDP_UPDATE_MESSAGE_QUEUE:
      status = update_message_queue_process_pending_messages(context->update);
      break;
 
    case FREERDP_INPUT_MESSAGE_QUEUE:
      status = input_message_queue_process_pending_messages(context->input);
      break;
    default:
      break;
  }
 
  return status;
}
 
static BOOL freerdp_send_channel_data(freerdp* instance, UINT16 channelId, const BYTE* data,
                                      size_t size)
{
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->rdp);
  return rdp_send_channel_data(instance->context->rdp, channelId, data, size);
}
 
static BOOL freerdp_send_channel_packet(freerdp* instance, UINT16 channelId, size_t totalSize,
                                        UINT32 flags, const BYTE* data, size_t chunkSize)
{
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->rdp);
  return rdp_channel_send_packet(instance->context->rdp, channelId, totalSize, flags, data,
                                 chunkSize);
}
 
BOOL freerdp_disconnect(freerdp* instance)
{
  BOOL rc = TRUE;
 
  if (!instance || !instance->context)
    return FALSE;
 
  rdpRdp* rdp = instance->context->rdp;
  if (rdp)
  {
    /* Try to send a [MS-RDPBCGR] 1.3.1.4.1 User-Initiated on Client PDU, we don't care about
     * success */
    if (freerdp_get_last_error(instance->context) == FREERDP_ERROR_CONNECT_CANCELLED)
    {
      (void)mcs_send_disconnect_provider_ultimatum(rdp->mcs,
                                                   Disconnect_Ultimatum_user_requested);
    }
  }
 
  utils_abort_connect(rdp);
 
  if (!rdp_client_disconnect(rdp))
    rc = FALSE;
 
  rdp_update_internal* up = NULL;
  if (rdp && rdp->update)
  {
    up = update_cast(rdp->update);
 
    update_post_disconnect(rdp->update);
  }
 
  IFCALL(instance->PostDisconnect, instance);
 
  if (up)
  {
    if (up->pcap_rfx)
    {
      up->dump_rfx = FALSE;
      pcap_close(up->pcap_rfx);
      up->pcap_rfx = NULL;
    }
  }
 
  freerdp_channels_close(instance->context->channels, instance);
 
  IFCALL(instance->PostFinalDisconnect, instance);
 
  freerdp_del_signal_cleanup_handler(instance->context, sig_abort_connect);
  return rc;
}
 
#if !defined(WITHOUT_FREERDP_3x_DEPRECATED)
BOOL freerdp_disconnect_before_reconnect(freerdp* instance)
{
  WINPR_ASSERT(instance);
  return freerdp_disconnect_before_reconnect_context(instance->context);
}
#endif
 
BOOL freerdp_disconnect_before_reconnect_context(rdpContext* context)
{
  rdpRdp* rdp = NULL;
 
  WINPR_ASSERT(context);
 
  rdp = context->rdp;
  return rdp_client_disconnect_and_clear(rdp);
}
 
BOOL freerdp_reconnect(freerdp* instance)
{
  rdpRdp* rdp = NULL;
 
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
 
  if (freerdp_get_last_error(instance->context) == FREERDP_ERROR_CONNECT_CANCELLED)
    return FALSE;
 
  rdp = instance->context->rdp;
 
  if (!utils_reset_abort(instance->context->rdp))
    return FALSE;
  return rdp_client_reconnect(rdp);
}
 
#if !defined(WITHOUT_FREERDP_3x_DEPRECATED)
BOOL freerdp_shall_disconnect(const freerdp* instance)
{
  if (!instance)
    return FALSE;
 
  return freerdp_shall_disconnect_context(instance->context);
}
#endif
 
BOOL freerdp_shall_disconnect_context(const rdpContext* context)
{
  if (!context)
    return FALSE;
 
  return utils_abort_event_is_set(context->rdp);
}
 
BOOL freerdp_focus_required(freerdp* instance)
{
  rdpRdp* rdp = NULL;
  BOOL bRetCode = FALSE;
 
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
 
  rdp = instance->context->rdp;
  WINPR_ASSERT(rdp);
 
  if (rdp->resendFocus)
  {
    bRetCode = TRUE;
    rdp->resendFocus = FALSE;
  }
 
  return bRetCode;
}
 
void freerdp_set_focus(freerdp* instance)
{
  rdpRdp* rdp = NULL;
 
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
 
  rdp = instance->context->rdp;
  WINPR_ASSERT(rdp);
 
  rdp->resendFocus = TRUE;
}
 
void freerdp_get_version(int* major, int* minor, int* revision)
{
  if (major != NULL)
    *major = FREERDP_VERSION_MAJOR;
 
  if (minor != NULL)
    *minor = FREERDP_VERSION_MINOR;
 
  if (revision != NULL)
    *revision = FREERDP_VERSION_REVISION;
}
 
const char* freerdp_get_version_string(void)
{
  return FREERDP_VERSION_FULL;
}
 
const char* freerdp_get_build_config(void)
{
  WINPR_PRAGMA_DIAG_PUSH
  WINPR_PRAGMA_DIAG_IGNORED_OVERLENGTH_STRINGS
  static const char build_config[] =
      "Build configuration: " FREERDP_BUILD_CONFIG "\n"
      "Build type:          " FREERDP_BUILD_TYPE "\n"
      "CFLAGS:              " FREERDP_CFLAGS "\n"
      "Compiler:            " FREERDP_COMPILER_ID ", " FREERDP_COMPILER_VERSION "\n"
      "Target architecture: " FREERDP_TARGET_ARCH "\n";
  WINPR_PRAGMA_DIAG_POP
  return build_config;
}
 
const char* freerdp_get_build_revision(void)
{
  return FREERDP_GIT_REVISION;
}
 
static wEventType FreeRDP_Events[] = {
  DEFINE_EVENT_ENTRY(WindowStateChange),   DEFINE_EVENT_ENTRY(ResizeWindow),
  DEFINE_EVENT_ENTRY(LocalResizeWindow),   DEFINE_EVENT_ENTRY(EmbedWindow),
  DEFINE_EVENT_ENTRY(PanningChange),       DEFINE_EVENT_ENTRY(ZoomingChange),
  DEFINE_EVENT_ENTRY(ErrorInfo),           DEFINE_EVENT_ENTRY(Terminate),
  DEFINE_EVENT_ENTRY(ConnectionResult),    DEFINE_EVENT_ENTRY(ChannelConnected),
  DEFINE_EVENT_ENTRY(ChannelDisconnected), DEFINE_EVENT_ENTRY(MouseEvent),
  DEFINE_EVENT_ENTRY(Activated),           DEFINE_EVENT_ENTRY(Timer),
  DEFINE_EVENT_ENTRY(GraphicsReset)
};
 
BOOL freerdp_context_new(freerdp* instance)
{
  return freerdp_context_new_ex(instance, NULL);
}
 
static BOOL freerdp_common_context(rdpContext* context, AccessTokenType tokenType, char** token,
                                   size_t count, ...)
{
  BOOL rc = FALSE;
 
  WINPR_ASSERT(context);
  if (!context->instance || !context->instance->GetAccessToken)
    return TRUE;
 
  va_list ap;
  va_start(ap, count);
  switch (tokenType)
  {
    case ACCESS_TOKEN_TYPE_AAD:
      if (count != 2)
      {
        WLog_ERR(TAG,
                 "ACCESS_TOKEN_TYPE_AAD expected 2 additional arguments, but got %" PRIuz
                 ", aborting",
                 count);
      }
      else
      {
        const char* scope = va_arg(ap, const char*);
        const char* req_cnf = va_arg(ap, const char*);
        rc = context->instance->GetAccessToken(context->instance, tokenType, token, count,
                                               scope, req_cnf);
      }
      break;
    case ACCESS_TOKEN_TYPE_AVD:
      if (count != 0)
      {
        WLog_WARN(TAG,
                  "ACCESS_TOKEN_TYPE_AVD expected 0 additional arguments, but got %" PRIuz
                  ", ignoring",
                  count);
      }
      else
      {
        rc = context->instance->GetAccessToken(context->instance, tokenType, token, count);
      }
      break;
    default:
      break;
  }
  va_end(ap);
 
  if (!rc)
    freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_ACCESS_DENIED);
 
  return rc;
}
 
BOOL freerdp_context_new_ex(freerdp* instance, rdpSettings* settings)
{
  rdpRdp* rdp = NULL;
  rdpContext* context = NULL;
  BOOL ret = TRUE;
 
  WINPR_ASSERT(instance);
 
  instance->context = context = (rdpContext*)calloc(1, instance->ContextSize);
 
  if (!context)
    return FALSE;
 
  context->log = WLog_Get(TAG);
  if (!context->log)
    goto fail;
 
  /* Set to external settings, prevents rdp_new from creating its own instance */
  context->settings = settings;
  context->instance = instance;
  context->ServerMode = FALSE;
  context->disconnectUltimatum = 0;
 
  context->metrics = metrics_new(context);
 
  if (!context->metrics)
    goto fail;
 
  rdp = rdp_new(context);
 
  if (!rdp)
    goto fail;
 
  context->rdp = rdp;
  context->pubSub = rdp->pubSub;
 
  if (!context->pubSub)
    goto fail;
 
  PubSub_AddEventTypes(rdp->pubSub, FreeRDP_Events, ARRAYSIZE(FreeRDP_Events));
 
#if defined(WITH_FREERDP_DEPRECATED)
  instance->input = rdp->input;
  instance->update = rdp->update;
  instance->settings = rdp->settings;
  instance->autodetect = rdp->autodetect;
#endif
 
  instance->heartbeat = rdp->heartbeat;
  context->graphics = graphics_new(context);
 
  if (!context->graphics)
    goto fail;
 
  context->input = rdp->input;
  context->update = rdp->update;
  context->settings = rdp->settings;
  context->autodetect = rdp->autodetect;
 
  if (!(context->errorDescription = calloc(1, 500)))
  {
    WLog_Print(context->log, WLOG_ERROR, "calloc failed!");
    goto fail;
  }
 
  if (!(context->channelErrorEvent = CreateEvent(NULL, TRUE, FALSE, NULL)))
  {
    WLog_Print(context->log, WLOG_ERROR, "CreateEvent failed!");
    goto fail;
  }
 
  update_register_client_callbacks(rdp->update);
 
  if (!(context->channels = freerdp_channels_new(instance)))
    goto fail;
 
  context->dump = stream_dump_new();
  if (!context->dump)
    goto fail;
 
  /* Fallback:
   * Client common library might set a function pointer to handle this, but here we provide a
   * default implementation that simply calls instance->GetAccessToken.
   */
  if (!freerdp_set_common_access_token(context, freerdp_common_context))
    goto fail;
 
  IFCALLRET(instance->ContextNew, ret, instance, context);
 
  if (!ret)
    goto fail;
 
  return TRUE;
 
fail:
  freerdp_context_free(instance);
  return FALSE;
}
 
BOOL freerdp_context_reset(freerdp* instance)
{
  if (!instance)
    return FALSE;
 
  WINPR_ASSERT(instance->context);
  rdpRdp* rdp = instance->context->rdp;
 
  return rdp_reset_runtime_settings(rdp);
}
 
void freerdp_context_free(freerdp* instance)
{
  rdpContext* ctx = NULL;
 
  if (!instance)
    return;
 
  if (!instance->context)
    return;
 
  ctx = instance->context;
 
  IFCALL(instance->ContextFree, instance, ctx);
  rdp_free(ctx->rdp);
  ctx->rdp = NULL;
  ctx->settings = NULL; /* owned by rdpRdp */
 
  graphics_free(ctx->graphics);
  ctx->graphics = NULL;
 
  metrics_free(ctx->metrics);
  ctx->metrics = NULL;
 
  if (ctx->channelErrorEvent)
    (void)CloseHandle(ctx->channelErrorEvent);
  ctx->channelErrorEvent = NULL;
 
  free(ctx->errorDescription);
  ctx->errorDescription = NULL;
 
  freerdp_channels_free(ctx->channels);
  ctx->channels = NULL;
 
  freerdp_client_codecs_free(ctx->codecs);
  ctx->codecs = NULL;
 
  stream_dump_free(ctx->dump);
  ctx->dump = NULL;
 
  ctx->input = NULL;      /* owned by rdpRdp */
  ctx->update = NULL;     /* owned by rdpRdp */
  ctx->settings = NULL;   /* owned by rdpRdp */
  ctx->autodetect = NULL; /* owned by rdpRdp */
 
  free(ctx);
  instance->context = NULL;
#if defined(WITH_FREERDP_DEPRECATED)
  instance->input = NULL;      /* owned by rdpRdp */
  instance->update = NULL;     /* owned by rdpRdp */
  instance->settings = NULL;   /* owned by rdpRdp */
  instance->autodetect = NULL; /* owned by rdpRdp */
#endif
  instance->heartbeat = NULL; /* owned by rdpRdp */
}
 
int freerdp_get_disconnect_ultimatum(const rdpContext* context)
{
  WINPR_ASSERT(context);
  return context->disconnectUltimatum;
}
 
UINT32 freerdp_error_info(const freerdp* instance)
{
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->rdp);
  return instance->context->rdp->errorInfo;
}
 
void freerdp_set_error_info(rdpRdp* rdp, UINT32 error)
{
  if (!rdp)
    return;
 
  rdp_set_error_info(rdp, error);
}
 
BOOL freerdp_send_error_info(rdpRdp* rdp)
{
  if (!rdp)
    return FALSE;
 
  return rdp_send_error_info(rdp);
}
 
UINT32 freerdp_get_last_error(const rdpContext* context)
{
  WINPR_ASSERT(context);
  return context->LastError;
}
 
const char* freerdp_get_last_error_name(UINT32 code)
{
  const char* name = NULL;
  const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
  const UINT32 type = GET_FREERDP_ERROR_TYPE(code);
 
  switch (cls)
  {
    case FREERDP_ERROR_ERRBASE_CLASS:
      name = freerdp_get_error_base_name(type);
      break;
 
    case FREERDP_ERROR_ERRINFO_CLASS:
      name = freerdp_get_error_info_name(type);
      break;
 
    case FREERDP_ERROR_CONNECT_CLASS:
      name = freerdp_get_error_connect_name(type);
      break;
 
    default:
      name = rpc_error_to_string(code);
      break;
  }
 
  return name;
}
 
const char* freerdp_get_last_error_string(UINT32 code)
{
  const char* string = NULL;
  const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
  const UINT32 type = GET_FREERDP_ERROR_TYPE(code);
 
  switch (cls)
  {
    case FREERDP_ERROR_ERRBASE_CLASS:
      string = freerdp_get_error_base_string(type);
      break;
 
    case FREERDP_ERROR_ERRINFO_CLASS:
      string = freerdp_get_error_info_string(type);
      break;
 
    case FREERDP_ERROR_CONNECT_CLASS:
      string = freerdp_get_error_connect_string(type);
      break;
 
    default:
      string = rpc_error_to_string(code);
      break;
  }
 
  return string;
}
 
const char* freerdp_get_last_error_category(UINT32 code)
{
  const char* string = NULL;
  const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
  const UINT32 type = GET_FREERDP_ERROR_TYPE(code);
 
  switch (cls)
  {
    case FREERDP_ERROR_ERRBASE_CLASS:
      string = freerdp_get_error_base_category(type);
      break;
 
    case FREERDP_ERROR_ERRINFO_CLASS:
      string = freerdp_get_error_info_category(type);
      break;
 
    case FREERDP_ERROR_CONNECT_CLASS:
      string = freerdp_get_error_connect_category(type);
      break;
 
    default:
      string = rpc_error_to_category(code);
      break;
  }
 
  return string;
}
 
void freerdp_set_last_error_ex(rdpContext* context, UINT32 lastError, const char* fkt,
                               const char* file, int line)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(line >= 0);
 
  if (lastError)
  {
    if (WLog_IsLevelActive(context->log, WLOG_ERROR))
    {
      WLog_PrintTextMessage(context->log, WLOG_ERROR, (size_t)line, file, fkt,
                            "%s [0x%08" PRIX32 "]", freerdp_get_last_error_name(lastError),
                            lastError);
    }
  }
 
  if (lastError == FREERDP_ERROR_SUCCESS)
  {
    if (WLog_IsLevelActive(context->log, WLOG_DEBUG))
      WLog_PrintTextMessage(context->log, WLOG_DEBUG, (size_t)line, file, fkt,
                            "resetting error state");
  }
  else if (context->LastError != FREERDP_ERROR_SUCCESS)
  {
    if (WLog_IsLevelActive(context->log, WLOG_ERROR))
    {
      WLog_PrintTextMessage(context->log, WLOG_ERROR, (size_t)line, file, fkt,
                            "TODO: Trying to set error code %s, but %s already set!",
                            freerdp_get_last_error_name(lastError),
                            freerdp_get_last_error_name(context->LastError));
    }
  }
  context->LastError = lastError;
}
 
const char* freerdp_get_logon_error_info_type_ex(UINT32 type, char* buffer, size_t size)
{
  const char* str = freerdp_get_logon_error_info_type(type);
  (void)_snprintf(buffer, size, "%s(0x%04" PRIx32 ")", str, type);
  return buffer;
}
 
const char* freerdp_get_logon_error_info_type(UINT32 type)
{
  switch (type)
  {
    case LOGON_MSG_DISCONNECT_REFUSED:
      return "LOGON_MSG_DISCONNECT_REFUSED";
 
    case LOGON_MSG_NO_PERMISSION:
      return "LOGON_MSG_NO_PERMISSION";
 
    case LOGON_MSG_BUMP_OPTIONS:
      return "LOGON_MSG_BUMP_OPTIONS";
 
    case LOGON_MSG_RECONNECT_OPTIONS:
      return "LOGON_MSG_RECONNECT_OPTIONS";
 
    case LOGON_MSG_SESSION_TERMINATE:
      return "LOGON_MSG_SESSION_TERMINATE";
 
    case LOGON_MSG_SESSION_CONTINUE:
      return "LOGON_MSG_SESSION_CONTINUE";
 
    default:
      return "UNKNOWN";
  }
}
 
const char* freerdp_get_logon_error_info_data(UINT32 data)
{
  switch (data)
  {
    case LOGON_FAILED_BAD_PASSWORD:
      return "LOGON_FAILED_BAD_PASSWORD";
 
    case LOGON_FAILED_UPDATE_PASSWORD:
      return "LOGON_FAILED_UPDATE_PASSWORD";
 
    case LOGON_FAILED_OTHER:
      return "LOGON_FAILED_OTHER";
 
    case LOGON_WARNING:
      return "LOGON_WARNING";
 
    default:
      return "SESSION_ID";
  }
}
 
const char* freerdp_get_logon_error_info_data_ex(UINT32 data, char* buffer, size_t size)
{
  const char* str = freerdp_get_logon_error_info_data(data);
  (void)_snprintf(buffer, size, "%s(0x%04" PRIx32 ")", str, data);
  return buffer;
}
 
freerdp* freerdp_new(void)
{
  freerdp* instance = NULL;
  instance = (freerdp*)calloc(1, sizeof(freerdp));
 
  if (!instance)
    return NULL;
 
  instance->ContextSize = sizeof(rdpContext);
  instance->SendChannelData = freerdp_send_channel_data;
  instance->SendChannelPacket = freerdp_send_channel_packet;
  instance->ReceiveChannelData = freerdp_channels_data;
  return instance;
}
 
void freerdp_free(freerdp* instance)
{
  free(instance);
}
 
ULONG freerdp_get_transport_sent(const rdpContext* context, BOOL resetCount)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->rdp);
  UINT64 rc = transport_get_bytes_sent(context->rdp->transport, resetCount);
  return WINPR_CXX_COMPAT_CAST(ULONG, MIN(rc, UINT32_MAX));
}
 
BOOL freerdp_nla_impersonate(rdpContext* context)
{
  rdpNla* nla = NULL;
 
  if (!context)
    return FALSE;
 
  if (!context->rdp)
    return FALSE;
 
  if (!context->rdp->transport)
    return FALSE;
 
  nla = transport_get_nla(context->rdp->transport);
  return nla_impersonate(nla);
}
 
BOOL freerdp_nla_revert_to_self(rdpContext* context)
{
  rdpNla* nla = NULL;
 
  if (!context)
    return FALSE;
 
  if (!context->rdp)
    return FALSE;
 
  if (!context->rdp->transport)
    return FALSE;
 
  nla = transport_get_nla(context->rdp->transport);
  return nla_revert_to_self(nla);
}
 
UINT32 freerdp_get_nla_sspi_error(const rdpContext* context)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->rdp);
  WINPR_ASSERT(context->rdp->transport);
 
  rdpNla* nla = transport_get_nla(context->rdp->transport);
  return (UINT32)nla_get_sspi_error(nla);
}
 
BOOL freerdp_nla_encrypt(rdpContext* context, const SecBuffer* inBuffer, SecBuffer* outBuffer)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->rdp);
 
  rdpNla* nla = context->rdp->nla;
  return nla_encrypt(nla, inBuffer, outBuffer);
}
 
BOOL freerdp_nla_decrypt(rdpContext* context, const SecBuffer* inBuffer, SecBuffer* outBuffer)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->rdp);
 
  rdpNla* nla = context->rdp->nla;
  return nla_decrypt(nla, inBuffer, outBuffer);
}
 
SECURITY_STATUS freerdp_nla_QueryContextAttributes(rdpContext* context, DWORD ulAttr, PVOID pBuffer)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->rdp);
 
  rdpNla* nla = context->rdp->nla;
  if (!nla)
    nla = transport_get_nla(context->rdp->transport);
 
  WINPR_ASSERT(nla);
 
  return nla_QueryContextAttributes(nla, ulAttr, pBuffer);
}
 
HANDLE getChannelErrorEventHandle(rdpContext* context)
{
  WINPR_ASSERT(context);
  return context->channelErrorEvent;
}
 
BOOL checkChannelErrorEvent(rdpContext* context)
{
  WINPR_ASSERT(context);
 
  if (WaitForSingleObject(context->channelErrorEvent, 0) == WAIT_OBJECT_0)
  {
    WLog_Print(context->log, WLOG_ERROR, "%s. Error was %" PRIu32 "", context->errorDescription,
               context->channelErrorNum);
    return FALSE;
  }
 
  return TRUE;
}
 
UINT getChannelError(const rdpContext* context)
{
  WINPR_ASSERT(context);
  return context->channelErrorNum;
}
 
const char* getChannelErrorDescription(const rdpContext* context)
{
  WINPR_ASSERT(context);
  return context->errorDescription;
}
 
void clearChannelError(rdpContext* context)
{
  WINPR_ASSERT(context);
  context->channelErrorNum = 0;
  memset(context->errorDescription, 0, 500);
  (void)ResetEvent(context->channelErrorEvent);
}
 
WINPR_ATTR_FORMAT_ARG(3, 4)
void setChannelError(rdpContext* context, UINT errorNum, WINPR_FORMAT_ARG const char* format, ...)
{
  va_list ap = { 0 };
  va_start(ap, format);
 
  WINPR_ASSERT(context);
 
  context->channelErrorNum = errorNum;
  (void)vsnprintf(context->errorDescription, 499, format, ap);
  va_end(ap);
  (void)SetEvent(context->channelErrorEvent);
}
 
const char* freerdp_nego_get_routing_token(const rdpContext* context, DWORD* length)
{
  if (!context || !context->rdp)
    return NULL;
 
  return (const char*)nego_get_routing_token(context->rdp->nego, length);
}
 
BOOL freerdp_io_callback_set_event(rdpContext* context, BOOL set)
{
  WINPR_ASSERT(context);
  return rdp_io_callback_set_event(context->rdp, set);
}
 
const rdpTransportIo* freerdp_get_io_callbacks(rdpContext* context)
{
  WINPR_ASSERT(context);
  return rdp_get_io_callbacks(context->rdp);
}
 
BOOL freerdp_set_io_callbacks(rdpContext* context, const rdpTransportIo* io_callbacks)
{
  WINPR_ASSERT(context);
  return rdp_set_io_callbacks(context->rdp, io_callbacks);
}
 
BOOL freerdp_set_io_callback_context(rdpContext* context, void* usercontext)
{
  WINPR_ASSERT(context);
  return rdp_set_io_callback_context(context->rdp, usercontext);
}
 
void* freerdp_get_io_callback_context(rdpContext* context)
{
  WINPR_ASSERT(context);
  return rdp_get_io_callback_context(context->rdp);
}
 
CONNECTION_STATE freerdp_get_state(const rdpContext* context)
{
  WINPR_ASSERT(context);
  return rdp_get_state(context->rdp);
}
 
const char* freerdp_state_string(CONNECTION_STATE state)
{
  return rdp_state_string(state);
}
 
BOOL freerdp_is_active_state(const rdpContext* context)
{
  WINPR_ASSERT(context);
  return rdp_is_active_state(context->rdp);
}
 
BOOL freerdp_channels_from_mcs(rdpSettings* settings, const rdpContext* context)
{
  WINPR_ASSERT(context);
  return rdp_channels_from_mcs(settings, context->rdp);
}
 
HANDLE freerdp_abort_event(rdpContext* context)
{
  WINPR_ASSERT(context);
  return utils_get_abort_event(context->rdp);
}
 
static void test_mcs_free(rdpMcs* mcs)
{
  if (!mcs)
    return;
 
  if (mcs->context)
  {
    rdpSettings* settings = mcs->context->settings;
    freerdp_settings_free(settings);
  }
  free(mcs->context);
 
  mcs_free(mcs);
}
 
static rdpMcs* test_mcs_new(void)
{
  rdpSettings* settings = freerdp_settings_new(0);
  rdpContext* context = calloc(1, sizeof(rdpContext));
 
  if (!settings)
    goto fail;
  if (!freerdp_settings_set_bool(settings, FreeRDP_TransportDumpReplay, TRUE))
    goto fail;
 
  if (!context)
    goto fail;
  context->settings = settings;
  return mcs_new(context);
 
fail:
  free(context);
  freerdp_settings_free(settings);
 
  return NULL;
}
 
BOOL freerdp_is_valid_mcs_create_request(const BYTE* data, size_t size)
{
 
  wStream sbuffer = { 0 };
  wStream* s = Stream_StaticConstInit(&sbuffer, data, size);
 
  WINPR_ASSERT(data || (size == 0));
  WINPR_ASSERT(s);
 
  rdpMcs* mcs = test_mcs_new();
  WINPR_ASSERT(mcs);
 
  BOOL result = mcs_recv_connect_initial(mcs, s);
  test_mcs_free(mcs);
  return result;
}
 
BOOL freerdp_is_valid_mcs_create_response(const BYTE* data, size_t size)
{
 
  wStream sbuffer = { 0 };
  wStream* s = Stream_StaticConstInit(&sbuffer, data, size);
 
  WINPR_ASSERT(data || (size == 0));
  WINPR_ASSERT(s);
 
  rdpMcs* mcs = test_mcs_new();
  WINPR_ASSERT(mcs);
 
  BOOL result = mcs_recv_connect_response(mcs, s);
  test_mcs_free(mcs);
  return result;
}
 
BOOL freerdp_persist_credentials(rdpContext* context)
{
  if (!context)
    return FALSE;
  WINPR_ASSERT(context->rdp);
  return utils_persist_credentials(context->rdp->originalSettings, context->rdp->settings);
}
 
const char* freerdp_disconnect_reason_string(int reason)
{
  switch (reason)
  {
    case Disconnect_Ultimatum_domain_disconnected:
      return "rn-domain-disconnected";
    case Disconnect_Ultimatum_provider_initiated:
      return "rn-provider-initiated";
    case Disconnect_Ultimatum_token_purged:
      return "rn-token-purged";
    case Disconnect_Ultimatum_user_requested:
      return "rn-user-requested";
    case Disconnect_Ultimatum_channel_purged:
      return "rn-channel-purged";
    default:
      return "rn-unknown";
  }
}
 
BOOL freerdp_set_common_access_token(rdpContext* context,
                                     pGetCommonAccessToken GetCommonAccessToken)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->rdp);
  context->rdp->GetCommonAccessToken = GetCommonAccessToken;
  return TRUE;
}
 
pGetCommonAccessToken freerdp_get_common_access_token(const rdpContext* context)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->rdp);
  return context->rdp->GetCommonAccessToken;
}