Blame libfreerdp/core/server.c

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 Service 5a9772
#define DEBUG_DVC(...) \
Packit Service 5a9772
	do                 \
Packit Service 5a9772
	{                  \
Packit Service 5a9772
	} 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 Service 5a9772
static rdpPeerChannel* wts_get_dvc_channel_by_id(WTSVirtualChannelManager* vcm, 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 Service 5a9772
		channel = (rdpPeerChannel*)ArrayList_GetItem(vcm->dynamicVirtualChannels, 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 Service 5a9772
static BOOL wts_queue_receive_data(rdpPeerChannel* channel, const BYTE* Buffer, UINT32 Length)
Packit 1fb8d4
{
Packit 1fb8d4
	BYTE* buffer;
Packit 1fb8d4
	wtsChannelMessage* messageCtx;
Packit Service 5a9772
	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 Service 5a9772
static BOOL wts_queue_send_item(rdpPeerChannel* channel, BYTE* Buffer, 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 Service 5a9772
	return MessageQueue_Post(channel->vcm->queue, (void*)(UINT_PTR)channelId, 0, (void*)buffer,
Packit Service 5a9772
	                         (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 Service 5a9772
static BOOL wts_read_drdynvc_capabilities_response(rdpPeerChannel* channel, 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 Service 5a9772
	DEBUG_DVC("Version: %" PRIu16 "", Version);
Packit 1fb8d4
	channel->vcm->drdynvc_state = DRDYNVC_STATE_READY;
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
static BOOL wts_read_drdynvc_create_response(rdpPeerChannel* channel, 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 Service 5a9772
	if ((INT32)CreationStatus < 0)
Packit 1fb8d4
	{
Packit Service 5a9772
		DEBUG_DVC("ChannelId %" PRIu32 " creation failed (%" PRId32 ")", channel->channelId,
Packit Service 5a9772
		          (INT32)CreationStatus);
Packit 1fb8d4
		channel->dvc_open_state = DVC_OPEN_STATE_FAILED;
Packit 1fb8d4
	}
Packit 1fb8d4
	else
Packit 1fb8d4
	{
Packit Service 5a9772
		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 Service 5a9772
static BOOL wts_read_drdynvc_data_first(rdpPeerChannel* channel, wStream* s, int cbLen,
Packit Service 5a9772
                                        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 Service 5a9772
	if (!Stream_EnsureRemainingCapacity(channel->receiveData, 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 Service 5a9772
static BOOL wts_read_drdynvc_data(rdpPeerChannel* channel, wStream* s, UINT32 length)
Packit 1fb8d4
{
Packit 1fb8d4
	BOOL ret = FALSE;
Packit 1fb8d4
Packit 1fb8d4
	if (channel->dvc_total_length > 0)
Packit 1fb8d4
	{
Packit Service 5a9772
		if (Stream_GetPosition(channel->receiveData) + length > 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 Service 5a9772
		if (Stream_GetPosition(channel->receiveData) >= 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 Service 5a9772
		else
Packit Service 5a9772
			ret = TRUE;
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 Service 5a9772
	DEBUG_DVC("ChannelId %" PRIu32 " close response", channel->channelId);
Packit 1fb8d4
	channel->dvc_open_state = DVC_OPEN_STATE_CLOSED;
Packit Service 5a9772
	MessageQueue_PostQuit(channel->queue, 0);
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 Service 5a9772
		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 Service 5a9772
					WLog_ERR(TAG, "Cmd %d not recognized.", Cmd);
Packit 1fb8d4
					break;
Packit 1fb8d4
			}
Packit 1fb8d4
		}
Packit 1fb8d4
		else
Packit 1fb8d4
		{
Packit Service 5a9772
			DEBUG_DVC("ChannelId %" PRIu32 " not exists.", ChannelId);
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
	else
Packit 1fb8d4
	{
Packit Service 5a9772
		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 Service 5a9772
static BOOL wts_write_drdynvc_create_request(wStream* s, UINT32 ChannelId, const char* ChannelName)
Packit 1fb8d4
{
Packit Service 5a9772
	size_t len;
Packit 1fb8d4
	wts_write_drdynvc_header(s, CREATE_REQUEST_PDU, ChannelId);
Packit 1fb8d4
	len = strlen(ChannelName) + 1;
Packit 1fb8d4
Packit Service 5a9772
	if (!Stream_EnsureRemainingCapacity(s, len))
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit 1fb8d4
	Stream_Write(s, ChannelName, len);
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
static BOOL WTSProcessChannelData(rdpPeerChannel* channel, UINT16 channelId, const BYTE* data,
Packit Service 5a9772
                                  size_t s, UINT32 flags, size_t t)
Packit 1fb8d4
{
Packit 1fb8d4
	BOOL ret = TRUE;
Packit Service 5a9772
	const size_t size = (size_t)s;
Packit Service 5a9772
	const size_t totalSize = (size_t)t;
Packit Service 5a9772
Packit Service 5a9772
	WINPR_UNUSED(channelId);
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 Service 5a9772
static BOOL WTSReceiveChannelData(freerdp_peer* client, UINT16 channelId, const BYTE* data,
Packit Service 5a9772
                                  size_t size, UINT32 flags, size_t totalSize)
Packit 1fb8d4
{
Packit 1fb8d4
	UINT32 i;
Packit 1fb8d4
	BOOL status = FALSE;
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
		{
Packit Service 5a9772
			rdpPeerChannel* channel = (rdpPeerChannel*)mcs->channels[i].handle;
Packit Service 5a9772
Packit Service 5a9772
			if (channel)
Packit Service 5a9772
				status = WTSProcessChannelData(channel, channelId, data, size, flags, totalSize);
Packit Service 5a9772
			break;
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return status;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
void WTSVirtualChannelManagerGetFileDescriptor(HANDLE hServer, void** fds, int* fds_count)
Packit 1fb8d4
{
Packit 1fb8d4
	void* fd;
Packit Service 5a9772
	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 Service 5a9772
	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 Service 5a9772
		channel =
Packit Service 5a9772
		    (rdpPeerChannel*)WTSVirtualChannelOpen((HANDLE)vcm, 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 Service 5a9772
			if (!WTSVirtualChannelWrite(channel, (PCHAR)&dynvc_caps, sizeof(dynvc_caps), &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 Service 5a9772
		channelId = (UINT16)(UINT_PTR)message.context;
Packit Service 5a9772
		buffer = (BYTE*)message.wParam;
Packit Service 5a9772
		length = (UINT32)(UINT_PTR)message.lParam;
Packit 1fb8d4
Packit Service 5a9772
		if (!vcm->client->SendChannelData(vcm->client, channelId, buffer, length))
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 Service 5a9772
	WTSVirtualChannelManager* vcm = (WTSVirtualChannelManager*)hServer;
Packit 1fb8d4
	return MessageQueue_Event(vcm->queue);
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
static rdpMcsChannel* wts_get_joined_channel_by_name(rdpMcs* mcs, const char* channel_name)
Packit 1fb8d4
{
Packit 1fb8d4
	UINT32 index;
Packit 1fb8d4
Packit Service 5a9772
	if (!mcs || !channel_name || !strnlen(channel_name, CHANNEL_NAME_LEN))
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 Service 5a9772
			              strnlen(channel_name, CHANNEL_NAME_LEN)) == 0)
Packit 1fb8d4
				return &mcs->channels[index];
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return NULL;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
static rdpMcsChannel* wts_get_joined_channel_by_id(rdpMcs* mcs, 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 Service 5a9772
	return wts_get_joined_channel_by_name(client->context->rdp->mcs, channel_name) == NULL ? FALSE
Packit Service 5a9772
	                                                                                       : 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 Service 5a9772
	return wts_get_joined_channel_by_id(client->context->rdp->mcs, channel_id) == NULL ? FALSE
Packit Service 5a9772
	                                                                                   : TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL WTSVirtualChannelManagerIsChannelJoined(HANDLE hServer, const char* name)
Packit 1fb8d4
{
Packit Service 5a9772
	WTSVirtualChannelManager* vcm = (WTSVirtualChannelManager*)hServer;
Packit 1fb8d4
Packit 1fb8d4
	if (!vcm || !vcm->rdp)
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit Service 5a9772
	return wts_get_joined_channel_by_name(vcm->rdp->mcs, name) == NULL ? FALSE : TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BYTE WTSVirtualChannelManagerGetDrdynvcState(HANDLE hServer)
Packit 1fb8d4
{
Packit Service 5a9772
	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 Service 5a9772
	channel = wts_get_joined_channel_by_name(client->context->rdp->mcs, channel_name);
Packit 1fb8d4
Packit 1fb8d4
	if (!channel)
Packit 1fb8d4
		return 0;
Packit 1fb8d4
Packit 1fb8d4
	return channel->ChannelId;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WTSChannelSetHandleByName(freerdp_peer* client, const char* channel_name, 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 Service 5a9772
	channel = wts_get_joined_channel_by_name(client->context->rdp->mcs, 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 Service 5a9772
BOOL WTSChannelSetHandleById(freerdp_peer* client, const UINT16 channel_id, 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 Service 5a9772
	channel = wts_get_joined_channel_by_name(client->context->rdp->mcs, 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 Service 5a9772
const char* WTSChannelGetName(freerdp_peer* client, UINT16 channel_id)
Packit Service 5a9772
{
Packit Service 5a9772
	rdpMcsChannel* channel;
Packit Service 5a9772
Packit Service 5a9772
	if (!client || !client->context || !client->context->rdp)
Packit Service 5a9772
		return NULL;
Packit Service 5a9772
Packit Service 5a9772
	channel = wts_get_joined_channel_by_id(client->context->rdp->mcs, channel_id);
Packit Service 5a9772
Packit Service 5a9772
	if (!channel)
Packit Service 5a9772
		return NULL;
Packit Service 5a9772
Packit Service 5a9772
	return (const char*)channel->Name;
Packit Service 5a9772
}
Packit Service 5a9772
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionW(LPWSTR pTargetServerName, ULONG TargetLogonId,
Packit Service 5a9772
                                                  BYTE HotkeyVk, USHORT HotkeyModifiers)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionA(LPSTR pTargetServerName, ULONG TargetLogonId,
Packit Service 5a9772
                                                  BYTE HotkeyVk, USHORT HotkeyModifiers)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionExW(LPWSTR pTargetServerName, ULONG TargetLogonId,
Packit Service 5a9772
                                                    BYTE HotkeyVk, USHORT HotkeyModifiers,
Packit Service 5a9772
                                                    DWORD flags)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionExA(LPSTR pTargetServerName, ULONG TargetLogonId,
Packit Service 5a9772
                                                    BYTE HotkeyVk, USHORT HotkeyModifiers,
Packit Service 5a9772
                                                    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 Service 5a9772
BOOL WINAPI FreeRDP_WTSConnectSessionW(ULONG LogonId, ULONG TargetLogonId, PWSTR pPassword,
Packit Service 5a9772
                                       BOOL bWait)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSConnectSessionA(ULONG LogonId, ULONG TargetLogonId, PSTR pPassword,
Packit Service 5a9772
                                       BOOL bWait)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateServersW(LPWSTR pDomainName, DWORD Reserved, DWORD Version,
Packit Service 5a9772
                                         PWTS_SERVER_INFOW* ppServerInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateServersA(LPSTR pDomainName, DWORD Reserved, DWORD Version,
Packit Service 5a9772
                                         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 Service 5a9772
static void wts_virtual_channel_manager_free_message(void* obj)
Packit Service 5a9772
{
Packit Service 5a9772
	wMessage* msg = (wMessage*)obj;
Packit Service 5a9772
Packit Service 5a9772
	if (msg)
Packit Service 5a9772
	{
Packit Service 5a9772
		BYTE* buffer = (BYTE*)msg->wParam;
Packit Service 5a9772
Packit Service 5a9772
		if (buffer)
Packit Service 5a9772
			free(buffer);
Packit Service 5a9772
	}
Packit Service 5a9772
}
Packit Service 5a9772
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 Service 5a9772
	wObject queueCallbacks = { 0 };
Packit Service 5a9772
Packit Service 5a9772
	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 Service 5a9772
	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 Service 5a9772
	if (HashTable_Add(g_ServerHandles, (void*)(UINT_PTR)vcm->SessionId, (void*)vcm) < 0)
Packit 1fb8d4
		goto error_free;
Packit 1fb8d4
Packit Service 5a9772
	queueCallbacks.fnObjectFree = wts_virtual_channel_manager_free_message;
Packit Service 5a9772
	vcm->queue = MessageQueue_New(&queueCallbacks);
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 Service 5a9772
	hServer = (HANDLE)vcm;
Packit 1fb8d4
	return hServer;
Packit 1fb8d4
error_dynamicVirtualChannels:
Packit 1fb8d4
	MessageQueue_Free(vcm->queue);
Packit 1fb8d4
error_queue:
Packit Service 5a9772
	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 Service 5a9772
	vcm = (WTSVirtualChannelManager*)hServer;
Packit 1fb8d4
Packit 1fb8d4
	if (vcm)
Packit 1fb8d4
	{
Packit Service 5a9772
		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 Service 5a9772
			channel = (rdpPeerChannel*)ArrayList_GetItem(vcm->dynamicVirtualChannels, 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 Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateSessionsW(HANDLE hServer, DWORD Reserved, DWORD Version,
Packit Service 5a9772
                                          PWTS_SESSION_INFOW* ppSessionInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateSessionsA(HANDLE hServer, DWORD Reserved, DWORD Version,
Packit Service 5a9772
                                          PWTS_SESSION_INFOA* ppSessionInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateSessionsExW(HANDLE hServer, DWORD* pLevel, DWORD Filter,
Packit Service 5a9772
                                            PWTS_SESSION_INFO_1W* ppSessionInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateSessionsExA(HANDLE hServer, DWORD* pLevel, DWORD Filter,
Packit Service 5a9772
                                            PWTS_SESSION_INFO_1A* ppSessionInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateProcessesW(HANDLE hServer, DWORD Reserved, DWORD Version,
Packit Service 5a9772
                                           PWTS_PROCESS_INFOW* ppProcessInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateProcessesA(HANDLE hServer, DWORD Reserved, DWORD Version,
Packit Service 5a9772
                                           PWTS_PROCESS_INFOA* ppProcessInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSTerminateProcess(HANDLE hServer, DWORD ProcessId, DWORD ExitCode)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL WINAPI FreeRDP_WTSQuerySessionInformationW(HANDLE hServer, DWORD SessionId,
Packit Service 5a9772
                                                WTS_INFO_CLASS WTSInfoClass, LPWSTR* ppBuffer,
Packit Service 5a9772
                                                DWORD* pBytesReturned)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL WINAPI FreeRDP_WTSQuerySessionInformationA(HANDLE hServer, DWORD SessionId,
Packit Service 5a9772
                                                WTS_INFO_CLASS WTSInfoClass, LPSTR* ppBuffer,
Packit Service 5a9772
                                                DWORD* pBytesReturned)
Packit 1fb8d4
{
Packit 1fb8d4
	DWORD BytesReturned;
Packit 1fb8d4
	WTSVirtualChannelManager* vcm;
Packit Service 5a9772
	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 Service 5a9772
		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 Service 5a9772
		*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 Service 5a9772
                                        WTS_CONFIG_CLASS WTSConfigClass, LPWSTR* ppBuffer,
Packit Service 5a9772
                                        DWORD* pBytesReturned)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL WINAPI FreeRDP_WTSQueryUserConfigA(LPSTR pServerName, LPSTR pUserName,
Packit Service 5a9772
                                        WTS_CONFIG_CLASS WTSConfigClass, LPSTR* ppBuffer,
Packit Service 5a9772
                                        DWORD* pBytesReturned)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL WINAPI FreeRDP_WTSSetUserConfigW(LPWSTR pServerName, LPWSTR pUserName,
Packit Service 5a9772
                                      WTS_CONFIG_CLASS WTSConfigClass, LPWSTR pBuffer,
Packit Service 5a9772
                                      DWORD DataLength)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL WINAPI FreeRDP_WTSSetUserConfigA(LPSTR pServerName, LPSTR pUserName,
Packit Service 5a9772
                                      WTS_CONFIG_CLASS WTSConfigClass, LPSTR pBuffer,
Packit Service 5a9772
                                      DWORD DataLength)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSSendMessageW(HANDLE hServer, DWORD SessionId, LPWSTR pTitle,
Packit Service 5a9772
                                    DWORD TitleLength, LPWSTR pMessage, DWORD MessageLength,
Packit Service 5a9772
                                    DWORD Style, DWORD Timeout, DWORD* pResponse, BOOL bWait)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSSendMessageA(HANDLE hServer, DWORD SessionId, LPSTR pTitle,
Packit Service 5a9772
                                    DWORD TitleLength, LPSTR pMessage, DWORD MessageLength,
Packit Service 5a9772
                                    DWORD Style, DWORD Timeout, DWORD* pResponse, BOOL bWait)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSDisconnectSession(HANDLE hServer, DWORD SessionId, BOOL bWait)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSLogoffSession(HANDLE hServer, DWORD SessionId, 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 Service 5a9772
BOOL WINAPI FreeRDP_WTSWaitSystemEvent(HANDLE hServer, DWORD EventMask, DWORD* pEventFlags)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
static void peer_channel_queue_free_message(void* obj)
Packit Service 5a9772
{
Packit Service 5a9772
	wMessage* msg = (wMessage*)obj;
Packit Service 5a9772
	if (!msg)
Packit Service 5a9772
		return;
Packit Service 5a9772
Packit Service 5a9772
	free(msg->context);
Packit Service 5a9772
}
Packit Service 5a9772
Packit Service 5a9772
static void channel_free(rdpPeerChannel* channel)
Packit Service 5a9772
{
Packit Service 5a9772
	if (!channel)
Packit Service 5a9772
		return;
Packit Service 5a9772
Packit Service 5a9772
	MessageQueue_Free(channel->queue);
Packit Service 5a9772
	Stream_Free(channel->receiveData, TRUE);
Packit Service 5a9772
	free(channel);
Packit Service 5a9772
}
Packit Service 5a9772
Packit Service 5a9772
static rdpPeerChannel* channel_new(WTSVirtualChannelManager* vcm, freerdp_peer* client,
Packit Service 5a9772
                                   UINT32 ChannelId, UINT16 index, UINT16 type, size_t chunkSize)
Packit Service 5a9772
{
Packit Service 5a9772
	wObject queueCallbacks = { 0 };
Packit Service 5a9772
	rdpPeerChannel* channel = (rdpPeerChannel*)calloc(1, sizeof(rdpPeerChannel));
Packit Service 5a9772
Packit Service 5a9772
	if (!channel)
Packit Service 5a9772
		goto fail;
Packit Service 5a9772
Packit Service 5a9772
	channel->vcm = vcm;
Packit Service 5a9772
	channel->client = client;
Packit Service 5a9772
	channel->channelId = ChannelId;
Packit Service 5a9772
	channel->index = index;
Packit Service 5a9772
	channel->channelType = type;
Packit Service 5a9772
	channel->receiveData = Stream_New(NULL, chunkSize);
Packit Service 5a9772
Packit Service 5a9772
	if (!channel->receiveData)
Packit Service 5a9772
		goto fail;
Packit Service 5a9772
Packit Service 5a9772
	queueCallbacks.fnObjectFree = peer_channel_queue_free_message;
Packit Service 5a9772
	channel->queue = MessageQueue_New(&queueCallbacks);
Packit Service 5a9772
Packit Service 5a9772
	if (!channel->queue)
Packit Service 5a9772
		goto fail;
Packit Service 5a9772
Packit Service 5a9772
	return channel;
Packit Service 5a9772
fail:
Packit Service 5a9772
	channel_free(channel);
Packit Service 5a9772
	return NULL;
Packit Service 5a9772
}
Packit Service 5a9772
Packit Service 5a9772
HANDLE WINAPI FreeRDP_WTSVirtualChannelOpen(HANDLE hServer, DWORD SessionId, LPSTR pVirtualName)
Packit 1fb8d4
{
Packit Service 5a9772
	size_t 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 Service 5a9772
	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 Service 5a9772
		if (mcs->channels[index].joined &&
Packit Service 5a9772
		    (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 Service 5a9772
	channel = (rdpPeerChannel*)mcs->channels[index].handle;
Packit 1fb8d4
Packit 1fb8d4
	if (!channel)
Packit 1fb8d4
	{
Packit Service 5a9772
		channel = channel_new(vcm, client, mcs->channels[index].ChannelId, index,
Packit Service 5a9772
		                      RDP_PEER_CHANNEL_TYPE_SVC, client->settings->VirtualChannelChunkSize);
Packit 1fb8d4
Packit 1fb8d4
		if (!channel)
Packit Service 5a9772
			goto fail;
Packit 1fb8d4
Packit 1fb8d4
		mcs->channels[index].handle = channel;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit Service 5a9772
	hChannelHandle = (HANDLE)channel;
Packit 1fb8d4
	return hChannelHandle;
Packit Service 5a9772
fail:
Packit Service 5a9772
	channel_free(channel);
Packit 1fb8d4
	SetLastError(ERROR_NOT_ENOUGH_MEMORY);
Packit 1fb8d4
	return NULL;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
HANDLE WINAPI FreeRDP_WTSVirtualChannelOpenEx(DWORD SessionId, LPSTR pVirtualName, DWORD flags)
Packit 1fb8d4
{
Packit 1fb8d4
	UINT32 index;
Packit Service 5a9772
	wStream* s = NULL;
Packit 1fb8d4
	rdpMcs* mcs;
Packit 1fb8d4
	BOOL joined = FALSE;
Packit 1fb8d4
	freerdp_peer* client;
Packit Service 5a9772
	rdpPeerChannel* channel = NULL;
Packit 1fb8d4
	ULONG written;
Packit Service 5a9772
	WTSVirtualChannelManager* vcm = NULL;
Packit 1fb8d4
Packit 1fb8d4
	if (SessionId == WTS_CURRENT_SESSION)
Packit 1fb8d4
		return NULL;
Packit 1fb8d4
Packit Service 5a9772
	vcm = (WTSVirtualChannelManager*)HashTable_GetItemValue(g_ServerHandles,
Packit Service 5a9772
	                                                        (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 Service 5a9772
		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 Service 5a9772
		if (mcs->channels[index].joined && (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 Service 5a9772
	channel = channel_new(vcm, client, 0, 0, RDP_PEER_CHANNEL_TYPE_DVC,
Packit Service 5a9772
	                      client->settings->VirtualChannelChunkSize);
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->channelId = InterlockedIncrement(&vcm->dvc_channel_id_seq);
Packit 1fb8d4
Packit 1fb8d4
	if (ArrayList_Add(vcm->dynamicVirtualChannels, channel) < 0)
Packit Service 5a9772
		goto fail;
Packit 1fb8d4
Packit 1fb8d4
	s = Stream_New(NULL, 64);
Packit 1fb8d4
Packit 1fb8d4
	if (!s)
Packit Service 5a9772
		goto fail;
Packit 1fb8d4
Packit 1fb8d4
	if (!wts_write_drdynvc_create_request(s, channel->channelId, pVirtualName))
Packit Service 5a9772
		goto fail;
Packit 1fb8d4
Packit Service 5a9772
	if (!WTSVirtualChannelWrite(vcm->drdynvc_channel, (PCHAR)Stream_Buffer(s),
Packit 1fb8d4
	                            Stream_GetPosition(s), &written))
Packit Service 5a9772
		goto fail;
Packit 1fb8d4
Packit 1fb8d4
	Stream_Free(s, TRUE);
Packit 1fb8d4
	return channel;
Packit Service 5a9772
fail:
Packit 1fb8d4
	Stream_Free(s, TRUE);
Packit Service 5a9772
	if (vcm)
Packit Service 5a9772
		ArrayList_Remove(vcm->dynamicVirtualChannels, channel);
Packit Service 5a9772
	channel_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 Service 5a9772
	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 Service 5a9772
					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 Service 5a9772
BOOL WINAPI FreeRDP_WTSVirtualChannelRead(HANDLE hChannelHandle, ULONG TimeOut, PCHAR Buffer,
Packit Service 5a9772
                                          ULONG BufferSize, PULONG pBytesRead)
Packit 1fb8d4
{
Packit 1fb8d4
	BYTE* buffer;
Packit 1fb8d4
	wMessage message;
Packit 1fb8d4
	wtsChannelMessage* messageCtx;
Packit Service 5a9772
	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 Service 5a9772
	messageCtx = (wtsChannelMessage*)(UINT_PTR)message.context;
Packit Service 5a9772
Packit Service 5a9772
	if (messageCtx == NULL)
Packit Service 5a9772
		return FALSE;
Packit Service 5a9772
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 Service 5a9772
		peer_channel_queue_free_message(&message);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSVirtualChannelWrite(HANDLE hChannelHandle, PCHAR Buffer, ULONG Length,
Packit Service 5a9772
                                           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 Service 5a9772
	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 Service 5a9772
	else if (!channel->vcm->drdynvc_channel || (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 Service 5a9772
			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 Service 5a9772
BOOL WINAPI FreeRDP_WTSVirtualChannelQuery(HANDLE hChannelHandle, WTS_VIRTUAL_CLASS WtsVirtualClass,
Packit Service 5a9772
                                           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 Service 5a9772
	rdpPeerChannel* channel = (rdpPeerChannel*)hChannelHandle;
Packit 1fb8d4
	ZeroMemory(fds, sizeof(fds));
Packit 1fb8d4
	hEvent = MessageQueue_Event(channel->queue);
Packit 1fb8d4
Packit Service 5a9772
	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 Service 5a9772
BOOL WINAPI FreeRDP_WTSRegisterSessionNotificationEx(HANDLE hServer, HWND hWnd, DWORD dwFlags)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSUnRegisterSessionNotificationEx(HANDLE hServer, 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 Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateProcessesExW(HANDLE hServer, DWORD* pLevel, DWORD SessionId,
Packit Service 5a9772
                                             LPWSTR* ppProcessInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateProcessesExA(HANDLE hServer, DWORD* pLevel, DWORD SessionId,
Packit Service 5a9772
                                             LPSTR* ppProcessInfo, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateListenersW(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                           PWTSLISTENERNAMEW pListeners, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSEnumerateListenersA(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                           PWTSLISTENERNAMEA pListeners, DWORD* pCount)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSQueryListenerConfigW(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                            LPWSTR pListenerName, PWTSLISTENERCONFIGW pBuffer)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSQueryListenerConfigA(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                            LPSTR pListenerName, PWTSLISTENERCONFIGA pBuffer)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSCreateListenerW(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                       LPWSTR pListenerName, PWTSLISTENERCONFIGW pBuffer,
Packit Service 5a9772
                                       DWORD flag)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSCreateListenerA(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit 1fb8d4
                                       LPSTR pListenerName, PWTSLISTENERCONFIGA pBuffer, DWORD flag)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSSetListenerSecurityW(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                            LPWSTR pListenerName,
Packit Service 5a9772
                                            SECURITY_INFORMATION SecurityInformation,
Packit Service 5a9772
                                            PSECURITY_DESCRIPTOR pSecurityDescriptor)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSSetListenerSecurityA(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                            LPSTR pListenerName,
Packit Service 5a9772
                                            SECURITY_INFORMATION SecurityInformation,
Packit Service 5a9772
                                            PSECURITY_DESCRIPTOR pSecurityDescriptor)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSGetListenerSecurityW(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                            LPWSTR pListenerName,
Packit Service 5a9772
                                            SECURITY_INFORMATION SecurityInformation,
Packit Service 5a9772
                                            PSECURITY_DESCRIPTOR pSecurityDescriptor, DWORD nLength,
Packit Service 5a9772
                                            LPDWORD lpnLengthNeeded)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL WINAPI FreeRDP_WTSGetListenerSecurityA(HANDLE hServer, PVOID pReserved, DWORD Reserved,
Packit Service 5a9772
                                            LPSTR pListenerName,
Packit Service 5a9772
                                            SECURITY_INFORMATION SecurityInformation,
Packit Service 5a9772
                                            PSECURITY_DESCRIPTOR pSecurityDescriptor, DWORD nLength,
Packit Service 5a9772
                                            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 Service 5a9772
BOOL WINAPI FreeRDP_WTSLogonUser(HANDLE hServer, LPCSTR username, LPCSTR password, LPCSTR domain)
Packit 1fb8d4
{
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}