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

1 files 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);