IntelSiliconPkg VTdPmrPei: Add PcdVTdPeiDmaBufferSize(S3)

[mirror_edk2.git] / IntelSiliconPkg / Feature / VTd / IntelVTdPmrPei / IntelVTdPmrPei.c

diff --git a/IntelSiliconPkg/Feature/VTd/IntelVTdPmrPei/IntelVTdPmrPei.c b/IntelSiliconPkg/Feature/VTd/IntelVTdPmrPei/IntelVTdPmrPei.c
index 2b2df1f425aef0041525d5aa90d8485ff02cda90..27847f43317bf8e85fa43b4e6d916e6dfa182576 100644 (file)
--- a/IntelSiliconPkg/Feature/VTd/IntelVTdPmrPei/IntelVTdPmrPei.c
+++ b/IntelSiliconPkg/Feature/VTd/IntelVTdPmrPei/IntelVTdPmrPei.c
@@ -29,9 +29,6 @@
 \r
 #include "IntelVTdPmrPei.h"\r
 \r
-#define  TOTAL_DMA_BUFFER_SIZE    SIZE_4MB\r
-#define  TOTAL_DMA_BUFFER_SIZE_S3 SIZE_1MB\r
-\r
 EFI_GUID mVTdInfoGuid = {\r
   0x222f5e30, 0x5cd, 0x49c6, { 0x8a, 0xc, 0x36, 0xd6, 0x58, 0x41, 0xe0, 0x82 }\r
 };\r
@@ -798,9 +795,9 @@ IntelVTdPmrInitialize (
   PeiServicesGetBootMode (&BootMode);\r
 \r
   if (BootMode == BOOT_ON_S3_RESUME) {\r
-    DmaBufferInfo->DmaBufferSize = TOTAL_DMA_BUFFER_SIZE_S3;\r
+    DmaBufferInfo->DmaBufferSize = PcdGet32 (PcdVTdPeiDmaBufferSizeS3);\r
   } else {\r
-    DmaBufferInfo->DmaBufferSize = TOTAL_DMA_BUFFER_SIZE;\r
+    DmaBufferInfo->DmaBufferSize = PcdGet32 (PcdVTdPeiDmaBufferSize);\r
   }\r
 \r
   Status = PeiServicesNotifyPpi (&mVTdInfoNotifyDesc);\r