Update PcdSetPtr() API to set the output buffer size to the max buffer size when...

Signed-off-by: lgao4
Reviewed-by: jlin16
git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12875 6f19259b-4bc3-4df7-8a09-765794883524

diff --git a/MdeModulePkg/Universal/PCD/Dxe/Service.c b/MdeModulePkg/Universal/PCD/Dxe/Service.c
index bf8e74dfb328e79be067ad29c7b022f13f893c62..8bfd199e4826bb25ac77a95f1fce599821e0e73b 100644 (file)
--- a/MdeModulePkg/Universal/PCD/Dxe/Service.c
+++ b/MdeModulePkg/Universal/PCD/Dxe/Service.c
@@ -746,6 +746,7 @@ SetWorker (
     //\r
     GetPtrTypeSize (TokenNumber, &MaxSize);\r
     if (*Size > MaxSize) {\r
+      *Size = MaxSize;\r
       return EFI_INVALID_PARAMETER;\r
     }\r
   } else {\r

diff --git a/MdeModulePkg/Universal/PCD/Pei/Service.c b/MdeModulePkg/Universal/PCD/Pei/Service.c
index 4e7366e7596f87d790eff2510fc1e2e8bd5c4446..4769600da762d9f7811b99efd45ce0ff0a7fa437 100644 (file)
--- a/MdeModulePkg/Universal/PCD/Pei/Service.c
+++ b/MdeModulePkg/Universal/PCD/Pei/Service.c
@@ -391,6 +391,7 @@ SetWorker (
     //\r
     GetPtrTypeSize (TokenNumber, &MaxSize, PeiPcdDb);\r
     if (*Size > MaxSize) {\r
+      *Size = MaxSize;\r
       return EFI_INVALID_PARAMETER;\r
     }\r
   } else {\r