Merge branch 'master' of github.com:awakecoding/FreeRDP into ports

libfreerdp/core/freerdp.c

@@ -192,11 +192,6 @@ BOOL freerdp_connect(freerdp* instance)
 		freerdp_set_last_error(instance->context, FREERDP_ERROR_INSUFFICIENT_PRIVILEGES);
 	}
 
-	if (!connectErrorCode)
-	{
-		connectErrorCode = UNDEFINEDCONNECTERROR;
-	}
-
 	SetEvent(rdp->transport->connectedEvent);
 
 	freerdp_connect_finally:

libfreerdp/core/transport.c

@@ -317,6 +317,7 @@ BOOL transport_connect_nla(rdpTransport* transport)
 	if (!transport->credssp)
 	{
 		transport->credssp = credssp_new(instance, transport, settings);
+		transport_set_nla_mode(transport, TRUE);
 
 		if (settings->AuthenticationServiceClass)
 		{
@@ -338,11 +339,14 @@ BOOL transport_connect_nla(rdpTransport* transport)
 		fprintf(stderr, "Authentication failure, check credentials.\n"
 			"If credentials are valid, the NTLMSSP implementation may be to blame.\n");
 
+		transport_set_nla_mode(transport, FALSE);
 		credssp_free(transport->credssp);
 		transport->credssp = NULL;
+
 		return FALSE;
 	}
 
+	transport_set_nla_mode(transport, FALSE);
 	credssp_free(transport->credssp);
 	transport->credssp = NULL;
 
@@ -481,11 +485,16 @@ BOOL transport_accept_nla(rdpTransport* transport)
 		return TRUE;
 
 	if (!transport->credssp)
+	{
 		transport->credssp = credssp_new(instance, transport, settings);
+		transport_set_nla_mode(transport, TRUE);
+	}
 
 	if (credssp_authenticate(transport->credssp) < 0)
 	{
 		fprintf(stderr, "client authentication failure\n");
+
+		transport_set_nla_mode(transport, FALSE);
 		credssp_free(transport->credssp);
 		transport->credssp = NULL;
 
@@ -495,6 +504,7 @@ BOOL transport_accept_nla(rdpTransport* transport)
 	}
 
 	/* don't free credssp module yet, we need to copy the credentials from it first */
+	transport_set_nla_mode(transport, FALSE);
 
 	return TRUE;
 }
@@ -643,14 +653,11 @@ int transport_read(rdpTransport* transport, wStream* s)
 
 	CopyMemory(header, Stream_Buffer(s), 4); /* peek at first 4 bytes */
 
-	/* if header is present, read in exactly one PDU */
+	/* if header is present, read exactly one PDU */
-	if (header[0] == 0x03)
-	{
-		/* TPKT header */
 
-		pduLength = (header[2] << 8) | header[3];
+	if (transport->NlaMode)
-	}
+	{
-	else if (header[0] == 0x30)
+		if (header[0] == 0x30)
 		{
 			/* TSRequest (NLA) */
 
@@ -678,6 +685,15 @@ int transport_read(rdpTransport* transport, wStream* s)
 				pduLength += 2;
 			}
 		}
+	}
+	else
+	{
+		if (header[0] == 0x03)
+		{
+			/* TPKT header */
+
+			pduLength = (header[2] << 8) | header[3];
+		}
 		else
 		{
 			/* Fast-Path Header */
@@ -687,6 +703,7 @@ int transport_read(rdpTransport* transport, wStream* s)
 			else
 				pduLength = header[1];
 		}
+	}
 
 	status = transport_read_layer(transport, Stream_Buffer(s) + position, pduLength - position);
 
@@ -889,32 +906,23 @@ int transport_check_fds(rdpTransport* transport)
 	 * Loop through and read all available PDUs.  Since multiple
 	 * PDUs can exist, it's important to deliver them all before
 	 * returning.  Otherwise we run the risk of having a thread
-	 * wait for a socket to get signalled that data is available
+	 * wait for a socket to get signaled that data is available
 	 * (which may never happen).
 	 */
 	for (;;)
 	{
 		status = transport_read_nonblocking(transport);
 
-		if (status < 0 || Stream_GetPosition(transport->ReceiveBuffer) == 0)
+		if ((status <= 0) || (Stream_GetPosition(transport->ReceiveBuffer) < 2))
 			return status;
 
 		while ((pos = Stream_GetPosition(transport->ReceiveBuffer)) > 0)
 		{
 			Stream_SetPosition(transport->ReceiveBuffer, 0);
 
-			if (tpkt_verify_header(transport->ReceiveBuffer)) /* TPKT */
+			if (transport->NlaMode)
 			{
-				/* Ensure the TPKT header is available. */
+				if (nla_verify_header(transport->ReceiveBuffer))
-				if (pos <= 4)
-				{
-					Stream_SetPosition(transport->ReceiveBuffer, pos);
-					return 0;
-				}
-
-				length = tpkt_read_header(transport->ReceiveBuffer);
-			}
-			else if (nla_verify_header(transport->ReceiveBuffer))
 				{
 					/* TSRequest */
 
@@ -936,6 +944,20 @@ int transport_check_fds(rdpTransport* transport)
 
 					length = nla_read_header(transport->ReceiveBuffer);
 				}
+			}
+			else
+			{
+				if (tpkt_verify_header(transport->ReceiveBuffer)) /* TPKT */
+				{
+					/* Ensure the TPKT header is available. */
+					if (pos <= 4)
+					{
+						Stream_SetPosition(transport->ReceiveBuffer, pos);
+						return 0;
+					}
+
+					length = tpkt_read_header(transport->ReceiveBuffer);
+				}
 				else /* Fast Path */
 				{
 					/* Ensure the Fast Path header is available. */
@@ -956,6 +978,7 @@ int transport_check_fds(rdpTransport* transport)
 
 					length = fastpath_read_header(NULL, transport->ReceiveBuffer);
 				}
+			}
 
 			if (length == 0)
 			{
@@ -1039,6 +1062,11 @@ void transport_set_gateway_enabled(rdpTransport* transport, BOOL GatewayEnabled)
 	transport->GatewayEnabled = GatewayEnabled;
 }
 
+void transport_set_nla_mode(rdpTransport* transport, BOOL NlaMode)
+{
+	transport->NlaMode = NlaMode;
+}
+
 static void* transport_client_thread(void* arg)
 {
 	DWORD status;

libfreerdp/core/transport.h

@@ -75,6 +75,7 @@ struct rdp_transport
 	HANDLE stopEvent;
 	HANDLE thread;
 	BOOL async;
+	BOOL NlaMode;
 	BOOL GatewayEnabled;
 	CRITICAL_SECTION ReadLock;
 	CRITICAL_SECTION WriteLock;
@@ -99,6 +100,7 @@ void transport_get_fds(rdpTransport* transport, void** rfds, int* rcount);
 int transport_check_fds(rdpTransport* transport);
 BOOL transport_set_blocking_mode(rdpTransport* transport, BOOL blocking);
 void transport_set_gateway_enabled(rdpTransport* transport, BOOL GatewayEnabled);
+void transport_set_nla_mode(rdpTransport* transport, BOOL NlaMode);
 void transport_get_read_handles(rdpTransport* transport, HANDLE* events, DWORD* count);
 
 wStream* transport_receive_pool_take(rdpTransport* transport);