NetworkPkg/TlsDxe: Fix failure to process multiple TLS records.

[mirror_edk2.git] / NetworkPkg / TlsDxe / TlsImpl.c

diff --git a/NetworkPkg/TlsDxe/TlsImpl.c b/NetworkPkg/TlsDxe/TlsImpl.c
index ea83dbd04f539820d2a8a6e193a02c5dab654cf1..2d4169b0a5d0763d232fd2fadd43bade53be3868 100644 (file)
--- a/NetworkPkg/TlsDxe/TlsImpl.c
+++ b/NetworkPkg/TlsDxe/TlsImpl.c
@@ -144,7 +144,7 @@ TlsEncryptPacket (
     BufferOutSize += ThisMessageSize;\r
 \r
     BufferInPtr += TLS_RECORD_HEADER_LENGTH + ThisPlainMessageSize;\r
-    TempRecordHeader += ThisMessageSize;\r
+    TempRecordHeader = (TLS_RECORD_HEADER *)((UINT8 *)TempRecordHeader + ThisMessageSize);\r
   }\r
 \r
   FreePool (BufferIn);\r
@@ -317,7 +317,7 @@ TlsDecryptPacket (
     BufferOutSize += TLS_RECORD_HEADER_LENGTH + ThisPlainMessageSize;\r
 \r
     BufferInPtr += TLS_RECORD_HEADER_LENGTH + ThisCipherMessageSize;\r
-    TempRecordHeader += TLS_RECORD_HEADER_LENGTH + ThisPlainMessageSize;\r
+    TempRecordHeader = (TLS_RECORD_HEADER *)((UINT8 *)TempRecordHeader + TLS_RECORD_HEADER_LENGTH + ThisPlainMessageSize);\r
   }\r
 \r
   FreePool (BufferIn);\r