1. Add EFI_COMPONENT_NAME2_PROTOCOL.GetControllerName() support.

[mirror_edk2.git] / NetworkPkg / Ip6Dxe / Ip6Common.c

/** @file\r
  The implementation of common functions shared by IP6 driver.\r
\r
  Copyright (c) 2009 - 2012, 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
  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 "Ip6Impl.h"\r
\r
/**\r
  Build a array of EFI_IP6_ADDRESS_INFO to be returned to the caller. The number\r
  of EFI_IP6_ADDRESS_INFO is also returned. If AddressList is NULL,\r
  only the address count is returned.\r
\r
  @param[in]  IpSb              The IP6 service binding instance.\r
  @param[out] AddressCount      The number of returned addresses.\r
  @param[out] AddressList       The pointer to the array of EFI_IP6_ADDRESS_INFO.\r
                                This is an optional parameter.\r
\r
\r
  @retval EFI_SUCCESS           The address array successfully built.\r
  @retval EFI_OUT_OF_RESOURCES  Failed to allocate the memory for the address info.\r
  @retval EFI_INVALID_PARAMETER Any input parameter is invalid.\r
\r
**/\r
EFI_STATUS\r
Ip6BuildEfiAddressList (\r
  IN IP6_SERVICE            *IpSb,\r
  OUT UINT32                *AddressCount,\r
  OUT EFI_IP6_ADDRESS_INFO  **AddressList OPTIONAL\r
  )\r
{\r
  UINT32                Count;\r
  LIST_ENTRY            *Entry;\r
  EFI_IP6_ADDRESS_INFO  *EfiAddrInfo;\r
  IP6_ADDRESS_INFO      *AddrInfo;\r
\r
  if (AddressCount == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (IpSb->LinkLocalOk) {\r
    Count = 1 + IpSb->DefaultInterface->AddressCount;\r
  } else {\r
    Count = 0;\r
  }\r
\r
  *AddressCount = Count;\r
\r
  if ((AddressList == NULL) || (Count == 0)) {\r
    return EFI_SUCCESS;\r
  }\r
\r
  if (*AddressList == NULL) {\r
    *AddressList = AllocatePool (sizeof (EFI_IP6_ADDRESS_INFO) * Count);\r
    if (*AddressList == NULL) {\r
      return EFI_OUT_OF_RESOURCES;\r
    }\r
  }\r
\r
  EfiAddrInfo = *AddressList;\r
\r
  IP6_COPY_ADDRESS (&EfiAddrInfo->Address, &IpSb->LinkLocalAddr);\r
  EfiAddrInfo->PrefixLength = IP6_LINK_LOCAL_PREFIX_LENGTH;\r
\r
  EfiAddrInfo++;\r
  Count = 1;\r
\r
  NET_LIST_FOR_EACH (Entry, &IpSb->DefaultInterface->AddressList) {\r
    AddrInfo = NET_LIST_USER_STRUCT_S (Entry, IP6_ADDRESS_INFO, Link, IP6_ADDR_INFO_SIGNATURE);\r
\r
    IP6_COPY_ADDRESS (&EfiAddrInfo->Address, &AddrInfo->Address);\r
    EfiAddrInfo->PrefixLength = AddrInfo->PrefixLength;\r
\r
    EfiAddrInfo++;\r
    Count++;\r
  }\r
\r
  ASSERT (Count == *AddressCount);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Generate the multicast addresses identify the group of all IPv6 nodes or IPv6\r
  routers defined in RFC4291.\r
\r
  All Nodes Addresses: FF01::1, FF02::1.\r
  All Router Addresses: FF01::2, FF02::2, FF05::2.\r
\r
  @param[in]  Router            If TRUE, generate all routers addresses,\r
                                else generate all node addresses.\r
  @param[in]  Scope             interface-local(1), link-local(2), or site-local(5)\r
  @param[out] Ip6Addr           The generated multicast address.\r
\r
  @retval EFI_INVALID_PARAMETER Any input parameter is invalid.\r
  @retval EFI_SUCCESS           The address is generated.\r
\r
**/\r
EFI_STATUS\r
Ip6SetToAllNodeMulticast (\r
  IN  BOOLEAN          Router,\r
  IN  UINT8            Scope,\r
  OUT EFI_IPv6_ADDRESS *Ip6Addr\r
  )\r
{\r
  if (Ip6Addr == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (!Router && Scope == IP6_SITE_LOCAL_SCOPE) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  ZeroMem (Ip6Addr, sizeof (EFI_IPv6_ADDRESS));\r
  Ip6Addr->Addr[0] = 0xFF;\r
  Ip6Addr->Addr[1] = Scope;\r
\r
  if (!Router) {\r
    Ip6Addr->Addr[15] = 0x1;\r
  } else {\r
    Ip6Addr->Addr[15] = 0x2;\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  This function converts MAC address to 64 bits interface ID according to RFC4291\r
  and returns the interface ID. Currently only 48-bit MAC address is supported by\r
  this function.\r
\r
  @param[in, out]  IpSb      The IP6 service binding instance.\r
\r
  @retval          NULL      The operation fails.\r
  @return                    Pointer to the generated interface ID.\r
\r
**/\r
UINT8 *\r
Ip6CreateInterfaceID (\r
  IN OUT IP6_SERVICE         *IpSb\r
  )\r
{\r
  UINT8                      InterfaceId[8];\r
  UINT8                      Byte;\r
  EFI_MAC_ADDRESS            *MacAddr;\r
  UINT32                     AddrLen;\r
\r
  NET_CHECK_SIGNATURE (IpSb, IP6_SERVICE_SIGNATURE);\r
\r
  AddrLen = IpSb->SnpMode.HwAddressSize;\r
\r
  //\r
  // Currently only IEEE 802 48-bit MACs are supported to create link local address.\r
  //\r
  if (AddrLen != IP6_MAC_LEN || IpSb->InterfaceIdLen != IP6_IF_ID_LEN) {\r
    return NULL;\r
  }\r
\r
  MacAddr = &IpSb->SnpMode.CurrentAddress;\r
\r
  //\r
  // Convert MAC address to 64 bits interface ID according to Appendix A of RFC4291:\r
  // 1. Insert 0xFFFE to the middle\r
  // 2. Invert the universal/local bit - bit 6 in network order\r
  //\r
  CopyMem (InterfaceId, MacAddr, 3);\r
  InterfaceId[3] = 0xFF;\r
  InterfaceId[4] = 0xFE;\r
  CopyMem (&InterfaceId[5], &MacAddr->Addr[3], 3);\r
\r
  Byte = (UINT8) (InterfaceId[0] & IP6_U_BIT);\r
  if (Byte == IP6_U_BIT) {\r
    InterfaceId[0] &= ~IP6_U_BIT;\r
  } else {\r
    InterfaceId[0] |= IP6_U_BIT;\r
  }\r
\r
  //\r
  // Return the interface ID.\r
  //\r
  return AllocateCopyPool (IpSb->InterfaceIdLen, InterfaceId);\r
}\r
\r
/**\r
  This function creates link-local address from interface identifier. The\r
  interface identifier is normally created from MAC address. It might be manually\r
  configured by administrator if the link-local address created from MAC address\r
  is a duplicate address.\r
\r
  @param[in, out]  IpSb      The IP6 service binding instance.\r
\r
  @retval          NULL      If the operation fails.\r
  @return                    The generated Link Local address, in network order.\r
\r
**/\r
EFI_IPv6_ADDRESS *\r
Ip6CreateLinkLocalAddr (\r
  IN OUT IP6_SERVICE           *IpSb\r
  )\r
{\r
  EFI_IPv6_ADDRESS             *Ip6Addr;\r
  EFI_IP6_CONFIG_PROTOCOL      *Ip6Config;\r
  UINTN                        DataSize;\r
  EFI_IP6_CONFIG_INTERFACE_ID  InterfaceId;\r
  EFI_STATUS                   Status;\r
\r
  NET_CHECK_SIGNATURE (IpSb, IP6_SERVICE_SIGNATURE);\r
\r
  if (IpSb->InterfaceId != NULL) {\r
    FreePool (IpSb->InterfaceId);\r
  }\r
\r
  //\r
  // Get the interface id if it is manully configured.\r
  //\r
  Ip6Config = &IpSb->Ip6ConfigInstance.Ip6Config;\r
  DataSize  = sizeof (EFI_IP6_CONFIG_INTERFACE_ID);\r
  ZeroMem (&InterfaceId, DataSize);\r
\r
  Status = Ip6Config->GetData (\r
                        Ip6Config,\r
                        Ip6ConfigDataTypeAltInterfaceId,\r
                        &DataSize,\r
                        &InterfaceId\r
                        );\r
  if (Status == EFI_NOT_FOUND) {\r
    //\r
    // Since the interface id is not configured, generate the interface id from\r
    // MAC address.\r
    //\r
    IpSb->InterfaceId = Ip6CreateInterfaceID (IpSb);\r
    if (IpSb->InterfaceId == NULL) {\r
      return NULL;\r
    }\r
\r
    CopyMem (&InterfaceId, IpSb->InterfaceId, IpSb->InterfaceIdLen);\r
    //\r
    // Record the interface id.\r
    //\r
    Status = Ip6Config->SetData (\r
                          Ip6Config,\r
                          Ip6ConfigDataTypeAltInterfaceId,\r
                          DataSize,\r
                          &InterfaceId\r
                          );\r
    if (EFI_ERROR (Status)) {\r
      FreePool (IpSb->InterfaceId);\r
      IpSb->InterfaceId = NULL;\r
      return NULL;\r
    }\r
  } else if (!EFI_ERROR (Status)) {\r
    IpSb->InterfaceId = AllocateCopyPool (DataSize, &InterfaceId);\r
    if (IpSb->InterfaceId == NULL) {\r
      return NULL;\r
    }\r
  } else {\r
    return NULL;\r
  }\r
\r
  //\r
  // Append FE80::/64 to the left of IPv6 address then return.\r
  //\r
  Ip6Addr = AllocateZeroPool (sizeof (EFI_IPv6_ADDRESS));\r
  if (Ip6Addr == NULL) {\r
    FreePool (IpSb->InterfaceId);\r
    IpSb->InterfaceId = NULL;\r
    return NULL;\r
  }\r
\r
  CopyMem (&Ip6Addr->Addr[8], IpSb->InterfaceId, IpSb->InterfaceIdLen);\r
  Ip6Addr->Addr[1] = 0x80;\r
  Ip6Addr->Addr[0] = 0xFE;\r
\r
  return Ip6Addr;\r
}\r
\r
/**\r
  Compute the solicited-node multicast address for an unicast or anycast address,\r
  by taking the low-order 24 bits of this address, and appending those bits to\r
  the prefix FF02:0:0:0:0:1:FF00::/104.\r
\r
  @param[in]  Ip6Addr               The unicast or anycast address, in network order.\r
  @param[out] MulticastAddr         The generated solicited-node multicast address,\r
                                    in network order.\r
\r
**/\r
VOID\r
Ip6CreateSNMulticastAddr (\r
  IN EFI_IPv6_ADDRESS  *Ip6Addr,\r
  OUT EFI_IPv6_ADDRESS *MulticastAddr\r
  )\r
{\r
  ASSERT (Ip6Addr != NULL && MulticastAddr != NULL);\r
\r
  ZeroMem (MulticastAddr, sizeof (EFI_IPv6_ADDRESS));\r
\r
  MulticastAddr->Addr[0]  = 0xFF;\r
  MulticastAddr->Addr[1]  = 0x02;\r
  MulticastAddr->Addr[11] = 0x1;\r
  MulticastAddr->Addr[12] = 0xFF;\r
\r
  CopyMem (&MulticastAddr->Addr[13], &Ip6Addr->Addr[13], 3);\r
}\r
\r
/**\r
  Insert a node IP6_ADDRESS_INFO to an IP6 interface.\r
\r
  @param[in, out]  IpIf             Points to an IP6 interface.\r
  @param[in]       AddrInfo         Points to IP6_ADDRESS_INFO\r
\r
**/\r
VOID\r
Ip6AddAddr (\r
  IN OUT IP6_INTERFACE *IpIf,\r
  IN IP6_ADDRESS_INFO  *AddrInfo\r
  )\r
{\r
  InsertHeadList (&IpIf->AddressList, &AddrInfo->Link);\r
  IpIf->AddressCount++;\r
}\r
\r
/**\r
  Callback function which provided by user to remove one node in NetDestroyLinkList process.\r
  \r
  @param[in]    Entry           The entry to be removed.\r
  @param[in]    Context         Pointer to the callback context corresponds to the Context in NetDestroyLinkList.\r
\r
  @retval EFI_SUCCESS           The entry has been removed successfully.\r
  @retval Others                Fail to remove the entry.\r
\r
**/\r
EFI_STATUS\r
Ip6DestroyChildEntryByAddr (\r
  IN LIST_ENTRY         *Entry,\r
  IN VOID               *Context\r
)\r
{\r
  IP6_PROTOCOL                  *Instance;\r
  EFI_SERVICE_BINDING_PROTOCOL  *ServiceBinding;\r
  EFI_IPv6_ADDRESS              *Address;\r
  \r
  Instance = NET_LIST_USER_STRUCT_S (Entry, IP6_PROTOCOL, Link, IP6_PROTOCOL_SIGNATURE);\r
  ServiceBinding = ((IP6_DESTROY_CHILD_BY_ADDR_CALLBACK_CONTEXT*) Context)->ServiceBinding;\r
  Address = ((IP6_DESTROY_CHILD_BY_ADDR_CALLBACK_CONTEXT*) Context)->Address;\r
\r
  if ((Instance->State == IP6_STATE_CONFIGED) && EFI_IP6_EQUAL (&Instance->ConfigData.StationAddress, Address)) {\r
    return ServiceBinding->DestroyChild (ServiceBinding, Instance->Handle);\r
  }\r
  \r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Destroy the IP instance if its StationAddress is removed. It is the help function\r
  for Ip6RemoveAddr().\r
\r
  @param[in, out]  IpSb             Points to an IP6 service binding instance.\r
  @param[in]       Address          The to be removed address\r
\r
**/\r
VOID\r
Ip6DestroyInstanceByAddress (\r
  IN OUT IP6_SERVICE   *IpSb,\r
  IN EFI_IPv6_ADDRESS  *Address\r
  )\r
{\r
  LIST_ENTRY                    *List;\r
  IP6_DESTROY_CHILD_BY_ADDR_CALLBACK_CONTEXT  Context;\r
\r
  NET_CHECK_SIGNATURE (IpSb, IP6_SERVICE_SIGNATURE);\r
\r
  List = &IpSb->Children;\r
  Context.ServiceBinding = &IpSb->ServiceBinding;\r
  Context.Address = Address;\r
  NetDestroyLinkList (\r
    List,\r
    Ip6DestroyChildEntryByAddr,\r
    &Context,\r
    NULL\r
    );\r
}\r
\r
/**\r
  Remove the IPv6 address from the address list node points to IP6_ADDRESS_INFO.\r
\r
  This function removes the matching IPv6 addresses from the address list and\r
  adjusts the address count of the address list. If IpSb is not NULL, this function\r
  calls Ip6LeaveGroup to see whether it should call Mnp->Groups() to remove the\r
  its solicited-node multicast MAC address from the filter list and sends out\r
  a Multicast Listener Done. If Prefix is NULL, all address in the address list\r
  will be removed. If Prefix is not NULL, the address that matching the Prefix\r
  with PrefixLength in the address list will be removed.\r
\r
  @param[in]       IpSb             NULL or points to IP6 service binding instance.\r
  @param[in, out]  AddressList      Address list array.\r
  @param[in, out]  AddressCount     The count of addresses in address list array.\r
  @param[in]       Prefix           NULL or an IPv6 address prefix.\r
  @param[in]       PrefixLength     The length of Prefix.\r
\r
  @retval    EFI_SUCCESS            The operation completed successfully.\r
  @retval    EFI_NOT_FOUND          The address matching the Prefix with PrefixLength\r
                                    cannot be found in the address list.\r
  @retval    EFI_INVALID_PARAMETER  Any input parameter is invalid.\r
\r
**/\r
EFI_STATUS\r
Ip6RemoveAddr (\r
  IN IP6_SERVICE       *IpSb          OPTIONAL,\r
  IN OUT LIST_ENTRY    *AddressList,\r
  IN OUT UINT32        *AddressCount,\r
  IN EFI_IPv6_ADDRESS  *Prefix        OPTIONAL,\r
  IN UINT8             PrefixLength\r
  )\r
{\r
  EFI_STATUS           Status;\r
  LIST_ENTRY           *Entry;\r
  LIST_ENTRY           *Next;\r
  IP6_ADDRESS_INFO     *AddrInfo;\r
  EFI_IPv6_ADDRESS     SnMCastAddr;\r
\r
  if (IsListEmpty (AddressList) || *AddressCount < 1 || PrefixLength > IP6_PREFIX_NUM) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status = EFI_NOT_FOUND;\r
\r
  NET_LIST_FOR_EACH_SAFE (Entry, Next, AddressList) {\r
    AddrInfo = NET_LIST_USER_STRUCT_S (Entry, IP6_ADDRESS_INFO, Link, IP6_ADDR_INFO_SIGNATURE);\r
\r
    if (Prefix == NULL ||\r
        (PrefixLength == 128 && EFI_IP6_EQUAL (Prefix, &AddrInfo->Address)) ||\r
        (PrefixLength == AddrInfo->PrefixLength && NetIp6IsNetEqual (Prefix, &AddrInfo->Address, PrefixLength))\r
        ) {\r
      if (IpSb != NULL) {\r
        NET_CHECK_SIGNATURE (IpSb, IP6_SERVICE_SIGNATURE);\r
        Ip6CreateSNMulticastAddr (&AddrInfo->Address, &SnMCastAddr);\r
        Ip6LeaveGroup (IpSb, &SnMCastAddr);\r
\r
        //\r
        // Destroy any instance who is using the dying address as the source address.\r
        //\r
        Ip6DestroyInstanceByAddress (IpSb, &AddrInfo->Address);\r
      }\r
\r
      RemoveEntryList (Entry);\r
      FreePool (AddrInfo);\r
      (*AddressCount)--;\r
\r
      Status = EFI_SUCCESS;\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  Check whether the incoming Ipv6 address is a solicited-node multicast address.\r
\r
  @param[in]  Ip6               Ip6 address, in network order.\r
\r
  @retval TRUE                  Yes, solicited-node multicast address\r
  @retval FALSE                 No\r
\r
**/\r
BOOLEAN\r
Ip6IsSNMulticastAddr (\r
  IN EFI_IPv6_ADDRESS *Ip6\r
  )\r
{\r
  EFI_IPv6_ADDRESS    Sn;\r
  BOOLEAN             Flag;\r
\r
  Ip6CreateSNMulticastAddr (Ip6, &Sn);\r
  Flag = FALSE;\r
\r
  if (CompareMem (Sn.Addr, Ip6->Addr, 13) == 0) {\r
    Flag = TRUE;\r
  }\r
\r
  return Flag;\r
}\r
\r
/**\r
  Check whether the incoming IPv6 address is one of the maintained addresses in\r
  the IP6 service binding instance.\r
\r
  @param[in]  IpSb              Points to a IP6 service binding instance.\r
  @param[in]  Address           The IP6 address to be checked.\r
  @param[out] Interface         If not NULL, output the IP6 interface which\r
                                maintains the Address.\r
  @param[out] AddressInfo       If not NULL, output the IP6 address information\r
                                of the Address.\r
\r
  @retval TRUE                  Yes, it is one of the maintained address.\r
  @retval FALSE                 No, it is not one of the maintained address.\r
\r
**/\r
BOOLEAN\r
Ip6IsOneOfSetAddress (\r
  IN  IP6_SERVICE           *IpSb,\r
  IN  EFI_IPv6_ADDRESS      *Address,\r
  OUT IP6_INTERFACE         **Interface   OPTIONAL,\r
  OUT IP6_ADDRESS_INFO      **AddressInfo OPTIONAL\r
  )\r
{\r
  LIST_ENTRY                *Entry;\r
  LIST_ENTRY                *Entry2;\r
  IP6_INTERFACE             *IpIf;\r
  IP6_ADDRESS_INFO          *TmpAddressInfo;\r
\r
  //\r
  // Check link-local address first\r
  //\r
  if (IpSb->LinkLocalOk && EFI_IP6_EQUAL (&IpSb->LinkLocalAddr, Address)) {\r
    if (Interface != NULL) {\r
      *Interface = IpSb->DefaultInterface;\r
    }\r
\r
    if (AddressInfo != NULL) {\r
      *AddressInfo = NULL;\r
    }\r
\r
    return TRUE;\r
  }\r
\r
  NET_LIST_FOR_EACH (Entry, &IpSb->Interfaces) {\r
    IpIf = NET_LIST_USER_STRUCT_S (Entry, IP6_INTERFACE, Link, IP6_INTERFACE_SIGNATURE);\r
\r
    NET_LIST_FOR_EACH (Entry2, &IpIf->AddressList) {\r
      TmpAddressInfo = NET_LIST_USER_STRUCT_S (Entry2, IP6_ADDRESS_INFO, Link, IP6_ADDR_INFO_SIGNATURE);\r
\r
      if (EFI_IP6_EQUAL (&TmpAddressInfo->Address, Address)) {\r
        if (Interface != NULL) {\r
          *Interface = IpIf;\r
        }\r
\r
        if (AddressInfo != NULL) {\r
          *AddressInfo = TmpAddressInfo;\r
        }\r
\r
        return TRUE;\r
      }\r
    }\r
  }\r
\r
  return FALSE;\r
}\r
\r
/**\r
  Check whether the incoming MAC address is valid.\r
\r
  @param[in]  IpSb              Points to a IP6 service binding instance.\r
  @param[in]  LinkAddress       The MAC address.\r
\r
  @retval TRUE                  Yes, it is valid.\r
  @retval FALSE                 No, it is not valid.\r
\r
**/\r
BOOLEAN\r
Ip6IsValidLinkAddress (\r
  IN  IP6_SERVICE      *IpSb,\r
  IN  EFI_MAC_ADDRESS  *LinkAddress\r
  )\r
{\r
  UINT32               Index;\r
\r
  //\r
  // TODO: might be updated later to be more acceptable.\r
  //\r
  for (Index = IpSb->SnpMode.HwAddressSize; Index < sizeof (EFI_MAC_ADDRESS); Index++) {\r
    if (LinkAddress->Addr[Index] != 0) {\r
      return FALSE;\r
    }\r
  }\r
\r
  return TRUE;\r
}\r
\r
/**\r
  Copy the PrefixLength bits from Src to Dest.\r
\r
  @param[out] Dest              A pointer to the buffer to copy to.\r
  @param[in]  Src               A pointer to the buffer to copy from.\r
  @param[in]  PrefixLength      The number of bits to copy.\r
\r
**/\r
VOID\r
Ip6CopyAddressByPrefix (\r
  OUT EFI_IPv6_ADDRESS *Dest,\r
  IN  EFI_IPv6_ADDRESS *Src,\r
  IN  UINT8            PrefixLength\r
  )\r
{\r
  UINT8 Byte;\r
  UINT8 Bit;\r
  UINT8 Mask;\r
\r
  ASSERT (Dest != NULL && Src != NULL);\r
  ASSERT (PrefixLength < IP6_PREFIX_NUM);\r
\r
  Byte = (UINT8) (PrefixLength / 8);\r
  Bit  = (UINT8) (PrefixLength % 8);\r
\r
  ZeroMem (Dest, sizeof (EFI_IPv6_ADDRESS));\r
\r
  CopyMem (Dest, Src, Byte);\r
\r
  if (Bit > 0) {\r
    Mask = (UINT8) (0xFF << (8 - Bit));\r
    ASSERT (Byte < 16);\r
    Dest->Addr[Byte] = (UINT8) (Src->Addr[Byte] & Mask);\r
  }\r
}\r
\r
/**\r
  Get the MAC address for a multicast IP address. Call\r
  Mnp's McastIpToMac to find the MAC address instead of\r
  hard-coding the NIC to be Ethernet.\r
\r
  @param[in]  Mnp                   The Mnp instance to get the MAC address.\r
  @param[in]  Multicast             The multicast IP address to translate.\r
  @param[out] Mac                   The buffer to hold the translated address.\r
\r
  @retval EFI_SUCCESS               The multicast IP successfully\r
                                    translated to a multicast MAC address.\r
  @retval Other                     The address is not converted because an error occurred.\r
\r
**/\r
EFI_STATUS\r
Ip6GetMulticastMac (\r
  IN  EFI_MANAGED_NETWORK_PROTOCOL *Mnp,\r
  IN  EFI_IPv6_ADDRESS             *Multicast,\r
  OUT EFI_MAC_ADDRESS              *Mac\r
  )\r
{\r
  EFI_IP_ADDRESS        EfiIp;\r
\r
  IP6_COPY_ADDRESS (&EfiIp.v6, Multicast);\r
\r
  return Mnp->McastIpToMac (Mnp, TRUE, &EfiIp, Mac);\r
}\r
\r
/**\r
  Set the Ip6 variable data.\r
\r
  @param[in]  IpSb              Points to an IP6 service binding instance.\r
\r
  @retval EFI_OUT_OF_RESOURCES  There are not enough resources to set the variable.\r
  @retval other                 Set variable failed.\r
\r
**/\r
EFI_STATUS\r
Ip6SetVariableData (\r
  IN IP6_SERVICE  *IpSb\r
  )\r
{\r
  UINT32                 NumConfiguredInstance;\r
  LIST_ENTRY             *Entry;\r
  UINTN                  VariableDataSize;\r
  EFI_IP6_VARIABLE_DATA  *Ip6VariableData;\r
  EFI_IP6_ADDRESS_PAIR   *Ip6AddressPair;\r
  IP6_PROTOCOL           *IpInstance;\r
  CHAR16                 *NewMacString;\r
  EFI_STATUS             Status;\r
\r
  NumConfiguredInstance = 0;\r
\r
  //\r
  // Go through the children list to count the configured children.\r
  //\r
  NET_LIST_FOR_EACH (Entry, &IpSb->Children) {\r
    IpInstance = NET_LIST_USER_STRUCT_S (Entry, IP6_PROTOCOL, Link, IP6_PROTOCOL_SIGNATURE);\r
\r
    if (IpInstance->State == IP6_STATE_CONFIGED) {\r
      NumConfiguredInstance++;\r
    }\r
  }\r
\r
  //\r
  // Calculate the size of the Ip6VariableData. As there may be no IP child,\r
  // we should add extra buffer for the address paris only if the number of configured\r
  // children is more than 1.\r
  //\r
  VariableDataSize = sizeof (EFI_IP6_VARIABLE_DATA);\r
\r
  if (NumConfiguredInstance > 1) {\r
    VariableDataSize += sizeof (EFI_IP6_ADDRESS_PAIR) * (NumConfiguredInstance - 1);\r
  }\r
\r
  Ip6VariableData = AllocatePool (VariableDataSize);\r
  if (Ip6VariableData == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  Ip6VariableData->DriverHandle = IpSb->Image;\r
  Ip6VariableData->AddressCount = NumConfiguredInstance;\r
\r
  Ip6AddressPair                = &Ip6VariableData->AddressPairs[0];\r
\r
  //\r
  // Go through the children list to fill the configured children's address pairs.\r
  //\r
  NET_LIST_FOR_EACH (Entry, &IpSb->Children) {\r
    IpInstance = NET_LIST_USER_STRUCT_S (Entry, IP6_PROTOCOL, Link, IP6_PROTOCOL_SIGNATURE);\r
\r
    if (IpInstance->State == IP6_STATE_CONFIGED) {\r
      Ip6AddressPair->InstanceHandle = IpInstance->Handle;\r
      Ip6AddressPair->PrefixLength   = IpInstance->PrefixLength;\r
      IP6_COPY_ADDRESS (&Ip6AddressPair->Ip6Address, &IpInstance->ConfigData.StationAddress);\r
\r
      Ip6AddressPair++;\r
    }\r
  }\r
\r
  //\r
  // Get the mac string.\r
  //\r
  Status = NetLibGetMacString (IpSb->Controller, IpSb->Image, &NewMacString);\r
  if (EFI_ERROR (Status)) {\r
    goto Exit;\r
  }\r
\r
  if (IpSb->MacString != NULL) {\r
    //\r
    // The variable is set already, we're going to update it.\r
    //\r
    if (StrCmp (IpSb->MacString, NewMacString) != 0) {\r
      //\r
      // The mac address is changed, delete the previous variable first.\r
      //\r
      gRT->SetVariable (\r
             IpSb->MacString,\r
             &gEfiIp6ServiceBindingProtocolGuid,\r
             EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
             0,\r
             NULL\r
             );\r
    }\r
\r
    FreePool (IpSb->MacString);\r
  }\r
\r
  IpSb->MacString = NewMacString;\r
\r
  Status = gRT->SetVariable (\r
                  IpSb->MacString,\r
                  &gEfiIp6ServiceBindingProtocolGuid,\r
                  EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
                  VariableDataSize,\r
                  (VOID *) Ip6VariableData\r
                  );\r
\r
Exit:\r
  FreePool (Ip6VariableData);\r
  return Status;\r
}\r
\r
/**\r
  Clear the variable and free the resource.\r
\r
  @param[in]  IpSb                  Ip6 service binding instance.\r
\r
**/\r
VOID\r
Ip6ClearVariableData (\r
  IN IP6_SERVICE  *IpSb\r
  )\r
{\r
  ASSERT (IpSb->MacString != NULL);\r
\r
  gRT->SetVariable (\r
         IpSb->MacString,\r
         &gEfiIp6ServiceBindingProtocolGuid,\r
         EFI_VARIABLE_BOOTSERVICE_ACCESS,\r
         0,\r
         NULL\r
         );\r
\r
  FreePool (IpSb->MacString);\r
  IpSb->MacString = NULL;\r
}\r
\r
/**\r
  Convert the multibyte field in IP header's byter order.\r
  In spite of its name, it can also be used to convert from\r
  host to network byte order.\r
\r
  @param[in, out]  Head                  The IP head to convert.\r
\r
  @return Point to the converted IP head.\r
\r
**/\r
EFI_IP6_HEADER *\r
Ip6NtohHead (\r
  IN OUT EFI_IP6_HEADER *Head\r
  )\r
{\r
  Head->FlowLabelL    = NTOHS (Head->FlowLabelL);\r
  Head->PayloadLength = NTOHS (Head->PayloadLength);\r
\r
  return Head;\r
}\r
\r