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

View File

@ -221,7 +221,6 @@ int transport_write(rdpTransport* transport, STREAM* s)
{
int status = -1;
int length;
int sent = 0;
length = stream_get_length(s);
stream_set_pos(s, 0);
@ -234,7 +233,7 @@ int transport_write(rdpTransport* transport, STREAM* s)
}
#endif
while (sent < length)
while (length > 0)
{
if (transport->layer == TRANSPORT_LAYER_TLS)
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);
}