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_
 | 
						|
 |