2012-09-20 07:51:34 +04:00
|
|
|
/**
|
|
|
|
* xrdp: A Remote Desktop Protocol server.
|
|
|
|
*
|
2014-03-02 11:26:40 +04:00
|
|
|
* Copyright (C) Jay Sorg 2004-2014
|
2012-09-20 07:51:34 +04:00
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*
|
|
|
|
* generic transport
|
|
|
|
*/
|
2008-08-22 10:08:11 +04:00
|
|
|
|
|
|
|
#if !defined(TRANS_H)
|
|
|
|
#define TRANS_H
|
|
|
|
|
|
|
|
#include "arch.h"
|
|
|
|
#include "parse.h"
|
|
|
|
|
2019-06-30 09:59:18 +03:00
|
|
|
#define TRANS_MODE_TCP 1 /* tcp6 if defined, else tcp4 */
|
2010-05-04 11:37:22 +04:00
|
|
|
#define TRANS_MODE_UNIX 2
|
2017-09-22 02:13:05 +03:00
|
|
|
#define TRANS_MODE_VSOCK 3
|
2019-06-30 09:59:18 +03:00
|
|
|
#define TRANS_MODE_TCP4 4 /* tcp4 only */
|
|
|
|
#define TRANS_MODE_TCP6 6 /* tcp6 only */
|
2017-09-22 02:13:05 +03:00
|
|
|
|
2010-05-04 11:37:22 +04:00
|
|
|
#define TRANS_TYPE_LISTENER 1
|
|
|
|
#define TRANS_TYPE_SERVER 2
|
|
|
|
#define TRANS_TYPE_CLIENT 3
|
|
|
|
|
|
|
|
#define TRANS_STATUS_DOWN 0
|
|
|
|
#define TRANS_STATUS_UP 1
|
|
|
|
|
2008-08-22 10:08:11 +04:00
|
|
|
struct trans; /* forward declaration */
|
2014-07-23 16:31:45 +04:00
|
|
|
struct xrdp_tls;
|
2008-08-22 10:08:11 +04:00
|
|
|
|
2021-03-06 18:45:27 +03:00
|
|
|
typedef int (*ttrans_data_in)(struct trans *self);
|
|
|
|
typedef int (*ttrans_conn_in)(struct trans *self,
|
|
|
|
struct trans *new_self);
|
2017-03-12 19:35:00 +03:00
|
|
|
typedef int (*tis_term)(void);
|
|
|
|
typedef int (*trans_recv_proc) (struct trans *self, char *ptr, int len);
|
|
|
|
typedef int (*trans_send_proc) (struct trans *self, const char *data, int len);
|
|
|
|
typedef int (*trans_can_recv_proc) (struct trans *self, int sck, int millis);
|
2008-08-22 10:08:11 +04:00
|
|
|
|
2015-07-06 09:14:46 +03:00
|
|
|
/* optional source info */
|
|
|
|
|
2020-08-07 13:48:22 +03:00
|
|
|
enum xrdp_source
|
|
|
|
{
|
|
|
|
XRDP_SOURCE_NONE = 0,
|
|
|
|
XRDP_SOURCE_CLIENT,
|
|
|
|
XRDP_SOURCE_SESMAN,
|
|
|
|
XRDP_SOURCE_CHANSRV,
|
|
|
|
XRDP_SOURCE_MOD,
|
|
|
|
|
|
|
|
XRDP_SOURCE_MAX_COUNT
|
|
|
|
};
|
2015-07-06 09:14:46 +03:00
|
|
|
|
2020-08-07 13:48:22 +03:00
|
|
|
/*
|
|
|
|
* @brief Provide flow control mechanism for (primarily) xrdp
|
|
|
|
*
|
|
|
|
* There is one of these data structures per-program.
|
|
|
|
*
|
|
|
|
* While input is being read from a 'struct trans' and processed, the
|
|
|
|
* cur_source member is set to the my_source member from the transport.
|
|
|
|
* During this processing, trans_write_copy() may be called to send output
|
2022-09-03 02:48:01 +03:00
|
|
|
* on another struct trans. If this happens, and the output needs to be
|
2020-08-07 13:48:22 +03:00
|
|
|
* buffered, trans_write_copy() can add the number of bytes generated by
|
|
|
|
* the input trans to the source field for the cur_source. This allows us to
|
|
|
|
* see how much output has been buffered for each input source.
|
|
|
|
*
|
|
|
|
* When the program assembles 'struct trans' objects to scan for input
|
|
|
|
* (normally in trans_get_wait_objs()), it is able to see how much buffered
|
|
|
|
* output is registered for each input. Inputs which have too much buffered
|
|
|
|
* output owing are skipped, and not considered for input.
|
|
|
|
*
|
|
|
|
* This provides a simple means of providing back-pressure on an input
|
|
|
|
* where the data it is providing is being processed and then sent out on
|
|
|
|
* a much slower link.
|
|
|
|
*/
|
2015-07-06 09:14:46 +03:00
|
|
|
struct source_info
|
|
|
|
{
|
2020-08-07 13:48:22 +03:00
|
|
|
enum xrdp_source cur_source;
|
|
|
|
int source[XRDP_SOURCE_MAX_COUNT];
|
2015-07-06 09:14:46 +03:00
|
|
|
};
|
|
|
|
|
2008-08-22 10:08:11 +04:00
|
|
|
struct trans
|
|
|
|
{
|
2014-04-21 07:24:05 +04:00
|
|
|
tbus sck; /* socket handle */
|
2017-09-22 02:13:05 +03:00
|
|
|
int mode; /* 1 tcp, 2 unix socket, 3 vsock */
|
2014-04-21 07:24:05 +04:00
|
|
|
int status;
|
|
|
|
int type1; /* 1 listener 2 server 3 client */
|
|
|
|
ttrans_data_in trans_data_in;
|
|
|
|
ttrans_conn_in trans_conn_in;
|
2021-03-06 18:45:27 +03:00
|
|
|
void *callback_data;
|
2022-12-07 12:44:56 +03:00
|
|
|
unsigned int header_size;
|
2021-03-06 18:45:27 +03:00
|
|
|
struct stream *in_s;
|
|
|
|
struct stream *out_s;
|
|
|
|
char *listen_filename;
|
2014-04-21 07:24:05 +04:00
|
|
|
tis_term is_term; /* used to test for exit */
|
2021-03-06 18:45:27 +03:00
|
|
|
struct stream *wait_s;
|
2014-04-21 07:24:05 +04:00
|
|
|
int no_stream_init_on_data_in;
|
|
|
|
int extra_flags; /* user defined */
|
2022-02-14 12:14:22 +03:00
|
|
|
void *extra_data; /* user defined */
|
|
|
|
void (*extra_destructor)(struct trans *); /* user defined */
|
|
|
|
|
2014-11-26 05:55:37 +03:00
|
|
|
struct ssl_tls *tls;
|
2016-09-09 09:42:04 +03:00
|
|
|
const char *ssl_protocol; /* e.g. TLSv1, TLSv1.1, TLSv1.2, unknown */
|
|
|
|
const char *cipher_name; /* e.g. AES256-GCM-SHA384 */
|
2014-11-22 07:49:01 +03:00
|
|
|
trans_recv_proc trans_recv;
|
|
|
|
trans_send_proc trans_send;
|
|
|
|
trans_can_recv_proc trans_can_recv;
|
2015-07-06 09:14:46 +03:00
|
|
|
struct source_info *si;
|
2020-08-07 13:48:22 +03:00
|
|
|
enum xrdp_source my_source;
|
2008-08-22 10:08:11 +04:00
|
|
|
};
|
|
|
|
|
2021-03-06 18:45:27 +03:00
|
|
|
struct trans *
|
2008-08-22 10:08:11 +04:00
|
|
|
trans_create(int mode, int in_size, int out_size);
|
2017-03-12 19:35:00 +03:00
|
|
|
void
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_delete(struct trans *self);
|
2017-06-22 21:38:37 +03:00
|
|
|
void
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_delete_from_child(struct trans *self);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_get_wait_objs(struct trans *self, tbus *objs, int *count);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2013-12-05 03:26:23 +04:00
|
|
|
trans_get_wait_objs_rw(struct trans *self,
|
|
|
|
tbus *robjs, int *rcount,
|
2015-07-13 11:10:48 +03:00
|
|
|
tbus *wobjs, int *wcount, int *timeout);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_check_wait_objs(struct trans *self);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_force_read_s(struct trans *self, struct stream *in_s, int size);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_force_write_s(struct trans *self, struct stream *out_s);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_force_read(struct trans *self, int size);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_force_write(struct trans *self);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_write_copy(struct trans *self);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_write_copy_s(struct trans *self, struct stream *out_s);
|
Rework transport connect logic
There are a number of ways the existing transport connect logic in
trans_connect could be improved for POSIX compatibility, and also
slightly tidied up:-
1) The same socket is re-used for multiple connect attempts following
failure which isn't behaviour defined by POSIX.1-2017 (although it
works on Linux).
2) An asynchronous connect is started, and then after a short
delay connect() is called again on the same socket. POSIX.1-2017
is clear that in this situation EALREADY is returned before the
connection is established, but is silent on the behaviour expected
when the connection is established. Returning success is an option,
but so is returning EISCONN. The current code assumes the connect()
call will succeed.
3) The code contains two virtually identical, quite complex loops for
TCP and UNIX sockets, differing only in the calls to create a socket
and connect it.
4) trans_connect() contains looping and retry logic, but this isn't
seen as sufficient by the chansrv connect code in xrdp/xrdp_mm.c and
the Xorg connect code in xup/xup.c. Both of these implement their own
looping and retry logic on top of the logic in trans_connect(),
resulting in slightly unpredictable behaviour with regard to
timeouts.
5) A socket number can technically be zero, but in a couple of places
this isn't allowed for.
This PR attempts to correct the implementation of trans_connect(),
and also to simplify the areas it is called from.
As part of the PR, the signature of the server_is_term member of the
xrdp module interface is changed to match the signature expected by the
is_term member of a struct trans. This allows for trans_connect()
in xrdp modules to directly access g_is_term() within the main xrdp
executable. At the moment this functionality is only used by the xup
module.
2022-03-31 16:33:06 +03:00
|
|
|
/**
|
|
|
|
* Connect the transport to the specified destination
|
|
|
|
*
|
|
|
|
* @param self Transport
|
|
|
|
* @param server Destination server (TCP transports only)
|
|
|
|
* @param port TCP port, or UNIX socket to connect to
|
|
|
|
* @param timeout in milli-seconds for the operation
|
|
|
|
* @return 0 for success
|
|
|
|
*
|
|
|
|
* Multiple connection attempts may be made within the timeout period.
|
|
|
|
*
|
|
|
|
* If the operation is successful, 0 is returned and self->status will
|
|
|
|
* be TRANS_STATUS_UP
|
|
|
|
*/
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-03-06 18:45:27 +03:00
|
|
|
trans_connect(struct trans *self, const char *server, const char *port,
|
2008-08-22 10:08:11 +04:00
|
|
|
int timeout);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-04-19 18:32:04 +03:00
|
|
|
trans_listen_address(struct trans *self, const char *port, const char *address);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2021-04-19 18:32:04 +03:00
|
|
|
trans_listen(struct trans *self, const char *port);
|
2021-03-06 18:45:27 +03:00
|
|
|
struct stream *
|
|
|
|
trans_get_in_s(struct trans *self);
|
|
|
|
struct stream *
|
|
|
|
trans_get_out_s(struct trans *self, int size);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2016-08-25 21:20:47 +03:00
|
|
|
trans_set_tls_mode(struct trans *self, const char *key, const char *cert,
|
2016-12-13 09:49:13 +03:00
|
|
|
long ssl_protocols, const char *tls_ciphers);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2014-07-23 16:31:45 +04:00
|
|
|
trans_shutdown_tls_mode(struct trans *self);
|
2017-03-12 19:35:00 +03:00
|
|
|
int
|
2014-07-23 16:31:45 +04:00
|
|
|
trans_tcp_force_read_s(struct trans *self, struct stream *in_s, int size);
|
2008-08-22 10:08:11 +04:00
|
|
|
|
|
|
|
#endif
|