ATL源码学习－－－线程模型支持

[c-sharp] view plain copy

// COM Sync Classes 
class CComCriticalSection  
{  
public:  
    CComCriticalSection() throw()  
    {  
        memset(&m_sec, 0, sizeof(CRITICAL_SECTION));  
    }  
    ~CComCriticalSection()  
    {  
    }  
    HRESULT Lock() throw()  
    {  
        EnterCriticalSection(&m_sec);  
 return S_OK;  
    }  
    HRESULT Unlock() throw()  
    {  
        LeaveCriticalSection(&m_sec);  
 return S_OK;  
    }  
    HRESULT Init() throw()  
    {  
        HRESULT hRes = E_FAIL;  
        __try  
        {  
            InitializeCriticalSection(&m_sec);  
            hRes = S_OK;  
        }  
 // structured exception may be raised in low memory situations 
        __except(STATUS_NO_MEMORY == GetExceptionCode())  
        {             
            hRes = E_OUTOFMEMORY;         
        }  
 return hRes;  
    }  
    HRESULT Term() throw()  
    {  
        DeleteCriticalSection(&m_sec);  
 return S_OK;  
    }     
    CRITICAL_SECTION m_sec;  
};  
class CComAutoCriticalSection : public CComCriticalSection  
{  
public:  
    CComAutoCriticalSection()  
    {  
        HRESULT hr = CComCriticalSection::Init();  
 if (FAILED(hr))  
            AtlThrow(hr);  
    }  
    ~CComAutoCriticalSection() throw()  
    {  
        CComCriticalSection::Term();  
    }  
private :  
    HRESULT Init(); // Not implemented. CComAutoCriticalSection::Init should never be called 
    HRESULT Term(); // Not implemented. CComAutoCriticalSection::Term should never be called 
};  
class CComSafeDeleteCriticalSection : public CComCriticalSection  
{  
public:  
    CComSafeDeleteCriticalSection(): m_bInitialized(false)   
    {  
    }  
    ~CComSafeDeleteCriticalSection() throw()  
    {  
 if (!m_bInitialized)  
        {  
 return;  
        }  
        m_bInitialized = false;  
        CComCriticalSection::Term();  
    }  
    HRESULT Init() throw()  
    {  
        ATLASSERT( !m_bInitialized );  
        HRESULT hr = CComCriticalSection::Init();  
 if (SUCCEEDED(hr))  
        {  
            m_bInitialized = true;  
        }  
 return hr;  
    }  
    HRESULT Term() throw()  
    {  
 if (!m_bInitialized)  
        {  
 return S_OK;  
        }  
        m_bInitialized = false;  
 return CComCriticalSection::Term();  
    }  
    HRESULT Lock()  
    {  
 // CComSafeDeleteCriticalSection::Init or CComAutoDeleteCriticalSection::Init 
 // not called or failed. 
 // m_critsec member of CComObjectRootEx is now of type  
 // CComAutoDeleteCriticalSection. It has to be initialized 
 // by calling CComObjectRootEx::_AtlInitialConstruct 
        ATLASSUME(m_bInitialized);  
 return CComCriticalSection::Lock();  
    }  
private:  
 bool m_bInitialized;  
};  
class CComAutoDeleteCriticalSection : public CComSafeDeleteCriticalSection  
{  
private:  
 // CComAutoDeleteCriticalSection::Term should never be called 
    HRESULT Term() throw();  
};  
class CComFakeCriticalSection  
{  
public:  
    HRESULT Lock() throw() { return S_OK; }  
    HRESULT Unlock() throw() { return S_OK; }  
    HRESULT Init() throw() { return S_OK; }  
    HRESULT Term() throw() { return S_OK; }  
};  

[cpp] view plain copy

template< class TLock >  
class CComCritSecLock  
{  
public:  
    CComCritSecLock( TLock& cs, bool bInitialLock = true );  
    ~CComCritSecLock() throw();  
 HRESULT Lock() throw();  
 void Unlock() throw();  
// Implementation 
private:  
    TLock& m_cs;  
 bool m_bLocked;  
// Private to avoid accidental use 
    CComCritSecLock( const CComCritSecLock& ) throw();  
    CComCritSecLock& operator=( const CComCritSecLock& ) throw();  
};  
template< class TLock >  
inline CComCritSecLock< TLock >::CComCritSecLock( TLock& cs, bool bInitialLock ) :  
    m_cs( cs ),  
    m_bLocked( false )  
{  
 if( bInitialLock )  
    {  
 HRESULT hr;  
        hr = Lock();  
 if( FAILED( hr ) )  
        {  
            AtlThrow( hr );  
        }  
    }  
}  
template< class TLock >  
inline CComCritSecLock< TLock >::~CComCritSecLock() throw()  
{  
 if( m_bLocked )  
    {  
        Unlock();  
    }  
}  
template< class TLock >  
inline HRESULT CComCritSecLock< TLock >::Lock() throw()  
{  
 HRESULT hr;  
    ATLASSERT( !m_bLocked );  
    hr = m_cs.Lock();  
 if( FAILED( hr ) )  
    {  
 return( hr );  
    }  
    m_bLocked = true;  
 return( S_OK );  
}  
template< class TLock >  
inline void CComCritSecLock< TLock >::Unlock() throw()  
{  
    ATLASSUME( m_bLocked );  
    m_cs.Unlock();  
    m_bLocked = false;  
}  

[cpp] view plain copy

class CComMultiThreadModel  
{  
public:  
 static ULONG WINAPI Increment(LPLONG p) throw() {return InterlockedIncrement(p);}  
 static ULONG WINAPI Decrement(LPLONG p) throw() {return InterlockedDecrement(p);}  
 typedef CComAutoCriticalSection AutoCriticalSection;  
 typedef CComAutoDeleteCriticalSection AutoDeleteCriticalSection;  
 typedef CComCriticalSection CriticalSection;  
 typedef CComMultiThreadModelNoCS ThreadModelNoCS;  
};  
class CComSingleThreadModel  
{  
public:  
 static ULONG WINAPI Increment(LPLONG p) throw() {return ++(*p);}  
 static ULONG WINAPI Decrement(LPLONG p) throw() {return --(*p);}  
 typedef CComFakeCriticalSection AutoCriticalSection;  
 typedef CComFakeCriticalSection AutoDeleteCriticalSection;  
 typedef CComFakeCriticalSection CriticalSection;  
 typedef CComSingleThreadModel ThreadModelNoCS;  
};  
class CComMultiThreadModelNoCS  
{  
public:  
 static ULONG WINAPI Increment(LPLONG p) throw() {return InterlockedIncrement(p);}  
 static ULONG WINAPI Decrement(LPLONG p) throw() {return InterlockedDecrement(p);}  
 typedef CComFakeCriticalSection AutoCriticalSection;  
 typedef CComFakeCriticalSection AutoDeleteCriticalSection;  
 typedef CComFakeCriticalSection CriticalSection;  
 typedef CComMultiThreadModelNoCS ThreadModelNoCS;  
};  

[cpp] view plain copy

#if defined(_ATL_SINGLE_THREADED) 
#if defined(_ATL_APARTMENT_THREADED) || defined(_ATL_FREE_THREADED) 
#pragma message ("More than one global threading model defined.") 
#endif 
 typedef CComSingleThreadModel CComObjectThreadModel;  
 typedef CComSingleThreadModel CComGlobalsThreadModel;  
#elif defined(_ATL_APARTMENT_THREADED) 
#if defined(_ATL_SINGLE_THREADED) || defined(_ATL_FREE_THREADED) 
#pragma message ("More than one global threading model defined.") 
#endif 
 typedef CComSingleThreadModel CComObjectThreadModel;  
 typedef CComMultiThreadModel CComGlobalsThreadModel;  
#elif defined(_ATL_FREE_THREADED) 
#if defined(_ATL_SINGLE_THREADED) || defined(_ATL_APARTMENT_THREADED) 
#pragma message ("More than one global threading model defined.") 
#endif 
 typedef CComMultiThreadModel CComObjectThreadModel;  
 typedef CComMultiThreadModel CComGlobalsThreadModel;  
#else 
#pragma message ("No global threading model defined") 
#endif