1. Add EFI_COMPONENT_NAME2_PROTOCOL.GetControllerName() support.

[mirror_edk2.git] / MdeModulePkg / Universal / Network / MnpDxe / ComponentName.c

diff --git a/MdeModulePkg/Universal/Network/MnpDxe/ComponentName.c b/MdeModulePkg/Universal/Network/MnpDxe/ComponentName.c
index c56ad061e137472d4180acc8c35342acf6dd71fb..9e66dc254b4d041adcf94bda21a68168958cd25e 100644 (file)
--- a/MdeModulePkg/Universal/Network/MnpDxe/ComponentName.c
+++ b/MdeModulePkg/Universal/Network/MnpDxe/ComponentName.c
@@ -1,7 +1,7 @@
 /** @file\r
   UEFI Component Name(2) protocol implementation for MnpDxe driver.\r
 \r
-Copyright (c) 2005 - 2011, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2005 - 2012, Intel Corporation. All rights reserved.<BR>\r
 This program and the accompanying materials\r
 are licensed and made available under the terms and conditions\r
 of the BSD License which accompanies this distribution.  The full\r
@@ -13,7 +13,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 **/\r
 \r
-#include  "MnpDriver.h"\r
+#include  "MnpImpl.h"\r
 \r
 //\r
 // EFI Component Name Protocol\r
@@ -44,6 +44,8 @@ GLOBAL_REMOVE_IF_UNREFERENCED EFI_UNICODE_STRING_TABLE      mMnpDriverNameTable[
   }\r
 };\r
 \r
+GLOBAL_REMOVE_IF_UNREFERENCED EFI_UNICODE_STRING_TABLE    *gMnpControllerNameTable = NULL;\r
+\r
 /**\r
   Retrieves a Unicode string that is the user readable name of the driver.\r
 \r
@@ -100,6 +102,106 @@ MnpComponentNameGetDriverName (
            );\r
 }\r
 \r
+/**\r
+  Update the component name for the MNP child handle.\r
+\r
+  @param  Mnp[in]                 A pointer to the EFI_MANAGED_NETWORK_PROTOCOL.\r
+\r
+  \r
+  @retval EFI_SUCCESS             Update the ControllerNameTable of this instance successfully.\r
+  @retval EFI_INVALID_PARAMETER   The input parameter is invalid.\r
+  \r
+**/\r
+EFI_STATUS\r
+UpdateName (\r
+  IN   EFI_MANAGED_NETWORK_PROTOCOL     *Mnp\r
+  )\r
+{\r
+  EFI_STATUS                       Status;\r
+  MNP_INSTANCE_DATA                *Instance;\r
+  CHAR16                           HandleName[80];\r
+  EFI_MANAGED_NETWORK_CONFIG_DATA  MnpConfigData;\r
+  EFI_SIMPLE_NETWORK_MODE          SnpModeData;\r
+  UINTN                            OffSet;\r
+  UINTN                            Index;\r
+\r
+  if (Mnp == NULL) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  Instance = MNP_INSTANCE_DATA_FROM_THIS (Mnp);\r
+  //\r
+  // Format the child name into the string buffer as:\r
+  // MNP (MAC=FF-FF-FF-FF-FF-FF, ProtocolType=0x0800, VlanId=0)\r
+  //\r
+  Status = Mnp->GetModeData (Mnp, &MnpConfigData, &SnpModeData);\r
+  if (!EFI_ERROR (Status)) {\r
+    OffSet = 0;\r
+    //\r
+    // Print the MAC address.\r
+    //\r
+    OffSet += UnicodeSPrint (\r
+                HandleName,\r
+                sizeof (HandleName),\r
+                L"MNP (MAC="\r
+                );\r
+    for (Index = 0; Index < SnpModeData.HwAddressSize; Index++) {\r
+      OffSet += UnicodeSPrint (\r
+                  HandleName + OffSet,\r
+                  sizeof (HandleName) - OffSet,\r
+                  L"%02X-",\r
+                  SnpModeData.CurrentAddress.Addr[Index]\r
+                  );\r
+    }\r
+    //\r
+    // Remove the last '-'\r
+    //\r
+    OffSet--;  \r
+    //\r
+    // Print the ProtocolType and VLAN ID for this instance.\r
+    //\r
+    OffSet += UnicodeSPrint (\r
+                HandleName + OffSet,\r
+                sizeof (HandleName) - OffSet,\r
+                L", ProtocolType=0x%X, VlanId=%d)",\r
+                MnpConfigData.ProtocolTypeFilter,\r
+                Instance->MnpServiceData->VlanId\r
+                );\r
+  } else if (Status == EFI_NOT_STARTED) {\r
+    UnicodeSPrint (\r
+      HandleName,\r
+      sizeof (HandleName),\r
+      L"MNP (Not started)"\r
+      );\r
+  } else {\r
+    return Status;\r
+  }\r
+  \r
+  if (gMnpControllerNameTable != NULL) {\r
+    FreeUnicodeStringTable (gMnpControllerNameTable);\r
+    gMnpControllerNameTable = NULL;\r
+  }\r
+  \r
+  Status = AddUnicodeString2 (\r
+             "eng",\r
+             gMnpComponentName.SupportedLanguages,\r
+             &gMnpControllerNameTable,\r
+             HandleName,\r
+             TRUE\r
+             );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+  \r
+  return AddUnicodeString2 (\r
+           "en",\r
+           gMnpComponentName2.SupportedLanguages,\r
+           &gMnpControllerNameTable,\r
+           HandleName,\r
+           FALSE\r
+           );\r
+}\r
+\r
 /**\r
   Retrieves a Unicode string that is the user readable name of the controller\r
   that is being managed by a driver.\r
@@ -178,5 +280,68 @@ MnpComponentNameGetControllerName (
      OUT CHAR16                        **ControllerName\r
   )\r
 {\r
-  return EFI_UNSUPPORTED;\r
+  EFI_STATUS                    Status;\r
+  EFI_MANAGED_NETWORK_PROTOCOL  *Mnp;\r
+\r
+  //\r
+  // Only provide names for MNP child handles.\r
+  //\r
+  if (ChildHandle == NULL) {\r
+    return EFI_UNSUPPORTED;\r
+  }\r
+ \r
+  // \r
+  // Make sure this driver is currently managing ControllerHandle  \r
+  // \r
+  Status = EfiTestManagedDevice (\r
+             ControllerHandle,\r
+             gMnpDriverBinding.DriverBindingHandle,\r
+             &gEfiSimpleNetworkProtocolGuid\r
+             );\r
+  if (EFI_ERROR (Status)) { \r
+    return Status;\r
+  }\r
+  \r
+  // \r
+  // Make sure this driver produced ChildHandle \r
+  // \r
+  Status = EfiTestChildHandle (\r
+             ControllerHandle,\r
+             ChildHandle,\r
+             &gEfiManagedNetworkServiceBindingProtocolGuid\r
+             );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  // \r
+  // Retrieve an instance of a produced protocol from ChildHandle  \r
+  // \r
+  Status = gBS->OpenProtocol (\r
+                  ChildHandle,\r
+                  &gEfiManagedNetworkProtocolGuid,\r
+                  (VOID **)&Mnp,\r
+                  NULL,\r
+                  NULL,\r
+                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  //\r
+  // Update the component name for this child handle.\r
+  //\r
+  Status = UpdateName (Mnp);\r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  return LookupUnicodeString2 (\r
+           Language,\r
+           This->SupportedLanguages,\r
+           gMnpControllerNameTable,\r
+           ControllerName,\r
+           (BOOLEAN)(This == &gMnpComponentName)\r
+           );\r
 }\r