[mirror_edk2.git] / NetworkPkg / DnsDxe / DnsDhcp.c

/** @file\r
Functions implementation related with DHCPv4/v6 for DNS driver.\r
\r
Copyright (c) 2015 - 2018, 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
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "DnsImpl.h"\r
\r
/**\r
  This function initialize the DHCP4 message instance.\r
\r
  This function will pad each item of dhcp4 message packet.\r
\r
  @param  Seed             Pointer to the message instance of the DHCP4 packet.\r
  @param  InterfaceInfo    Pointer to the EFI_IP4_CONFIG2_INTERFACE_INFO instance.\r
\r
**/\r
VOID\r
DnsInitSeedPacket (\r
  OUT EFI_DHCP4_PACKET               *Seed,\r
  IN  EFI_IP4_CONFIG2_INTERFACE_INFO *InterfaceInfo\r
  )\r
{\r
  EFI_DHCP4_HEADER           *Header;\r
\r
  //\r
  // Get IfType and HwAddressSize from SNP mode data.\r
  //\r
  Seed->Size            = sizeof (EFI_DHCP4_PACKET);\r
  Seed->Length          = sizeof (Seed->Dhcp4);\r
  Header                = &Seed->Dhcp4.Header;\r
  ZeroMem (Header, sizeof (EFI_DHCP4_HEADER));\r
  Header->OpCode        = DHCP4_OPCODE_REQUEST;\r
  Header->HwType        = InterfaceInfo->IfType;\r
  Header->HwAddrLen     = (UINT8) InterfaceInfo->HwAddressSize;\r
  CopyMem (Header->ClientHwAddr, &(InterfaceInfo->HwAddress), Header->HwAddrLen);\r
\r
  Seed->Dhcp4.Magik     = DHCP4_MAGIC;\r
  Seed->Dhcp4.Option[0] = DHCP4_TAG_EOP;\r
}\r
\r
/**\r
  The common notify function.\r
\r
  @param[in]  Event   The event signaled.\r
  @param[in]  Context The context.\r
\r
**/\r
VOID\r
EFIAPI\r
DhcpCommonNotify (\r
  IN EFI_EVENT  Event,\r
  IN VOID       *Context\r
  )\r
{\r
  if ((Event == NULL) || (Context == NULL)) {\r
    return ;\r
  }\r
\r
  *((BOOLEAN *) Context) = TRUE;\r
}\r
\r
/**\r
  Parse the ACK to get required information\r
\r
  @param  Dhcp4            The DHCP4 protocol.\r
  @param  Packet           Packet waiting for parse.\r
  @param  DnsServerInfor   The required Dns4 server information.\r
\r
  @retval EFI_SUCCESS           The DNS information is got from the DHCP ACK.\r
  @retval EFI_NO_MAPPING        DHCP failed to acquire address and other information.\r
  @retval EFI_DEVICE_ERROR      Other errors as indicated.\r
  @retval EFI_OUT_OF_RESOURCES  Failed to allocate memory.\r
\r
**/\r
EFI_STATUS\r
ParseDhcp4Ack (\r
  IN EFI_DHCP4_PROTOCOL         *Dhcp4,\r
  IN EFI_DHCP4_PACKET           *Packet,\r
  IN DNS4_SERVER_INFOR          *DnsServerInfor\r
  )\r
{\r
  EFI_STATUS              Status;\r
  UINT32                  OptionCount;\r
  EFI_DHCP4_PACKET_OPTION **OptionList;\r
  UINT32                  ServerCount;\r
  EFI_IPv4_ADDRESS        *ServerList;\r
  UINT32                  Index;\r
  UINT32                  Count;\r
\r
  ServerCount = 0;\r
  ServerList = NULL;\r
\r
  OptionCount = 0;\r
  OptionList  = NULL;\r
\r
  Status      = Dhcp4->Parse (Dhcp4, Packet, &OptionCount, OptionList);\r
  if (Status != EFI_BUFFER_TOO_SMALL) {\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  OptionList = AllocatePool (OptionCount * sizeof (EFI_DHCP4_PACKET_OPTION *));\r
  if (OptionList == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  Status = Dhcp4->Parse (Dhcp4, Packet, &OptionCount, OptionList);\r
  if (EFI_ERROR (Status)) {\r
    gBS->FreePool (OptionList);\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  Status = EFI_NOT_FOUND;\r
\r
  for (Index = 0; Index < OptionCount; Index++) {\r
    //\r
    // Get DNS server addresses\r
    //\r
    if (OptionList[Index]->OpCode == DHCP4_TAG_DNS_SERVER) {\r
\r
      if (((OptionList[Index]->Length & 0x3) != 0) || (OptionList[Index]->Length == 0)) {\r
        Status = EFI_DEVICE_ERROR;\r
        break;\r
      }\r
\r
      ServerCount = OptionList[Index]->Length/4;\r
      ServerList = AllocatePool (ServerCount * sizeof (EFI_IPv4_ADDRESS));\r
      if (ServerList == NULL) {\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
\r
      for(Count=0; Count < ServerCount; Count++){\r
        CopyMem (ServerList + Count, &OptionList[Index]->Data[4 * Count], sizeof (EFI_IPv4_ADDRESS));\r
      }\r
\r
      *(DnsServerInfor->ServerCount) = ServerCount;\r
      DnsServerInfor->ServerList     = ServerList;\r
\r
      Status = EFI_SUCCESS;\r
    }\r
  }\r
\r
  gBS->FreePool (OptionList);\r
\r
  return Status;\r
}\r
\r
/**\r
  EFI_DHCP6_INFO_CALLBACK is provided by the consumer of the EFI DHCPv6 Protocol\r
  instance to intercept events that occurs in the DHCPv6 Information Request\r
  exchange process.\r
\r
  @param  This                  Pointer to the EFI_DHCP6_PROTOCOL instance that\r
                                is used to configure this  callback function.\r
  @param  Context               Pointer to the context that is initialized in\r
                                the EFI_DHCP6_PROTOCOL.InfoRequest().\r
  @param  Packet                Pointer to Reply packet that has been received.\r
                                The EFI DHCPv6 Protocol instance is responsible\r
                                for freeing the buffer.\r
\r
  @retval EFI_SUCCESS           The DNS information is got from the DHCP ACK.\r
  @retval EFI_DEVICE_ERROR      Other errors as indicated.\r
  @retval EFI_OUT_OF_RESOURCES  Failed to allocate memory.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ParseDhcp6Ack (\r
  IN EFI_DHCP6_PROTOCOL          *This,\r
  IN VOID                        *Context,\r
  IN EFI_DHCP6_PACKET            *Packet\r
  )\r
{\r
  EFI_STATUS                  Status;\r
  UINT32                      OptionCount;\r
  EFI_DHCP6_PACKET_OPTION     **OptionList;\r
  DNS6_SERVER_INFOR           *DnsServerInfor;\r
  UINT32                      ServerCount;\r
  EFI_IPv6_ADDRESS            *ServerList;\r
  UINT32                      Index;\r
  UINT32                      Count;\r
\r
  OptionCount = 0;\r
  ServerCount = 0;\r
  ServerList  = NULL;\r
\r
  Status      = This->Parse (This, Packet, &OptionCount, NULL);\r
  if (Status != EFI_BUFFER_TOO_SMALL) {\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  OptionList = AllocateZeroPool (OptionCount * sizeof (EFI_DHCP6_PACKET_OPTION *));\r
  if (OptionList == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  Status = This->Parse (This, Packet, &OptionCount, OptionList);\r
  if (EFI_ERROR (Status)) {\r
    gBS->FreePool (OptionList);\r
    return EFI_DEVICE_ERROR;\r
  }\r
\r
  DnsServerInfor = (DNS6_SERVER_INFOR *) Context;\r
\r
  for (Index = 0; Index < OptionCount; Index++) {\r
    OptionList[Index]->OpCode = NTOHS (OptionList[Index]->OpCode);\r
    OptionList[Index]->OpLen  = NTOHS (OptionList[Index]->OpLen);\r
\r
    //\r
    // Get DNS server addresses from this reply packet.\r
    //\r
    if (OptionList[Index]->OpCode == DHCP6_TAG_DNS_SERVER) {\r
\r
      if (((OptionList[Index]->OpLen & 0xf) != 0) || (OptionList[Index]->OpLen == 0)) {\r
        Status = EFI_DEVICE_ERROR;\r
        gBS->FreePool (OptionList);\r
        return Status;\r
      }\r
\r
      ServerCount = OptionList[Index]->OpLen/16;\r
      ServerList = AllocatePool (ServerCount * sizeof (EFI_IPv6_ADDRESS));\r
      if (ServerList == NULL) {\r
        gBS->FreePool (OptionList);\r
        return EFI_OUT_OF_RESOURCES;\r
      }\r
\r
      for(Count=0; Count < ServerCount; Count++){\r
        CopyMem (ServerList + Count, &OptionList[Index]->Data[16 * Count], sizeof (EFI_IPv6_ADDRESS));\r
      }\r
\r
      *(DnsServerInfor->ServerCount) = ServerCount;\r
      DnsServerInfor->ServerList     = ServerList;\r
    }\r
  }\r
\r
  gBS->FreePool (OptionList);\r
\r
  return Status;\r
\r
}\r
\r
/**\r
  Parse the DHCP ACK to get Dns4 server information.\r
\r
  @param  Instance         The DNS instance.\r
  @param  DnsServerCount   Retrieved Dns4 server Ip count.\r
  @param  DnsServerList    Retrieved Dns4 server Ip list.\r
\r
  @retval EFI_SUCCESS           The Dns4 information is got from the DHCP ACK.\r
  @retval EFI_OUT_OF_RESOURCES  Failed to allocate memory.\r
  @retval EFI_NO_MEDIA          There was a media error.\r
  @retval Others                Other errors as indicated.\r
\r
**/\r
EFI_STATUS\r
GetDns4ServerFromDhcp4 (\r
  IN  DNS_INSTANCE               *Instance,\r
  OUT UINT32                     *DnsServerCount,\r
  OUT EFI_IPv4_ADDRESS           **DnsServerList\r
  )\r
{\r
  EFI_STATUS                          Status;\r
  EFI_HANDLE                          Image;\r
  EFI_HANDLE                          Controller;\r
  EFI_STATUS                          MediaStatus;\r
  EFI_HANDLE                          MnpChildHandle;\r
  EFI_MANAGED_NETWORK_PROTOCOL        *Mnp;\r
  EFI_MANAGED_NETWORK_CONFIG_DATA     MnpConfigData;\r
  EFI_HANDLE                          Dhcp4Handle;\r
  EFI_DHCP4_PROTOCOL                  *Dhcp4;\r
  EFI_IP4_CONFIG2_PROTOCOL            *Ip4Config2;\r
  UINTN                               DataSize;\r
  VOID                                *Data;\r
  EFI_IP4_CONFIG2_INTERFACE_INFO      *InterfaceInfo;\r
  EFI_DHCP4_PACKET                    SeedPacket;\r
  EFI_DHCP4_PACKET_OPTION             *ParaList[2];\r
  DNS4_SERVER_INFOR                   DnsServerInfor;\r
\r
  EFI_DHCP4_TRANSMIT_RECEIVE_TOKEN    Token;\r
  BOOLEAN                             IsDone;\r
  UINTN                               Index;\r
\r
  Image                      = Instance->Service->ImageHandle;\r
  Controller                 = Instance->Service->ControllerHandle;\r
\r
  MnpChildHandle             = NULL;\r
  Mnp                        = NULL;\r
\r
  Dhcp4Handle                = NULL;\r
  Dhcp4                      = NULL;\r
\r
  Ip4Config2                 = NULL;\r
  DataSize                   = 0;\r
  Data                       = NULL;\r
  InterfaceInfo              = NULL;\r
\r
  ZeroMem ((UINT8 *) ParaList, sizeof (ParaList));\r
\r
  ZeroMem (&MnpConfigData, sizeof (EFI_MANAGED_NETWORK_CONFIG_DATA));\r
\r
  ZeroMem (&DnsServerInfor, sizeof (DNS4_SERVER_INFOR));\r
\r
  ZeroMem (&Token, sizeof (EFI_DHCP4_TRANSMIT_RECEIVE_TOKEN));\r
\r
  DnsServerInfor.ServerCount = DnsServerCount;\r
\r
  IsDone = FALSE;\r
\r
  //\r
  // Check media.\r
  //\r
  MediaStatus = EFI_SUCCESS;\r
  NetLibDetectMediaWaitTimeout (Controller, DNS_CHECK_MEDIA_GET_DHCP_WAITING_TIME, &MediaStatus);\r
  if (MediaStatus != EFI_SUCCESS) {\r
    return EFI_NO_MEDIA;\r
  }\r
\r
  //\r
  // Create a Mnp child instance, get the protocol and config for it.\r
  //\r
  Status = NetLibCreateServiceChild (\r
             Controller,\r
             Image,\r
             &gEfiManagedNetworkServiceBindingProtocolGuid,\r
             &MnpChildHandle\r
             );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                  MnpChildHandle,\r
                  &gEfiManagedNetworkProtocolGuid,\r
                  (VOID **) &Mnp,\r
                  Image,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  MnpConfigData.ReceivedQueueTimeoutValue = 0;\r
  MnpConfigData.TransmitQueueTimeoutValue = 0;\r
  MnpConfigData.ProtocolTypeFilter        = IP4_ETHER_PROTO;\r
  MnpConfigData.EnableUnicastReceive      = TRUE;\r
  MnpConfigData.EnableMulticastReceive    = TRUE;\r
  MnpConfigData.EnableBroadcastReceive    = TRUE;\r
  MnpConfigData.EnablePromiscuousReceive  = FALSE;\r
  MnpConfigData.FlushQueuesOnReset        = TRUE;\r
  MnpConfigData.EnableReceiveTimestamps   = FALSE;\r
  MnpConfigData.DisableBackgroundPolling  = FALSE;\r
\r
  Status = Mnp->Configure(Mnp, &MnpConfigData);\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  //\r
  // Create a DHCP4 child instance and get the protocol.\r
  //\r
  Status = NetLibCreateServiceChild (\r
             Controller,\r
             Image,\r
             &gEfiDhcp4ServiceBindingProtocolGuid,\r
             &Dhcp4Handle\r
             );\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                  Dhcp4Handle,\r
                  &gEfiDhcp4ProtocolGuid,\r
                  (VOID **) &Dhcp4,\r
                  Image,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  //\r
  // Get Ip4Config2 instance info.\r
  //\r
  Status = gBS->HandleProtocol (Controller, &gEfiIp4Config2ProtocolGuid, (VOID **) &Ip4Config2);\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  Status = Ip4Config2->GetData (Ip4Config2, Ip4Config2DataTypeInterfaceInfo, &DataSize, Data);\r
  if (EFI_ERROR (Status) && Status != EFI_BUFFER_TOO_SMALL) {\r
    goto ON_EXIT;\r
  }\r
\r
  Data = AllocateZeroPool (DataSize);\r
  if (Data == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  Status = Ip4Config2->GetData (Ip4Config2, Ip4Config2DataTypeInterfaceInfo, &DataSize, Data);\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  InterfaceInfo = (EFI_IP4_CONFIG2_INTERFACE_INFO *)Data;\r
\r
  //\r
  // Build required Token.\r
  //\r
  Status = gBS->CreateEvent (\r
                  EVT_NOTIFY_SIGNAL,\r
                  TPL_NOTIFY,\r
                  DhcpCommonNotify,\r
                  &IsDone,\r
                  &Token.CompletionEvent\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  SetMem (&Token.RemoteAddress, sizeof (EFI_IPv4_ADDRESS), 0xff);\r
\r
  Token.RemotePort = 67;\r
\r
  Token.ListenPointCount = 1;\r
\r
  Token.ListenPoints = AllocateZeroPool (Token.ListenPointCount * sizeof (EFI_DHCP4_LISTEN_POINT));\r
  if (Token.ListenPoints == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  if (Instance->Dns4CfgData.UseDefaultSetting) {\r
    CopyMem (&(Token.ListenPoints[0].ListenAddress), &(InterfaceInfo->StationAddress), sizeof (EFI_IPv4_ADDRESS));\r
    CopyMem (&(Token.ListenPoints[0].SubnetMask), &(InterfaceInfo->SubnetMask), sizeof (EFI_IPv4_ADDRESS));\r
  } else {\r
    CopyMem (&(Token.ListenPoints[0].ListenAddress), &(Instance->Dns4CfgData.StationIp), sizeof (EFI_IPv4_ADDRESS));\r
    CopyMem (&(Token.ListenPoints[0].SubnetMask), &(Instance->Dns4CfgData.SubnetMask), sizeof (EFI_IPv4_ADDRESS));\r
  }\r
\r
  Token.ListenPoints[0].ListenPort = 68;\r
\r
  Token.TimeoutValue = DNS_TIME_TO_GETMAP;\r
\r
  DnsInitSeedPacket (&SeedPacket, InterfaceInfo);\r
\r
  ParaList[0] = AllocateZeroPool (sizeof (EFI_DHCP4_PACKET_OPTION));\r
  if (ParaList[0] == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  ParaList[0]->OpCode  = DHCP4_TAG_TYPE;\r
  ParaList[0]->Length  = 1;\r
  ParaList[0]->Data[0] = DHCP4_MSG_REQUEST;\r
\r
  ParaList[1] = AllocateZeroPool (sizeof (EFI_DHCP4_PACKET_OPTION));\r
  if (ParaList[1] == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  ParaList[1]->OpCode  = DHCP4_TAG_PARA_LIST;\r
  ParaList[1]->Length  = 1;\r
  ParaList[1]->Data[0] = DHCP4_TAG_DNS_SERVER;\r
\r
  Status = Dhcp4->Build (Dhcp4, &SeedPacket, 0, NULL, 2, ParaList, &Token.Packet);\r
\r
  Token.Packet->Dhcp4.Header.Xid      = HTONL(NET_RANDOM (NetRandomInitSeed ()));\r
\r
  Token.Packet->Dhcp4.Header.Reserved = HTONS ((UINT16)0x8000);\r
\r
  if (Instance->Dns4CfgData.UseDefaultSetting) {\r
    CopyMem (&(Token.Packet->Dhcp4.Header.ClientAddr), &(InterfaceInfo->StationAddress), sizeof (EFI_IPv4_ADDRESS));\r
  } else {\r
    CopyMem (&(Token.Packet->Dhcp4.Header.ClientAddr), &(Instance->Dns4CfgData.StationIp), sizeof (EFI_IPv4_ADDRESS));\r
  }\r
\r
  CopyMem (Token.Packet->Dhcp4.Header.ClientHwAddr, &(InterfaceInfo->HwAddress), InterfaceInfo->HwAddressSize);\r
\r
  Token.Packet->Dhcp4.Header.HwAddrLen = (UINT8)(InterfaceInfo->HwAddressSize);\r
\r
  //\r
  // TransmitReceive Token\r
  //\r
  Status = Dhcp4->TransmitReceive (Dhcp4, &Token);\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  //\r
  // Poll the packet\r
  //\r
  do {\r
    Status = Mnp->Poll (Mnp);\r
  } while (!IsDone);\r
\r
  //\r
  // Parse the ACK to get required information if received done.\r
  //\r
  if (IsDone && !EFI_ERROR (Token.Status)) {\r
    for (Index = 0; Index < Token.ResponseCount; Index++) {\r
      Status = ParseDhcp4Ack (Dhcp4, &Token.ResponseList[Index], &DnsServerInfor);\r
      if (!EFI_ERROR (Status)) {\r
        break;\r
      }\r
    }\r
\r
    *DnsServerList = DnsServerInfor.ServerList;\r
  } else {\r
    Status = Token.Status;\r
  }\r
\r
ON_EXIT:\r
\r
  if (Data != NULL) {\r
    FreePool (Data);\r
  }\r
\r
  for (Index = 0; Index < 2; Index++) {\r
    if (ParaList[Index] != NULL) {\r
      FreePool (ParaList[Index]);\r
    }\r
  }\r
\r
  if (Token.ListenPoints) {\r
    FreePool (Token.ListenPoints);\r
  }\r
\r
  if (Token.Packet) {\r
    FreePool (Token.Packet);\r
  }\r
\r
  if (Token.ResponseList != NULL) {\r
    FreePool (Token.ResponseList);\r
  }\r
\r
  if (Token.CompletionEvent != NULL) {\r
    gBS->CloseEvent (Token.CompletionEvent);\r
  }\r
\r
  if (Dhcp4 != NULL) {\r
    Dhcp4->Stop (Dhcp4);\r
    Dhcp4->Configure (Dhcp4, NULL);\r
\r
    gBS->CloseProtocol (\r
           Dhcp4Handle,\r
           &gEfiDhcp4ProtocolGuid,\r
           Image,\r
           Controller\r
           );\r
  }\r
\r
  if (Dhcp4Handle != NULL) {\r
    NetLibDestroyServiceChild (\r
      Controller,\r
      Image,\r
      &gEfiDhcp4ServiceBindingProtocolGuid,\r
      Dhcp4Handle\r
      );\r
  }\r
\r
  if (Mnp != NULL) {\r
    Mnp->Configure (Mnp, NULL);\r
\r
    gBS->CloseProtocol (\r
           MnpChildHandle,\r
           &gEfiManagedNetworkProtocolGuid,\r
           Image,\r
           Controller\r
           );\r
  }\r
\r
  NetLibDestroyServiceChild (\r
    Controller,\r
    Image,\r
    &gEfiManagedNetworkServiceBindingProtocolGuid,\r
    MnpChildHandle\r
    );\r
\r
  return Status;\r
}\r
\r
/**\r
  Parse the DHCP ACK to get Dns6 server information.\r
\r
  @param  Image            The handle of the driver image.\r
  @param  Controller       The handle of the controller.\r
  @param  DnsServerCount   Retrieved Dns6 server Ip count.\r
  @param  DnsServerList    Retrieved Dns6 server Ip list.\r
\r
  @retval EFI_SUCCESS           The Dns6 information is got from the DHCP ACK.\r
  @retval EFI_OUT_OF_RESOURCES  Failed to allocate memory.\r
  @retval EFI_NO_MEDIA          There was a media error.\r
  @retval Others                Other errors as indicated.\r
\r
**/\r
EFI_STATUS\r
GetDns6ServerFromDhcp6 (\r
  IN  EFI_HANDLE                 Image,\r
  IN  EFI_HANDLE                 Controller,\r
  OUT UINT32                     *DnsServerCount,\r
  OUT EFI_IPv6_ADDRESS           **DnsServerList\r
  )\r
{\r
  EFI_HANDLE                Dhcp6Handle;\r
  EFI_DHCP6_PROTOCOL        *Dhcp6;\r
  EFI_STATUS                Status;\r
  EFI_STATUS                TimerStatus;\r
  EFI_DHCP6_PACKET_OPTION   *Oro;\r
  EFI_DHCP6_RETRANSMISSION  InfoReqReXmit;\r
  EFI_EVENT                 Timer;\r
  EFI_STATUS                MediaStatus;\r
  DNS6_SERVER_INFOR         DnsServerInfor;\r
\r
  Dhcp6Handle = NULL;\r
  Dhcp6       = NULL;\r
  Oro         = NULL;\r
  Timer       = NULL;\r
\r
  ZeroMem (&DnsServerInfor, sizeof (DNS6_SERVER_INFOR));\r
\r
  DnsServerInfor.ServerCount = DnsServerCount;\r
\r
  //\r
  // Check media status before doing DHCP.\r
  //\r
  MediaStatus = EFI_SUCCESS;\r
  NetLibDetectMediaWaitTimeout (Controller, DNS_CHECK_MEDIA_GET_DHCP_WAITING_TIME, &MediaStatus);\r
  if (MediaStatus != EFI_SUCCESS) {\r
    return EFI_NO_MEDIA;\r
  }\r
\r
  //\r
  // Create a DHCP6 child instance and get the protocol.\r
  //\r
  Status = NetLibCreateServiceChild (\r
             Controller,\r
             Image,\r
             &gEfiDhcp6ServiceBindingProtocolGuid,\r
             &Dhcp6Handle\r
             );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                  Dhcp6Handle,\r
                  &gEfiDhcp6ProtocolGuid,\r
                  (VOID **) &Dhcp6,\r
                  Image,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    goto ON_EXIT;\r
  }\r
\r
  Oro = AllocateZeroPool (sizeof (EFI_DHCP6_PACKET_OPTION) + 1);\r
  if (Oro == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  //\r
  // Ask the server to reply with DNS options.\r
  // All members in EFI_DHCP6_PACKET_OPTION are in network order.\r
  //\r
  Oro->OpCode  = HTONS (DHCP6_TAG_DNS_REQUEST);\r
  Oro->OpLen   = HTONS (2);\r
  Oro->Data[1] = DHCP6_TAG_DNS_SERVER;\r
\r
  InfoReqReXmit.Irt = 4;\r
  InfoReqReXmit.Mrc = 1;\r
  InfoReqReXmit.Mrt = 10;\r
  InfoReqReXmit.Mrd = 30;\r
\r
  Status = Dhcp6->InfoRequest (\r
                    Dhcp6,\r
                    TRUE,\r
                    Oro,\r
                    0,\r
                    NULL,\r
                    &InfoReqReXmit,\r
                    NULL,\r
                    ParseDhcp6Ack,\r
                    &DnsServerInfor\r
                    );\r
  if (Status == EFI_NO_MAPPING) {\r
    Status = gBS->CreateEvent (EVT_TIMER, TPL_CALLBACK, NULL, NULL, &Timer);\r
    if (EFI_ERROR (Status)) {\r
      goto ON_EXIT;\r
    }\r
\r
    Status = gBS->SetTimer (\r
                    Timer,\r
                    TimerRelative,\r
                    DNS_TIME_TO_GETMAP * TICKS_PER_SECOND\r
                    );\r
\r
    if (EFI_ERROR (Status)) {\r
      goto ON_EXIT;\r
    }\r
\r
    do {\r
      TimerStatus = gBS->CheckEvent (Timer);\r
      if (!EFI_ERROR (TimerStatus)) {\r
        Status = Dhcp6->InfoRequest (\r
                          Dhcp6,\r
                          TRUE,\r
                          Oro,\r
                          0,\r
                          NULL,\r
                          &InfoReqReXmit,\r
                          NULL,\r
                          ParseDhcp6Ack,\r
                          &DnsServerInfor\r
                          );\r
      }\r
    } while (TimerStatus == EFI_NOT_READY);\r
  }\r
\r
  *DnsServerList  = DnsServerInfor.ServerList;\r
\r
ON_EXIT:\r
\r
  if (Oro != NULL) {\r
    FreePool (Oro);\r
  }\r
\r
  if (Timer != NULL) {\r
    gBS->CloseEvent (Timer);\r
  }\r
\r
  if (Dhcp6 != NULL) {\r
    gBS->CloseProtocol (\r
           Dhcp6Handle,\r
           &gEfiDhcp6ProtocolGuid,\r
           Image,\r
           Controller\r
           );\r
  }\r
\r
  NetLibDestroyServiceChild (\r
    Controller,\r
    Image,\r
    &gEfiDhcp6ServiceBindingProtocolGuid,\r
    Dhcp6Handle\r
    );\r
\r
  return Status;\r
\r
}\r
\r