[ros-diffs] [janderwald] 39784: - Add sanity checks - Fix memory leaks

[janderwald at svn.reactos.org]

Author: janderwald
Date: Fri Feb 27 16:19:10 2009
New Revision: 39784

URL: http://svn.reactos.org/svn/reactos?rev=39784&view=rev
Log:
- Add sanity checks
- Fix memory leaks

Modified:
    trunk/reactos/drivers/wdm/audio/sysaudio/control.c

Modified: trunk/reactos/drivers/wdm/audio/sysaudio/control.c
URL: http://svn.reactos.org/svn/reactos/trunk/reactos/drivers/wdm/audio/sysaudio/control.c?rev=39784&r1=39783&r2=39784&view=diff
==============================================================================
--- trunk/reactos/drivers/wdm/audio/sysaudio/control.c [iso-8859-1] (original)
+++ trunk/reactos/drivers/wdm/audio/sysaudio/control.c [iso-8859-1] Fri Feb 27 16:19:10 2009
@@ -155,15 +155,24 @@
 
     WorkerContext->PinConnect->PinToHandle = NULL;
 
+    DPRINT1("CreatePinWorkerRoutine entered\n");
+
+    ASSERT(WorkerContext->Entry);
+    ASSERT(WorkerContext->PinConnect);
+    ASSERT(WorkerContext->Entry->Pins);
+    ASSERT(WorkerContext->Entry->NumberOfPins > WorkerContext->PinConnect->PinId);
 
     if (WorkerContext->CreateRealPin)
     {
         /* create the real pin */
+        DPRINT("Creating real pin\n");
         Status = KsCreatePin(WorkerContext->Entry->Handle, WorkerContext->PinConnect, GENERIC_READ | GENERIC_WRITE, &RealPinHandle);
+        DPRINT1("Status %x\n", Status);
         if (!NT_SUCCESS(Status))
         {
             DPRINT1("Failed to create Pin with %x\n", Status);
             SetIrpIoStatus(WorkerContext->Irp, STATUS_UNSUCCESSFUL, 0);
+            ExFreePool(WorkerContext->DispatchContext);
             ExFreePool(WorkerContext);
             return;
         }
@@ -173,6 +182,16 @@
                                            GENERIC_READ | GENERIC_WRITE, 
                                            IoFileObjectType, KernelMode, (PVOID*)&FileObject, NULL);
 
+        DPRINT1("Status %x\n", Status);
+        if (!NT_SUCCESS(Status))
+        {
+            DPRINT1("Failed to get file object with %x\n", Status);
+            SetIrpIoStatus(WorkerContext->Irp, STATUS_UNSUCCESSFUL, 0);
+            ExFreePool(WorkerContext->DispatchContext);
+            ExFreePool(WorkerContext);
+            return;
+        }
+
         if (WorkerContext->Entry->Pins[WorkerContext->PinConnect->PinId].MaxPinInstanceCount == 1)
         {
             /* store the pin handle there is the pin can only be instantiated once*/
@@ -183,11 +202,7 @@
         WorkerContext->DispatchContext->Handle = RealPinHandle;
         WorkerContext->DispatchContext->PinId = WorkerContext->PinConnect->PinId;
         WorkerContext->DispatchContext->AudioEntry = WorkerContext->Entry;
-
-        if (NT_SUCCESS(Status))
-            WorkerContext->DispatchContext->FileObject = FileObject;
-        else
-            WorkerContext->DispatchContext->FileObject = NULL;
+        WorkerContext->DispatchContext->FileObject = FileObject;
     }
     else
     {
@@ -204,6 +219,7 @@
         {
             DPRINT1("Failed to get file object with %x\n", Status);
             SetIrpIoStatus(WorkerContext->Irp, STATUS_UNSUCCESSFUL, 0);
+            ExFreePool(WorkerContext->DispatchContext);
             ExFreePool(WorkerContext);
             return;
         }
@@ -224,6 +240,7 @@
         }
 
         SetIrpIoStatus(WorkerContext->Irp, STATUS_UNSUCCESSFUL, 0);
+        ExFreePool(WorkerContext->DispatchContext);
         ExFreePool(WorkerContext);
         return;
     }
@@ -236,14 +253,10 @@
     if (!NT_SUCCESS(Status))
     {
         DPRINT1("Failed to get file object with %x\n", Status);
-        if (WorkerContext->CreateRealPin)
-        {
-            /* mark pin as free to use */
-            WorkerContext->Entry->Pins[WorkerContext->PinConnect->PinId].References = 0;
-        }
 
         ZwClose(VirtualPinHandle);
         SetIrpIoStatus(WorkerContext->Irp, STATUS_UNSUCCESSFUL, 0);
+        ExFreePool(WorkerContext->DispatchContext);
         ExFreePool(WorkerContext);
         return;
     }
@@ -256,8 +269,6 @@
     ASSERT(WorkerContext->AudioClient->NumDevices > 0);
     ASSERT(WorkerContext->AudioClient->Devs != NULL);
     ASSERT(WorkerContext->Entry->Pins != NULL);
-    ASSERT(WorkerContext->Entry->NumberOfPins > WorkerContext->PinConnect->PinId);
-
 
     AudioClient = WorkerContext->AudioClient;
     NumHandels = AudioClient->Devs[AudioClient->NumDevices -1].ClientHandlesCount;