wf_client.c

 
#include <freerdp/config.h>
 
#include <winpr/windows.h>
#include <winpr/library.h>
 
#include <winpr/crt.h>
#include <winpr/assert.h>
 
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <tchar.h>
#include <winpr/assert.h>
#include <sys/types.h>
#include <io.h>
 
#ifdef WITH_PROGRESS_BAR
#include <shobjidl.h>
#endif
 
#ifdef WITH_WINDOWS_CERT_STORE
#include <wincrypt.h>
#endif
 
#include <freerdp/log.h>
#include <freerdp/freerdp.h>
#include <freerdp/constants.h>
#include <freerdp/settings.h>
 
#include <freerdp/locale/locale.h>
#include <freerdp/locale/keyboard.h>
#include <freerdp/codec/region.h>
#include <freerdp/client/cmdline.h>
#include <freerdp/client/channels.h>
#include <freerdp/channels/channels.h>
 
#include "wf_gdi.h"
#include "wf_rail.h"
#include "wf_channels.h"
#include "wf_graphics.h"
 
#include "resource/resource.h"
 
#define TAG CLIENT_TAG("windows")
 
#define WM_FREERDP_SHOWWINDOW (WM_USER + 100)
 
static BOOL wf_has_console(void)
{
#ifdef WITH_WIN_CONSOLE
  int file = _fileno(stdin);
  int tty = _isatty(file);
#else
  int file = -1;
  int tty = 0;
#endif
 
  WLog_INFO(TAG, "Detected stdin=%d -> %s mode", file, tty ? "console" : "gui");
  return tty;
}
 
static BOOL wf_end_paint(rdpContext* context)
{
  RECT updateRect = { 0 };
  REGION16 invalidRegion = { 0 };
  RECTANGLE_16 invalidRect = { 0 };
  const RECTANGLE_16* extents = NULL;
 
  WINPR_ASSERT(context);
 
  wfContext* wfc = (wfContext*)context;
  rdpGdi* gdi = context->gdi;
  WINPR_ASSERT(gdi);
 
  HGDI_DC hdc = gdi->primary->hdc;
  WINPR_ASSERT(hdc);
  if (!hdc->hwnd)
    return TRUE;
 
  HGDI_WND hwnd = hdc->hwnd;
  WINPR_ASSERT(hwnd->invalid || (hwnd->ninvalid == 0));
 
  if (hwnd->invalid->null)
    return TRUE;
 
  const int ninvalid = hwnd->ninvalid;
  const HGDI_RGN cinvalid = hwnd->cinvalid;
 
  if (ninvalid < 1)
    return TRUE;
 
  region16_init(&invalidRegion);
 
  for (int i = 0; i < ninvalid; i++)
  {
    invalidRect.left = cinvalid[i].x;
    invalidRect.top = cinvalid[i].y;
    invalidRect.right = cinvalid[i].x + cinvalid[i].w;
    invalidRect.bottom = cinvalid[i].y + cinvalid[i].h;
    region16_union_rect(&invalidRegion, &invalidRegion, &invalidRect);
  }
 
  if (!region16_is_empty(&invalidRegion))
  {
    extents = region16_extents(&invalidRegion);
    updateRect.left = extents->left;
    updateRect.top = extents->top;
    updateRect.right = extents->right;
    updateRect.bottom = extents->bottom;
 
    wf_scale_rect(wfc, &updateRect);
 
    InvalidateRect(wfc->hwnd, &updateRect, FALSE);
 
    if (wfc->rail)
      wf_rail_invalidate_region(wfc, &invalidRegion);
  }
 
  region16_uninit(&invalidRegion);
 
  if (!wfc->is_shown)
  {
    wfc->is_shown = TRUE;
 
#ifdef WITH_PROGRESS_BAR
    if (wfc->taskBarList)
    {
      wfc->taskBarList->lpVtbl->SetProgressState(wfc->taskBarList, wfc->hwnd,
                                                 TBPF_NOPROGRESS);
    }
#endif
 
    PostMessage(wfc->hwnd, WM_FREERDP_SHOWWINDOW, 0, 0);
    WLog_INFO(TAG, "Window is shown!");
  }
  return TRUE;
}
 
static BOOL wf_begin_paint(rdpContext* context)
{
  HGDI_DC hdc;
 
  if (!context || !context->gdi || !context->gdi->primary || !context->gdi->primary->hdc)
    return FALSE;
 
  hdc = context->gdi->primary->hdc;
 
  if (!hdc || !hdc->hwnd || !hdc->hwnd->invalid)
    return FALSE;
 
  hdc->hwnd->invalid->null = TRUE;
  hdc->hwnd->ninvalid = 0;
  return TRUE;
}
 
static BOOL wf_desktop_resize(rdpContext* context)
{
  BOOL same;
  RECT rect;
  rdpSettings* settings;
  wfContext* wfc = (wfContext*)context;
 
  if (!context || !context->settings)
    return FALSE;
 
  settings = context->settings;
 
  if (wfc->primary)
  {
    same = (wfc->primary == wfc->drawing) ? TRUE : FALSE;
    wf_image_free(wfc->primary);
    wfc->primary =
        wf_image_new(wfc, freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth),
                     freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight),
                     context->gdi->dstFormat, NULL);
  }
 
  if (!gdi_resize_ex(context->gdi, freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth),
                     freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight), 0,
                     context->gdi->dstFormat, wfc->primary->pdata, NULL))
    return FALSE;
 
  if (same)
    wfc->drawing = wfc->primary;
 
  if (wfc->fullscreen != TRUE)
  {
    if (wfc->hwnd && !freerdp_settings_get_bool(settings, FreeRDP_SmartSizing))
      SetWindowPos(wfc->hwnd, HWND_TOP, -1, -1,
                   freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth) + wfc->diff.x,
                   freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight) + wfc->diff.y,
                   SWP_NOMOVE);
  }
  else
  {
    wf_update_offset(wfc);
    GetWindowRect(wfc->hwnd, &rect);
    InvalidateRect(wfc->hwnd, &rect, TRUE);
  }
 
  return TRUE;
}
 
static BOOL wf_pre_connect(freerdp* instance)
{
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->settings);
 
  rdpContext* context = instance->context;
  wfContext* wfc = (wfContext*)instance->context;
  rdpSettings* settings = context->settings;
  if (!freerdp_settings_set_uint32(settings, FreeRDP_OsMajorType, OSMAJORTYPE_WINDOWS))
    return FALSE;
  if (!freerdp_settings_set_uint32(settings, FreeRDP_OsMinorType, OSMINORTYPE_WINDOWS_NT))
    return FALSE;
  wfc->fullscreen = freerdp_settings_get_bool(settings, FreeRDP_Fullscreen);
  wfc->fullscreen_toggle = freerdp_settings_get_bool(settings, FreeRDP_ToggleFullscreen);
  UINT32 desktopWidth = freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth);
  UINT32 desktopHeight = freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight);
 
  if (wfc->percentscreen > 0)
  {
    desktopWidth = (GetSystemMetrics(SM_CXSCREEN) * wfc->percentscreen) / 100;
    if (!freerdp_settings_set_uint32(settings, FreeRDP_DesktopWidth, desktopWidth))
      return FALSE;
    desktopHeight = (GetSystemMetrics(SM_CYSCREEN) * wfc->percentscreen) / 100;
    if (!freerdp_settings_set_uint32(settings, FreeRDP_DesktopHeight, desktopHeight))
      return FALSE;
  }
 
  if (wfc->fullscreen)
  {
    if (freerdp_settings_get_bool(settings, FreeRDP_UseMultimon))
    {
      desktopWidth = GetSystemMetrics(SM_CXVIRTUALSCREEN);
      desktopHeight = GetSystemMetrics(SM_CYVIRTUALSCREEN);
    }
    else
    {
      desktopWidth = GetSystemMetrics(SM_CXSCREEN);
      desktopHeight = GetSystemMetrics(SM_CYSCREEN);
    }
  }
 
  /* FIXME: desktopWidth has a limitation that it should be divisible by 4,
   *        otherwise the screen will crash when connecting to an XP desktop.*/
  desktopWidth = (desktopWidth + 3) & (~3);
 
  if (desktopWidth != freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth))
  {
    if (!freerdp_settings_set_uint32(settings, FreeRDP_DesktopWidth, desktopWidth))
      return FALSE;
  }
 
  if (desktopHeight != freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight))
  {
    if (!freerdp_settings_set_uint32(settings, FreeRDP_DesktopHeight, desktopHeight))
      return FALSE;
  }
 
  DWORD keyboardLayoutId = freerdp_settings_get_uint32(settings, FreeRDP_KeyboardLayout);
 
  {
    CHAR name[KL_NAMELENGTH + 1] = { 0 };
    if (GetKeyboardLayoutNameA(name))
    {
      ULONG rc = 0;
 
      errno = 0;
      rc = strtoul(name, NULL, 16);
      if (errno == 0)
        keyboardLayoutId = rc;
    }
 
    if (keyboardLayoutId == 0)
    {
      const HKL layout = GetKeyboardLayout(0);
      const uint32_t masked = (uint32_t)(((uintptr_t)layout >> 16) & 0xFFFF);
      keyboardLayoutId = masked;
    }
  }
 
  if (keyboardLayoutId == 0)
    freerdp_detect_keyboard_layout_from_system_locale(&keyboardLayoutId);
  if (keyboardLayoutId == 0)
    keyboardLayoutId = ENGLISH_UNITED_STATES;
  if (!freerdp_settings_set_uint32(settings, FreeRDP_KeyboardLayout, keyboardLayoutId))
    return FALSE;
  PubSub_SubscribeChannelConnected(instance->context->pubSub, wf_OnChannelConnectedEventHandler);
  PubSub_SubscribeChannelDisconnected(instance->context->pubSub,
                                      wf_OnChannelDisconnectedEventHandler);
  return TRUE;
}
 
static void wf_append_item_to_system_menu(HMENU hMenu, UINT fMask, UINT wID, const wchar_t* text,
                                          wfContext* wfc)
{
  MENUITEMINFO item_info = { 0 };
  item_info.fMask = fMask;
  item_info.cbSize = sizeof(MENUITEMINFO);
  item_info.wID = wID;
  item_info.fType = MFT_STRING;
  item_info.dwTypeData = _wcsdup(text);
  item_info.cch = (UINT)_wcslen(text);
  if (wfc)
    item_info.dwItemData = (ULONG_PTR)wfc;
  InsertMenuItem(hMenu, wfc->systemMenuInsertPosition++, TRUE, &item_info);
}
 
static void wf_add_system_menu(wfContext* wfc)
{
  HMENU hMenu;
 
  if (wfc->fullscreen && !wfc->fullscreen_toggle)
  {
    return;
  }
 
  if (freerdp_settings_get_bool(wfc->common.context.settings, FreeRDP_DynamicResolutionUpdate))
  {
    return;
  }
 
  hMenu = GetSystemMenu(wfc->hwnd, FALSE);
 
  wf_append_item_to_system_menu(hMenu,
                                MIIM_CHECKMARKS | MIIM_FTYPE | MIIM_ID | MIIM_STRING | MIIM_DATA,
                                SYSCOMMAND_ID_SMARTSIZING, L"Smart sizing", wfc);
 
  if (freerdp_settings_get_bool(wfc->common.context.settings, FreeRDP_SmartSizing))
  {
    CheckMenuItem(hMenu, SYSCOMMAND_ID_SMARTSIZING, MF_CHECKED);
  }
 
  if (freerdp_settings_get_bool(wfc->common.context.settings, FreeRDP_RemoteAssistanceMode))
    wf_append_item_to_system_menu(hMenu, MIIM_FTYPE | MIIM_ID | MIIM_STRING,
                                  SYSCOMMAND_ID_REQUEST_CONTROL, L"Request control", wfc);
}
 
static WCHAR* wf_window_get_title(rdpSettings* settings)
{
  BOOL port;
  WCHAR* windowTitle = NULL;
  size_t size;
  WCHAR prefix[] = L"FreeRDP:";
 
  if (!settings)
    return NULL;
 
  const char* name = freerdp_settings_get_string(settings, FreeRDP_ServerHostname);
 
  if (freerdp_settings_get_string(settings, FreeRDP_WindowTitle))
    return ConvertUtf8ToWCharAlloc(freerdp_settings_get_string(settings, FreeRDP_WindowTitle),
                                   NULL);
 
  port = (freerdp_settings_get_uint32(settings, FreeRDP_ServerPort) != 3389);
  size = strlen(name) + 16 + wcslen(prefix);
  windowTitle = calloc(size, sizeof(WCHAR));
 
  if (!windowTitle)
    return NULL;
 
  if (!port)
    _snwprintf_s(windowTitle, size, _TRUNCATE, L"%s %S", prefix, name);
  else
    _snwprintf_s(windowTitle, size, _TRUNCATE, L"%s %S:%u", prefix, name,
                 freerdp_settings_get_uint32(settings, FreeRDP_ServerPort));
 
  return windowTitle;
}
 
static BOOL wf_post_connect(freerdp* instance)
{
  rdpGdi* gdi;
  DWORD dwStyle;
  rdpCache* cache;
  wfContext* wfc;
  rdpContext* context;
  rdpSettings* settings;
  EmbedWindowEventArgs e;
  const UINT32 format = PIXEL_FORMAT_BGRX32;
 
  WINPR_ASSERT(instance);
 
  context = instance->context;
  WINPR_ASSERT(context);
 
  settings = context->settings;
  WINPR_ASSERT(settings);
 
  wfc = (wfContext*)instance->context;
  WINPR_ASSERT(wfc);
 
  wfc->primary =
      wf_image_new(wfc, freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth),
                   freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight), format, NULL);
 
  if (!wfc->primary)
  {
    WLog_ERR(TAG, "Failed to allocate primary surface");
    return FALSE;
  }
 
  if (!gdi_init_ex(instance, format, 0, wfc->primary->pdata, NULL))
    return FALSE;
 
  cache = instance->context->cache;
  WINPR_ASSERT(cache);
 
  gdi = instance->context->gdi;
 
  if (!freerdp_settings_get_bool(settings, FreeRDP_SoftwareGdi))
  {
    wf_gdi_register_update_callbacks(context->update);
  }
 
  wfc->window_title = wf_window_get_title(settings);
 
  if (!wfc->window_title)
    return FALSE;
 
  if (freerdp_settings_get_bool(settings, FreeRDP_EmbeddedWindow))
  {
    if (!freerdp_settings_set_bool(settings, FreeRDP_Decorations, FALSE))
      return FALSE;
  }
 
  if (wfc->fullscreen)
    dwStyle = WS_POPUP;
  else if (!freerdp_settings_get_bool(settings, FreeRDP_Decorations))
    dwStyle = WS_CHILD | WS_BORDER;
  else
    dwStyle =
        WS_CAPTION | WS_OVERLAPPED | WS_SYSMENU | WS_MINIMIZEBOX | WS_SIZEBOX | WS_MAXIMIZEBOX;
 
  if (!wfc->hwnd)
  {
    wfc->hwnd = CreateWindowEx(0, wfc->wndClassName, wfc->window_title, dwStyle, 0, 0, 0, 0,
                               wfc->hWndParent, NULL, wfc->hInstance, NULL);
    if (!wfc->hwnd)
    {
      WLog_ERR(TAG, "CreateWindowEx failed with error: %lu", GetLastError());
      return FALSE;
    }
    SetWindowLongPtr(wfc->hwnd, GWLP_USERDATA, (LONG_PTR)wfc);
  }
 
  wf_resize_window(wfc);
  wf_add_system_menu(wfc);
  BitBlt(wfc->primary->hdc, 0, 0, freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth),
         freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight), NULL, 0, 0, BLACKNESS);
  wfc->drawing = wfc->primary;
  EventArgsInit(&e, "wfreerdp");
  e.embed = FALSE;
  e.handle = (void*)wfc->hwnd;
  PubSub_OnEmbedWindow(context->pubSub, context, &e);
#ifdef WITH_PROGRESS_BAR
  if (wfc->taskBarList)
  {
    ShowWindow(wfc->hwnd, SW_SHOWMINIMIZED);
    wfc->taskBarList->lpVtbl->SetProgressState(wfc->taskBarList, wfc->hwnd, TBPF_INDETERMINATE);
  }
#endif
  UpdateWindow(wfc->hwnd);
  context->update->BeginPaint = wf_begin_paint;
  context->update->DesktopResize = wf_desktop_resize;
  context->update->EndPaint = wf_end_paint;
  context->update->SetKeyboardIndicators = wf_keyboard_set_indicators;
  wf_register_pointer(context->graphics);
 
  wfc->floatbar = wf_floatbar_new(wfc, wfc->hInstance,
                                  freerdp_settings_get_uint32(settings, FreeRDP_Floatbar));
 
  wf_event_focus_in(wfc);
 
  return TRUE;
}
 
static void wf_post_disconnect(freerdp* instance)
{
  wfContext* wfc;
 
  if (!instance || !instance->context)
    return;
 
  wfc = (wfContext*)instance->context;
  free(wfc->window_title);
}
 
static CREDUI_INFOW wfUiInfo = { sizeof(CREDUI_INFOW), NULL, L"Enter your credentials",
                               L"Remote Desktop Security", NULL };
 
static BOOL wf_authenticate_ex(freerdp* instance, char** username, char** password, char** domain,
                               rdp_auth_reason reason)
{
  wfContext* wfc;
  BOOL fSave;
  DWORD status;
  DWORD dwFlags;
  WCHAR UserNameW[CREDUI_MAX_USERNAME_LENGTH + 1] = { 0 };
  WCHAR UserW[CREDUI_MAX_USERNAME_LENGTH + 1] = { 0 };
  WCHAR DomainW[CREDUI_MAX_DOMAIN_TARGET_LENGTH + 1] = { 0 };
  WCHAR PasswordW[CREDUI_MAX_PASSWORD_LENGTH + 1] = { 0 };
 
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->settings);
 
  wfc = (wfContext*)instance->context;
  WINPR_ASSERT(wfc);
 
  WINPR_ASSERT(username);
  WINPR_ASSERT(domain);
  WINPR_ASSERT(password);
 
  const WCHAR auth[] = L"Target credentials requested";
  const WCHAR authPin[] = L"PIN requested";
  const WCHAR gwAuth[] = L"Gateway credentials requested";
  const WCHAR* titleW = auth;
 
  fSave = FALSE;
  dwFlags = CREDUI_FLAGS_DO_NOT_PERSIST | CREDUI_FLAGS_EXCLUDE_CERTIFICATES |
            CREDUI_FLAGS_USERNAME_TARGET_CREDENTIALS;
  switch (reason)
  {
    case AUTH_NLA:
      break;
    case AUTH_TLS:
    case AUTH_RDP:
      if ((*username) && (*password))
        return TRUE;
      break;
    case AUTH_SMARTCARD_PIN:
      dwFlags &= ~CREDUI_FLAGS_USERNAME_TARGET_CREDENTIALS;
      dwFlags |= CREDUI_FLAGS_PASSWORD_ONLY_OK | CREDUI_FLAGS_KEEP_USERNAME;
      titleW = authPin;
      if (*password)
        return TRUE;
      if (!(*username))
        *username = _strdup("PIN");
      break;
    case GW_AUTH_HTTP:
    case GW_AUTH_RDG:
    case GW_AUTH_RPC:
      titleW = gwAuth;
      break;
    default:
      return FALSE;
  }
 
  if (*username)
  {
    (void)ConvertUtf8ToWChar(*username, UserNameW, ARRAYSIZE(UserNameW));
    (void)ConvertUtf8ToWChar(*username, UserW, ARRAYSIZE(UserW));
  }
 
  if (*password)
    (void)ConvertUtf8ToWChar(*password, PasswordW, ARRAYSIZE(PasswordW));
 
  if (*domain)
    (void)ConvertUtf8ToWChar(*domain, DomainW, ARRAYSIZE(DomainW));
 
  if (_wcsnlen(PasswordW, ARRAYSIZE(PasswordW)) == 0)
  {
    if (!wfc->isConsole &&
        freerdp_settings_get_bool(wfc->common.context.settings, FreeRDP_CredentialsFromStdin))
      WLog_ERR(TAG, "Flag for stdin read present but stdin is redirected; using GUI");
    if (wfc->isConsole &&
        freerdp_settings_get_bool(wfc->common.context.settings, FreeRDP_CredentialsFromStdin))
      status = CredUICmdLinePromptForCredentialsW(titleW, NULL, 0, UserNameW,
                                                  ARRAYSIZE(UserNameW), PasswordW,
                                                  ARRAYSIZE(PasswordW), &fSave, dwFlags);
    else
      status = CredUIPromptForCredentialsW(&wfUiInfo, titleW, NULL, 0, UserNameW,
                                           ARRAYSIZE(UserNameW), PasswordW,
                                           ARRAYSIZE(PasswordW), &fSave, dwFlags);
    if (status != NO_ERROR)
    {
      WLog_ERR(TAG, "CredUIPromptForCredentials unexpected status: 0x%08lX", status);
      return FALSE;
    }
 
    if ((dwFlags & CREDUI_FLAGS_KEEP_USERNAME) == 0)
    {
      status = CredUIParseUserNameW(UserNameW, UserW, ARRAYSIZE(UserW), DomainW,
                                    ARRAYSIZE(DomainW));
      if (status != NO_ERROR)
      {
        CHAR User[CREDUI_MAX_USERNAME_LENGTH + 1] = { 0 };
        CHAR UserName[CREDUI_MAX_USERNAME_LENGTH + 1] = { 0 };
        CHAR Domain[CREDUI_MAX_DOMAIN_TARGET_LENGTH + 1] = { 0 };
 
        (void)ConvertWCharNToUtf8(UserNameW, ARRAYSIZE(UserNameW), UserName,
                                  ARRAYSIZE(UserName));
        (void)ConvertWCharNToUtf8(UserW, ARRAYSIZE(UserW), User, ARRAYSIZE(User));
        (void)ConvertWCharNToUtf8(DomainW, ARRAYSIZE(DomainW), Domain, ARRAYSIZE(Domain));
        WLog_ERR(TAG, "Failed to parse UserName: %s into User: %s Domain: %s", UserName,
                 User, Domain);
        return FALSE;
      }
    }
  }
 
  *username = ConvertWCharNToUtf8Alloc(UserW, ARRAYSIZE(UserW), NULL);
  if (!(*username))
  {
    WLog_ERR(TAG, "ConvertWCharNToUtf8Alloc failed", status);
    return FALSE;
  }
 
  if (_wcsnlen(DomainW, ARRAYSIZE(DomainW)) > 0)
    *domain = ConvertWCharNToUtf8Alloc(DomainW, ARRAYSIZE(DomainW), NULL);
  else
    *domain = _strdup("\0");
 
  if (!(*domain))
  {
    free(*username);
    WLog_ERR(TAG, "strdup failed", status);
    return FALSE;
  }
 
  *password = ConvertWCharNToUtf8Alloc(PasswordW, ARRAYSIZE(PasswordW), NULL);
  if (!(*password))
  {
    free(*username);
    free(*domain);
    return FALSE;
  }
 
  return TRUE;
}
 
static WCHAR* wf_format_text(const WCHAR* fmt, ...)
{
  int rc;
  size_t size = 0;
  WCHAR* buffer = NULL;
 
  do
  {
    WCHAR* tmp;
    va_list ap;
    va_start(ap, fmt);
    rc = _vsnwprintf(buffer, size, fmt, ap);
    va_end(ap);
    if (rc <= 0)
      goto fail;
 
    if ((size_t)rc < size)
      return buffer;
 
    size = (size_t)rc + 1;
    tmp = realloc(buffer, size * sizeof(WCHAR));
    if (!tmp)
      goto fail;
 
    buffer = tmp;
  } while (TRUE);
 
fail:
  free(buffer);
  return NULL;
}
 
#ifdef WITH_WINDOWS_CERT_STORE
/* https://stackoverflow.com/questions/1231178/load-an-pem-encoded-x-509-certificate-into-windows-cryptoapi/3803333#3803333
 */
/* https://github.com/microsoft/Windows-classic-samples/blob/main/Samples/Win7Samples/security/cryptoapi/peertrust/cpp/peertrust.cpp
 */
/* https://stackoverflow.com/questions/7340504/whats-the-correct-way-to-verify-an-ssl-certificate-in-win32
 */
 
static void wf_report_error(char* wszMessage, DWORD dwErrCode)
{
  LPSTR pwszMsgBuf = NULL;
 
  if (NULL != wszMessage && 0 != *wszMessage)
  {
    WLog_ERR(TAG, "%s", wszMessage);
  }
 
  FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
                 NULL,                                      // Location of message
                                                            //  definition ignored
                 dwErrCode,                                 // Message identifier for
                                                            //  the requested message
                 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Language identifier for
                                                            //  the requested message
                 (LPSTR)&pwszMsgBuf,                        // Buffer that receives
                                                            //  the formatted message
                 0,                                         // Size of output buffer
                                                            //  not needed as allocate
                                                            //  buffer flag is set
                 NULL                                       // Array of insert values
  );
 
  if (NULL != pwszMsgBuf)
  {
    WLog_ERR(TAG, "Error: 0x%08x (%d) %s", dwErrCode, dwErrCode, pwszMsgBuf);
    LocalFree(pwszMsgBuf);
  }
  else
  {
    WLog_ERR(TAG, "Error: 0x%08x (%d)", dwErrCode, dwErrCode);
  }
}
 
static DWORD wf_is_x509_certificate_trusted(const char* common_name, const char* subject,
                                            const char* issuer, const char* fingerprint)
{
  HRESULT hr = CRYPT_E_NOT_FOUND;
 
  DWORD dwChainFlags = CERT_CHAIN_REVOCATION_CHECK_CHAIN_EXCLUDE_ROOT;
  PCCERT_CONTEXT pCert = NULL;
  HCERTCHAINENGINE hChainEngine = NULL;
  PCCERT_CHAIN_CONTEXT pChainContext = NULL;
 
  CERT_ENHKEY_USAGE EnhkeyUsage = { 0 };
  CERT_USAGE_MATCH CertUsage = { 0 };
  CERT_CHAIN_PARA ChainPara = { 0 };
  CERT_CHAIN_POLICY_PARA ChainPolicy = { 0 };
  CERT_CHAIN_POLICY_STATUS PolicyStatus = { 0 };
  CERT_CHAIN_ENGINE_CONFIG EngineConfig = { 0 };
 
  DWORD derPubKeyLen = WINPR_ASSERTING_INT_CAST(uint32_t, strlen(fingerprint));
  char* derPubKey = calloc(derPubKeyLen, sizeof(char));
  if (NULL == derPubKey)
  {
    WLog_ERR(TAG, "Could not allocate derPubKey");
    goto CleanUp;
  }
 
  /*
   * Convert from PEM format to DER format - removes header and footer and decodes from base64
   */
  if (!CryptStringToBinaryA(fingerprint, 0, CRYPT_STRING_BASE64HEADER, derPubKey, &derPubKeyLen,
                            NULL, NULL))
  {
    WLog_ERR(TAG, "CryptStringToBinary failed. Err: %d", GetLastError());
    goto CleanUp;
  }
 
  //---------------------------------------------------------
  // Initialize data structures for chain building.
 
  EnhkeyUsage.cUsageIdentifier = 0;
  EnhkeyUsage.rgpszUsageIdentifier = NULL;
 
  CertUsage.dwType = USAGE_MATCH_TYPE_AND;
  CertUsage.Usage = EnhkeyUsage;
 
  ChainPara.cbSize = sizeof(ChainPara);
  ChainPara.RequestedUsage = CertUsage;
 
  ChainPolicy.cbSize = sizeof(ChainPolicy);
 
  PolicyStatus.cbSize = sizeof(PolicyStatus);
 
  EngineConfig.cbSize = sizeof(EngineConfig);
  EngineConfig.dwUrlRetrievalTimeout = 0;
 
  pCert = CertCreateCertificateContext(X509_ASN_ENCODING, derPubKey, derPubKeyLen);
  if (NULL == pCert)
  {
    WLog_ERR(TAG, "FAILED: Certificate could not be parsed.");
    goto CleanUp;
  }
 
  dwChainFlags |= CERT_CHAIN_ENABLE_PEER_TRUST;
 
  // When this flag is set, end entity certificates in the
  // Trusted People store are trusted without doing any chain building
  // This optimizes the chain building process.
 
  //---------------------------------------------------------
  // Create chain engine.
 
  if (!CertCreateCertificateChainEngine(&EngineConfig, &hChainEngine))
  {
    hr = HRESULT_FROM_WIN32(GetLastError());
    goto CleanUp;
  }
 
  //-------------------------------------------------------------------
  // Build a chain using CertGetCertificateChain
 
  if (!CertGetCertificateChain(hChainEngine, // use the default chain engine
                               pCert,        // pointer to the end certificate
                               NULL,         // use the default time
                               NULL,         // search no additional stores
                               &ChainPara,   // use AND logic and enhanced key usage
                                             //  as indicated in the ChainPara
                                             //  data structure
                               dwChainFlags,
                               NULL,            // currently reserved
                               &pChainContext)) // return a pointer to the chain created
  {
    hr = HRESULT_FROM_WIN32(GetLastError());
    goto CleanUp;
  }
 
  //---------------------------------------------------------------
  // Verify that the chain complies with policy
 
  if (!CertVerifyCertificateChainPolicy(CERT_CHAIN_POLICY_BASE, // use the base policy
                                        pChainContext,          // pointer to the chain
                                        &ChainPolicy,
                                        &PolicyStatus)) // return a pointer to the policy status
  {
    hr = HRESULT_FROM_WIN32(GetLastError());
    goto CleanUp;
  }
 
  if (PolicyStatus.dwError != S_OK)
  {
    wf_report_error("CertVerifyCertificateChainPolicy: Chain Status", PolicyStatus.dwError);
    hr = PolicyStatus.dwError;
    // Instruction: If the PolicyStatus.dwError is CRYPT_E_NO_REVOCATION_CHECK or
    // CRYPT_E_REVOCATION_OFFLINE, it indicates errors in obtaining
    //        revocation information. These can be ignored since the retrieval of
    // revocation information depends on network availability
 
    if (PolicyStatus.dwError == CRYPT_E_NO_REVOCATION_CHECK ||
        PolicyStatus.dwError == CRYPT_E_REVOCATION_OFFLINE)
    {
      hr = S_OK;
    }
 
    goto CleanUp;
  }
 
  WLog_INFO(TAG, "CertVerifyCertificateChainPolicy succeeded for %s (%s) issued by %s",
            common_name, subject, issuer);
 
  hr = S_OK;
CleanUp:
 
  if (FAILED(hr))
  {
    WLog_INFO(TAG, "CertVerifyCertificateChainPolicy failed for %s (%s) issued by %s",
              common_name, subject, issuer);
    wf_report_error(NULL, hr);
  }
 
  free(derPubKey);
 
  if (NULL != pChainContext)
  {
    CertFreeCertificateChain(pChainContext);
  }
 
  if (NULL != hChainEngine)
  {
    CertFreeCertificateChainEngine(hChainEngine);
  }
 
  if (NULL != pCert)
  {
    CertFreeCertificateContext(pCert);
  }
 
  return (DWORD)hr;
}
#endif
 
static DWORD wf_cli_verify_certificate_ex(freerdp* instance, const char* host, UINT16 port,
                                          const char* common_name, const char* subject,
                                          const char* issuer, const char* fingerprint, DWORD flags)
{
#ifdef WITH_WINDOWS_CERT_STORE
  if (flags & VERIFY_CERT_FLAG_FP_IS_PEM && !(flags & VERIFY_CERT_FLAG_MISMATCH))
  {
    if (wf_is_x509_certificate_trusted(common_name, subject, issuer, fingerprint) == S_OK)
    {
      return 2;
    }
  }
#endif
 
  return client_cli_verify_certificate_ex(instance, host, port, common_name, subject, issuer,
                                          fingerprint, flags);
}
 
static DWORD wf_verify_certificate_ex(freerdp* instance, const char* host, UINT16 port,
                                      const char* common_name, const char* subject,
                                      const char* issuer, const char* fingerprint, DWORD flags)
{
  WCHAR* buffer;
  WCHAR* caption;
  int what = IDCANCEL;
 
#ifdef WITH_WINDOWS_CERT_STORE
  if (flags & VERIFY_CERT_FLAG_FP_IS_PEM && !(flags & VERIFY_CERT_FLAG_MISMATCH))
  {
    if (wf_is_x509_certificate_trusted(common_name, subject, issuer, fingerprint) == S_OK)
    {
      return 2;
    }
  }
#endif
 
  buffer = wf_format_text(
      L"Certificate details:\n"
      L"\tCommonName: %S\n"
      L"\tSubject: %S\n"
      L"\tIssuer: %S\n"
      L"\tThumbprint: %S\n"
      L"\tHostMismatch: %S\n"
      L"\n"
      L"The above X.509 certificate could not be verified, possibly because you do not have "
      L"the CA certificate in your certificate store, or the certificate has expired. "
      L"Please look at the OpenSSL documentation on how to add a private CA to the store.\n"
      L"\n"
      L"YES\tAccept permanently\n"
      L"NO\tAccept for this session only\n"
      L"CANCEL\tAbort connection\n",
      common_name, subject, issuer, fingerprint,
      flags & VERIFY_CERT_FLAG_MISMATCH ? "Yes" : "No");
  caption = wf_format_text(L"Verify certificate for %S:%hu", host, port);
 
  WINPR_UNUSED(instance);
 
  if (!buffer || !caption)
    goto fail;
 
  what = MessageBoxW(NULL, buffer, caption, MB_YESNOCANCEL);
fail:
  free(buffer);
  free(caption);
 
  /* return 1 to accept and store a certificate, 2 to accept
   * a certificate only for this session, 0 otherwise */
  switch (what)
  {
    case IDYES:
      return 1;
    case IDNO:
      return 2;
    default:
      return 0;
  }
}
 
static DWORD wf_verify_changed_certificate_ex(freerdp* instance, const char* host, UINT16 port,
                                              const char* common_name, const char* subject,
                                              const char* issuer, const char* new_fingerprint,
                                              const char* old_subject, const char* old_issuer,
                                              const char* old_fingerprint, DWORD flags)
{
  WCHAR* buffer;
  WCHAR* caption;
  int what = IDCANCEL;
 
  buffer = wf_format_text(
      L"New Certificate details:\n"
      L"\tCommonName: %S\n"
      L"\tSubject: %S\n"
      L"\tIssuer: %S\n"
      L"\tThumbprint: %S\n"
      L"\tHostMismatch: %S\n"
      L"\n"
      L"Old Certificate details:\n"
      L"\tSubject: %S\n"
      L"\tIssuer: %S\n"
      L"\tThumbprint: %S"
      L"The above X.509 certificate could not be verified, possibly because you do not have "
      L"the CA certificate in your certificate store, or the certificate has expired. "
      L"Please look at the OpenSSL documentation on how to add a private CA to the store.\n"
      L"\n"
      L"YES\tAccept permanently\n"
      L"NO\tAccept for this session only\n"
      L"CANCEL\tAbort connection\n",
      common_name, subject, issuer, new_fingerprint,
      flags & VERIFY_CERT_FLAG_MISMATCH ? "Yes" : "No", old_subject, old_issuer, old_fingerprint);
  caption = wf_format_text(L"Verify certificate change for %S:%hu", host, port);
 
  WINPR_UNUSED(instance);
  if (!buffer || !caption)
    goto fail;
 
  what = MessageBoxW(NULL, buffer, caption, MB_YESNOCANCEL);
fail:
  free(buffer);
  free(caption);
 
  /* return 1 to accept and store a certificate, 2 to accept
   * a certificate only for this session, 0 otherwise */
  switch (what)
  {
    case IDYES:
      return 1;
    case IDNO:
      return 2;
    default:
      return 0;
  }
}
 
static BOOL wf_present_gateway_message(freerdp* instance, UINT32 type, BOOL isDisplayMandatory,
                                       BOOL isConsentMandatory, size_t length, const WCHAR* message)
{
  if (!isDisplayMandatory && !isConsentMandatory)
    return TRUE;
 
  /* special handling for consent messages (show modal dialog) */
  if (type == GATEWAY_MESSAGE_CONSENT && isConsentMandatory)
  {
    int mbRes;
    WCHAR* msg;
 
    msg = wf_format_text(L"%.*s\n\nI understand and agree to the terms of this policy", length,
                         message);
    mbRes = MessageBoxW(NULL, msg, L"Consent Message", MB_YESNO);
    free(msg);
 
    if (mbRes != IDYES)
      return FALSE;
  }
  else
    return client_cli_present_gateway_message(instance, type, isDisplayMandatory,
                                              isConsentMandatory, length, message);
 
  return TRUE;
}
 
static DWORD WINAPI wf_client_thread(LPVOID lpParam)
{
  MSG msg = { 0 };
  int width = 0;
  int height = 0;
  BOOL msg_ret = FALSE;
  int quit_msg = 0;
  DWORD error = 0;
 
  freerdp* instance = (freerdp*)lpParam;
  WINPR_ASSERT(instance);
 
  if (!freerdp_connect(instance))
    goto end;
 
  rdpContext* context = instance->context;
  WINPR_ASSERT(context);
 
  wfContext* wfc = (wfContext*)instance->context;
  WINPR_ASSERT(wfc);
 
  rdpChannels* channels = context->channels;
  WINPR_ASSERT(channels);
 
  rdpSettings* settings = context->settings;
  WINPR_ASSERT(settings);
 
  while (!freerdp_shall_disconnect_context(instance->context))
  {
    HANDLE handles[MAXIMUM_WAIT_OBJECTS] = { 0 };
    DWORD nCount = 0;
 
    if (freerdp_focus_required(instance))
    {
      wf_event_focus_in(wfc);
      wf_event_focus_in(wfc);
    }
 
    {
      DWORD tmp = freerdp_get_event_handles(context, &handles[nCount], 64 - nCount);
 
      if (tmp == 0)
      {
        WLog_ERR(TAG, "freerdp_get_event_handles failed");
        break;
      }
 
      nCount += tmp;
    }
 
    DWORD status = MsgWaitForMultipleObjectsEx(nCount, handles, 5 * 1000, QS_ALLINPUT,
                                               MWMO_ALERTABLE | MWMO_INPUTAVAILABLE);
    if (status == WAIT_FAILED)
    {
      WLog_ERR(TAG, "wfreerdp_run: WaitForMultipleObjects failed: 0x%08lX", GetLastError());
      break;
    }
 
    {
      if (!freerdp_check_event_handles(context))
      {
        if (client_auto_reconnect(instance))
          continue;
 
        WLog_ERR(TAG, "Failed to check FreeRDP file descriptor");
        break;
      }
    }
 
    if (freerdp_shall_disconnect_context(instance->context))
      break;
 
    quit_msg = FALSE;
 
    while (PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE))
    {
      msg_ret = GetMessage(&msg, NULL, 0, 0);
 
      if (freerdp_settings_get_bool(settings, FreeRDP_EmbeddedWindow))
      {
        if ((msg.message == WM_SETFOCUS) && (msg.lParam == 1))
        {
          PostMessage(wfc->hwnd, WM_SETFOCUS, 0, 0);
        }
        else if ((msg.message == WM_KILLFOCUS) && (msg.lParam == 1))
        {
          PostMessage(wfc->hwnd, WM_KILLFOCUS, 0, 0);
        }
      }
 
      switch (msg.message)
      {
        case WM_SIZE:
        {
          width = LOWORD(msg.lParam);
          height = HIWORD(msg.lParam);
          SetWindowPos(wfc->hwnd, HWND_TOP, 0, 0, width, height, SWP_FRAMECHANGED);
          break;
        }
        case WM_FREERDP_SHOWWINDOW:
        {
          ShowWindow(wfc->hwnd, SW_NORMAL);
          break;
        }
        default:
          break;
      }
 
      if ((msg_ret == 0) || (msg_ret == -1))
      {
        quit_msg = TRUE;
        break;
      }
 
      TranslateMessage(&msg);
      DispatchMessage(&msg);
    }
 
    if (quit_msg)
      break;
  }
 
  /* cleanup */
  freerdp_disconnect(instance);
 
end:
  error = freerdp_get_last_error(instance->context);
  WLog_DBG(TAG, "Main thread exited with %" PRIu32, error);
  ExitThread(error);
  return error;
}
 
static DWORD WINAPI wf_keyboard_thread(LPVOID lpParam)
{
  MSG msg;
  BOOL status;
  wfContext* wfc;
  HHOOK hook_handle;
  wfc = (wfContext*)lpParam;
  WINPR_ASSERT(NULL != wfc);
  hook_handle = SetWindowsHookEx(WH_KEYBOARD_LL, wf_ll_kbd_proc, wfc->hInstance, 0);
 
  if (hook_handle)
  {
    while ((status = GetMessage(&msg, NULL, 0, 0)) != 0)
    {
      if (status == -1)
      {
        WLog_ERR(TAG, "keyboard thread error getting message");
        break;
      }
      else
      {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
      }
    }
 
    UnhookWindowsHookEx(hook_handle);
  }
  else
  {
    WLog_ERR(TAG, "failed to install keyboard hook");
  }
 
  WLog_DBG(TAG, "Keyboard thread exited.");
  ExitThread(0);
  return 0;
}
 
int freerdp_client_set_window_size(wfContext* wfc, int width, int height)
{
  WLog_DBG(TAG, "freerdp_client_set_window_size %d, %d", width, height);
 
  if ((width != wfc->client_width) || (height != wfc->client_height))
  {
    PostThreadMessage(wfc->mainThreadId, WM_SIZE, SIZE_RESTORED,
                      ((UINT)height << 16) | (UINT)width);
  }
 
  return 0;
}
 
void wf_size_scrollbars(wfContext* wfc, UINT32 client_width, UINT32 client_height)
{
  const rdpSettings* settings;
  WINPR_ASSERT(wfc);
 
  settings = wfc->common.context.settings;
  WINPR_ASSERT(settings);
 
  if (wfc->disablewindowtracking)
    return;
 
  // prevent infinite message loop
  wfc->disablewindowtracking = TRUE;
 
  if (freerdp_settings_get_bool(settings, FreeRDP_SmartSizing) ||
      freerdp_settings_get_bool(settings, FreeRDP_DynamicResolutionUpdate))
  {
    wfc->xCurrentScroll = 0;
    wfc->yCurrentScroll = 0;
 
    if (wfc->xScrollVisible || wfc->yScrollVisible)
    {
      if (ShowScrollBar(wfc->hwnd, SB_BOTH, FALSE))
      {
        wfc->xScrollVisible = FALSE;
        wfc->yScrollVisible = FALSE;
      }
    }
  }
  else
  {
    SCROLLINFO si;
    BOOL horiz = wfc->xScrollVisible;
    BOOL vert = wfc->yScrollVisible;
 
    if (!horiz && client_width < freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth))
    {
      horiz = TRUE;
    }
    else if (horiz &&
             client_width >=
                 freerdp_settings_get_uint32(
                     settings, FreeRDP_DesktopWidth) /* - GetSystemMetrics(SM_CXVSCROLL)*/)
    {
      horiz = FALSE;
    }
 
    if (!vert && client_height < freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight))
    {
      vert = TRUE;
    }
    else if (vert &&
             client_height >=
                 freerdp_settings_get_uint32(
                     settings, FreeRDP_DesktopHeight) /* - GetSystemMetrics(SM_CYHSCROLL)*/)
    {
      vert = FALSE;
    }
 
    if (horiz == vert && (horiz != wfc->xScrollVisible && vert != wfc->yScrollVisible))
    {
      if (ShowScrollBar(wfc->hwnd, SB_BOTH, horiz))
      {
        wfc->xScrollVisible = horiz;
        wfc->yScrollVisible = vert;
      }
    }
 
    if (horiz != wfc->xScrollVisible)
    {
      if (ShowScrollBar(wfc->hwnd, SB_HORZ, horiz))
      {
        wfc->xScrollVisible = horiz;
      }
    }
 
    if (vert != wfc->yScrollVisible)
    {
      if (ShowScrollBar(wfc->hwnd, SB_VERT, vert))
      {
        wfc->yScrollVisible = vert;
      }
    }
 
    if (horiz)
    {
      // The horizontal scrolling range is defined by
      // (bitmap_width) - (client_width). The current horizontal
      // scroll value remains within the horizontal scrolling range.
      wfc->xMaxScroll =
          MAX(freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth) - client_width, 0);
      wfc->xCurrentScroll = MIN(wfc->xCurrentScroll, wfc->xMaxScroll);
      si.cbSize = sizeof(si);
      si.fMask = SIF_RANGE | SIF_PAGE | SIF_POS;
      si.nMin = wfc->xMinScroll;
      si.nMax = freerdp_settings_get_uint32(settings, FreeRDP_DesktopWidth);
      si.nPage = client_width;
      si.nPos = wfc->xCurrentScroll;
      SetScrollInfo(wfc->hwnd, SB_HORZ, &si, TRUE);
    }
 
    if (vert)
    {
      // The vertical scrolling range is defined by
      // (bitmap_height) - (client_height). The current vertical
      // scroll value remains within the vertical scrolling range.
      wfc->yMaxScroll = MAX(
          freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight) - client_height, 0);
      wfc->yCurrentScroll = MIN(wfc->yCurrentScroll, wfc->yMaxScroll);
      si.cbSize = sizeof(si);
      si.fMask = SIF_RANGE | SIF_PAGE | SIF_POS;
      si.nMin = wfc->yMinScroll;
      si.nMax = freerdp_settings_get_uint32(settings, FreeRDP_DesktopHeight);
      si.nPage = client_height;
      si.nPos = wfc->yCurrentScroll;
      SetScrollInfo(wfc->hwnd, SB_VERT, &si, TRUE);
    }
  }
 
  wfc->disablewindowtracking = FALSE;
  wf_update_canvas_diff(wfc);
}
 
static BOOL wfreerdp_client_global_init(void)
{
  WSADATA wsaData;
 
  WSAStartup(0x101, &wsaData);
 
  freerdp_register_addin_provider(freerdp_channels_load_static_addin_entry, 0);
 
  return TRUE;
}
 
static void wfreerdp_client_global_uninit(void)
{
  WSACleanup();
}
 
static BOOL wfreerdp_client_new(freerdp* instance, rdpContext* context)
{
  wfContext* wfc = (wfContext*)context;
  if (!wfc)
    return FALSE;
 
  // AttachConsole and stdin do not work well.
  // Use GUI input dialogs instead of command line ones.
  wfc->isConsole = wf_has_console();
 
  if (!(wfreerdp_client_global_init()))
    return FALSE;
 
  WINPR_ASSERT(instance);
  instance->PreConnect = wf_pre_connect;
  instance->PostConnect = wf_post_connect;
  instance->PostDisconnect = wf_post_disconnect;
  instance->AuthenticateEx = wf_authenticate_ex;
 
#ifdef WITH_WINDOWS_CERT_STORE
  if (!freerdp_settings_set_bool(context->settings, FreeRDP_CertificateCallbackPreferPEM, TRUE))
    return FALSE;
#endif
 
  if (!wfc->isConsole)
  {
    instance->VerifyCertificateEx = wf_verify_certificate_ex;
    instance->VerifyChangedCertificateEx = wf_verify_changed_certificate_ex;
    instance->PresentGatewayMessage = wf_present_gateway_message;
  }
 
#ifdef WITH_PROGRESS_BAR
  CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
  CoCreateInstance(&CLSID_TaskbarList, NULL, CLSCTX_ALL, &IID_ITaskbarList3,
                   (void**)&wfc->taskBarList);
#endif
 
  return TRUE;
}
 
static void wfreerdp_client_free(freerdp* instance, rdpContext* context)
{
  WINPR_UNUSED(instance);
  if (!context)
    return;
 
#ifdef WITH_PROGRESS_BAR
  CoUninitialize();
#endif
}
 
static int wfreerdp_client_start(rdpContext* context)
{
  wfContext* wfc = (wfContext*)context;
 
  WINPR_ASSERT(context);
  WINPR_ASSERT(context->settings);
 
  freerdp* instance = context->instance;
  WINPR_ASSERT(instance);
 
  rdpSettings* settings = context->settings;
  WINPR_ASSERT(settings);
 
  HINSTANCE hInstance = GetModuleHandle(NULL);
  HWND hWndParent = (HWND)freerdp_settings_get_uint64(settings, FreeRDP_ParentWindowId);
  if (!freerdp_settings_set_bool(settings, FreeRDP_EmbeddedWindow, (hWndParent) ? TRUE : FALSE))
    return -1;
 
  wfc->hWndParent = hWndParent;
 
  if (freerdp_settings_get_bool(settings, FreeRDP_EmbeddedWindow))
  {
    typedef UINT(WINAPI * GetDpiForWindow_t)(HWND hwnd);
    typedef BOOL(WINAPI * SetProcessDPIAware_t)(void);
 
    HMODULE module = GetModuleHandle(_T("User32"));
    if (module)
    {
      GetDpiForWindow_t pGetDpiForWindow =
          GetProcAddressAs(module, "GetDpiForWindow", GetDpiForWindow_t);
      SetProcessDPIAware_t pSetProcessDPIAware =
          GetProcAddressAs(module, "SetProcessDPIAware", SetProcessDPIAware_t);
      if (pGetDpiForWindow && pSetProcessDPIAware)
      {
        const UINT dpiAwareness = pGetDpiForWindow(hWndParent);
        if (dpiAwareness != USER_DEFAULT_SCREEN_DPI)
          pSetProcessDPIAware();
      }
      FreeLibrary(module);
    }
  }
 
  /* initial windows system item position where we will insert new menu item
   * after default 5 items (restore, move, size, minimize, maximize)
   * gets incremented each time wf_append_item_to_system_menu is called
   * or maybe could use GetMenuItemCount() to get initial item count ? */
  wfc->systemMenuInsertPosition = 6;
 
  wfc->hInstance = hInstance;
  wfc->cursor = LoadCursor(NULL, IDC_ARROW);
  wfc->icon = LoadIcon(GetModuleHandle(NULL), MAKEINTRESOURCE(IDI_ICON1));
  wfc->wndClassName = _tcsdup(_T("FreeRDP"));
  wfc->wndClass.cbSize = sizeof(WNDCLASSEX);
  wfc->wndClass.style = CS_HREDRAW | CS_VREDRAW;
  wfc->wndClass.lpfnWndProc = wf_event_proc;
  wfc->wndClass.cbClsExtra = 0;
  wfc->wndClass.cbWndExtra = 0;
  wfc->wndClass.hCursor = NULL;
  wfc->wndClass.hbrBackground = (HBRUSH)GetStockObject(BLACK_BRUSH);
  wfc->wndClass.lpszMenuName = NULL;
  wfc->wndClass.lpszClassName = wfc->wndClassName;
  wfc->wndClass.hInstance = hInstance;
  wfc->wndClass.hIcon = wfc->icon;
  wfc->wndClass.hIconSm = wfc->icon;
  RegisterClassEx(&(wfc->wndClass));
  wfc->keyboardThread =
      CreateThread(NULL, 0, wf_keyboard_thread, (void*)wfc, 0, &wfc->keyboardThreadId);
 
  if (!wfc->keyboardThread)
    return -1;
 
  wfc->common.thread =
      CreateThread(NULL, 0, wf_client_thread, (void*)instance, 0, &wfc->mainThreadId);
 
  if (!wfc->common.thread)
    return -1;
 
  return 0;
}
 
static int wfreerdp_client_stop(rdpContext* context)
{
  int rc;
  wfContext* wfc = (wfContext*)context;
 
  WINPR_ASSERT(wfc);
  PostThreadMessage(wfc->mainThreadId, WM_QUIT, 0, 0);
  rc = freerdp_client_common_stop(context);
  wfc->mainThreadId = 0;
 
  if (wfc->keyboardThread)
  {
    PostThreadMessage(wfc->keyboardThreadId, WM_QUIT, 0, 0);
    (void)WaitForSingleObject(wfc->keyboardThread, INFINITE);
    (void)CloseHandle(wfc->keyboardThread);
    wfc->keyboardThread = NULL;
    wfc->keyboardThreadId = 0;
  }
 
  return 0;
}
 
int RdpClientEntry(RDP_CLIENT_ENTRY_POINTS* pEntryPoints)
{
  pEntryPoints->Version = 1;
  pEntryPoints->Size = sizeof(RDP_CLIENT_ENTRY_POINTS_V1);
  pEntryPoints->GlobalInit = wfreerdp_client_global_init;
  pEntryPoints->GlobalUninit = wfreerdp_client_global_uninit;
  pEntryPoints->ContextSize = sizeof(wfContext);
  pEntryPoints->ClientNew = wfreerdp_client_new;
  pEntryPoints->ClientFree = wfreerdp_client_free;
  pEntryPoints->ClientStart = wfreerdp_client_start;
  pEntryPoints->ClientStop = wfreerdp_client_stop;
  return 0;
}