From 52c0d06b94665def4977e13ea329dccb17f46da5 Mon Sep 17 00:00:00 2001
From: hhuan13 <hhuan13@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Tue, 2 Mar 2010 02:58:03 +0000
Subject: Fix K8 scan issues

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10151 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Core/PiSmmCore/Locate.c | 1 +
 MdeModulePkg/Core/PiSmmCore/Pool.c   | 1 +
 2 files changed, 2 insertions(+)

(limited to 'MdeModulePkg/Core/PiSmmCore')

diff --git a/MdeModulePkg/Core/PiSmmCore/Locate.c b/MdeModulePkg/Core/PiSmmCore/Locate.c
index ca95ee4e17..00b2c892e7 100644
--- a/MdeModulePkg/Core/PiSmmCore/Locate.c
+++ b/MdeModulePkg/Core/PiSmmCore/Locate.c
@@ -398,6 +398,7 @@ SmmLocateHandle (
     *BufferSize = ResultSize;
 
     if (SearchType == ByRegisterNotify && !EFI_ERROR(Status)) {
+      ASSERT (SearchKey != NULL);
       //
       // If this is a search by register notify and a handle was
       // returned, update the register notification position
diff --git a/MdeModulePkg/Core/PiSmmCore/Pool.c b/MdeModulePkg/Core/PiSmmCore/Pool.c
index 443faab34f..d7f80f4f8c 100644
--- a/MdeModulePkg/Core/PiSmmCore/Pool.c
+++ b/MdeModulePkg/Core/PiSmmCore/Pool.c
@@ -97,6 +97,7 @@ InternalAllocPoolByIndex (
   EFI_STATUS        Status;
   FREE_POOL_HEADER  *Hdr;
 
+  ASSERT (PoolIndex <= MAX_POOL_INDEX);
   Status = EFI_SUCCESS;
   if (PoolIndex == MAX_POOL_INDEX) {
     Hdr = (FREE_POOL_HEADER *)AllocatePages (EFI_SIZE_TO_PAGES (MAX_POOL_SIZE << 1));
-- 
cgit v1.2.3