1*a20fd023SAndrew Rist /************************************************************** 2cdf0e10cSrcweir * 3*a20fd023SAndrew Rist * Licensed to the Apache Software Foundation (ASF) under one 4*a20fd023SAndrew Rist * or more contributor license agreements. See the NOTICE file 5*a20fd023SAndrew Rist * distributed with this work for additional information 6*a20fd023SAndrew Rist * regarding copyright ownership. The ASF licenses this file 7*a20fd023SAndrew Rist * to you under the Apache License, Version 2.0 (the 8*a20fd023SAndrew Rist * "License"); you may not use this file except in compliance 9*a20fd023SAndrew Rist * with the License. You may obtain a copy of the License at 10*a20fd023SAndrew Rist * 11*a20fd023SAndrew Rist * http://www.apache.org/licenses/LICENSE-2.0 12*a20fd023SAndrew Rist * 13*a20fd023SAndrew Rist * Unless required by applicable law or agreed to in writing, 14*a20fd023SAndrew Rist * software distributed under the License is distributed on an 15*a20fd023SAndrew Rist * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 16*a20fd023SAndrew Rist * KIND, either express or implied. See the License for the 17*a20fd023SAndrew Rist * specific language governing permissions and limitations 18*a20fd023SAndrew Rist * under the License. 19*a20fd023SAndrew Rist * 20*a20fd023SAndrew Rist *************************************************************/ 21*a20fd023SAndrew Rist 22*a20fd023SAndrew Rist 23cdf0e10cSrcweir 24cdf0e10cSrcweir #ifndef _INPROCSERV_SMARTPOINTER_HXX_ 25cdf0e10cSrcweir #define _INPROCSERV_SMARTPOINTER_HXX_ 26cdf0e10cSrcweir 27cdf0e10cSrcweir // #define OWNDEBUG 28cdf0e10cSrcweir 29cdf0e10cSrcweir #ifdef OWNDEBUG 30cdf0e10cSrcweir #define WRITEDEBUGINFOINTERN( x ) WriteDebugInfo( (DWORD)this, x, sizeof( x ) ) 31cdf0e10cSrcweir #define WRITEDEBUGINFO( x ) WRITEDEBUGINFOINTERN( x ":" MY_STRING_LINE "\n" ) 32cdf0e10cSrcweir #define TO_STRING( x ) #x 33cdf0e10cSrcweir #define MACRO_VALUE_TO_STRING( x ) TO_STRING( x ) 34cdf0e10cSrcweir #define MY_STRING_LINE MACRO_VALUE_TO_STRING( __LINE__ ) 35cdf0e10cSrcweir #else 36cdf0e10cSrcweir #define WRITEDEBUGINFO( x ) void() 37cdf0e10cSrcweir #define MY_STRING_LINE 38cdf0e10cSrcweir #endif 39cdf0e10cSrcweir 40cdf0e10cSrcweir 41cdf0e10cSrcweir namespace inprocserv{ 42cdf0e10cSrcweir 43cdf0e10cSrcweir void WriteDebugInfo( DWORD pThis, char* pString, DWORD nToWrite ); 44cdf0e10cSrcweir 45cdf0e10cSrcweir template< class T > class ComSmart 46cdf0e10cSrcweir { 47cdf0e10cSrcweir T* m_pInterface; 48cdf0e10cSrcweir OwnRelease()49cdf0e10cSrcweir void OwnRelease() 50cdf0e10cSrcweir { 51cdf0e10cSrcweir if ( m_pInterface ) 52cdf0e10cSrcweir { 53cdf0e10cSrcweir T* pInterface = m_pInterface; 54cdf0e10cSrcweir m_pInterface = NULL; 55cdf0e10cSrcweir pInterface->Release(); 56cdf0e10cSrcweir } 57cdf0e10cSrcweir } 58cdf0e10cSrcweir 59cdf0e10cSrcweir public: ComSmart()60cdf0e10cSrcweir ComSmart() 61cdf0e10cSrcweir : m_pInterface( NULL ) 62cdf0e10cSrcweir {} 63cdf0e10cSrcweir ComSmart(const ComSmart<T> & rObj)64cdf0e10cSrcweir ComSmart( const ComSmart<T>& rObj ) 65cdf0e10cSrcweir : m_pInterface( rObj.m_pInterface ) 66cdf0e10cSrcweir { 67cdf0e10cSrcweir if ( m_pInterface != NULL ) 68cdf0e10cSrcweir m_pInterface->AddRef(); 69cdf0e10cSrcweir } 70cdf0e10cSrcweir ComSmart(T * pInterface)71cdf0e10cSrcweir ComSmart( T* pInterface ) 72cdf0e10cSrcweir : m_pInterface( pInterface ) 73cdf0e10cSrcweir { 74cdf0e10cSrcweir if ( m_pInterface != NULL ) 75cdf0e10cSrcweir m_pInterface->AddRef(); 76cdf0e10cSrcweir } 77cdf0e10cSrcweir ~ComSmart()78cdf0e10cSrcweir ~ComSmart() 79cdf0e10cSrcweir { 80cdf0e10cSrcweir OwnRelease(); 81cdf0e10cSrcweir } 82cdf0e10cSrcweir operator =(const ComSmart<T> & rObj)83cdf0e10cSrcweir ComSmart& operator=( const ComSmart<T>& rObj ) 84cdf0e10cSrcweir { 85cdf0e10cSrcweir OwnRelease(); 86cdf0e10cSrcweir 87cdf0e10cSrcweir m_pInterface = rObj.m_pInterface; 88cdf0e10cSrcweir 89cdf0e10cSrcweir if ( m_pInterface != NULL ) 90cdf0e10cSrcweir m_pInterface->AddRef(); 91cdf0e10cSrcweir 92cdf0e10cSrcweir return *this; 93cdf0e10cSrcweir } 94cdf0e10cSrcweir operator =(T * pInterface)95cdf0e10cSrcweir ComSmart<T>& operator=( T* pInterface ) 96cdf0e10cSrcweir { 97cdf0e10cSrcweir OwnRelease(); 98cdf0e10cSrcweir 99cdf0e10cSrcweir m_pInterface = pInterface; 100cdf0e10cSrcweir 101cdf0e10cSrcweir if ( m_pInterface != NULL ) 102cdf0e10cSrcweir m_pInterface->AddRef(); 103cdf0e10cSrcweir 104cdf0e10cSrcweir return *this; 105cdf0e10cSrcweir } 106cdf0e10cSrcweir operator T*() const107cdf0e10cSrcweir operator T*() const 108cdf0e10cSrcweir { 109cdf0e10cSrcweir return m_pInterface; 110cdf0e10cSrcweir } 111cdf0e10cSrcweir operator *() const112cdf0e10cSrcweir T& operator*() const 113cdf0e10cSrcweir { 114cdf0e10cSrcweir return *m_pInterface; 115cdf0e10cSrcweir } 116cdf0e10cSrcweir operator &()117cdf0e10cSrcweir T** operator&() 118cdf0e10cSrcweir { 119cdf0e10cSrcweir OwnRelease(); 120cdf0e10cSrcweir 121cdf0e10cSrcweir m_pInterface = NULL; 122cdf0e10cSrcweir 123cdf0e10cSrcweir return &m_pInterface; 124cdf0e10cSrcweir } 125cdf0e10cSrcweir operator ->() const126cdf0e10cSrcweir T* operator->() const 127cdf0e10cSrcweir { 128cdf0e10cSrcweir return m_pInterface; 129cdf0e10cSrcweir } 130cdf0e10cSrcweir operator ==(const ComSmart<T> & rObj) const131cdf0e10cSrcweir BOOL operator==( const ComSmart<T>& rObj ) const 132cdf0e10cSrcweir { 133cdf0e10cSrcweir return ( m_pInterface == rObj.m_pInterface ); 134cdf0e10cSrcweir } 135cdf0e10cSrcweir operator !=(const ComSmart<T> & rObj) const136cdf0e10cSrcweir BOOL operator!=( const ComSmart<T>& rObj ) const 137cdf0e10cSrcweir { 138cdf0e10cSrcweir return ( m_pInterface != rObj.m_pInterface ); 139cdf0e10cSrcweir } 140cdf0e10cSrcweir operator ==(const T * pInterface) const141cdf0e10cSrcweir BOOL operator==( const T* pInterface ) const 142cdf0e10cSrcweir { 143cdf0e10cSrcweir return ( m_pInterface == pInterface ); 144cdf0e10cSrcweir } 145cdf0e10cSrcweir operator !=(const T * pInterface) const146cdf0e10cSrcweir BOOL operator!=( const T* pInterface ) const 147cdf0e10cSrcweir { 148cdf0e10cSrcweir return ( m_pInterface != pInterface ); 149cdf0e10cSrcweir } 150cdf0e10cSrcweir }; 151cdf0e10cSrcweir 152cdf0e10cSrcweir class CSGuard 153cdf0e10cSrcweir { 154cdf0e10cSrcweir CRITICAL_SECTION* m_pCriticalSection; 155cdf0e10cSrcweir 156cdf0e10cSrcweir public: CSGuard(CRITICAL_SECTION * pCS)157cdf0e10cSrcweir CSGuard( CRITICAL_SECTION* pCS ) 158cdf0e10cSrcweir : m_pCriticalSection( pCS ) 159cdf0e10cSrcweir { 160cdf0e10cSrcweir if ( m_pCriticalSection ) 161cdf0e10cSrcweir EnterCriticalSection( m_pCriticalSection ); 162cdf0e10cSrcweir } 163cdf0e10cSrcweir ~CSGuard()164cdf0e10cSrcweir ~CSGuard() 165cdf0e10cSrcweir { 166cdf0e10cSrcweir if ( m_pCriticalSection ) 167cdf0e10cSrcweir LeaveCriticalSection( m_pCriticalSection ); 168cdf0e10cSrcweir } 169cdf0e10cSrcweir }; 170cdf0e10cSrcweir 171cdf0e10cSrcweir class ULONGGuard 172cdf0e10cSrcweir { 173cdf0e10cSrcweir ULONG* m_pValue; 174cdf0e10cSrcweir 175cdf0e10cSrcweir public: ULONGGuard(ULONG * pValue)176cdf0e10cSrcweir ULONGGuard( ULONG* pValue ) 177cdf0e10cSrcweir : m_pValue( pValue ) 178cdf0e10cSrcweir { 179cdf0e10cSrcweir if ( m_pValue ) 180cdf0e10cSrcweir (*m_pValue)++; 181cdf0e10cSrcweir } 182cdf0e10cSrcweir ~ULONGGuard()183cdf0e10cSrcweir ~ULONGGuard() 184cdf0e10cSrcweir { 185cdf0e10cSrcweir if ( m_pValue ) 186cdf0e10cSrcweir (*m_pValue)--; 187cdf0e10cSrcweir } 188cdf0e10cSrcweir }; 189cdf0e10cSrcweir 190cdf0e10cSrcweir } // namespace inprocserv 191cdf0e10cSrcweir 192cdf0e10cSrcweir #endif 193cdf0e10cSrcweir 194