From 8ec8bed4394046cf1909b39cc42ef8829ddb28a3 Mon Sep 17 00:00:00 2001
From: klu2 <klu2@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Thu, 18 Mar 2010 06:24:53 +0000
Subject: 1, For Pcd_PPI, UnregistedCallBack should return
 EFI_INVALID_PARAMETER if Callback function can not be found. 2,
 PCD_PPI/PROTOCOL should return EFI_INVALID_PARAMETER if size is unmatched

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10276 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Universal/PCD/Dxe/Service.c | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'MdeModulePkg/Universal/PCD/Dxe')

diff --git a/MdeModulePkg/Universal/PCD/Dxe/Service.c b/MdeModulePkg/Universal/PCD/Dxe/Service.c
index fc6f6b4414..5d4fbbc082 100644
--- a/MdeModulePkg/Universal/PCD/Dxe/Service.c
+++ b/MdeModulePkg/Universal/PCD/Dxe/Service.c
@@ -740,8 +740,14 @@ SetWorker (
   //
   ASSERT (TokenNumber + 1 < PCD_TOTAL_TOKEN_NUMBER + 1);
 
-  if ((!PtrType) && (*Size != DxePcdGetSize (TokenNumber + 1))) {
-    return EFI_INVALID_PARAMETER;
+  if (PtrType) {
+    if (*Size > DxePcdGetSize (TokenNumber + 1)) {
+      return EFI_INVALID_PARAMETER;
+    }
+  } else {
+    if (*Size != DxePcdGetSize (TokenNumber + 1)) {
+      return EFI_INVALID_PARAMETER;
+    }
   }
 
   //
-- 
cgit v1.2.3