20#include <freerdp/config.h>
24#include <winpr/assert.h>
27#include <winpr/synch.h>
28#include <winpr/print.h>
29#include <winpr/stream.h>
30#include <winpr/winsock.h>
31#include <winpr/crypto.h>
33#include <freerdp/log.h>
34#include <freerdp/error.h>
35#include <freerdp/utils/ringbuffer.h>
37#include <openssl/bio.h>
44#include <sys/socket.h>
47#ifdef FREERDP_HAVE_VALGRIND_MEMCHECK_H
48#include <valgrind/memcheck.h>
58#include "childsession.h"
60#include "gateway/rdg.h"
61#include "gateway/wst.h"
62#include "gateway/arm.h"
64#define TAG FREERDP_TAG("core.transport")
66#define BUFFER_SIZE 16384
70 TRANSPORT_LAYER layer;
80 TransportRecv ReceiveCallback;
81 wStreamPool* ReceivePool;
82 HANDLE connectedEvent;
92 BOOL haveMoreBytesToRead;
103 void* userContextShadowPtr;
104} rdpTransportLayerInt;
106static void transport_ssl_cb(
const SSL* ssl,
int where,
int ret)
108 if (where & SSL_CB_ALERT)
110 rdpTransport* transport = (rdpTransport*)SSL_get_app_data(ssl);
111 WINPR_ASSERT(transport);
115 case (SSL3_AL_FATAL << 8) | SSL_AD_ACCESS_DENIED:
117 if (!freerdp_get_last_error(transport_get_context(transport)))
119 WLog_Print(transport->log, WLOG_ERROR,
"ACCESS DENIED");
120 freerdp_set_last_error_log(transport_get_context(transport),
121 FREERDP_ERROR_AUTHENTICATION_FAILED);
126 case (SSL3_AL_FATAL << 8) | SSL_AD_INTERNAL_ERROR:
128 if (transport->NlaMode)
130 if (!freerdp_get_last_error(transport_get_context(transport)))
134 kret = nla_get_error(transport->nla);
136 kret = FREERDP_ERROR_CONNECT_PASSWORD_CERTAINLY_EXPIRED;
137 freerdp_set_last_error_log(transport_get_context(transport), kret);
143 case (SSL3_AL_WARNING << 8) | SSL3_AD_CLOSE_NOTIFY:
147 WLog_Print(transport->log, WLOG_WARN,
148 "Unhandled SSL error (where=%d, ret=%d [%s, %s])", where, ret,
149 SSL_alert_type_string_long(ret), SSL_alert_desc_string_long(ret));
156wStream* transport_send_stream_init(WINPR_ATTR_UNUSED rdpTransport* transport,
size_t size)
158 WINPR_ASSERT(transport);
160 return Stream_New(NULL, size);
163BOOL transport_attach(rdpTransport* transport,
int sockfd)
167 return IFCALLRESULT(FALSE, transport->io.TransportAttach, transport, sockfd);
170static BOOL transport_default_attach(rdpTransport* transport,
int sockfd)
172 BIO* socketBio = NULL;
173 BIO* bufferedBio = NULL;
174 const rdpSettings* settings = NULL;
175 rdpContext* context = transport_get_context(transport);
179 WLog_WARN(TAG,
"Running peer without socket (sockfd=%d)", sockfd);
183 settings = context->settings;
184 WINPR_ASSERT(settings);
188 if (!freerdp_tcp_set_keep_alive_mode(settings, sockfd))
191 socketBio = BIO_new(BIO_s_simple_socket());
197 bufferedBio = BIO_new(BIO_s_buffered_socket());
203 bufferedBio = BIO_push(bufferedBio, socketBio);
212 BIO_set_fd(socketBio, sockfd, BIO_CLOSE);
214 EnterCriticalSection(&(transport->ReadLock));
215 EnterCriticalSection(&(transport->WriteLock));
216 transport->frontBio = bufferedBio;
217 LeaveCriticalSection(&(transport->WriteLock));
218 LeaveCriticalSection(&(transport->ReadLock));
224 BIO_free_all(socketBio);
226 closesocket((SOCKET)sockfd);
231BOOL transport_connect_rdp(rdpTransport* transport)
236 switch (utils_authenticate(transport_get_context(transport)->instance, AUTH_RDP, FALSE))
240 case AUTH_NO_CREDENTIALS:
243 freerdp_set_last_error_if_not(transport_get_context(transport),
244 FREERDP_ERROR_CONNECT_CANCELLED);
251BOOL transport_connect_tls(rdpTransport* transport)
253 const rdpSettings* settings = NULL;
254 rdpContext* context = transport_get_context(transport);
256 settings = context->settings;
257 WINPR_ASSERT(settings);
260 if (settings->SelectedProtocol == PROTOCOL_SSL)
262 switch (utils_authenticate(context->instance, AUTH_TLS, FALSE))
266 case AUTH_NO_CREDENTIALS:
269 freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_CANCELLED);
276 return IFCALLRESULT(FALSE, transport->io.TLSConnect, transport);
279static BOOL transport_default_connect_tls(rdpTransport* transport)
283 rdpContext* context = NULL;
284 rdpSettings* settings = NULL;
286 WINPR_ASSERT(transport);
288 context = transport_get_context(transport);
289 WINPR_ASSERT(context);
291 settings = context->settings;
292 WINPR_ASSERT(settings);
294 if (!(tls = freerdp_tls_new(context)))
297 transport->tls = tls;
299 if (transport->GatewayEnabled)
300 transport->layer = TRANSPORT_LAYER_TSG_TLS;
302 transport->layer = TRANSPORT_LAYER_TLS;
304 tls->hostname = settings->ServerHostname;
305 tls->serverName = settings->UserSpecifiedServerName;
306 tls->port = WINPR_ASSERTING_INT_CAST(int32_t, MIN(UINT16_MAX, settings->ServerPort));
311 tls->isGatewayTransport = FALSE;
312 tlsStatus = freerdp_tls_connect(tls, transport->frontBio);
318 freerdp_set_last_error_if_not(context, FREERDP_ERROR_TLS_CONNECT_FAILED);
322 freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_CANCELLED);
328 transport->frontBio = tls->bio;
337 typedef void (*ssl_cb_t)(
const SSL* ssl,
int type,
int val);
338 ssl_cb_t fkt = transport_ssl_cb;
340 BIO_info_cb* bfkt = WINPR_FUNC_PTR_CAST(fkt, BIO_info_cb*);
341 BIO_callback_ctrl(tls->bio, BIO_CTRL_SET_CALLBACK, bfkt);
342 SSL_set_app_data(tls->ssl, transport);
344 if (!transport->frontBio)
346 WLog_Print(transport->log, WLOG_ERROR,
"unable to prepend a filtering TLS bio");
353BOOL transport_connect_nla(rdpTransport* transport, BOOL earlyUserAuth)
355 rdpContext* context = NULL;
356 rdpSettings* settings = NULL;
361 context = transport_get_context(transport);
362 WINPR_ASSERT(context);
364 settings = context->settings;
365 WINPR_ASSERT(settings);
370 if (!transport_connect_tls(transport))
373 if (!settings->Authentication)
377 rdp->nla = nla_new(context, transport);
382 nla_set_early_user_auth(rdp->nla, earlyUserAuth);
384 transport_set_nla_mode(transport, TRUE);
386 if (settings->AuthenticationServiceClass)
388 if (!nla_set_service_principal(rdp->nla, settings->AuthenticationServiceClass,
393 if (nla_client_begin(rdp->nla) < 0)
395 WLog_Print(transport->log, WLOG_ERROR,
"NLA begin failed");
397 freerdp_set_last_error_if_not(context, FREERDP_ERROR_AUTHENTICATION_FAILED);
399 transport_set_nla_mode(transport, FALSE);
403 return rdp_client_transition_to_state(rdp, CONNECTION_STATE_NLA);
406BOOL transport_connect_rdstls(rdpTransport* transport)
409 rdpRdstls* rdstls = NULL;
410 rdpContext* context = NULL;
412 WINPR_ASSERT(transport);
414 context = transport_get_context(transport);
415 WINPR_ASSERT(context);
417 if (!transport_connect_tls(transport))
420 rdstls = rdstls_new(context, transport);
424 transport_set_rdstls_mode(transport, TRUE);
426 if (rdstls_authenticate(rdstls) < 0)
428 WLog_Print(transport->log, WLOG_ERROR,
"RDSTLS authentication failed");
429 freerdp_set_last_error_if_not(context, FREERDP_ERROR_AUTHENTICATION_FAILED);
433 transport_set_rdstls_mode(transport, FALSE);
440BOOL transport_connect_aad(rdpTransport* transport)
442 rdpContext* context = NULL;
443 rdpSettings* settings = NULL;
448 context = transport_get_context(transport);
449 WINPR_ASSERT(context);
451 settings = context->settings;
452 WINPR_ASSERT(settings);
457 if (!transport_connect_tls(transport))
460 if (!settings->Authentication)
466 transport_set_aad_mode(transport, TRUE);
468 if (aad_client_begin(rdp->aad) < 0)
470 WLog_Print(transport->log, WLOG_ERROR,
"AAD begin failed");
472 freerdp_set_last_error_if_not(context, FREERDP_ERROR_AUTHENTICATION_FAILED);
474 transport_set_aad_mode(transport, FALSE);
478 return rdp_client_transition_to_state(rdp, CONNECTION_STATE_AAD);
481static BOOL transport_can_retry(
const rdpContext* context, BOOL status)
483 switch (freerdp_get_last_error(context))
485 case FREERDP_ERROR_CONNECT_TARGET_BOOTING:
492BOOL transport_connect(rdpTransport* transport,
const char* hostname, UINT16 port, DWORD timeout)
495 rdpSettings* settings = NULL;
496 rdpContext* context = transport_get_context(transport);
497 BOOL rpcFallback = 0;
499 WINPR_ASSERT(context);
500 WINPR_ASSERT(hostname);
502 settings = context->settings;
503 WINPR_ASSERT(settings);
505 rpcFallback = !settings->GatewayHttpTransport;
507 if (transport->GatewayEnabled)
509 if (settings->GatewayUrl)
511 WINPR_ASSERT(!transport->wst);
512 transport->wst = wst_new(context);
517 status = wst_connect(transport->wst, timeout);
521 transport->frontBio = wst_get_front_bio_and_take_ownership(transport->wst);
522 WINPR_ASSERT(transport->frontBio);
523 BIO_set_nonblock(transport->frontBio, 0);
524 transport->layer = TRANSPORT_LAYER_TSG;
529 wst_free(transport->wst);
530 transport->wst = NULL;
533 if (transport_can_retry(transport->context, status) && settings->GatewayHttpTransport)
535 WINPR_ASSERT(!transport->rdg);
536 transport->rdg = rdg_new(context);
541 status = rdg_connect(transport->rdg, timeout, &rpcFallback);
545 transport->frontBio = rdg_get_front_bio_and_take_ownership(transport->rdg);
546 WINPR_ASSERT(transport->frontBio);
547 BIO_set_nonblock(transport->frontBio, 0);
548 transport->layer = TRANSPORT_LAYER_TSG;
553 rdg_free(transport->rdg);
554 transport->rdg = NULL;
558 if (transport_can_retry(transport->context, status) && settings->GatewayRpcTransport &&
561 WINPR_ASSERT(!transport->tsg);
562 transport->tsg = tsg_new(transport);
568 freerdp_set_last_error_log(context, FREERDP_ERROR_SUCCESS);
569 status = tsg_connect(transport->tsg, hostname, port, timeout);
573 transport->frontBio = tsg_get_bio(transport->tsg);
574 transport->layer = TRANSPORT_LAYER_TSG;
579 tsg_free(transport->tsg);
580 transport->tsg = NULL;
587 const char* proxyHostname = NULL;
588 const char* proxyUsername = NULL;
589 const char* proxyPassword = NULL;
590 BOOL isProxyConnection =
591 proxy_prepare(settings, &proxyHostname, &peerPort, &proxyUsername, &proxyPassword);
594 if (isProxyConnection)
595 layer = transport_connect_layer(transport, proxyHostname, peerPort, timeout);
597 layer = transport_connect_layer(transport, hostname, port, timeout);
602 if (!transport_attach_layer(transport, layer))
604 transport_layer_free(layer);
608 if (isProxyConnection)
610 if (!proxy_connect(context, transport->frontBio, proxyUsername, proxyPassword, hostname,
621BOOL transport_connect_childsession(rdpTransport* transport)
623 WINPR_ASSERT(transport);
625 transport->frontBio = createChildSessionBio();
626 if (!transport->frontBio)
629 transport->layer = TRANSPORT_LAYER_TSG;
633BOOL transport_accept_rdp(rdpTransport* transport)
641BOOL transport_accept_tls(rdpTransport* transport)
645 return IFCALLRESULT(FALSE, transport->io.TLSAccept, transport);
648static BOOL transport_default_accept_tls(rdpTransport* transport)
650 rdpContext* context = transport_get_context(transport);
651 rdpSettings* settings = NULL;
653 WINPR_ASSERT(context);
655 settings = context->settings;
656 WINPR_ASSERT(settings);
659 transport->tls = freerdp_tls_new(context);
661 transport->layer = TRANSPORT_LAYER_TLS;
663 if (!freerdp_tls_accept(transport->tls, transport->frontBio, settings))
666 transport->frontBio = transport->tls->bio;
670BOOL transport_accept_nla(rdpTransport* transport)
672 rdpContext* context = transport_get_context(transport);
673 rdpSettings* settings = NULL;
675 WINPR_ASSERT(context);
677 settings = context->settings;
678 WINPR_ASSERT(settings);
680 if (!IFCALLRESULT(FALSE, transport->io.TLSAccept, transport))
685 if (!settings->Authentication)
690 transport->nla = nla_new(context, transport);
691 transport_set_nla_mode(transport, TRUE);
694 if (nla_authenticate(transport->nla) < 0)
696 WLog_Print(transport->log, WLOG_ERROR,
"client authentication failure");
697 transport_set_nla_mode(transport, FALSE);
698 nla_free(transport->nla);
699 transport->nla = NULL;
700 freerdp_tls_set_alert_code(transport->tls, TLS_ALERT_LEVEL_FATAL,
701 TLS_ALERT_DESCRIPTION_ACCESS_DENIED);
702 freerdp_tls_send_alert(transport->tls);
707 transport_set_nla_mode(transport, FALSE);
711BOOL transport_accept_rdstls(rdpTransport* transport)
714 rdpRdstls* rdstls = NULL;
715 rdpContext* context = NULL;
717 WINPR_ASSERT(transport);
719 context = transport_get_context(transport);
720 WINPR_ASSERT(context);
722 if (!IFCALLRESULT(FALSE, transport->io.TLSAccept, transport))
725 rdstls = rdstls_new(context, transport);
729 transport_set_rdstls_mode(transport, TRUE);
731 if (rdstls_authenticate(rdstls) < 0)
733 WLog_Print(transport->log, WLOG_ERROR,
"client authentication failure");
734 freerdp_tls_set_alert_code(transport->tls, TLS_ALERT_LEVEL_FATAL,
735 TLS_ALERT_DESCRIPTION_ACCESS_DENIED);
736 freerdp_tls_send_alert(transport->tls);
740 transport_set_rdstls_mode(transport, FALSE);
747#define WLog_ERR_BIO(transport, biofunc, bio) \
748 transport_bio_error_log(transport, biofunc, bio, __FILE__, __func__, __LINE__)
750static void transport_bio_error_log(rdpTransport* transport, LPCSTR biofunc,
751 WINPR_ATTR_UNUSED BIO* bio, LPCSTR file, LPCSTR func,
754 unsigned long sslerr = 0;
758 WINPR_ASSERT(transport);
763 if (level < WLog_GetLogLevel(transport->log))
766 if (ERR_peek_error() == 0)
768 char ebuffer[256] = { 0 };
771 WLog_PrintTextMessage(transport->log, level, line, file, func,
"%s retries exceeded",
774 WLog_PrintTextMessage(transport->log, level, line, file, func,
775 "%s returned a system error %d: %s", biofunc, saveerrno,
776 winpr_strerror(saveerrno, ebuffer,
sizeof(ebuffer)));
780 while ((sslerr = ERR_get_error()))
782 char buf[120] = { 0 };
784 ERR_error_string_n(sslerr, buf, 120);
785 WLog_PrintTextMessage(transport->log, level, line, file, func,
"%s returned an error: %s",
790static SSIZE_T transport_read_layer(rdpTransport* transport, BYTE* data,
size_t bytes)
794 rdpContext* context = NULL;
796 WINPR_ASSERT(transport);
798 context = transport_get_context(transport);
799 WINPR_ASSERT(context);
804 if (!transport->frontBio || (bytes > SSIZE_MAX))
806 transport->layer = TRANSPORT_LAYER_CLOSED;
807 freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
811 while (read < (SSIZE_T)bytes)
813 const SSIZE_T tr = (SSIZE_T)bytes - read;
814 int r = (int)((tr > INT_MAX) ? INT_MAX : tr);
816 int status = BIO_read(transport->frontBio, data + read, r);
818 if (freerdp_shall_disconnect_context(context))
823 if (!transport->frontBio || !BIO_should_retry(transport->frontBio))
826 if (!transport->frontBio)
828 WLog_Print(transport->log, WLOG_ERROR,
"BIO_read: transport->frontBio null");
832 WLog_ERR_BIO(transport,
"BIO_read", transport->frontBio);
833 transport->layer = TRANSPORT_LAYER_CLOSED;
834 freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
839 if (!transport->blocking)
844 if (BIO_wait_read(transport->frontBio, 100) < 0)
846 WLog_ERR_BIO(transport,
"BIO_wait_read", transport->frontBio);
853#ifdef FREERDP_HAVE_VALGRIND_MEMCHECK_H
854 VALGRIND_MAKE_MEM_DEFINED(data + read, bytes - read);
857 rdp->inBytes += WINPR_ASSERTING_INT_CAST(uint64_t, status);
877static SSIZE_T transport_read_layer_bytes(rdpTransport* transport,
wStream* s,
size_t toRead)
883 if (toRead > SSIZE_MAX)
886 status = IFCALLRESULT(-1, transport->io.ReadBytes, transport, Stream_Pointer(s), toRead);
891 Stream_Seek(s, (
size_t)status);
892 return status == (SSIZE_T)toRead ? 1 : 0;
907int transport_read_pdu(rdpTransport* transport,
wStream* s)
911 return IFCALLRESULT(-1, transport->io.ReadPdu, transport, s);
914static SSIZE_T parse_nla_mode_pdu(rdpTransport* transport,
wStream* stream)
916 SSIZE_T pduLength = 0;
918 wStream* s = Stream_StaticConstInit(&sbuffer, Stream_Buffer(stream), Stream_Length(stream));
925 UINT8 typeEncoding = 0;
926 if (Stream_GetRemainingLength(s) < 1)
928 Stream_Read_UINT8(s, typeEncoding);
929 if (typeEncoding == 0x30)
932 UINT8 lengthEncoding = 0;
933 if (Stream_GetRemainingLength(s) < 1)
935 Stream_Read_UINT8(s, lengthEncoding);
936 if (lengthEncoding & 0x80)
938 if ((lengthEncoding & ~(0x80)) == 1)
941 if (Stream_GetRemainingLength(s) < 1)
943 Stream_Read_UINT8(s, length);
947 else if ((lengthEncoding & ~(0x80)) == 2)
951 if (Stream_GetRemainingLength(s) < 2)
953 Stream_Read_UINT16_BE(s, length);
959 WLog_Print(transport->log, WLOG_ERROR,
"Error reading TSRequest!");
965 pduLength = lengthEncoding;
973static SSIZE_T parse_default_mode_pdu(rdpTransport* transport,
wStream* stream)
975 SSIZE_T pduLength = 0;
977 wStream* s = Stream_StaticConstInit(&sbuffer, Stream_Buffer(stream), Stream_Length(stream));
980 if (Stream_GetRemainingLength(s) < 1)
982 Stream_Read_UINT8(s, version);
987 if (Stream_GetRemainingLength(s) < 3)
990 Stream_Read_UINT16_BE(s, length);
994 if ((pduLength < 7) || (pduLength > 0xFFFF))
996 WLog_Print(transport->log, WLOG_ERROR,
"tpkt - invalid pduLength: %" PRIdz, pduLength);
1004 if (Stream_GetRemainingLength(s) < 1)
1006 Stream_Read_UINT8(s, length1);
1010 if (Stream_GetRemainingLength(s) < 1)
1012 Stream_Read_UINT8(s, length2);
1013 pduLength = ((length1 & 0x7F) << 8) | length2;
1016 pduLength = length1;
1023 if (pduLength < 3 || pduLength > 0x8000)
1025 WLog_Print(transport->log, WLOG_ERROR,
"fast path - invalid pduLength: %" PRIdz,
1034SSIZE_T transport_parse_pdu(rdpTransport* transport,
wStream* s, BOOL* incomplete)
1036 SSIZE_T pduLength = 0;
1047 Stream_SealLength(s);
1048 if (transport->NlaMode)
1049 pduLength = parse_nla_mode_pdu(transport, s);
1050 else if (transport->RdstlsMode)
1051 pduLength = rdstls_parse_pdu(transport->log, s);
1053 pduLength = parse_default_mode_pdu(transport, s);
1058 const size_t len = Stream_Length(s);
1059 if (len > WINPR_ASSERTING_INT_CAST(
size_t, pduLength))
1063 *incomplete = len < WINPR_ASSERTING_INT_CAST(
size_t, pduLength);
1068static int transport_default_read_pdu(rdpTransport* transport,
wStream* s)
1070 BOOL incomplete = 0;
1072 size_t pduLength = 0;
1073 size_t position = 0;
1075 WINPR_ASSERT(transport);
1080 if (transport->AadMode)
1085 const SSIZE_T rc = transport_read_layer(transport, &c, 1);
1087 return (rc == 0) ? 0 : -1;
1088 if (!Stream_EnsureRemainingCapacity(s, 1))
1090 Stream_Write_UINT8(s, c);
1091 }
while (c !=
'\0');
1093 else if (transport->earlyUserAuth)
1095 if (!Stream_EnsureCapacity(s, 4))
1097 const SSIZE_T rc = transport_read_layer_bytes(transport, s, 4);
1099 return (rc == 0) ? 0 : -1;
1104 status = transport_parse_pdu(transport, s, &incomplete);
1105 while ((status == 0) && incomplete)
1107 if (!Stream_EnsureRemainingCapacity(s, 1))
1109 SSIZE_T rc = transport_read_layer_bytes(transport, s, 1);
1114 status = transport_parse_pdu(transport, s, &incomplete);
1120 pduLength = (size_t)status;
1123 if (!Stream_EnsureCapacity(s, pduLength))
1126 position = Stream_GetPosition(s);
1127 if (position > pduLength)
1129 else if (position < pduLength)
1131 status = transport_read_layer_bytes(transport, s, pduLength - position);
1134 if ((status < INT32_MIN) || (status > INT32_MAX))
1140 if (Stream_GetPosition(s) >= pduLength)
1141 WLog_Packet(transport->log, WLOG_TRACE, Stream_Buffer(s), pduLength,
1142 WLOG_PACKET_INBOUND);
1145 Stream_SealLength(s);
1146 Stream_SetPosition(s, 0);
1147 const size_t len = Stream_Length(s);
1148 if (len > INT32_MAX)
1153int transport_write(rdpTransport* transport,
wStream* s)
1158 return IFCALLRESULT(-1, transport->io.WritePdu, transport, s);
1161static int transport_default_write(rdpTransport* transport,
wStream* s)
1164 rdpContext* context = transport_get_context(transport);
1166 WINPR_ASSERT(transport);
1167 WINPR_ASSERT(context);
1174 rdpRdp* rdp = context->rdp;
1178 EnterCriticalSection(&(transport->WriteLock));
1179 if (!transport->frontBio)
1182 size_t length = Stream_GetPosition(s);
1183 size_t writtenlength = length;
1184 Stream_SetPosition(s, 0);
1188 rdp->outBytes += length;
1189 WLog_Packet(transport->log, WLOG_TRACE, Stream_Buffer(s), length, WLOG_PACKET_OUTBOUND);
1195 const int towrite = (length > INT32_MAX) ? INT32_MAX : (int)length;
1196 status = BIO_write(transport->frontBio, Stream_ConstPointer(s), towrite);
1204 if (!BIO_should_retry(transport->frontBio))
1206 WLog_ERR_BIO(transport,
"BIO_should_retry", transport->frontBio);
1211 if (!transport->blocking)
1213 WLog_ERR_BIO(transport,
"BIO_write", transport->frontBio);
1217 if (BIO_wait_write(transport->frontBio, 100) < 0)
1219 WLog_ERR_BIO(transport,
"BIO_wait_write", transport->frontBio);
1227 WINPR_ASSERT(context->settings);
1228 if (transport->blocking || context->settings->WaitForOutputBufferFlush)
1230 while (BIO_write_blocked(transport->frontBio))
1232 if (BIO_wait_write(transport->frontBio, 100) < 0)
1234 WLog_Print(transport->log, WLOG_ERROR,
"error when selecting for write");
1239 if (BIO_flush(transport->frontBio) < 1)
1241 WLog_Print(transport->log, WLOG_ERROR,
"error when flushing outputBuffer");
1248 const size_t ustatus = (size_t)status;
1249 if (ustatus > length)
1256 Stream_Seek(s, ustatus);
1259 transport->written += writtenlength;
1265 transport->layer = TRANSPORT_LAYER_CLOSED;
1266 freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
1269 LeaveCriticalSection(&(transport->WriteLock));
1275BOOL transport_get_public_key(rdpTransport* transport,
const BYTE** data, DWORD* length)
1277 return IFCALLRESULT(FALSE, transport->io.GetPublicKey, transport, data, length);
1280static BOOL transport_default_get_public_key(rdpTransport* transport,
const BYTE** data,
1283 rdpTls* tls = transport_get_tls(transport);
1287 *data = tls->PublicKey;
1288 *length = tls->PublicKeyLength;
1293DWORD transport_get_event_handles(rdpTransport* transport, HANDLE* events, DWORD count)
1297 WINPR_ASSERT(transport);
1298 WINPR_ASSERT(events);
1299 WINPR_ASSERT(count > 0);
1305 WLog_Print(transport->log, WLOG_ERROR,
"provided handles array is too small");
1309 events[nCount++] = transport->rereadEvent;
1311 if (transport->useIoEvent)
1315 events[nCount++] = transport->ioEvent;
1319 if (!transport->GatewayEnabled)
1323 if (nCount >= count)
1325 WLog_Print(transport->log, WLOG_ERROR,
1326 "provided handles array is too small (count=%" PRIu32
" nCount=%" PRIu32
1332 if (transport->frontBio)
1334 if (BIO_get_event(transport->frontBio, &events[nCount]) != 1)
1336 WLog_Print(transport->log, WLOG_ERROR,
"error getting the frontBio handle");
1348 rdg_get_event_handles(transport->rdg, &events[nCount], count - nCount);
1355 else if (transport->tsg)
1358 tsg_get_event_handles(transport->tsg, &events[nCount], count - nCount);
1365 else if (transport->wst)
1368 wst_get_event_handles(transport->wst, &events[nCount], count - nCount);
1380#if defined(WITH_FREERDP_DEPRECATED)
1381void transport_get_fds(rdpTransport* transport,
void** rfds,
int* rcount)
1384 HANDLE events[MAXIMUM_WAIT_OBJECTS] = { 0 };
1386 WINPR_ASSERT(transport);
1388 WINPR_ASSERT(rcount);
1390 nCount = transport_get_event_handles(transport, events, ARRAYSIZE(events));
1391 *rcount = nCount + 1;
1393 for (DWORD index = 0; index < nCount; index++)
1395 rfds[index] = GetEventWaitObject(events[index]);
1398 rfds[nCount] = GetEventWaitObject(transport->rereadEvent);
1402BOOL transport_is_write_blocked(rdpTransport* transport)
1404 WINPR_ASSERT(transport);
1405 WINPR_ASSERT(transport->frontBio);
1406 return BIO_write_blocked(transport->frontBio) != 0;
1409int transport_drain_output_buffer(rdpTransport* transport)
1411 BOOL status = FALSE;
1413 WINPR_ASSERT(transport);
1414 WINPR_ASSERT(transport->frontBio);
1415 if (BIO_write_blocked(transport->frontBio))
1417 if (BIO_flush(transport->frontBio) < 1)
1420 const long rc = BIO_write_blocked(transport->frontBio);
1427int transport_check_fds(rdpTransport* transport)
1430 state_run_t recv_status = STATE_RUN_FAILED;
1432 rdpContext* context = transport_get_context(transport);
1434 WINPR_ASSERT(context);
1436 if (transport->layer == TRANSPORT_LAYER_CLOSED)
1438 WLog_Print(transport->log, WLOG_DEBUG,
"transport_check_fds: transport layer closed");
1439 freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
1452 if ((status = transport_read_pdu(transport, transport->ReceiveBuffer)) <= 0)
1455 WLog_Print(transport->log, WLOG_DEBUG,
"transport_check_fds: transport_read_pdu() - %i",
1457 if (transport->haveMoreBytesToRead)
1459 transport->haveMoreBytesToRead = FALSE;
1460 (void)ResetEvent(transport->rereadEvent);
1465 received = transport->ReceiveBuffer;
1466 transport->ReceiveBuffer = StreamPool_Take(transport->ReceivePool, 0);
1467 if (!transport->ReceiveBuffer)
1469 Stream_Release(received);
1479 WINPR_ASSERT(transport->ReceiveCallback);
1480 recv_status = transport->ReceiveCallback(transport, received, transport->ReceiveExtra);
1481 Stream_Release(received);
1483 if (state_run_failed(recv_status))
1485 char buffer[64] = { 0 };
1486 WLog_Print(transport->log, WLOG_ERROR,
1487 "transport_check_fds: transport->ReceiveCallback() - %s",
1488 state_run_result_string(recv_status, buffer, ARRAYSIZE(buffer)));
1495 if (!transport->haveMoreBytesToRead)
1497 transport->haveMoreBytesToRead = TRUE;
1498 (void)SetEvent(transport->rereadEvent);
1503BOOL transport_set_blocking_mode(rdpTransport* transport, BOOL blocking)
1505 WINPR_ASSERT(transport);
1507 return IFCALLRESULT(FALSE, transport->io.SetBlockingMode, transport, blocking);
1510static BOOL transport_default_set_blocking_mode(rdpTransport* transport, BOOL blocking)
1512 WINPR_ASSERT(transport);
1514 transport->blocking = blocking;
1516 if (transport->frontBio)
1518 if (!BIO_set_nonblock(transport->frontBio, blocking ? FALSE : TRUE))
1525rdpTransportLayer* transport_connect_layer(rdpTransport* transport,
const char* hostname,
int port,
1528 WINPR_ASSERT(transport);
1530 return IFCALLRESULT(NULL, transport->io.ConnectLayer, transport, hostname, port, timeout);
1533static rdpTransportLayer* transport_default_connect_layer(rdpTransport* transport,
1534 const char* hostname,
int port,
1537 rdpContext* context = transport_get_context(transport);
1538 WINPR_ASSERT(context);
1540 return freerdp_tcp_connect_layer(context, hostname, port, timeout);
1545 WINPR_ASSERT(transport);
1546 WINPR_ASSERT(layer);
1548 return IFCALLRESULT(FALSE, transport->io.AttachLayer, transport, layer);
1551static BOOL transport_default_attach_layer(rdpTransport* transport,
rdpTransportLayer* layer)
1553 BIO* layerBio = BIO_new(BIO_s_transport_layer());
1557 BIO* bufferedBio = BIO_new(BIO_s_buffered_socket());
1561 bufferedBio = BIO_push(bufferedBio, layerBio);
1566 BIO_set_data(layerBio, layer);
1568 transport->frontBio = bufferedBio;
1574 BIO_free_all(layerBio);
1579void transport_set_gateway_enabled(rdpTransport* transport, BOOL GatewayEnabled)
1581 WINPR_ASSERT(transport);
1582 transport->GatewayEnabled = GatewayEnabled;
1585void transport_set_nla_mode(rdpTransport* transport, BOOL NlaMode)
1587 WINPR_ASSERT(transport);
1588 transport->NlaMode = NlaMode;
1591void transport_set_rdstls_mode(rdpTransport* transport, BOOL RdstlsMode)
1593 WINPR_ASSERT(transport);
1594 transport->RdstlsMode = RdstlsMode;
1597void transport_set_aad_mode(rdpTransport* transport, BOOL AadMode)
1599 WINPR_ASSERT(transport);
1600 transport->AadMode = AadMode;
1603BOOL transport_disconnect(rdpTransport* transport)
1607 return IFCALLRESULT(FALSE, transport->io.TransportDisconnect, transport);
1610static BOOL transport_default_disconnect(rdpTransport* transport)
1617 EnterCriticalSection(&(transport->ReadLock));
1618 EnterCriticalSection(&(transport->WriteLock));
1621 freerdp_tls_free(transport->tls);
1622 transport->tls = NULL;
1626 if (transport->frontBio)
1627 BIO_free_all(transport->frontBio);
1632 tsg_free(transport->tsg);
1633 transport->tsg = NULL;
1638 rdg_free(transport->rdg);
1639 transport->rdg = NULL;
1644 wst_free(transport->wst);
1645 transport->wst = NULL;
1648 transport->frontBio = NULL;
1649 transport->layer = TRANSPORT_LAYER_TCP;
1650 transport->earlyUserAuth = FALSE;
1651 LeaveCriticalSection(&(transport->WriteLock));
1652 LeaveCriticalSection(&(transport->ReadLock));
1656rdpTransport* transport_new(rdpContext* context)
1658 rdpTransport* transport = (rdpTransport*)calloc(1,
sizeof(rdpTransport));
1660 WINPR_ASSERT(context);
1664 transport->log = WLog_Get(TAG);
1666 if (!transport->log)
1669 transport->context = context;
1670 transport->ReceivePool = StreamPool_New(TRUE, BUFFER_SIZE);
1672 if (!transport->ReceivePool)
1676 transport->ReceiveBuffer = StreamPool_Take(transport->ReceivePool, 0);
1678 if (!transport->ReceiveBuffer)
1681 transport->connectedEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1683 if (!transport->connectedEvent || transport->connectedEvent == INVALID_HANDLE_VALUE)
1686 transport->rereadEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1688 if (!transport->rereadEvent || transport->rereadEvent == INVALID_HANDLE_VALUE)
1691 transport->ioEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
1693 if (!transport->ioEvent || transport->ioEvent == INVALID_HANDLE_VALUE)
1696 transport->haveMoreBytesToRead = FALSE;
1697 transport->blocking = TRUE;
1698 transport->GatewayEnabled = FALSE;
1699 transport->layer = TRANSPORT_LAYER_TCP;
1701 if (!InitializeCriticalSectionAndSpinCount(&(transport->ReadLock), 4000))
1704 if (!InitializeCriticalSectionAndSpinCount(&(transport->WriteLock), 4000))
1708 transport->io.TCPConnect = freerdp_tcp_default_connect;
1709 transport->io.TLSConnect = transport_default_connect_tls;
1710 transport->io.TLSAccept = transport_default_accept_tls;
1711 transport->io.TransportAttach = transport_default_attach;
1712 transport->io.TransportDisconnect = transport_default_disconnect;
1713 transport->io.ReadPdu = transport_default_read_pdu;
1714 transport->io.WritePdu = transport_default_write;
1715 transport->io.ReadBytes = transport_read_layer;
1716 transport->io.GetPublicKey = transport_default_get_public_key;
1717 transport->io.SetBlockingMode = transport_default_set_blocking_mode;
1718 transport->io.ConnectLayer = transport_default_connect_layer;
1719 transport->io.AttachLayer = transport_default_attach_layer;
1723 WINPR_PRAGMA_DIAG_PUSH
1724 WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
1725 transport_free(transport);
1726 WINPR_PRAGMA_DIAG_POP
1730void transport_free(rdpTransport* transport)
1735 transport_disconnect(transport);
1737 EnterCriticalSection(&(transport->ReadLock));
1738 if (transport->ReceiveBuffer)
1739 Stream_Release(transport->ReceiveBuffer);
1740 LeaveCriticalSection(&(transport->ReadLock));
1742 (void)StreamPool_WaitForReturn(transport->ReceivePool, INFINITE);
1744 EnterCriticalSection(&(transport->ReadLock));
1745 EnterCriticalSection(&(transport->WriteLock));
1747 nla_free(transport->nla);
1748 StreamPool_Free(transport->ReceivePool);
1749 (void)CloseHandle(transport->connectedEvent);
1750 (void)CloseHandle(transport->rereadEvent);
1751 (void)CloseHandle(transport->ioEvent);
1753 LeaveCriticalSection(&(transport->ReadLock));
1754 DeleteCriticalSection(&(transport->ReadLock));
1756 LeaveCriticalSection(&(transport->WriteLock));
1757 DeleteCriticalSection(&(transport->WriteLock));
1761BOOL transport_set_io_callbacks(rdpTransport* transport,
const rdpTransportIo* io_callbacks)
1763 if (!transport || !io_callbacks)
1766 transport->io = *io_callbacks;
1770const rdpTransportIo* transport_get_io_callbacks(
const rdpTransport* transport)
1774 return &transport->io;
1777rdpContext* transport_get_context(rdpTransport* transport)
1779 WINPR_ASSERT(transport);
1780 return transport->context;
1783rdpTransport* freerdp_get_transport(rdpContext* context)
1785 WINPR_ASSERT(context);
1786 WINPR_ASSERT(context->rdp);
1787 return context->rdp->transport;
1790BOOL transport_set_nla(rdpTransport* transport, rdpNla* nla)
1792 WINPR_ASSERT(transport);
1793 nla_free(transport->nla);
1794 transport->nla = nla;
1798rdpNla* transport_get_nla(rdpTransport* transport)
1800 WINPR_ASSERT(transport);
1801 return transport->nla;
1804BOOL transport_set_tls(rdpTransport* transport, rdpTls* tls)
1806 WINPR_ASSERT(transport);
1807 freerdp_tls_free(transport->tls);
1808 transport->tls = tls;
1812rdpTls* transport_get_tls(rdpTransport* transport)
1814 WINPR_ASSERT(transport);
1815 return transport->tls;
1818BOOL transport_set_tsg(rdpTransport* transport, rdpTsg* tsg)
1820 WINPR_ASSERT(transport);
1821 tsg_free(transport->tsg);
1822 transport->tsg = tsg;
1826rdpTsg* transport_get_tsg(rdpTransport* transport)
1828 WINPR_ASSERT(transport);
1829 return transport->tsg;
1832wStream* transport_take_from_pool(rdpTransport* transport,
size_t size)
1834 WINPR_ASSERT(transport);
1835 if (!transport->frontBio)
1837 return StreamPool_Take(transport->ReceivePool, size);
1840UINT64 transport_get_bytes_sent(rdpTransport* transport, BOOL resetCount)
1843 WINPR_ASSERT(transport);
1844 rc = transport->written;
1846 transport->written = 0;
1850TRANSPORT_LAYER transport_get_layer(rdpTransport* transport)
1852 WINPR_ASSERT(transport);
1853 return transport->layer;
1856BOOL transport_set_layer(rdpTransport* transport, TRANSPORT_LAYER layer)
1858 WINPR_ASSERT(transport);
1859 transport->layer = layer;
1863BOOL transport_set_connected_event(rdpTransport* transport)
1865 WINPR_ASSERT(transport);
1866 return SetEvent(transport->connectedEvent);
1869BOOL transport_set_recv_callbacks(rdpTransport* transport, TransportRecv recv,
void* extra)
1871 WINPR_ASSERT(transport);
1872 transport->ReceiveCallback = recv;
1873 transport->ReceiveExtra = extra;
1877BOOL transport_get_blocking(
const rdpTransport* transport)
1879 WINPR_ASSERT(transport);
1880 return transport->blocking;
1883BOOL transport_set_blocking(rdpTransport* transport, BOOL blocking)
1885 WINPR_ASSERT(transport);
1886 transport->blocking = blocking;
1890BOOL transport_have_more_bytes_to_read(rdpTransport* transport)
1892 WINPR_ASSERT(transport);
1893 return transport->haveMoreBytesToRead;
1896int transport_tcp_connect(rdpTransport* transport,
const char* hostname,
int port, DWORD timeout)
1898 rdpContext* context = transport_get_context(transport);
1899 WINPR_ASSERT(context);
1900 WINPR_ASSERT(context->settings);
1901 return IFCALLRESULT(-1, transport->io.TCPConnect, context, context->settings, hostname, port,
1905HANDLE transport_get_front_bio(rdpTransport* transport)
1907 HANDLE hEvent = NULL;
1908 WINPR_ASSERT(transport);
1909 WINPR_ASSERT(transport->frontBio);
1911 BIO_get_event(transport->frontBio, &hEvent);
1915BOOL transport_io_callback_set_event(rdpTransport* transport, BOOL set)
1917 WINPR_ASSERT(transport);
1918 transport->useIoEvent = TRUE;
1920 return ResetEvent(transport->ioEvent);
1921 return SetEvent(transport->ioEvent);
1924void transport_set_early_user_auth_mode(rdpTransport* transport, BOOL EUAMode)
1926 WINPR_ASSERT(transport);
1927 transport->earlyUserAuth = EUAMode;
1928 WLog_Print(transport->log, WLOG_DEBUG,
"Early User Auth Mode: %s", EUAMode ?
"on" :
"off");
1931rdpTransportLayer* transport_layer_new(WINPR_ATTR_UNUSED rdpTransport* transport,
1934 rdpTransportLayerInt* layer = (rdpTransportLayerInt*)calloc(1,
sizeof(rdpTransportLayerInt));
1940 layer->userContextShadowPtr = calloc(1, contextSize);
1941 if (!layer->userContextShadowPtr)
1947 layer->pub.userContext = layer->userContextShadowPtr;
1954 rdpTransportLayerInt* intern = (rdpTransportLayerInt*)layer;
1958 IFCALL(intern->pub.Close, intern->pub.userContext);
1959 free(intern->userContextShadowPtr);
1963static int transport_layer_bio_write(BIO* bio,
const char* buf,
int size)
1976 BIO_clear_flags(bio, BIO_FLAGS_WRITE | BIO_FLAGS_SHOULD_RETRY);
1979 const int status = IFCALLRESULT(-1, layer->Write, layer->userContext, buf, size);
1981 if (status >= 0 && status < size)
1982 BIO_set_flags(bio, (BIO_FLAGS_WRITE | BIO_FLAGS_SHOULD_RETRY));
1987 BIO_set_flags(bio, (BIO_FLAGS_WRITE | BIO_FLAGS_SHOULD_RETRY));
1996static int transport_layer_bio_read(BIO* bio,
char* buf,
int size)
2009 BIO_clear_flags(bio, BIO_FLAGS_READ | BIO_FLAGS_SHOULD_RETRY);
2011 const int status = IFCALLRESULT(-1, layer->Read, layer->userContext, buf, size);
2016 BIO_set_flags(bio, (BIO_FLAGS_READ | BIO_FLAGS_SHOULD_RETRY));
2025static int transport_layer_bio_puts(WINPR_ATTR_UNUSED BIO* bio, WINPR_ATTR_UNUSED
const char* str)
2030static int transport_layer_bio_gets(WINPR_ATTR_UNUSED BIO* bio, WINPR_ATTR_UNUSED
char* str,
2031 WINPR_ATTR_UNUSED
int size)
2036static long transport_layer_bio_ctrl(BIO* bio,
int cmd,
long arg1,
void* arg2)
2047 case BIO_C_GET_EVENT:
2048 *((HANDLE*)arg2) = IFCALLRESULT(NULL, layer->GetEvent, layer->userContext);
2052 case BIO_C_SET_NONBLOCK:
2056 case BIO_C_WAIT_READ:
2058 int timeout = (int)arg1;
2059 BOOL r = IFCALLRESULT(FALSE, layer->Wait, layer->userContext, FALSE,
2060 WINPR_ASSERTING_INT_CAST(uint32_t, timeout));
2072 case BIO_C_WAIT_WRITE:
2074 int timeout = (int)arg1;
2075 BOOL r = IFCALLRESULT(FALSE, layer->Wait, layer->userContext, TRUE,
2076 WINPR_ASSERTING_INT_CAST(uint32_t, timeout));
2088 case BIO_CTRL_GET_CLOSE:
2089 status = BIO_get_shutdown(bio);
2092 case BIO_CTRL_SET_CLOSE:
2093 BIO_set_shutdown(bio, (
int)arg1);
2097 case BIO_CTRL_FLUSH:
2110static int transport_layer_bio_new(BIO* bio)
2114 BIO_set_flags(bio, BIO_FLAGS_SHOULD_RETRY);
2115 BIO_set_init(bio, 1);
2119static int transport_layer_bio_free(BIO* bio)
2126 transport_layer_free(layer);
2128 BIO_set_data(bio, NULL);
2129 BIO_set_init(bio, 0);
2130 BIO_set_flags(bio, 0);
2135BIO_METHOD* BIO_s_transport_layer(
void)
2137 static BIO_METHOD* bio_methods = NULL;
2139 if (bio_methods == NULL)
2141 if (!(bio_methods = BIO_meth_new(BIO_TYPE_SIMPLE,
"TransportLayer")))
2144 BIO_meth_set_write(bio_methods, transport_layer_bio_write);
2145 BIO_meth_set_read(bio_methods, transport_layer_bio_read);
2146 BIO_meth_set_puts(bio_methods, transport_layer_bio_puts);
2147 BIO_meth_set_gets(bio_methods, transport_layer_bio_gets);
2148 BIO_meth_set_ctrl(bio_methods, transport_layer_bio_ctrl);
2149 BIO_meth_set_create(bio_methods, transport_layer_bio_new);
2150 BIO_meth_set_destroy(bio_methods, transport_layer_bio_free);
FREERDP_API const char * freerdp_settings_get_server_name(const rdpSettings *settings)
A helper function to return the correct server name.