Refine soma code to make code run safely.

[mirror_edk2.git] / MdeModulePkg / Universal / Network / SnpDxe / Mcast_ip_to_mac.c

diff --git a/MdeModulePkg/Universal/Network/SnpDxe/Mcast_ip_to_mac.c b/MdeModulePkg/Universal/Network/SnpDxe/Mcast_ip_to_mac.c
index f4294c03d1ee1a2a155c35e2901b79c74c6dc090..23dba901942ae54c9cc4ae6731576e8e4325892c 100644 (file)
--- a/MdeModulePkg/Universal/Network/SnpDxe/Mcast_ip_to_mac.c
+++ b/MdeModulePkg/Universal/Network/SnpDxe/Mcast_ip_to_mac.c
@@ -45,8 +45,8 @@ PxeIp2Mac (
   Db                  = Snp->Db;\r
   Snp->Cdb.OpCode     = PXE_OPCODE_MCAST_IP_TO_MAC;\r
   Snp->Cdb.OpFlags    = (UINT16) (IPv6 ? PXE_OPFLAGS_MCAST_IPV6_TO_MAC : PXE_OPFLAGS_MCAST_IPV4_TO_MAC);\r
-  Snp->Cdb.CPBsize    = sizeof (PXE_CPB_MCAST_IP_TO_MAC);\r
-  Snp->Cdb.DBsize     = sizeof (PXE_DB_MCAST_IP_TO_MAC);\r
+  Snp->Cdb.CPBsize    = (UINT16) sizeof (PXE_CPB_MCAST_IP_TO_MAC);\r
+  Snp->Cdb.DBsize     = (UINT16) sizeof (PXE_DB_MCAST_IP_TO_MAC);\r
 \r
   Snp->Cdb.CPBaddr    = (UINT64)(UINTN) Cpb;\r
   Snp->Cdb.DBaddr     = (UINT64)(UINTN) Db;\r