|
Packit |
1fb8d4 |
/**
|
|
Packit |
1fb8d4 |
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
Packit |
1fb8d4 |
* Server Channels
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
Packit |
1fb8d4 |
* Copyright 2015 Thincast Technologies GmbH
|
|
Packit |
1fb8d4 |
* Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
Packit |
1fb8d4 |
* you may not use this file except in compliance with the License.
|
|
Packit |
1fb8d4 |
* You may obtain a copy of the License at
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* http://www.apache.org/licenses/LICENSE-2.0
|
|
Packit |
1fb8d4 |
*
|
|
Packit |
1fb8d4 |
* Unless required by applicable law or agreed to in writing, software
|
|
Packit |
1fb8d4 |
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
Packit |
1fb8d4 |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
Packit |
1fb8d4 |
* See the License for the specific language governing permissions and
|
|
Packit |
1fb8d4 |
* limitations under the License.
|
|
Packit |
1fb8d4 |
*/
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#ifdef HAVE_CONFIG_H
|
|
Packit |
1fb8d4 |
#include "config.h"
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include <stdio.h>
|
|
Packit |
1fb8d4 |
#include <stdlib.h>
|
|
Packit |
1fb8d4 |
#include <string.h>
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include <winpr/crt.h>
|
|
Packit |
1fb8d4 |
#include <winpr/synch.h>
|
|
Packit |
1fb8d4 |
#include <winpr/stream.h>
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include <freerdp/log.h>
|
|
Packit |
1fb8d4 |
#include <freerdp/constants.h>
|
|
Packit |
1fb8d4 |
#include <freerdp/server/channels.h>
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include "rdp.h"
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#include "server.h"
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#define TAG FREERDP_TAG("core.server")
|
|
Packit |
1fb8d4 |
#ifdef WITH_DEBUG_DVC
|
|
Packit |
1fb8d4 |
#define DEBUG_DVC(...) WLog_DBG(TAG, __VA_ARGS__)
|
|
Packit |
1fb8d4 |
#else
|
|
Packit |
1fb8d4 |
#define DEBUG_DVC(...) do { } while (0)
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
struct _wtsChannelMessage
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT16 channelId;
|
|
Packit |
1fb8d4 |
UINT16 reserved;
|
|
Packit |
1fb8d4 |
UINT32 length;
|
|
Packit |
1fb8d4 |
UINT32 offset;
|
|
Packit |
1fb8d4 |
};
|
|
Packit |
1fb8d4 |
typedef struct _wtsChannelMessage wtsChannelMessage;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static DWORD g_SessionId = 1;
|
|
Packit |
1fb8d4 |
static wHashTable* g_ServerHandles = NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static rdpPeerChannel* wts_get_dvc_channel_by_id(WTSVirtualChannelManager* vcm,
|
|
Packit |
1fb8d4 |
UINT32 ChannelId)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
int index;
|
|
Packit |
1fb8d4 |
int count;
|
|
Packit |
1fb8d4 |
BOOL found = FALSE;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel = NULL;
|
|
Packit |
1fb8d4 |
ArrayList_Lock(vcm->dynamicVirtualChannels);
|
|
Packit |
1fb8d4 |
count = ArrayList_Count(vcm->dynamicVirtualChannels);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
for (index = 0; index < count; index++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
channel = (rdpPeerChannel*) ArrayList_GetItem(vcm->dynamicVirtualChannels,
|
|
Packit |
1fb8d4 |
index);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel->channelId == ChannelId)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
found = TRUE;
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
ArrayList_Unlock(vcm->dynamicVirtualChannels);
|
|
Packit |
1fb8d4 |
return found ? channel : NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL wts_queue_receive_data(rdpPeerChannel* channel, const BYTE* Buffer,
|
|
Packit |
1fb8d4 |
UINT32 Length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
BYTE* buffer;
|
|
Packit |
1fb8d4 |
wtsChannelMessage* messageCtx;
|
|
Packit |
1fb8d4 |
messageCtx = (wtsChannelMessage*) malloc(sizeof(wtsChannelMessage) + Length);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!messageCtx)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
messageCtx->channelId = channel->channelId;
|
|
Packit |
1fb8d4 |
messageCtx->length = Length;
|
|
Packit |
1fb8d4 |
messageCtx->offset = 0;
|
|
Packit |
1fb8d4 |
buffer = (BYTE*)(messageCtx + 1);
|
|
Packit |
1fb8d4 |
CopyMemory(buffer, Buffer, Length);
|
|
Packit |
1fb8d4 |
return MessageQueue_Post(channel->queue, messageCtx, 0, NULL, NULL);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL wts_queue_send_item(rdpPeerChannel* channel, BYTE* Buffer,
|
|
Packit |
1fb8d4 |
UINT32 Length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
BYTE* buffer;
|
|
Packit |
1fb8d4 |
UINT32 length;
|
|
Packit |
1fb8d4 |
UINT16 channelId;
|
|
Packit |
1fb8d4 |
buffer = Buffer;
|
|
Packit |
1fb8d4 |
length = Length;
|
|
Packit |
1fb8d4 |
channelId = channel->channelId;
|
|
Packit |
1fb8d4 |
return MessageQueue_Post(channel->vcm->queue, (void*)(UINT_PTR) channelId, 0,
|
|
Packit |
1fb8d4 |
(void*) buffer, (void*)(UINT_PTR) length);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static int wts_read_variable_uint(wStream* s, int cbLen, UINT32* val)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
switch (cbLen)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
case 0:
|
|
Packit |
1fb8d4 |
if (Stream_GetRemainingLength(s) < 1)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Read_UINT8(s, *val);
|
|
Packit |
1fb8d4 |
return 1;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
case 1:
|
|
Packit |
1fb8d4 |
if (Stream_GetRemainingLength(s) < 2)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Read_UINT16(s, *val);
|
|
Packit |
1fb8d4 |
return 2;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
default:
|
|
Packit |
1fb8d4 |
if (Stream_GetRemainingLength(s) < 4)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Read_UINT32(s, *val);
|
|
Packit |
1fb8d4 |
return 4;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL wts_read_drdynvc_capabilities_response(rdpPeerChannel* channel,
|
|
Packit |
1fb8d4 |
UINT32 length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT16 Version;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (length < 3)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Seek_UINT8(channel->receiveData); /* Pad (1 byte) */
|
|
Packit |
1fb8d4 |
Stream_Read_UINT16(channel->receiveData, Version);
|
|
Packit |
1fb8d4 |
DEBUG_DVC("Version: %"PRIu16"", Version);
|
|
Packit |
1fb8d4 |
channel->vcm->drdynvc_state = DRDYNVC_STATE_READY;
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL wts_read_drdynvc_create_response(rdpPeerChannel* channel,
|
|
Packit |
1fb8d4 |
wStream* s, UINT32 length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT32 CreationStatus;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (length < 4)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Read_UINT32(s, CreationStatus);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if ((INT32) CreationStatus < 0)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
DEBUG_DVC("ChannelId %"PRIu32" creation failed (%"PRId32")", channel->channelId,
|
|
Packit |
1fb8d4 |
(INT32) CreationStatus);
|
|
Packit |
1fb8d4 |
channel->dvc_open_state = DVC_OPEN_STATE_FAILED;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
DEBUG_DVC("ChannelId %"PRIu32" creation succeeded", channel->channelId);
|
|
Packit |
1fb8d4 |
channel->dvc_open_state = DVC_OPEN_STATE_SUCCEEDED;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL wts_read_drdynvc_data_first(rdpPeerChannel* channel, wStream* s,
|
|
Packit |
1fb8d4 |
int cbLen, UINT32 length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
int value;
|
|
Packit |
1fb8d4 |
value = wts_read_variable_uint(s, cbLen, &channel->dvc_total_length);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (value == 0)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
length -= value;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (length > channel->dvc_total_length)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_SetPosition(channel->receiveData, 0);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!Stream_EnsureRemainingCapacity(channel->receiveData,
|
|
Packit |
1fb8d4 |
(int) channel->dvc_total_length))
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Write(channel->receiveData, Stream_Pointer(s), length);
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL wts_read_drdynvc_data(rdpPeerChannel* channel, wStream* s,
|
|
Packit |
1fb8d4 |
UINT32 length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
BOOL ret = FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel->dvc_total_length > 0)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (Stream_GetPosition(channel->receiveData) + length >
|
|
Packit |
1fb8d4 |
channel->dvc_total_length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
channel->dvc_total_length = 0;
|
|
Packit |
1fb8d4 |
WLog_ERR(TAG, "incorrect fragment data, discarded.");
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Write(channel->receiveData, Stream_Pointer(s), length);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (Stream_GetPosition(channel->receiveData) >= (int) channel->dvc_total_length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
ret = wts_queue_receive_data(channel, Stream_Buffer(channel->receiveData),
|
|
Packit |
1fb8d4 |
channel->dvc_total_length);
|
|
Packit |
1fb8d4 |
channel->dvc_total_length = 0;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
ret = wts_queue_receive_data(channel, Stream_Pointer(s), length);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return ret;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static void wts_read_drdynvc_close_response(rdpPeerChannel* channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
DEBUG_DVC("ChannelId %"PRIu32" close response", channel->channelId);
|
|
Packit |
1fb8d4 |
channel->dvc_open_state = DVC_OPEN_STATE_CLOSED;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL wts_read_drdynvc_pdu(rdpPeerChannel* channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT32 length;
|
|
Packit |
1fb8d4 |
int value;
|
|
Packit |
1fb8d4 |
int Cmd;
|
|
Packit |
1fb8d4 |
int Sp;
|
|
Packit |
1fb8d4 |
int cbChId;
|
|
Packit |
1fb8d4 |
UINT32 ChannelId;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* dvc;
|
|
Packit |
1fb8d4 |
length = Stream_GetPosition(channel->receiveData);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (length < 1)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_SetPosition(channel->receiveData, 0);
|
|
Packit |
1fb8d4 |
Stream_Read_UINT8(channel->receiveData, value);
|
|
Packit |
1fb8d4 |
length--;
|
|
Packit |
1fb8d4 |
Cmd = (value & 0xf0) >> 4;
|
|
Packit |
1fb8d4 |
Sp = (value & 0x0c) >> 2;
|
|
Packit |
1fb8d4 |
cbChId = (value & 0x03) >> 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (Cmd == CAPABILITY_REQUEST_PDU)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return wts_read_drdynvc_capabilities_response(channel, length);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else if (channel->vcm->drdynvc_state == DRDYNVC_STATE_READY)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
value = wts_read_variable_uint(channel->receiveData, cbChId, &ChannelId);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (value == 0)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
length -= value;
|
|
Packit |
1fb8d4 |
DEBUG_DVC("Cmd %d ChannelId %"PRIu32" length %"PRIu32"", Cmd, ChannelId, length);
|
|
Packit |
1fb8d4 |
dvc = wts_get_dvc_channel_by_id(channel->vcm, ChannelId);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (dvc)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
switch (Cmd)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
case CREATE_REQUEST_PDU:
|
|
Packit |
1fb8d4 |
return wts_read_drdynvc_create_response(dvc, channel->receiveData, length);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
case DATA_FIRST_PDU:
|
|
Packit |
1fb8d4 |
return wts_read_drdynvc_data_first(dvc, channel->receiveData, Sp, length);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
case DATA_PDU:
|
|
Packit |
1fb8d4 |
return wts_read_drdynvc_data(dvc, channel->receiveData, length);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
case CLOSE_REQUEST_PDU:
|
|
Packit |
1fb8d4 |
wts_read_drdynvc_close_response(dvc);
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
default:
|
|
Packit |
1fb8d4 |
WLog_ERR(TAG, "Cmd %d not recognized.", Cmd);
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
DEBUG_DVC("ChannelId %"PRIu32" not exists.", ChannelId);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WLog_ERR(TAG, "received Cmd %d but channel is not ready.", Cmd);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static int wts_write_variable_uint(wStream* s, UINT32 val)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
int cb;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (val <= 0xFF)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
cb = 0;
|
|
Packit |
1fb8d4 |
Stream_Write_UINT8(s, val);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else if (val <= 0xFFFF)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
cb = 1;
|
|
Packit |
1fb8d4 |
Stream_Write_UINT16(s, val);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
cb = 2;
|
|
Packit |
1fb8d4 |
Stream_Write_UINT32(s, val);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return cb;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static void wts_write_drdynvc_header(wStream* s, BYTE Cmd, UINT32 ChannelId)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
BYTE* bm;
|
|
Packit |
1fb8d4 |
int cbChId;
|
|
Packit |
1fb8d4 |
Stream_GetPointer(s, bm);
|
|
Packit |
1fb8d4 |
Stream_Seek_UINT8(s);
|
|
Packit |
1fb8d4 |
cbChId = wts_write_variable_uint(s, ChannelId);
|
|
Packit |
1fb8d4 |
*bm = ((Cmd & 0x0F) << 4) | cbChId;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL wts_write_drdynvc_create_request(wStream* s, UINT32 ChannelId,
|
|
Packit |
1fb8d4 |
const char* ChannelName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT32 len;
|
|
Packit |
1fb8d4 |
wts_write_drdynvc_header(s, CREATE_REQUEST_PDU, ChannelId);
|
|
Packit |
1fb8d4 |
len = strlen(ChannelName) + 1;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!Stream_EnsureRemainingCapacity(s, (int) len))
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Write(s, ChannelName, len);
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static BOOL WTSProcessChannelData(rdpPeerChannel* channel, UINT16 channelId,
|
|
Packit |
1fb8d4 |
const BYTE* data, int size, int flags, int totalSize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
BOOL ret = TRUE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (flags & CHANNEL_FLAG_FIRST)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
Stream_SetPosition(channel->receiveData, 0);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!Stream_EnsureRemainingCapacity(channel->receiveData, size))
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Write(channel->receiveData, data, size);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (flags & CHANNEL_FLAG_LAST)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (Stream_GetPosition(channel->receiveData) != totalSize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WLog_ERR(TAG, "read error");
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel == channel->vcm->drdynvc_channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
ret = wts_read_drdynvc_pdu(channel);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
ret = wts_queue_receive_data(channel, Stream_Buffer(channel->receiveData),
|
|
Packit |
1fb8d4 |
Stream_GetPosition(channel->receiveData));
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_SetPosition(channel->receiveData, 0);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return ret;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static int WTSReceiveChannelData(freerdp_peer* client, UINT16 channelId,
|
|
Packit |
1fb8d4 |
const BYTE* data, int size, int flags, int totalSize)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT32 i;
|
|
Packit |
1fb8d4 |
BOOL status = FALSE;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel;
|
|
Packit |
1fb8d4 |
rdpMcs* mcs = client->context->rdp->mcs;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
for (i = 0; i < mcs->channelCount; i++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (mcs->channels[i].ChannelId == channelId)
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (i < mcs->channelCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
channel = (rdpPeerChannel*) mcs->channels[i].handle;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WTSProcessChannelData(channel, channelId, data, size, flags, totalSize);
|
|
Packit |
1fb8d4 |
status = TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return status;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
void WTSVirtualChannelManagerGetFileDescriptor(HANDLE hServer, void** fds,
|
|
Packit |
1fb8d4 |
int* fds_count)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
void* fd;
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm = (WTSVirtualChannelManager*) hServer;
|
|
Packit |
1fb8d4 |
fd = GetEventWaitObject(MessageQueue_Event(vcm->queue));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (fd)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
fds[*fds_count] = fd;
|
|
Packit |
1fb8d4 |
(*fds_count)++;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#if 0
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (vcm->drdynvc_channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
fd = GetEventWaitObject(vcm->drdynvc_channel->receiveEvent);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (fd)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
fds[*fds_count] = fd;
|
|
Packit |
1fb8d4 |
(*fds_count)++;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
#endif
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WTSVirtualChannelManagerCheckFileDescriptor(HANDLE hServer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
wMessage message;
|
|
Packit |
1fb8d4 |
BOOL status = TRUE;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel;
|
|
Packit |
1fb8d4 |
UINT32 dynvc_caps;
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm = (WTSVirtualChannelManager*) hServer;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if ((vcm->drdynvc_state == DRDYNVC_STATE_NONE) && vcm->client->activated)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
/* Initialize drdynvc channel once and only once. */
|
|
Packit |
1fb8d4 |
vcm->drdynvc_state = DRDYNVC_STATE_INITIALIZED;
|
|
Packit |
1fb8d4 |
channel = (rdpPeerChannel*) WTSVirtualChannelOpen((HANDLE) vcm,
|
|
Packit |
1fb8d4 |
WTS_CURRENT_SESSION, "drdynvc");
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
ULONG written;
|
|
Packit |
1fb8d4 |
vcm->drdynvc_channel = channel;
|
|
Packit |
1fb8d4 |
dynvc_caps = 0x00010050; /* DYNVC_CAPS_VERSION1 (4 bytes) */
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!WTSVirtualChannelWrite(channel, (PCHAR) &dynvc_caps, sizeof(dynvc_caps),
|
|
Packit |
1fb8d4 |
&written))
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
while (MessageQueue_Peek(vcm->queue, &message, TRUE))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
BYTE* buffer;
|
|
Packit |
1fb8d4 |
UINT32 length;
|
|
Packit |
1fb8d4 |
UINT16 channelId;
|
|
Packit |
1fb8d4 |
channelId = (UINT16)(UINT_PTR) message.context;
|
|
Packit |
1fb8d4 |
buffer = (BYTE*) message.wParam;
|
|
Packit |
1fb8d4 |
length = (UINT32)(UINT_PTR) message.lParam;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (vcm->client->SendChannelData(vcm->client, channelId, buffer,
|
|
Packit |
1fb8d4 |
length) == FALSE)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
status = FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
free(buffer);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!status)
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return status;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
HANDLE WTSVirtualChannelManagerGetEventHandle(HANDLE hServer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm = (WTSVirtualChannelManager*) hServer;
|
|
Packit |
1fb8d4 |
return MessageQueue_Event(vcm->queue);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static rdpMcsChannel* wts_get_joined_channel_by_name(rdpMcs* mcs,
|
|
Packit |
1fb8d4 |
const char* channel_name)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT32 index;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!mcs || !channel_name || !strlen(channel_name))
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
for (index = 0; index < mcs->channelCount; index++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (mcs->channels[index].joined)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (_strnicmp(mcs->channels[index].Name, channel_name,
|
|
Packit |
1fb8d4 |
strlen(channel_name)) == 0)
|
|
Packit |
1fb8d4 |
return &mcs->channels[index];
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
static rdpMcsChannel* wts_get_joined_channel_by_id(rdpMcs* mcs,
|
|
Packit |
1fb8d4 |
const UINT16 channel_id)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT32 index;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!mcs || !channel_id)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
for (index = 0; index < mcs->channelCount; index++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (mcs->channels[index].joined)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (mcs->channels[index].ChannelId == channel_id)
|
|
Packit |
1fb8d4 |
return &mcs->channels[index];
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WTSIsChannelJoinedByName(freerdp_peer* client, const char* channel_name)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (!client || !client->context || !client->context->rdp)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return wts_get_joined_channel_by_name(client->context->rdp->mcs,
|
|
Packit |
1fb8d4 |
channel_name) == NULL ? FALSE : TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WTSIsChannelJoinedById(freerdp_peer* client, const UINT16 channel_id)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (!client || !client->context || !client->context->rdp)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return wts_get_joined_channel_by_id(client->context->rdp->mcs,
|
|
Packit |
1fb8d4 |
channel_id) == NULL ? FALSE : TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WTSVirtualChannelManagerIsChannelJoined(HANDLE hServer, const char* name)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm = (WTSVirtualChannelManager*) hServer;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!vcm || !vcm->rdp)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return wts_get_joined_channel_by_name(vcm->rdp->mcs,
|
|
Packit |
1fb8d4 |
name) == NULL ? FALSE : TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BYTE WTSVirtualChannelManagerGetDrdynvcState(HANDLE hServer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm = (WTSVirtualChannelManager*) hServer;
|
|
Packit |
1fb8d4 |
return vcm->drdynvc_state;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
UINT16 WTSChannelGetId(freerdp_peer* client, const char* channel_name)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
rdpMcsChannel* channel;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!client || !client->context || !client->context->rdp)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel = wts_get_joined_channel_by_name(client->context->rdp->mcs,
|
|
Packit |
1fb8d4 |
channel_name);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
return 0;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return channel->ChannelId;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WTSChannelSetHandleByName(freerdp_peer* client, const char* channel_name,
|
|
Packit |
1fb8d4 |
void* handle)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
rdpMcsChannel* channel;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!client || !client->context || !client->context->rdp)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel = wts_get_joined_channel_by_name(client->context->rdp->mcs,
|
|
Packit |
1fb8d4 |
channel_name);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel->handle = handle;
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WTSChannelSetHandleById(freerdp_peer* client, const UINT16 channel_id,
|
|
Packit |
1fb8d4 |
void* handle)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
rdpMcsChannel* channel;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!client || !client->context || !client->context->rdp)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel = wts_get_joined_channel_by_id(client->context->rdp->mcs, channel_id);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel->handle = handle;
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
void* WTSChannelGetHandleByName(freerdp_peer* client, const char* channel_name)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
rdpMcsChannel* channel;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!client || !client->context || !client->context->rdp)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel = wts_get_joined_channel_by_name(client->context->rdp->mcs,
|
|
Packit |
1fb8d4 |
channel_name);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return channel->handle;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
void* WTSChannelGetHandleById(freerdp_peer* client, const UINT16 channel_id)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
rdpMcsChannel* channel;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!client || !client->context || !client->context->rdp)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel = wts_get_joined_channel_by_id(client->context->rdp->mcs, channel_id);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return channel->handle;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionW(LPWSTR pTargetServerName,
|
|
Packit |
1fb8d4 |
ULONG TargetLogonId, BYTE HotkeyVk, USHORT HotkeyModifiers)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionA(LPSTR pTargetServerName,
|
|
Packit |
1fb8d4 |
ULONG TargetLogonId, BYTE HotkeyVk, USHORT HotkeyModifiers)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionExW(LPWSTR pTargetServerName,
|
|
Packit |
1fb8d4 |
ULONG TargetLogonId, BYTE HotkeyVk, USHORT HotkeyModifiers, DWORD flags)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionExA(LPSTR pTargetServerName,
|
|
Packit |
1fb8d4 |
ULONG TargetLogonId, BYTE HotkeyVk, USHORT HotkeyModifiers, DWORD flags)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSStopRemoteControlSession(ULONG LogonId)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSConnectSessionW(ULONG LogonId, ULONG TargetLogonId,
|
|
Packit |
1fb8d4 |
PWSTR pPassword, BOOL bWait)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSConnectSessionA(ULONG LogonId, ULONG TargetLogonId,
|
|
Packit |
1fb8d4 |
PSTR pPassword, BOOL bWait)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateServersW(LPWSTR pDomainName, DWORD Reserved,
|
|
Packit |
1fb8d4 |
DWORD Version, PWTS_SERVER_INFOW* ppServerInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateServersA(LPSTR pDomainName, DWORD Reserved,
|
|
Packit |
1fb8d4 |
DWORD Version, PWTS_SERVER_INFOA* ppServerInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
HANDLE WINAPI FreeRDP_WTSOpenServerW(LPWSTR pServerName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return INVALID_HANDLE_VALUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
HANDLE WINAPI FreeRDP_WTSOpenServerA(LPSTR pServerName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
rdpContext* context;
|
|
Packit |
1fb8d4 |
freerdp_peer* client;
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm;
|
|
Packit |
1fb8d4 |
HANDLE hServer = INVALID_HANDLE_VALUE;
|
|
Packit |
1fb8d4 |
context = (rdpContext*) pServerName;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!context)
|
|
Packit |
1fb8d4 |
return INVALID_HANDLE_VALUE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
client = context->peer;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!client)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_INVALID_DATA);
|
|
Packit |
1fb8d4 |
return INVALID_HANDLE_VALUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
vcm = (WTSVirtualChannelManager*) calloc(1, sizeof(WTSVirtualChannelManager));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!vcm)
|
|
Packit |
1fb8d4 |
goto error_vcm_alloc;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
vcm->client = client;
|
|
Packit |
1fb8d4 |
vcm->rdp = context->rdp;
|
|
Packit |
1fb8d4 |
vcm->SessionId = g_SessionId++;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!g_ServerHandles)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
g_ServerHandles = HashTable_New(TRUE);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!g_ServerHandles)
|
|
Packit |
1fb8d4 |
goto error_free;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (HashTable_Add(g_ServerHandles, (void*)(UINT_PTR) vcm->SessionId,
|
|
Packit |
1fb8d4 |
(void*) vcm) < 0)
|
|
Packit |
1fb8d4 |
goto error_free;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
vcm->queue = MessageQueue_New(NULL);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!vcm->queue)
|
|
Packit |
1fb8d4 |
goto error_queue;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
vcm->dvc_channel_id_seq = 0;
|
|
Packit |
1fb8d4 |
vcm->dynamicVirtualChannels = ArrayList_New(TRUE);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!vcm->dynamicVirtualChannels)
|
|
Packit |
1fb8d4 |
goto error_dynamicVirtualChannels;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
client->ReceiveChannelData = WTSReceiveChannelData;
|
|
Packit |
1fb8d4 |
hServer = (HANDLE) vcm;
|
|
Packit |
1fb8d4 |
return hServer;
|
|
Packit |
1fb8d4 |
error_dynamicVirtualChannels:
|
|
Packit |
1fb8d4 |
MessageQueue_Free(vcm->queue);
|
|
Packit |
1fb8d4 |
error_queue:
|
|
Packit |
1fb8d4 |
HashTable_Remove(g_ServerHandles, (void*)(UINT_PTR) vcm->SessionId);
|
|
Packit |
1fb8d4 |
error_free:
|
|
Packit |
1fb8d4 |
free(vcm);
|
|
Packit |
1fb8d4 |
error_vcm_alloc:
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
Packit |
1fb8d4 |
return INVALID_HANDLE_VALUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
HANDLE WINAPI FreeRDP_WTSOpenServerExW(LPWSTR pServerName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return INVALID_HANDLE_VALUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
HANDLE WINAPI FreeRDP_WTSOpenServerExA(LPSTR pServerName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FreeRDP_WTSOpenServerA(pServerName);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
VOID WINAPI FreeRDP_WTSCloseServer(HANDLE hServer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
int index;
|
|
Packit |
1fb8d4 |
int count;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel;
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm;
|
|
Packit |
1fb8d4 |
vcm = (WTSVirtualChannelManager*) hServer;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (vcm)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
HashTable_Remove(g_ServerHandles, (void*)(UINT_PTR) vcm->SessionId);
|
|
Packit |
1fb8d4 |
ArrayList_Lock(vcm->dynamicVirtualChannels);
|
|
Packit |
1fb8d4 |
count = ArrayList_Count(vcm->dynamicVirtualChannels);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
for (index = 0; index < count; index++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
channel = (rdpPeerChannel*) ArrayList_GetItem(vcm->dynamicVirtualChannels,
|
|
Packit |
1fb8d4 |
index);
|
|
Packit |
1fb8d4 |
WTSVirtualChannelClose(channel);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
ArrayList_Unlock(vcm->dynamicVirtualChannels);
|
|
Packit |
1fb8d4 |
ArrayList_Free(vcm->dynamicVirtualChannels);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (vcm->drdynvc_channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WTSVirtualChannelClose(vcm->drdynvc_channel);
|
|
Packit |
1fb8d4 |
vcm->drdynvc_channel = NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
MessageQueue_Free(vcm->queue);
|
|
Packit |
1fb8d4 |
free(vcm);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateSessionsW(HANDLE hServer, DWORD Reserved,
|
|
Packit |
1fb8d4 |
DWORD Version, PWTS_SESSION_INFOW* ppSessionInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateSessionsA(HANDLE hServer, DWORD Reserved,
|
|
Packit |
1fb8d4 |
DWORD Version, PWTS_SESSION_INFOA* ppSessionInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateSessionsExW(HANDLE hServer, DWORD* pLevel,
|
|
Packit |
1fb8d4 |
DWORD Filter, PWTS_SESSION_INFO_1W* ppSessionInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateSessionsExA(HANDLE hServer, DWORD* pLevel,
|
|
Packit |
1fb8d4 |
DWORD Filter, PWTS_SESSION_INFO_1A* ppSessionInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateProcessesW(HANDLE hServer, DWORD Reserved,
|
|
Packit |
1fb8d4 |
DWORD Version, PWTS_PROCESS_INFOW* ppProcessInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateProcessesA(HANDLE hServer, DWORD Reserved,
|
|
Packit |
1fb8d4 |
DWORD Version, PWTS_PROCESS_INFOA* ppProcessInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSTerminateProcess(HANDLE hServer, DWORD ProcessId,
|
|
Packit |
1fb8d4 |
DWORD ExitCode)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSQuerySessionInformationW(HANDLE hServer, DWORD SessionId,
|
|
Packit |
1fb8d4 |
WTS_INFO_CLASS WTSInfoClass, LPWSTR* ppBuffer, DWORD* pBytesReturned)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSQuerySessionInformationA(HANDLE hServer, DWORD SessionId,
|
|
Packit |
1fb8d4 |
WTS_INFO_CLASS WTSInfoClass, LPSTR* ppBuffer, DWORD* pBytesReturned)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
DWORD BytesReturned;
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm;
|
|
Packit |
1fb8d4 |
vcm = (WTSVirtualChannelManager*) hServer;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!vcm)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (WTSInfoClass == WTSSessionId)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
ULONG* pBuffer;
|
|
Packit |
1fb8d4 |
BytesReturned = sizeof(ULONG);
|
|
Packit |
1fb8d4 |
pBuffer = (ULONG*) malloc(sizeof(BytesReturned));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!pBuffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(E_OUTOFMEMORY);
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
*pBuffer = vcm->SessionId;
|
|
Packit |
1fb8d4 |
*ppBuffer = (LPSTR) pBuffer;
|
|
Packit |
1fb8d4 |
*pBytesReturned = BytesReturned;
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSQueryUserConfigW(LPWSTR pServerName, LPWSTR pUserName,
|
|
Packit |
1fb8d4 |
WTS_CONFIG_CLASS WTSConfigClass, LPWSTR* ppBuffer, DWORD* pBytesReturned)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSQueryUserConfigA(LPSTR pServerName, LPSTR pUserName,
|
|
Packit |
1fb8d4 |
WTS_CONFIG_CLASS WTSConfigClass, LPSTR* ppBuffer, DWORD* pBytesReturned)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSSetUserConfigW(LPWSTR pServerName, LPWSTR pUserName,
|
|
Packit |
1fb8d4 |
WTS_CONFIG_CLASS WTSConfigClass, LPWSTR pBuffer, DWORD DataLength)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSSetUserConfigA(LPSTR pServerName, LPSTR pUserName,
|
|
Packit |
1fb8d4 |
WTS_CONFIG_CLASS WTSConfigClass, LPSTR pBuffer, DWORD DataLength)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSSendMessageW(HANDLE hServer, DWORD SessionId,
|
|
Packit |
1fb8d4 |
LPWSTR pTitle, DWORD TitleLength,
|
|
Packit |
1fb8d4 |
LPWSTR pMessage, DWORD MessageLength, DWORD Style, DWORD Timeout,
|
|
Packit |
1fb8d4 |
DWORD* pResponse, BOOL bWait)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSSendMessageA(HANDLE hServer, DWORD SessionId,
|
|
Packit |
1fb8d4 |
LPSTR pTitle, DWORD TitleLength,
|
|
Packit |
1fb8d4 |
LPSTR pMessage, DWORD MessageLength, DWORD Style, DWORD Timeout,
|
|
Packit |
1fb8d4 |
DWORD* pResponse, BOOL bWait)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSDisconnectSession(HANDLE hServer, DWORD SessionId,
|
|
Packit |
1fb8d4 |
BOOL bWait)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSLogoffSession(HANDLE hServer, DWORD SessionId,
|
|
Packit |
1fb8d4 |
BOOL bWait)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSShutdownSystem(HANDLE hServer, DWORD ShutdownFlag)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSWaitSystemEvent(HANDLE hServer, DWORD EventMask,
|
|
Packit |
1fb8d4 |
DWORD* pEventFlags)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
HANDLE WINAPI FreeRDP_WTSVirtualChannelOpen(HANDLE hServer, DWORD SessionId,
|
|
Packit |
1fb8d4 |
LPSTR pVirtualName)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
int length;
|
|
Packit |
1fb8d4 |
UINT32 index;
|
|
Packit |
1fb8d4 |
rdpMcs* mcs;
|
|
Packit |
1fb8d4 |
BOOL joined = FALSE;
|
|
Packit |
1fb8d4 |
freerdp_peer* client;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel;
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm;
|
|
Packit |
1fb8d4 |
HANDLE hChannelHandle = NULL;
|
|
Packit |
1fb8d4 |
vcm = (WTSVirtualChannelManager*) hServer;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!vcm)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_INVALID_DATA);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
client = vcm->client;
|
|
Packit |
1fb8d4 |
mcs = client->context->rdp->mcs;
|
|
Packit |
1fb8d4 |
length = strlen(pVirtualName);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (length > 8)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NOT_FOUND);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
for (index = 0; index < mcs->channelCount; index++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (mcs->channels[index].joined
|
|
Packit |
1fb8d4 |
&& (strncmp(mcs->channels[index].Name, pVirtualName, length) == 0))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
joined = TRUE;
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!joined)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NOT_FOUND);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel = (rdpPeerChannel*) mcs->channels[index].handle;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
channel = (rdpPeerChannel*) calloc(1, sizeof(rdpPeerChannel));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
goto error_channel_alloc;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel->vcm = vcm;
|
|
Packit |
1fb8d4 |
channel->client = client;
|
|
Packit |
1fb8d4 |
channel->channelId = mcs->channels[index].ChannelId;
|
|
Packit |
1fb8d4 |
channel->index = index;
|
|
Packit |
1fb8d4 |
channel->channelType = RDP_PEER_CHANNEL_TYPE_SVC;
|
|
Packit |
1fb8d4 |
channel->receiveData = Stream_New(NULL,
|
|
Packit |
1fb8d4 |
client->settings->VirtualChannelChunkSize);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel->receiveData)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WLog_ERR(TAG, "Stream_New failed!");
|
|
Packit |
1fb8d4 |
goto error_receiveData;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel->queue = MessageQueue_New(NULL);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel->queue)
|
|
Packit |
1fb8d4 |
goto error_queue;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
mcs->channels[index].handle = channel;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
hChannelHandle = (HANDLE) channel;
|
|
Packit |
1fb8d4 |
return hChannelHandle;
|
|
Packit |
1fb8d4 |
error_queue:
|
|
Packit |
1fb8d4 |
Stream_Free(channel->receiveData, TRUE);
|
|
Packit |
1fb8d4 |
error_receiveData:
|
|
Packit |
1fb8d4 |
free(channel);
|
|
Packit |
1fb8d4 |
error_channel_alloc:
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
HANDLE WINAPI FreeRDP_WTSVirtualChannelOpenEx(DWORD SessionId,
|
|
Packit |
1fb8d4 |
LPSTR pVirtualName, DWORD flags)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
UINT32 index;
|
|
Packit |
1fb8d4 |
wStream* s;
|
|
Packit |
1fb8d4 |
rdpMcs* mcs;
|
|
Packit |
1fb8d4 |
BOOL joined = FALSE;
|
|
Packit |
1fb8d4 |
freerdp_peer* client;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel;
|
|
Packit |
1fb8d4 |
ULONG written;
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (SessionId == WTS_CURRENT_SESSION)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
vcm = (WTSVirtualChannelManager*) HashTable_GetItemValue(g_ServerHandles,
|
|
Packit |
1fb8d4 |
(void*)(UINT_PTR) SessionId);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!vcm)
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!(flags & WTS_CHANNEL_OPTION_DYNAMIC))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FreeRDP_WTSVirtualChannelOpen((HANDLE) vcm, SessionId, pVirtualName);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
client = vcm->client;
|
|
Packit |
1fb8d4 |
mcs = client->context->rdp->mcs;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
for (index = 0; index < mcs->channelCount; index++)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (mcs->channels[index].joined
|
|
Packit |
1fb8d4 |
&& (strncmp(mcs->channels[index].Name, "drdynvc", 7) == 0))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
joined = TRUE;
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!joined)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NOT_FOUND);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!vcm->drdynvc_channel || (vcm->drdynvc_state != DRDYNVC_STATE_READY))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NOT_READY);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel = (rdpPeerChannel*) calloc(1, sizeof(rdpPeerChannel));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel->vcm = vcm;
|
|
Packit |
1fb8d4 |
channel->client = client;
|
|
Packit |
1fb8d4 |
channel->channelType = RDP_PEER_CHANNEL_TYPE_DVC;
|
|
Packit |
1fb8d4 |
channel->receiveData = Stream_New(NULL,
|
|
Packit |
1fb8d4 |
client->settings->VirtualChannelChunkSize);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel->receiveData)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WLog_ERR(TAG, "Stream_New failed!");
|
|
Packit |
1fb8d4 |
goto error_receiveData;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel->queue = MessageQueue_New(NULL);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel->queue)
|
|
Packit |
1fb8d4 |
goto error_queue;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
channel->channelId = InterlockedIncrement(&vcm->dvc_channel_id_seq);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (ArrayList_Add(vcm->dynamicVirtualChannels, channel) < 0)
|
|
Packit |
1fb8d4 |
goto error_add;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
s = Stream_New(NULL, 64);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!s)
|
|
Packit |
1fb8d4 |
goto error_s;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!wts_write_drdynvc_create_request(s, channel->channelId, pVirtualName))
|
|
Packit |
1fb8d4 |
goto error_create;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!WTSVirtualChannelWrite(vcm->drdynvc_channel, (PCHAR) Stream_Buffer(s),
|
|
Packit |
1fb8d4 |
Stream_GetPosition(s), &written))
|
|
Packit |
1fb8d4 |
goto error_create;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Free(s, TRUE);
|
|
Packit |
1fb8d4 |
return channel;
|
|
Packit |
1fb8d4 |
error_create:
|
|
Packit |
1fb8d4 |
Stream_Free(s, TRUE);
|
|
Packit |
1fb8d4 |
error_s:
|
|
Packit |
1fb8d4 |
ArrayList_Remove(vcm->dynamicVirtualChannels, channel);
|
|
Packit |
1fb8d4 |
error_add:
|
|
Packit |
1fb8d4 |
MessageQueue_Free(channel->queue);
|
|
Packit |
1fb8d4 |
error_queue:
|
|
Packit |
1fb8d4 |
Stream_Free(channel->receiveData, TRUE);
|
|
Packit |
1fb8d4 |
error_receiveData:
|
|
Packit |
1fb8d4 |
free(channel);
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
Packit |
1fb8d4 |
return NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSVirtualChannelClose(HANDLE hChannelHandle)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
wStream* s;
|
|
Packit |
1fb8d4 |
rdpMcs* mcs;
|
|
Packit |
1fb8d4 |
WTSVirtualChannelManager* vcm;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel = (rdpPeerChannel*) hChannelHandle;
|
|
Packit |
1fb8d4 |
BOOL ret = TRUE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
vcm = channel->vcm;
|
|
Packit |
1fb8d4 |
mcs = vcm->client->context->rdp->mcs;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel->channelType == RDP_PEER_CHANNEL_TYPE_SVC)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
if (channel->index < mcs->channelCount)
|
|
Packit |
1fb8d4 |
mcs->channels[channel->index].handle = NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
ArrayList_Remove(vcm->dynamicVirtualChannels, channel);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel->dvc_open_state == DVC_OPEN_STATE_SUCCEEDED)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
ULONG written;
|
|
Packit |
1fb8d4 |
s = Stream_New(NULL, 8);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!s)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WLog_ERR(TAG, "Stream_New failed!");
|
|
Packit |
1fb8d4 |
ret = FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
wts_write_drdynvc_header(s, CLOSE_REQUEST_PDU, channel->channelId);
|
|
Packit |
1fb8d4 |
ret = WTSVirtualChannelWrite(vcm->drdynvc_channel, (PCHAR) Stream_Buffer(s),
|
|
Packit |
1fb8d4 |
Stream_GetPosition(s), &written);
|
|
Packit |
1fb8d4 |
Stream_Free(s, TRUE);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel->receiveData)
|
|
Packit |
1fb8d4 |
Stream_Free(channel->receiveData, TRUE);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel->queue)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
MessageQueue_Free(channel->queue);
|
|
Packit |
1fb8d4 |
channel->queue = NULL;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
free(channel);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return ret;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSVirtualChannelRead(HANDLE hChannelHandle, ULONG TimeOut,
|
|
Packit |
1fb8d4 |
PCHAR Buffer, ULONG BufferSize, PULONG pBytesRead)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
BYTE* buffer;
|
|
Packit |
1fb8d4 |
wMessage message;
|
|
Packit |
1fb8d4 |
wtsChannelMessage* messageCtx;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel = (rdpPeerChannel*) hChannelHandle;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!MessageQueue_Peek(channel->queue, &message, FALSE))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(ERROR_NO_DATA);
|
|
Packit |
1fb8d4 |
*pBytesRead = 0;
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
messageCtx = (wtsChannelMessage*)(UINT_PTR) message.context;
|
|
Packit |
1fb8d4 |
buffer = (BYTE*)(messageCtx + 1);
|
|
Packit |
1fb8d4 |
*pBytesRead = messageCtx->length - messageCtx->offset;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (Buffer == NULL || BufferSize == 0)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (*pBytesRead > BufferSize)
|
|
Packit |
1fb8d4 |
*pBytesRead = BufferSize;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
CopyMemory(Buffer, buffer + messageCtx->offset, *pBytesRead);
|
|
Packit |
1fb8d4 |
messageCtx->offset += *pBytesRead;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (messageCtx->offset >= messageCtx->length)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
MessageQueue_Peek(channel->queue, &message, TRUE);
|
|
Packit |
1fb8d4 |
free(messageCtx);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSVirtualChannelWrite(HANDLE hChannelHandle, PCHAR Buffer,
|
|
Packit |
1fb8d4 |
ULONG Length, PULONG pBytesWritten)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
wStream* s;
|
|
Packit |
1fb8d4 |
int cbLen;
|
|
Packit |
1fb8d4 |
int cbChId;
|
|
Packit |
1fb8d4 |
int first;
|
|
Packit |
1fb8d4 |
BYTE* buffer;
|
|
Packit |
1fb8d4 |
UINT32 length;
|
|
Packit |
1fb8d4 |
UINT32 written;
|
|
Packit |
1fb8d4 |
UINT32 totalWritten = 0;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel = (rdpPeerChannel*) hChannelHandle;
|
|
Packit |
1fb8d4 |
BOOL ret = TRUE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!channel)
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (channel->channelType == RDP_PEER_CHANNEL_TYPE_SVC)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
length = Length;
|
|
Packit |
1fb8d4 |
buffer = (BYTE*)malloc(length);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!buffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(E_OUTOFMEMORY);
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
CopyMemory(buffer, Buffer, length);
|
|
Packit |
1fb8d4 |
totalWritten = Length;
|
|
Packit |
1fb8d4 |
ret = wts_queue_send_item(channel, buffer, length);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else if (!channel->vcm->drdynvc_channel
|
|
Packit |
1fb8d4 |
|| (channel->vcm->drdynvc_state != DRDYNVC_STATE_READY))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
DEBUG_DVC("drdynvc not ready");
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
first = TRUE;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
while (Length > 0)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
s = Stream_New(NULL, channel->client->settings->VirtualChannelChunkSize);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!s)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
WLog_ERR(TAG, "Stream_New failed!");
|
|
Packit |
1fb8d4 |
SetLastError(E_OUTOFMEMORY);
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
buffer = Stream_Buffer(s);
|
|
Packit |
1fb8d4 |
Stream_Seek_UINT8(s);
|
|
Packit |
1fb8d4 |
cbChId = wts_write_variable_uint(s, channel->channelId);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (first && (Length > (UINT32) Stream_GetRemainingLength(s)))
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
cbLen = wts_write_variable_uint(s, Length);
|
|
Packit |
1fb8d4 |
buffer[0] = (DATA_FIRST_PDU << 4) | (cbLen << 2) | cbChId;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
buffer[0] = (DATA_PDU << 4) | cbChId;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
first = FALSE;
|
|
Packit |
1fb8d4 |
written = Stream_GetRemainingLength(s);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (written > Length)
|
|
Packit |
1fb8d4 |
written = Length;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
Stream_Write(s, Buffer, written);
|
|
Packit |
1fb8d4 |
length = Stream_GetPosition(s);
|
|
Packit |
1fb8d4 |
Stream_Free(s, FALSE);
|
|
Packit |
1fb8d4 |
Length -= written;
|
|
Packit |
1fb8d4 |
Buffer += written;
|
|
Packit |
1fb8d4 |
totalWritten += written;
|
|
Packit |
1fb8d4 |
ret = wts_queue_send_item(channel->vcm->drdynvc_channel, buffer, length);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (pBytesWritten)
|
|
Packit |
1fb8d4 |
*pBytesWritten = totalWritten;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return ret;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSVirtualChannelPurgeInput(HANDLE hChannelHandle)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSVirtualChannelPurgeOutput(HANDLE hChannelHandle)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSVirtualChannelQuery(HANDLE hChannelHandle,
|
|
Packit |
1fb8d4 |
WTS_VIRTUAL_CLASS WtsVirtualClass, PVOID* ppBuffer, DWORD* pBytesReturned)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
void* pfd;
|
|
Packit |
1fb8d4 |
BOOL bval;
|
|
Packit |
1fb8d4 |
void* fds[10];
|
|
Packit |
1fb8d4 |
HANDLE hEvent;
|
|
Packit |
1fb8d4 |
int fds_count = 0;
|
|
Packit |
1fb8d4 |
BOOL status = FALSE;
|
|
Packit |
1fb8d4 |
rdpPeerChannel* channel = (rdpPeerChannel*) hChannelHandle;
|
|
Packit |
1fb8d4 |
ZeroMemory(fds, sizeof(fds));
|
|
Packit |
1fb8d4 |
hEvent = MessageQueue_Event(channel->queue);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
switch ((UINT32) WtsVirtualClass)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
case WTSVirtualFileHandle:
|
|
Packit |
1fb8d4 |
pfd = GetEventWaitObject(hEvent);
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (pfd)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
fds[fds_count] = pfd;
|
|
Packit |
1fb8d4 |
(fds_count)++;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
*ppBuffer = malloc(sizeof(void*));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!*ppBuffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(E_OUTOFMEMORY);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
CopyMemory(*ppBuffer, &fds[0], sizeof(void*));
|
|
Packit |
1fb8d4 |
*pBytesReturned = sizeof(void*);
|
|
Packit |
1fb8d4 |
status = TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
case WTSVirtualEventHandle:
|
|
Packit |
1fb8d4 |
*ppBuffer = malloc(sizeof(HANDLE));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!*ppBuffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(E_OUTOFMEMORY);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
CopyMemory(*ppBuffer, &(hEvent), sizeof(HANDLE));
|
|
Packit |
1fb8d4 |
*pBytesReturned = sizeof(void*);
|
|
Packit |
1fb8d4 |
status = TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
case WTSVirtualChannelReady:
|
|
Packit |
1fb8d4 |
if (channel->channelType == RDP_PEER_CHANNEL_TYPE_SVC)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
bval = TRUE;
|
|
Packit |
1fb8d4 |
status = TRUE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
switch (channel->dvc_open_state)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
case DVC_OPEN_STATE_NONE:
|
|
Packit |
1fb8d4 |
bval = FALSE;
|
|
Packit |
1fb8d4 |
status = TRUE;
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
case DVC_OPEN_STATE_SUCCEEDED:
|
|
Packit |
1fb8d4 |
bval = TRUE;
|
|
Packit |
1fb8d4 |
status = TRUE;
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
default:
|
|
Packit |
1fb8d4 |
bval = FALSE;
|
|
Packit |
1fb8d4 |
status = FALSE;
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
*ppBuffer = malloc(sizeof(BOOL));
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
if (!*ppBuffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
SetLastError(E_OUTOFMEMORY);
|
|
Packit |
1fb8d4 |
status = FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
else
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
CopyMemory(*ppBuffer, &bval, sizeof(BOOL));
|
|
Packit |
1fb8d4 |
*pBytesReturned = sizeof(BOOL);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
default:
|
|
Packit |
1fb8d4 |
break;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
return status;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
VOID WINAPI FreeRDP_WTSFreeMemory(PVOID pMemory)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
free(pMemory);
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSFreeMemoryExW(WTS_TYPE_CLASS WTSTypeClass, PVOID pMemory,
|
|
Packit |
1fb8d4 |
ULONG NumberOfEntries)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSFreeMemoryExA(WTS_TYPE_CLASS WTSTypeClass, PVOID pMemory,
|
|
Packit |
1fb8d4 |
ULONG NumberOfEntries)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSRegisterSessionNotification(HWND hWnd, DWORD dwFlags)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSUnRegisterSessionNotification(HWND hWnd)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSRegisterSessionNotificationEx(HANDLE hServer, HWND hWnd,
|
|
Packit |
1fb8d4 |
DWORD dwFlags)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSUnRegisterSessionNotificationEx(HANDLE hServer,
|
|
Packit |
1fb8d4 |
HWND hWnd)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSQueryUserToken(ULONG SessionId, PHANDLE phToken)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateProcessesExW(HANDLE hServer, DWORD* pLevel,
|
|
Packit |
1fb8d4 |
DWORD SessionId, LPWSTR* ppProcessInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateProcessesExA(HANDLE hServer, DWORD* pLevel,
|
|
Packit |
1fb8d4 |
DWORD SessionId, LPSTR* ppProcessInfo, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateListenersW(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved, PWTSLISTENERNAMEW pListeners, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSEnumerateListenersA(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved, PWTSLISTENERNAMEA pListeners, DWORD* pCount)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSQueryListenerConfigW(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved, LPWSTR pListenerName, PWTSLISTENERCONFIGW pBuffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSQueryListenerConfigA(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved, LPSTR pListenerName, PWTSLISTENERCONFIGA pBuffer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSCreateListenerW(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved,
|
|
Packit |
1fb8d4 |
LPWSTR pListenerName, PWTSLISTENERCONFIGW pBuffer, DWORD flag)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSCreateListenerA(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved,
|
|
Packit |
1fb8d4 |
LPSTR pListenerName, PWTSLISTENERCONFIGA pBuffer, DWORD flag)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSSetListenerSecurityW(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved,
|
|
Packit |
1fb8d4 |
LPWSTR pListenerName, SECURITY_INFORMATION SecurityInformation,
|
|
Packit |
1fb8d4 |
PSECURITY_DESCRIPTOR pSecurityDescriptor)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSSetListenerSecurityA(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved,
|
|
Packit |
1fb8d4 |
LPSTR pListenerName, SECURITY_INFORMATION SecurityInformation,
|
|
Packit |
1fb8d4 |
PSECURITY_DESCRIPTOR pSecurityDescriptor)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSGetListenerSecurityW(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved,
|
|
Packit |
1fb8d4 |
LPWSTR pListenerName, SECURITY_INFORMATION SecurityInformation,
|
|
Packit |
1fb8d4 |
PSECURITY_DESCRIPTOR pSecurityDescriptor, DWORD nLength,
|
|
Packit |
1fb8d4 |
LPDWORD lpnLengthNeeded)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSGetListenerSecurityA(HANDLE hServer, PVOID pReserved,
|
|
Packit |
1fb8d4 |
DWORD Reserved,
|
|
Packit |
1fb8d4 |
LPSTR pListenerName, SECURITY_INFORMATION SecurityInformation,
|
|
Packit |
1fb8d4 |
PSECURITY_DESCRIPTOR pSecurityDescriptor, DWORD nLength,
|
|
Packit |
1fb8d4 |
LPDWORD lpnLengthNeeded)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL CDECL FreeRDP_WTSEnableChildSessions(BOOL bEnable)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL CDECL FreeRDP_WTSIsChildSessionsEnabled(PBOOL pbEnabled)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL CDECL FreeRDP_WTSGetChildSessionId(PULONG pSessionId)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
DWORD WINAPI FreeRDP_WTSGetActiveConsoleSessionId(void)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return 0xFFFFFFFF;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSLogoffUser(HANDLE hServer)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|
|
Packit |
1fb8d4 |
|
|
Packit |
1fb8d4 |
BOOL WINAPI FreeRDP_WTSLogonUser(HANDLE hServer, LPCSTR username,
|
|
Packit |
1fb8d4 |
LPCSTR password, LPCSTR domain)
|
|
Packit |
1fb8d4 |
{
|
|
Packit |
1fb8d4 |
return FALSE;
|
|
Packit |
1fb8d4 |
}
|