/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * FreeRDP Interface
 *
 * Copyright 2009-2011 Jay Sorg
 * Copyright 2015 Thincast Technologies GmbH
 * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef FREERDP_H
#define FREERDP_H

typedef struct rdp_rdp rdpRdp;
typedef struct rdp_gdi rdpGdi;
typedef struct rdp_rail rdpRail;
typedef struct rdp_cache rdpCache;
typedef struct rdp_channels rdpChannels;
typedef struct rdp_graphics rdpGraphics;
typedef struct rdp_metrics rdpMetrics;
typedef struct rdp_codecs rdpCodecs;

typedef struct rdp_freerdp freerdp;
typedef struct rdp_context rdpContext;
typedef struct rdp_freerdp_peer freerdp_peer;

typedef struct rdp_client_context rdpClientContext;
typedef struct rdp_client_entry_points_v1 RDP_CLIENT_ENTRY_POINTS_V1;
typedef RDP_CLIENT_ENTRY_POINTS_V1 RDP_CLIENT_ENTRY_POINTS;

#include <freerdp/api.h>
#include <freerdp/types.h>
#include <freerdp/error.h>
#include <freerdp/event.h>
#include <freerdp/codecs.h>
#include <freerdp/metrics.h>
#include <freerdp/settings.h>
#include <freerdp/extension.h>

#include <winpr/stream.h>

#include <freerdp/input.h>
#include <freerdp/update.h>
#include <freerdp/message.h>
#include <freerdp/autodetect.h>

#ifdef __cplusplus
extern "C" {
#endif

/* Flags used by certificate callbacks */
#define VERIFY_CERT_FLAG_NONE     0x00
#define VERIFY_CERT_FLAG_LEGACY   0x02
#define VERIFY_CERT_FLAG_REDIRECT 0x10
#define VERIFY_CERT_FLAG_GATEWAY  0x20
#define VERIFY_CERT_FLAG_CHANGED  0x40
#define VERIFY_CERT_FLAG_MISMATCH 0x80

typedef BOOL (*pContextNew)(freerdp* instance, rdpContext* context);
typedef void (*pContextFree)(freerdp* instance, rdpContext* context);

typedef BOOL (*pPreConnect)(freerdp* instance);
typedef BOOL (*pPostConnect)(freerdp* instance);
typedef void (*pPostDisconnect)(freerdp* instance);
typedef BOOL (*pAuthenticate)(freerdp* instance, char** username,
                              char** password, char** domain);

/** @brief Callback used if user interaction is required to accept
 *         an unknown certificate.
 *
 *  @deprecated Use pVerifyCertificateEx
 *  @param common_name      The certificate registered hostname.
 *  @param subject          The common name of the certificate.
 *  @param issuer           The issuer of the certificate.
 *  @param fingerprint      The fingerprint of the certificate.
 *  @param host_mismatch    A flag indicating the certificate
 *                          subject does not match the host connecting to.
 *
 *  @return 1 to accept and store a certificate, 2 to accept
 *          a certificate only for this session, 0 otherwise.
 */
typedef DWORD (*pVerifyCertificate)(freerdp* instance,
                                    const char* common_name,
                                    const char* subject,
                                    const char* issuer,
                                    const char* fingerprint,
                                    BOOL host_mismatch);

/** @brief Callback used if user interaction is required to accept
 *         an unknown certificate.
 *
 *  @param host             The hostname connecting to.
 *  @param port             The port connecting to.
 *  @param common_name      The certificate registered hostname.
 *  @param subject          The common name of the certificate.
 *  @param issuer           The issuer of the certificate.
 *  @param fingerprint      The fingerprint of the certificate.
 *  @param flags            Flags of type VERIFY_CERT_FLAG*
 *
 *  @return 1 to accept and store a certificate, 2 to accept
 *          a certificate only for this session, 0 otherwise.
 */
typedef DWORD (*pVerifyCertificateEx)(freerdp* instance,
                                      const char* host,
                                      UINT16 port,
                                      const char* common_name,
                                      const char* subject,
                                      const char* issuer,
                                      const char* fingerprint,
                                      DWORD flags);

/** @brief Callback used if user interaction is required to accept
 *         a changed certificate.
 *
 *  @deprecated Use pVerifyChangedCertificateEx
 *  @param common_name      The certificate registered hostname.
 *  @param subject          The common name of the new certificate.
 *  @param issuer           The issuer of the new certificate.
 *  @param fingerprint      The fingerprint of the new certificate.
 *  @param old_subject      The common name of the old certificate.
 *  @param old_issuer       The issuer of the new certificate.
 *  @param old_fingerprint  The fingerprint of the old certificate.
 *
 *  @return 1 to accept and store a certificate, 2 to accept
 *          a certificate only for this session, 0 otherwise.
 */

typedef DWORD (*pVerifyChangedCertificate)(freerdp* instance,
        const char* common_name,
        const char* subject,
        const char* issuer,
        const char* new_fingerprint,
        const char* old_subject,
        const char* old_issuer,
        const char* old_fingerprint);

/** @brief Callback used if user interaction is required to accept
 *         a changed certificate.
 *
 *  @param host             The hostname connecting to.
 *  @param port             The port connecting to.
 *  @param common_name      The certificate registered hostname.
 *  @param subject          The common name of the new certificate.
 *  @param issuer           The issuer of the new certificate.
 *  @param fingerprint      The fingerprint of the new certificate.
 *  @param old_subject      The common name of the old certificate.
 *  @param old_issuer       The issuer of the new certificate.
 *  @param old_fingerprint  The fingerprint of the old certificate.
 *  @param flags            Flags of type VERIFY_CERT_FLAG*
 *
 *  @return 1 to accept and store a certificate, 2 to accept
 *          a certificate only for this session, 0 otherwise.
 */

typedef DWORD (*pVerifyChangedCertificateEx)(freerdp* instance,
        const char* host,
        UINT16 port,
        const char* common_name,
        const char* subject,
        const char* issuer,
        const char* new_fingerprint,
        const char* old_subject,
        const char* old_issuer,
        const char* old_fingerprint,
        DWORD flags);

/** @brief Callback used if user interaction is required to accept
 *         a certificate.
 *
 *  @param instance         Pointer to the freerdp instance.
 *  @param data             Pointer to certificate data in PEM format.
 *  @param length           The length of the certificate data.
 *  @param hostname         The hostname connecting to.
 *  @param port             The port connecting to.
 *  @param flags            Flags of type VERIFY_CERT_FLAG*
 *
 *  @return 1 to accept and store a certificate, 2 to accept
 *          a certificate only for this session, 0 otherwise.
 */
typedef int (*pVerifyX509Certificate)(freerdp* instance, const BYTE* data,
                                      size_t length, const char* hostname,
                                      UINT16 port, DWORD flags);

typedef int (*pLogonErrorInfo)(freerdp* instance, UINT32 data, UINT32 type);

typedef int (*pSendChannelData)(freerdp* instance, UINT16 channelId, BYTE* data,
                                int size);
typedef int (*pReceiveChannelData)(freerdp* instance, UINT16 channelId,
                                   BYTE* data, int size, int flags, int totalSize);

typedef BOOL (*pPresentGatewayMessage)(freerdp* instance, UINT32 type, BOOL isDisplayMandatory,
                                       BOOL isConsentMandatory, size_t length, const WCHAR* message);

/**
 * Defines the context for a given instance of RDP connection.
 * It is embedded in the rdp_freerdp structure, and allocated by a call to freerdp_context_new().
 * It is deallocated by a call to freerdp_context_free().
 */
struct rdp_context
{
	ALIGN64 freerdp* instance; /**< (offset 0)
						  Pointer to a rdp_freerdp structure.
						  This is a back-link to retrieve the freerdp instance from the context.
						  It is set by the freerdp_context_new() function */
	ALIGN64 freerdp_peer* peer; /**< (offset 1)
						   Pointer to the client peer.
						   This is set by a call to freerdp_peer_context_new() during peer initialization.
						   This field is used only on the server side. */
	ALIGN64 BOOL ServerMode; /**< (offset 2) true when context is in server mode */

	ALIGN64 UINT32 LastError; /* 3 */

	UINT64 paddingA[16 - 4]; /* 4 */

	ALIGN64 int argc;	/**< (offset 16)
				   Number of arguments given to the program at launch time.
				   Used to keep this data available and used later on, typically just before connection initialization.
				   @see freerdp_parse_args() */
	ALIGN64 char** argv; /**< (offset 17)
					List of arguments given to the program at launch time.
					Used to keep this data available and used later on, typically just before connection initialization.
					@see freerdp_parse_args() */

	ALIGN64 wPubSub* pubSub; /* (offset 18) */

	ALIGN64 HANDLE channelErrorEvent; /* (offset 19)*/
	ALIGN64 UINT channelErrorNum; /*(offset 20)*/
	ALIGN64 char* errorDescription; /*(offset 21)*/

	UINT64 paddingB[32 - 22]; /* 22 */

	ALIGN64 rdpRdp* rdp; /**< (offset 32)
					Pointer to a rdp_rdp structure used to keep the connection's parameters.
					It is allocated by freerdp_context_new() and deallocated by freerdp_context_free(), at the same
					time that this rdp_context structure - there is no need to specifically allocate/deallocate this. */
	ALIGN64 rdpGdi* gdi; /**< (offset 33)
					Pointer to a rdp_gdi structure used to keep the gdi settings.
					It is allocated by gdi_init() and deallocated by gdi_free().
					It must be deallocated before deallocating this rdp_context structure. */
	ALIGN64 rdpRail* rail; /* 34 */
	ALIGN64 rdpCache* cache; /* 35 */
	ALIGN64 rdpChannels* channels; /* 36 */
	ALIGN64 rdpGraphics* graphics; /* 37 */
	ALIGN64 rdpInput* input; /* 38 */
	ALIGN64 rdpUpdate* update; /* 39 */
	ALIGN64 rdpSettings* settings; /* 40 */
	ALIGN64 rdpMetrics* metrics; /* 41 */
	ALIGN64 rdpCodecs* codecs; /* 42 */
	ALIGN64 rdpAutoDetect* autodetect; /* 43 */
	ALIGN64 HANDLE abortEvent; /* 44 */
	ALIGN64 int  disconnectUltimatum; /* 45 */
	UINT64 paddingC[64 - 46]; /* 46 */

	UINT64 paddingD[96 - 64]; /* 64 */
	UINT64 paddingE[128 - 96]; /* 96 */
};

/**
 *  Defines the possible disconnect reasons in the MCS Disconnect Provider
 *  Ultimatum PDU
 */

enum Disconnect_Ultimatum
{
	Disconnect_Ultimatum_domain_disconnected = 0,
	Disconnect_Ultimatum_provider_initiated = 1,
	Disconnect_Ultimatum_token_purged = 2,
	Disconnect_Ultimatum_user_requested = 3,
	Disconnect_Ultimatum_channel_purged = 4
};

#include <freerdp/client.h>

/** Defines the options for a given instance of RDP connection.
 *  This is built by the client and given to the FreeRDP library to create the connection
 *  with the expected options.
 *  It is allocated by a call to freerdp_new() and deallocated by a call to freerdp_free().
 *  Some of its content need specific allocation/deallocation - see field description for details.
 */
struct rdp_freerdp
{
	ALIGN64 rdpContext* context; /**< (offset 0)
							  Pointer to a rdpContext structure.
							  Client applications can use the ContextSize field to register a context bigger than the rdpContext
							  structure. This allow clients to use additional context information.
							  When using this capability, client application should ALWAYS declare their structure with the
							  rdpContext field first, and any additional content following it.
							  Can be allocated by a call to freerdp_context_new().
							  Must be deallocated by a call to freerdp_context_free() before deallocating the current instance. */

	ALIGN64 RDP_CLIENT_ENTRY_POINTS* pClientEntryPoints;

	UINT64 paddingA[16 - 2]; /* 2 */

	ALIGN64 rdpInput* input; /* (offset 16)
						Input handle for the connection.
						Will be initialized by a call to freerdp_context_new() */
	ALIGN64 rdpUpdate* update; /* (offset 17)
						  Update display parameters. Used to register display events callbacks and settings.
						  Will be initialized by a call to freerdp_context_new() */
	ALIGN64 rdpSettings* settings; /**< (offset 18)
								Pointer to a rdpSettings structure. Will be used to maintain the required RDP settings.
								Will be initialized by a call to freerdp_context_new() */
	ALIGN64 rdpAutoDetect* autodetect; /* (offset 19)
									Auto-Detect handle for the connection.
									Will be initialized by a call to freerdp_context_new() */
	UINT64 paddingB[32 - 20]; /* 20 */

	ALIGN64 size_t ContextSize; /* (offset 32)
							Specifies the size of the 'context' field. freerdp_context_new() will use this size to allocate the context buffer.
							freerdp_new() sets it to sizeof(rdpContext).
							If modifying it, there should always be a minimum of sizeof(rdpContext), as the freerdp library will assume it can use the
							'context' field to set the required informations in it.
							Clients will typically make it bigger, and use a context structure embedding the rdpContext, and
							adding additional information after that.
						 */

	ALIGN64 pContextNew ContextNew; /**< (offset 33)
								 Callback for context allocation
								 Can be set before calling freerdp_context_new() to have it executed after allocation and initialization.
								 Must be set to NULL if not needed. */

	ALIGN64 pContextFree ContextFree; /**< (offset 34)
								   Callback for context deallocation
								   Can be set before calling freerdp_context_free() to have it executed before deallocation.
								   Must be set to NULL if not needed. */
	UINT64 paddingC[47 - 35]; /* 35 */

	ALIGN64 UINT ConnectionCallbackState; /* 47 */

	ALIGN64 pPreConnect PreConnect; /**< (offset 48)
								 Callback for pre-connect operations.
								 Can be set before calling freerdp_connect() to have it executed before the actual connection happens.
								 Must be set to NULL if not needed. */

	ALIGN64 pPostConnect PostConnect; /**< (offset 49)
								   Callback for post-connect operations.
								   Can be set before calling freerdp_connect() to have it executed after the actual connection has succeeded.
								   Must be set to NULL if not needed. */

	ALIGN64 pAuthenticate Authenticate; /**< (offset 50)
									 Callback for authentication.
									 It is used to get the username/password when it was not provided at connection time. */
	ALIGN64 pVerifyCertificate VerifyCertificate; /**< (offset 51)
											   Callback for certificate validation.
											   Used to verify that an unknown certificate is trusted.
 DEPRECATED: Use VerifyChangedCertificateEx*/
	ALIGN64 pVerifyChangedCertificate VerifyChangedCertificate; /**< (offset 52)
															 Callback for changed certificate validation.
															 Used when a certificate differs from stored fingerprint.
 DEPRECATED: Use VerifyChangedCertificateEx */

	ALIGN64 pVerifyX509Certificate
	VerifyX509Certificate;  /**< (offset 53)  Callback for X509 certificate verification (PEM format) */

	ALIGN64 pLogonErrorInfo
	LogonErrorInfo; /**< (offset 54)  Callback for logon error info, important for logon system messages with RemoteApp */

	ALIGN64 pPostDisconnect PostDisconnect; /**< (offset 55)
																						Callback for cleaning up resources allocated
																						by connect callbacks. */

	ALIGN64 pAuthenticate GatewayAuthenticate; /**< (offset 56)
									 Callback for gateway authentication.
									 It is used to get the username/password when it was not provided at connection time. */

	ALIGN64 pPresentGatewayMessage PresentGatewayMessage;/**< (offset 57)
									 Callback for gateway consent messages.
									 It is used to present consent messages to the user. */

	UINT64 paddingD[64 - 58]; /* 58 */

	ALIGN64 pSendChannelData SendChannelData; /* (offset 64)
										 Callback for sending data to a channel.
										 By default, it is set by freerdp_new() to freerdp_send_channel_data(), which eventually calls
										 freerdp_channel_send() */
	ALIGN64 pReceiveChannelData ReceiveChannelData; /* (offset 65)
											   Callback for receiving data from a channel.
											   This is called by freerdp_channel_process() (if not NULL).
											   Clients will typically use a function that calls freerdp_channels_data() to perform the needed tasks. */

	ALIGN64 pVerifyCertificateEx VerifyCertificateEx; /**< (offset 66)
											   Callback for certificate validation.
											   Used to verify that an unknown certificate is trusted. */
	ALIGN64 pVerifyChangedCertificateEx VerifyChangedCertificateEx; /**< (offset 67)
															 Callback for changed certificate validation.
															 Used when a certificate differs from stored fingerprint. */
	UINT64 paddingE[80 - 68]; /* 68 */
};

struct rdp_channel_handles
{
	wListDictionary* init;
	wListDictionary* open;
};
typedef struct rdp_channel_handles rdpChannelHandles;

FREERDP_API BOOL freerdp_context_new(freerdp* instance);
FREERDP_API void freerdp_context_free(freerdp* instance);

FREERDP_API BOOL freerdp_connect(freerdp* instance);
FREERDP_API BOOL freerdp_abort_connect(freerdp* instance);
FREERDP_API BOOL freerdp_shall_disconnect(freerdp* instance);
FREERDP_API BOOL freerdp_disconnect(freerdp* instance);

FREERDP_API BOOL freerdp_disconnect_before_reconnect(freerdp* instance);
FREERDP_API BOOL freerdp_reconnect(freerdp* instance);

FREERDP_API UINT freerdp_channel_add_init_handle_data(rdpChannelHandles* handles, void* pInitHandle,
        void* pUserData);
FREERDP_API void* freerdp_channel_get_init_handle_data(rdpChannelHandles* handles,
        void* pInitHandle);
FREERDP_API void freerdp_channel_remove_init_handle_data(rdpChannelHandles* handles,
        void* pInitHandle);

FREERDP_API UINT freerdp_channel_add_open_handle_data(rdpChannelHandles* handles, DWORD openHandle,
        void* pUserData);
FREERDP_API void* freerdp_channel_get_open_handle_data(rdpChannelHandles* handles,
        DWORD openHandle);
FREERDP_API void freerdp_channel_remove_open_handle_data(rdpChannelHandles* handles,
        DWORD openHandle);

FREERDP_API UINT freerdp_channels_attach(freerdp* instance);
FREERDP_API UINT freerdp_channels_detach(freerdp* instance);

FREERDP_API BOOL freerdp_get_fds(freerdp* instance, void** rfds, int* rcount,
                                 void** wfds, int* wcount);
FREERDP_API BOOL freerdp_check_fds(freerdp* instance);

FREERDP_API DWORD freerdp_get_event_handles(rdpContext* context, HANDLE* events,
        DWORD count);
FREERDP_API BOOL freerdp_check_event_handles(rdpContext* context);

FREERDP_API wMessageQueue* freerdp_get_message_queue(freerdp* instance,
        DWORD id);
FREERDP_API HANDLE freerdp_get_message_queue_event_handle(freerdp* instance,
        DWORD id);
FREERDP_API int freerdp_message_queue_process_message(freerdp* instance,
        DWORD id, wMessage* message);
FREERDP_API int freerdp_message_queue_process_pending_messages(
    freerdp* instance, DWORD id);

FREERDP_API UINT32 freerdp_error_info(freerdp* instance);
FREERDP_API void freerdp_set_error_info(rdpRdp* rdp, UINT32 error);

FREERDP_API void freerdp_get_version(int* major, int* minor, int* revision);
FREERDP_API const char* freerdp_get_version_string(void);
FREERDP_API const char* freerdp_get_build_date(void);
FREERDP_API const char* freerdp_get_build_revision(void);
FREERDP_API const char* freerdp_get_build_config(void);

FREERDP_API freerdp* freerdp_new(void);
FREERDP_API void freerdp_free(freerdp* instance);

FREERDP_API BOOL freerdp_focus_required(freerdp* instance);
FREERDP_API void freerdp_set_focus(freerdp* instance);

FREERDP_API int freerdp_get_disconnect_ultimatum(rdpContext* context);

FREERDP_API UINT32 freerdp_get_last_error(rdpContext* context);
FREERDP_API const char* freerdp_get_last_error_name(UINT32 error);
FREERDP_API const char* freerdp_get_last_error_string(UINT32 error);
FREERDP_API void freerdp_set_last_error(rdpContext* context, UINT32 lastError);

FREERDP_API const char* freerdp_get_logon_error_info_type(UINT32 type);
FREERDP_API const char* freerdp_get_logon_error_info_data(UINT32 data);

FREERDP_API ULONG freerdp_get_transport_sent(rdpContext* context,
        BOOL resetCount);

FREERDP_API BOOL freerdp_nla_impersonate(rdpContext* context);
FREERDP_API BOOL freerdp_nla_revert_to_self(rdpContext* context);

FREERDP_API void clearChannelError(rdpContext* context);
FREERDP_API HANDLE getChannelErrorEventHandle(rdpContext* context);
FREERDP_API UINT getChannelError(rdpContext* context);
FREERDP_API const char* getChannelErrorDescription(rdpContext* context);
FREERDP_API void setChannelError(rdpContext* context, UINT errorNum,
                                 char* description);
FREERDP_API BOOL checkChannelErrorEvent(rdpContext* context);

#ifdef __cplusplus
}
#endif

#endif /* FREERDP_H */
