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
23class _com_error;
24
25#ifndef WINAPI
26#define WINAPI __stdcall
27#endif
28
29void WINAPI _com_issue_error(HRESULT);
30struct IUnknown;
31
32template<typename _Interface,const IID *_IID >
33class _com_IIID {
34public:
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
41template<typename _IIID> class _com_ptr_t {
42public:
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,__uuidof(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); }
277private:
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(__uuidof(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(__uuidof(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
332template<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
337template<typename _Interface,typename _InterfacePtr> bool operator==(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p==i; }
338
339template<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
344template<typename _Interface,typename _InterfacePtr> bool operator!=(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p!=i; }
345
346template<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
351template<typename _Interface,typename _InterfacePtr> bool operator<(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p>i; }
352
353template<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
358template<typename _Interface,typename _InterfacePtr> bool operator>(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p<i; }
359
360template<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
365template<typename _Interface,typename _InterfacePtr> bool operator<=(_Interface *i,_com_ptr_t<_InterfacePtr> &p) { return p>=i; }
366
367template<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
372template<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