From 7a49cd087506d3c9d3b290e749b3191988880bbb Mon Sep 17 00:00:00 2001
From: ydong10 <ydong10@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Tue, 10 Apr 2012 02:27:05 +0000
Subject: Update code to avoid potential access violation.

Signed-off-by: Dong Eric <eric.dong@intel.com>
Reviewed-by: Ye Ting <ting.ye@intel.com>
Reviewed-by: Ouyang Qian <Ouyang.qian@intel.com>

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@13175 6f19259b-4bc3-4df7-8a09-765794883524
---
 NetworkPkg/Application/IpsecConfig/Helper.c               | 3 ++-
 NetworkPkg/Application/IpsecConfig/PolicyEntryOperation.c | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

(limited to 'NetworkPkg/Application')

diff --git a/NetworkPkg/Application/IpsecConfig/Helper.c b/NetworkPkg/Application/IpsecConfig/Helper.c
index 5013ad9b76..ae867ceee7 100644
--- a/NetworkPkg/Application/IpsecConfig/Helper.c
+++ b/NetworkPkg/Application/IpsecConfig/Helper.c
@@ -1,7 +1,7 @@
 /** @file
   The assistant function implementation for IpSecConfig application.
 
-  Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>
+  Copyright (c) 2009 - 2012, Intel Corporation. All rights reserved.<BR>
 
   This program and the accompanying materials
   are licensed and made available under the terms and conditions of the BSD License
@@ -93,6 +93,7 @@ GetNumber (
       //
       // Convert falied, so use String->Integer map.
       //
+      ASSERT (Map != NULL);
       Value64 = MapStringToInteger (ValueStr, Map);
       if (Value64 == (UINT32) -1) {
         //
diff --git a/NetworkPkg/Application/IpsecConfig/PolicyEntryOperation.c b/NetworkPkg/Application/IpsecConfig/PolicyEntryOperation.c
index 67209f6247..05df9fa919 100644
--- a/NetworkPkg/Application/IpsecConfig/PolicyEntryOperation.c
+++ b/NetworkPkg/Application/IpsecConfig/PolicyEntryOperation.c
@@ -1,7 +1,7 @@
 /** @file
   The implementation of policy entry operation function in IpSecConfig application.
 
-  Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>
+  Copyright (c) 2009 - 2012, Intel Corporation. All rights reserved.<BR>
 
   This program and the accompanying materials
   are licensed and made available under the terms and conditions of the BSD License
@@ -784,6 +784,7 @@ CreateSadEntry (
   if (ValueStr != NULL ) {
     (*Data)->AlgoInfo.EspAlgoInfo.EncKeyLength = EncKeyLength;
     AsciiStr = AllocateZeroPool (EncKeyLength + 1);
+    ASSERT (AsciiStr != NULL);
     UnicodeStrToAsciiStr (ValueStr, AsciiStr);
     CopyMem ((*Data)->AlgoInfo.EspAlgoInfo.EncKey,  AsciiStr, EncKeyLength);
     FreePool (AsciiStr);
-- 
cgit v1.2.3