Blame libfreerdp/core/transport.c

Packit 1fb8d4
/**
Packit 1fb8d4
 * FreeRDP: A Remote Desktop Protocol Implementation
Packit 1fb8d4
 * Network Transport Layer
Packit 1fb8d4
 *
Packit 1fb8d4
 * Copyright 2011 Vic Lee
Packit 1fb8d4
 *
Packit 1fb8d4
 * Licensed under the Apache License, Version 2.0 (the "License");
Packit 1fb8d4
 * you may not use this file except in compliance with the License.
Packit 1fb8d4
 * You may obtain a copy of the License at
Packit 1fb8d4
 *
Packit 1fb8d4
 *     http://www.apache.org/licenses/LICENSE-2.0
Packit 1fb8d4
 *
Packit 1fb8d4
 * Unless required by applicable law or agreed to in writing, software
Packit 1fb8d4
 * distributed under the License is distributed on an "AS IS" BASIS,
Packit 1fb8d4
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
Packit 1fb8d4
 * See the License for the specific language governing permissions and
Packit 1fb8d4
 * limitations under the License.
Packit 1fb8d4
 */
Packit 1fb8d4
Packit 1fb8d4
#ifdef HAVE_CONFIG_H
Packit 1fb8d4
#include "config.h"
Packit 1fb8d4
#endif
Packit 1fb8d4
Packit 1fb8d4
#include <assert.h>
Packit 1fb8d4
Packit 1fb8d4
#include <winpr/crt.h>
Packit 1fb8d4
#include <winpr/synch.h>
Packit 1fb8d4
#include <winpr/print.h>
Packit 1fb8d4
#include <winpr/stream.h>
Packit 1fb8d4
#include <winpr/winsock.h>
Packit 1fb8d4
#include <winpr/crypto.h>
Packit 1fb8d4
Packit 1fb8d4
#include <freerdp/log.h>
Packit 1fb8d4
#include <freerdp/error.h>
Packit 1fb8d4
#include <freerdp/utils/ringbuffer.h>
Packit 1fb8d4
Packit 1fb8d4
#include <openssl/bio.h>
Packit 1fb8d4
#include <time.h>
Packit 1fb8d4
#include <errno.h>
Packit 1fb8d4
#include <fcntl.h>
Packit 1fb8d4
Packit 1fb8d4
#ifndef _WIN32
Packit 1fb8d4
#include <netdb.h>
Packit 1fb8d4
#include <sys/socket.h>
Packit 1fb8d4
#endif /* _WIN32 */
Packit 1fb8d4
Packit 1fb8d4
#ifdef HAVE_VALGRIND_MEMCHECK_H
Packit 1fb8d4
#include <valgrind/memcheck.h>
Packit 1fb8d4
#endif
Packit 1fb8d4
Packit 1fb8d4
#include "tpkt.h"
Packit 1fb8d4
#include "fastpath.h"
Packit 1fb8d4
#include "transport.h"
Packit 1fb8d4
#include "rdp.h"
Packit 1fb8d4
#include "proxy.h"
Packit 1fb8d4
Packit 1fb8d4
#define TAG FREERDP_TAG("core.transport")
Packit 1fb8d4
Packit 1fb8d4
#define BUFFER_SIZE 16384
Packit 1fb8d4
Packit 1fb8d4
#ifdef WITH_GSSAPI
Packit 1fb8d4
Packit 1fb8d4
#include <krb5.h>
Packit 1fb8d4
#include <winpr/library.h>
Packit 1fb8d4
static UINT32 transport_krb5_check_account(rdpTransport* transport, char* username, char* domain,
Packit Service 5a9772
                                           char* passwd)
Packit 1fb8d4
{
Packit 1fb8d4
	krb5_error_code ret;
Packit 1fb8d4
	krb5_context context = NULL;
Packit 1fb8d4
	krb5_principal principal = NULL;
Packit 1fb8d4
	char address[256];
Packit 1fb8d4
	krb5_ccache ccache;
Packit 1fb8d4
	krb5_init_creds_context ctx = NULL;
Packit 1fb8d4
	_snprintf(address, sizeof(address), "%s@%s", username, domain);
Packit 1fb8d4
Packit 1fb8d4
	/* Create a krb5 library context */
Packit 1fb8d4
	if ((ret = krb5_init_context(&context)) != 0)
Packit 1fb8d4
		WLog_Print(transport->log, WLOG_ERROR, "krb5_init_context failed with error %d", (int)ret);
Packit 1fb8d4
	else if ((ret = krb5_parse_name_flags(context, address, 0, &principal)) != 0)
Packit Service 5a9772
		WLog_Print(transport->log, WLOG_ERROR, "krb5_parse_name_flags failed with error %d",
Packit Service 5a9772
		           (int)ret);
Packit 1fb8d4
	/* Find a credential cache with a specified client principal */
Packit 1fb8d4
	else if ((ret = krb5_cc_cache_match(context, principal, &ccache)) != 0)
Packit 1fb8d4
	{
Packit 1fb8d4
		if ((ret = krb5_cc_default(context, &ccache)) != 0)
Packit Service 5a9772
			WLog_Print(transport->log, WLOG_ERROR,
Packit Service 5a9772
			           "krb5 failed to resolve credentials cache with error %d", (int)ret);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (ret != KRB5KDC_ERR_NONE)
Packit 1fb8d4
		goto out;
Packit 1fb8d4
	/* Create a context for acquiring initial credentials */
Packit 1fb8d4
	else if ((ret = krb5_init_creds_init(context, principal, NULL, NULL, 0, NULL, &ctx)) != 0)
Packit 1fb8d4
	{
Packit 1fb8d4
		WLog_Print(transport->log, WLOG_WARN, "krb5_init_creds_init returned error %d", (int)ret);
Packit 1fb8d4
		goto out;
Packit 1fb8d4
	}
Packit 1fb8d4
	/* Set a password for acquiring initial credentials */
Packit 1fb8d4
	else if ((ret = krb5_init_creds_set_password(context, ctx, passwd)) != 0)
Packit 1fb8d4
	{
Packit Service 5a9772
		WLog_Print(transport->log, WLOG_WARN, "krb5_init_creds_set_password returned error %d",
Packit Service 5a9772
		           ret);
Packit 1fb8d4
		goto out;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	/* Acquire credentials using an initial credential context */
Packit 1fb8d4
	ret = krb5_init_creds_get(context, ctx);
Packit 1fb8d4
out:
Packit 1fb8d4
Packit 1fb8d4
	switch (ret)
Packit 1fb8d4
	{
Packit 1fb8d4
		case KRB5KDC_ERR_NONE:
Packit 1fb8d4
			break;
Packit 1fb8d4
Packit 1fb8d4
		case KRB5_KDC_UNREACH:
Packit 1fb8d4
			WLog_Print(transport->log, WLOG_WARN, "krb5_init_creds_get: KDC unreachable");
Packit 1fb8d4
			ret = FREERDP_ERROR_CONNECT_KDC_UNREACHABLE;
Packit 1fb8d4
			break;
Packit 1fb8d4
Packit 1fb8d4
		case KRB5KRB_AP_ERR_BAD_INTEGRITY:
Packit 1fb8d4
		case KRB5KRB_AP_ERR_MODIFIED:
Packit 1fb8d4
		case KRB5KDC_ERR_PREAUTH_FAILED:
Packit 1fb8d4
		case KRB5_GET_IN_TKT_LOOP:
Packit 1fb8d4
			WLog_Print(transport->log, WLOG_WARN, "krb5_init_creds_get: Password incorrect");
Packit 1fb8d4
			ret = FREERDP_ERROR_AUTHENTICATION_FAILED;
Packit 1fb8d4
			break;
Packit 1fb8d4
Packit 1fb8d4
		case KRB5KDC_ERR_KEY_EXP:
Packit 1fb8d4
			WLog_Print(transport->log, WLOG_WARN, "krb5_init_creds_get: Password has expired");
Packit 1fb8d4
			ret = FREERDP_ERROR_CONNECT_PASSWORD_EXPIRED;
Packit 1fb8d4
			break;
Packit 1fb8d4
Packit 1fb8d4
		case KRB5KDC_ERR_CLIENT_REVOKED:
Packit 1fb8d4
			WLog_Print(transport->log, WLOG_WARN, "krb5_init_creds_get: Password revoked");
Packit 1fb8d4
			ret = FREERDP_ERROR_CONNECT_CLIENT_REVOKED;
Packit 1fb8d4
			break;
Packit 1fb8d4
Packit 1fb8d4
		case KRB5KDC_ERR_POLICY:
Packit 1fb8d4
			ret = FREERDP_ERROR_INSUFFICIENT_PRIVILEGES;
Packit 1fb8d4
			break;
Packit 1fb8d4
Packit 1fb8d4
		default:
Packit 1fb8d4
			WLog_Print(transport->log, WLOG_WARN, "krb5_init_creds_get");
Packit 1fb8d4
			ret = FREERDP_ERROR_CONNECT_TRANSPORT_FAILED;
Packit 1fb8d4
			break;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (ctx)
Packit 1fb8d4
		krb5_init_creds_free(context, ctx);
Packit 1fb8d4
Packit 1fb8d4
	krb5_free_context(context);
Packit 1fb8d4
	return ret;
Packit 1fb8d4
}
Packit 1fb8d4
#endif /* WITH_GSSAPI */
Packit 1fb8d4
Packit 1fb8d4
static void transport_ssl_cb(SSL* ssl, int where, int ret)
Packit 1fb8d4
{
Packit 1fb8d4
	if (where & SSL_CB_ALERT)
Packit 1fb8d4
	{
Packit Service 5a9772
		rdpTransport* transport = (rdpTransport*)SSL_get_app_data(ssl);
Packit 1fb8d4
Packit 1fb8d4
		switch (ret)
Packit 1fb8d4
		{
Packit 1fb8d4
			case (SSL3_AL_FATAL << 8) | SSL_AD_ACCESS_DENIED:
Packit Service 5a9772
			{
Packit Service 5a9772
				if (!freerdp_get_last_error(transport->context))
Packit 1fb8d4
				{
Packit Service 5a9772
					WLog_Print(transport->log, WLOG_ERROR, "%s: ACCESS DENIED", __FUNCTION__);
Packit Service 5a9772
					freerdp_set_last_error_log(transport->context,
Packit Service 5a9772
					                           FREERDP_ERROR_AUTHENTICATION_FAILED);
Packit 1fb8d4
				}
Packit Service 5a9772
			}
Packit Service 5a9772
			break;
Packit 1fb8d4
Packit 1fb8d4
			case (SSL3_AL_FATAL << 8) | SSL_AD_INTERNAL_ERROR:
Packit Service 5a9772
			{
Packit Service 5a9772
				if (transport->NlaMode)
Packit 1fb8d4
				{
Packit Service 5a9772
					UINT32 kret = 0;
Packit 1fb8d4
#ifdef WITH_GSSAPI
Packit 1fb8d4
Packit Service 5a9772
					if ((strlen(transport->settings->Domain) != 0) &&
Packit Service 5a9772
					    (strncmp(transport->settings->Domain, ".", 1) != 0))
Packit Service 5a9772
					{
Packit Service 5a9772
						kret = transport_krb5_check_account(
Packit Service 5a9772
						    transport, transport->settings->Username, transport->settings->Domain,
Packit Service 5a9772
						    transport->settings->Password);
Packit Service 5a9772
					}
Packit Service 5a9772
					else
Packit 1fb8d4
#endif /* WITH_GSSAPI */
Packit Service 5a9772
						kret = FREERDP_ERROR_CONNECT_PASSWORD_CERTAINLY_EXPIRED;
Packit 1fb8d4
Packit Service 5a9772
					freerdp_set_last_error_if_not(transport->context, kret);
Packit Service 5a9772
				}
Packit 1fb8d4
Packit Service 5a9772
				break;
Packit 1fb8d4
Packit 1fb8d4
				case (SSL3_AL_WARNING << 8) | SSL3_AD_CLOSE_NOTIFY:
Packit 1fb8d4
					break;
Packit 1fb8d4
Packit 1fb8d4
				default:
Packit Service 5a9772
					WLog_Print(transport->log, WLOG_WARN,
Packit Service 5a9772
					           "Unhandled SSL error (where=%d, ret=%d [%s, %s])", where, ret,
Packit 1fb8d4
					           SSL_alert_type_string_long(ret), SSL_alert_desc_string_long(ret));
Packit 1fb8d4
					break;
Packit Service 5a9772
			}
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
wStream* transport_send_stream_init(rdpTransport* transport, int size)
Packit 1fb8d4
{
Packit 1fb8d4
	wStream* s;
Packit 1fb8d4
Packit 1fb8d4
	if (!(s = StreamPool_Take(transport->ReceivePool, size)))
Packit 1fb8d4
		return NULL;
Packit 1fb8d4
Packit 1fb8d4
	if (!Stream_EnsureCapacity(s, size))
Packit 1fb8d4
	{
Packit 1fb8d4
		Stream_Release(s);
Packit 1fb8d4
		return NULL;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	Stream_SetPosition(s, 0);
Packit 1fb8d4
	return s;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_attach(rdpTransport* transport, int sockfd)
Packit 1fb8d4
{
Packit 1fb8d4
	BIO* socketBio = NULL;
Packit 1fb8d4
	BIO* bufferedBio;
Packit 1fb8d4
	socketBio = BIO_new(BIO_s_simple_socket());
Packit 1fb8d4
Packit 1fb8d4
	if (!socketBio)
Packit 1fb8d4
		goto fail;
Packit 1fb8d4
Packit 1fb8d4
	BIO_set_fd(socketBio, sockfd, BIO_CLOSE);
Packit 1fb8d4
	bufferedBio = BIO_new(BIO_s_buffered_socket());
Packit 1fb8d4
Packit 1fb8d4
	if (!bufferedBio)
Packit 1fb8d4
		goto fail;
Packit 1fb8d4
Packit 1fb8d4
	bufferedBio = BIO_push(bufferedBio, socketBio);
Packit 1fb8d4
	transport->frontBio = bufferedBio;
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
fail:
Packit 1fb8d4
Packit 1fb8d4
	if (socketBio)
Packit 1fb8d4
		BIO_free_all(socketBio);
Packit 1fb8d4
	else
Packit 1fb8d4
		close(sockfd);
Packit 1fb8d4
Packit 1fb8d4
	return FALSE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_connect_rdp(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	/* RDP encryption */
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_connect_tls(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	int tlsStatus;
Packit 1fb8d4
	rdpTls* tls = NULL;
Packit 1fb8d4
	rdpContext* context = transport->context;
Packit 1fb8d4
	rdpSettings* settings = transport->settings;
Packit 1fb8d4
Packit 1fb8d4
	if (!(tls = tls_new(settings)))
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit 1fb8d4
	transport->tls = tls;
Packit 1fb8d4
Packit 1fb8d4
	if (transport->GatewayEnabled)
Packit 1fb8d4
		transport->layer = TRANSPORT_LAYER_TSG_TLS;
Packit 1fb8d4
	else
Packit 1fb8d4
		transport->layer = TRANSPORT_LAYER_TLS;
Packit 1fb8d4
Packit 1fb8d4
	tls->hostname = settings->ServerHostname;
Packit 1fb8d4
	tls->port = settings->ServerPort;
Packit 1fb8d4
Packit 1fb8d4
	if (tls->port == 0)
Packit 1fb8d4
		tls->port = 3389;
Packit 1fb8d4
Packit 1fb8d4
	tls->isGatewayTransport = FALSE;
Packit 1fb8d4
	tlsStatus = tls_connect(tls, transport->frontBio);
Packit 1fb8d4
Packit 1fb8d4
	if (tlsStatus < 1)
Packit 1fb8d4
	{
Packit 1fb8d4
		if (tlsStatus < 0)
Packit 1fb8d4
		{
Packit Service 5a9772
			freerdp_set_last_error_if_not(context, FREERDP_ERROR_TLS_CONNECT_FAILED);
Packit 1fb8d4
		}
Packit 1fb8d4
		else
Packit 1fb8d4
		{
Packit Service 5a9772
			freerdp_set_last_error_if_not(context, FREERDP_ERROR_CONNECT_CANCELLED);
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	transport->frontBio = tls->bio;
Packit Service 5a9772
	BIO_callback_ctrl(tls->bio, BIO_CTRL_SET_CALLBACK, (bio_info_cb*)(void*)transport_ssl_cb);
Packit 1fb8d4
	SSL_set_app_data(tls->ssl, transport);
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->frontBio)
Packit 1fb8d4
	{
Packit 1fb8d4
		WLog_Print(transport->log, WLOG_ERROR, "unable to prepend a filtering TLS bio");
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_connect_nla(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	rdpContext* context = transport->context;
Packit 1fb8d4
	rdpSettings* settings = context->settings;
Packit 1fb8d4
	freerdp* instance = context->instance;
Packit 1fb8d4
	rdpRdp* rdp = context->rdp;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport_connect_tls(transport))
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit 1fb8d4
	if (!settings->Authentication)
Packit 1fb8d4
		return TRUE;
Packit 1fb8d4
Packit Service 5a9772
	nla_free(rdp->nla);
Packit 1fb8d4
	rdp->nla = nla_new(instance, transport, settings);
Packit 1fb8d4
Packit 1fb8d4
	if (!rdp->nla)
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit 1fb8d4
	transport_set_nla_mode(transport, TRUE);
Packit 1fb8d4
Packit 1fb8d4
	if (settings->AuthenticationServiceClass)
Packit 1fb8d4
	{
Packit Service 5a9772
		if (!nla_set_service_principal(rdp->nla, nla_make_spn(settings->AuthenticationServiceClass,
Packit Service 5a9772
		                                                      settings->ServerHostname)))
Packit 1fb8d4
			return FALSE;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (nla_client_begin(rdp->nla) < 0)
Packit 1fb8d4
	{
Packit 1fb8d4
		WLog_Print(transport->log, WLOG_ERROR, "NLA begin failed");
Packit 1fb8d4
Packit Service 5a9772
		freerdp_set_last_error_if_not(context, FREERDP_ERROR_AUTHENTICATION_FAILED);
Packit 1fb8d4
Packit 1fb8d4
		transport_set_nla_mode(transport, FALSE);
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	rdp_client_transition_to_state(rdp, CONNECTION_STATE_NLA);
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
BOOL transport_connect(rdpTransport* transport, const char* hostname, UINT16 port, int timeout)
Packit 1fb8d4
{
Packit 1fb8d4
	int sockfd;
Packit 1fb8d4
	BOOL status = FALSE;
Packit 1fb8d4
	rdpSettings* settings = transport->settings;
Packit 1fb8d4
	rdpContext* context = transport->context;
Packit 1fb8d4
	BOOL rpcFallback = !settings->GatewayHttpTransport;
Packit 1fb8d4
Packit 1fb8d4
	if (transport->GatewayEnabled)
Packit 1fb8d4
	{
Packit 1fb8d4
		if (!status && settings->GatewayHttpTransport)
Packit 1fb8d4
		{
Packit 1fb8d4
			transport->rdg = rdg_new(context);
Packit 1fb8d4
Packit 1fb8d4
			if (!transport->rdg)
Packit 1fb8d4
				return FALSE;
Packit 1fb8d4
Packit 1fb8d4
			status = rdg_connect(transport->rdg, timeout, &rpcFallback);
Packit 1fb8d4
Packit 1fb8d4
			if (status)
Packit 1fb8d4
			{
Packit 1fb8d4
				transport->frontBio = rdg_get_front_bio_and_take_ownership(transport->rdg);
Packit 1fb8d4
				BIO_set_nonblock(transport->frontBio, 0);
Packit 1fb8d4
				transport->layer = TRANSPORT_LAYER_TSG;
Packit 1fb8d4
				status = TRUE;
Packit 1fb8d4
			}
Packit 1fb8d4
			else
Packit 1fb8d4
			{
Packit 1fb8d4
				rdg_free(transport->rdg);
Packit 1fb8d4
				transport->rdg = NULL;
Packit 1fb8d4
			}
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		if (!status && settings->GatewayRpcTransport && rpcFallback)
Packit 1fb8d4
		{
Packit 1fb8d4
			transport->tsg = tsg_new(transport);
Packit 1fb8d4
Packit 1fb8d4
			if (!transport->tsg)
Packit 1fb8d4
				return FALSE;
Packit 1fb8d4
Packit Service 5a9772
			/* Reset error condition from RDG */
Packit Service 5a9772
			freerdp_set_last_error_log(context, FREERDP_ERROR_SUCCESS);
Packit 1fb8d4
			status = tsg_connect(transport->tsg, hostname, port, timeout);
Packit 1fb8d4
Packit 1fb8d4
			if (status)
Packit 1fb8d4
			{
Packit 1fb8d4
				transport->frontBio = tsg_get_bio(transport->tsg);
Packit 1fb8d4
				transport->layer = TRANSPORT_LAYER_TSG;
Packit 1fb8d4
				status = TRUE;
Packit 1fb8d4
			}
Packit 1fb8d4
			else
Packit 1fb8d4
			{
Packit 1fb8d4
				tsg_free(transport->tsg);
Packit 1fb8d4
				transport->tsg = NULL;
Packit 1fb8d4
			}
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
	else
Packit 1fb8d4
	{
Packit 1fb8d4
		UINT16 peerPort;
Packit Service 5a9772
		const char *proxyHostname, *proxyUsername, *proxyPassword;
Packit Service 5a9772
		BOOL isProxyConnection =
Packit Service 5a9772
		    proxy_prepare(settings, &proxyHostname, &peerPort, &proxyUsername, &proxyPassword);
Packit 1fb8d4
Packit 1fb8d4
		if (isProxyConnection)
Packit 1fb8d4
			sockfd = freerdp_tcp_connect(context, settings, proxyHostname, peerPort, timeout);
Packit 1fb8d4
		else
Packit 1fb8d4
			sockfd = freerdp_tcp_connect(context, settings, hostname, port, timeout);
Packit 1fb8d4
Packit 1fb8d4
		if (sockfd < 0)
Packit 1fb8d4
			return FALSE;
Packit 1fb8d4
Packit 1fb8d4
		if (!transport_attach(transport, sockfd))
Packit 1fb8d4
			return FALSE;
Packit 1fb8d4
Packit 1fb8d4
		if (isProxyConnection)
Packit 1fb8d4
		{
Packit Service 5a9772
			if (!proxy_connect(settings, transport->frontBio, proxyUsername, proxyPassword,
Packit Service 5a9772
			                   hostname, port))
Packit 1fb8d4
				return FALSE;
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		status = TRUE;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return status;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_accept_rdp(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	/* RDP encryption */
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_accept_tls(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	rdpSettings* settings = transport->settings;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->tls)
Packit 1fb8d4
		transport->tls = tls_new(transport->settings);
Packit 1fb8d4
Packit 1fb8d4
	transport->layer = TRANSPORT_LAYER_TLS;
Packit 1fb8d4
Packit 1fb8d4
	if (!tls_accept(transport->tls, transport->frontBio, settings))
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit 1fb8d4
	transport->frontBio = transport->tls->bio;
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_accept_nla(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	rdpSettings* settings = transport->settings;
Packit Service 5a9772
	freerdp* instance = (freerdp*)settings->instance;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->tls)
Packit 1fb8d4
		transport->tls = tls_new(transport->settings);
Packit 1fb8d4
Packit 1fb8d4
	transport->layer = TRANSPORT_LAYER_TLS;
Packit 1fb8d4
Packit 1fb8d4
	if (!tls_accept(transport->tls, transport->frontBio, settings))
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit 1fb8d4
	transport->frontBio = transport->tls->bio;
Packit 1fb8d4
Packit 1fb8d4
	/* Network Level Authentication */
Packit 1fb8d4
Packit 1fb8d4
	if (!settings->Authentication)
Packit 1fb8d4
		return TRUE;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->nla)
Packit 1fb8d4
	{
Packit 1fb8d4
		transport->nla = nla_new(instance, transport, settings);
Packit 1fb8d4
		transport_set_nla_mode(transport, TRUE);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (nla_authenticate(transport->nla) < 0)
Packit 1fb8d4
	{
Packit 1fb8d4
		WLog_Print(transport->log, WLOG_ERROR, "client authentication failure");
Packit 1fb8d4
		transport_set_nla_mode(transport, FALSE);
Packit 1fb8d4
		nla_free(transport->nla);
Packit 1fb8d4
		transport->nla = NULL;
Packit 1fb8d4
		tls_set_alert_code(transport->tls, TLS_ALERT_LEVEL_FATAL,
Packit 1fb8d4
		                   TLS_ALERT_DESCRIPTION_ACCESS_DENIED);
Packit 1fb8d4
		tls_send_alert(transport->tls);
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	/* don't free nla module yet, we need to copy the credentials from it first */
Packit 1fb8d4
	transport_set_nla_mode(transport, FALSE);
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
#define WLog_ERR_BIO(transport, biofunc, bio) \
Packit 1fb8d4
	transport_bio_error_log(transport, biofunc, bio, __FILE__, __FUNCTION__, __LINE__)
Packit 1fb8d4
Packit Service 5a9772
static void transport_bio_error_log(rdpTransport* transport, LPCSTR biofunc, BIO* bio, LPCSTR file,
Packit Service 5a9772
                                    LPCSTR func, DWORD line)
Packit 1fb8d4
{
Packit 1fb8d4
	unsigned long sslerr;
Packit 1fb8d4
	char* buf;
Packit 1fb8d4
	int saveerrno;
Packit 1fb8d4
	DWORD level;
Packit 1fb8d4
	saveerrno = errno;
Packit 1fb8d4
	level = WLOG_ERROR;
Packit 1fb8d4
Packit 1fb8d4
	if (level < WLog_GetLogLevel(transport->log))
Packit 1fb8d4
		return;
Packit 1fb8d4
Packit 1fb8d4
	if (ERR_peek_error() == 0)
Packit 1fb8d4
	{
Packit 1fb8d4
		const char* fmt = "%s returned a system error %d: %s";
Packit 1fb8d4
		WLog_PrintMessage(transport->log, WLOG_MESSAGE_TEXT, level, line, file, func, fmt, biofunc,
Packit 1fb8d4
		                  saveerrno, strerror(saveerrno));
Packit 1fb8d4
		return;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	buf = malloc(120);
Packit 1fb8d4
Packit 1fb8d4
	if (buf)
Packit 1fb8d4
	{
Packit 1fb8d4
		const char* fmt = "%s returned an error: %s";
Packit 1fb8d4
Packit 1fb8d4
		while ((sslerr = ERR_get_error()))
Packit 1fb8d4
		{
Packit 1fb8d4
			ERR_error_string_n(sslerr, buf, 120);
Packit Service 5a9772
			WLog_PrintMessage(transport->log, WLOG_MESSAGE_TEXT, level, line, file, func, fmt,
Packit Service 5a9772
			                  biofunc, buf);
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		free(buf);
Packit 1fb8d4
	}
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
static SSIZE_T transport_read_layer(rdpTransport* transport, BYTE* data, size_t bytes)
Packit 1fb8d4
{
Packit Service 5a9772
	SSIZE_T read = 0;
Packit Service 5a9772
	rdpRdp* rdp = transport->context->rdp;
Packit 1fb8d4
Packit Service 5a9772
	if (!transport->frontBio || (bytes > SSIZE_MAX))
Packit 1fb8d4
	{
Packit 1fb8d4
		transport->layer = TRANSPORT_LAYER_CLOSED;
Packit Service 5a9772
		freerdp_set_last_error_if_not(transport->context, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
Packit 1fb8d4
		return -1;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit Service 5a9772
	while (read < (SSIZE_T)bytes)
Packit 1fb8d4
	{
Packit Service 5a9772
		const SSIZE_T tr = (SSIZE_T)bytes - read;
Packit Service 5a9772
		int r = (int)((tr > INT_MAX) ? INT_MAX : tr);
Packit Service 5a9772
		int status = BIO_read(transport->frontBio, data + read, r);
Packit 1fb8d4
Packit 1fb8d4
		if (status <= 0)
Packit 1fb8d4
		{
Packit 1fb8d4
			if (!transport->frontBio || !BIO_should_retry(transport->frontBio))
Packit 1fb8d4
			{
Packit 1fb8d4
				/* something unexpected happened, let's close */
Packit 1fb8d4
				if (!transport->frontBio)
Packit 1fb8d4
				{
Packit 1fb8d4
					WLog_Print(transport->log, WLOG_ERROR, "BIO_read: transport->frontBio null");
Packit 1fb8d4
					return -1;
Packit 1fb8d4
				}
Packit 1fb8d4
Packit 1fb8d4
				WLog_ERR_BIO(transport, "BIO_read", transport->frontBio);
Packit 1fb8d4
				transport->layer = TRANSPORT_LAYER_CLOSED;
Packit Service 5a9772
				freerdp_set_last_error_if_not(transport->context,
Packit Service 5a9772
				                              FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
Packit 1fb8d4
				return -1;
Packit 1fb8d4
			}
Packit 1fb8d4
Packit 1fb8d4
			/* non blocking will survive a partial read */
Packit 1fb8d4
			if (!transport->blocking)
Packit 1fb8d4
				return read;
Packit 1fb8d4
Packit Service 5a9772
			/* blocking means that we can't continue until we have read the number of requested
Packit Service 5a9772
			 * bytes */
Packit 1fb8d4
			if (BIO_wait_read(transport->frontBio, 100) < 0)
Packit 1fb8d4
			{
Packit 1fb8d4
				WLog_ERR_BIO(transport, "BIO_wait_read", transport->frontBio);
Packit 1fb8d4
				return -1;
Packit 1fb8d4
			}
Packit 1fb8d4
Packit 1fb8d4
			continue;
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
#ifdef HAVE_VALGRIND_MEMCHECK_H
Packit 1fb8d4
		VALGRIND_MAKE_MEM_DEFINED(data + read, bytes - read);
Packit 1fb8d4
#endif
Packit 1fb8d4
		read += status;
Packit Service 5a9772
		rdp->inBytes += status;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return read;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
/**
Packit 1fb8d4
 * @brief Tries to read toRead bytes from the specified transport
Packit 1fb8d4
 *
Packit 1fb8d4
 * Try to read toRead bytes from the transport to the stream.
Packit Service 5a9772
 * In case it was not possible to read toRead bytes 0 is returned. The stream is always advanced by
Packit Service 5a9772
 * the number of bytes read.
Packit 1fb8d4
 *
Packit 1fb8d4
 * The function assumes that the stream has enough capacity to hold the data.
Packit 1fb8d4
 *
Packit 1fb8d4
 * @param[in] transport rdpTransport
Packit 1fb8d4
 * @param[in] s wStream
Packit 1fb8d4
 * @param[in] toRead number of bytes to read
Packit 1fb8d4
 * @return < 0 on error; 0 if not enough data is available (non blocking mode); 1 toRead bytes read
Packit 1fb8d4
 */
Packit Service 5a9772
static SSIZE_T transport_read_layer_bytes(rdpTransport* transport, wStream* s, size_t toRead)
Packit 1fb8d4
{
Packit Service 5a9772
	SSIZE_T status;
Packit Service 5a9772
	if (toRead > SSIZE_MAX)
Packit Service 5a9772
		return 0;
Packit Service 5a9772
Packit 1fb8d4
	status = transport_read_layer(transport, Stream_Pointer(s), toRead);
Packit 1fb8d4
Packit 1fb8d4
	if (status <= 0)
Packit 1fb8d4
		return status;
Packit 1fb8d4
Packit Service 5a9772
	Stream_Seek(s, (size_t)status);
Packit Service 5a9772
	return status == (SSIZE_T)toRead ? 1 : 0;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
/**
Packit 1fb8d4
 * @brief Try to read a complete PDU (NLA, fast-path or tpkt) from the underlying transport.
Packit 1fb8d4
 *
Packit 1fb8d4
 * If possible a complete PDU is read, in case of non blocking transport this might not succeed.
Packit 1fb8d4
 * Except in case of an error the passed stream will point to the last byte read (correct
Packit 1fb8d4
 * position). When the pdu read is completed the stream is sealed and the pointer set to 0
Packit 1fb8d4
 *
Packit 1fb8d4
 * @param[in] transport rdpTransport
Packit 1fb8d4
 * @param[in] s wStream
Packit 1fb8d4
 * @return < 0 on error; 0 if not enough data is available (non blocking mode); > 0 number of
Packit 1fb8d4
 * bytes of the *complete* pdu read
Packit 1fb8d4
 */
Packit 1fb8d4
int transport_read_pdu(rdpTransport* transport, wStream* s)
Packit 1fb8d4
{
Packit 1fb8d4
	int status;
Packit 1fb8d4
	size_t position;
Packit Service 5a9772
	size_t pduLength;
Packit 1fb8d4
	BYTE* header;
Packit 1fb8d4
	pduLength = 0;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport)
Packit 1fb8d4
		return -1;
Packit 1fb8d4
Packit 1fb8d4
	if (!s)
Packit 1fb8d4
		return -1;
Packit 1fb8d4
Packit 1fb8d4
	position = Stream_GetPosition(s);
Packit 1fb8d4
Packit 1fb8d4
	/* Make sure there is enough space for the longest header within the stream */
Packit 1fb8d4
	if (!Stream_EnsureCapacity(s, 4))
Packit 1fb8d4
		return -1;
Packit 1fb8d4
Packit 1fb8d4
	/* Make sure at least two bytes are read for further processing */
Packit Service 5a9772
	if (position < 2 && (status = transport_read_layer_bytes(transport, s, 2 - position)) != 1)
Packit 1fb8d4
	{
Packit 1fb8d4
		/* No data available at the moment */
Packit 1fb8d4
		return status;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	/* update position value for further checks */
Packit 1fb8d4
	position = Stream_GetPosition(s);
Packit 1fb8d4
	header = Stream_Buffer(s);
Packit 1fb8d4
Packit 1fb8d4
	if (transport->NlaMode)
Packit 1fb8d4
	{
Packit 1fb8d4
		/*
Packit 1fb8d4
		 * In case NlaMode is set TSRequest package(s) are expected
Packit 1fb8d4
		 * 0x30 = DER encoded data with these bits set:
Packit 1fb8d4
		 * bit 6 P/C constructed
Packit 1fb8d4
		 * bit 5 tag number - sequence
Packit 1fb8d4
		 */
Packit 1fb8d4
		if (header[0] == 0x30)
Packit 1fb8d4
		{
Packit 1fb8d4
			/* TSRequest (NLA) */
Packit 1fb8d4
			if (header[1] & 0x80)
Packit 1fb8d4
			{
Packit 1fb8d4
				if ((header[1] & ~(0x80)) == 1)
Packit 1fb8d4
				{
Packit 1fb8d4
					/* check for header bytes already was readed in previous calls */
Packit Service 5a9772
					if (position < 3 &&
Packit Service 5a9772
					    (status = transport_read_layer_bytes(transport, s, 3 - position)) != 1)
Packit 1fb8d4
						return status;
Packit 1fb8d4
Packit 1fb8d4
					pduLength = header[2];
Packit 1fb8d4
					pduLength += 3;
Packit 1fb8d4
				}
Packit 1fb8d4
				else if ((header[1] & ~(0x80)) == 2)
Packit 1fb8d4
				{
Packit 1fb8d4
					/* check for header bytes already was readed in previous calls */
Packit Service 5a9772
					if (position < 4 &&
Packit Service 5a9772
					    (status = transport_read_layer_bytes(transport, s, 4 - position)) != 1)
Packit 1fb8d4
						return status;
Packit 1fb8d4
Packit 1fb8d4
					pduLength = (header[2] << 8) | header[3];
Packit 1fb8d4
					pduLength += 4;
Packit 1fb8d4
				}
Packit 1fb8d4
				else
Packit 1fb8d4
				{
Packit 1fb8d4
					WLog_Print(transport->log, WLOG_ERROR, "Error reading TSRequest!");
Packit 1fb8d4
					return -1;
Packit 1fb8d4
				}
Packit 1fb8d4
			}
Packit 1fb8d4
			else
Packit 1fb8d4
			{
Packit 1fb8d4
				pduLength = header[1];
Packit 1fb8d4
				pduLength += 2;
Packit 1fb8d4
			}
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
	else
Packit 1fb8d4
	{
Packit 1fb8d4
		if (header[0] == 0x03)
Packit 1fb8d4
		{
Packit 1fb8d4
			/* TPKT header */
Packit 1fb8d4
			/* check for header bytes already was readed in previous calls */
Packit Service 5a9772
			if (position < 4 &&
Packit Service 5a9772
			    (status = transport_read_layer_bytes(transport, s, 4 - position)) != 1)
Packit 1fb8d4
				return status;
Packit 1fb8d4
Packit 1fb8d4
			pduLength = (header[2] << 8) | header[3];
Packit 1fb8d4
Packit 1fb8d4
			/* min and max values according to ITU-T Rec. T.123 (01/2007) section 8 */
Packit 1fb8d4
			if (pduLength < 7 || pduLength > 0xFFFF)
Packit 1fb8d4
			{
Packit Service 5a9772
				WLog_Print(transport->log, WLOG_ERROR, "tpkt - invalid pduLength: %" PRIdz,
Packit Service 5a9772
				           pduLength);
Packit 1fb8d4
				return -1;
Packit 1fb8d4
			}
Packit 1fb8d4
		}
Packit 1fb8d4
		else
Packit 1fb8d4
		{
Packit 1fb8d4
			/* Fast-Path Header */
Packit 1fb8d4
			if (header[1] & 0x80)
Packit 1fb8d4
			{
Packit 1fb8d4
				/* check for header bytes already was readed in previous calls */
Packit Service 5a9772
				if (position < 3 &&
Packit Service 5a9772
				    (status = transport_read_layer_bytes(transport, s, 3 - position)) != 1)
Packit 1fb8d4
					return status;
Packit 1fb8d4
Packit 1fb8d4
				pduLength = ((header[1] & 0x7F) << 8) | header[2];
Packit 1fb8d4
			}
Packit 1fb8d4
			else
Packit 1fb8d4
				pduLength = header[1];
Packit 1fb8d4
Packit 1fb8d4
			/*
Packit 1fb8d4
			 * fast-path has 7 bits for length so the maximum size, including headers is 0x8000
Packit 1fb8d4
			 * The theoretical minimum fast-path PDU consists only of two header bytes plus one
Packit 1fb8d4
			 * byte for data (e.g. fast-path input synchronize pdu)
Packit 1fb8d4
			 */
Packit 1fb8d4
			if (pduLength < 3 || pduLength > 0x8000)
Packit 1fb8d4
			{
Packit Service 5a9772
				WLog_Print(transport->log, WLOG_ERROR, "fast path - invalid pduLength: %" PRIdz,
Packit Service 5a9772
				           pduLength);
Packit 1fb8d4
				return -1;
Packit 1fb8d4
			}
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (!Stream_EnsureCapacity(s, Stream_GetPosition(s) + pduLength))
Packit 1fb8d4
		return -1;
Packit 1fb8d4
Packit 1fb8d4
	status = transport_read_layer_bytes(transport, s, pduLength - Stream_GetPosition(s));
Packit 1fb8d4
Packit 1fb8d4
	if (status != 1)
Packit 1fb8d4
		return status;
Packit 1fb8d4
Packit 1fb8d4
	if (Stream_GetPosition(s) >= pduLength)
Packit Service 5a9772
		WLog_Packet(transport->log, WLOG_TRACE, Stream_Buffer(s), pduLength, WLOG_PACKET_INBOUND);
Packit 1fb8d4
Packit 1fb8d4
	Stream_SealLength(s);
Packit 1fb8d4
	Stream_SetPosition(s, 0);
Packit 1fb8d4
	return Stream_Length(s);
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
int transport_write(rdpTransport* transport, wStream* s)
Packit 1fb8d4
{
Packit 1fb8d4
	size_t length;
Packit 1fb8d4
	int status = -1;
Packit 1fb8d4
	int writtenlength = 0;
Packit Service 5a9772
	rdpRdp* rdp = transport->context->rdp;
Packit 1fb8d4
Packit 1fb8d4
	if (!s)
Packit 1fb8d4
		return -1;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport)
Packit 1fb8d4
		goto fail;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->frontBio)
Packit 1fb8d4
	{
Packit 1fb8d4
		transport->layer = TRANSPORT_LAYER_CLOSED;
Packit Service 5a9772
		freerdp_set_last_error_if_not(transport->context, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
Packit 1fb8d4
		goto fail;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	EnterCriticalSection(&(transport->WriteLock));
Packit 1fb8d4
	length = Stream_GetPosition(s);
Packit 1fb8d4
	writtenlength = length;
Packit 1fb8d4
	Stream_SetPosition(s, 0);
Packit 1fb8d4
Packit 1fb8d4
	if (length > 0)
Packit 1fb8d4
	{
Packit Service 5a9772
		rdp->outBytes += length;
Packit Service 5a9772
		WLog_Packet(transport->log, WLOG_TRACE, Stream_Buffer(s), length, WLOG_PACKET_OUTBOUND);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	while (length > 0)
Packit 1fb8d4
	{
Packit 1fb8d4
		status = BIO_write(transport->frontBio, Stream_Pointer(s), length);
Packit 1fb8d4
Packit 1fb8d4
		if (status <= 0)
Packit 1fb8d4
		{
Packit 1fb8d4
			/* the buffered BIO that is at the end of the chain always says OK for writing,
Packit 1fb8d4
			 * so a retry means that for any reason we need to read. The most probable
Packit 1fb8d4
			 * is a SSL or TSG BIO in the chain.
Packit 1fb8d4
			 */
Packit 1fb8d4
			if (!BIO_should_retry(transport->frontBio))
Packit 1fb8d4
			{
Packit 1fb8d4
				WLog_ERR_BIO(transport, "BIO_should_retry", transport->frontBio);
Packit 1fb8d4
				goto out_cleanup;
Packit 1fb8d4
			}
Packit 1fb8d4
Packit 1fb8d4
			/* non-blocking can live with blocked IOs */
Packit 1fb8d4
			if (!transport->blocking)
Packit 1fb8d4
			{
Packit 1fb8d4
				WLog_ERR_BIO(transport, "BIO_write", transport->frontBio);
Packit 1fb8d4
				goto out_cleanup;
Packit 1fb8d4
			}
Packit 1fb8d4
Packit 1fb8d4
			if (BIO_wait_write(transport->frontBio, 100) < 0)
Packit 1fb8d4
			{
Packit 1fb8d4
				WLog_ERR_BIO(transport, "BIO_wait_write", transport->frontBio);
Packit 1fb8d4
				status = -1;
Packit 1fb8d4
				goto out_cleanup;
Packit 1fb8d4
			}
Packit 1fb8d4
Packit 1fb8d4
			continue;
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		if (transport->blocking || transport->settings->WaitForOutputBufferFlush)
Packit 1fb8d4
		{
Packit 1fb8d4
			while (BIO_write_blocked(transport->frontBio))
Packit 1fb8d4
			{
Packit 1fb8d4
				if (BIO_wait_write(transport->frontBio, 100) < 0)
Packit 1fb8d4
				{
Packit 1fb8d4
					WLog_Print(transport->log, WLOG_ERROR, "error when selecting for write");
Packit 1fb8d4
					status = -1;
Packit 1fb8d4
					goto out_cleanup;
Packit 1fb8d4
				}
Packit 1fb8d4
Packit 1fb8d4
				if (BIO_flush(transport->frontBio) < 1)
Packit 1fb8d4
				{
Packit 1fb8d4
					WLog_Print(transport->log, WLOG_ERROR, "error when flushing outputBuffer");
Packit 1fb8d4
					status = -1;
Packit 1fb8d4
					goto out_cleanup;
Packit 1fb8d4
				}
Packit 1fb8d4
			}
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		length -= status;
Packit 1fb8d4
		Stream_Seek(s, status);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	transport->written += writtenlength;
Packit 1fb8d4
out_cleanup:
Packit 1fb8d4
Packit 1fb8d4
	if (status < 0)
Packit 1fb8d4
	{
Packit 1fb8d4
		/* A write error indicates that the peer has dropped the connection */
Packit 1fb8d4
		transport->layer = TRANSPORT_LAYER_CLOSED;
Packit Service 5a9772
		freerdp_set_last_error_if_not(transport->context, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	LeaveCriticalSection(&(transport->WriteLock));
Packit 1fb8d4
fail:
Packit 1fb8d4
	Stream_Release(s);
Packit 1fb8d4
	return status;
Packit 1fb8d4
}
Packit 1fb8d4
Packit Service 5a9772
DWORD transport_get_event_handles(rdpTransport* transport, HANDLE* events, DWORD count)
Packit 1fb8d4
{
Packit 1fb8d4
	DWORD nCount = 1; /* always the reread Event */
Packit 1fb8d4
	DWORD tmp;
Packit 1fb8d4
Packit 1fb8d4
	if (events)
Packit 1fb8d4
	{
Packit 1fb8d4
		if (count < 1)
Packit 1fb8d4
		{
Packit Service 5a9772
			WLog_Print(transport->log, WLOG_ERROR, "%s: provided handles array is too small",
Packit Service 5a9772
			           __FUNCTION__);
Packit 1fb8d4
			return 0;
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		events[0] = transport->rereadEvent;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->GatewayEnabled)
Packit 1fb8d4
	{
Packit 1fb8d4
		nCount++;
Packit 1fb8d4
Packit 1fb8d4
		if (events)
Packit 1fb8d4
		{
Packit 1fb8d4
			if (nCount > count)
Packit 1fb8d4
			{
Packit 1fb8d4
				WLog_Print(transport->log, WLOG_ERROR,
Packit Service 5a9772
				           "%s: provided handles array is too small (count=%" PRIu32
Packit Service 5a9772
				           " nCount=%" PRIu32 ")",
Packit 1fb8d4
				           __FUNCTION__, count, nCount);
Packit 1fb8d4
				return 0;
Packit 1fb8d4
			}
Packit 1fb8d4
Packit 1fb8d4
			if (BIO_get_event(transport->frontBio, &events[1]) != 1)
Packit 1fb8d4
			{
Packit Service 5a9772
				WLog_Print(transport->log, WLOG_ERROR, "%s: error getting the frontBio handle",
Packit Service 5a9772
				           __FUNCTION__);
Packit 1fb8d4
				return 0;
Packit 1fb8d4
			}
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
	else
Packit 1fb8d4
	{
Packit 1fb8d4
		if (transport->rdg)
Packit 1fb8d4
		{
Packit 1fb8d4
			tmp = rdg_get_event_handles(transport->rdg, &events[1], count - 1);
Packit 1fb8d4
Packit 1fb8d4
			if (tmp == 0)
Packit 1fb8d4
				return 0;
Packit 1fb8d4
Packit 1fb8d4
			nCount += tmp;
Packit 1fb8d4
		}
Packit 1fb8d4
		else if (transport->tsg)
Packit 1fb8d4
		{
Packit 1fb8d4
			tmp = tsg_get_event_handles(transport->tsg, &events[1], count - 1);
Packit 1fb8d4
Packit 1fb8d4
			if (tmp == 0)
Packit 1fb8d4
				return 0;
Packit 1fb8d4
Packit 1fb8d4
			nCount += tmp;
Packit 1fb8d4
		}
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return nCount;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
void transport_get_fds(rdpTransport* transport, void** rfds, int* rcount)
Packit 1fb8d4
{
Packit 1fb8d4
	DWORD index;
Packit 1fb8d4
	DWORD nCount;
Packit 1fb8d4
	HANDLE events[64];
Packit 1fb8d4
	nCount = transport_get_event_handles(transport, events, 64);
Packit 1fb8d4
	*rcount = nCount + 1;
Packit 1fb8d4
Packit 1fb8d4
	for (index = 0; index < nCount; index++)
Packit 1fb8d4
	{
Packit 1fb8d4
		rfds[index] = GetEventWaitObject(events[index]);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	rfds[nCount] = GetEventWaitObject(transport->rereadEvent);
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_is_write_blocked(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	return BIO_write_blocked(transport->frontBio);
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
int transport_drain_output_buffer(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	BOOL status = FALSE;
Packit 1fb8d4
Packit 1fb8d4
	if (BIO_write_blocked(transport->frontBio))
Packit 1fb8d4
	{
Packit 1fb8d4
		if (BIO_flush(transport->frontBio) < 1)
Packit 1fb8d4
			return -1;
Packit 1fb8d4
Packit 1fb8d4
		status |= BIO_write_blocked(transport->frontBio);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return status;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
int transport_check_fds(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	int status;
Packit 1fb8d4
	int recv_status;
Packit 1fb8d4
	wStream* received;
Packit Service 5a9772
	UINT64 now = GetTickCount64();
Packit Service 5a9772
	UINT64 dueDate = 0;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport)
Packit 1fb8d4
		return -1;
Packit 1fb8d4
Packit Service 5a9772
	if (transport->layer == TRANSPORT_LAYER_CLOSED)
Packit Service 5a9772
	{
Packit Service 5a9772
		WLog_Print(transport->log, WLOG_DEBUG, "transport_check_fds: transport layer closed");
Packit Service 5a9772
		freerdp_set_last_error_if_not(transport->context, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED);
Packit Service 5a9772
		return -1;
Packit Service 5a9772
	}
Packit Service 5a9772
Packit 1fb8d4
	dueDate = now + transport->settings->MaxTimeInCheckLoop;
Packit 1fb8d4
Packit 1fb8d4
	if (transport->haveMoreBytesToRead)
Packit 1fb8d4
	{
Packit 1fb8d4
		transport->haveMoreBytesToRead = FALSE;
Packit 1fb8d4
		ResetEvent(transport->rereadEvent);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	while (now < dueDate)
Packit 1fb8d4
	{
Packit 1fb8d4
		if (freerdp_shall_disconnect(transport->context->instance))
Packit 1fb8d4
		{
Packit 1fb8d4
			return -1;
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		/**
Packit 1fb8d4
		 * Note: transport_read_pdu tries to read one PDU from
Packit 1fb8d4
		 * the transport layer.
Packit 1fb8d4
		 * The ReceiveBuffer might have a position > 0 in case of a non blocking
Packit 1fb8d4
		 * transport. If transport_read_pdu returns 0 the pdu couldn't be read at
Packit 1fb8d4
		 * this point.
Packit 1fb8d4
		 * Note that transport->ReceiveBuffer is replaced after each iteration
Packit 1fb8d4
		 * of this loop with a fresh stream instance from a pool.
Packit 1fb8d4
		 */
Packit 1fb8d4
		if ((status = transport_read_pdu(transport, transport->ReceiveBuffer)) <= 0)
Packit 1fb8d4
		{
Packit 1fb8d4
			if (status < 0)
Packit Service 5a9772
				WLog_Print(transport->log, WLOG_DEBUG,
Packit Service 5a9772
				           "transport_check_fds: transport_read_pdu() - %i", status);
Packit 1fb8d4
Packit 1fb8d4
			return status;
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		received = transport->ReceiveBuffer;
Packit 1fb8d4
Packit 1fb8d4
		if (!(transport->ReceiveBuffer = StreamPool_Take(transport->ReceivePool, 0)))
Packit 1fb8d4
			return -1;
Packit 1fb8d4
Packit 1fb8d4
		/**
Packit 1fb8d4
		 * status:
Packit 1fb8d4
		 * 	-1: error
Packit 1fb8d4
		 * 	 0: success
Packit 1fb8d4
		 * 	 1: redirection
Packit 1fb8d4
		 */
Packit 1fb8d4
		recv_status = transport->ReceiveCallback(transport, received, transport->ReceiveExtra);
Packit 1fb8d4
		Stream_Release(received);
Packit 1fb8d4
Packit 1fb8d4
		/* session redirection or activation */
Packit 1fb8d4
		if (recv_status == 1 || recv_status == 2)
Packit 1fb8d4
		{
Packit 1fb8d4
			return recv_status;
Packit 1fb8d4
		}
Packit 1fb8d4
Packit 1fb8d4
		if (recv_status < 0)
Packit 1fb8d4
		{
Packit Service 5a9772
			WLog_Print(transport->log, WLOG_ERROR,
Packit Service 5a9772
			           "transport_check_fds: transport->ReceiveCallback() - %i", recv_status);
Packit 1fb8d4
			return -1;
Packit 1fb8d4
		}
Packit 1fb8d4
Packit Service 5a9772
		now = GetTickCount64();
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (now >= dueDate)
Packit 1fb8d4
	{
Packit 1fb8d4
		SetEvent(transport->rereadEvent);
Packit 1fb8d4
		transport->haveMoreBytesToRead = TRUE;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	return 0;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_set_blocking_mode(rdpTransport* transport, BOOL blocking)
Packit 1fb8d4
{
Packit 1fb8d4
	transport->blocking = blocking;
Packit 1fb8d4
Packit 1fb8d4
	if (!BIO_set_nonblock(transport->frontBio, blocking ? FALSE : TRUE))
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit 1fb8d4
	return TRUE;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
void transport_set_gateway_enabled(rdpTransport* transport, BOOL GatewayEnabled)
Packit 1fb8d4
{
Packit 1fb8d4
	transport->GatewayEnabled = GatewayEnabled;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
void transport_set_nla_mode(rdpTransport* transport, BOOL NlaMode)
Packit 1fb8d4
{
Packit 1fb8d4
	transport->NlaMode = NlaMode;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
BOOL transport_disconnect(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	BOOL status = TRUE;
Packit 1fb8d4
Packit 1fb8d4
	if (!transport)
Packit 1fb8d4
		return FALSE;
Packit 1fb8d4
Packit 1fb8d4
	if (transport->tls)
Packit 1fb8d4
	{
Packit 1fb8d4
		tls_free(transport->tls);
Packit 1fb8d4
		transport->tls = NULL;
Packit 1fb8d4
	}
Packit 1fb8d4
	else
Packit 1fb8d4
	{
Packit 1fb8d4
		if (transport->frontBio)
Packit 1fb8d4
			BIO_free_all(transport->frontBio);
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (transport->tsg)
Packit 1fb8d4
	{
Packit 1fb8d4
		tsg_free(transport->tsg);
Packit 1fb8d4
		transport->tsg = NULL;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	if (transport->rdg)
Packit 1fb8d4
	{
Packit 1fb8d4
		rdg_free(transport->rdg);
Packit 1fb8d4
		transport->rdg = NULL;
Packit 1fb8d4
	}
Packit 1fb8d4
Packit 1fb8d4
	transport->frontBio = NULL;
Packit 1fb8d4
	transport->layer = TRANSPORT_LAYER_TCP;
Packit 1fb8d4
	return status;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
rdpTransport* transport_new(rdpContext* context)
Packit 1fb8d4
{
Packit 1fb8d4
	rdpTransport* transport;
Packit Service 5a9772
	transport = (rdpTransport*)calloc(1, sizeof(rdpTransport));
Packit 1fb8d4
Packit 1fb8d4
	if (!transport)
Packit 1fb8d4
		return NULL;
Packit 1fb8d4
Packit 1fb8d4
	transport->log = WLog_Get(TAG);
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->log)
Packit 1fb8d4
		goto out_free_transport;
Packit 1fb8d4
Packit 1fb8d4
	transport->context = context;
Packit 1fb8d4
	transport->settings = context->settings;
Packit 1fb8d4
	transport->ReceivePool = StreamPool_New(TRUE, BUFFER_SIZE);
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->ReceivePool)
Packit 1fb8d4
		goto out_free_transport;
Packit 1fb8d4
Packit 1fb8d4
	/* receive buffer for non-blocking read. */
Packit 1fb8d4
	transport->ReceiveBuffer = StreamPool_Take(transport->ReceivePool, 0);
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->ReceiveBuffer)
Packit 1fb8d4
		goto out_free_receivepool;
Packit 1fb8d4
Packit 1fb8d4
	transport->connectedEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
Packit 1fb8d4
Packit Service 5a9772
	if (!transport->connectedEvent || transport->connectedEvent == INVALID_HANDLE_VALUE)
Packit 1fb8d4
		goto out_free_receivebuffer;
Packit 1fb8d4
Packit 1fb8d4
	transport->rereadEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
Packit 1fb8d4
Packit 1fb8d4
	if (!transport->rereadEvent || transport->rereadEvent == INVALID_HANDLE_VALUE)
Packit 1fb8d4
		goto out_free_connectedEvent;
Packit 1fb8d4
Packit 1fb8d4
	transport->haveMoreBytesToRead = FALSE;
Packit 1fb8d4
	transport->blocking = TRUE;
Packit 1fb8d4
	transport->GatewayEnabled = FALSE;
Packit 1fb8d4
	transport->layer = TRANSPORT_LAYER_TCP;
Packit 1fb8d4
Packit 1fb8d4
	if (!InitializeCriticalSectionAndSpinCount(&(transport->ReadLock), 4000))
Packit 1fb8d4
		goto out_free_rereadEvent;
Packit 1fb8d4
Packit 1fb8d4
	if (!InitializeCriticalSectionAndSpinCount(&(transport->WriteLock), 4000))
Packit 1fb8d4
		goto out_free_readlock;
Packit 1fb8d4
Packit 1fb8d4
	return transport;
Packit 1fb8d4
out_free_readlock:
Packit 1fb8d4
	DeleteCriticalSection(&(transport->ReadLock));
Packit 1fb8d4
out_free_rereadEvent:
Packit 1fb8d4
	CloseHandle(transport->rereadEvent);
Packit 1fb8d4
out_free_connectedEvent:
Packit 1fb8d4
	CloseHandle(transport->connectedEvent);
Packit 1fb8d4
out_free_receivebuffer:
Packit 1fb8d4
	StreamPool_Return(transport->ReceivePool, transport->ReceiveBuffer);
Packit 1fb8d4
out_free_receivepool:
Packit 1fb8d4
	StreamPool_Free(transport->ReceivePool);
Packit 1fb8d4
out_free_transport:
Packit 1fb8d4
	free(transport);
Packit 1fb8d4
	return NULL;
Packit 1fb8d4
}
Packit 1fb8d4
Packit 1fb8d4
void transport_free(rdpTransport* transport)
Packit 1fb8d4
{
Packit 1fb8d4
	if (!transport)
Packit 1fb8d4
		return;
Packit 1fb8d4
Packit 1fb8d4
	transport_disconnect(transport);
Packit 1fb8d4
Packit 1fb8d4
	if (transport->ReceiveBuffer)
Packit 1fb8d4
		Stream_Release(transport->ReceiveBuffer);
Packit 1fb8d4
Packit Service 5a9772
	nla_free(transport->nla);
Packit 1fb8d4
	StreamPool_Free(transport->ReceivePool);
Packit 1fb8d4
	CloseHandle(transport->connectedEvent);
Packit 1fb8d4
	CloseHandle(transport->rereadEvent);
Packit 1fb8d4
	DeleteCriticalSection(&(transport->ReadLock));
Packit 1fb8d4
	DeleteCriticalSection(&(transport->WriteLock));
Packit 1fb8d4
	free(transport);
Packit 1fb8d4
}