mirrors/FreeRDP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

server: proxy: use peer load balance info when host is fetched from config

Browse Source
This commit is contained in:
Kobi Mizrachi 2020-05-12 13:30:20 +03:00 committed by akallabeth
parent c6e37dce79
commit d28ad95fbe
1 changed files with 69 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
server/proxy/pf_client.c
View File

@ -72,10 +72,11 @@ static void pf_client_on_error_info(void* ctx, ErrorInfoEventArgs* e)
freerdp_send_error_info(ps->context.rdp); freerdp_send_error_info(ps->context.rdp);
} }
static BOOL pf_client_load_rdpsnd(pClientContext* pc, proxyConfig* config) static BOOL pf_client_load_rdpsnd(pClientContext* pc)
{ {
rdpContext* context = (rdpContext*)pc; rdpContext* context = (rdpContext*)pc;
pServerContext* ps = pc->pdata->ps; pServerContext* ps = pc->pdata->ps;
proxyConfig* config = pc->pdata->config;
/* /*
* if AudioOutput is enabled in proxy and client connected with rdpsnd, use proxy as rdpsnd * if AudioOutput is enabled in proxy and client connected with rdpsnd, use proxy as rdpsnd
@ -98,6 +99,63 @@ static BOOL pf_client_load_rdpsnd(pClientContext* pc, proxyConfig* config)
return TRUE; return TRUE;
} }
static BOOL pf_client_passthrough_channels_init(pClientContext* pc)
{
pServerContext* ps = pc->pdata->ps;
rdpSettings* settings = pc->context.settings;
proxyConfig* config = pc->pdata->config;
size_t i;
if (settings->ChannelCount + config->PassthroughCount >= settings->ChannelDefArraySize)
{
LOG_ERR(TAG, pc, "too many channels");
return FALSE;
}
for (i = 0; i < config->PassthroughCount; i++)
{
const char* channel_name = config->Passthrough[i];
CHANNEL_DEF channel = { 0 };
/* only connect connect this channel if already joined in peer connection */
if (!WTSVirtualChannelManagerIsChannelJoined(ps->vcm, channel_name))
{
LOG_INFO(TAG, ps, "client did not connected with channel %s, skipping passthrough",
channel_name);
continue;
}
channel.options = CHANNEL_OPTION_INITIALIZED; /* TODO: Export to config. */
strncpy(channel.name, channel_name, CHANNEL_NAME_LEN);
settings->ChannelDefArray[settings->ChannelCount++] = channel;
}
return TRUE;
}
static BOOL pf_client_use_peer_load_balance_info(pClientContext* pc)
{
pServerContext* ps = pc->pdata->ps;
rdpSettings* settings = pc->context.settings;
DWORD lb_info_len;
const char* lb_info = freerdp_nego_get_routing_token(&ps->context, &lb_info_len);
if (!lb_info)
return TRUE;
free(settings->LoadBalanceInfo);
settings->LoadBalanceInfoLength = lb_info_len;
settings->LoadBalanceInfo = malloc(settings->LoadBalanceInfoLength);
if (!settings->LoadBalanceInfo)
return FALSE;
CopyMemory(settings->LoadBalanceInfo, lb_info, settings->LoadBalanceInfoLength);
return TRUE;
}
/** /**
* Called before a connection is established. * Called before a connection is established.
* *
@ -135,6 +193,7 @@ static BOOL pf_client_pre_connect(freerdp* instance)
settings->DynamicResolutionUpdate = config->DisplayControl; settings->DynamicResolutionUpdate = config->DisplayControl;
settings->AutoReconnectionEnabled = TRUE; settings->AutoReconnectionEnabled = TRUE;
/** /**
* Register the channel listeners. * Register the channel listeners.
* They are required to set up / tear down channels if they are loaded. * They are required to set up / tear down channels if they are loaded.
@ -144,45 +203,26 @@ static BOOL pf_client_pre_connect(freerdp* instance)
PubSub_SubscribeChannelDisconnected(instance->context->pubSub, PubSub_SubscribeChannelDisconnected(instance->context->pubSub,
pf_channels_on_client_channel_disconnect); pf_channels_on_client_channel_disconnect);
PubSub_SubscribeErrorInfo(instance->context->pubSub, pf_client_on_error_info); PubSub_SubscribeErrorInfo(instance->context->pubSub, pf_client_on_error_info);
/** /**
* Load all required plugins / channels / libraries specified by current * Load all required plugins / channels / libraries specified by current
* settings. * settings.
*/ */
LOG_INFO(TAG, pc, "Loading addins"); LOG_INFO(TAG, pc, "Loading addins");
if (!config->UseLoadBalanceInfo)
{ {
/* add passthrough channels to channel def array */ /* if target is static (i.e fetched from config). make sure to use peer's load-balance info,
size_t i; * in order to support broker redirection.
*/
if (settings->ChannelCount + config->PassthroughCount >= settings->ChannelDefArraySize) if (!pf_client_use_peer_load_balance_info(pc))
{
LOG_ERR(TAG, pc, "too many channels");
return FALSE; return FALSE;
} }
for (i = 0; i < config->PassthroughCount; i++) if (!pf_client_passthrough_channels_init(pc))
{ return FALSE;
const char* channel_name = config->Passthrough[i];
CHANNEL_DEF channel = { 0 };
/* only connect connect this channel if already joined in peer connection */ if (!pf_client_load_rdpsnd(pc))
if (!WTSVirtualChannelManagerIsChannelJoined(ps->vcm, channel_name))
{
LOG_INFO(TAG, ps, "client did not connected with channel %s, skipping passthrough",
channel_name);
continue;
}
channel.options = CHANNEL_OPTION_INITIALIZED; /* TODO: Export to config. */
strncpy(channel.name, channel_name, CHANNEL_NAME_LEN);
settings->ChannelDefArray[settings->ChannelCount++] = channel;
}
}
if (!pf_client_load_rdpsnd(pc, config))
{ {
LOG_ERR(TAG, pc, "Failed to load rdpsnd client"); LOG_ERR(TAG, pc, "Failed to load rdpsnd client");
return FALSE; return FALSE;