USB: attach memory pool tag for control transfer

imbushuo · Aug 5, 2019 · ba8c0b6 · ba8c0b6
1 parent 1a88b18
commit ba8c0b6
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 5 deletions.
diff --git a/src/AmtPtpDeviceUsbKm/Device.c b/src/AmtPtpDeviceUsbKm/Device.c
@@ -523,7 +523,7 @@ AmtPtpSetWellspringMode(
 	status = WdfMemoryCreate(
 		WDF_NO_OBJECT_ATTRIBUTES,
 		PagedPool,
-		0,
+		POOL_TAG_PTP_CONTROL,
 		DeviceContext->DeviceInfo->um_size,
 		&bufHandle,
 		&buffer
@@ -626,6 +626,7 @@ AmtPtpSetWellspringMode(
 		"%!FUNC! Exit"
 	);
 
+	WdfObjectDelete(bufHandle);
 	bufHandle = NULL;
 	return status;
 }
diff --git a/src/AmtPtpDeviceUsbKm/Device.h b/src/AmtPtpDeviceUsbKm/Device.h
@@ -53,6 +53,11 @@ typedef struct _DEVICE_CONTEXT
 //
 WDF_DECLARE_CONTEXT_TYPE_WITH_NAME(DEVICE_CONTEXT, DeviceGetContext)
 
+//
+// Pool tags
+//
+#define POOL_TAG_PTP_CONTROL 'PTPC'
+
 //
 // Function to initialize the device's queues and callbacks
 //

diff --git a/src/AmtPtpDeviceUsbUm/Device.c b/src/AmtPtpDeviceUsbUm/Device.c
@@ -340,7 +340,7 @@ AmtPtpGetWellspringMode(
 	status = WdfMemoryCreate(
 		WDF_NO_OBJECT_ATTRIBUTES,
 		PagedPool,
-		0,
+		POOL_TAG_PTP_CONTROL,
 		DeviceContext->DeviceInfo->um_size,
 		&bufHandle,
 		&buffer
@@ -407,7 +407,7 @@ AmtPtpGetWellspringMode(
 		"%!FUNC! Exit"
 	);
 
-	bufHandle = NULL;
+	WdfObjectDelete(bufHandle);
 	return status;
 
 }
@@ -443,7 +443,7 @@ AmtPtpSetWellspringMode(
 	status = WdfMemoryCreate(
 		WDF_NO_OBJECT_ATTRIBUTES, 
 		PagedPool, 
-		0, 
+		POOL_TAG_PTP_CONTROL, 
 		DeviceContext->DeviceInfo->um_size, 
 		&bufHandle, 
 		&buffer
@@ -546,7 +546,7 @@ AmtPtpSetWellspringMode(
 		"%!FUNC! Exit"
 	);
 
-	bufHandle = NULL;
+	WdfObjectDelete(bufHandle);
 	return status;
 
 }

diff --git a/src/AmtPtpDeviceUsbUm/Device.h b/src/AmtPtpDeviceUsbUm/Device.h
@@ -44,6 +44,11 @@ typedef struct _DEVICE_CONTEXT
 //
 WDF_DECLARE_CONTEXT_TYPE_WITH_NAME(DEVICE_CONTEXT, DeviceGetContext)
 
+//
+// Pool tags
+//
+#define POOL_TAG_PTP_CONTROL 'PTPC'
+
 //
 // Function to initialize the device's queues and callbacks
 //