Refine soma code to make code run safely.

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

diff --git a/MdeModulePkg/Universal/Network/SnpDxe/Initialize.c b/MdeModulePkg/Universal/Network/SnpDxe/Initialize.c
index 56f7283b5e9b3ee35b18518343af4e1c65a990b2..3f40ef3ce7e2149297f261167d16e90051e74d4e 100644 (file)
--- a/MdeModulePkg/Universal/Network/SnpDxe/Initialize.c
+++ b/MdeModulePkg/Universal/Network/SnpDxe/Initialize.c
@@ -84,8 +84,8 @@ PxeInit (
   Snp->Cdb.OpCode     = PXE_OPCODE_INITIALIZE;\r
   Snp->Cdb.OpFlags    = CableDetectFlag;\r
 \r
-  Snp->Cdb.CPBsize    = sizeof (PXE_CPB_INITIALIZE);\r
-  Snp->Cdb.DBsize     = sizeof (PXE_DB_INITIALIZE);\r
+  Snp->Cdb.CPBsize    = (UINT16) sizeof (PXE_CPB_INITIALIZE);\r
+  Snp->Cdb.DBsize     = (UINT16) sizeof (PXE_DB_INITIALIZE);\r
 \r
   Snp->Cdb.CPBaddr    = (UINT64)(UINTN) Snp->Cpb;\r
   Snp->Cdb.DBaddr     = (UINT64)(UINTN) Snp->Db;\r