mirrors
/
FreeRDP
mirror of https://github.com/FreeRDP/FreeRDP
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Handle short writes properly in transport_write(). 

Before this change if short write (send) occured, we were trying to
resend without updating length variable, so we were trying to send
too much and the following undefined data were causing the other
side to terminate connection.
This commit is contained in:
Pawel Jakub Dawidek 2012-01-24 11:42:30 +01:00
parent 89253f027c
commit 609d177412
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libfreerdp-core/transport.c
View File

@ -221,7 +221,6 @@ int transport_write(rdpTransport* transport, STREAM* s)
{ {
int status = -1; int status = -1;
int length; int length;
int sent = 0;
length = stream_get_length(s); length = stream_get_length(s);
stream_set_pos(s, 0); stream_set_pos(s, 0);
@ -234,7 +233,7 @@ int transport_write(rdpTransport* transport, STREAM* s)
} }
#endif #endif
while (sent < length) while (length > 0)
{ {
if (transport->layer == TRANSPORT_LAYER_TLS) if (transport->layer == TRANSPORT_LAYER_TLS)
status = tls_write(transport->tls, stream_get_tail(s), length); status = tls_write(transport->tls, stream_get_tail(s), length);
@ -258,7 +257,7 @@ int transport_write(rdpTransport* transport, STREAM* s)
} }
} }
sent += status; length -= status;
stream_seek(s, status); stream_seek(s, status);
} }