[cgutman] 43154: - We want socket closure notification so we don't leak our socket context when the socket is destroyed - Free socket context in WSHNotify

Author: cgutman
Date: Sat Sep 26 01:44:51 2009
New Revision: 43154
URL: http://svn.reactos.org/svn/reactos?rev=43154&view=rev
Log:
 - We want socket closure notification so we don't leak our socket context when the
socket is destroyed
 - Free socket context in WSHNotify
Modified:
    trunk/reactos/dll/win32/wshtcpip/wshtcpip.c
Modified: trunk/reactos/dll/win32/wshtcpip/wshtcpip.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/wshtcpip/wshtcpi…
==============================================================================
--- trunk/reactos/dll/win32/wshtcpip/wshtcpip.c [iso-8859-1] (original)
+++ trunk/reactos/dll/win32/wshtcpip/wshtcpip.c [iso-8859-1] Sat Sep 26 01:44:51 2009
@@ -312,7 +312,16 @@
     IN  HANDLE TdiConnectionObjectHandle,
     IN  DWORD NotifyEvent)
 {
-    UNIMPLEMENTED
+    switch (NotifyEvent)
+    {
+        case WSH_NOTIFY_CLOSE:
+        HeapFree(GetProcessHeap(), 0, HelperDllSocketContext);
+        break;
+
+        default:
+        DPRINT1("Unwanted notification received! (%d)\n", NotifyEvent);
+        break;
+    }
     return 0;
 }
@@ -446,6 +455,7 @@
     Context->Flags         = Flags;
     *HelperDllSocketContext = Context;
+    *NotificationEvents = WSH_NOTIFY_CLOSE;
     return NO_ERROR;
 }