morgan/FreeRDP
1
0
Fork 0
mirror of https://github.com/morgan9e/FreeRDP synced 2026-04-15 08:54:38 +09:00
Code Issues Packages Projects Releases Wiki Activity

[core,rdstls] ensure stream length on empy return

Browse Source
This commit is contained in:
akallabeth
2023-03-08 08:29:34 +01:00
committed by akallabeth
parent 7c24da917e
commit f5a8da4f62
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
libfreerdp/core/rdstls.c
View File

@@ -138,6 +138,9 @@ static SSIZE_T rdstls_write_string(wStream* s, const char* str)
{
/* Write unicode null */
Stream_Write_UINT16(s, 2);
if (!Stream_EnsureRemainingCapacity(s, 2))
return -1;
Stream_Write_UINT16(s, 0);
return (SSIZE_T)(Stream_GetPosition(s) - pos);
}
@@ -162,8 +165,12 @@ static BOOL rdstls_write_data(wStream* s, UINT32 length, const BYTE* data)
if (!data)
{
return -1;
/* Write unicode null */
Stream_Write_UINT16(s, 2);
if (!Stream_EnsureRemainingCapacity(s, 2))
return -1;
Stream_Write_UINT16(s, 0);
return TRUE;
}