346 lines
13 KiB
C
346 lines
13 KiB
C
|
/**
|
||
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
||
|
|
* FreeRDP Proxy Server
|
||
|
|
*
|
||
|
|
* Copyright 2019 Mati Shabtay <matishabtay@gmail.com>
|
||
|
|
* Copyright 2019 Kobi Mizrachi <kmizrachi18@gmail.com>
|
||
|
|
* Copyright 2019 Idan Freiberg <speidy@gmail.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/client/rail.h>
|
||
|
|
#include <freerdp/server/rail.h>
|
||
|
|
|
||
|
|
#include "pf_rail.h"
|
||
|
|
#include "pf_context.h"
|
||
|
|
#include "pf_log.h"
|
||
|
|
|
||
|
|
#define TAG PROXY_TAG("rail")
|
||
|
|
|
||
|
|
BOOL pf_rail_context_init(pServerContext* ps)
|
||
|
|
{
|
||
|
|
RailServerContext* rail;
|
||
|
|
rail = ps->rail = rail_server_context_new(ps->vcm);
|
||
|
|
|
||
|
|
if (!rail)
|
||
|
|
{
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* when mstsc reconnects, it doesn't wait for a second handshake, so update all handshake flags
|
||
|
|
* to be SET, then set them again when the remote server sends his handshake.
|
||
|
|
*/
|
||
|
|
rail_server_set_handshake_ex_flags(rail,
|
||
|
|
TS_RAIL_ORDER_HANDSHAKEEX_FLAGS_HIDEF |
|
||
|
|
TS_RAIL_ORDER_HANDSHAKE_EX_FLAGS_EXTENDED_SPI_SUPPORTED |
|
||
|
|
TS_RAIL_ORDER_HANDSHAKE_EX_FLAGS_SNAP_ARRANGE_SUPPORTED);
|
||
|
|
|
||
|
|
rail->rdpcontext = (rdpContext*)ps;
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_on_open(RailClientContext* context, BOOL* sendHandshake)
|
||
|
|
{
|
||
|
|
if (NULL != sendHandshake)
|
||
|
|
*sendHandshake = FALSE;
|
||
|
|
|
||
|
|
return CHANNEL_RC_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Callbacks from client side */
|
||
|
|
static UINT pf_rail_server_handshake(RailClientContext* client,
|
||
|
|
const RAIL_HANDSHAKE_ORDER* handshake)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerHandshake(server, handshake);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_handshake_ex(RailClientContext* client,
|
||
|
|
const RAIL_HANDSHAKE_EX_ORDER* handshakeEx)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerHandshakeEx(server, handshakeEx);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_sysparam(RailClientContext* client, const RAIL_SYSPARAM_ORDER* sysparam)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerSysparam(server, sysparam);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_local_move_size(RailClientContext* client,
|
||
|
|
const RAIL_LOCALMOVESIZE_ORDER* localMoveSize)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerLocalMoveSize(server, localMoveSize);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_min_max_info(RailClientContext* client,
|
||
|
|
const RAIL_MINMAXINFO_ORDER* minMaxInfo)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerMinMaxInfo(server, minMaxInfo);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_taskbar_info(RailClientContext* client,
|
||
|
|
const RAIL_TASKBAR_INFO_ORDER* taskbarInfo)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerTaskbarInfo(server, taskbarInfo);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_langbar_info(RailClientContext* client,
|
||
|
|
const RAIL_LANGBAR_INFO_ORDER* langbarInfo)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerLangbarInfo(server, langbarInfo);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_exec_result(RailClientContext* client,
|
||
|
|
const RAIL_EXEC_RESULT_ORDER* execResult)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerExecResult(server, execResult);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_z_order_sync(RailClientContext* client,
|
||
|
|
const RAIL_ZORDER_SYNC* zOrderSync)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerZOrderSync(server, zOrderSync);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_cloak(RailClientContext* client, const RAIL_CLOAK* cloak)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerCloak(server, cloak);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT
|
||
|
|
pf_rail_server_power_display_request(RailClientContext* client,
|
||
|
|
const RAIL_POWER_DISPLAY_REQUEST* powerDisplayRequest)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerPowerDisplayRequest(server, powerDisplayRequest);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_get_appid_resp(RailClientContext* client,
|
||
|
|
const RAIL_GET_APPID_RESP_ORDER* getAppidResp)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerGetAppidResp(server, getAppidResp);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_server_get_appid_resp_ex(RailClientContext* client,
|
||
|
|
const RAIL_GET_APPID_RESP_EX* getAppidRespEx)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)client->custom;
|
||
|
|
RailServerContext* server = (RailServerContext*)pdata->ps->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return server->ServerGetAppidRespEx(server, getAppidRespEx);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Callbacks from server side */
|
||
|
|
|
||
|
|
static UINT pf_rail_client_handshake(RailServerContext* server,
|
||
|
|
const RAIL_HANDSHAKE_ORDER* handshake)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientHandshake(client, handshake);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_client_status(RailServerContext* server,
|
||
|
|
const RAIL_CLIENT_STATUS_ORDER* clientStatus)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientInformation(client, clientStatus);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_exec(RailServerContext* server, const RAIL_EXEC_ORDER* exec)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientExecute(client, exec);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_sysparam(RailServerContext* server, const RAIL_SYSPARAM_ORDER* sysparam)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientSystemParam(client, sysparam);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_activate(RailServerContext* server, const RAIL_ACTIVATE_ORDER* activate)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientActivate(client, activate);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_sysmenu(RailServerContext* server, const RAIL_SYSMENU_ORDER* sysmenu)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientSystemMenu(client, sysmenu);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_syscommand(RailServerContext* server,
|
||
|
|
const RAIL_SYSCOMMAND_ORDER* syscommand)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientSystemCommand(client, syscommand);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_notify_event(RailServerContext* server,
|
||
|
|
const RAIL_NOTIFY_EVENT_ORDER* notifyEvent)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientNotifyEvent(client, notifyEvent);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_window_move(RailServerContext* server,
|
||
|
|
const RAIL_WINDOW_MOVE_ORDER* windowMove)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientWindowMove(client, windowMove);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_snap_arrange(RailServerContext* server,
|
||
|
|
const RAIL_SNAP_ARRANGE* snapArrange)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientSnapArrange(client, snapArrange);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_get_appid_req(RailServerContext* server,
|
||
|
|
const RAIL_GET_APPID_REQ_ORDER* getAppidReq)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientGetAppIdRequest(client, getAppidReq);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_langbar_info(RailServerContext* server,
|
||
|
|
const RAIL_LANGBAR_INFO_ORDER* langbarInfo)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientLanguageBarInfo(client, langbarInfo);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_language_ime_info(RailServerContext* server,
|
||
|
|
const RAIL_LANGUAGEIME_INFO_ORDER* languageImeInfo)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientLanguageIMEInfo(client, languageImeInfo);
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_compartment_info(RailServerContext* server,
|
||
|
|
const RAIL_COMPARTMENT_INFO_ORDER* compartmentInfo)
|
||
|
|
{
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static UINT pf_rail_client_cloak(RailServerContext* server, const RAIL_CLOAK* cloak)
|
||
|
|
{
|
||
|
|
proxyData* pdata = (proxyData*)server->custom;
|
||
|
|
RailClientContext* client = (RailClientContext*)pdata->pc->rail;
|
||
|
|
WLog_DBG(TAG, __FUNCTION__);
|
||
|
|
return client->ClientCloak(client, cloak);
|
||
|
|
}
|
||
|
|
|
||
|
|
void pf_rail_pipeline_init(RailClientContext* client, RailServerContext* server, proxyData* pdata)
|
||
|
|
{
|
||
|
|
/* Set server and client side references to proxy data */
|
||
|
|
client->custom = (void*)pdata;
|
||
|
|
server->custom = (void*)pdata;
|
||
|
|
/* Set client callbacks */
|
||
|
|
client->OnOpen = pf_rail_client_on_open;
|
||
|
|
client->ServerHandshake = pf_rail_server_handshake;
|
||
|
|
client->ServerHandshakeEx = pf_rail_server_handshake_ex;
|
||
|
|
client->ServerSystemParam = pf_rail_server_sysparam;
|
||
|
|
client->ServerLocalMoveSize = pf_rail_server_local_move_size;
|
||
|
|
client->ServerMinMaxInfo = pf_rail_server_min_max_info;
|
||
|
|
client->ServerTaskBarInfo = pf_rail_server_taskbar_info;
|
||
|
|
client->ServerLanguageBarInfo = pf_rail_server_langbar_info;
|
||
|
|
client->ServerExecuteResult = pf_rail_server_exec_result;
|
||
|
|
client->ServerZOrderSync = pf_rail_server_z_order_sync;
|
||
|
|
client->ServerCloak = pf_rail_server_cloak;
|
||
|
|
client->ServerPowerDisplayRequest = pf_rail_server_power_display_request;
|
||
|
|
client->ServerGetAppIdResponse = pf_rail_server_get_appid_resp;
|
||
|
|
client->ServerGetAppidResponseExtended = pf_rail_server_get_appid_resp_ex;
|
||
|
|
/* Set server callbacks */
|
||
|
|
server->ClientHandshake = pf_rail_client_handshake;
|
||
|
|
server->ClientClientStatus = pf_rail_client_client_status;
|
||
|
|
server->ClientExec = pf_rail_client_exec;
|
||
|
|
server->ClientSysparam = pf_rail_client_sysparam;
|
||
|
|
server->ClientActivate = pf_rail_client_activate;
|
||
|
|
server->ClientSysmenu = pf_rail_client_sysmenu;
|
||
|
|
server->ClientSyscommand = pf_rail_client_syscommand;
|
||
|
|
server->ClientNotifyEvent = pf_rail_client_notify_event;
|
||
|
|
server->ClientGetAppidReq = pf_rail_client_get_appid_req;
|
||
|
|
server->ClientWindowMove = pf_rail_client_window_move;
|
||
|
|
server->ClientSnapArrange = pf_rail_client_snap_arrange;
|
||
|
|
server->ClientLangbarInfo = pf_rail_client_langbar_info;
|
||
|
|
server->ClientLanguageImeInfo = pf_rail_client_language_ime_info;
|
||
|
|
server->ClientCompartmentInfo = pf_rail_client_compartment_info;
|
||
|
|
server->ClientCloak = pf_rail_client_cloak;
|
||
|
|
}
|