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