xref: /reactos/sdk/include/vcruntime/comip.h (revision d6eebaa4) 
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 _INC_COMIP
7 #define _INC_COMIP
8 
9 #include <_mingw.h>
10 
11 #include <ole2.h>
12 #include <malloc.h>
13 
14 #include <comutil.h>
15 
16 #ifdef __cplusplus
17 
18 #pragma push_macro("new")
19 #undef new
20 
21 #include <new.h>
22 
23 class _com_error;
24 
25 #ifndef WINAPI
26 #define WINAPI __stdcall
27 #endif
28 
29 void WINAPI _com_issue_error(HRESULT);
30 struct IUnknown;
31 
32 template<typename _Interface,const IID *_IID >
33 class _com_IIID {
34 public:
35   typedef _Interface Interface;
36   static _Interface *GetInterfacePtr() throw() { return NULL; }
37   static _Interface& GetInterface() throw() { return *GetInterfacePtr(); }
38   static const IID& GetIID() throw() { return *_IID; }
39 };
40 
41 template<typename _IIID> class _com_ptr_t {
42 public:
43   typedef _IIID ThisIIID;
44   typedef typename _IIID::Interface Interface;
45   static const IID& GetIID() throw() { return ThisIIID::GetIID(); }
46   template<typename _OtherIID> _com_ptr_t(const _com_ptr_t<_OtherIID> &p) : m_pInterface(NULL) {
47     HRESULT hr = _QueryInterface(p);
48     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
49   }
50   template<typename _InterfaceType> _com_ptr_t(_InterfaceType *p) : m_pInterface(NULL) {
51     HRESULT hr = _QueryInterface(p);
52     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
53   }
54   template<typename _X> _com_ptr_t(LPSTR str) { new(this) _com_ptr_t(static_cast<LPCSTR> (str),NULL); }
55   template<typename _X> _com_ptr_t(LPWSTR str) { new(this) _com_ptr_t(static_cast<LPCWSTR> (str),NULL); }
56   template<typename _X> explicit _com_ptr_t(_com_ptr_t *p) : m_pInterface(NULL) {
57     if(!p) { _com_issue_error(E_POINTER); }
58     else {
59       m_pInterface = p->m_pInterface;
60       AddRef();
61     }
62   }
63   _com_ptr_t() throw() : m_pInterface(NULL) { }
64   _com_ptr_t(int null) : m_pInterface(NULL) {
65     if(null!=0) { _com_issue_error(E_POINTER); }
66   }
67 
68 #ifdef _NATIVE_NULLPTR_SUPPORTED
69   _com_ptr_t(decltype(nullptr)) : m_pInterface(NULL) {}
70 #endif
71 
72   _com_ptr_t(const _com_ptr_t &cp) throw() : m_pInterface(cp.m_pInterface) { _AddRef(); }
73   template<typename _X> _com_ptr_t(Interface *pInterface) throw() : m_pInterface(pInterface) { _AddRef(); }
74   _com_ptr_t(Interface *pInterface,bool fAddRef) throw() : m_pInterface(pInterface) {
75     if(fAddRef) _AddRef();
76   }
77   _com_ptr_t(const _variant_t& varSrc) : m_pInterface(NULL) {
78     HRESULT hr = QueryStdInterfaces(varSrc);
79     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
80   }
81   explicit _com_ptr_t(const CLSID &clsid,IUnknown *pOuter = NULL,DWORD dwClsContext = CLSCTX_ALL) : m_pInterface(NULL) {
82     HRESULT hr = CreateInstance(clsid,pOuter,dwClsContext);
83     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
84   }
85   explicit _com_ptr_t(LPCWSTR str,IUnknown *pOuter = NULL,DWORD dwClsContext = CLSCTX_ALL) : m_pInterface(NULL) {
86     HRESULT hr = CreateInstance(str,pOuter,dwClsContext);
87     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
88   }
89   explicit _com_ptr_t(LPCSTR str,IUnknown *pOuter = NULL,DWORD dwClsContext = CLSCTX_ALL) : m_pInterface(NULL) {
90     HRESULT hr = CreateInstance(str,pOuter,dwClsContext);
91     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
92   }
93   template<typename _OtherIID> _com_ptr_t &operator=(const _com_ptr_t<_OtherIID> &p) {
94     HRESULT hr = _QueryInterface(p);
95     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
96     return *this;
97   }
98   template<typename _InterfaceType> _com_ptr_t &operator=(_InterfaceType *p) {
99     HRESULT hr = _QueryInterface(p);
100     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
101     return *this;
102   }
103   template<typename _X> _com_ptr_t &operator=(Interface *pInterface) throw() {
104     if(m_pInterface!=pInterface) {
105       Interface *pOldInterface = m_pInterface;
106       m_pInterface = pInterface;
107       _AddRef();
108       if(pOldInterface!=NULL) pOldInterface->Release();
109     }
110     return *this;
111   }
112   _com_ptr_t &operator=(const _com_ptr_t &cp) throw() { return operator=(cp.m_pInterface); }
113   _com_ptr_t &operator=(int null) {
114     if(null!=0) { _com_issue_error(E_POINTER); }
115     return operator=(reinterpret_cast<Interface*>(NULL));
116   }
117   _com_ptr_t &operator=(const _variant_t& varSrc) {
118     HRESULT hr = QueryStdInterfaces(varSrc);
119     if(FAILED(hr) && (hr!=E_NOINTERFACE)) { _com_issue_error(hr); }
120     return *this;
121   }
122   ~_com_ptr_t() throw() { _Release(); }
123   void Attach(Interface *pInterface) throw() {
124     _Release();
125     m_pInterface = pInterface;
126   }
127   void Attach(Interface *pInterface,bool fAddRef) throw() {
128     _Release();
129     m_pInterface = pInterface;
130     if(fAddRef) {
131       if(!pInterface) { _com_issue_error(E_POINTER); }
132       else pInterface->AddRef();
133     }
134   }
135   Interface *Detach() throw() {
136     Interface *const old = m_pInterface;
137     m_pInterface = NULL;
138     return old;
139   }
140   operator Interface*() const throw() { return m_pInterface; }
141   operator Interface&() const {
142     if(!m_pInterface) { _com_issue_error(E_POINTER); }
143     return *m_pInterface;
144   }
145   Interface& operator*() const {
146     if(!m_pInterface) { _com_issue_error(E_POINTER); }
147     return *m_pInterface;
148   }
149   Interface **operator&() throw() {
150     _Release();
151     m_pInterface = NULL;
152     return &m_pInterface;
153   }
154   Interface *operator->() const {
155     if(!m_pInterface) { _com_issue_error(E_POINTER); }
156     return m_pInterface;
157   }
158   operator bool() const throw() { return m_pInterface!=NULL; }
159   template<typename _OtherIID> bool operator==(const _com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)==0; }
160   template<typename _OtherIID> bool operator==(_com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)==0; }
161   template<typename _InterfaceType> bool operator==(_InterfaceType *p) { return _CompareUnknown(p)==0; }
162   template<typename _X> bool operator==(Interface *p) { return (m_pInterface==p) ? true : _CompareUnknown(p)==0; }
163   template<typename _X> bool operator==(const _com_ptr_t &p) throw() { return operator==(p.m_pInterface); }
164   template<typename _X> bool operator==(_com_ptr_t &p) throw() { return operator==(p.m_pInterface); }
165   bool operator==(int null) {
166     if(null!=0) { _com_issue_error(E_POINTER); }
167     return !m_pInterface;
168   }
169   template<typename _OtherIID> bool operator!=(const _com_ptr_t<_OtherIID> &p) { return !(operator==(p)); }
170   template<typename _OtherIID> bool operator!=(_com_ptr_t<_OtherIID> &p) { return !(operator==(p)); }
171   template<typename _InterfaceType> bool operator!=(_InterfaceType *p) { return !(operator==(p)); }
172   bool operator!=(int null) { return !(operator==(null)); }
173   template<typename _OtherIID> bool operator<(const _com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)<0; }
174   template<typename _OtherIID> bool operator<(_com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)<0; }
175   template<typename _InterfaceType> bool operator<(_InterfaceType *p) { return _CompareUnknown(p)<0; }
176   template<typename _OtherIID> bool operator>(const _com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)>0; }
177   template<typename _OtherIID> bool operator>(_com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)>0; }
178   template<typename _InterfaceType> bool operator>(_InterfaceType *p) { return _CompareUnknown(p)>0; }
179   template<typename _OtherIID> bool operator<=(const _com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)<=0; }
180   template<typename _OtherIID> bool operator<=(_com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)<=0; }
181   template<typename _InterfaceType> bool operator<=(_InterfaceType *p) { return _CompareUnknown(p)<=0; }
182   template<typename _OtherIID> bool operator>=(const _com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)>=0; }
183   template<typename _OtherIID> bool operator>=(_com_ptr_t<_OtherIID> &p) { return _CompareUnknown(p)>=0; }
184   template<typename _InterfaceType> bool operator>=(_InterfaceType *p) { return _CompareUnknown(p)>=0; }
185   void Release() {
186     if(!m_pInterface) { _com_issue_error(E_POINTER); }
187     else {
188       m_pInterface->Release();
189       m_pInterface = NULL;
190     }
191   }
192   void AddRef() {
193     if(!m_pInterface) { _com_issue_error(E_POINTER); }
194     else m_pInterface->AddRef();
195   }
196   Interface *GetInterfacePtr() const throw() { return m_pInterface; }
197   Interface*& GetInterfacePtr() throw() { return m_pInterface; }
198   HRESULT CreateInstance(const CLSID &rclsid,IUnknown *pOuter = NULL,DWORD dwClsContext = CLSCTX_ALL) throw() {
199     HRESULT hr;
200     _Release();
201     if(dwClsContext & (CLSCTX_LOCAL_SERVER | CLSCTX_REMOTE_SERVER)) {
202       IUnknown *pIUnknown;
203       hr = CoCreateInstance(rclsid,pOuter,dwClsContext,IID_IUnknown,reinterpret_cast<void**>(&pIUnknown));
204       if(SUCCEEDED(hr)) {
205         hr = OleRun(pIUnknown);
206         if(SUCCEEDED(hr)) hr = pIUnknown->QueryInterface(GetIID(),reinterpret_cast<void**>(&m_pInterface));
207         pIUnknown->Release();
208       }
209     } else hr = CoCreateInstance(rclsid,pOuter,dwClsContext,GetIID(),reinterpret_cast<void**>(&m_pInterface));
210     if(FAILED(hr)) m_pInterface = NULL;
211     return hr;
212   }
213   HRESULT CreateInstance(LPCWSTR clsidString,IUnknown *pOuter = NULL,DWORD dwClsContext = CLSCTX_ALL) throw() {
214     if(!clsidString) return E_INVALIDARG;
215     CLSID clsid;
216     HRESULT hr;
217     if(clsidString[0]==L'{') hr = CLSIDFromString(const_cast<LPWSTR> (clsidString),&clsid);
218     else hr = CLSIDFromProgID(const_cast<LPWSTR> (clsidString),&clsid);
219     if(FAILED(hr)) return hr;
220     return CreateInstance(clsid,pOuter,dwClsContext);
221   }
222   HRESULT CreateInstance(LPCSTR clsidStringA,IUnknown *pOuter = NULL,DWORD dwClsContext = CLSCTX_ALL) throw() {
223     if(!clsidStringA) return E_INVALIDARG;
224     int size = lstrlenA(clsidStringA) + 1;
225     int destSize = MultiByteToWideChar(CP_ACP,0,clsidStringA,size,NULL,0);
226     if(destSize==0) return HRESULT_FROM_WIN32(GetLastError());
227     LPWSTR clsidStringW;
228     clsidStringW = static_cast<LPWSTR>(_malloca(destSize*sizeof(WCHAR)));
229     if(!clsidStringW) return E_OUTOFMEMORY;
230     if(MultiByteToWideChar(CP_ACP,0,clsidStringA,size,clsidStringW,destSize)==0) {
231       _freea(clsidStringW);
232       return HRESULT_FROM_WIN32(GetLastError());
233     }
234     HRESULT hr=CreateInstance(clsidStringW,pOuter,dwClsContext);
235     _freea(clsidStringW);
236     return hr;
237   }
238   HRESULT GetActiveObject(const CLSID &rclsid) throw() {
239     _Release();
240     IUnknown *pIUnknown;
241     HRESULT hr = ::GetActiveObject(rclsid,NULL,&pIUnknown);
242     if(SUCCEEDED(hr)) {
243       hr = pIUnknown->QueryInterface(GetIID(),reinterpret_cast<void**>(&m_pInterface));
244       pIUnknown->Release();
245     }
246     if(FAILED(hr)) m_pInterface = NULL;
247     return hr;
248   }
249   HRESULT GetActiveObject(LPCWSTR clsidString) throw() {
250     if(!clsidString) return E_INVALIDARG;
251     CLSID clsid;
252     HRESULT hr;
253     if(clsidString[0]=='{') hr = CLSIDFromString(const_cast<LPWSTR> (clsidString),&clsid);
254     else hr = CLSIDFromProgID(const_cast<LPWSTR> (clsidString),&clsid);
255     if(FAILED(hr)) return hr;
256     return GetActiveObject(clsid);
257   }
258   HRESULT GetActiveObject(LPCSTR clsidStringA) throw() {
259     if(!clsidStringA) return E_INVALIDARG;
260     int size = lstrlenA(clsidStringA) + 1;
261     int destSize = MultiByteToWideChar(CP_ACP,0,clsidStringA,size,NULL,0);
262     LPWSTR clsidStringW;
263     try {
264       clsidStringW = static_cast<LPWSTR>(_alloca(destSize*sizeof(WCHAR)));
265     } catch (...) {
266       clsidStringW = NULL;
267     }
268     if(!clsidStringW) return E_OUTOFMEMORY;
269     if(MultiByteToWideChar(CP_ACP,0,clsidStringA,size,clsidStringW,destSize)==0) return HRESULT_FROM_WIN32(GetLastError());
270     return GetActiveObject(clsidStringW);
271   }
272   template<typename _InterfaceType> HRESULT QueryInterface(const IID& iid,_InterfaceType*& p) throw () {
273     if(m_pInterface!=NULL) return m_pInterface->QueryInterface(iid,reinterpret_cast<void**>(&p));
274     return E_POINTER;
275   }
276   template<typename _InterfaceType> HRESULT QueryInterface(const IID& iid,_InterfaceType **p) throw() { return QueryInterface(iid,*p); }
277 private:
278   Interface *m_pInterface;
279   void _Release() throw() {
280     if(m_pInterface!=NULL) m_pInterface->Release();
281   }
282   void _AddRef() throw() {
283     if(m_pInterface!=NULL) m_pInterface->AddRef();
284   }
285   template<typename _InterfacePtr> HRESULT _QueryInterface(_InterfacePtr p) throw() {
286     HRESULT hr;
287     if(p!=NULL) {
288       Interface *pInterface;
289       hr = p->QueryInterface(GetIID(),reinterpret_cast<void**>(&pInterface));
290       Attach(SUCCEEDED(hr)? pInterface: NULL);
291     } else {
292       operator=(static_cast<Interface*>(NULL));
293       hr = E_NOINTERFACE;
294     }
295     return hr;
296   }
297   template<typename _InterfacePtr> int _CompareUnknown(_InterfacePtr p) {
298     IUnknown *pu1,*pu2;
299     if(m_pInterface!=NULL) {
300       HRESULT hr = m_pInterface->QueryInterface(IID_IUnknown,reinterpret_cast<void**>(&pu1));
301       if(FAILED(hr)) {
302         _com_issue_error(hr);
303         pu1 = NULL;
304       } else pu1->Release();
305     } else pu1 = NULL;
306     if(p!=NULL) {
307       HRESULT hr = p->QueryInterface(IID_IUnknown,reinterpret_cast<void**>(&pu2));
308       if(FAILED(hr)) {
309         _com_issue_error(hr);
310         pu2 = NULL;
311       } else pu2->Release();
312     } else pu2 = NULL;
313     return pu1 - pu2;
314   }
315   HRESULT QueryStdInterfaces(const _variant_t& varSrc) throw() {
316     if(V_VT(&varSrc)==VT_DISPATCH) return _QueryInterface(V_DISPATCH(&varSrc));
317     if(V_VT(&varSrc)==VT_UNKNOWN) return _QueryInterface(V_UNKNOWN(&varSrc));
318     VARIANT varDest;
319     VariantInit(&varDest);
320     HRESULT hr = VariantChangeType(&varDest,const_cast<VARIANT*>(static_cast<const VARIANT*>(&varSrc)),0,VT_DISPATCH);
321     if(SUCCEEDED(hr)) hr = _QueryInterface(V_DISPATCH(&varSrc));
322     if(hr==E_NOINTERFACE) {
323       VariantInit(&varDest);
324       hr = VariantChangeType(&varDest,const_cast<VARIANT*>(static_cast<const VARIANT*>(&varSrc)),0,VT_UNKNOWN);
325       if(SUCCEEDED(hr)) hr = _QueryInterface(V_UNKNOWN(&varSrc));
326     }
327     VariantClear(&varDest);
328     return hr;
329   }
330 };
331 
332 template<typename _InterfaceType> bool operator==(int null,_com_ptr_t<_InterfaceType> &p) {
333   if(null!=0) { _com_issue_error(E_POINTER); }
334   return !p;
335 }
336 
337 template<typename _Interface,typename _InterfacePtr> bool operator==(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p==i; }
338 
339 template<typename _Interface> bool operator!=(int null,_com_ptr_t<_Interface> &p) {
340   if(null!=0) { _com_issue_error(E_POINTER); }
341   return p!=NULL;
342 }
343 
344 template<typename _Interface,typename _InterfacePtr> bool operator!=(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p!=i; }
345 
346 template<typename _Interface> bool operator<(int null,_com_ptr_t<_Interface> &p) {
347   if(null!=0) { _com_issue_error(E_POINTER); }
348   return p>NULL;
349 }
350 
351 template<typename _Interface,typename _InterfacePtr> bool operator<(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p>i; }
352 
353 template<typename _Interface> bool operator>(int null,_com_ptr_t<_Interface> &p) {
354   if(null!=0) { _com_issue_error(E_POINTER); }
355   return p<NULL;
356 }
357 
358 template<typename _Interface,typename _InterfacePtr> bool operator>(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p<i; }
359 
360 template<typename _Interface> bool operator<=(int null,_com_ptr_t<_Interface> &p) {
361   if(null!=0) { _com_issue_error(E_POINTER); }
362   return p>=NULL;
363 }
364 
365 template<typename _Interface,typename _InterfacePtr> bool operator<=(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p>=i; }
366 
367 template<typename _Interface> bool operator>=(int null,_com_ptr_t<_Interface> &p) {
368   if(null!=0) { _com_issue_error(E_POINTER); }
369   return p<=NULL;
370 }
371 
372 template<typename _Interface,typename _InterfacePtr> bool operator>=(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p<=i; }
373 
374 #pragma pop_macro("new")
375 
376 #endif /* __cplusplus */
377 
378 #endif /* _INC_COMIP */
379