From e3b7ea5642567fd2cd0d4228adcb32f55be47989 Mon Sep 17 00:00:00 2001
From: natede-MSFT <natede@microsoft.com>
Date: Wed, 2 Jun 2021 17:33:55 -0500
Subject: [PATCH] Remove deprecated API usage in DV-FailDriver-WDM sample.

---
 tools/dv/samples/DV-FailDriver-WDM/defect_toastmon.c | 10 +++++-----
 tools/dv/samples/DV-FailDriver-WDM/wmi.c             |  4 ++--
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/tools/dv/samples/DV-FailDriver-WDM/defect_toastmon.c b/tools/dv/samples/DV-FailDriver-WDM/defect_toastmon.c
index f8c545d29..f90a1a865 100644
--- a/tools/dv/samples/DV-FailDriver-WDM/defect_toastmon.c
+++ b/tools/dv/samples/DV-FailDriver-WDM/defect_toastmon.c
@@ -828,7 +828,7 @@ Return Value:
         // Allocate memory for the deviceinfo
         //
 
-        list = ExAllocatePoolWithTag(PagedPool,sizeof(DEVICE_INFO),DRIVER_TAG);
+        list = ExAllocatePool2(POOL_FLAG_PAGED, sizeof(DEVICE_INFO), DRIVER_TAG);
         if (list == NULL)
         {
             goto Error;
@@ -841,10 +841,10 @@ Return Value:
         list->SymbolicLink.MaximumLength = symbolicLinkName->Length +
                                           sizeof(UNICODE_NULL);
         list->SymbolicLink.Length = symbolicLinkName->Length;
-        list->SymbolicLink.Buffer = ExAllocatePoolWithTag (
-                                       PagedPool,
-                                       list->SymbolicLink.MaximumLength,
-                                       DRIVER_TAG);
+        list->SymbolicLink.Buffer = ExAllocatePool2(
+                                        POOL_FLAG_PAGED,
+                                        list->SymbolicLink.MaximumLength,
+                                        DRIVER_TAG);
         if (list->SymbolicLink.Buffer == NULL)
         {
             goto Error;
diff --git a/tools/dv/samples/DV-FailDriver-WDM/wmi.c b/tools/dv/samples/DV-FailDriver-WDM/wmi.c
index ee856d59d..098bd8d1e 100644
--- a/tools/dv/samples/DV-FailDriver-WDM/wmi.c
+++ b/tools/dv/samples/DV-FailDriver-WDM/wmi.c
@@ -191,8 +191,8 @@ Return Value:
         }
     }
 
-    valueInfo = ExAllocatePoolWithTag (NonPagedPool, resultLength,
-                                                DRIVER_TAG);
+    valueInfo = ExAllocatePool2(POOL_FLAG_NON_PAGED, resultLength,
+                                    DRIVER_TAG);
     if (NULL == valueInfo) {
         status = STATUS_INSUFFICIENT_RESOURCES;
         goto Error;