/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * Server Channels

 *

 * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>

 * Copyright 2015 Copyright 2015 Thincast Technologies GmbH

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#ifndef FREERDP_LIB_CORE_SERVER_H

#define FREERDP_LIB_CORE_SERVER_H

#include <freerdp/freerdp.h>

#include <freerdp/api.h>

#include <freerdp/channels/wtsvc.h>

#include <winpr/synch.h>

#include <winpr/stream.h>

#include <winpr/collections.h>

typedef struct rdp_peer_channel rdpPeerChannel;

typedef struct WTSVirtualChannelManager WTSVirtualChannelManager;

#include "rdp.h"

#include "mcs.h"

#define CREATE_REQUEST_PDU 0x01

#define DATA_FIRST_PDU 0x02

#define DATA_PDU 0x03

#define CLOSE_REQUEST_PDU 0x04

#define CAPABILITY_REQUEST_PDU 0x05

enum

{

	RDP_PEER_CHANNEL_TYPE_SVC = 0,

	RDP_PEER_CHANNEL_TYPE_DVC = 1

};

enum

{

	DVC_OPEN_STATE_NONE = 0,

	DVC_OPEN_STATE_SUCCEEDED = 1,

	DVC_OPEN_STATE_FAILED = 2,

	DVC_OPEN_STATE_CLOSED = 3

};

struct rdp_peer_channel

{

	WTSVirtualChannelManager* vcm;

	freerdp_peer* client;

	void* extra;

	UINT16 index;

	UINT32 channelId;

	UINT16 channelType;

	UINT32 channelFlags;

	wStream* receiveData;

	wMessageQueue* queue;

	BYTE dvc_open_state;

	UINT32 dvc_total_length;

	rdpMcsChannel* mcsChannel;

};

struct WTSVirtualChannelManager

{

	rdpRdp* rdp;

	freerdp_peer* client;

	DWORD SessionId;

	wMessageQueue* queue;

	rdpPeerChannel* drdynvc_channel;

	BYTE drdynvc_state;

	LONG dvc_channel_id_seq;

	wArrayList* dynamicVirtualChannels;

};

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionW(LPWSTR pTargetServerName,

                                                                ULONG TargetLogonId, BYTE HotkeyVk,

                                                                USHORT HotkeyModifiers);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionA(LPSTR pTargetServerName,

                                                                ULONG TargetLogonId, BYTE HotkeyVk,

                                                                USHORT HotkeyModifiers);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionExW(LPWSTR pTargetServerName,

                                                                  ULONG TargetLogonId,

                                                                  BYTE HotkeyVk,

                                                                  USHORT HotkeyModifiers,

                                                                  DWORD flags);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSStartRemoteControlSessionExA(LPSTR pTargetServerName,

                                                                  ULONG TargetLogonId,

                                                                  BYTE HotkeyVk,

                                                                  USHORT HotkeyModifiers,

                                                                  DWORD flags);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSStopRemoteControlSession(ULONG LogonId);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSConnectSessionW(ULONG LogonId, ULONG TargetLogonId,

                                                     PWSTR pPassword, BOOL bWait);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSConnectSessionA(ULONG LogonId, ULONG TargetLogonId,

                                                     PSTR pPassword, BOOL bWait);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateServersW(LPWSTR pDomainName, DWORD Reserved,

                                                       DWORD Version,

                                                       PWTS_SERVER_INFOW* ppServerInfo,

                                                       DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateServersA(LPSTR pDomainName, DWORD Reserved,

                                                       DWORD Version,

                                                       PWTS_SERVER_INFOA* ppServerInfo,

                                                       DWORD* pCount);

FREERDP_LOCAL HANDLE WINAPI FreeRDP_WTSOpenServerW(LPWSTR pServerName);

FREERDP_LOCAL HANDLE WINAPI FreeRDP_WTSOpenServerA(LPSTR pServerName);

FREERDP_LOCAL HANDLE WINAPI FreeRDP_WTSOpenServerExW(LPWSTR pServerName);

FREERDP_LOCAL HANDLE WINAPI FreeRDP_WTSOpenServerExA(LPSTR pServerName);

FREERDP_LOCAL VOID WINAPI FreeRDP_WTSCloseServer(HANDLE hServer);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateSessionsW(HANDLE hServer, DWORD Reserved,

                                                        DWORD Version,

                                                        PWTS_SESSION_INFOW* ppSessionInfo,

                                                        DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateSessionsA(HANDLE hServer, DWORD Reserved,

                                                        DWORD Version,

                                                        PWTS_SESSION_INFOA* ppSessionInfo,

                                                        DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateSessionsExW(HANDLE hServer, DWORD* pLevel,

                                                          DWORD Filter,

                                                          PWTS_SESSION_INFO_1W* ppSessionInfo,

                                                          DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateSessionsExA(HANDLE hServer, DWORD* pLevel,

                                                          DWORD Filter,

                                                          PWTS_SESSION_INFO_1A* ppSessionInfo,

                                                          DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateProcessesW(HANDLE hServer, DWORD Reserved,

                                                         DWORD Version,

                                                         PWTS_PROCESS_INFOW* ppProcessInfo,

                                                         DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateProcessesA(HANDLE hServer, DWORD Reserved,

                                                         DWORD Version,

                                                         PWTS_PROCESS_INFOA* ppProcessInfo,

                                                         DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSTerminateProcess(HANDLE hServer, DWORD ProcessId,

                                                      DWORD ExitCode);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSQuerySessionInformationW(HANDLE hServer, DWORD SessionId,

                                                              WTS_INFO_CLASS WTSInfoClass,

                                                              LPWSTR* ppBuffer,

                                                              DWORD* pBytesReturned);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSQuerySessionInformationA(HANDLE hServer, DWORD SessionId,

                                                              WTS_INFO_CLASS WTSInfoClass,

                                                              LPSTR* ppBuffer,

                                                              DWORD* pBytesReturned);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSQueryUserConfigW(LPWSTR pServerName, LPWSTR pUserName,

                                                      WTS_CONFIG_CLASS WTSConfigClass,

                                                      LPWSTR* ppBuffer, DWORD* pBytesReturned);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSQueryUserConfigA(LPSTR pServerName, LPSTR pUserName,

                                                      WTS_CONFIG_CLASS WTSConfigClass,

                                                      LPSTR* ppBuffer, DWORD* pBytesReturned);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSSetUserConfigW(LPWSTR pServerName, LPWSTR pUserName,

                                                    WTS_CONFIG_CLASS WTSConfigClass, LPWSTR pBuffer,

                                                    DWORD DataLength);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSSetUserConfigA(LPSTR pServerName, LPSTR pUserName,

                                                    WTS_CONFIG_CLASS WTSConfigClass, LPSTR pBuffer,

                                                    DWORD DataLength);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSSendMessageW(HANDLE hServer, DWORD SessionId, LPWSTR pTitle,

                                                  DWORD TitleLength, LPWSTR pMessage,

                                                  DWORD MessageLength, DWORD Style, DWORD Timeout,

                                                  DWORD* pResponse, BOOL bWait);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSSendMessageA(HANDLE hServer, DWORD SessionId, LPSTR pTitle,

                                                  DWORD TitleLength, LPSTR pMessage,

                                                  DWORD MessageLength, DWORD Style, DWORD Timeout,

                                                  DWORD* pResponse, BOOL bWait);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSDisconnectSession(HANDLE hServer, DWORD SessionId, BOOL bWait);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSLogoffSession(HANDLE hServer, DWORD SessionId, BOOL bWait);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSShutdownSystem(HANDLE hServer, DWORD ShutdownFlag);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSWaitSystemEvent(HANDLE hServer, DWORD EventMask,

                                                     DWORD* pEventFlags);

FREERDP_LOCAL HANDLE WINAPI FreeRDP_WTSVirtualChannelOpen(HANDLE hServer, DWORD SessionId,

                                                          LPSTR pVirtualName);

FREERDP_LOCAL HANDLE WINAPI FreeRDP_WTSVirtualChannelOpenEx(DWORD SessionId, LPSTR pVirtualName,

                                                            DWORD flags);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSVirtualChannelClose(HANDLE hChannelHandle);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSVirtualChannelRead(HANDLE hChannelHandle, ULONG TimeOut,

                                                        PCHAR Buffer, ULONG BufferSize,

                                                        PULONG pBytesRead);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSVirtualChannelWrite(HANDLE hChannelHandle, PCHAR Buffer,

                                                         ULONG Length, PULONG pBytesWritten);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSVirtualChannelPurgeInput(HANDLE hChannelHandle);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSVirtualChannelPurgeOutput(HANDLE hChannelHandle);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSVirtualChannelQuery(HANDLE hChannelHandle,

                                                         WTS_VIRTUAL_CLASS WtsVirtualClass,

                                                         PVOID* ppBuffer, DWORD* pBytesReturned);

FREERDP_LOCAL VOID WINAPI FreeRDP_WTSFreeMemory(PVOID pMemory);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSFreeMemoryExW(WTS_TYPE_CLASS WTSTypeClass, PVOID pMemory,

                                                   ULONG NumberOfEntries);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSFreeMemoryExA(WTS_TYPE_CLASS WTSTypeClass, PVOID pMemory,

                                                   ULONG NumberOfEntries);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSRegisterSessionNotification(HWND hWnd, DWORD dwFlags);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSUnRegisterSessionNotification(HWND hWnd);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSRegisterSessionNotificationEx(HANDLE hServer, HWND hWnd,

                                                                   DWORD dwFlags);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSUnRegisterSessionNotificationEx(HANDLE hServer, HWND hWnd);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSQueryUserToken(ULONG SessionId, PHANDLE phToken);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateProcessesExW(HANDLE hServer, DWORD* pLevel,

                                                           DWORD SessionId, LPWSTR* ppProcessInfo,

                                                           DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateProcessesExA(HANDLE hServer, DWORD* pLevel,

                                                           DWORD SessionId, LPSTR* ppProcessInfo,

                                                           DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateListenersW(HANDLE hServer, PVOID pReserved,

                                                         DWORD Reserved,

                                                         PWTSLISTENERNAMEW pListeners,

                                                         DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSEnumerateListenersA(HANDLE hServer, PVOID pReserved,

                                                         DWORD Reserved,

                                                         PWTSLISTENERNAMEA pListeners,

                                                         DWORD* pCount);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSQueryListenerConfigW(HANDLE hServer, PVOID pReserved,

                                                          DWORD Reserved, LPWSTR pListenerName,

                                                          PWTSLISTENERCONFIGW pBuffer);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSQueryListenerConfigA(HANDLE hServer, PVOID pReserved,

                                                          DWORD Reserved, LPSTR pListenerName,

                                                          PWTSLISTENERCONFIGA pBuffer);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSCreateListenerW(HANDLE hServer, PVOID pReserved,

                                                     DWORD Reserved, LPWSTR pListenerName,

                                                     PWTSLISTENERCONFIGW pBuffer, DWORD flag);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSCreateListenerA(HANDLE hServer, PVOID pReserved,

                                                     DWORD Reserved, LPSTR pListenerName,

                                                     PWTSLISTENERCONFIGA pBuffer, DWORD flag);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSSetListenerSecurityW(HANDLE hServer, PVOID pReserved,

                                                          DWORD Reserved, LPWSTR pListenerName,

                                                          SECURITY_INFORMATION SecurityInformation,

                                                          PSECURITY_DESCRIPTOR pSecurityDescriptor);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSSetListenerSecurityA(HANDLE hServer, PVOID pReserved,

                                                          DWORD Reserved, LPSTR pListenerName,

                                                          SECURITY_INFORMATION SecurityInformation,

                                                          PSECURITY_DESCRIPTOR pSecurityDescriptor);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSGetListenerSecurityW(HANDLE hServer, PVOID pReserved,

                                                          DWORD Reserved, LPWSTR pListenerName,

                                                          SECURITY_INFORMATION SecurityInformation,

                                                          PSECURITY_DESCRIPTOR pSecurityDescriptor,

                                                          DWORD nLength, LPDWORD lpnLengthNeeded);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSGetListenerSecurityA(HANDLE hServer, PVOID pReserved,

                                                          DWORD Reserved, LPSTR pListenerName,

                                                          SECURITY_INFORMATION SecurityInformation,

                                                          PSECURITY_DESCRIPTOR pSecurityDescriptor,

                                                          DWORD nLength, LPDWORD lpnLengthNeeded);

FREERDP_LOCAL BOOL CDECL FreeRDP_WTSEnableChildSessions(BOOL bEnable);

FREERDP_LOCAL BOOL CDECL FreeRDP_WTSIsChildSessionsEnabled(PBOOL pbEnabled);

FREERDP_LOCAL BOOL CDECL FreeRDP_WTSGetChildSessionId(PULONG pSessionId);

FREERDP_LOCAL DWORD WINAPI FreeRDP_WTSGetActiveConsoleSessionId(void);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSLogoffUser(HANDLE hServer);

FREERDP_LOCAL BOOL WINAPI FreeRDP_WTSLogonUser(HANDLE hServer, LPCSTR username, LPCSTR password,

                                               LPCSTR domain);

#endif /* FREERDP_LIB_CORE_SERVER_H */