mirrors/FreeRDP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use static wStream for packet logger

Browse Source
This commit is contained in:
Armin Novak 2022-10-10 14:46:46 +02:00 committed by akallabeth
parent 43ea3c0132
commit d879d87610
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
winpr/libwinpr/utils/wlog/PacketMessage.c
View File

@ -266,21 +266,22 @@ void Pcap_Close(wPcap* pcap)
static BOOL WLog_PacketMessage_Write_EthernetHeader(wPcap* pcap, wEthernetHeader* ethernet)
{
wStream* s;
BYTE buffer[14];
wStream sbuffer = { 0 };
BYTE buffer[14] = { 0 };
BOOL ret = TRUE;
if (!pcap || !pcap->fp || !ethernet)
return FALSE;
s = Stream_New(buffer, 14);
s = Stream_StaticInit(&sbuffer, buffer, sizeof(buffer));
if (!s)
return FALSE;
Stream_Write(s, ethernet->Destination, 6);
Stream_Write(s, ethernet->Source, 6);
Stream_Write_UINT16_BE(s, ethernet->Type);
if (fwrite(buffer, 14, 1, pcap->fp) != 1)
if (fwrite(buffer, sizeof(buffer), 1, pcap->fp) != 1)
ret = FALSE;
Stream_Free(s, FALSE);
return ret;
}
@ -309,13 +310,14 @@ static UINT16 IPv4Checksum(BYTE* ipv4, int length)
static BOOL WLog_PacketMessage_Write_IPv4Header(wPcap* pcap, wIPv4Header* ipv4)
{
wStream* s;
BYTE buffer[20];
wStream sbuffer = { 0 };
BYTE buffer[20] = { 0 };
int ret = TRUE;
if (!pcap || !pcap->fp || !ipv4)
return FALSE;
s = Stream_New(buffer, 20);
s = Stream_StaticInit(&sbuffer, buffer, sizeof(buffer));
if (!s)
return FALSE;
Stream_Write_UINT8(s, (ipv4->Version << 4) | ipv4->InternetHeaderLength);
@ -331,23 +333,24 @@ static BOOL WLog_PacketMessage_Write_IPv4Header(wPcap* pcap, wIPv4Header* ipv4)
ipv4->HeaderChecksum = IPv4Checksum((BYTE*)buffer, 20);
Stream_Rewind(s, 10);
Stream_Write_UINT16(s, ipv4->HeaderChecksum);
Stream_Seek(s, 8);
if (fwrite(buffer, 20, 1, pcap->fp) != 1)
if (fwrite(buffer, sizeof(buffer), 1, pcap->fp) != 1)
ret = FALSE;
Stream_Free(s, FALSE);
return ret;
}
static BOOL WLog_PacketMessage_Write_TcpHeader(wPcap* pcap, wTcpHeader* tcp)
{
wStream* s;
BYTE buffer[20];
wStream sbuffer = { 0 };
BYTE buffer[20] = { 0 };
BOOL ret = TRUE;
if (!pcap || !pcap->fp || !tcp)
return FALSE;
s = Stream_New(buffer, 20);
s = Stream_StaticInit(&sbuffer, buffer, sizeof(buffer));
if (!s)
return FALSE;
Stream_Write_UINT16_BE(s, tcp->SourcePort);
@ -362,11 +365,10 @@ static BOOL WLog_PacketMessage_Write_TcpHeader(wPcap* pcap, wTcpHeader* tcp)
if (pcap->fp)
{
if (fwrite(buffer, 20, 1, pcap->fp) != 1)
if (fwrite(buffer, sizeof(buffer), 1, pcap->fp) != 1)
ret = FALSE;
}
Stream_Free(s, FALSE);
return ret;
}