[AFD] Introduce and use pool tags. Thanks go to Arty for assisting me with this....

[reactos.git] / drivers / network / afd / afd / tdiconn.c

diff --git a/drivers/network/afd/afd/tdiconn.c b/drivers/network/afd/afd/tdiconn.c
index a0c27c9..afa9f0f 100644 (file)
--- a/drivers/network/afd/afd/tdiconn.c
+++ b/drivers/network/afd/afd/tdiconn.c
@@ -71,26 +71,6 @@ UINT TaLengthOfTransportAddressByType(UINT AddressType)
     return AddrLen;
 }
 
-VOID TaCopyAddressInPlace( PTA_ADDRESS Target,
-                           PTA_ADDRESS Source ) {
-    UINT AddrLen = TaLengthOfAddress( Source );
-    RtlCopyMemory( Target, Source, AddrLen );
-}
-
-PTA_ADDRESS TaCopyAddress( PTA_ADDRESS Source ) {
-    UINT AddrLen = TaLengthOfAddress( Source );
-    PVOID Buffer;
-    if (!AddrLen)
-        return NULL;
-
-    Buffer = ExAllocatePool( NonPagedPool, AddrLen );
-
-    if (Buffer)
-       RtlCopyMemory( Buffer, Source, AddrLen );
-
-    return Buffer;
-}
-
 VOID TaCopyTransportAddressInPlace( PTRANSPORT_ADDRESS Target,
                                     PTRANSPORT_ADDRESS Source ) {
     UINT AddrLen = TaLengthOfTransportAddress( Source );
@@ -105,7 +85,9 @@ PTRANSPORT_ADDRESS TaCopyTransportAddress( PTRANSPORT_ADDRESS OtherAddress ) {
     if (!AddrLen)
         return NULL;
 
-    A = ExAllocatePool( NonPagedPool, AddrLen );
+    A = ExAllocatePoolWithTag(NonPagedPool,
+                              AddrLen,
+                              TAG_AFD_TRANSPORT_ADDRESS);
 
     if( A )
         TaCopyTransportAddressInPlace( A, OtherAddress );
@@ -137,13 +119,13 @@ PTRANSPORT_ADDRESS TaBuildNullTransportAddress(UINT AddressType)
     if (!AddrLen)
         return NULL;
 
-    A = ExAllocatePool(NonPagedPool, AddrLen);
+    A = ExAllocatePoolWithTag(NonPagedPool, AddrLen, TAG_AFD_TRANSPORT_ADDRESS);
 
     if (A)
     {
         if (TdiBuildNullTransportAddressInPlace(A, AddressType) != STATUS_SUCCESS)
         {
-            ExFreePool(A);
+            ExFreePoolWithTag(A, TAG_AFD_TRANSPORT_ADDRESS);
             return NULL;
         }
     }
@@ -210,9 +192,9 @@ NTSTATUS TdiBuildNullConnectionInfo
     }
 
     ConnInfo = (PTDI_CONNECTION_INFORMATION)
-        ExAllocatePool(NonPagedPool,
-                       sizeof(TDI_CONNECTION_INFORMATION) +
-                       TdiAddressSize);
+        ExAllocatePoolWithTag(NonPagedPool,
+                              sizeof(TDI_CONNECTION_INFORMATION) + TdiAddressSize,
+                              TAG_AFD_TDI_CONNECTION_INFORMATION);
     if (!ConnInfo) {
         *ConnectionInfo = NULL;
         return STATUS_INSUFFICIENT_RESOURCES;
@@ -222,7 +204,7 @@ NTSTATUS TdiBuildNullConnectionInfo
 
     if (!NT_SUCCESS(Status))
     {
-        ExFreePool( ConnInfo );
+        ExFreePoolWithTag(ConnInfo, TAG_AFD_TDI_CONNECTION_INFORMATION);
         ConnInfo = NULL;
     }