Remove deprecated API usage in DV-FailDriver-WDM sample.

microsoft · Jun 2, 2021 · e3b7ea5 · e3b7ea5
1 parent 10037b5
commit e3b7ea5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git 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
@@ -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;