Use string pointer instead string buffer to avoid string copy operation.

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

diff --git a/NetworkPkg/Dhcp6Dxe/ComponentName.c b/NetworkPkg/Dhcp6Dxe/ComponentName.c
index c7fd6075df92121eabe7481efdf37c83e8df74a6..927a7fe7ecc7468fc7c50c39a33ca0dd8c3aa308 100644 (file)
--- a/NetworkPkg/Dhcp6Dxe/ComponentName.c
+++ b/NetworkPkg/Dhcp6Dxe/ComponentName.c
@@ -259,7 +259,7 @@ UpdateName (
 {\r
   EFI_STATUS                       Status;\r
   EFI_DHCP6_MODE_DATA              Dhcp6ModeData;\r
-  CHAR16                           HandleName[64];\r
+  CHAR16                           *HandleName;\r
 \r
   if (Dhcp6 == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -279,9 +279,12 @@ UpdateName (
   }\r
   \r
   if (Dhcp6ModeData.Ia == NULL) {\r
-    UnicodeSPrint (HandleName, sizeof (HandleName), L"DHCPv6 (No configured IA)");\r
+    HandleName = L"DHCPv6 (No configured IA)";\r
   } else {\r
-    StrnCpy (HandleName, mDhcp6ControllerName[Dhcp6ModeData.Ia->State], sizeof (HandleName) / sizeof (CHAR16) - 1);\r
+    if (Dhcp6ModeData.Ia->State > Dhcp6Rebinding) {\r
+      return EFI_DEVICE_ERROR;\r
+    }\r
+    HandleName = mDhcp6ControllerName[Dhcp6ModeData.Ia->State];\r
   }\r
   \r
   Status = AddUnicodeString2 (\r