NetworkPkg:Fix Network memory leak when calling GetModeData interface

[mirror_edk2.git] / NetworkPkg / Dhcp6Dxe / ComponentName.c

diff --git a/NetworkPkg/Dhcp6Dxe/ComponentName.c b/NetworkPkg/Dhcp6Dxe/ComponentName.c
index 927a7fe7ecc7468fc7c50c39a33ca0dd8c3aa308..ba612156a09f1001807eff58127731d258799848 100644 (file)
--- a/NetworkPkg/Dhcp6Dxe/ComponentName.c
+++ b/NetworkPkg/Dhcp6Dxe/ComponentName.c
@@ -1,7 +1,7 @@
 /** @file\r
   UEFI Component Name(2) protocol implementation for Dhcp6 driver.\r
 \r
-  Copyright (c) 2009 - 2014, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2009 - 2016, Intel Corporation. All rights reserved.<BR>\r
 \r
   This program and the accompanying materials\r
   are licensed and made available under the terms and conditions of the BSD License\r
@@ -286,7 +286,14 @@ UpdateName (
     }\r
     HandleName = mDhcp6ControllerName[Dhcp6ModeData.Ia->State];\r
   }\r
-  \r
+\r
+  if (Dhcp6ModeData.Ia != NULL) {\r
+    FreePool (Dhcp6ModeData.Ia);\r
+  }\r
+  if (Dhcp6ModeData.ClientId != NULL) {\r
+    FreePool (Dhcp6ModeData.ClientId);\r
+  }\r
+\r
   Status = AddUnicodeString2 (\r
              "eng",\r
              gDhcp6ComponentName.SupportedLanguages,\r