]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Library/DxeNetLib/DxeNetLib.c
1. Fix a bug in PXE driver that the PXE boot do not restart if a new boot option...
[mirror_edk2.git] / MdeModulePkg / Library / DxeNetLib / DxeNetLib.c
index 5bad2f13a6b519db1e82bab82d1d1ccc368b8e78..956f733400baf26f1319540b323cb0535eb82227 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
   Network library.\r
 \r
-Copyright (c) 2005 - 2010, 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 of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
@@ -13,6 +13,8 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 #include <Uefi.h>\r
 \r
+#include <IndustryStandard/SmBios.h>\r
+\r
 #include <Protocol/DriverBinding.h>\r
 #include <Protocol/ServiceBinding.h>\r
 #include <Protocol/SimpleNetwork.h>\r
@@ -23,6 +25,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Protocol/HiiConfigAccess.h>\r
 \r
 #include <Guid/NicIp4ConfigNvData.h>\r
+#include <Guid/SmBios.h>\r
 \r
 #include <Library/NetLib.h>\r
 #include <Library/BaseLib.h>\r
@@ -1826,7 +1829,7 @@ NetLibCreateServiceChild (
 \r
 \r
 /**\r
-  Destory a child of the service that is identified by ServiceBindingGuid.\r
+  Destroy a child of the service that is identified by ServiceBindingGuid.\r
 \r
   Get the ServiceBinding Protocol first, then use it to destroy a child.\r
 \r
@@ -1835,10 +1838,10 @@ NetLibCreateServiceChild (
   @param[in]   Controller            The controller which has the service installed.\r
   @param[in]   Image                 The image handle used to open service.\r
   @param[in]   ServiceBindingGuid    The service's Guid.\r
-  @param[in]   ChildHandle           The child to destory.\r
+  @param[in]   ChildHandle           The child to destroy.\r
 \r
-  @retval EFI_SUCCESS           The child is successfully destoried.\r
-  @retval Others                Failed to destory the child.\r
+  @retval EFI_SUCCESS           The child is successfully destroyed.\r
+  @retval Others                Failed to destroy the child.\r
 \r
 **/\r
 EFI_STATUS\r
@@ -1872,7 +1875,7 @@ NetLibDestroyServiceChild (
   }\r
 \r
   //\r
-  // destory the child\r
+  // destroy the child\r
   //\r
   Status = Service->DestroyChild (Service, ChildHandle);\r
   return Status;\r
@@ -2127,6 +2130,7 @@ NetLibGetMacAddress (
                     (VOID **) &Mnp\r
                     );\r
     if (EFI_ERROR (Status)) {\r
+      MnpSb->DestroyChild (MnpSb, MnpChildHandle);\r
       return Status;\r
     }\r
 \r
@@ -2135,6 +2139,7 @@ NetLibGetMacAddress (
     //\r
     Status = Mnp->GetModeData (Mnp, NULL, &SnpModeData);\r
     if (EFI_ERROR (Status)) {\r
+      MnpSb->DestroyChild (MnpSb, MnpChildHandle);\r
       return Status;\r
     }\r
     SnpMode = &SnpModeData;\r
@@ -2163,7 +2168,8 @@ NetLibGetMacAddress (
   @param[in]   ServiceHandle         The handle where network service binding protocol is\r
                                      installed on.\r
   @param[in]   ImageHandle           The image handle used to act as the agent handle to\r
-                                     get the simple network protocol.\r
+                                     get the simple network protocol. This parameter is\r
+                                     optional and may be NULL.\r
   @param[out]  MacString             The pointer to store the address of the string\r
                                      representation of  the mac address.\r
 \r
@@ -2176,7 +2182,7 @@ EFI_STATUS
 EFIAPI\r
 NetLibGetMacString (\r
   IN  EFI_HANDLE            ServiceHandle,\r
-  IN  EFI_HANDLE            ImageHandle,\r
+  IN  EFI_HANDLE            ImageHandle, OPTIONAL\r
   OUT CHAR16                **MacString\r
   )\r
 {\r
@@ -2573,7 +2579,6 @@ ON_EXIT:
 \r
   The header type of IPv4 device path node is MESSAGING_DEVICE_PATH.\r
   The header subtype of IPv4 device path node is MSG_IPv4_DP.\r
-  The length of the IPv4 device path node in bytes is 19.\r
   Get other info from parameters to make up the whole IPv4 device path node.\r
 \r
   @param[in, out]  Node                  Pointer to the IPv4 device path node.\r
@@ -2601,7 +2606,7 @@ NetLibCreateIPv4DPathNode (
 {\r
   Node->Header.Type    = MESSAGING_DEVICE_PATH;\r
   Node->Header.SubType = MSG_IPv4_DP;\r
-  SetDevicePathNodeLength (&Node->Header, 19);\r
+  SetDevicePathNodeLength (&Node->Header, sizeof (IPv4_DEVICE_PATH));\r
 \r
   CopyMem (&Node->LocalIpAddress, &LocalIp, sizeof (EFI_IPv4_ADDRESS));\r
   CopyMem (&Node->RemoteIpAddress, &RemoteIp, sizeof (EFI_IPv4_ADDRESS));\r
@@ -2616,6 +2621,14 @@ NetLibCreateIPv4DPathNode (
   } else {\r
     Node->StaticIpAddress = NetLibDefaultAddressIsStatic (Controller);\r
   }\r
+\r
+  //\r
+  // Set the Gateway IP address to default value 0:0:0:0.\r
+  // Set the Subnet mask to default value 255:255:255:0.\r
+  //\r
+  ZeroMem (&Node->GatewayIpAddress, sizeof (EFI_IPv4_ADDRESS));\r
+  SetMem (&Node->SubnetMask, sizeof (EFI_IPv4_ADDRESS), 0xff);\r
+  Node->SubnetMask.Addr[3] = 0;\r
 }\r
 \r
 /**\r
@@ -2657,7 +2670,14 @@ NetLibCreateIPv6DPathNode (
   Node->RemotePort = RemotePort;\r
 \r
   Node->Protocol        = Protocol;\r
-  Node->StaticIpAddress = FALSE;\r
+\r
+  //\r
+  // Set default value to IPAddressOrigin, PrefixLength.\r
+  // Set the Gateway IP address to unspecified address.\r
+  //\r
+  Node->IpAddressOrigin = 0;\r
+  Node->PrefixLength    = IP6_PREFIX_LENGTH;\r
+  ZeroMem (&Node->GatewayIpAddress, sizeof (EFI_IPv6_ADDRESS));\r
 }\r
 \r
 /**\r
@@ -2949,10 +2969,10 @@ NetLibAsciiStrToIp6 (
         LeadZeroCnt++;\r
       }\r
     } else {\r
-      if ((Cnt == 4) && (*TempStr  == '0') && (LeadZero == FALSE)) {\r
+      if ((Cnt == 4) && (*TempStr  == '0') && !LeadZero) {\r
         return EFI_INVALID_PARAMETER;\r
       }\r
-      if ((Cnt != 0) && (Cnt < 4) && (LeadZero == TRUE)) {\r
+      if ((Cnt != 0) && (Cnt < 4) && LeadZero) {\r
         return EFI_INVALID_PARAMETER;\r
       }\r
     } \r
@@ -3155,3 +3175,86 @@ Exit:
   return Status;\r
 }\r
 \r
+\r
+\r
+/**\r
+  This function obtains the system guid from the smbios table.\r
+\r
+  @param[out]  SystemGuid     The pointer of the returned system guid.\r
+\r
+  @retval EFI_SUCCESS         Successfully obtained the system guid.\r
+  @retval EFI_NOT_FOUND       Did not find the SMBIOS table.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+NetLibGetSystemGuid (\r
+  OUT EFI_GUID              *SystemGuid\r
+  )\r
+{\r
+  EFI_STATUS                Status;\r
+  SMBIOS_TABLE_ENTRY_POINT  *SmbiosTable;\r
+  SMBIOS_STRUCTURE_POINTER  Smbios;\r
+  SMBIOS_STRUCTURE_POINTER  SmbiosEnd;\r
+  CHAR8                     *String;\r
+\r
+  SmbiosTable = NULL;\r
+  Status      = EfiGetSystemConfigurationTable (&gEfiSmbiosTableGuid, (VOID **) &SmbiosTable);\r
+\r
+  if (EFI_ERROR (Status) || SmbiosTable == NULL) {\r
+    return EFI_NOT_FOUND;\r
+  }\r
+\r
+  Smbios.Hdr    = (SMBIOS_STRUCTURE *) (UINTN) SmbiosTable->TableAddress;\r
+  SmbiosEnd.Raw = (UINT8 *) (UINTN) (SmbiosTable->TableAddress + SmbiosTable->TableLength);\r
+\r
+  do {\r
+    if (Smbios.Hdr->Type == 1) {\r
+      if (Smbios.Hdr->Length < 0x19) {\r
+        //\r
+        // Older version did not support UUID.\r
+        //\r
+        return EFI_NOT_FOUND;\r
+      }\r
+      \r
+      //\r
+      // SMBIOS tables are byte packed so we need to do a byte copy to\r
+      // prevend alignment faults on Itanium-based platform.\r
+      //\r
+      CopyMem (SystemGuid, &Smbios.Type1->Uuid, sizeof (EFI_GUID));\r
+      return EFI_SUCCESS;\r
+    }\r
+\r
+    //\r
+    // Go to the next SMBIOS structure. Each SMBIOS structure may include 2 parts:\r
+    // 1. Formatted section; 2. Unformatted string section. So, 2 steps are needed\r
+    // to skip one SMBIOS structure.\r
+    //\r
+    \r
+    //\r
+    // Step 1: Skip over formatted section.\r
+    //\r
+    String = (CHAR8 *) (Smbios.Raw + Smbios.Hdr->Length);\r
+  \r
+    //\r
+    // Step 2: Skip over unformated string section.\r
+    //\r
+    do {\r
+      //\r
+      // Each string is terminated with a NULL(00h) BYTE and the sets of strings\r
+      // is terminated with an additional NULL(00h) BYTE.\r
+      //\r
+      for ( ; *String != 0; String++) {\r
+      }\r
+\r
+      if (*(UINT8*)++String == 0) {\r
+        //\r
+        // Pointer to the next SMBIOS structure.\r
+        //\r
+        Smbios.Raw = (UINT8 *)++String;\r
+        break;\r
+      }    \r
+    } while (TRUE);\r
+  } while (Smbios.Raw < SmbiosEnd.Raw);\r
+  return EFI_NOT_FOUND;\r
+}\r