]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcImpl.c
1.Fix a bug in Dhcp4Dxe driver to correct the ‘secs’ field in DHCP message.
[mirror_edk2.git] / MdeModulePkg / Universal / Network / UefiPxeBcDxe / PxeBcImpl.c
index 989da79b850f042341631f7eb7d4ce7efcd5546c..6f6d7f92ae58a797e1a181f2357b4190f9a60782 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
   Interface routines for PxeBc.\r
 \r
-Copyright (c) 2007 - 2010, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2007 - 2011, 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
@@ -561,15 +561,16 @@ EfiPxeBcDhcp (
   IN BOOLEAN                          SortOffers\r
   )\r
 {\r
-  PXEBC_PRIVATE_DATA      *Private;\r
-  EFI_PXE_BASE_CODE_MODE  *Mode;\r
-  EFI_DHCP4_PROTOCOL      *Dhcp4;\r
-  EFI_DHCP4_CONFIG_DATA   Dhcp4CfgData;\r
-  EFI_DHCP4_MODE_DATA     Dhcp4Mode;\r
-  EFI_DHCP4_PACKET_OPTION *OptList[PXEBC_DHCP4_MAX_OPTION_NUM];\r
-  UINT32                  OptCount;\r
-  EFI_STATUS              Status;\r
-  EFI_ARP_CONFIG_DATA     ArpConfigData;\r
+  PXEBC_PRIVATE_DATA           *Private;\r
+  EFI_PXE_BASE_CODE_MODE       *Mode;\r
+  EFI_DHCP4_PROTOCOL           *Dhcp4;\r
+  EFI_DHCP4_CONFIG_DATA        Dhcp4CfgData;\r
+  EFI_DHCP4_MODE_DATA          Dhcp4Mode;\r
+  EFI_DHCP4_PACKET_OPTION      *OptList[PXEBC_DHCP4_MAX_OPTION_NUM];\r
+  UINT32                       OptCount;\r
+  EFI_STATUS                   Status;\r
+  EFI_ARP_CONFIG_DATA          ArpConfigData;\r
+  EFI_PXE_BASE_CODE_IP_FILTER  IpFilter;\r
 \r
   if (This == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -588,6 +589,11 @@ EfiPxeBcDhcp (
 \r
   Mode->IcmpErrorReceived = FALSE;\r
 \r
+  //\r
+  // Stop Udp4Read instance\r
+  //\r
+  Private->Udp4Read->Configure (Private->Udp4Read, NULL);\r
+\r
   //\r
   // Initialize the DHCP options and build the option list\r
   //\r
@@ -671,7 +677,7 @@ ON_EXIT:
       ZeroMem (&ArpConfigData, sizeof (EFI_ARP_CONFIG_DATA));\r
 \r
       ArpConfigData.SwAddressType   = 0x0800;\r
-      ArpConfigData.SwAddressLength = sizeof (EFI_IPv4_ADDRESS);\r
+      ArpConfigData.SwAddressLength = (UINT8) sizeof (EFI_IPv4_ADDRESS);\r
       ArpConfigData.StationAddress  = &Private->StationIp.v4;\r
 \r
       Private->Arp->Configure (Private->Arp, NULL);\r
@@ -694,9 +700,43 @@ ON_EXIT:
         Mode->RouteTable[1].SubnetMask.Addr[0] = 0;\r
         Mode->RouteTable[1].GwAddr.Addr[0]     = Private->GatewayIp.Addr[0];\r
       }\r
+\r
+      //\r
+      // Flush new station IP address into Udp4CfgData and Ip4ConfigData\r
+      //\r
+      CopyMem (&Private->Udp4CfgData.StationAddress, &Private->StationIp, sizeof (EFI_IPv4_ADDRESS));\r
+      CopyMem (&Private->Udp4CfgData.SubnetMask, &Private->SubnetMask, sizeof (EFI_IPv4_ADDRESS));\r
+      CopyMem (&Private->Ip4ConfigData.StationAddress, &Private->StationIp, sizeof (EFI_IPv4_ADDRESS));\r
+      CopyMem (&Private->Ip4ConfigData.SubnetMask, &Private->SubnetMask, sizeof (EFI_IPv4_ADDRESS));\r
+  \r
+      //\r
+      // Reconfigure the Ip4 instance to capture background ICMP packets with new station Ip address.\r
+      //\r
+      Private->Ip4->Cancel (Private->Ip4, &Private->IcmpErrorRcvToken);\r
+      Private->Ip4->Configure (Private->Ip4, NULL);\r
+  \r
+      Status = Private->Ip4->Configure (Private->Ip4, &Private->Ip4ConfigData);\r
+      if (EFI_ERROR (Status)) {\r
+        goto ON_EXIT;\r
+      }\r
+  \r
+      Status = Private->Ip4->Receive (Private->Ip4, &Private->IcmpErrorRcvToken);\r
+      if (EFI_ERROR (Status)) {\r
+        goto ON_EXIT;\r
+      } \r
     }\r
   }\r
 \r
+  Status = Private->Udp4Read->Configure (Private->Udp4Read, &Private->Udp4CfgData);\r
+\r
+  //\r
+  // Dhcp(), Discover(), and Mtftp() set the IP filter, and return with the IP \r
+  // receive filter list emptied and the filter set to EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP.\r
+  //\r
+  ZeroMem(&IpFilter, sizeof (EFI_PXE_BASE_CODE_IP_FILTER));\r
+  IpFilter.Filters = EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP;\r
+  This->SetIpFilter (This, &IpFilter);\r
+  \r
   return Status;\r
 }\r
 \r
@@ -765,6 +805,7 @@ EfiPxeBcDiscover (
   UINT16                          Index;\r
   EFI_STATUS                      Status;\r
   PXEBC_BOOT_SVR_ENTRY            *BootSvrEntry;\r
+  EFI_PXE_BASE_CODE_IP_FILTER     IpFilter;\r
 \r
   if (This == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -786,6 +827,11 @@ EfiPxeBcDiscover (
     return EFI_NOT_STARTED;\r
   }\r
 \r
+  //\r
+  // Stop Udp4Read instance\r
+  //\r
+  Private->Udp4Read->Configure (Private->Udp4Read, NULL);\r
+\r
   Mode->IcmpErrorReceived = FALSE;\r
 \r
   //\r
@@ -798,7 +844,8 @@ EfiPxeBcDiscover (
 \r
     if (!Mode->PxeDiscoverValid || !Mode->PxeReplyReceived || (!Mode->PxeBisReplyReceived && UseBis)) {\r
 \r
-      return EFI_INVALID_PARAMETER;\r
+      Status = EFI_INVALID_PARAMETER;\r
+      goto ON_EXIT;  \r
     }\r
 \r
     DefaultInfo.IpCnt                 = 1;\r
@@ -821,7 +868,8 @@ EfiPxeBcDiscover (
       //\r
       // Address is not acquired or no discovery options.\r
       //\r
-      return EFI_INVALID_PARAMETER;\r
+      Status = EFI_INVALID_PARAMETER;\r
+      goto ON_EXIT;  \r
     }\r
 \r
     DefaultInfo.UseMCast    = (BOOLEAN)!IS_DISABLE_MCAST_DISCOVER (VendorOpt->DiscoverCtrl);\r
@@ -859,7 +907,7 @@ EfiPxeBcDiscover (
       }\r
 \r
       if (EFI_ERROR (Status)) {\r
-        return Status;\r
+        goto ON_EXIT;\r
       }\r
 \r
       DefaultInfo.IpCnt = BootSvrEntry->IpCnt;\r
@@ -867,7 +915,9 @@ EfiPxeBcDiscover (
       if (DefaultInfo.IpCnt >= 1) {\r
         CreatedInfo = AllocatePool (sizeof (DefaultInfo) + (DefaultInfo.IpCnt - 1) * sizeof (*SrvList));\r
         if (CreatedInfo == NULL) {\r
-          return EFI_OUT_OF_RESOURCES;\r
+          Status = EFI_OUT_OF_RESOURCES;\r
+          goto ON_EXIT;\r
+          \r
         }     \r
       \r
         CopyMem (CreatedInfo, &DefaultInfo, sizeof (DefaultInfo));\r
@@ -895,14 +945,16 @@ EfiPxeBcDiscover (
       }\r
 \r
       if (Index != Info->IpCnt) {\r
-        return EFI_INVALID_PARAMETER;\r
+        Status = EFI_INVALID_PARAMETER;\r
+        goto ON_EXIT;        \r
       }\r
     }\r
   }\r
 \r
   if ((!Info->UseUCast && !Info->UseBCast && !Info->UseMCast) || (Info->MustUseList && Info->IpCnt == 0)) {\r
 \r
-    return EFI_INVALID_PARAMETER;\r
+    Status = EFI_INVALID_PARAMETER;\r
+    goto ON_EXIT;\r
   }\r
   //\r
   // Execute discover by UniCast/BroadCast/MultiCast\r
@@ -972,6 +1024,7 @@ EfiPxeBcDiscover (
     } else {\r
       Status = EFI_DEVICE_ERROR;\r
     }\r
+    return Status;\r
   } else {\r
     PxeBcParseCachedDhcpPacket (&Private->PxeReply);\r
   }\r
@@ -987,6 +1040,18 @@ EfiPxeBcDiscover (
   if (CreatedInfo != NULL) {\r
     FreePool (CreatedInfo);\r
   }\r
+\r
+ON_EXIT:\r
+\r
+  Status = Private->Udp4Read->Configure (Private->Udp4Read, &Private->Udp4CfgData);\r
+  \r
+  //\r
+  // Dhcp(), Discover(), and Mtftp() set the IP filter, and return with the IP \r
+  // receive filter list emptied and the filter set to EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP.\r
+  //\r
+  ZeroMem(&IpFilter, sizeof (EFI_PXE_BASE_CODE_IP_FILTER));\r
+  IpFilter.Filters = EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP;\r
+  This->SetIpFilter (This, &IpFilter);\r
   \r
   return Status;\r
 }\r
@@ -1078,11 +1143,12 @@ EfiPxeBcMtftp (
   IN BOOLEAN                          DontUseBuffer\r
   )\r
 {\r
-  PXEBC_PRIVATE_DATA      *Private;\r
-  EFI_MTFTP4_CONFIG_DATA  Mtftp4Config;\r
-  EFI_STATUS              Status;\r
-  EFI_PXE_BASE_CODE_MODE  *Mode;\r
-  EFI_MAC_ADDRESS         TempMacAddr;\r
+  PXEBC_PRIVATE_DATA           *Private;\r
+  EFI_MTFTP4_CONFIG_DATA       Mtftp4Config;\r
+  EFI_STATUS                   Status;\r
+  EFI_PXE_BASE_CODE_MODE       *Mode;\r
+  EFI_MAC_ADDRESS              TempMacAddr;\r
+  EFI_PXE_BASE_CODE_IP_FILTER  IpFilter;\r
 \r
   if ((This == NULL)                                                          ||\r
       (Filename == NULL)                                                      ||\r
@@ -1108,6 +1174,11 @@ EfiPxeBcMtftp (
     }\r
   }\r
 \r
+  //\r
+  // Stop Udp4Read instance\r
+  //\r
+  Private->Udp4Read->Configure (Private->Udp4Read, NULL);\r
+\r
   Mode->TftpErrorReceived = FALSE;\r
   Mode->IcmpErrorReceived = FALSE;\r
 \r
@@ -1208,6 +1279,19 @@ EfiPxeBcMtftp (
     Mode->IcmpErrorReceived = TRUE;\r
   }\r
 \r
+  if (EFI_ERROR (Status)) {\r
+    return Status;\r
+  }\r
+\r
+  Status = Private->Udp4Read->Configure (Private->Udp4Read, &Private->Udp4CfgData);\r
+  //\r
+  // Dhcp(), Discover(), and Mtftp() set the IP filter, and return with the IP \r
+  // receive filter list emptied and the filter set to EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP.\r
+  //\r
+  ZeroMem(&IpFilter, sizeof (EFI_PXE_BASE_CODE_IP_FILTER));\r
+  IpFilter.Filters = EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP;\r
+  This->SetIpFilter (This, &IpFilter);\r
+\r
   return Status;\r
 }\r
 \r
@@ -1333,20 +1417,20 @@ EfiPxeBcUdpWrite (
     if (SrcPort != NULL) {\r
       Private->CurrentUdpSrcPort = *SrcPort;\r
     }\r
-\r
-    Status = PxeBcConfigureUdpWriteInstance (\r
-               Udp4,\r
-               &Private->StationIp.v4,\r
-               &Private->SubnetMask.v4,\r
-               &Private->GatewayIp.v4,\r
-               &Private->CurrentUdpSrcPort\r
-               );\r
-    if (EFI_ERROR (Status)) {\r
-      Private->CurrentUdpSrcPort = 0;\r
-      return EFI_INVALID_PARAMETER;\r
-    }\r
   }\r
 \r
+  Status = PxeBcConfigureUdpWriteInstance (\r
+             Udp4,\r
+             &Private->StationIp.v4,\r
+             &Private->SubnetMask.v4,\r
+             &Private->GatewayIp.v4,\r
+             &Private->CurrentUdpSrcPort\r
+             );\r
+  if (EFI_ERROR (Status)) {\r
+    Private->CurrentUdpSrcPort = 0;\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+  \r
   ZeroMem (&Token, sizeof (EFI_UDP4_COMPLETION_TOKEN));\r
   ZeroMem (&Udp4Session, sizeof (EFI_UDP4_SESSION_DATA));\r
 \r
@@ -1418,6 +1502,10 @@ ON_EXIT:
 \r
   FreePool (Udp4TxData);\r
 \r
+  //\r
+  // Reset the instance.\r
+  //\r
+  Udp4->Configure (Udp4, NULL);\r
   return Status;\r
 }\r
 \r
@@ -1543,7 +1631,13 @@ EfiPxeBcUdpRead (
   EFI_STATUS                Status;\r
   BOOLEAN                   IsDone;\r
   BOOLEAN                   Matched;\r
-  UINTN                     CopyLen;\r
+  UINTN                     CopiedLen;\r
+  UINTN                     HeaderLen;\r
+  UINTN                     HeaderCopiedLen;\r
+  UINTN                     BufferCopiedLen;\r
+  UINT32                    FragmentLength;\r
+  UINTN                     FragmentIndex;\r
+  UINT8                     *FragmentBuffer;\r
 \r
   if (This == NULL || DestIp == NULL || DestPort == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -1700,26 +1794,51 @@ TRY_AGAIN:
     }\r
 \r
     if (Matched) {\r
+      ASSERT (RxData != NULL);\r
 \r
-      CopyLen = 0;\r
-\r
+      HeaderLen = 0;\r
       if (HeaderSize != NULL) {\r
-        CopyLen = MIN (*HeaderSize, RxData->DataLength);\r
-        CopyMem (HeaderPtr, RxData->FragmentTable[0].FragmentBuffer, CopyLen);\r
-        *HeaderSize = CopyLen;\r
+        HeaderLen = MIN (*HeaderSize, RxData->DataLength);\r
       }\r
 \r
-      if (RxData->DataLength - CopyLen > *BufferSize) {\r
-\r
+      if (RxData->DataLength - HeaderLen > *BufferSize) {\r
         Status = EFI_BUFFER_TOO_SMALL;\r
       } else {\r
-\r
-        *BufferSize = RxData->DataLength - CopyLen;\r
-        CopyMem (\r
-          BufferPtr,\r
-          (UINT8 *) RxData->FragmentTable[0].FragmentBuffer + CopyLen,\r
-          *BufferSize\r
-          );\r
+        *HeaderSize = HeaderLen;\r
+        *BufferSize = RxData->DataLength - HeaderLen;\r
+\r
+        HeaderCopiedLen = 0;\r
+        BufferCopiedLen = 0;\r
+        for (FragmentIndex = 0; FragmentIndex < RxData->FragmentCount; FragmentIndex++) {\r
+          FragmentLength = RxData->FragmentTable[FragmentIndex].FragmentLength;\r
+          FragmentBuffer = RxData->FragmentTable[FragmentIndex].FragmentBuffer;\r
+          if (HeaderCopiedLen + FragmentLength < HeaderLen) {\r
+            //\r
+            // Copy the header part of received data.\r
+            //\r
+            CopyMem ((UINT8 *) HeaderPtr + HeaderCopiedLen, FragmentBuffer, FragmentLength);\r
+            HeaderCopiedLen += FragmentLength;\r
+          } else if (HeaderCopiedLen < HeaderLen) {\r
+            //\r
+            // Copy the header part of received data.\r
+            //\r
+            CopiedLen = HeaderLen - HeaderCopiedLen;\r
+            CopyMem ((UINT8 *) HeaderPtr + HeaderCopiedLen, FragmentBuffer, CopiedLen);\r
+            HeaderCopiedLen += CopiedLen;\r
+\r
+            //\r
+            // Copy the other part of received data.\r
+            //\r
+            CopyMem ((UINT8 *) BufferPtr + BufferCopiedLen, FragmentBuffer + CopiedLen, FragmentLength - CopiedLen);\r
+            BufferCopiedLen += (FragmentLength - CopiedLen);\r
+          } else {\r
+            //\r
+            // Copy the other part of received data.\r
+            //\r
+            CopyMem ((UINT8 *) BufferPtr + BufferCopiedLen, FragmentBuffer, FragmentLength);\r
+            BufferCopiedLen += FragmentLength;\r
+          }\r
+        }\r
       }\r
     } else {\r
 \r
@@ -1788,7 +1907,11 @@ EfiPxeBcSetIpFilter (
   PXEBC_PRIVATE_DATA        *Private;\r
   EFI_PXE_BASE_CODE_MODE    *Mode;\r
   UINTN                     Index;\r
+  EFI_UDP4_CONFIG_DATA      *Udp4Cfg;\r
   BOOLEAN                   PromiscuousNeed;\r
+  BOOLEAN                   AcceptPromiscuous;\r
+  BOOLEAN                   AcceptBroadcast;\r
+  BOOLEAN                   MultiCastUpdate;\r
 \r
   if (This == NULL) {\r
     DEBUG ((EFI_D_ERROR, "This == NULL.\n"));\r
@@ -1813,6 +1936,11 @@ EfiPxeBcSetIpFilter (
     return EFI_NOT_STARTED;\r
   }\r
 \r
+  if (Mode->UsingIpv6) {\r
+    DEBUG ((EFI_D_ERROR, "This driver is PXE for IPv4 Only.\n"));\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
   PromiscuousNeed = FALSE;\r
 \r
   for (Index = 0; Index < NewFilter->IpCnt; ++Index) {\r
@@ -1834,54 +1962,71 @@ EfiPxeBcSetIpFilter (
     }\r
   }\r
 \r
-  //\r
-  // Clear the UDP instance configuration, all joined groups will be left\r
-  // during the operation.\r
-  //\r
-  Private->Udp4Read->Configure (Private->Udp4Read, NULL);\r
-  Private->Udp4CfgData.AcceptPromiscuous  = FALSE;\r
-  Private->Udp4CfgData.AcceptBroadcast    = FALSE;\r
+  AcceptPromiscuous = FALSE;\r
+  AcceptBroadcast   = FALSE;\r
+  MultiCastUpdate   = FALSE;\r
 \r
   if (PromiscuousNeed ||\r
       ((NewFilter->Filters & EFI_PXE_BASE_CODE_IP_FILTER_PROMISCUOUS) != 0) ||\r
       ((NewFilter->Filters & EFI_PXE_BASE_CODE_IP_FILTER_PROMISCUOUS_MULTICAST) != 0)\r
      ) {\r
     //\r
-    // Configure the udp4 filter to receive all packages\r
+    // Configure the udp4 filter to receive all packages.\r
     //\r
-    Private->Udp4CfgData.AcceptPromiscuous  = TRUE;\r
-\r
+    AcceptPromiscuous  = TRUE;\r
+  } else if ((NewFilter->Filters & EFI_PXE_BASE_CODE_IP_FILTER_BROADCAST) != 0) {\r
     //\r
-    // Configure the UDP instance with the new configuration.\r
+    // Configure the udp4 filter to receive all broadcast packages.\r
     //\r
-    Status = Private->Udp4Read->Configure (Private->Udp4Read, &Private->Udp4CfgData);\r
-    if (EFI_ERROR (Status)) {\r
-      return Status;\r
-    }\r
-\r
-  } else {\r
+    AcceptBroadcast   = TRUE;\r
+  }\r
 \r
-    if ((NewFilter->Filters & EFI_PXE_BASE_CODE_IP_FILTER_BROADCAST) != 0) {\r
-      //\r
-      // Configure the udp4 filter to receive all broadcast packages\r
-      //\r
-      Private->Udp4CfgData.AcceptBroadcast    = TRUE;\r
+  //\r
+  // In multicast condition when Promiscuous FALSE and IpCnt no-zero.\r
+  // Here check if there is any update of the multicast ip address. If yes,\r
+  // we need leave the old multicast group (by Config UDP instance to NULL),\r
+  // and join the new multicast group.\r
+  //\r
+  if (!AcceptPromiscuous) {\r
+    if ((NewFilter->Filters & EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP) != 0) {\r
+      if (Mode->IpFilter.IpCnt != NewFilter->IpCnt) {\r
+        MultiCastUpdate = TRUE;\r
+      } else if (CompareMem (Mode->IpFilter.IpList, NewFilter->IpList, NewFilter->IpCnt * sizeof (EFI_IP_ADDRESS)) != 0 ) {\r
+        MultiCastUpdate = TRUE;\r
+      }\r
     }\r
+  }\r
+  \r
+  //\r
+  // Check whether we need reconfigure the UDP instance.\r
+  //\r
+  Udp4Cfg = &Private->Udp4CfgData;\r
+  if ((AcceptPromiscuous != Udp4Cfg->AcceptPromiscuous) ||\r
+         (AcceptBroadcast != Udp4Cfg->AcceptBroadcast)     || MultiCastUpdate) {\r
+    //\r
+    // Clear the UDP instance configuration, all joined groups will be left\r
+    // during the operation.\r
+    //\r
+    Private->Udp4Read->Configure (Private->Udp4Read, NULL);\r
 \r
     //\r
     // Configure the UDP instance with the new configuration.\r
     //\r
-    Status = Private->Udp4Read->Configure (Private->Udp4Read, &Private->Udp4CfgData);\r
+    Udp4Cfg->AcceptPromiscuous = AcceptPromiscuous;\r
+    Udp4Cfg->AcceptBroadcast   = AcceptBroadcast;\r
+    Status = Private->Udp4Read->Configure (Private->Udp4Read, Udp4Cfg);\r
     if (EFI_ERROR (Status)) {\r
       return Status;\r
     }\r
 \r
-    if ((NewFilter->Filters & EFI_PXE_BASE_CODE_IP_FILTER_STATION_IP) != 0) {\r
-\r
+    //\r
+    // In not Promiscuous mode, need to join the new multicast group.\r
+    //\r
+    if (!AcceptPromiscuous) {\r
       for (Index = 0; Index < NewFilter->IpCnt; ++Index) {\r
         if (IP4_IS_MULTICAST (EFI_NTOHL (NewFilter->IpList[Index].v4))) {\r
           //\r
-          // Join the mutilcast group\r
+          // Join the mutilcast group.\r
           //\r
           Status = Private->Udp4Read->Groups (Private->Udp4Read, TRUE, &NewFilter->IpList[Index].v4);\r
           if (EFI_ERROR (Status)) {\r
@@ -2185,7 +2330,7 @@ EfiPxeBcSetStationIP (
     ZeroMem (&ArpConfigData, sizeof (EFI_ARP_CONFIG_DATA));\r
 \r
     ArpConfigData.SwAddressType   = 0x0800;\r
-    ArpConfigData.SwAddressLength = sizeof (EFI_IPv4_ADDRESS);\r
+    ArpConfigData.SwAddressLength = (UINT8) sizeof (EFI_IPv4_ADDRESS);\r
     ArpConfigData.StationAddress  = &Private->StationIp.v4;\r
 \r
     Private->Arp->Configure (Private->Arp, NULL);\r