From 03bc7c2b09c6f36930a5138c7727c0c4048539a7 Mon Sep 17 00:00:00 2001 From: Shumin Qiu Date: Thu, 9 Jan 2014 00:30:27 +0000 Subject: Add code to check whether the pointer 'NewBuffer' in 'FileHandelWrappers.c' and the pointer 'FoundVarName' in 'Dmpstore.c' are NULL before used. Signed-off-by: Shumin Qiu Reviewed-by: Jaben Carsey git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15060 6f19259b-4bc3-4df7-8a09-765794883524 --- ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c') diff --git a/ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c b/ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c index afeedb0959..b02048302e 100644 --- a/ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c +++ b/ShellPkg/Library/UefiShellDebug1CommandsLib/DmpStore.c @@ -121,11 +121,15 @@ CascadeProcessVariables ( if (Status == EFI_BUFFER_TOO_SMALL) { SHELL_FREE_NON_NULL(FoundVarName); FoundVarName = AllocateZeroPool (NameSize); - if (PrevName != NULL) { - StrCpy(FoundVarName, PrevName); - } + if (FoundVarName != NULL) { + if (PrevName != NULL) { + StrCpy(FoundVarName, PrevName); + } - Status = gRT->GetNextVariableName (&NameSize, FoundVarName, &FoundVarGuid); + Status = gRT->GetNextVariableName (&NameSize, FoundVarName, &FoundVarGuid); + } else { + Status = EFI_OUT_OF_RESOURCES; + } } // -- cgit v1.2.3