diff options
author | Eric Dong <eric.dong@intel.com> | 2014-06-26 01:38:46 +0000 |
---|---|---|
committer | ydong10 <ydong10@6f19259b-4bc3-4df7-8a09-765794883524> | 2014-06-26 01:38:46 +0000 |
commit | 81cd2f536ef0097392b51f96405321ce7b01350c (patch) | |
tree | ce44817456c749b0663e8fb845c9d91b8aefbce3 /ShellPkg/Application/Shell | |
parent | 45295cf5a58578b84e5092c3a3dd145c069c1db5 (diff) | |
download | edk2-platforms-81cd2f536ef0097392b51f96405321ce7b01350c.tar.xz |
Refine code to make it more safely.
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Eric Dong <eric.dong@intel.com>
Reviewed-by: Jaben Carsey <Jaben.carsey@intel.com>
git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15593 6f19259b-4bc3-4df7-8a09-765794883524
Diffstat (limited to 'ShellPkg/Application/Shell')
-rw-r--r-- | ShellPkg/Application/Shell/Shell.c | 2 | ||||
-rw-r--r-- | ShellPkg/Application/Shell/ShellProtocol.c | 1 |
2 files changed, 3 insertions, 0 deletions
diff --git a/ShellPkg/Application/Shell/Shell.c b/ShellPkg/Application/Shell/Shell.c index 4abeafcd44..2f1e3eb65c 100644 --- a/ShellPkg/Application/Shell/Shell.c +++ b/ShellPkg/Application/Shell/Shell.c @@ -509,6 +509,7 @@ UefiMain ( // Reset page break back to default.
//
ShellInfoObject.PageBreakEnabled = PcdGetBool(PcdShellPageBreakDefault);
+ ASSERT (ShellInfoObject.ConsoleInfo != NULL);
ShellInfoObject.ConsoleInfo->Enabled = TRUE;
ShellInfoObject.ConsoleInfo->RowCounter = 0;
@@ -2091,6 +2092,7 @@ ProcessCommandLineToFinal( if (EFI_ERROR(Status)) {
return (Status);
}
+ ASSERT (*CmdLine != NULL);
TrimSpaces(CmdLine);
diff --git a/ShellPkg/Application/Shell/ShellProtocol.c b/ShellPkg/Application/Shell/ShellProtocol.c index e4680139d0..3d01283b60 100644 --- a/ShellPkg/Application/Shell/ShellProtocol.c +++ b/ShellPkg/Application/Shell/ShellProtocol.c @@ -460,6 +460,7 @@ EfiShellGetFilePathFromDevicePath( ASSERT((PathForReturn == NULL && PathSize == 0) || (PathForReturn != NULL));
AlignedNode = AllocateCopyPool (DevicePathNodeLength(FilePath), FilePath);
+ ASSERT (AlignedNode != NULL);
// File Path Device Path Nodes 'can optionally add a "\" separator to
// the beginning and/or the end of the Path Name string.'
|