From 532691c8ba90e5022174503ab0781322c6f79cab Mon Sep 17 00:00:00 2001
From: jcarsey <jcarsey@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Fri, 14 Oct 2011 19:21:13 +0000
Subject: ShellPkg: Add checking for memory allocation and pointer returns from
 functions.

signed-off-by: jcarsey
reviewed-by: geekboy15a

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12540 6f19259b-4bc3-4df7-8a09-765794883524
---
 ShellPkg/Library/UefiShellLevel3CommandsLib/Touch.c | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'ShellPkg/Library/UefiShellLevel3CommandsLib')

diff --git a/ShellPkg/Library/UefiShellLevel3CommandsLib/Touch.c b/ShellPkg/Library/UefiShellLevel3CommandsLib/Touch.c
index 7eb3c8c0e8..29f6be3931 100644
--- a/ShellPkg/Library/UefiShellLevel3CommandsLib/Touch.c
+++ b/ShellPkg/Library/UefiShellLevel3CommandsLib/Touch.c
@@ -79,11 +79,16 @@ DoTouchByHandle (
 
   if (FS == NULL) {
     FS = StrnCatGrow(&FS, NULL, Name, 0);
-    TempSpot = StrStr(FS, L"\\");
-    if (TempSpot != NULL) {
-      *TempSpot = CHAR_NULL;
+    if (FS != NULL) {
+      TempSpot = StrStr(FS, L"\\");
+      if (TempSpot != NULL) {
+        *TempSpot = CHAR_NULL;
+      }
     }
   }
+  if (FS == NULL) {
+    return (EFI_INVALID_PARAMETER);
+  }
 
   //
   // do it
-- 
cgit v1.2.3