Refine soma code to make code run safely.

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

diff --git a/MdeModulePkg/Universal/Network/SnpDxe/Snp.c b/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
index 13f57a1f54d2b7476969480642ec8777b1f1a62e..e90e4e0244075d836f3a8934833a003d227c6762 100644 (file)
--- a/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
+++ b/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
@@ -483,8 +483,8 @@ SimpleNetworkDriverStart (
   Snp->Cdb.CPBsize    = PXE_CPBSIZE_NOT_USED;\r
   Snp->Cdb.CPBaddr    = PXE_DBADDR_NOT_USED;\r
 \r
-  Snp->Cdb.DBsize     = sizeof Snp->InitInfo;\r
-  Snp->Cdb.DBaddr     = (UINT64)(UINTN) &Snp->InitInfo;\r
+  Snp->Cdb.DBsize     = (UINT16) sizeof (Snp->InitInfo);\r
+  Snp->Cdb.DBaddr     = (UINT64)(UINTN) (&Snp->InitInfo);\r
 \r
   Snp->Cdb.StatCode   = PXE_STATCODE_INITIALIZE;\r
   Snp->Cdb.StatFlags  = PXE_STATFLAGS_INITIALIZE;\r
@@ -513,7 +513,7 @@ SimpleNetworkDriverStart (
   Snp->Cdb.CPBsize    = PXE_CPBSIZE_NOT_USED;\r
   Snp->Cdb.CPBaddr    = PXE_DBADDR_NOT_USED;\r
 \r
-  Snp->Cdb.DBsize     = sizeof ConfigInfo;\r
+  Snp->Cdb.DBsize     = (UINT16) sizeof (ConfigInfo);\r
   Snp->Cdb.DBaddr     = (UINT64)(UINTN) &ConfigInfo;\r
 \r
   Snp->Cdb.StatCode   = PXE_STATCODE_INITIALIZE;\r