/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * FreeRDP Proxy Server

 *

 * Copyright 2019 Mati Shabtay <matishabtay@gmail.com>

 * Copyright 2019 Kobi Mizrachi <kmizrachi18@gmail.com>

 * Copyright 2019 Idan Freiberg <speidy@gmail.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_SERVER_PROXY_PFCONTEXT_H

#define FREERDP_SERVER_PROXY_PFCONTEXT_H

#include <freerdp/freerdp.h>

#include <freerdp/channels/wtsvc.h>

#include <freerdp/client/rdpei.h>

#include <freerdp/client/rail.h>

#include <freerdp/server/rail.h>

#include <freerdp/client/rdpgfx.h>

#include <freerdp/server/rdpgfx.h>

#include <freerdp/client/disp.h>

#include <freerdp/server/disp.h>

#include <freerdp/server/cliprdr.h>

#include <freerdp/server/rdpsnd.h>

#include "pf_config.h"

#include "pf_server.h"

typedef struct proxy_data proxyData;

/**

 * Wraps rdpContext and holds the state for the proxy's server.

 */

struct p_server_context

{

	rdpContext context;

	proxyData* pdata;

	HANDLE vcm;

	HANDLE dynvcReady;

	RailServerContext* rail;

	RdpgfxServerContext* gfx;

	DispServerContext* disp;

	CliprdrServerContext* cliprdr;

	RdpsndServerContext* rdpsnd;

	HANDLE* vc_handles; /* static virtual channels open handles */

	wHashTable* vc_ids; /* channel_name -> channel_id map */

};

typedef struct p_server_context pServerContext;

/**

 * Wraps rdpContext and holds the state for the proxy's client.

 */

struct p_client_context

{

	rdpContext context;

	proxyData* pdata;

	RdpeiClientContext* rdpei;

	RdpgfxClientContext* gfx_proxy;

	RdpgfxClientContext* gfx_decoder;

	DispClientContext* disp;

	CliprdrClientContext* cliprdr;

	RailClientContext* rail;

	/*

	 * In a case when freerdp_connect fails,

	 * Used for NLA fallback feature, to check if the server should close the connection.

	 * When it is set to TRUE, proxy's client knows it shouldn't signal the server thread to

	 * closed the connection when pf_client_post_disconnect is called, because it is trying to

	 * connect reconnect without NLA. It must be set to TRUE before the first try, and to FALSE

	 * after the connection fully established, to ensure graceful shutdown of the connection

	 * when it will be closed.

	 */

	BOOL allow_next_conn_failure;

	/* session capture */

	char* frames_dir;

	UINT64 frames_count;

	wHashTable* vc_ids; /* channel_name -> channel_id map */

};

typedef struct p_client_context pClientContext;

/**

 * Holds data common to both sides of a proxy's session.

 */

struct proxy_data

{

	proxyConfig* config;

	pServerContext* ps;

	pClientContext* pc;

	HANDLE abort_event;

	HANDLE client_thread;

	HANDLE gfx_server_ready;

	char* session_id;

	/* used to external modules to store per-session info */

	wHashTable* modules_info;

};

BOOL pf_context_copy_settings(rdpSettings* dst, const rdpSettings* src);

BOOL pf_context_init_server_context(freerdp_peer* client);

pClientContext* pf_context_create_client_context(rdpSettings* clientSettings);

proxyData* proxy_data_new(void);

void proxy_data_set_client_context(proxyData* pdata, pClientContext* context);

void proxy_data_set_server_context(proxyData* pdata, pServerContext* context);

void proxy_data_free(proxyData* pdata);

BOOL proxy_data_shall_disconnect(proxyData* pdata);

void proxy_data_abort_connect(proxyData* pdata);

#endif /* FREERDP_SERVER_PROXY_PFCONTEXT_H */