FreeRDP/libfreerdp/core/freerdp.c

1095 lines
26 KiB
C
Raw Normal View History

/**
2012-10-09 07:02:04 +04:00
* FreeRDP: A Remote Desktop Protocol Implementation
* FreeRDP Core
*
* Copyright 2011 Marc-Andre Moreau <marcandre.moreau@gmail.com>
* Copyright 2015 Thincast Technologies GmbH
* Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.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 <stdarg.h>
#include "rdp.h"
#include "input.h"
#include "update.h"
#include "surface.h"
#include "transport.h"
2011-07-28 19:09:51 +04:00
#include "connection.h"
#include "message.h"
2016-01-28 16:26:50 +03:00
#include "buildflags.h"
#include "gateway/rpc_fault.h"
2021-06-09 15:03:34 +03:00
#include <winpr/assert.h>
2018-08-24 14:49:19 +03:00
#include <winpr/crt.h>
2015-03-30 19:17:07 +03:00
#include <winpr/string.h>
#include <winpr/stream.h>
#include <winpr/wtsapi.h>
Winpr/openssl: Fix digests initialization in multi-thread SSL functions like OpenSSL_add_all_digests should be invoked at very beginning as they are not MT safe. If not we might meet double free exception as following: #0 0x00007f23ddd71c37 in raise () from /lib/x86_64-linux-gnu/libc.so.6 #1 0x00007f23ddd75028 in abort () from /lib/x86_64-linux-gnu/libc.so.6 #2 0x00007f23dddae2a4 in ?? () from /lib/x86_64-linux-gnu/libc.so.6 #3 0x00007f23dddba55e in ?? () from /lib/x86_64-linux-gnu/libc.so.6 #4 0x00007f23dc6ecfcd in CRYPTO_free () from /lib/x86_64-linux-gnu/libcrypto.so.1.0.0 #5 0x00007f23dc6ef8d1 in OBJ_NAME_add () from /lib/x86_64-linux-gnu/libcrypto.so.1.0.0 #6 0x00007f23dc77dcd8 in EVP_add_digest () from /lib/x86_64-linux-gnu/libcrypto.so.1.0.0 #7 0x00007f23dc782321 in OpenSSL_add_all_digests () from /lib/x86_64-linux-gnu/libcrypto.so.1.0.0 #8 0x00007f23c781da28 in winpr_openssl_get_evp_md (md=4) at /home/zihao/workspace/zihao_FreeRDP/winpr/libwinpr/crypto/hash.c:52 #9 0x00007f23c781dccb in winpr_Digest_Init (ctx=0x7f22d064d470, md=<optimized out>) at /home/zihao/workspace/zihao_FreeRDP/winpr/libwinpr/crypto/hash.c:344 #10 0x00007f23d486139b in security_salted_mac_signature (rdp=0x7f23859f5a20, data=0x7f238542d4fb "\004\204\022\004", length=4743, encryption=<optimized out>, output=0x7 at /home/zihao/workspace/zihao_FreeRDP/libfreerdp/core/security.c:378 #11 0x00007f23d488d73f in fastpath_send_update_pdu (fastpath=<optimized out>, updateCode=4 '\004', s=0x7f23859f5f40, skipCompression=true) at /home/zihao/workspace/zihao_FreeRDP/libfreerdp/core/fastpath.c:1076 #12 0x00007f23d4891c4f in update_send_surface_frame_bits (context=0x7f23859f5540, cmd=0x7f22b2ffcc80, first=true, last=true, frameId=6) at /home/zihao/workspace/zihao_FreeRDP/libfreerdp/core/update.c:1041 Related reports: https://rt.openssl.org/Ticket/Display.html?id=2216&user=guest&pass=guest
2016-12-26 17:21:36 +03:00
#include <winpr/ssl.h>
#include <winpr/debug.h>
#include <freerdp/freerdp.h>
2012-12-14 09:25:48 +04:00
#include <freerdp/error.h>
2013-06-19 00:55:23 +04:00
#include <freerdp/event.h>
2012-08-31 03:57:21 +04:00
#include <freerdp/locale/keyboard.h>
#include <freerdp/channels/channels.h>
#include <freerdp/version.h>
#include <freerdp/log.h>
#include <freerdp/cache/pointer.h>
#include "settings.h"
#define TAG FREERDP_TAG("core")
2012-02-23 19:30:04 +04:00
/** Creates a new connection based on the settings found in the "instance" parameter
* It will use the callbacks registered on the structure to process the pre/post connect operations
* that the caller requires.
* @see struct rdp_freerdp in freerdp.h
*
2019-11-06 17:24:51 +03:00
* @param instance - pointer to a rdp_freerdp structure that contains base information to establish
* the connection. On return, this function will be initialized with the new connection's settings.
*
* @return TRUE if successful. FALSE otherwise.
2012-02-23 19:30:04 +04:00
*
*/
BOOL freerdp_connect(freerdp* instance)
{
2016-10-11 12:20:54 +03:00
UINT status2 = CHANNEL_RC_OK;
rdpRdp* rdp;
2015-02-06 01:01:56 +03:00
BOOL status = TRUE;
rdpSettings* settings;
2013-07-04 00:18:45 +04:00
ConnectionResultEventArgs e;
2012-05-22 00:01:24 +04:00
if (!instance)
return FALSE;
WINPR_ASSERT(instance->context);
2012-04-13 15:16:08 +04:00
/* We always set the return code to 0 before we start the connect sequence*/
instance->ConnectionCallbackState = CLIENT_STATE_INITIAL;
freerdp_set_last_error_log(instance->context, FREERDP_ERROR_SUCCESS);
2015-09-03 14:43:08 +03:00
clearChannelError(instance->context);
ResetEvent(instance->context->abortEvent);
rdp = instance->context->rdp;
WINPR_ASSERT(rdp);
settings = instance->settings;
WINPR_ASSERT(settings);
freerdp_channels_register_instance(instance->context->channels, instance);
if (!freerdp_settings_set_default_order_support(settings))
return FALSE;
2015-02-06 22:21:26 +03:00
IFCALLRET(instance->PreConnect, status, instance);
instance->ConnectionCallbackState = CLIENT_STATE_PRECONNECT_PASSED;
2016-08-10 14:33:34 +03:00
if (status)
2019-11-06 17:24:51 +03:00
status2 = freerdp_channels_pre_connect(instance->context->channels, instance);
2016-08-10 14:33:34 +03:00
if (settings->KeyboardLayout == KBD_JAPANESE_INPUT_SYSTEM_MS_IME2002)
{
settings->KeyboardType = 7;
settings->KeyboardSubType = 2;
settings->KeyboardFunctionKey = 12;
}
2016-08-10 14:33:34 +03:00
if (!status || (status2 != CHANNEL_RC_OK))
{
freerdp_set_last_error_if_not(instance->context, FREERDP_ERROR_PRE_CONNECT_FAILED);
WLog_ERR(TAG, "freerdp_pre_connect failed");
goto freerdp_connect_finally;
}
status = rdp_client_connect(rdp);
2021-09-20 09:05:52 +03:00
/* Pointers might have changed inbetween */
if (rdp && rdp->settings)
{
/* --authonly tests the connection without a UI */
if (rdp->settings->AuthenticationOnly)
{
WLog_ERR(TAG, "Authentication only, exit status %" PRId32 "", status);
goto freerdp_connect_finally;
}
if (rdp->settings->DumpRemoteFx)
{
rdp->update->pcap_rfx = pcap_open(rdp->settings->DumpRemoteFxFile, TRUE);
2016-08-10 14:33:34 +03:00
if (rdp->update->pcap_rfx)
rdp->update->dump_rfx = TRUE;
}
}
if (status)
{
pointer_cache_register_callbacks(instance->context->update);
2015-02-06 22:21:26 +03:00
IFCALLRET(instance->PostConnect, status, instance);
instance->ConnectionCallbackState = CLIENT_STATE_POSTCONNECT_PASSED;
2015-02-06 01:01:56 +03:00
2016-08-10 14:33:34 +03:00
if (status)
status2 = freerdp_channels_post_connect(instance->context->channels, instance);
}
else
{
if (freerdp_get_last_error(instance->context) == FREERDP_ERROR_CONNECT_TRANSPORT_FAILED)
status = freerdp_reconnect(instance);
else
goto freerdp_connect_finally;
}
2016-08-10 14:33:34 +03:00
2019-11-06 17:24:51 +03:00
if (!status || (status2 != CHANNEL_RC_OK) || !update_post_connect(instance->update))
{
WLog_ERR(TAG, "freerdp_post_connect failed");
freerdp_set_last_error_if_not(instance->context, FREERDP_ERROR_POST_CONNECT_FAILED);
status = FALSE;
goto freerdp_connect_finally;
}
if (instance->settings->PlayRemoteFx)
{
wStream* s;
rdpUpdate* update;
pcap_record record;
update = instance->update;
update->pcap_rfx = pcap_open(settings->PlayRemoteFxFile, FALSE);
status = FALSE;
if (!update->pcap_rfx)
goto freerdp_connect_finally;
else
update->play_rfx = TRUE;
status = TRUE;
while (pcap_has_next_record(update->pcap_rfx) && status)
{
pcap_get_next_record_header(update->pcap_rfx, &record);
2021-09-06 12:01:36 +03:00
s = transport_take_from_pool(rdp->transport, record.length);
if (!s)
break;
record.data = Stream_Buffer(s);
pcap_get_next_record_content(update->pcap_rfx, &record);
Stream_SetLength(s, record.length);
Stream_SetPosition(s, 0);
if (!update_begin_paint(update))
status = FALSE;
else
{
if (update_recv_surfcmds(update, s) < 0)
status = FALSE;
if (!update_end_paint(update))
status = FALSE;
}
Stream_Release(s);
}
pcap_close(update->pcap_rfx);
update->pcap_rfx = NULL;
goto freerdp_connect_finally;
}
2012-05-22 00:01:24 +04:00
if (rdp->errorInfo == ERRINFO_SERVER_INSUFFICIENT_PRIVILEGES)
freerdp_set_last_error_log(instance->context, FREERDP_ERROR_INSUFFICIENT_PRIVILEGES);
2021-09-06 12:01:36 +03:00
transport_set_connected_event(rdp->transport);
2015-02-09 19:33:43 +03:00
freerdp_connect_finally:
2013-07-04 00:18:45 +04:00
EventArgsInit(&e, "freerdp");
e.result = status ? 0 : -1;
PubSub_OnConnectionResult(instance->context->pubSub, instance->context, &e);
if (!status)
freerdp_disconnect(instance);
return status;
}
BOOL freerdp_abort_connect(freerdp* instance)
{
if (!instance || !instance->context)
return FALSE;
return SetEvent(instance->context->abortEvent);
}
2019-11-06 17:24:51 +03:00
BOOL freerdp_get_fds(freerdp* instance, void** rfds, int* rcount, void** wfds, int* wcount)
{
rdpRdp* rdp;
WINPR_ASSERT(instance);
WINPR_ASSERT(instance->context);
rdp = instance->context->rdp;
WINPR_ASSERT(rdp);
transport_get_fds(rdp->transport, rfds, rcount);
return TRUE;
}
BOOL freerdp_check_fds(freerdp* instance)
{
int status;
rdpRdp* rdp;
if (!instance)
return FALSE;
if (!instance->context)
return FALSE;
if (!instance->context->rdp)
return FALSE;
rdp = instance->context->rdp;
status = rdp_check_fds(rdp);
if (status < 0)
2013-06-19 00:55:23 +04:00
{
TerminateEventArgs e;
rdpContext* context = instance->context;
WLog_DBG(TAG, "rdp_check_fds() - %i", status);
2013-06-19 00:55:23 +04:00
EventArgsInit(&e, "freerdp");
e.code = 0;
PubSub_OnTerminate(context->pubSub, context, &e);
return FALSE;
2013-06-19 00:55:23 +04:00
}
return TRUE;
2011-07-28 19:09:51 +04:00
}
2019-11-06 17:24:51 +03:00
DWORD freerdp_get_event_handles(rdpContext* context, HANDLE* events, DWORD count)
{
2015-04-21 13:35:55 +03:00
DWORD nCount = 0;
WINPR_ASSERT(context);
WINPR_ASSERT(context->rdp);
WINPR_ASSERT(events || (count == 0));
nCount += transport_get_event_handles(context->rdp->transport, events, count);
if (nCount == 0)
return 0;
2016-09-13 14:42:42 +03:00
if (events && (nCount < count + 2))
{
events[nCount++] = freerdp_channels_get_event_handle(context->instance);
events[nCount++] = getChannelErrorEventHandle(context);
2016-09-13 14:42:42 +03:00
events[nCount++] = context->abortEvent;
}
else
return 0;
WINPR_ASSERT(context->settings);
if (context->settings->AsyncInput)
{
if (nCount >= count)
return 0;
2016-10-11 12:20:54 +03:00
2019-11-06 17:24:51 +03:00
events[nCount++] =
freerdp_get_message_queue_event_handle(context->instance, FREERDP_INPUT_MESSAGE_QUEUE);
}
return nCount;
}
BOOL freerdp_check_event_handles(rdpContext* context)
{
BOOL status;
WINPR_ASSERT(context);
status = freerdp_check_fds(context->instance);
if (!status)
2015-05-02 06:26:08 +03:00
{
if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
2019-11-06 17:24:51 +03:00
WLog_ERR(TAG, "freerdp_check_fds() failed - %" PRIi32 "", status);
return FALSE;
}
status = freerdp_channels_check_fds(context->channels, context->instance);
2015-05-02 06:26:08 +03:00
if (!status)
{
if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
2019-11-06 17:24:51 +03:00
WLog_ERR(TAG, "freerdp_channels_check_fds() failed - %" PRIi32 "", status);
return FALSE;
}
status = checkChannelErrorEvent(context);
if (!status)
2018-07-09 18:45:50 +03:00
{
if (freerdp_get_last_error(context) == FREERDP_ERROR_SUCCESS)
2019-11-06 17:24:51 +03:00
WLog_ERR(TAG, "checkChannelErrorEvent() failed - %" PRIi32 "", status);
return FALSE;
2018-07-09 18:45:50 +03:00
}
WINPR_ASSERT(context->settings);
if (context->settings->AsyncInput)
{
2019-11-06 17:24:51 +03:00
int rc = freerdp_message_queue_process_pending_messages(context->instance,
FREERDP_INPUT_MESSAGE_QUEUE);
2018-07-09 18:45:50 +03:00
if (rc < 0)
return FALSE;
else
status = TRUE;
}
return status;
}
wMessageQueue* freerdp_get_message_queue(freerdp* instance, DWORD id)
{
wMessageQueue* queue = NULL;
WINPR_ASSERT(instance);
switch (id)
{
case FREERDP_UPDATE_MESSAGE_QUEUE:
WINPR_ASSERT(instance->update);
queue = instance->update->queue;
break;
case FREERDP_INPUT_MESSAGE_QUEUE:
WINPR_ASSERT(instance->input);
queue = instance->input->queue;
break;
}
return queue;
}
HANDLE freerdp_get_message_queue_event_handle(freerdp* instance, DWORD id)
{
HANDLE event = NULL;
wMessageQueue* queue = freerdp_get_message_queue(instance, id);
if (queue)
event = MessageQueue_Event(queue);
return event;
}
2019-11-06 17:24:51 +03:00
int freerdp_message_queue_process_message(freerdp* instance, DWORD id, wMessage* message)
{
int status = -1;
WINPR_ASSERT(instance);
switch (id)
{
case FREERDP_UPDATE_MESSAGE_QUEUE:
status = update_message_queue_process_message(instance->update, message);
break;
case FREERDP_INPUT_MESSAGE_QUEUE:
status = input_message_queue_process_message(instance->input, message);
break;
}
return status;
}
int freerdp_message_queue_process_pending_messages(freerdp* instance, DWORD id)
{
int status = -1;
WINPR_ASSERT(instance);
switch (id)
{
case FREERDP_UPDATE_MESSAGE_QUEUE:
status = update_message_queue_process_pending_messages(instance->update);
break;
case FREERDP_INPUT_MESSAGE_QUEUE:
status = input_message_queue_process_pending_messages(instance->input);
break;
}
return status;
}
static BOOL freerdp_send_channel_data(freerdp* instance, UINT16 channelId, const BYTE* data,
size_t size)
{
WINPR_ASSERT(instance);
WINPR_ASSERT(instance->context);
WINPR_ASSERT(instance->context->rdp);
return rdp_send_channel_data(instance->context->rdp, channelId, data, size);
}
static BOOL freerdp_send_channel_packet(freerdp* instance, UINT16 channelId, size_t totalSize,
UINT32 flags, const BYTE* data, size_t chunkSize)
{
WINPR_ASSERT(instance);
WINPR_ASSERT(instance->context);
WINPR_ASSERT(instance->context->rdp);
return rdp_channel_send_packet(instance->context->rdp, channelId, totalSize, flags, data,
chunkSize);
}
BOOL freerdp_disconnect(freerdp* instance)
{
BOOL rc = TRUE;
rdpRdp* rdp;
if (!instance || !instance->context)
return FALSE;
rdp = instance->context->rdp;
if (!rdp_client_disconnect(rdp))
rc = FALSE;
update_post_disconnect(instance->update);
2016-08-09 23:37:02 +03:00
if (instance->settings->AsyncInput)
{
2019-11-06 17:24:51 +03:00
wMessageQueue* inputQueue =
freerdp_get_message_queue(instance, FREERDP_INPUT_MESSAGE_QUEUE);
2016-10-11 12:20:54 +03:00
MessageQueue_PostQuit(inputQueue, 0);
}
2016-09-06 15:20:07 +03:00
IFCALL(instance->PostDisconnect, instance);
if (instance->update->pcap_rfx)
{
instance->update->dump_rfx = FALSE;
pcap_close(instance->update->pcap_rfx);
instance->update->pcap_rfx = NULL;
}
2016-08-09 23:37:02 +03:00
freerdp_channels_close(instance->context->channels, instance);
return rc;
}
BOOL freerdp_disconnect_before_reconnect(freerdp* instance)
{
rdpRdp* rdp;
WINPR_ASSERT(instance);
WINPR_ASSERT(instance->context);
rdp = instance->context->rdp;
return rdp_client_disconnect_and_clear(rdp);
}
BOOL freerdp_reconnect(freerdp* instance)
{
rdpRdp* rdp;
WINPR_ASSERT(instance);
WINPR_ASSERT(instance->context);
rdp = instance->context->rdp;
WINPR_ASSERT(instance->context->abortEvent);
ResetEvent(instance->context->abortEvent);
return rdp_client_reconnect(rdp);
}
BOOL freerdp_shall_disconnect(freerdp* instance)
{
if (!instance || !instance->context)
return FALSE;
if (WaitForSingleObject(instance->context->abortEvent, 0) != WAIT_OBJECT_0)
2015-07-30 09:18:57 +03:00
return FALSE;
return TRUE;
}
BOOL freerdp_focus_required(freerdp* instance)
{
rdpRdp* rdp;
BOOL bRetCode = FALSE;
rdp = instance->context->rdp;
if (rdp->resendFocus)
{
bRetCode = TRUE;
rdp->resendFocus = FALSE;
}
return bRetCode;
}
void freerdp_set_focus(freerdp* instance)
{
rdpRdp* rdp;
rdp = instance->context->rdp;
rdp->resendFocus = TRUE;
}
void freerdp_get_version(int* major, int* minor, int* revision)
{
if (major != NULL)
*major = FREERDP_VERSION_MAJOR;
if (minor != NULL)
*minor = FREERDP_VERSION_MINOR;
if (revision != NULL)
*revision = FREERDP_VERSION_REVISION;
}
2015-03-30 19:09:02 +03:00
const char* freerdp_get_version_string(void)
{
return FREERDP_VERSION_FULL;
}
2016-01-28 16:26:50 +03:00
const char* freerdp_get_build_config(void)
{
static const char build_config[] =
"Build configuration: " BUILD_CONFIG "\n"
"Build type: " BUILD_TYPE "\n"
"CFLAGS: " CFLAGS "\n"
"Compiler: " COMPILER_ID ", " COMPILER_VERSION "\n"
"Target architecture: " TARGET_ARCH "\n";
2016-01-28 16:26:50 +03:00
return build_config;
}
2015-03-30 19:09:02 +03:00
const char* freerdp_get_build_revision(void)
{
return FREERDP_GIT_REVISION;
}
2019-11-06 17:24:51 +03:00
static wEventType FreeRDP_Events[] = {
DEFINE_EVENT_ENTRY(WindowStateChange) DEFINE_EVENT_ENTRY(ResizeWindow)
DEFINE_EVENT_ENTRY(LocalResizeWindow) DEFINE_EVENT_ENTRY(EmbedWindow)
DEFINE_EVENT_ENTRY(PanningChange) DEFINE_EVENT_ENTRY(ZoomingChange)
DEFINE_EVENT_ENTRY(ErrorInfo) DEFINE_EVENT_ENTRY(Terminate)
DEFINE_EVENT_ENTRY(ConnectionResult) DEFINE_EVENT_ENTRY(ChannelConnected)
DEFINE_EVENT_ENTRY(ChannelDisconnected) DEFINE_EVENT_ENTRY(MouseEvent)
DEFINE_EVENT_ENTRY(Activated) DEFINE_EVENT_ENTRY(Timer)
DEFINE_EVENT_ENTRY(GraphicsReset)
};
/** Allocator function for a rdp context.
* The function will allocate a rdpRdp structure using rdp_new(), then copy
2012-02-23 19:30:04 +04:00
* its contents to the appropriate fields in the rdp_freerdp structure given in parameters.
* It will also initialize the 'context' field in the rdp_freerdp structure as needed.
2019-11-06 17:24:51 +03:00
* If the caller has set the ContextNew callback in the 'instance' parameter, it will be called at
* the end of the function.
*
2019-11-06 17:24:51 +03:00
* @param instance - Pointer to the rdp_freerdp structure that will be initialized with the new
* context.
*/
BOOL freerdp_context_new(freerdp* instance)
{
rdpRdp* rdp;
2013-06-14 06:11:23 +04:00
rdpContext* context;
BOOL ret = TRUE;
2019-11-06 17:24:51 +03:00
instance->context = (rdpContext*)calloc(1, instance->ContextSize);
if (!instance->context)
return FALSE;
2013-06-19 00:55:23 +04:00
context = instance->context;
context->instance = instance;
context->ServerMode = FALSE;
context->settings = instance->settings;
context->disconnectUltimatum = 0;
2013-06-19 00:55:23 +04:00
context->pubSub = PubSub_New(TRUE);
if (!context->pubSub)
goto fail;
2013-06-19 00:55:23 +04:00
2019-11-06 17:24:51 +03:00
PubSub_AddEventTypes(context->pubSub, FreeRDP_Events, ARRAYSIZE(FreeRDP_Events));
context->metrics = metrics_new(context);
if (!context->metrics)
goto fail;
rdp = rdp_new(context);
if (!rdp)
goto fail;
context->rdp = rdp;
instance->input = rdp->input;
instance->update = rdp->update;
instance->settings = rdp->settings;
instance->autodetect = rdp->autodetect;
instance->heartbeat = rdp->heartbeat;
2013-06-14 06:11:23 +04:00
context->graphics = graphics_new(context);
if (!context->graphics)
goto fail;
2013-06-14 06:11:23 +04:00
context->input = instance->input;
context->update = instance->update;
context->settings = instance->settings;
context->autodetect = instance->autodetect;
instance->update->context = instance->context;
instance->update->pointer->context = instance->context;
instance->update->primary->context = instance->context;
instance->update->secondary->context = instance->context;
instance->update->altsec->context = instance->context;
2013-06-14 06:11:23 +04:00
instance->input->context = context;
instance->autodetect->context = context;
if (!(context->errorDescription = calloc(1, 500)))
{
WLog_ERR(TAG, "calloc failed!");
goto fail;
}
if (!(context->channelErrorEvent = CreateEvent(NULL, TRUE, FALSE, NULL)))
{
WLog_ERR(TAG, "CreateEvent failed!");
goto fail;
}
update_register_client_callbacks(rdp->update);
instance->context->abortEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
if (!instance->context->abortEvent)
goto fail;
if (!(context->channels = freerdp_channels_new(instance)))
goto fail;
IFCALLRET(instance->ContextNew, ret, instance, instance->context);
if (ret)
return TRUE;
2015-08-04 14:34:28 +03:00
fail:
freerdp_context_free(instance);
return FALSE;
}
/** Deallocator function for a rdp context.
2019-11-06 17:24:51 +03:00
* The function will deallocate the resources from the 'instance' parameter that were allocated
* from a call to freerdp_context_new(). If the ContextFree callback is set in the 'instance'
* parameter, it will be called before deallocation occurs.
*
2019-11-06 17:24:51 +03:00
* @param instance - Pointer to the rdp_freerdp structure that was initialized by a call to
* freerdp_context_new(). On return, the fields associated to the context are invalid.
*/
void freerdp_context_free(freerdp* instance)
{
rdpContext* ctx;
if (!instance)
return;
2013-04-12 21:44:23 +04:00
if (!instance->context)
return;
ctx = instance->context;
IFCALL(instance->ContextFree, instance, ctx);
rdp_free(ctx->rdp);
ctx->rdp = NULL;
ctx->settings = NULL; /* owned by rdpRdp */
graphics_free(ctx->graphics);
ctx->graphics = NULL;
PubSub_Free(ctx->pubSub);
ctx->pubSub = NULL;
metrics_free(ctx->metrics);
ctx->metrics = NULL;
2021-09-20 09:05:52 +03:00
if (ctx->channelErrorEvent)
CloseHandle(ctx->channelErrorEvent);
ctx->channelErrorEvent = NULL;
free(ctx->errorDescription);
ctx->errorDescription = NULL;
2021-09-20 09:05:52 +03:00
if (ctx->abortEvent)
CloseHandle(ctx->abortEvent);
ctx->abortEvent = NULL;
freerdp_channels_free(ctx->channels);
ctx->channels = NULL;
codecs_free(ctx->codecs);
ctx->codecs = NULL;
free(ctx);
instance->context = NULL;
instance->settings = NULL; /* owned by rdpRdp */
}
int freerdp_get_disconnect_ultimatum(rdpContext* context)
{
return context->disconnectUltimatum;
}
2012-10-09 11:26:39 +04:00
UINT32 freerdp_error_info(freerdp* instance)
{
return instance->context->rdp->errorInfo;
}
void freerdp_set_error_info(rdpRdp* rdp, UINT32 error)
{
if (!rdp)
return;
rdp_set_error_info(rdp, error);
}
2019-05-19 14:34:00 +03:00
BOOL freerdp_send_error_info(rdpRdp* rdp)
{
if (!rdp)
return FALSE;
return rdp_send_error_info(rdp);
}
UINT32 freerdp_get_last_error(rdpContext* context)
{
return context->LastError;
}
const char* freerdp_get_last_error_name(UINT32 code)
{
const char* name = NULL;
const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
const UINT32 type = GET_FREERDP_ERROR_TYPE(code);
switch (cls)
{
case FREERDP_ERROR_ERRBASE_CLASS:
name = freerdp_get_error_base_name(type);
break;
case FREERDP_ERROR_ERRINFO_CLASS:
name = freerdp_get_error_info_name(type);
break;
case FREERDP_ERROR_CONNECT_CLASS:
name = freerdp_get_error_connect_name(type);
break;
default:
name = rpc_error_to_string(code);
break;
}
return name;
}
const char* freerdp_get_last_error_string(UINT32 code)
{
const char* string = NULL;
const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
const UINT32 type = GET_FREERDP_ERROR_TYPE(code);
switch (cls)
{
case FREERDP_ERROR_ERRBASE_CLASS:
string = freerdp_get_error_base_string(type);
break;
case FREERDP_ERROR_ERRINFO_CLASS:
string = freerdp_get_error_info_string(type);
break;
case FREERDP_ERROR_CONNECT_CLASS:
string = freerdp_get_error_connect_string(type);
break;
default:
string = rpc_error_to_string(code);
break;
}
return string;
}
const char* freerdp_get_last_error_category(UINT32 code)
{
const char* string = NULL;
const UINT32 cls = GET_FREERDP_ERROR_CLASS(code);
const UINT32 type = GET_FREERDP_ERROR_TYPE(code);
switch (cls)
{
2019-11-06 17:24:51 +03:00
case FREERDP_ERROR_ERRBASE_CLASS:
string = freerdp_get_error_base_category(type);
break;
2019-11-06 17:24:51 +03:00
case FREERDP_ERROR_ERRINFO_CLASS:
string = freerdp_get_error_info_category(type);
break;
2019-11-06 17:24:51 +03:00
case FREERDP_ERROR_CONNECT_CLASS:
string = freerdp_get_error_connect_category(type);
break;
2019-11-06 17:24:51 +03:00
default:
string = rpc_error_to_category(code);
break;
}
return string;
}
void freerdp_set_last_error(rdpContext* context, UINT32 lastError)
{
freerdp_set_last_error_ex(context, lastError, NULL, NULL, -1);
}
void freerdp_set_last_error_ex(rdpContext* context, UINT32 lastError, const char* fkt,
const char* file, int line)
{
if (lastError)
WLog_ERR(TAG, "%s:%s %s [0x%08" PRIX32 "]", fkt, __FUNCTION__,
2016-10-11 12:20:54 +03:00
freerdp_get_last_error_name(lastError), lastError);
if (lastError == FREERDP_ERROR_SUCCESS)
{
WLog_INFO(TAG, "%s:%s resetting error state", fkt, __FUNCTION__);
}
else if (context->LastError != FREERDP_ERROR_SUCCESS)
{
WLog_ERR(TAG, "%s: TODO: Trying to set error code %s, but %s already set!", fkt,
freerdp_get_last_error_name(lastError),
freerdp_get_last_error_name(context->LastError));
}
context->LastError = lastError;
}
const char* freerdp_get_logon_error_info_type(UINT32 type)
{
switch (type)
{
case LOGON_MSG_DISCONNECT_REFUSED:
return "LOGON_MSG_DISCONNECT_REFUSED";
case LOGON_MSG_NO_PERMISSION:
return "LOGON_MSG_NO_PERMISSION";
case LOGON_MSG_BUMP_OPTIONS:
return "LOGON_MSG_BUMP_OPTIONS";
case LOGON_MSG_RECONNECT_OPTIONS:
return "LOGON_MSG_RECONNECT_OPTIONS";
case LOGON_MSG_SESSION_TERMINATE:
return "LOGON_MSG_SESSION_TERMINATE";
case LOGON_MSG_SESSION_CONTINUE:
return "LOGON_MSG_SESSION_CONTINUE";
default:
return "UNKNOWN";
}
}
const char* freerdp_get_logon_error_info_data(UINT32 data)
{
switch (data)
{
case LOGON_FAILED_BAD_PASSWORD:
return "LOGON_FAILED_BAD_PASSWORD";
case LOGON_FAILED_UPDATE_PASSWORD:
return "LOGON_FAILED_UPDATE_PASSWORD";
case LOGON_FAILED_OTHER:
return "LOGON_FAILED_OTHER";
case LOGON_WARNING:
return "LOGON_WARNING";
default:
return "SESSION_ID";
}
}
/** Allocator function for the rdp_freerdp structure.
* @return an allocated structure filled with 0s. Need to be deallocated using freerdp_free()
*/
freerdp* freerdp_new(void)
{
freerdp* instance;
2019-11-06 17:24:51 +03:00
instance = (freerdp*)calloc(1, sizeof(freerdp));
if (!instance)
return NULL;
instance->ContextSize = sizeof(rdpContext);
instance->SendChannelData = freerdp_send_channel_data;
instance->SendChannelPacket = freerdp_send_channel_packet;
instance->ReceiveChannelData = freerdp_channels_data;
return instance;
}
/** Deallocator function for the rdp_freerdp structure.
* @param instance - pointer to the rdp_freerdp structure to deallocate.
* On return, this pointer is not valid anymore.
*/
void freerdp_free(freerdp* instance)
{
2015-05-11 10:07:39 +03:00
free(instance);
}
ULONG freerdp_get_transport_sent(rdpContext* context, BOOL resetCount)
{
2021-09-06 12:01:36 +03:00
WINPR_ASSERT(context);
WINPR_ASSERT(context->rdp);
return transport_get_bytes_sent(context->rdp->transport, resetCount);
}
BOOL freerdp_nla_impersonate(rdpContext* context)
{
rdpNla* nla;
if (!context)
return FALSE;
if (!context->rdp)
return FALSE;
if (!context->rdp->transport)
return FALSE;
2021-09-06 12:01:36 +03:00
nla = transport_get_nla(context->rdp->transport);
return nla_impersonate(nla);
}
BOOL freerdp_nla_revert_to_self(rdpContext* context)
{
rdpNla* nla;
if (!context)
return FALSE;
if (!context->rdp)
return FALSE;
if (!context->rdp->transport)
return FALSE;
2021-09-06 12:01:36 +03:00
nla = transport_get_nla(context->rdp->transport);
return nla_revert_to_self(nla);
}
HANDLE getChannelErrorEventHandle(rdpContext* context)
{
return context->channelErrorEvent;
}
BOOL checkChannelErrorEvent(rdpContext* context)
{
if (WaitForSingleObject(context->channelErrorEvent, 0) == WAIT_OBJECT_0)
{
2019-11-06 17:24:51 +03:00
WLog_ERR(TAG, "%s. Error was %" PRIu32 "", context->errorDescription,
2016-10-11 12:20:54 +03:00
context->channelErrorNum);
return FALSE;
}
return TRUE;
}
/**
* Function description
*
* @return 0 on success, otherwise a Win32 error code
*/
UINT getChannelError(rdpContext* context)
{
return context->channelErrorNum;
}
2015-09-03 14:43:08 +03:00
const char* getChannelErrorDescription(rdpContext* context)
{
return context->errorDescription;
}
2015-09-03 14:43:08 +03:00
void clearChannelError(rdpContext* context)
2015-09-03 14:43:08 +03:00
{
context->channelErrorNum = 0;
memset(context->errorDescription, 0, 500);
ResetEvent(context->channelErrorEvent);
}
void setChannelError(rdpContext* context, UINT errorNum, const char* format, ...)
{
va_list ap;
va_start(ap, format);
context->channelErrorNum = errorNum;
vsnprintf(context->errorDescription, 499, format, ap);
va_end(ap);
SetEvent(context->channelErrorEvent);
}
First version of an RDP proxy (#5372) * server: Add proxy dir with barebones server * sever/proxy: Remove licensing * server/proxy: Add client files * server/proxy: rm binary * server/proxy: Formatting * server/proxy: Fixed includes and added basic client creation functionality * server/proxy: Remove licensing and fix ifndef * proxy/server: Fix cmake indentation * server/proxy: Fix licensing * server/proxy: Forward connection on peer_post_connect * server/proxy: Fix function signature * server/proxy: Changed function signature of proxy_client_start * server/proxy: Now peer_post_connect calls proxy_client_start in a new thread * pfreerdp.c: Clean up useless comments and logs * server/proxy: Fix license * server/proxy: Remove all non-connection related data from proxy_context * server/proxy: Move Log Tag definition to pf_log.h * server/proxy: Move context definition to pf_context * server/proxy: Delete pfreerdp.h * pfreerdp.c: Move context callbacks to pf_context.c * server/proxy: Update CMakeLists.txt * pf_channels: Use new proxy context API * pf_client: Move context to pf_context * pf_client.c: Remove unnessecary event handling * server/proxy: Formatting * proxy/server: Move server logic to pf_server.c * server/proxy: Handle client disconnection * Merge stash * pf_server.c: Open GFX Connection to client * server: CMakeLists: build proxy along with other servers * server: proxy: get target server from rdpNego->RoutingToken Iv'e omitted a check from which im not sure is right. Should check in docs * server/proxy: Handle remote server -> client disconnection * server/proxy: Move common function to pf_common.c * server/proxy: Move common function to pf_common.c * rdpgfx.h: Add reference to freerdp.h for rdpContext * pf_channels: Pipe GFX on channel connection * server/proxy: Add pf_rdpgfx for proxy gfx callbacks * pf_client: Declare dynvc and gfx capabilities on connection * server/proxy: Add graphics callbacks * server/proxy: Add graphics callbacks * pf_server: Listen to channel events * Pass user settings to server * pf_server: Proxy mouse events * fixup! server/proxy: Add graphics callbacks * pf_client: Fix setting initialization * Merge feat/proxy-gfx to feat/proxy * pf_server: Fix double freed credentials * server/proxy: Remove unnecessary call to freerdp_client_settings_parse_command_line * server/proxy: Refactor re-activation code * server/proxy: Run format scripts * server/proxy: Fix segfault when post_disconnect return FALSE * server/proxy: Refactor proxy_settings_mirror * server/proxy: Redirect credentials * server/proxy: move proxy_settings_mirror to pf_common.c * server/proxy: Redirect desktop_resize event * pf_client: Remove interactive CLI auth methods * fixup! server/proxy: Redirect credentials * server/proxy: Rename proxy_mirror_settings to pf_common_copy_settings * pf_server.c: Fixed non-freed context When the disconnection is forced by the target server, the function `pf_server_handle_client_disconnection` isn't called. Therefore, the context of the connection between the proxy to tagrget isn't freed. * fixup! pf_server.c: Fixed non-freed context * pf_client: Prefix all client methods with pf_client * pf_context: Add init client to proxy context method * pf_server: Confirm all GFX caps regardless of settings * pf_server: Prefix all methods with pf_server * pf_server: Move variable decleration to start of method * pf_server: Fix client setting * pf_server: Fix GFX init method * pf_server: Move variable decleration to start of methods * server/proxy: Formatting * Merge feat/proxy * pf_server: Proxy synchronize event * pf_server: Proxy refresh rect update events * pf_server: Proxy suppress output messages * server/proxy: Fix licensing * server/proxy: Move client input callbacks to pf_input * server/proxy: Move client update callbacks to pf_update * server/proxy: Fix non-terminated target host string * Feat/proxy config (#2) * server/proxy: Add config loading support * server/proxy: Add config file * server/proxy: Format code * server/proxy: Code refactor, rename update_register_callbacks and input_register_callbacks * server/proxy: Update config file * server/proxy: Remove config.ini from root directory * Remove comment from config file * server/proxy: Fix leak in pf_server_load_config * server/proxy: Add rdpServerProxy struct and embed it in proxyContext * server/proxy: Load configuration and pass it inside every proxyContext instance * server/proxy: Move rdpProxyServer to proxy.h * server/proxy: Use configuration while proxying input events * server/proxy: Update CMakeLists * server/proxy: Refactor pf_input.c * server/proxy: Add AllowedChannels, DeniedChannels in configuration * server/proxy: Remove unnecessary variable from parse_channels_from_str * server/proxy: Update config file * server/proxy: config: Rename to * server/proxy: config: Add mode - blacklist/whitelist * server/proxy: Refactor, fix NULL deref * server/proxy: Add license to proxy.h * server/proxy: Fix newline in pf_config.c * server/proxy: config: Rename Mode to WhitelistMode * Add target in config. Add checks for configuration validity (#3) * Add target in config. Add checks for configuration validity * Update config file * libfreerdp: nego: revert commented out check of routingToken length * pf_server: Fix target host info from RoutingToken * pf_server: Remove hardcoded lenght of routing token prefix * Feat/refactor context (#8) * Refactor main structs * Update CMakeLists.txt * pf_server.c: Free pdata at the end of the connection * Run format scripts * Rename tf to pc * Fix licenses * pf_server: Refactor names of structs and functions * proxy: gfx: sync caps (#4) * proxy: gfx: sync caps * proxy: gfx: sync caps, hook gfx client's OnClose() call and close server resources * fixup! Feat/refactor context (#8) * fixup! fixup! Feat/refactor context (#8) * rdpgfx/client: Fix rdpgfx_recv_caps_confirm_pdu caps set length parsing * Run format scripts * proxy config.ini: Change default port to 3389 * pf_rdpgfx: Limit caps version to freerdp's supported versions * Gfx OnOpen() wait for dynvc ready (#10) * proxy/gfx: Wait for dynvc ready state before open * pf_channels: Initialize pc->gfx * pf_rdpgfx: Add log and fix comments * rdpgfx: Fix GFX v10.6 PDUs parsing and naming according to the spec * pf_rdpgfx: Proxy rdpgfx v10.6 PDUs * gfx client: Publish FrameAck sending and add auto ack flag * proxy/gfx: Forward frame ack messages * pf_context: Forward domain on connection * pf_rdpgfx: Change max supported caps to 10.6 * proxy: Update config * server/proxy: Use configuration in pf_server_handle_client * rdpgfx/client: Fix size of surface_to_scaled_window, surface_to_window * pf_rdpgfx: Fix formatting * pf_server.c: Fix comments * Move pf_server_rdpgfx_init to pf_rdpgfx * server/proxy/CMakeLists.txt: Fix formatting * pf_client.c: Add comment in proxy_server_reactivate * Fixed const correctness of gfx function pointer Signed-off-by: Mati Shabtay <matishabtay@gmail.com> * server: proxy: update copyright * server: proxy: wrap rdpNego and add a getter for routing token * Refactor routing token getter (#14) * Refactor routing token getter * pf_server_parse_target_from_routing_token change routing_token_length to be DWORD * libfreerdp/core/nego.c: Run format script * pf_server: Run format script * server/proxy: Fix os msbuild tests * pf_channels.c: Remove unused channels * pf_client: Remove unused callbacks * proxy: Remove encomsp callbacks from proxy's client * client/rdpgfx_main.c: Fix msbuild test * pf_config.c: Use StrSep instead of strsep for Windows builds * Removed nego struct from direct access. Signed-off-by: Mati Shabtay <matishabtay@gmail.com> * proxy: Rename binary to freerdp-proxy * rdpgfx_main.c: Revert unwanted double change to send_supported_caps * Cleaned up proxy server code. * All internal functions static * Added simple command line argument to supply a config file * Silence compiler warnings Signed-off-by: kubistika <kmizrachi18@gmail.com>
2019-05-17 15:32:54 +03:00
const char* freerdp_nego_get_routing_token(rdpContext* context, DWORD* length)
{
if (!context || !context->rdp)
return NULL;
return (const char*)nego_get_routing_token(context->rdp->nego, length);
}
const rdpTransportIo* freerdp_get_io_callbacks(rdpContext* context)
{
WINPR_ASSERT(context);
return rdp_get_io_callbacks(context->rdp);
}
BOOL freerdp_set_io_callbacks(rdpContext* context, const rdpTransportIo* io_callbacks)
{
WINPR_ASSERT(context);
return rdp_set_io_callbacks(context->rdp, io_callbacks);
}
CONNECTION_STATE freerdp_get_state(const rdpContext* context)
{
WINPR_ASSERT(context);
return rdp_get_state(context->rdp);
}
const char* freerdp_state_string(CONNECTION_STATE state)
{
return rdp_state_string(state);
}
BOOL freerdp_channels_from_mcs(rdpSettings* settings, const rdpContext* context)
{
WINPR_ASSERT(context);
return rdp_channels_from_mcs(settings, context->rdp);
}