xf_reconnect.c

 
#include <freerdp/config.h>
#include <freerdp/client.h>
#include "xf_reconnect.h"
#include "xf_utils.h"
 
static void xf_reconnect_draw(xfContext* xfc)
{
  WINPR_ASSERT(xfc);
  LogDynAndXFlush(xfc->log, xfc->display);
  const int screen = DefaultScreen(xfc->display);
  GC gc = LogDynAndXCreateGC(xfc->log, xfc->display, xfc->reconnectWindow, 0, nullptr);
  if (!gc)
    return;
  LogDynAndXSetForeground(xfc->log, xfc->display, gc, BlackPixel(xfc->display, screen));
  XFontStruct* font = XLoadQueryFont(xfc->display, "10x20");
  if (font)
    XSetFont(xfc->display, gc, font->fid);
  LogDynAndXClearWindow(xfc->log, xfc->display, xfc->reconnectWindow);
  XDrawString(xfc->display, xfc->reconnectWindow, gc, 16, 30, xfc->reconnectMessage,
              (int)strlen(xfc->reconnectMessage));
  const char cancel[] = "Close this window to cancel.";
  XDrawString(xfc->display, xfc->reconnectWindow, gc, 16, 60, cancel, sizeof(cancel) - 1);
  if (font)
    XFreeFont(xfc->display, font);
  LogDynAndXFreeGC(xfc->log, xfc->display, gc);
  LogDynAndXFlush(xfc->log, xfc->display);
}
 
SSIZE_T xf_retry_dialog(freerdp* instance, const char* what, size_t current, void* userarg)
{
  WINPR_ASSERT(instance);
  const SSIZE_T delay = client_common_retry_dialog(instance, what, current, userarg);
  xfContext* xfc = (xfContext*)instance->context;
  WINPR_ASSERT(xfc);
  if ((delay < 0) || !xfc->display || strcmp(what, "connection") != 0)
    return delay;
 
  xf_lock_x11(xfc);
  if (!xfc->reconnectWindow)
  {
    const int screen = DefaultScreen(xfc->display);
    xfc->reconnectWindow =
        XCreateSimpleWindow(xfc->display, DefaultRootWindow(xfc->display), 0, 0, 420, 90, 1,
                            BlackPixel(xfc->display, screen), WhitePixel(xfc->display, screen));
    XStoreName(xfc->display, xfc->reconnectWindow, "FreeRDP - Reconnecting");
    LogDynAndXSelectInput(xfc->log, xfc->display, xfc->reconnectWindow, ExposureMask);
    XSetWMProtocols(xfc->display, xfc->reconnectWindow, &xfc->WM_DELETE_WINDOW, 1);
    XMapRaised(xfc->display, xfc->reconnectWindow);
  }
  const UINT32 max =
      freerdp_settings_get_uint32(instance->context->settings, FreeRDP_AutoReconnectMaxRetries);
  if (max)
    (void)snprintf(xfc->reconnectMessage, sizeof(xfc->reconnectMessage),
                   "Reconnecting: attempt %zu of %" PRIu32, current + 1, max);
  else
    (void)snprintf(xfc->reconnectMessage, sizeof(xfc->reconnectMessage),
                   "Reconnecting: attempt %zu", current + 1);
  xf_reconnect_draw(xfc);
  xf_unlock_x11(xfc);
  return delay;
}
 
BOOL xf_reconnect_event(xfContext* xfc, const XEvent* event)
{
  WINPR_ASSERT(xfc);
  WINPR_ASSERT(event);
 
  if (!xfc->reconnectWindow || (event->xany.window != xfc->reconnectWindow))
    return TRUE;
 
  if (event->type == Expose)
    xf_reconnect_draw(xfc);
  else if (event->type == ClientMessage && event->xclient.format == 32 &&
           event->xclient.message_type == xfc->WM_PROTOCOLS &&
           (Atom)event->xclient.data.l[0] == xfc->WM_DELETE_WINDOW)
  {
    return FALSE;
  }
  return TRUE;
}
 
void xf_reconnect_close(xfContext* xfc)
{
  WINPR_ASSERT(xfc);
  if (xfc->display && xfc->reconnectWindow)
  {
    xf_lock_x11(xfc);
    LogDynAndXDestroyWindow(xfc->log, xfc->display, xfc->reconnectWindow);
    xfc->reconnectWindow = 0;
    LogDynAndXFlush(xfc->log, xfc->display);
    xf_unlock_x11(xfc);
  }
}