Win32 API doesn't provide pthreads-like destructors, so we need to emulate them to free per-thread allocated memory. git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@63653 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
		
			
				
	
	
		
			119 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			119 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| ///////////////////////////////////////////////////////////////////////////////
 | |
| // Name:        wx/os2/tls.h
 | |
| // Purpose:     OS/2 implementation of wxTlsValue<>
 | |
| // Author:      Stefan Neis
 | |
| // Created:     2008-08-30
 | |
| // RCS-ID:      $Id$
 | |
| // Copyright:   (c) 2008 Stefan Neis
 | |
| // Licence:     wxWindows licence
 | |
| ///////////////////////////////////////////////////////////////////////////////
 | |
| 
 | |
| #ifndef _WX_OS2_TLS_H_
 | |
| #define _WX_OS2_TLS_H_
 | |
| 
 | |
| #include "wx/os2/private.h"
 | |
| #include "wx/thread.h"
 | |
| #include "wx/vector.h"
 | |
| 
 | |
| // ----------------------------------------------------------------------------
 | |
| // wxTlsKey is a helper class encapsulating a TLS slot
 | |
| // ----------------------------------------------------------------------------
 | |
| 
 | |
| class wxTlsKey
 | |
| {
 | |
| public:
 | |
|     // ctor allocates a new key
 | |
|     wxTlsKey(wxTlsDestructorFunction destructor)
 | |
|     {
 | |
|         m_destructor = destructor;
 | |
|         APIRET rc = ::DosAllocThreadLocalMemory(1, &m_slot);
 | |
|         if (rc != NO_ERROR)
 | |
|             m_slot = NULL;
 | |
|     }
 | |
| 
 | |
|     // return true if the key was successfully allocated
 | |
|     bool IsOk() const { return m_slot != NULL; }
 | |
| 
 | |
|     // get the key value, there is no error return
 | |
|     void *Get() const
 | |
|     {
 | |
|         return (void *)m_slot;
 | |
|     }
 | |
| 
 | |
|     // change the key value, return true if ok
 | |
|     bool Set(void *value)
 | |
|     {
 | |
|         void *old = Get();
 | |
| 
 | |
|         m_slot = (ULONG*)value;
 | |
| 
 | |
|         if ( old )
 | |
|             m_destructor(old);
 | |
| 
 | |
|         // update m_allValues list of all values - remove old, add new
 | |
|         wxCriticalSectionLocker lock(m_csAllValues);
 | |
|         if ( old )
 | |
|         {
 | |
|             for ( wxVector<void*>::iterator i = m_allValues.begin();
 | |
|                   i != m_allValues.end();
 | |
|                   ++i )
 | |
|             {
 | |
|                 if ( *i == old )
 | |
|                 {
 | |
|                     if ( value )
 | |
|                         *i = value;
 | |
|                     else
 | |
|                         m_allValues.erase(i);
 | |
|                     return true;
 | |
|                 }
 | |
|             }
 | |
|             wxFAIL_MSG( "previous wxTlsKey value not recorded in m_allValues" );
 | |
|         }
 | |
| 
 | |
|         if ( value )
 | |
|             m_allValues.push_back(value);
 | |
| 
 | |
|         return true;
 | |
|     }
 | |
| 
 | |
|     // free the key
 | |
|     ~wxTlsKey()
 | |
|     {
 | |
|         if ( !IsOk() )
 | |
|             return;
 | |
| 
 | |
|         // Win32 and OS/2 API doesn't have the equivalent of pthread's
 | |
|         // destructor, so we have to keep track of all allocated values and
 | |
|         // destroy them manually; ideally we'd do that at thread exit time, but
 | |
|         // since we could only do that with wxThread and not otherwise created
 | |
|         // threads, we do it here.
 | |
|         //
 | |
|         // TODO: We should still call destructors for wxTlsKey used in the
 | |
|         //       thread from wxThread's thread shutdown code, *in addition*
 | |
|         //       to doing it in ~wxTlsKey.
 | |
|         //
 | |
|         // NB: No need to lock m_csAllValues, by the time this code is called,
 | |
|         //     no other thread can be using this key.
 | |
|         for ( wxVector<void*>::iterator i = m_allValues.begin();
 | |
|               i != m_allValues.end();
 | |
|               ++i )
 | |
|         {
 | |
|             m_destructor(*i);
 | |
|         }
 | |
| 
 | |
|         ::DosFreeThreadLocalMemory(m_slot);
 | |
|     }
 | |
| 
 | |
| private:
 | |
|     wxTlsDestructorFunction m_destructor;
 | |
|     ULONG* m_slot;
 | |
| 
 | |
|     wxVector<void*> m_allValues;
 | |
|     wxCriticalSection m_csAllValues;
 | |
| 
 | |
|     wxDECLARE_NO_COPY_CLASS(wxTlsKey);
 | |
| };
 | |
| 
 | |
| #endif // _WX_OS2_TLS_H_
 | |
| 
 |