304 lines
7.3 KiB
C
304 lines
7.3 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.
|
||
|
*/
|
||
|
|
||
|
#ifdef HAVE_CONFIG_H
|
||
|
#include "config.h"
|
||
|
#endif
|
||
|
|
||
|
#include <freerdp/gdi/gfx.h>
|
||
|
|
||
|
#include <freerdp/client/rdpei.h>
|
||
|
#include <freerdp/client/rail.h>
|
||
|
#include <freerdp/client/cliprdr.h>
|
||
|
#include <freerdp/client/rdpgfx.h>
|
||
|
#include <freerdp/client/disp.h>
|
||
|
|
||
|
#include "pf_channels.h"
|
||
|
#include "pf_client.h"
|
||
|
#include "pf_context.h"
|
||
|
#include "pf_rail.h"
|
||
|
#include "pf_rdpgfx.h"
|
||
|
#include "pf_cliprdr.h"
|
||
|
#include "pf_disp.h"
|
||
|
#include "pf_log.h"
|
||
|
#include "pf_modules.h"
|
||
|
#include "pf_rdpsnd.h"
|
||
|
|
||
|
#define TAG PROXY_TAG("channels")
|
||
|
|
||
|
static void pf_channels_wait_for_server_dynvc(pServerContext* ps)
|
||
|
{
|
||
|
WLog_DBG(TAG, "pf_channels_wait_for_server_dynvc(): waiting for server's drdynvc to be ready");
|
||
|
WaitForSingleObject(ps->dynvcReady, INFINITE);
|
||
|
WLog_DBG(TAG, "pf_channels_wait_for_server_dynvc(): server's drdynvc is ready!");
|
||
|
}
|
||
|
|
||
|
void pf_channels_on_client_channel_connect(void* data, ChannelConnectedEventArgs* e)
|
||
|
{
|
||
|
pClientContext* pc = (pClientContext*)data;
|
||
|
pServerContext* ps = pc->pdata->ps;
|
||
|
LOG_INFO(TAG, pc, "Channel connected: %s", e->name);
|
||
|
|
||
|
if (strcmp(e->name, RDPEI_DVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
pc->rdpei = (RdpeiClientContext*)e->pInterface;
|
||
|
}
|
||
|
else if (strcmp(e->name, RAIL_SVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
pc->rail = (RailClientContext*)e->pInterface;
|
||
|
|
||
|
if (ps->rail->Start(ps->rail) != CHANNEL_RC_OK)
|
||
|
{
|
||
|
WLog_ERR(TAG, "failed to start RAIL server");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
pf_rail_pipeline_init(pc->rail, ps->rail, pc->pdata);
|
||
|
}
|
||
|
else if (strcmp(e->name, RDPGFX_DVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
pf_channels_wait_for_server_dynvc(ps);
|
||
|
pc->gfx_proxy = (RdpgfxClientContext*)e->pInterface;
|
||
|
pf_rdpgfx_pipeline_init(pc->gfx_proxy, ps->gfx, pc->pdata);
|
||
|
|
||
|
if (!ps->gfx->Open(ps->gfx))
|
||
|
{
|
||
|
WLog_ERR(TAG, "failed to open GFX server");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
SetEvent(pc->pdata->gfx_server_ready);
|
||
|
}
|
||
|
else if (strcmp(e->name, DISP_DVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
UINT ret;
|
||
|
|
||
|
ret = ps->disp->Open(ps->disp);
|
||
|
if (ret != CHANNEL_RC_OK)
|
||
|
{
|
||
|
if (ret == ERROR_NOT_FOUND)
|
||
|
{
|
||
|
/* client did not connect with disp */
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
pf_channels_wait_for_server_dynvc(ps);
|
||
|
if (ps->disp->Open(ps->disp) != CHANNEL_RC_OK)
|
||
|
{
|
||
|
WLog_ERR(TAG, "failed to open disp channel");
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pc->disp = (DispClientContext*)e->pInterface;
|
||
|
pf_disp_register_callbacks(pc->disp, ps->disp, pc->pdata);
|
||
|
}
|
||
|
else if (strcmp(e->name, CLIPRDR_SVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
if (ps->cliprdr->Start(ps->cliprdr) != CHANNEL_RC_OK)
|
||
|
{
|
||
|
WLog_ERR(TAG, "failed to open cliprdr channel");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
pc->cliprdr = (CliprdrClientContext*)e->pInterface;
|
||
|
pf_cliprdr_register_callbacks(pc->cliprdr, ps->cliprdr, pc->pdata);
|
||
|
}
|
||
|
else if (strcmp(e->name, "rdpsnd") == 0)
|
||
|
{
|
||
|
/* sound is disabled */
|
||
|
if (ps->rdpsnd == NULL)
|
||
|
return;
|
||
|
|
||
|
if (ps->rdpsnd->Initialize(ps->rdpsnd, TRUE) != CHANNEL_RC_OK)
|
||
|
{
|
||
|
WLog_ERR(TAG, "failed to open rdpsnd channel");
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void pf_channels_on_client_channel_disconnect(void* data, ChannelDisconnectedEventArgs* e)
|
||
|
{
|
||
|
rdpContext* context = (rdpContext*)data;
|
||
|
pClientContext* pc = (pClientContext*)context;
|
||
|
pServerContext* ps = pc->pdata->ps;
|
||
|
LOG_INFO(TAG, pc, "Channel disconnected: %s", e->name);
|
||
|
|
||
|
if (strcmp(e->name, RDPEI_DVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
pc->rdpei = NULL;
|
||
|
}
|
||
|
else if (strcmp(e->name, RDPGFX_DVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
if (!ps->gfx->Close(ps->gfx))
|
||
|
WLog_ERR(TAG, "failed to close gfx server");
|
||
|
|
||
|
gdi_graphics_pipeline_uninit(context->gdi, pc->gfx_decoder);
|
||
|
rdpgfx_client_context_free(pc->gfx_decoder);
|
||
|
}
|
||
|
else if (strcmp(e->name, RAIL_SVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
if (!ps->rail->Stop(ps->rail))
|
||
|
WLog_ERR(TAG, "failed to close rail server");
|
||
|
|
||
|
pc->rail = NULL;
|
||
|
}
|
||
|
else if (strcmp(e->name, DISP_DVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
if (ps->disp->Close(ps->disp) != CHANNEL_RC_OK)
|
||
|
WLog_ERR(TAG, "failed to close disp server");
|
||
|
|
||
|
pc->disp = NULL;
|
||
|
}
|
||
|
else if (strcmp(e->name, CLIPRDR_SVC_CHANNEL_NAME) == 0)
|
||
|
{
|
||
|
if (ps->cliprdr->Stop(ps->cliprdr) != CHANNEL_RC_OK)
|
||
|
WLog_ERR(TAG, "failed to stop cliprdr server");
|
||
|
|
||
|
pc->cliprdr = NULL;
|
||
|
}
|
||
|
else if (strcmp(e->name, "rdpsnd") == 0)
|
||
|
{
|
||
|
/* sound is disabled */
|
||
|
if (ps->rdpsnd == NULL)
|
||
|
return;
|
||
|
|
||
|
if (ps->rdpsnd->Stop(ps->rdpsnd) != CHANNEL_RC_OK)
|
||
|
WLog_ERR(TAG, "failed to close rdpsnd server");
|
||
|
}
|
||
|
}
|
||
|
|
||
|
BOOL pf_server_channels_init(pServerContext* ps)
|
||
|
{
|
||
|
rdpContext* context = (rdpContext*)ps;
|
||
|
rdpContext* client = (rdpContext*)ps->pdata->pc;
|
||
|
proxyConfig* config = ps->pdata->config;
|
||
|
|
||
|
if (context->settings->SupportGraphicsPipeline && config->GFX)
|
||
|
{
|
||
|
if (!pf_server_rdpgfx_init(ps))
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (config->DisplayControl)
|
||
|
{
|
||
|
if (!pf_server_disp_init(ps))
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (config->Clipboard &&
|
||
|
WTSVirtualChannelManagerIsChannelJoined(ps->vcm, CLIPRDR_SVC_CHANNEL_NAME))
|
||
|
{
|
||
|
client->settings->RedirectClipboard = TRUE;
|
||
|
|
||
|
if (!pf_server_cliprdr_init(ps))
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (config->AudioOutput && WTSVirtualChannelManagerIsChannelJoined(ps->vcm, "rdpsnd"))
|
||
|
{
|
||
|
if (!pf_server_rdpsnd_init(ps))
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
if (config->RemoteApp &&
|
||
|
WTSVirtualChannelManagerIsChannelJoined(ps->vcm, RAIL_SVC_CHANNEL_NAME))
|
||
|
{
|
||
|
if (!pf_rail_context_init(ps))
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
{
|
||
|
/* open static channels for passthrough */
|
||
|
size_t i;
|
||
|
|
||
|
for (i = 0; i < config->PassthroughCount; i++)
|
||
|
{
|
||
|
char* channel_name = config->Passthrough[i];
|
||
|
UINT64 channel_id;
|
||
|
|
||
|
/* only open channel if client joined with it */
|
||
|
if (!WTSVirtualChannelManagerIsChannelJoined(ps->vcm, channel_name))
|
||
|
continue;
|
||
|
|
||
|
ps->vc_handles[i] = WTSVirtualChannelOpen(ps->vcm, WTS_CURRENT_SESSION, channel_name);
|
||
|
if (!ps->vc_handles[i])
|
||
|
{
|
||
|
LOG_ERR(TAG, ps, "WTSVirtualChannelOpen failed for passthrough channel: %s",
|
||
|
channel_name);
|
||
|
|
||
|
return FALSE;
|
||
|
}
|
||
|
|
||
|
channel_id = (UINT64)WTSChannelGetId(ps->context.peer, channel_name);
|
||
|
HashTable_Add(ps->vc_ids, channel_name, (void*)channel_id);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return pf_modules_run_hook(HOOK_TYPE_SERVER_CHANNELS_INIT, ps->pdata);
|
||
|
}
|
||
|
|
||
|
void pf_server_channels_free(pServerContext* ps)
|
||
|
{
|
||
|
if (ps->gfx)
|
||
|
{
|
||
|
rdpgfx_server_context_free(ps->gfx);
|
||
|
ps->gfx = NULL;
|
||
|
}
|
||
|
|
||
|
if (ps->disp)
|
||
|
{
|
||
|
disp_server_context_free(ps->disp);
|
||
|
ps->disp = NULL;
|
||
|
}
|
||
|
|
||
|
if (ps->cliprdr)
|
||
|
{
|
||
|
cliprdr_server_context_free(ps->cliprdr);
|
||
|
ps->cliprdr = NULL;
|
||
|
}
|
||
|
|
||
|
if (ps->rdpsnd)
|
||
|
{
|
||
|
rdpsnd_server_context_free(ps->rdpsnd);
|
||
|
ps->rdpsnd = NULL;
|
||
|
}
|
||
|
|
||
|
if (ps->rail)
|
||
|
{
|
||
|
rail_server_context_free(ps->rail);
|
||
|
ps->rail = NULL;
|
||
|
}
|
||
|
|
||
|
{
|
||
|
/* close passthrough channels */
|
||
|
size_t i;
|
||
|
|
||
|
for (i = 0; i < ps->pdata->config->PassthroughCount; i++)
|
||
|
WTSVirtualChannelClose(ps->vc_handles[i]);
|
||
|
}
|
||
|
|
||
|
pf_modules_run_hook(HOOK_TYPE_SERVER_CHANNELS_FREE, ps->pdata);
|
||
|
}
|