Changeset 8839 for trunk/src/kernel32/overlappedio.cpp

Timestamp:

Jul 5, 2002, 7:58:26 PM (23 years ago)

Author:

sandervl

Message:

Keep reference count for overlapped IO objects to avoid premature destruction

File:

• trunk/src/kernel32/overlappedio.cpp (modified) (7 diffs)

trunk/src/kernel32/overlappedio.cpp

@@ -1 @@
-/* $Id: overlappedio.cpp,v 1.17 2002-06-11 16:36:06 sandervl Exp $ */
+/* $Id: overlappedio.cpp,v 1.18 2002-07-05 17:58:26 sandervl Exp $ */
 
 /*
@@ -29 +29 @@
                                          LPOVERLAPPED_HANDLER lpPollHandler,
                                          BOOL fFullDuplex) :
-                   hThreadRead(0), hThreadWrite(0), hThreadPoll(0)
+                   hThreadRead(0), hThreadWrite(0), hThreadPoll(0), refCount(0)
 {
     OverlappedIOError errcode = OutOfMemory;
@@ -75 +75 @@
     threadparam->lpOverlappedObj = this;
     hThreadRead  = ::CreateThread(NULL, 32*1024, OverlappedIOThread, (LPVOID)threadparam, 0, &dwThreadId);
+    if(hThreadRead) {//thread uses this object; keep reference count to avoid premature destruction
+        AddRef();
+    }
 
     if(lpWriteHandler && fFullDuplex) {
@@ -84 +87 @@
         threadparam->lpOverlappedObj = this;
         hThreadWrite = ::CreateThread(NULL, 32*1024, OverlappedIOThread, (LPVOID)threadparam, 0, &dwThreadId);
+        if(hThreadWrite) {//thread uses this object; keep reference count to avoid premature destruction
+            AddRef();
+        }
     }
 
@@ -95 +101 @@
         hThreadPoll  = ::CreateThread(NULL, 32*1024, OverlappedIOThread, (LPVOID)threadparam, 0, &dwThreadId);
         SetThreadPriority(hThreadPoll, THREAD_PRIORITY_TIME_CRITICAL);
+        if(hThreadPoll) {//thread uses this object; keep reference count to avoid premature destruction
+            AddRef();
+        }
     }
 
@@ -146 +155 @@
 
     DeleteCriticalSection(&critsect);
+}
+//******************************************************************************
+//******************************************************************************
+DWORD OverlappedIOHandler::AddRef()
+{
+    return InterlockedIncrement(&refCount);
+}
+//******************************************************************************
+//******************************************************************************
+DWORD OverlappedIOHandler::Release(BOOL fSignalExit)
+{
+    if(fSignalExit) {
+        ::SetEvent(hEventExit);
+    }
+    if(InterlockedDecrement(&refCount) == 0) {
+        dprintf(("OverlappedIOHandler::Release -> delete now"));
+        delete this;
+        return 0;
+    }
+    return refCount;
 }
 //******************************************************************************
@@ -314 +343 @@
         } //while(TRUE)
     }
+    Release();  //decrease reference count
     return 0;
 }