GUACAMOLE-249: Add common SVC plugin implementation as future simplified basis for RDPSND, RDPDR, etc.
This commit is contained in:
parent
7b93b3d2e9
commit
352b9c517c
@ -41,6 +41,7 @@ libguac_client_rdp_la_SOURCES = \
|
|||||||
channels.c \
|
channels.c \
|
||||||
client.c \
|
client.c \
|
||||||
clipboard.c \
|
clipboard.c \
|
||||||
|
common-svc.c \
|
||||||
decompose.c \
|
decompose.c \
|
||||||
disp.c \
|
disp.c \
|
||||||
error.c \
|
error.c \
|
||||||
@ -83,6 +84,7 @@ noinst_HEADERS = \
|
|||||||
client.h \
|
client.h \
|
||||||
clipboard.h \
|
clipboard.h \
|
||||||
channels.h \
|
channels.h \
|
||||||
|
common-svc.h \
|
||||||
decompose.h \
|
decompose.h \
|
||||||
disp.h \
|
disp.h \
|
||||||
error.h \
|
error.h \
|
||||||
@ -130,14 +132,34 @@ libguac_client_rdp_la_LIBADD = \
|
|||||||
# Plugins for FreeRDP
|
# Plugins for FreeRDP
|
||||||
#
|
#
|
||||||
|
|
||||||
freerdp_LTLIBRARIES = \
|
freerdp_LTLIBRARIES = \
|
||||||
libguacai-client.la \
|
libguac-common-svc-client.la \
|
||||||
libguacdr-client.la \
|
libguacai-client.la \
|
||||||
libguacsnd-client.la \
|
libguacdr-client.la \
|
||||||
|
libguacsnd-client.la \
|
||||||
libguacsvc-client.la
|
libguacsvc-client.la
|
||||||
|
|
||||||
freerdpdir = ${libdir}/freerdp2
|
freerdpdir = ${libdir}/freerdp2
|
||||||
|
|
||||||
|
#
|
||||||
|
# Common SVC plugin (shared by RDPDR, RDPSND, etc.)
|
||||||
|
#
|
||||||
|
|
||||||
|
libguac_common_svc_client_la_SOURCES = \
|
||||||
|
guac-common-svc/common-svc-main.c
|
||||||
|
|
||||||
|
libguac_common_svc_client_la_CFLAGS = \
|
||||||
|
-Werror -Wall -Iinclude \
|
||||||
|
@LIBGUAC_INCLUDE@ \
|
||||||
|
@RDP_CFLAGS@
|
||||||
|
|
||||||
|
libguac_common_svc_client_la_LDFLAGS = \
|
||||||
|
-module -avoid-version -shared \
|
||||||
|
@RDP_LIBS@
|
||||||
|
|
||||||
|
libguac_common_svc_client_la_LIBADD = \
|
||||||
|
@LIBGUAC_LTLIB@
|
||||||
|
|
||||||
#
|
#
|
||||||
# RDPDR
|
# RDPDR
|
||||||
#
|
#
|
||||||
|
103
src/protocols/rdp/common-svc.c
Normal file
103
src/protocols/rdp/common-svc.c
Normal file
@ -0,0 +1,103 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one
|
||||||
|
* or more contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. The ASF licenses this file
|
||||||
|
* to you 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 "config.h"
|
||||||
|
#include "channels.h"
|
||||||
|
#include "common-svc.h"
|
||||||
|
#include "rdp.h"
|
||||||
|
|
||||||
|
#include <freerdp/svc.h>
|
||||||
|
#include <guacamole/client.h>
|
||||||
|
#include <guacamole/stream.h>
|
||||||
|
#include <guacamole/string.h>
|
||||||
|
#include <winpr/stream.h>
|
||||||
|
#include <winpr/wtsapi.h>
|
||||||
|
#include <winpr/wtypes.h>
|
||||||
|
|
||||||
|
int guac_rdp_common_svc_load_plugin(rdpContext* context,
|
||||||
|
char* name, ULONG channel_options,
|
||||||
|
guac_rdp_common_svc_connect_handler* connect_handler,
|
||||||
|
guac_rdp_common_svc_receive_handler* receive_handler,
|
||||||
|
guac_rdp_common_svc_terminate_handler* terminate_handler) {
|
||||||
|
|
||||||
|
guac_client* client = ((rdp_freerdp_context*) context)->client;
|
||||||
|
|
||||||
|
guac_rdp_common_svc* svc = calloc(1, sizeof(guac_rdp_common_svc));
|
||||||
|
svc->client = client;
|
||||||
|
svc->name = svc->_channel_def.name;
|
||||||
|
svc->_connect_handler = connect_handler;
|
||||||
|
svc->_receive_handler = receive_handler;
|
||||||
|
svc->_terminate_handler = terminate_handler;
|
||||||
|
|
||||||
|
/* Init FreeRDP channel definition */
|
||||||
|
int name_length = guac_strlcpy(svc->_channel_def.name, name, GUAC_RDP_SVC_MAX_LENGTH);
|
||||||
|
svc->_channel_def.options =
|
||||||
|
CHANNEL_OPTION_INITIALIZED
|
||||||
|
| CHANNEL_OPTION_ENCRYPT_RDP
|
||||||
|
| channel_options;
|
||||||
|
|
||||||
|
/* Warn about name length */
|
||||||
|
if (name_length >= GUAC_RDP_SVC_MAX_LENGTH)
|
||||||
|
guac_client_log(client, GUAC_LOG_WARNING,
|
||||||
|
"Static channel name \"%s\" exceeds maximum length of %i "
|
||||||
|
"characters and will be truncated to \"%s\".",
|
||||||
|
name, GUAC_RDP_SVC_MAX_LENGTH - 1, svc->name);
|
||||||
|
|
||||||
|
int result = guac_freerdp_channels_load_plugin(context->channels,
|
||||||
|
context->settings, "guac-common-svc", svc);
|
||||||
|
|
||||||
|
/* Attempt to load the common SVC plugin for new static channel */
|
||||||
|
if (result) {
|
||||||
|
guac_client_log(client, GUAC_LOG_WARNING, "Cannot create static "
|
||||||
|
"channel \"%s\": failed to load \"guac-common-svc\" plugin "
|
||||||
|
"for FreeRDP.", svc->name);
|
||||||
|
free(svc);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Store and log on success (SVC structure will be freed on channel termination) */
|
||||||
|
else
|
||||||
|
guac_client_log(client, GUAC_LOG_DEBUG, "Support for static channel "
|
||||||
|
"\"%s\" loaded.", svc->name);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
void guac_rdp_common_svc_write(guac_rdp_common_svc* svc,
|
||||||
|
wStream* output_stream) {
|
||||||
|
|
||||||
|
/* Do not write if plugin not associated */
|
||||||
|
if (!svc->_open_handle) {
|
||||||
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "%i bytes of data "
|
||||||
|
"written to SVC \"%s\" are being dropped because the remote "
|
||||||
|
"desktop side of that SVC is not yet connected.",
|
||||||
|
Stream_Length(output_stream), svc->name);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* NOTE: Data sent via pVirtualChannelWriteEx MUST always be dynamically
|
||||||
|
* allocated, as it will be automatically freed using free(). If provided,
|
||||||
|
* the last parameter (user data) MUST be a pointer to a wStream, as it
|
||||||
|
* will automatically be freed by FreeRDP using Stream_Free() */
|
||||||
|
svc->_entry_points.pVirtualChannelWriteEx(svc->_init_handle,
|
||||||
|
svc->_open_handle, Stream_Buffer(output_stream),
|
||||||
|
Stream_GetPosition(output_stream), output_stream);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
230
src/protocols/rdp/common-svc.h
Normal file
230
src/protocols/rdp/common-svc.h
Normal file
@ -0,0 +1,230 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one
|
||||||
|
* or more contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. The ASF licenses this file
|
||||||
|
* to you 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef GUAC_RDP_SVC_H
|
||||||
|
#define GUAC_RDP_SVC_H
|
||||||
|
|
||||||
|
#include "config.h"
|
||||||
|
|
||||||
|
#include <freerdp/svc.h>
|
||||||
|
#include <guacamole/client.h>
|
||||||
|
#include <guacamole/stream.h>
|
||||||
|
#include <winpr/stream.h>
|
||||||
|
#include <winpr/wtsapi.h>
|
||||||
|
#include <winpr/wtypes.h>
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The maximum number of bytes to allow within each channel name, including
|
||||||
|
* null terminator.
|
||||||
|
*/
|
||||||
|
#define GUAC_RDP_SVC_MAX_LENGTH 8
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The maximum number of bytes that the RDP server will be allowed to send
|
||||||
|
* within any single write operation, regardless of the number of chunks that
|
||||||
|
* write is split into. Bytes beyond this limit may be dropped.
|
||||||
|
*/
|
||||||
|
#define GUAC_SVC_MAX_ASSEMBLED_LENGTH 1048576
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Structure describing a static virtual channel, and the corresponding
|
||||||
|
* Guacamole pipes and FreeRDP resources.
|
||||||
|
*/
|
||||||
|
typedef struct guac_rdp_common_svc guac_rdp_common_svc;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handler which is invoked when a CHANNEL_EVENT_CONNECTED event has been
|
||||||
|
* processed and the connection/initialization process of the SVC is now
|
||||||
|
* complete.
|
||||||
|
*
|
||||||
|
* @param svc
|
||||||
|
* The guac_rdp_common_svc structure representing the SVC that is now
|
||||||
|
* connected.
|
||||||
|
*/
|
||||||
|
typedef void guac_rdp_common_svc_connect_handler(guac_rdp_common_svc* svc);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handler which is invoked when a logical block of data has been received
|
||||||
|
* along an SVC, having been reassembled from a series of
|
||||||
|
* CHANNEL_EVENT_DATA_RECEIVED events.
|
||||||
|
*
|
||||||
|
* @param svc
|
||||||
|
* The guac_rdp_common_svc structure representing the SVC that received the
|
||||||
|
* data.
|
||||||
|
*
|
||||||
|
* @param input_stream
|
||||||
|
* The reassembled block of data that was received.
|
||||||
|
*/
|
||||||
|
typedef void guac_rdp_common_svc_receive_handler(guac_rdp_common_svc* svc, wStream* input_stream);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handler which is invoked when a CHANNEL_EVENT_TERMINATED event has been
|
||||||
|
* processed and all resources associated with the SVC must now be freed.
|
||||||
|
*
|
||||||
|
* @param svc
|
||||||
|
* The guac_rdp_common_svc structure representing the SVC that has been
|
||||||
|
* terminated.
|
||||||
|
*/
|
||||||
|
typedef void guac_rdp_common_svc_terminate_handler(guac_rdp_common_svc* svc);
|
||||||
|
|
||||||
|
struct guac_rdp_common_svc {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Reference to the client owning this static channel.
|
||||||
|
*/
|
||||||
|
guac_client* client;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The name of the static virtual channel, as specified to
|
||||||
|
* guac_rdp_common_svc_load_plugin(). This value is stored and defined
|
||||||
|
* internally by the CHANNEL_DEF.
|
||||||
|
*/
|
||||||
|
const char* name;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Arbitrary channel-specific data which may be assigned and referenced by
|
||||||
|
* channel implementations leveraging the "guac-common-svc" plugin.
|
||||||
|
*/
|
||||||
|
void* data;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handler which is invoked when handling a CHANNEL_EVENT_CONNECTED event.
|
||||||
|
*/
|
||||||
|
guac_rdp_common_svc_connect_handler* _connect_handler;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handler which is invoked when all chunks of data for a single logical
|
||||||
|
* block have been received via CHANNEL_EVENT_DATA_RECEIVED events and
|
||||||
|
* reassembled.
|
||||||
|
*/
|
||||||
|
guac_rdp_common_svc_receive_handler* _receive_handler;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handler which is invokved when the SVC has been disconnected and is
|
||||||
|
* about to be freed.
|
||||||
|
*/
|
||||||
|
guac_rdp_common_svc_terminate_handler* _terminate_handler;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The definition of this static virtual channel, including its name.
|
||||||
|
*/
|
||||||
|
CHANNEL_DEF _channel_def;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Functions and data specific to the FreeRDP side of the virtual channel
|
||||||
|
* and plugin.
|
||||||
|
*/
|
||||||
|
CHANNEL_ENTRY_POINTS_FREERDP_EX _entry_points;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handle which identifies the client connection, typically referred to
|
||||||
|
* within the FreeRDP source as pInitHandle. This handle is provided to the
|
||||||
|
* channel entry point and the channel init event handler. The handle must
|
||||||
|
* eventually be used within the channel open event handler to obtain a
|
||||||
|
* handle to the channel itself.
|
||||||
|
*/
|
||||||
|
PVOID _init_handle;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handle which identifies the channel itself, typically referred to within
|
||||||
|
* the FreeRDP source as OpenHandle. This handle is obtained through a call
|
||||||
|
* to entry_points.pVirtualChannelOpenEx() in response to receiving a
|
||||||
|
* CHANNEL_EVENT_CONNECTED event via the init event handler.
|
||||||
|
*
|
||||||
|
* Data is received in CHANNEL_EVENT_DATA_RECEIVED events via the open
|
||||||
|
* event handler, and data is written through calls to
|
||||||
|
* entry_points.pVirtualChannelWriteEx().
|
||||||
|
*/
|
||||||
|
DWORD _open_handle;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* All data that has been received thus far from the current RDP server
|
||||||
|
* write operation. Data received along virtual channels is sent in chunks
|
||||||
|
* (typically 1600 bytes), and thus must be gradually reassembled as it is
|
||||||
|
* received.
|
||||||
|
*/
|
||||||
|
wStream* _input_stream;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Initializes arbitrary static virtual channel (SVC) support for RDP, loading
|
||||||
|
* a new instance of Guacamole's arbitrary SVC plugin for FreeRDP ("guacsvc")
|
||||||
|
* supporting the channel having the given name. Data sent from within the RDP
|
||||||
|
* session using this channel will be sent along an identically-named pipe
|
||||||
|
* stream to the Guacamole client, and data sent along a pipe stream having the
|
||||||
|
* same name will be written to the SVC and received within the RDP session. If
|
||||||
|
* failures occur while loading the plugin, messages noting the specifics of
|
||||||
|
* those failures will be logged, and support for the given channel will not be
|
||||||
|
* functional.
|
||||||
|
*
|
||||||
|
* This MUST be called within the PreConnect callback of the freerdp instance
|
||||||
|
* for static virtual channel support to be loaded.
|
||||||
|
*
|
||||||
|
* @param rdpContext
|
||||||
|
* The rdpContext associated with the FreeRDP side of the RDP connection.
|
||||||
|
*
|
||||||
|
* @param name
|
||||||
|
* The name of the SVC which should be handled by the new instance of the
|
||||||
|
* plugin.
|
||||||
|
*
|
||||||
|
* @param channel_options
|
||||||
|
* Bitwise OR of any of the several CHANNEL_OPTION_* flags. Regardless of
|
||||||
|
* whether specified here, the CHANNEL_OPTION_INTIALIZED and
|
||||||
|
* CHANNEL_OPTION_ENCRYPT_RDP flags will automatically be set.
|
||||||
|
*
|
||||||
|
* @param connect_handler
|
||||||
|
* The function to invoke when the SVC has been connected.
|
||||||
|
*
|
||||||
|
* @param receive_handler
|
||||||
|
* The function to invoke when the SVC has received a logical block of
|
||||||
|
* data, reassembled from perhaps several smaller chunks of data.
|
||||||
|
*
|
||||||
|
* @param terminate_handler
|
||||||
|
* The function to invoke when the SVC has been disconnected and is about
|
||||||
|
* to be freed.
|
||||||
|
*
|
||||||
|
* @return
|
||||||
|
* Zero if the plugin was loaded successfully, non-zero if the plugin could
|
||||||
|
* not be loaded.
|
||||||
|
*/
|
||||||
|
int guac_rdp_common_svc_load_plugin(rdpContext* context,
|
||||||
|
char* name, ULONG channel_options,
|
||||||
|
guac_rdp_common_svc_connect_handler* connect_handler,
|
||||||
|
guac_rdp_common_svc_receive_handler* receive_handler,
|
||||||
|
guac_rdp_common_svc_terminate_handler* terminate_handler);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Writes the given data to the virtual channel such that it can be received
|
||||||
|
* within the RDP session. The given data MUST be dynamically allocated, as the
|
||||||
|
* write operation may be queued and the actual write may not occur until
|
||||||
|
* later. The provided wStream and the buffer it points to will be
|
||||||
|
* automatically freed after the write occurs.
|
||||||
|
*
|
||||||
|
* @param svc
|
||||||
|
* The static virtual channel to write data to.
|
||||||
|
*
|
||||||
|
* @param output_stream
|
||||||
|
* The data to write, which MUST be dynamically allocated.
|
||||||
|
*/
|
||||||
|
void guac_rdp_common_svc_write(guac_rdp_common_svc* svc,
|
||||||
|
wStream* output_stream);
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
310
src/protocols/rdp/guac-common-svc/common-svc-main.c
Normal file
310
src/protocols/rdp/guac-common-svc/common-svc-main.c
Normal file
@ -0,0 +1,310 @@
|
|||||||
|
/*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one
|
||||||
|
* or more contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. The ASF licenses this file
|
||||||
|
* to you 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 "config.h"
|
||||||
|
#include "common-svc.h"
|
||||||
|
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include <freerdp/constants.h>
|
||||||
|
#include <guacamole/client.h>
|
||||||
|
#include <guacamole/protocol.h>
|
||||||
|
#include <guacamole/socket.h>
|
||||||
|
#include <guacamole/string.h>
|
||||||
|
#include <winpr/stream.h>
|
||||||
|
#include <winpr/wtypes.h>
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Event handler for events which deal with data transmitted over an open SVC.
|
||||||
|
* This specific implementation of the event handler currently handles only the
|
||||||
|
* CHANNEL_EVENT_DATA_RECEIVED event, delegating actual handling of that event
|
||||||
|
* to guac_rdp_common_svc_process_receive().
|
||||||
|
*
|
||||||
|
* The FreeRDP requirements for this function follow those of the
|
||||||
|
* VirtualChannelOpenEventEx callback defined within Microsoft's RDP API:
|
||||||
|
*
|
||||||
|
* https://docs.microsoft.com/en-us/previous-versions/windows/embedded/aa514754%28v%3dmsdn.10%29
|
||||||
|
*
|
||||||
|
* @param user_param
|
||||||
|
* The pointer to arbitrary data originally passed via the first parameter
|
||||||
|
* of the pVirtualChannelInitEx() function call when the associated channel
|
||||||
|
* was initialized. The pVirtualChannelInitEx() function is exposed within
|
||||||
|
* the channel entry points structure.
|
||||||
|
*
|
||||||
|
* @param open_handle
|
||||||
|
* The handle which identifies the channel itself, typically referred to
|
||||||
|
* within the FreeRDP source as OpenHandle.
|
||||||
|
*
|
||||||
|
* @param event
|
||||||
|
* An integer representing the event that should be handled. This will be
|
||||||
|
* either CHANNEL_EVENT_DATA_RECEIVED, CHANNEL_EVENT_WRITE_CANCELLED, or
|
||||||
|
* CHANNEL_EVENT_WRITE_COMPLETE.
|
||||||
|
*
|
||||||
|
* @param data
|
||||||
|
* The data received, for CHANNEL_EVENT_DATA_RECEIVED events, and the value
|
||||||
|
* passed as user data to pVirtualChannelWriteEx() for
|
||||||
|
* CHANNEL_EVENT_WRITE_* events (note that user data for
|
||||||
|
* pVirtualChannelWriteEx() as implemented by FreeRDP MUST either be NULL
|
||||||
|
* or a wStream containing the data written).
|
||||||
|
*
|
||||||
|
* @param data_length
|
||||||
|
* The number of bytes of event-specific data.
|
||||||
|
*
|
||||||
|
* @param total_length
|
||||||
|
* The total number of bytes expected to be received from the RDP server
|
||||||
|
* due to this single write (from the server's perspective). Each write may
|
||||||
|
* actually be split into multiple chunks, thus resulting in multiple
|
||||||
|
* receive events for the same logical block of data. The relationship
|
||||||
|
* between chunks is indicated with the CHANNEL_FLAG_FIRST and
|
||||||
|
* CHANNEL_FLAG_LAST flags.
|
||||||
|
*
|
||||||
|
* @param data_flags
|
||||||
|
* The result of a bitwise OR of the CHANNEL_FLAG_* flags which apply to
|
||||||
|
* the data received. This value is relevant only to
|
||||||
|
* CHANNEL_EVENT_DATA_RECEIVED events. Valid flags are CHANNEL_FLAG_FIRST,
|
||||||
|
* CHANNEL_FLAG_LAST, and CHANNEL_FLAG_ONLY. The flag CHANNEL_FLAG_MIDDLE
|
||||||
|
* is not itself a flag, but the absence of both CHANNEL_FLAG_FIRST and
|
||||||
|
* CHANNEL_FLAG_LAST.
|
||||||
|
*/
|
||||||
|
static VOID guac_rdp_common_svc_handle_open_event(LPVOID user_param,
|
||||||
|
DWORD open_handle, UINT event, LPVOID data, UINT32 data_length,
|
||||||
|
UINT32 total_length, UINT32 data_flags) {
|
||||||
|
|
||||||
|
/* Ignore all events except for received data */
|
||||||
|
if (event != CHANNEL_EVENT_DATA_RECEIVED)
|
||||||
|
return;
|
||||||
|
|
||||||
|
guac_rdp_common_svc* svc = (guac_rdp_common_svc*) user_param;
|
||||||
|
|
||||||
|
/* Validate relevant handle matches that of SVC */
|
||||||
|
if (open_handle != svc->_open_handle) {
|
||||||
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "%i bytes of data "
|
||||||
|
"received from within the remote desktop session for SVC "
|
||||||
|
"\"%s\" are being dropped because the relevant open handle "
|
||||||
|
"(0x%X) does not match the open handle of the SVC (0x%X).",
|
||||||
|
data_length, svc->name, open_handle, svc->_open_handle);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* If receiving first chunk, allocate sufficient space for all remaining
|
||||||
|
* chunks */
|
||||||
|
if (data_flags & CHANNEL_FLAG_FIRST) {
|
||||||
|
|
||||||
|
/* Limit maximum received size */
|
||||||
|
if (total_length > GUAC_SVC_MAX_ASSEMBLED_LENGTH) {
|
||||||
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "RDP server has "
|
||||||
|
"requested to send a sequence of %i bytes, but this "
|
||||||
|
"exceeds the maximum buffer space of %i bytes. Received "
|
||||||
|
"data may be truncated.", total_length,
|
||||||
|
GUAC_SVC_MAX_ASSEMBLED_LENGTH);
|
||||||
|
total_length = GUAC_SVC_MAX_ASSEMBLED_LENGTH;
|
||||||
|
}
|
||||||
|
|
||||||
|
svc->_input_stream = Stream_New(NULL, total_length);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Add chunk to buffer only if sufficient space remains */
|
||||||
|
if (Stream_EnsureRemainingCapacity(svc->_input_stream, data_length))
|
||||||
|
Stream_Write(svc->_input_stream, data, data_length);
|
||||||
|
else
|
||||||
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "%i bytes of data "
|
||||||
|
"received from within the remote desktop session for SVC "
|
||||||
|
"\"%s\" are being dropped because the maximum available "
|
||||||
|
"space for received data has been exceeded.", data_length,
|
||||||
|
svc->name);
|
||||||
|
|
||||||
|
/* Fire event once last chunk has been received */
|
||||||
|
if (data_flags & CHANNEL_FLAG_LAST) {
|
||||||
|
|
||||||
|
Stream_SealLength(svc->_input_stream);
|
||||||
|
Stream_SetPosition(svc->_input_stream, 0);
|
||||||
|
|
||||||
|
/* Handle channel-specific data receipt tasks, if any */
|
||||||
|
if (svc->_receive_handler)
|
||||||
|
svc->_receive_handler(svc, svc->_input_stream);
|
||||||
|
|
||||||
|
Stream_Free(svc->_input_stream, TRUE);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Processes a CHANNEL_EVENT_CONNECTED event, completing the
|
||||||
|
* connection/initialization process of the channel.
|
||||||
|
*
|
||||||
|
* @param rdpsnd
|
||||||
|
* The guac_rdp_common_svc structure representing the channel.
|
||||||
|
*/
|
||||||
|
static void guac_rdp_common_svc_process_connect(guac_rdp_common_svc* svc) {
|
||||||
|
|
||||||
|
/* Open FreeRDP side of connected channel */
|
||||||
|
UINT32 open_status =
|
||||||
|
svc->_entry_points.pVirtualChannelOpenEx(svc->_init_handle,
|
||||||
|
&svc->_open_handle, svc->_channel_def.name,
|
||||||
|
guac_rdp_common_svc_handle_open_event);
|
||||||
|
|
||||||
|
/* Warn if the channel cannot be opened after all */
|
||||||
|
if (open_status != CHANNEL_RC_OK) {
|
||||||
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "SVC \"%s\" could not "
|
||||||
|
"be opened: %s (error %i)", svc->name,
|
||||||
|
WTSErrorToString(open_status), open_status);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Handle channel-specific connect tasks, if any */
|
||||||
|
if (svc->_connect_handler)
|
||||||
|
svc->_connect_handler(svc);
|
||||||
|
|
||||||
|
/* Channel is now ready */
|
||||||
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "SVC \"%s\" connected.",
|
||||||
|
svc->name);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Processes a CHANNEL_EVENT_TERMINATED event, freeing all resources associated
|
||||||
|
* with the channel.
|
||||||
|
*
|
||||||
|
* @param svc
|
||||||
|
* The guac_rdp_common_svc structure representing the channel.
|
||||||
|
*/
|
||||||
|
static void guac_rdp_common_svc_process_terminate(guac_rdp_common_svc* svc) {
|
||||||
|
|
||||||
|
/* Handle channel-specific termination tasks, if any */
|
||||||
|
if (svc->_terminate_handler)
|
||||||
|
svc->_terminate_handler(svc);
|
||||||
|
|
||||||
|
guac_client_log(svc->client, GUAC_LOG_DEBUG, "SVC \"%s\" disconnected.",
|
||||||
|
svc->name);
|
||||||
|
free(svc);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Event handler for events which deal with the overall lifecycle of an SVC.
|
||||||
|
* This specific implementation of the event handler currently handles only
|
||||||
|
* CHANNEL_EVENT_CONNECTED and CHANNEL_EVENT_TERMINATED events, delegating
|
||||||
|
* actual handling of those events to guac_rdp_common_svc_process_connect() and
|
||||||
|
* guac_rdp_common_svc_process_terminate() respectively.
|
||||||
|
*
|
||||||
|
* The FreeRDP requirements for this function follow those of the
|
||||||
|
* VirtualChannelInitEventEx callback defined within Microsoft's RDP API:
|
||||||
|
*
|
||||||
|
* https://docs.microsoft.com/en-us/previous-versions/windows/embedded/aa514727%28v%3dmsdn.10%29
|
||||||
|
*
|
||||||
|
* @param user_param
|
||||||
|
* The pointer to arbitrary data originally passed via the first parameter
|
||||||
|
* of the pVirtualChannelInitEx() function call when the associated channel
|
||||||
|
* was initialized. The pVirtualChannelInitEx() function is exposed within
|
||||||
|
* the channel entry points structure.
|
||||||
|
*
|
||||||
|
* @param init_handle
|
||||||
|
* The handle which identifies the client connection, typically referred to
|
||||||
|
* within the FreeRDP source as pInitHandle.
|
||||||
|
*
|
||||||
|
* @param event
|
||||||
|
* An integer representing the event that should be handled. This will be
|
||||||
|
* either CHANNEL_EVENT_CONNECTED, CHANNEL_EVENT_DISCONNECTED,
|
||||||
|
* CHANNEL_EVENT_INITIALIZED, CHANNEL_EVENT_TERMINATED, or
|
||||||
|
* CHANNEL_EVENT_V1_CONNECTED.
|
||||||
|
*
|
||||||
|
* @param data
|
||||||
|
* NULL in all cases except the CHANNEL_EVENT_CONNECTED event, in which
|
||||||
|
* case this is a null-terminated string containing the name of the server.
|
||||||
|
*
|
||||||
|
* @param data_length
|
||||||
|
* The number of bytes of data, if any.
|
||||||
|
*/
|
||||||
|
static VOID guac_rdp_common_svc_handle_init_event(LPVOID user_param,
|
||||||
|
LPVOID init_handle, UINT event, LPVOID data, UINT data_length) {
|
||||||
|
|
||||||
|
guac_rdp_common_svc* svc = (guac_rdp_common_svc*) user_param;
|
||||||
|
|
||||||
|
/* Validate relevant handle matches that of SVC */
|
||||||
|
if (init_handle != svc->_init_handle) {
|
||||||
|
guac_client_log(svc->client, GUAC_LOG_WARNING, "An init event (#%i) "
|
||||||
|
"for SVC \"%s\" has been dropped because the relevant init "
|
||||||
|
"handle (0x%X) does not match the init handle of the SVC "
|
||||||
|
"(0x%X).", event, svc->name, init_handle, svc->_init_handle);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (event) {
|
||||||
|
|
||||||
|
/* The remote desktop side of the SVC has been connected */
|
||||||
|
case CHANNEL_EVENT_CONNECTED:
|
||||||
|
guac_rdp_common_svc_process_connect(svc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
/* The channel has disconnected and now must be cleaned up */
|
||||||
|
case CHANNEL_EVENT_TERMINATED:
|
||||||
|
guac_rdp_common_svc_process_terminate(svc);
|
||||||
|
break;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Entry point for FreeRDP plugins. This function is automatically invoked when
|
||||||
|
* the plugin is loaded.
|
||||||
|
*
|
||||||
|
* @param entry_points
|
||||||
|
* Functions and data specific to the FreeRDP side of the virtual channel
|
||||||
|
* and plugin. This structure must be copied within implementation-specific
|
||||||
|
* storage such that the functions it references can be invoked when
|
||||||
|
* needed.
|
||||||
|
*
|
||||||
|
* @param init_handle
|
||||||
|
* The handle which identifies the client connection, typically referred to
|
||||||
|
* within the FreeRDP source as pInitHandle. This handle is also provided
|
||||||
|
* to the channel init event handler. The handle must eventually be used
|
||||||
|
* within the channel open event handler to obtain a handle to the channel
|
||||||
|
* itself.
|
||||||
|
*
|
||||||
|
* @return
|
||||||
|
* TRUE if the plugin has initialized successfully, FALSE otherwise.
|
||||||
|
*/
|
||||||
|
BOOL VirtualChannelEntryEx(PCHANNEL_ENTRY_POINTS entry_points,
|
||||||
|
PVOID init_handle) {
|
||||||
|
|
||||||
|
CHANNEL_ENTRY_POINTS_FREERDP_EX* entry_points_ex =
|
||||||
|
(CHANNEL_ENTRY_POINTS_FREERDP_EX*) entry_points;
|
||||||
|
|
||||||
|
/* Get structure representing the Guacamole side of the SVC from plugin
|
||||||
|
* parameters */
|
||||||
|
guac_rdp_common_svc* svc = (guac_rdp_common_svc*) entry_points_ex->pExtendedData;
|
||||||
|
|
||||||
|
/* Copy FreeRDP data into SVC structure for future reference */
|
||||||
|
svc->_entry_points = *entry_points_ex;
|
||||||
|
svc->_init_handle = init_handle;
|
||||||
|
|
||||||
|
/* Complete initialization */
|
||||||
|
if (svc->_entry_points.pVirtualChannelInitEx(svc, svc, init_handle,
|
||||||
|
&svc->_channel_def, 1, VIRTUAL_CHANNEL_VERSION_WIN2000,
|
||||||
|
guac_rdp_common_svc_handle_init_event) != CHANNEL_RC_OK) {
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
return TRUE;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user