[WS2_32]

[reactos.git] / reactos / dll / win32 / ws2_32 / src / dprovide.c

diff --git a/reactos/dll/win32/ws2_32/src/dprovide.c b/reactos/dll/win32/ws2_32/src/dprovide.c
index 41c389d..bc0f2f0 100644 (file)
--- a/reactos/dll/win32/ws2_32/src/dprovide.c
+++ b/reactos/dll/win32/ws2_32/src/dprovide.c
@@ -21,12 +21,13 @@ WsTpAllocate(VOID)
 {
     PTPROVIDER Provider;
 
-    DPRINT("WsTpAllocate: WsSockHeap %d\n", WsSockHeap);
     /* Allocate the object */
     Provider = HeapAlloc(WsSockHeap, HEAP_ZERO_MEMORY, sizeof(*Provider));
-
-    /* Setup non-zero data */
-    Provider->RefCount = 1;
+    if (Provider)
+    {
+        /* Setup non-zero data */
+        Provider->RefCount = 1;
+    }
 
     /* Return it */
     return Provider;
@@ -130,10 +131,11 @@ WsTpDelete(IN PTPROVIDER Provider)
 
         /* Unload the library */
         FreeLibrary(Provider->DllHandle);
-
-        /* Clear the handle value */
         Provider->DllHandle = NULL;
     }
+
+    /* Delete us */
+    HeapFree(WsSockHeap, 0, Provider);
 }
 
 VOID