[ros-diffs] [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

[cgutman at svn.reactos.org]

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/wshtcpip.c?rev=43154&r1=43153&r2=43154&view=diff
==============================================================================
--- 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;
 }