From 6c295504112992500f872c72dc053b29415fafcb Mon Sep 17 00:00:00 2001
From: qhuang8 <qhuang8@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Tue, 4 Mar 2008 11:36:52 +0000
Subject: [PATCH] Add NULL parameter checking for notify function.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@4791 6f19259b-4bc3-4df7-8a09-765794883524
---
 Nt32Pkg/WinNtGopDxe/WinNtGopInput.c | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/Nt32Pkg/WinNtGopDxe/WinNtGopInput.c b/Nt32Pkg/WinNtGopDxe/WinNtGopInput.c
index 0ea802355d..f78247c23f 100644
--- a/Nt32Pkg/WinNtGopDxe/WinNtGopInput.c
+++ b/Nt32Pkg/WinNtGopDxe/WinNtGopInput.c
@@ -326,7 +326,7 @@ Returns:
   Private->CapsLock                = FALSE;
   Private->NumLock                 = FALSE;
   Private->ScrollLock              = FALSE;
-  
+ 
   Private->KeyState.KeyShiftState  = EFI_SHIFT_STATE_VALID;
   Private->KeyState.KeyToggleState = EFI_TOGGLE_STATE_VALID;
 
@@ -734,9 +734,9 @@ WinNtGopSimpleTextInExRegisterKeyNotify (
   LIST_ENTRY                         *Link;
   WIN_NT_GOP_SIMPLE_TEXTIN_EX_NOTIFY *NewNotify;      
 
-  if (KeyData == NULL || NotifyHandle == NULL) {
+  if (KeyData == NULL || KeyNotificationFunction == NULL || NotifyHandle == NULL) {
     return EFI_INVALID_PARAMETER;
-  }  
+  }
 
   Private = GOP_PRIVATE_DATA_FROM_TEXT_IN_EX_THIS (This);