131 lines
3.7 KiB
C
131 lines
3.7 KiB
C
/**
|
|
* 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"
|
|
|
|
#define PROXY_SESSION_ID_LENGTH 32
|
|
|
|
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;
|
|
|
|
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[PROXY_SESSION_ID_LENGTH + 1];
|
|
|
|
/* 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 */
|