MdePkg: Refine UefiFileHandleLib to avoid write non-ASCII char into ASCII file.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Qiu Shumin <shumin.qiu@intel.com>
Reviewed-by: Liming Gao <liming.gao@intel.com>
Reviewed-by: Jaben Carsey <jaben.carsey@intel.com>

git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@18430 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 7 insertions, 0 deletions

diff --git a/MdePkg/Library/UefiFileHandleLib/UefiFileHandleLib.c b/MdePkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
index 04a2f189ec..dfec5fa4d4 100644
--- a/MdePkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
+++ b/MdePkg/Library/UefiFileHandleLib/UefiFileHandleLib.c
@@ -1079,6 +1079,7 @@ FileHandleWriteLine(
   EFI_STATUS  Status;

   CHAR16      CharBuffer;

   UINTN       Size;

+  UINTN       Index;

   UINTN       CharSize;

   UINT64      FileSize;

   UINT64      OriginalFilePosition;

@@ -1136,6 +1137,12 @@ FileHandleWriteLine(
       return EFI_OUT_OF_RESOURCES;

     }

     UnicodeStrToAsciiStr (Buffer, AsciiBuffer);

+    for (Index = 0; Index < Size; Index++) {

+      if (!((AsciiBuffer[Index] >= 0) && (AsciiBuffer[Index] < 128))){

+        FreePool(AsciiBuffer);

+        return EFI_INVALID_PARAMETER;

+      }

+    }

     

     Size = AsciiStrSize(AsciiBuffer) - sizeof(CHAR8);

     Status = FileHandleWrite(Handle, &Size, AsciiBuffer);