mirror of
https://github.com/FreeRDP/FreeRDP.git
synced 2025-06-03 00:00:20 +00:00
976c3c2ab9
* Split out proxy headers and moved to public API to allow external modules to be built. * Split proxy into proxy library and proxy binary. The library can be used by other applications and provides a simple API * Improved channel passthrough, now all channels including dynamic channels work. * Extended module API to hook more events, improved module samples * Cleaned up proxy code, removed global static variables used, added WINPR_ASSERT
328 lines
8.4 KiB
C++
328 lines
8.4 KiB
C++
/**
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
* FreeRDP Proxy Server Demo C++ Module
|
|
*
|
|
* Copyright 2019 Kobi Mizrachi <kmizrachi18@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 <iostream>
|
|
|
|
#include <freerdp/server/proxy/proxy_modules_api.h>
|
|
|
|
#define TAG MODULE_TAG("demo")
|
|
|
|
static constexpr char plugin_name[] = "demo";
|
|
static constexpr char plugin_desc[] = "this is a test plugin";
|
|
|
|
static BOOL demo_plugin_unload(proxyPlugin* plugin)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
|
|
std::cout << "C++ demo plugin: unloading..." << std::endl;
|
|
|
|
/* Here we have to free up our custom data storage. */
|
|
if (plugin)
|
|
free(plugin->custom);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_init_connect(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_uninit_connect(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_pre_connect(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_post_connect(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_post_disconnect(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_x509_certificate(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_login_failure(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_end_paint(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_server_post_connect(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_server_peer_activate(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_server_channels_init(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_server_channels_free(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_server_session_end(proxyPlugin* plugin, proxyData* pdata, void* custom)
|
|
{
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(custom);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_filter_keyboard_event(proxyPlugin* plugin, proxyData* pdata, void* param)
|
|
{
|
|
proxyPluginsManager* mgr;
|
|
auto event_data = static_cast<const proxyKeyboardEventInfo*>(param);
|
|
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(event_data);
|
|
|
|
mgr = plugin->mgr;
|
|
WINPR_ASSERT(mgr);
|
|
|
|
if (event_data == nullptr)
|
|
return FALSE;
|
|
|
|
if (event_data->rdp_scan_code == RDP_SCANCODE_KEY_B)
|
|
{
|
|
/* user typed 'B', that means bye :) */
|
|
std::cout << "C++ demo plugin: aborting connection" << std::endl;
|
|
mgr->AbortConnect(mgr, pdata);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_mouse_event(proxyPlugin* plugin, proxyData* pdata, void* param)
|
|
{
|
|
auto event_data = static_cast<const proxyMouseEventInfo*>(param);
|
|
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(event_data);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_client_channel_data(proxyPlugin* plugin, proxyData* pdata, void* param)
|
|
{
|
|
const proxyChannelDataEventInfo* channel = static_cast<const proxyChannelDataEventInfo*>(param);
|
|
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(channel);
|
|
|
|
WLog_INFO(TAG, "%s: %s [0x%04" PRIx16 "] got %" PRIuz, __FUNCTION__, channel->channel_name,
|
|
channel->channel_id, channel->data_len);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_server_channel_data(proxyPlugin* plugin, proxyData* pdata, void* param)
|
|
{
|
|
const proxyChannelDataEventInfo* channel = static_cast<const proxyChannelDataEventInfo*>(param);
|
|
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(channel);
|
|
|
|
WLog_WARN(TAG, "%s: %s [0x%04" PRIx16 "] got %" PRIuz, __FUNCTION__, channel->channel_name,
|
|
channel->channel_id, channel->data_len);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_dynamic_channel_create(proxyPlugin* plugin, proxyData* pdata, void* param)
|
|
{
|
|
const proxyChannelDataEventInfo* channel = static_cast<const proxyChannelDataEventInfo*>(param);
|
|
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(channel);
|
|
|
|
WLog_WARN(TAG, "%s: %s [0x%04" PRIx16 "]", __FUNCTION__, channel->channel_name,
|
|
channel->channel_id);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_server_fetch_target_addr(proxyPlugin* plugin, proxyData* pdata, void* param)
|
|
{
|
|
auto event_data = static_cast<const proxyFetchTargetEventInfo*>(param);
|
|
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(event_data);
|
|
|
|
WLog_INFO(TAG, "%s", __FUNCTION__);
|
|
return TRUE;
|
|
}
|
|
|
|
static BOOL demo_server_peer_logon(proxyPlugin* plugin, proxyData* pdata, void* param)
|
|
{
|
|
auto info = static_cast<const proxyServerPeerLogon*>(param);
|
|
WINPR_ASSERT(plugin);
|
|
WINPR_ASSERT(pdata);
|
|
WINPR_ASSERT(info);
|
|
WINPR_ASSERT(info->identity);
|
|
|
|
WLog_INFO(TAG, "%s: %d", __FUNCTION__, info->automatic);
|
|
return TRUE;
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
extern "C"
|
|
{
|
|
#endif
|
|
FREERDP_API BOOL proxy_module_entry_point(proxyPluginsManager* plugins_manager, void* userdata);
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
BOOL proxy_module_entry_point(proxyPluginsManager* plugins_manager, void* userdata)
|
|
{
|
|
struct demo_custom_data
|
|
{
|
|
proxyPluginsManager* mgr;
|
|
int somesetting;
|
|
};
|
|
struct demo_custom_data* custom;
|
|
proxyPlugin plugin;
|
|
|
|
plugin.name = plugin_name;
|
|
plugin.description = plugin_desc;
|
|
plugin.PluginUnload = demo_plugin_unload;
|
|
plugin.ClientInitConnect = demo_client_init_connect;
|
|
plugin.ClientUninitConnect = demo_client_uninit_connect;
|
|
plugin.ClientPreConnect = demo_client_pre_connect;
|
|
plugin.ClientPostConnect = demo_client_post_connect;
|
|
plugin.ClientPostDisconnect = demo_client_post_disconnect;
|
|
plugin.ClientX509Certificate = demo_client_x509_certificate;
|
|
plugin.ClientLoginFailure = demo_client_login_failure;
|
|
plugin.ClientEndPaint = demo_client_end_paint;
|
|
plugin.ServerPostConnect = demo_server_post_connect;
|
|
plugin.ServerPeerActivate = demo_server_peer_activate;
|
|
plugin.ServerChannelsInit = demo_server_channels_init;
|
|
plugin.ServerChannelsFree = demo_server_channels_free;
|
|
plugin.ServerSessionEnd = demo_server_session_end;
|
|
plugin.KeyboardEvent = demo_filter_keyboard_event;
|
|
plugin.MouseEvent = demo_mouse_event;
|
|
plugin.ClientChannelData = demo_client_channel_data;
|
|
plugin.ServerChannelData = demo_server_channel_data;
|
|
plugin.DynamicChannelCreate = demo_dynamic_channel_create;
|
|
plugin.ServerFetchTargetAddr = demo_server_fetch_target_addr;
|
|
plugin.ServerPeerLogon = demo_server_peer_logon;
|
|
plugin.userdata = userdata;
|
|
|
|
custom = (struct demo_custom_data*)calloc(1, sizeof(struct demo_custom_data));
|
|
if (!custom)
|
|
return FALSE;
|
|
|
|
custom->mgr = plugins_manager;
|
|
custom->somesetting = 42;
|
|
|
|
plugin.custom = custom;
|
|
plugin.userdata = userdata;
|
|
|
|
return plugins_manager->RegisterPlugin(plugins_manager, &plugin);
|
|
}
|