1 /** 2 * This file has no copyright assigned and is placed in the Public Domain. 3 * This file is part of the mingw-w64 runtime package. 4 * No warranty is given; refer to the file DISCLAIMER.PD within this package. 5 */ 6 #ifndef __MSPUTILS_H_ 7 #define __MSPUTILS_H_ 8 9 #if _ATL_VER >= 0x0300 10 #define DECLARE_VQI() 11 #else 12 #define DECLARE_VQI() STDMETHOD(QueryInterface)(REFIID iid,void **ppvObject) = 0; STDMETHOD_(ULONG,AddRef)() = 0; STDMETHOD_(ULONG,Release)() = 0; 13 #endif 14 15 #define MSP_(hr) (FAILED(hr)?MSP_ERROR:MSP_TRACE) 16 17 extern __inline WINBOOL IsValidAggregatedMediaType(DWORD dwAggregatedMediaType) { 18 const DWORD dwAllPossibleMediaTypes = TAPIMEDIATYPE_AUDIO | TAPIMEDIATYPE_VIDEO | TAPIMEDIATYPE_DATAMODEM | TAPIMEDIATYPE_G3FAX | TAPIMEDIATYPE_MULTITRACK; 19 WINBOOL bValidMediaType = FALSE; 20 if((0==(dwAggregatedMediaType & dwAllPossibleMediaTypes)) || (0!=(dwAggregatedMediaType & (~dwAllPossibleMediaTypes)))) { 21 bValidMediaType = FALSE; 22 } else { 23 bValidMediaType = TRUE; 24 } 25 return bValidMediaType; 26 } 27 28 extern __inline WINBOOL IsSingleMediaType(DWORD dwMediaType) { return !((dwMediaType==0) || ((dwMediaType & (dwMediaType - 1))!=0)); } 29 extern __inline WINBOOL IsValidSingleMediaType(DWORD dwMediaType,DWORD dwMask) { return IsSingleMediaType(dwMediaType) && ((dwMediaType & dwMask)==dwMediaType); } 30 31 const DWORD INITIAL = 8; 32 const DWORD DELTA = 8; 33 34 template <class T,DWORD dwInitial = INITIAL,DWORD dwDelta = DELTA> class CMSPArray { 35 protected: 36 T *m_aT; 37 int m_nSize; 38 int m_nAllocSize; 39 public: 40 CMSPArray() : m_aT(NULL),m_nSize(0),m_nAllocSize(0) { } 41 ~CMSPArray() { RemoveAll(); } 42 int GetSize() const { return m_nSize; } 43 WINBOOL Grow() { 44 T *aT; 45 int nNewAllocSize = (m_nAllocSize==0) ? dwInitial : (m_nSize + DELTA); 46 aT = (T *)realloc(m_aT,nNewAllocSize *sizeof(T)); 47 if(!aT) return FALSE; 48 m_nAllocSize = nNewAllocSize; 49 m_aT = aT; 50 return TRUE; 51 } 52 WINBOOL Add(T &t) { 53 if(m_nSize==m_nAllocSize) { 54 if(!Grow()) return FALSE; 55 } 56 m_nSize++; 57 SetAtIndex(m_nSize - 1,t); 58 return TRUE; 59 } 60 WINBOOL Remove(T &t) { 61 int nIndex = Find(t); 62 if(nIndex==-1) return FALSE; 63 return RemoveAt(nIndex); 64 } 65 WINBOOL RemoveAt(int nIndex) { 66 if(nIndex!=(m_nSize - 1)) 67 memmove((void*)&m_aT[nIndex],(void*)&m_aT[nIndex + 1],(m_nSize - (nIndex + 1))*sizeof(T)); 68 m_nSize--; 69 return TRUE; 70 } 71 void RemoveAll() { 72 if(m_nAllocSize > 0) { 73 free(m_aT); 74 m_aT = NULL; 75 m_nSize = 0; 76 m_nAllocSize = 0; 77 } 78 } 79 T &operator[] (int nIndex) const { 80 _ASSERTE(nIndex >= 0 && nIndex < m_nSize); 81 return m_aT[nIndex]; 82 } 83 T *GetData() const { return m_aT; } 84 void SetAtIndex(int nIndex,T &t) { 85 _ASSERTE(nIndex >= 0 && nIndex < m_nSize); 86 m_aT[nIndex] = t; 87 } 88 int Find(T &t) const { 89 for(int i = 0;i < m_nSize;i++) { 90 if(m_aT[i]==t) return i; 91 } 92 return -1; 93 } 94 }; 95 96 class CMSPCritSection { 97 private: 98 CRITICAL_SECTION m_CritSec; 99 public: 100 CMSPCritSection() { InitializeCriticalSection(&m_CritSec); } 101 ~CMSPCritSection() { DeleteCriticalSection(&m_CritSec); } 102 void Lock() { EnterCriticalSection(&m_CritSec); } 103 WINBOOL TryLock() { return TryEnterCriticalSection(&m_CritSec); } 104 void Unlock() { LeaveCriticalSection(&m_CritSec); } 105 }; 106 107 class CLock { 108 private: 109 CMSPCritSection &m_CriticalSection; 110 public: 111 CLock(CMSPCritSection &CriticalSection) : m_CriticalSection(CriticalSection) { 112 m_CriticalSection.Lock(); 113 } 114 ~CLock() { m_CriticalSection.Unlock(); } 115 }; 116 117 class CCSLock { 118 private: 119 CRITICAL_SECTION *m_pCritSec; 120 public: 121 CCSLock(CRITICAL_SECTION *pCritSec) : m_pCritSec(pCritSec) { 122 EnterCriticalSection(m_pCritSec); 123 } 124 ~CCSLock() { LeaveCriticalSection(m_pCritSec); } 125 }; 126 127 #ifndef CONTAINING_RECORD 128 #define CONTAINING_RECORD(address,type,field) ((type *)((PCHAR)(address) - (ULONG_PTR)(&((type *)0)->field))) 129 #endif 130 131 #ifndef InitializeListHead 132 #define InitializeListHead(ListHead) ((ListHead)->Flink = (ListHead)->Blink = (ListHead)) 133 #define IsListEmpty(ListHead) ((ListHead)->Flink==(ListHead)) 134 #define RemoveHeadList(ListHead) (ListHead)->Flink; {RemoveEntryList((ListHead)->Flink)} 135 #define RemoveTailList(ListHead) (ListHead)->Blink; {RemoveEntryList((ListHead)->Blink)} 136 #define RemoveEntryList(Entry) { PLIST_ENTRY _EX_Blink; PLIST_ENTRY _EX_Flink; _EX_Flink = (Entry)->Flink; _EX_Blink = (Entry)->Blink; _EX_Blink->Flink = _EX_Flink; _EX_Flink->Blink = _EX_Blink; } 137 #define InsertTailList(ListHead,Entry) { PLIST_ENTRY _EX_Blink; PLIST_ENTRY _EX_ListHead; _EX_ListHead = (ListHead); _EX_Blink = _EX_ListHead->Blink; (Entry)->Flink = _EX_ListHead; (Entry)->Blink = _EX_Blink; _EX_Blink->Flink = (Entry); _EX_ListHead->Blink = (Entry); } 138 #define InsertHeadList(ListHead,Entry) { PLIST_ENTRY _EX_Flink; PLIST_ENTRY _EX_ListHead; _EX_ListHead = (ListHead); _EX_Flink = _EX_ListHead->Flink; (Entry)->Flink = _EX_Flink; (Entry)->Blink = _EX_ListHead; _EX_Flink->Blink = (Entry); _EX_ListHead->Flink = (Entry); } 139 140 WINBOOL IsNodeOnList(PLIST_ENTRY ListHead,PLIST_ENTRY Entry); 141 #endif 142 143 template <class T> ULONG MSPAddRefHelper (T *pMyThis) { 144 LOG((MSP_INFO,"MSPAddRefHelper - this = 0x%08x",pMyThis)); 145 typedef CComAggObject<T> AggClass; 146 AggClass *p = CONTAINING_RECORD(pMyThis,AggClass,m_contained); 147 return p->AddRef(); 148 } 149 150 template <class T> ULONG MSPReleaseHelper (T *pMyThis) { 151 LOG((MSP_INFO,"MSPReleaseHelper - this = 0x%08x",pMyThis)); 152 typedef CComAggObject<T> AggClass; 153 AggClass *p = CONTAINING_RECORD(pMyThis,AggClass,m_contained); 154 return p->Release(); 155 } 156 157 #include <objsafe.h> 158 159 class CMSPObjectSafetyImpl : public IObjectSafety { 160 public: 161 CMSPObjectSafetyImpl() : m_dwSafety(0) { } 162 enum { 163 SUPPORTED_SAFETY_OPTIONS = INTERFACESAFE_FOR_UNTRUSTED_CALLER | INTERFACESAFE_FOR_UNTRUSTED_DATA 164 }; 165 STDMETHOD(SetInterfaceSafetyOptions)(REFIID riid,DWORD dwOptionSetMask,DWORD dwEnabledOptions) { 166 if((~SUPPORTED_SAFETY_OPTIONS & dwOptionSetMask)!=0) return E_FAIL; 167 IUnknown *pUnk = NULL; 168 HRESULT hr = QueryInterface(riid,(void**)&pUnk); 169 if(SUCCEEDED(hr)) { 170 pUnk->Release(); 171 pUnk = NULL; 172 s_CritSection.Lock(); 173 m_dwSafety = (dwEnabledOptions & dwOptionSetMask) | (m_dwSafety & ~dwOptionSetMask); 174 s_CritSection.Unlock(); 175 } 176 return hr; 177 } 178 STDMETHOD(GetInterfaceSafetyOptions)(REFIID riid,DWORD *pdwSupportedOptions,DWORD *pdwEnabledOptions) { 179 if(IsBadWritePtr(pdwSupportedOptions,sizeof(DWORD)) || IsBadWritePtr(pdwEnabledOptions,sizeof(DWORD))) return E_POINTER; 180 *pdwSupportedOptions = 0; 181 *pdwEnabledOptions = 0; 182 IUnknown *pUnk = NULL; 183 HRESULT hr = QueryInterface(riid,(void**)&pUnk); 184 if(SUCCEEDED(hr)) { 185 pUnk->Release(); 186 pUnk = NULL; 187 *pdwSupportedOptions = SUPPORTED_SAFETY_OPTIONS; 188 s_CritSection.Lock(); 189 *pdwEnabledOptions = m_dwSafety; 190 s_CritSection.Unlock(); 191 } 192 return hr; 193 } 194 private: 195 DWORD m_dwSafety; 196 static CMSPCritSection s_CritSection; 197 }; 198 199 #endif 200
