pf_input.c

 
#include <winpr/assert.h>
 
#include "pf_input.h"
#include "pf_client.h"
#include "pf_server.h"
#include <freerdp/server/proxy/proxy_config.h>
 
#include "proxy_modules.h"
 
WINPR_ATTR_NODISCARD
static BOOL pf_server_check_and_sync_input_state(pClientContext* pc)
{
  WINPR_ASSERT(pc);
 
  if (!freerdp_is_active_state(&pc->cctx.context))
    return FALSE;
  if (pc->input_state_sync_pending)
  {
    BOOL rc = freerdp_input_send_synchronize_event(pc->cctx.context.input, pc->input_state);
    if (rc)
      pc->input_state_sync_pending = FALSE;
  }
  return TRUE;
}
 
WINPR_ATTR_NODISCARD
static BOOL pf_server_synchronize_event(rdpInput* input, UINT32 flags)
{
  WINPR_ASSERT(input);
  pServerContext* ps = (pServerContext*)input->context;
  WINPR_ASSERT(ps);
  WINPR_ASSERT(ps->pdata);
 
  pClientContext* pc = proxy_data_get_client_context(ps->pdata);
  if (!pc)
    return TRUE;
 
  pc->input_state = flags;
  pc->input_state_sync_pending = TRUE;
 
  if (!pf_server_check_and_sync_input_state(pc))
    return TRUE;
  return TRUE;
}
 
WINPR_ATTR_NODISCARD
static BOOL pf_server_keyboard_event(rdpInput* input, UINT16 flags, UINT8 code)
{
  proxyKeyboardEventInfo event = WINPR_C_ARRAY_INIT;
 
  WINPR_ASSERT(input);
  pServerContext* ps = (pServerContext*)input->context;
  WINPR_ASSERT(ps);
  WINPR_ASSERT(ps->pdata);
 
  pClientContext* pc = proxy_data_get_client_context(ps->pdata);
  if (!pc)
    return TRUE;
 
  const proxyConfig* config = ps->pdata->config;
  WINPR_ASSERT(config);
 
  if (!pf_server_check_and_sync_input_state(pc))
    return TRUE;
 
  if (!config->Keyboard)
    return TRUE;
 
  event.flags = flags;
  event.rdp_scan_code = code;
 
  if (pf_modules_run_filter(pc->pdata->module, FILTER_TYPE_KEYBOARD, pc->pdata, &event))
    return freerdp_input_send_keyboard_event(pc->cctx.context.input, flags, code);
 
  return TRUE;
}
 
WINPR_ATTR_NODISCARD
static BOOL pf_server_unicode_keyboard_event(rdpInput* input, UINT16 flags, UINT16 code)
{
  proxyUnicodeEventInfo event = WINPR_C_ARRAY_INIT;
 
  WINPR_ASSERT(input);
  pServerContext* ps = (pServerContext*)input->context;
  WINPR_ASSERT(ps);
  WINPR_ASSERT(ps->pdata);
 
  pClientContext* pc = proxy_data_get_client_context(ps->pdata);
  if (!pc)
    return TRUE;
 
  const proxyConfig* config = ps->pdata->config;
  WINPR_ASSERT(config);
 
  if (!pf_server_check_and_sync_input_state(pc))
    return TRUE;
 
  if (!config->Keyboard)
    return TRUE;
 
  event.flags = flags;
  event.code = code;
  if (pf_modules_run_filter(pc->pdata->module, FILTER_TYPE_UNICODE, pc->pdata, &event))
    return freerdp_input_send_unicode_keyboard_event(pc->cctx.context.input, flags, code);
  return TRUE;
}
 
WINPR_ATTR_NODISCARD
static BOOL pf_server_mouse_event(rdpInput* input, UINT16 flags, UINT16 x, UINT16 y)
{
  proxyMouseEventInfo event = WINPR_C_ARRAY_INIT;
 
  WINPR_ASSERT(input);
  pServerContext* ps = (pServerContext*)input->context;
  WINPR_ASSERT(ps);
  WINPR_ASSERT(ps->pdata);
 
  pClientContext* pc = proxy_data_get_client_context(ps->pdata);
  if (!pc)
    return TRUE;
 
  const proxyConfig* config = ps->pdata->config;
  WINPR_ASSERT(config);
 
  if (!pf_server_check_and_sync_input_state(pc))
    return TRUE;
 
  if (!config->Mouse)
    return TRUE;
 
  event.flags = flags;
  event.x = x;
  event.y = y;
 
  if (pf_modules_run_filter(pc->pdata->module, FILTER_TYPE_MOUSE, pc->pdata, &event))
    return freerdp_input_send_mouse_event(pc->cctx.context.input, flags, x, y);
 
  return TRUE;
}
 
WINPR_ATTR_NODISCARD
static BOOL pf_server_extended_mouse_event(rdpInput* input, UINT16 flags, UINT16 x, UINT16 y)
{
  proxyMouseExEventInfo event = WINPR_C_ARRAY_INIT;
 
  WINPR_ASSERT(input);
 
  pServerContext* ps = (pServerContext*)input->context;
  WINPR_ASSERT(ps);
  WINPR_ASSERT(ps->pdata);
 
  pClientContext* pc = proxy_data_get_client_context(ps->pdata);
  if (!pc)
    return TRUE;
 
  const proxyConfig* config = ps->pdata->config;
  WINPR_ASSERT(config);
 
  if (!pf_server_check_and_sync_input_state(pc))
    return TRUE;
 
  if (!config->Mouse)
    return TRUE;
 
  event.flags = flags;
  event.x = x;
  event.y = y;
  if (pf_modules_run_filter(pc->pdata->module, FILTER_TYPE_MOUSE_EX, pc->pdata, &event))
    return freerdp_input_send_extended_mouse_event(pc->cctx.context.input, flags, x, y);
  return TRUE;
}
 
void pf_server_register_input_callbacks(rdpInput* input)
{
  WINPR_ASSERT(input);
 
  input->SynchronizeEvent = pf_server_synchronize_event;
  input->KeyboardEvent = pf_server_keyboard_event;
  input->UnicodeKeyboardEvent = pf_server_unicode_keyboard_event;
  input->MouseEvent = pf_server_mouse_event;
  input->ExtendedMouseEvent = pf_server_extended_mouse_event;
}