Append the terminating null character at the end of the string to avoid buffer overflow.
Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Qiu Shumin <shumin.qiu@intel.com> Reviewed-by: Fu Siyuan <siyuan.fu@intel.com> git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15888 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
@@ -69,6 +69,7 @@ PxeBcMtftp6CheckPacket (
|
||||
(CHAR8 *) Packet->Error.ErrorMessage,
|
||||
PXE_MTFTP_ERROR_STRING_LENGTH
|
||||
);
|
||||
Private->Mode.TftpError.ErrorString[PXE_MTFTP_ERROR_STRING_LENGTH - 1] = '\0';
|
||||
}
|
||||
|
||||
if (Callback != NULL) {
|
||||
@@ -182,6 +183,7 @@ PxeBcMtftp6GetFileSize (
|
||||
(CHAR8 *) Packet->Error.ErrorMessage,
|
||||
PXE_MTFTP_ERROR_STRING_LENGTH
|
||||
);
|
||||
Private->Mode.TftpError.ErrorString[PXE_MTFTP_ERROR_STRING_LENGTH - 1] = '\0';
|
||||
}
|
||||
goto ON_ERROR;
|
||||
}
|
||||
@@ -511,6 +513,7 @@ PxeBcMtftp4CheckPacket (
|
||||
(CHAR8 *) Packet->Error.ErrorMessage,
|
||||
PXE_MTFTP_ERROR_STRING_LENGTH
|
||||
);
|
||||
Private->Mode.TftpError.ErrorString[PXE_MTFTP_ERROR_STRING_LENGTH - 1] = '\0';
|
||||
}
|
||||
|
||||
if (Callback != NULL) {
|
||||
@@ -624,6 +627,7 @@ PxeBcMtftp4GetFileSize (
|
||||
(CHAR8 *) Packet->Error.ErrorMessage,
|
||||
PXE_MTFTP_ERROR_STRING_LENGTH
|
||||
);
|
||||
Private->Mode.TftpError.ErrorString[PXE_MTFTP_ERROR_STRING_LENGTH - 1] = '\0';
|
||||
}
|
||||
goto ON_ERROR;
|
||||
}
|
||||
|
Reference in New Issue
Block a user