MalwareSourceCode/Win32/Proof of Concepts/herpaderping/ext/submodules/wil/com.h

2833 lines
125 KiB
C
Raw Normal View History

2022-04-12 01:00:13 +00:00
//*********************************************************
//
// Copyright (c) Microsoft. All rights reserved.
// This code is licensed under the MIT License.
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF
// ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
// TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
// PARTICULAR PURPOSE AND NONINFRINGEMENT.
//
//*********************************************************
#ifndef __WIL_COM_INCLUDED
#define __WIL_COM_INCLUDED
#include <WeakReference.h>
#include <combaseapi.h>
#include "result.h"
#include "resource.h" // last to ensure _COMBASEAPI_H_ protected definitions are available
// Forward declaration within WIL (see https://msdn.microsoft.com/en-us/library/br244983.aspx)
/// @cond
namespace Microsoft
{
namespace WRL
{
template <typename T>
class ComPtr;
}
}
/// @endcond
namespace wil
{
/// @cond
namespace details
{
// We can't directly use wistd::is_convertible as it returns TRUE for an ambiguous conversion.
// Adding is_abstract to the mix, enables us to allow conversion for interfaces, but deny it for
// classes (where the multiple inheritance causes ambiguity).
// NOTE: I've reached out to vcsig on this topic and it turns out that __is_convertible_to should NEVER
// return true for ambiguous conversions. This was a bug in our compiler that has since been fixed.
// Eventually, once that fix propagates we can move to a more efficient __is_convertible_to without
// the added complexity.
template <class TFrom, class TTo>
struct is_com_convertible :
wistd::bool_constant<__is_convertible_to(TFrom, TTo) && (__is_abstract(TFrom) || wistd::is_same<TFrom, TTo>::value)>
{
};
typedef wistd::integral_constant<char, 0> tag_com_query;
typedef wistd::integral_constant<char, 1> tag_try_com_query;
typedef wistd::integral_constant<char, 2> tag_com_copy;
typedef wistd::integral_constant<char, 3> tag_try_com_copy;
class default_query_policy
{
public:
template <typename T>
inline static HRESULT query(_In_ T* ptr, REFIID riid, _COM_Outptr_ void** result)
{
return ptr->QueryInterface(riid, result);
}
template <typename T, typename TResult>
inline static HRESULT query(_In_ T* ptr, _COM_Outptr_ TResult** result)
{
return query_dispatch(ptr, typename details::is_com_convertible<T*, TResult*>::type(), result);
}
private:
template <typename T, typename TResult>
inline static HRESULT query_dispatch(_In_ T* ptr, wistd::true_type, _COM_Outptr_ TResult** result) // convertible
{
*result = ptr;
(*result)->AddRef();
return S_OK;
}
template <typename T, typename TResult>
inline static HRESULT query_dispatch(_In_ T* ptr, wistd::false_type, _COM_Outptr_ TResult** result) // not convertible
{
auto hr = ptr->QueryInterface(IID_PPV_ARGS(result));
__analysis_assume(SUCCEEDED(hr) || (*result == nullptr));
return hr;
}
};
template <typename T>
struct query_policy_helper
{
typedef default_query_policy type;
};
class weak_query_policy
{
public:
inline static HRESULT query(_In_ IWeakReference* ptr, REFIID riid, _COM_Outptr_ void** result)
{
WI_ASSERT_MSG(riid != __uuidof(IWeakReference), "Cannot resolve a weak reference to IWeakReference");
*result = nullptr;
IInspectable* temp;
HRESULT hr = ptr->Resolve(__uuidof(IInspectable), reinterpret_cast<IInspectable**>(&temp));
if (SUCCEEDED(hr))
{
if (temp == nullptr)
{
return E_NOT_SET;
}
hr = temp->QueryInterface(riid, result);
__analysis_assume(SUCCEEDED(hr) || (*result == nullptr));
temp->Release();
}
return hr;
}
template <typename TResult>
inline static HRESULT query(_In_ IWeakReference* ptr, _COM_Outptr_ TResult** result)
{
static_assert(!wistd::is_same<IWeakReference, TResult>::value, "Cannot resolve a weak reference to IWeakReference");
return query_dispatch(ptr, wistd::is_base_of<IInspectable, TResult>(), result);
}
private:
template <typename TResult>
static HRESULT query_dispatch(_In_ IWeakReference* ptr, wistd::true_type, _COM_Outptr_ TResult** result)
{
auto hr = ptr->Resolve(__uuidof(TResult), reinterpret_cast<IInspectable**>(result));
if (SUCCEEDED(hr) && (*result == nullptr))
{
hr = E_NOT_SET;
}
__analysis_assume(SUCCEEDED(hr) || (*result == nullptr));
return hr;
}
template <typename TResult>
static HRESULT query_dispatch(_In_ IWeakReference* ptr, wistd::false_type, _COM_Outptr_ TResult** result)
{
return query(ptr, IID_PPV_ARGS(result));
}
};
template <>
struct query_policy_helper<IWeakReference>
{
typedef weak_query_policy type;
};
#if (NTDDI_VERSION >= NTDDI_WINBLUE)
class agile_query_policy
{
public:
inline static HRESULT query(_In_ IAgileReference* ptr, REFIID riid, _COM_Outptr_ void** result)
{
WI_ASSERT_MSG(riid != __uuidof(IAgileReference), "Cannot resolve a agile reference to IAgileReference");
auto hr = ptr->Resolve(riid, result);
__analysis_assume(SUCCEEDED(hr) || (*result == nullptr)); // IAgileReference::Resolve not annotated correctly
return hr;
}
template <typename TResult>
static HRESULT query(_In_ IAgileReference* ptr, _COM_Outptr_ TResult** result)
{
static_assert(!wistd::is_same<IAgileReference, TResult>::value, "Cannot resolve a agile reference to IAgileReference");
return query(ptr, __uuidof(TResult), reinterpret_cast<void**>(result));
}
};
template <>
struct query_policy_helper<IAgileReference>
{
typedef agile_query_policy type;
};
#endif
template <typename T>
using query_policy_t = typename query_policy_helper<typename wistd::remove_pointer<T>::type>::type;
} // details
/// @endcond
//! Represents the base template type that implements com_ptr, com_weak_ref, and com_agile_ref.
//! See @ref page_comptr for more background. See @ref page_query for more information on querying with WIL.
//! @tparam T Represents the type being held by the com_ptr_t.
//! For com_ptr, this will always be the interface being represented. For com_weak_ref, this will always be
//! IWeakReference. For com_agile_ref, this will always be IAgileReference.
//! @tparam err_policy Represents the error policy for the class (error codes, exceptions, or fail fast; see @ref page_errors)
template <typename T, typename err_policy = err_exception_policy>
class com_ptr_t
{
private:
typedef typename wistd::add_lvalue_reference<T>::type element_type_reference;
typedef details::query_policy_t<T> query_policy;
public:
//! The function return result (HRESULT or void) for the given err_policy (see @ref page_errors).
typedef typename err_policy::result result;
//! The template type `T` being held by the com_ptr_t.
typedef T element_type;
//! A pointer to the template type `T` being held by the com_ptr_t (what `get()` returns).
typedef T* pointer;
//! @name Constructors
//! @{
//! Default constructor (holds nullptr).
com_ptr_t() WI_NOEXCEPT :
m_ptr(nullptr)
{
}
//! Implicit construction from nullptr_t (holds nullptr).
com_ptr_t(wistd::nullptr_t) WI_NOEXCEPT :
com_ptr_t()
{
}
//! Implicit construction from a compatible raw interface pointer (AddRef's the parameter).
com_ptr_t(pointer ptr) WI_NOEXCEPT :
m_ptr(ptr)
{
if (m_ptr)
{
m_ptr->AddRef();
}
}
//! Copy-construction from a like `com_ptr_t` (copies and AddRef's the parameter).
com_ptr_t(const com_ptr_t& other) WI_NOEXCEPT :
com_ptr_t(other.get())
{
}
//! Copy-construction from a convertible `com_ptr_t` (copies and AddRef's the parameter).
template <class U, typename err, class = wistd::enable_if_t<__is_convertible_to(U*, pointer)>>
com_ptr_t(const com_ptr_t<U, err>& other) WI_NOEXCEPT :
com_ptr_t(static_cast<pointer>(other.get()))
{
}
//! Move construction from a like `com_ptr_t` (avoids AddRef/Release by moving from the parameter).
com_ptr_t(com_ptr_t&& other) WI_NOEXCEPT :
m_ptr(other.detach())
{
}
//! Move construction from a compatible `com_ptr_t` (avoids AddRef/Release by moving from the parameter).
template <class U, typename err, class = wistd::enable_if_t<__is_convertible_to(U*, pointer)>>
com_ptr_t(com_ptr_t<U, err>&& other) WI_NOEXCEPT :
m_ptr(other.detach())
{
}
//! @}
//! Destructor (releases the pointer).
~com_ptr_t() WI_NOEXCEPT
{
if (m_ptr)
{
m_ptr->Release();
}
}
//! @name Assignment operators
//! @{
//! Assign to nullptr (releases the current pointer, holds nullptr).
com_ptr_t& operator=(wistd::nullptr_t) WI_NOEXCEPT
{
reset();
return *this;
}
//! Assign a compatible raw interface pointer (releases current pointer, copies and AddRef's the parameter).
com_ptr_t& operator=(pointer other) WI_NOEXCEPT
{
auto ptr = m_ptr;
m_ptr = other;
if (m_ptr)
{
m_ptr->AddRef();
}
if (ptr)
{
ptr->Release();
}
return *this;
}
//! Assign a like `com_ptr_t` (releases current pointer, copies and AddRef's the parameter).
com_ptr_t& operator=(const com_ptr_t& other) WI_NOEXCEPT
{
return operator=(other.get());
}
//! Assign a convertible `com_ptr_t` (releases current pointer, copies and AddRef's the parameter).
template <class U, typename err, class = wistd::enable_if_t<__is_convertible_to(U*, pointer)>>
com_ptr_t& operator=(const com_ptr_t<U, err>& other) WI_NOEXCEPT
{
return operator=(static_cast<pointer>(other.get()));
}
//! Move assign from a like `com_ptr_t` (releases current pointer, avoids AddRef/Release by moving the parameter).
com_ptr_t& operator=(com_ptr_t&& other) WI_NOEXCEPT
{
attach(other.detach());
return *this;
}
//! Move assignment from a compatible `com_ptr_t` (releases current pointer, avoids AddRef/Release by moving from the parameter).
template <class U, typename err, class = wistd::enable_if_t<__is_convertible_to(U*, pointer)>>
com_ptr_t& operator=(com_ptr_t<U, err>&& other) WI_NOEXCEPT
{
attach(other.detach());
return *this;
}
//! @}
//! @name Modifiers
//! @{
//! Swap pointers with an another named com_ptr_t object.
template <typename err>
void swap(com_ptr_t<T, err>& other) WI_NOEXCEPT
{
auto ptr = m_ptr;
m_ptr = other.m_ptr;
other.m_ptr = ptr;
}
//! Swap pointers with a rvalue reference to another com_ptr_t object.
template <typename err>
void swap(com_ptr_t<T, err>&& other) WI_NOEXCEPT
{
swap(other);
}
//! Releases the pointer and sets it to nullptr.
void reset() WI_NOEXCEPT
{
auto ptr = m_ptr;
m_ptr = nullptr;
if (ptr)
{
ptr->Release();
}
}
//! Releases the pointer and sets it to nullptr.
void reset(wistd::nullptr_t) WI_NOEXCEPT
{
reset();
}
//! Takes ownership of a compatible raw interface pointer (releases pointer, copies but DOES NOT AddRef the parameter).
void attach(pointer other) WI_NOEXCEPT
{
auto ptr = m_ptr;
m_ptr = other;
if (ptr)
{
ULONG ref;
ref = ptr->Release();
WI_ASSERT_MSG(((other != ptr) || (ref > 0)), "Bug: Attaching the same already assigned, destructed pointer");
}
}
//! Relinquishes ownership and returns the internal interface pointer (DOES NOT release the detached pointer, sets class pointer to null).
WI_NODISCARD pointer detach() WI_NOEXCEPT
{
auto temp = m_ptr;
m_ptr = nullptr;
return temp;
}
//! Returns the address of the internal pointer (releases ownership of the pointer BEFORE returning the address).
//! The pointer is explicitly released to prevent accidental leaks of the pointer. Coding standards generally indicate that
//! there is little valid `_Inout_` use of `IInterface**`, making this safe to do under typical use.
//! @see addressof
//! ~~~~
//! STDAPI GetMuffin(IMuffin **muffin);
//! wil::com_ptr<IMuffin> myMuffin;
//! THROW_IF_FAILED(GetMuffin(myMuffin.put()));
//! ~~~~
pointer* put() WI_NOEXCEPT
{
reset();
return &m_ptr;
}
//! Returns the address of the internal pointer casted to void** (releases ownership of the pointer BEFORE returning the address).
//! @see put
void** put_void() WI_NOEXCEPT
{
return reinterpret_cast<void**>(put());
}
//! Returns the address of the internal pointer casted to IUnknown** (releases ownership of the pointer BEFORE returning the address).
//! @see put
::IUnknown** put_unknown() WI_NOEXCEPT
{
return reinterpret_cast<::IUnknown**>(put());
}
//! Returns the address of the internal pointer (releases ownership of the pointer BEFORE returning the address).
//! The pointer is explicitly released to prevent accidental leaks of the pointer. Coding standards generally indicate that
//! there is little valid `_Inout_` use of `IInterface**`, making this safe to do under typical use. Since this behavior is not always immediately
//! apparent, prefer to scope variables as close to use as possible (generally avoiding use of the same com_ptr variable in successive calls to
//! receive an output interface).
//! @see addressof
pointer* operator&() WI_NOEXCEPT
{
return put();
}
//! Returns the address of the internal pointer (does not release the pointer; should not be used for `_Out_` parameters)
pointer* addressof() WI_NOEXCEPT
{
return &m_ptr;
}
//! @}
//! @name Inspection
//! @{
//! Returns the address of the const internal pointer (does not release the pointer)
const pointer* addressof() const WI_NOEXCEPT
{
return &m_ptr;
}
//! Returns 'true' if the pointer is assigned (NOT nullptr)
explicit operator bool() const WI_NOEXCEPT
{
return (m_ptr != nullptr);
}
//! Returns the pointer
pointer get() const WI_NOEXCEPT
{
return m_ptr;
}
//! Allows direct calls against the pointer (AV on internal nullptr)
pointer operator->() const WI_NOEXCEPT
{
return m_ptr;
}
//! Dereferences the pointer (AV on internal nullptr)
element_type_reference operator*() const WI_NOEXCEPT
{
return *m_ptr;
}
//! @}
//! @name Query helpers
//! * Retrieves the requested interface
//! * AV if the pointer is null
//! * Produce an error if the requested interface is unsupported
//!
//! See @ref page_query for more information
//! @{
//! Query and return a smart pointer matching the interface specified by 'U': `auto foo = m_ptr.query<IFoo>();`.
//! See @ref page_query for more information.
//!
//! This method is the primary method that should be used to query a com_ptr in exception-based or fail-fast based code.
//! Error-code returning code should use @ref query_to so that the returned HRESULT can be examined. In the following
//! examples, `m_ptr` is an exception-based or fail-fast based com_ptr, com_weak_ref, or com_agile_ref:
//! ~~~~
//! auto foo = ptr.query<IFoo>();
//! foo->Method1();
//! foo->Method2();
//! ~~~~
//! For simple single-method calls, this method allows removing the temporary that holds the com_ptr:
//! ~~~~
//! ptr.query<IFoo>()->Method1();
//! ~~~~
//! @tparam U Represents the interface being queried
//! @return A `com_ptr_t` pointer to the given interface `U`. The pointer is guaranteed not null. The returned
//! `com_ptr_t` type will be @ref com_ptr or @ref com_ptr_failfast (matching the error handling form of the
//! pointer being queried (exception based or fail-fast).
template <class U>
inline com_ptr_t<U, err_policy> query() const
{
static_assert(wistd::is_same<void, result>::value, "query requires exceptions or fail fast; use try_query or query_to");
return com_ptr_t<U, err_policy>(m_ptr, details::tag_com_query());
}
//! Query for the interface of the given out parameter `U`: `ptr.query_to(&foo);`.
//! See @ref page_query for more information.
//!
//! For fail-fast and exception-based behavior this routine should primarily be used to write to out parameters and @ref query should
//! be used to perform most queries. For error-code based code, this routine is the primary method that should be used to query a com_ptr.
//!
//! Error-code based samples:
//! ~~~~
//! // class member being queried:
//! wil::com_ptr_nothrow<IUnknown> m_ptr;
//!
//! // simple query example:
//! wil::com_ptr_nothrow<IFoo> foo;
//! RETURN_IF_FAILED(m_ptr.query_to(&foo));
//! foo->FooMethod1();
//!
//! // output parameter example:
//! HRESULT GetFoo(_COM_Outptr_ IFoo** fooPtr)
//! {
//! RETURN_IF_FAILED(m_ptr.query_to(fooPtr));
//! return S_OK;
//! }
//! ~~~~
//! Exception or fail-fast samples:
//! ~~~~
//! // class member being queried
//! wil::com_ptr<IUnknown> m_ptr;
//!
//! void GetFoo(_COM_Outptr_ IFoo** fooPtr)
//! {
//! m_ptr.query_to(fooPtr);
//! }
//! ~~~~
//! @tparam U Represents the interface being queried (type of the output parameter). This interface does not need to
//! be specified directly. Rely upon template type deduction to pick up the type from the output parameter.
//! @param ptrResult The output pointer that will receive the newly queried interface. This pointer will be assigned null on failure.
//! @return For the nothrow (error code-based) classes (@ref com_ptr_nothrow, @ref com_weak_ref_nothrow, @ref com_agile_ref_nothrow) this
//! method returns an `HRESULT` indicating whether the query was successful. Exception-based and fail-fast based classes
//! do not return a value (void).
template <class U>
result query_to(_COM_Outptr_ U** ptrResult) const
{
// Prefast cannot see through the error policy + query_policy mapping and as a result fires 6388 and 28196 for this function.
// Suppression is also not working. Wrapping this entire function in #pragma warning(disable: 6388 28196) does not stop all of the prefast errors
// from being emitted.
#if defined(_PREFAST_)
*ptrResult = nullptr;
return err_policy::HResult(E_NOINTERFACE);
#else
return err_policy::HResult(query_policy::query(m_ptr, ptrResult));
#endif
}
//! Query for the requested interface using the iid, ppv pattern: `ptr.query_to(riid, ptr);`.
//! See @ref page_query for more information.
//!
//! This method is built to implement an API boundary that exposes a returned pointer to a caller through the REFIID and void** pointer
//! pattern (like QueryInterface). This pattern should not be used outside of that pattern (through IID_PPV_ARGS) as it is less efficient
//! than the typed version of @ref query_to which can elide the QueryInterface in favor of AddRef when the types are convertible.
//! ~~~~
//! // class member being queried:
//! wil::com_ptr_nothrow<IUnknown> m_ptr;
//!
//! // output parameter example:
//! HRESULT GetFoo(REFIID riid, _COM_Outptr_ void** ptrResult)
//! {
//! RETURN_IF_FAILED(m_ptr.query_to(riid, ptrResult));
//! return S_OK;
//! }
//! ~~~~
//! @param riid The interface to query for.
//! @param ptrResult The output pointer that will receive the newly queried interface. This pointer will be assigned null on failure.
//! @return For the nothrow (error code-based) classes (@ref com_ptr_nothrow, @ref com_weak_ref_nothrow, @ref com_agile_ref_nothrow) this
//! method returns an `HRESULT` indicating whether the query was successful. Exception-based and fail-fast based classes
//! do not return a value (void).
result query_to(REFIID riid, _COM_Outptr_ void** ptrResult) const
{
// Prefast cannot see through the error policy + query_policy mapping and as a result and as a result fires 6388 and 28196 for this function.
// Suppression is also not working. Wrapping this entire function in #pragma warning(disable: 6388 28196) does not stop the prefast errors
// from being emitted.
#if defined(_PREFAST_)
*ptrResult = nullptr;
return err_policy::HResult(E_NOINTERFACE);
#else
return err_policy::HResult(query_policy::query(m_ptr, riid, ptrResult));
#endif
}
//! @}
//! @name Try query helpers
//! * Attempts to retrieves the requested interface
//! * AV if the pointer is null
//! * Produce null if the requested interface is unsupported
//! * bool returns 'true' when query was successful
//!
//! See @ref page_query for more information.
//! @{
//! Attempt a query and return a smart pointer matching the interface specified by 'U': `auto foo = m_ptr.try_query<IFoo>();` (null result when interface is unsupported).
//! See @ref page_query for more information.
//!
//! This method can be used to query a com_ptr for an interface when it's known that support for that interface is
//! optional (failing the query should not produce an error). The caller must examine the returned pointer to see
//! if it's null before using it:
//! ~~~~
//! auto foo = ptr.try_query<IFoo>();
//! if (foo)
//! {
//! foo->Method1();
//! foo->Method2();
//! }
//! ~~~~
//! @tparam U Represents the interface being queried
//! @return A `com_ptr_t` pointer to the given interface `U`. The returned pointer will be null if the interface is
//! not supported. The returned `com_ptr_t` will have the same error handling policy (exceptions, failfast or error codes) as
//! the pointer being queried.
template <class U>
inline com_ptr_t<U, err_policy> try_query() const
{
return com_ptr_t<U, err_policy>(m_ptr, details::tag_try_com_query());
}
//! Attempts to query for the interface matching the given output parameter; returns a bool indicating if the query was successful (non-null).
//! See @ref page_query for more information.
//!
//! This method can be used to perform a query against a non-null interface when it's known that support for that interface is
//! optional (failing the query should not produce an error). The caller must examine the returned bool before using the returned pointer.
//! ~~~~
//! wil::com_ptr_nothrow<IFoo> foo;
//! if (ptr.try_query_to(&foo))
//! {
//! foo->Method1();
//! foo->Method2();
//! }
//! ~~~~
//! @param ptrResult The pointer to query for. The interface to query is deduced from the type of this out parameter; do not specify
//! the type directly to the template.
//! @return A `bool` indicating `true` of the query was successful (the returned parameter is non-null).
template <class U>
_Success_return_ bool try_query_to(_COM_Outptr_ U** ptrResult) const
{
return SUCCEEDED(query_policy::query(m_ptr, ptrResult));
}
//! Attempts a query for the requested interface using the iid, ppv pattern: `ptr.try_query_to(riid, ptr);`.
//! See @ref page_query for more information.
//!
//! This method is built to implement an API boundary that exposes a returned pointer to a caller through the REFIID and void** pointer
//! pattern (like QueryInterface). The key distinction is that this routine does not produce an error if the request isn't fulfilled, so
//! it's appropriate for `_COM_Outptr_result_maybenull_` cases. This pattern should not be used outside of that pattern (through IID_PPV_ARGS) as
//! it is less efficient than the typed version of @ref try_query_to which can elide the QueryInterface in favor of AddRef when the types are convertible.
//! The caller must examine the returned bool before using the returned pointer.
//! ~~~~
//! // class member being queried:
//! wil::com_ptr_nothrow<IUnknown> m_ptr;
//!
//! // output parameter example (result may be null):
//! HRESULT GetFoo(REFIID riid, _COM_Outptr_result_maybenull_ void** ptrResult)
//! {
//! m_ptr.try_query_to(riid, ptrResult);
//! return S_OK;
//! }
//! ~~~~
//! @param riid The interface to query for.
//! @param ptrResult The output pointer that will receive the newly queried interface. This pointer will be assigned null on failure.
//! @return A `bool` indicating `true` of the query was successful (the returned parameter is non-null).
_Success_return_ bool try_query_to(REFIID riid, _COM_Outptr_ void** ptrResult) const
{
return SUCCEEDED(query_policy::query(m_ptr, riid, ptrResult));
}
//! @}
//! @name Copy helpers
//! * Retrieves the requested interface
//! * Succeeds with null if the pointer is null
//! * Produce an error if the requested interface is unsupported
//!
//! See @ref page_query for more information.
//! @{
//! Query and return a smart pointer matching the interface specified by 'U': `auto foo = m_ptr.copy<IFoo>();` (succeeds and returns a null ptr if the queried pointer is null).
//! See @ref page_query for more information.
//!
//! This method is identical to @ref query with the exception that it can be used when the pointer is null. When used
//! against a null pointer, the returned pointer will always be null and an error will not be produced. Like query it will
//! produce an error for a non-null pointer that does not support the requested interface.
//! @tparam U Represents the interface being queried
//! @return A `com_ptr_t` pointer to the given interface `U`. The pointer will be null ONLY if the pointer being queried is null. The returned
//! `com_ptr_t` type will be @ref com_ptr or @ref com_ptr_failfast (matching the error handling form of the
//! pointer being queried (exception based or fail-fast).
template <class U>
inline com_ptr_t<U, err_policy> copy() const
{
static_assert(wistd::is_same<void, result>::value, "copy requires exceptions or fail fast; use the try_copy or copy_to method");
return com_ptr_t<U, err_policy>(m_ptr, details::tag_com_copy());
}
//! Query for the interface of the given out parameter `U`: `ptr.copy_to(&foo);` (succeeds and returns null ptr if the queried pointer is null).
//! See @ref page_query for more information.
//!
//! This method is identical to @ref query_to with the exception that it can be used when the pointer is null. When used
//! against a null pointer, the returned pointer will always be null and an error will not be produced. Like query_to it will
//! produce an error for a non-null pointer that does not support the requested interface.
//! @tparam U Represents the interface being queried (type of the output parameter). This interface does not need to
//! be specified directly. Rely upon template type deduction to pick up the type from the output parameter.
//! @param ptrResult The output pointer that will receive the newly queried interface. This pointer will be assigned null on failure OR assigned null
//! when the source pointer is null.
//! @return For the nothrow (error code-based) classes (@ref com_ptr_nothrow, @ref com_weak_ref_nothrow, @ref com_agile_ref_nothrow) this
//! method returns an `HRESULT` indicating whether the query was successful. Copying a null value is considered success. Exception-based
//! and fail-fast based classes do not return a value (void).
template <class U>
result copy_to(_COM_Outptr_result_maybenull_ U** ptrResult) const
{
if (m_ptr)
{
// Prefast cannot see through the error policy + query_policy mapping and as a result and as a result fires 6388 and 28196 for this function.
// Suppression is also not working. Wrapping this entire function in #pragma warning(disable: 6388 28196) does not stop the prefast errors
// from being emitted.
#if defined(_PREFAST_)
*ptrResult = nullptr;
return err_policy::HResult(E_NOINTERFACE);
#else
return err_policy::HResult(query_policy::query(m_ptr, ptrResult));
#endif
}
*ptrResult = nullptr;
return err_policy::OK();
}
//! Query for the requested interface using the iid, ppv pattern: `ptr.copy_to(riid, ptr);`. (succeeds and returns null ptr if the queried pointer is null).
//! See @ref page_query for more information.
//!
//! Identical to the corresponding @ref query_to method with the exception that it can be used when the pointer is null. When used
//! against a null pointer, the returned pointer will always be null and an error will not be produced. Like query_to it will
//! produce an error for a non-null pointer that does not support the requested interface.
//! @param riid The interface to query for.
//! @param ptrResult The output pointer that will receive the newly queried interface. This pointer will be assigned null on failure OR assigned null
//! when the source pointer is null.
//! @return For the nothrow (error code-based) classes (@ref com_ptr_nothrow, @ref com_weak_ref_nothrow, @ref com_agile_ref_nothrow) this
//! method returns an `HRESULT` indicating whether the query was successful. Copying a null value is considered success. Exception-based
//! and fail-fast based classes do not return a value (void).
result copy_to(REFIID riid, _COM_Outptr_result_maybenull_ void** ptrResult) const
{
if (m_ptr)
{
// Prefast cannot see through the error policy + query_policy mapping and as a result and as a result fires 6388 and 28196 for this function.
// Suppression is also not working. Wrapping this entire function in #pragma warning(disable: 6388 28196) does not stop the prefast errors
// from being emitted.
#if defined(_PREFAST_)
*ptrResult = nullptr;
return err_policy::HResult(E_NOINTERFACE);
#else
return err_policy::HResult(query_policy::query(m_ptr, riid, ptrResult));
#endif
}
*ptrResult = nullptr;
return err_policy::OK();
}
//! @}
//! @name Try copy helpers
//! * Attempts to retrieves the requested interface
//! * Successfully produces null if the queried pointer is already null
//! * Produce null if the requested interface is unsupported
//! * bool returns 'false' ONLY when the queried pointer is not null and the requested interface is unsupported
//!
//! See @ref page_query for more information.
//! @{
//! Attempt a query and return a smart pointer matching the interface specified by 'U': `auto foo = m_ptr.try_query<IFoo>();` (null result when interface is unsupported or queried pointer is null).
//! See @ref page_query for more information.
//!
//! Identical to the corresponding @ref try_query method with the exception that it can be used when the pointer is null. When used
//! against a null pointer, the returned pointer will always be null and an error will not be produced.
//! @tparam U Represents the interface being queried
//! @return A `com_ptr_t` pointer to the given interface `U`. The returned pointer will be null if the interface was
//! not supported or the pointer being queried is null. The returned `com_ptr_t` will have the same error handling
//! policy (exceptions, failfast or error codes) as the pointer being queried.
template <class U>
inline com_ptr_t<U, err_policy> try_copy() const
{
return com_ptr_t<U, err_policy>(m_ptr, details::tag_try_com_copy());
}
//! Attempts to query for the interface matching the given output parameter; returns a bool indicating if the query was successful (returns `false` if the pointer is null).
//! See @ref page_query for more information.
//!
//! Identical to the corresponding @ref try_query_to method with the exception that it can be used when the pointer is null. When used
//! against a null pointer, the returned pointer will be null and the return value will be `false`.
//! @param ptrResult The pointer to query for. The interface to query is deduced from the type of this out parameter; do not specify
//! the type directly to the template.
//! @return A `bool` indicating `true` of the query was successful (the returned parameter is non-null).
template <class U>
_Success_return_ bool try_copy_to(_COM_Outptr_result_maybenull_ U** ptrResult) const
{
if (m_ptr)
{
return SUCCEEDED(query_policy::query(m_ptr, ptrResult));
}
*ptrResult = nullptr;
return false;
}
//! Attempts a query for the requested interface using the iid, ppv pattern: `ptr.try_query_to(riid, ptr);` (returns `false` if the pointer is null)
//! See @ref page_query for more information.
//!
//! Identical to the corresponding @ref try_query_to method with the exception that it can be used when the pointer is null. When used
//! against a null pointer, the returned pointer will be null and the return value will be `false`.
//! @param riid The interface to query for.
//! @param ptrResult The output pointer that will receive the newly queried interface. This pointer will be assigned null on failure or
//! if the source pointer being queried is null.
//! @return A `bool` indicating `true` of the query was successful (the returned parameter is non-null). Querying a null
//! pointer will return `false` with a null result.
_Success_return_ bool try_copy_to(REFIID riid, _COM_Outptr_result_maybenull_ void** ptrResult) const
{
if (m_ptr)
{
return SUCCEEDED(query_policy::query(m_ptr, riid, ptrResult));
}
*ptrResult = nullptr;
return false;
}
//! @}
//! @name WRL compatibility
//! @{
//! Copy construct from a compatible WRL ComPtr<T>.
template <class U, class = wistd::enable_if_t<__is_convertible_to(U*, pointer)>>
com_ptr_t(const Microsoft::WRL::ComPtr<U>& other) WI_NOEXCEPT :
com_ptr_t(static_cast<pointer>(other.Get()))
{
}
//! Move construct from a compatible WRL ComPtr<T>.
template <class U, class = wistd::enable_if_t<__is_convertible_to(U*, pointer)>>
com_ptr_t(Microsoft::WRL::ComPtr<U>&& other) WI_NOEXCEPT :
m_ptr(other.Detach())
{
}
//! Assign from a compatible WRL ComPtr<T>.
template <class U, class = wistd::enable_if_t<__is_convertible_to(U*, pointer)>>
com_ptr_t& operator=(const Microsoft::WRL::ComPtr<U>& other) WI_NOEXCEPT
{
return operator=(static_cast<pointer>(other.Get()));
}
//! Move assign from a compatible WRL ComPtr<T>.
template <class U, class = wistd::enable_if_t<__is_convertible_to(U*, pointer)>>
com_ptr_t& operator=(Microsoft::WRL::ComPtr<U>&& other) WI_NOEXCEPT
{
attach(other.Detach());
return *this;
}
//! Swap pointers with a WRL ComPtr<T> to the same interface.
void swap(Microsoft::WRL::ComPtr<T>& other) WI_NOEXCEPT
{
auto ptr = m_ptr;
m_ptr = other.Detach();
other.Attach(ptr);
}
//! Swap pointers with a rvalue reference to a WRL ComPtr<T> to the same interface.
void swap(Microsoft::WRL::ComPtr<T>&& other) WI_NOEXCEPT
{
swap(other);
}
//! @} // WRL compatibility
public:
// Internal Helpers
/// @cond
template <class U>
inline com_ptr_t(_In_ U* ptr, details::tag_com_query)
{
err_policy::HResult(details::query_policy_t<U>::query(ptr, &m_ptr));
}
template <class U>
inline com_ptr_t(_In_ U* ptr, details::tag_try_com_query) WI_NOEXCEPT : m_ptr(nullptr)
{
details::query_policy_t<U>::query(ptr, &m_ptr);
}
template <class U>
inline com_ptr_t(_In_opt_ U* ptr, details::tag_com_copy)
{
if (ptr)
{
err_policy::HResult(details::query_policy_t<U>::query(ptr, &m_ptr));
return;
}
m_ptr = nullptr;
}
template <class U>
inline com_ptr_t(_In_opt_ U* ptr, details::tag_try_com_copy) WI_NOEXCEPT : m_ptr(nullptr)
{
if (ptr)
{
details::query_policy_t<U>::query(ptr, &m_ptr);
}
}
/// @endcond
private:
pointer m_ptr;
};
// Error-policy driven forms of com_ptr
#ifdef WIL_ENABLE_EXCEPTIONS
//! COM pointer, errors throw exceptions (see @ref com_ptr_t for details)
template <typename T>
using com_ptr = com_ptr_t<T, err_exception_policy>;
#endif
//! COM pointer, errors return error codes (see @ref com_ptr_t for details)
template <typename T>
using com_ptr_nothrow = com_ptr_t<T, err_returncode_policy>;
//! COM pointer, errors fail-fast (see @ref com_ptr_t for details)
template <typename T>
using com_ptr_failfast = com_ptr_t<T, err_failfast_policy>;
// Global operators / swap
//! Swaps the given com pointers that have different error handling.
//! Note that there are also corresponding versions to allow you to swap any wil com_ptr<T> with a WRL ComPtr<T>.
template <typename T, typename ErrLeft, typename ErrRight>
inline void swap(com_ptr_t<T, ErrLeft>& left, com_ptr_t<T, ErrRight>& right) WI_NOEXCEPT
{
left.swap(right);
}
//! Swaps the given com pointers that have the same error handling.
template <typename T, typename Err>
inline void swap(com_ptr_t<T, Err>& left, com_ptr_t<T, Err>& right) WI_NOEXCEPT
{
left.swap(right);
}
//! Compare two com pointers.
//! Compares the two raw com pointers for equivalence. Does NOT compare object identity with a QI for IUnknown.
//!
//! Note that documentation for all of the various comparators has not been generated to reduce global function
//! clutter, but ALL standard comparison operators are supported between wil com_ptr<T> objects, nullptr_t, and
//! WRL ComPtr<T>.
template <typename TLeft, typename ErrLeft, typename TRight, typename ErrRight>
inline bool operator==(const com_ptr_t<TLeft, ErrLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left.get() == right.get());
}
// We don't document all of the global comparison operators (reduce clutter)
/// @cond
template <typename TLeft, typename ErrLeft, typename TRight, typename ErrRight>
inline bool operator<(const com_ptr_t<TLeft, ErrLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left.get() < right.get());
}
template <typename TLeft, typename ErrLeft>
inline bool operator==(const com_ptr_t<TLeft, ErrLeft>& left, wistd::nullptr_t) WI_NOEXCEPT
{
return (left.get() == nullptr);
}
template <typename TLeft, typename ErrLeft, typename TRight, typename ErrRight>
inline bool operator!=(const com_ptr_t<TLeft, ErrLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(left == right)); }
template <typename TLeft, typename ErrLeft, typename TRight, typename ErrRight>
inline bool operator>=(const com_ptr_t<TLeft, ErrLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(left < right)); }
template <typename TLeft, typename ErrLeft, typename TRight, typename ErrRight>
inline bool operator>(const com_ptr_t<TLeft, ErrLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (right < left); }
template <typename TLeft, typename ErrLeft, typename TRight, typename ErrRight>
inline bool operator<=(const com_ptr_t<TLeft, ErrLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(right < left)); }
template <typename TRight, typename ErrRight>
inline bool operator==(wistd::nullptr_t, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{
return (right.get() == nullptr);
}
template <typename TLeft, typename ErrLeft>
inline bool operator!=(const com_ptr_t<TLeft, ErrLeft>& left, wistd::nullptr_t) WI_NOEXCEPT
{ return (!(left == nullptr)); }
template <typename TRight, typename ErrRight>
inline bool operator!=(wistd::nullptr_t, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(right == nullptr)); }
// WRL ComPtr support
template <typename T, typename ErrLeft>
inline void swap(com_ptr_t<T, ErrLeft>& left, Microsoft::WRL::ComPtr<T>& right) WI_NOEXCEPT
{
left.swap(right);
}
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator==(const com_ptr_t<TLeft, ErrLeft>& left, const Microsoft::WRL::ComPtr<TRight>& right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left.get() == right.Get());
}
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator<(const com_ptr_t<TLeft, ErrLeft>& left, const Microsoft::WRL::ComPtr<TRight>& right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left.get() < right.Get());
}
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator!=(const com_ptr_t<TLeft, ErrLeft>& left, const Microsoft::WRL::ComPtr<TRight>& right) WI_NOEXCEPT
{ return (!(left == right)); }
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator>=(const com_ptr_t<TLeft, ErrLeft>& left, const Microsoft::WRL::ComPtr<TRight>& right) WI_NOEXCEPT
{ return (!(left < right)); }
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator>(const com_ptr_t<TLeft, ErrLeft>& left, const Microsoft::WRL::ComPtr<TRight>& right) WI_NOEXCEPT
{ return (right < left); }
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator<=(const com_ptr_t<TLeft, ErrLeft>& left, const Microsoft::WRL::ComPtr<TRight>& right) WI_NOEXCEPT
{ return (!(right < left)); }
template <typename T, typename ErrRight>
inline void swap(Microsoft::WRL::ComPtr<T>& left, com_ptr_t<T, ErrRight>& right) WI_NOEXCEPT
{
right.swap(left);
}
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator==(const Microsoft::WRL::ComPtr<TLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left.Get() == right.get());
}
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator<(const Microsoft::WRL::ComPtr<TLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left.Get() < right.get());
}
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator!=(const Microsoft::WRL::ComPtr<TLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(left == right)); }
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator>=(const Microsoft::WRL::ComPtr<TLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(left < right)); }
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator>(const Microsoft::WRL::ComPtr<TLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (right < left); }
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator<=(const Microsoft::WRL::ComPtr<TLeft>& left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(right < left)); }
// raw COM pointer support
//
// Use these for convenience and to avoid unnecessary AddRef/Release cyles when using raw
// pointers to access STL containers. Specify std::less<> to benefit from operator<.
//
// Example: std::set<wil::com_ptr<IUnknown>, std::less<>> set;
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator==(const com_ptr_t<TLeft, ErrLeft>& left, TRight* right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left.get() == right);
}
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator<(const com_ptr_t<TLeft, ErrLeft>& left, TRight* right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left.get() < right);
}
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator!=(const com_ptr_t<TLeft, ErrLeft>& left, TRight* right) WI_NOEXCEPT
{ return (!(left == right)); }
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator>=(const com_ptr_t<TLeft, ErrLeft>& left, TRight* right) WI_NOEXCEPT
{ return (!(left < right)); }
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator>(const com_ptr_t<TLeft, ErrLeft>& left, TRight* right) WI_NOEXCEPT
{ return (right < left); }
template <typename TLeft, typename ErrLeft, typename TRight>
inline bool operator<=(const com_ptr_t<TLeft, ErrLeft>& left, TRight* right) WI_NOEXCEPT
{ return (!(right < left)); }
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator==(TLeft* left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left == right.get());
}
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator<(TLeft* left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{
static_assert(__is_convertible_to(TLeft*, TRight*) || __is_convertible_to(TRight*, TLeft*), "comparison operator requires left and right pointers to be compatible");
return (left < right.get());
}
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator!=(TLeft* left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(left == right)); }
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator>=(TLeft* left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(left < right)); }
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator>(TLeft* left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (right < left); }
template <typename TLeft, typename TRight, typename ErrRight>
inline bool operator<=(TLeft* left, const com_ptr_t<TRight, ErrRight>& right) WI_NOEXCEPT
{ return (!(right < left)); }
// suppress documentation of every single comparison operator
/// @endcond
//! An overloaded function that retrieves the raw com pointer from a raw pointer, wil::com_ptr_t<T>, WRL ComPtr<T>, or Platform::Object^.
//! This function is primarily useful by library or helper code. It allows code to be written to accept a forwarding reference
//! template that can be used as an input com pointer. That input com pointer is allowed to be any of:
//! * Raw Pointer: `T* com_raw_ptr(T* ptr)`
//! * Wil com_ptr: `T* com_raw_ptr(const wil::com_ptr_t<T, err>& ptr)`
//! * WRL ComPtr: `T* com_raw_ptr(const Microsoft::WRL::ComPtr<T>& ptr)`
//! * C++/CX hat: `IInspectable* com_raw_ptr(Platform::Object^ ptr)`
//!
//! Which in turn allows code like the following to be written:
//! ~~~~
//! template <typename U, typename T>
//! void com_query_to(T&& ptrSource, _COM_Outptr_ U** ptrResult)
//! {
//! auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
//! // decltype(raw) has the type of the inner pointer and raw is guaranteed to be a raw com pointer
//! ~~~~
template <typename T>
T* com_raw_ptr(T* ptr)
{
return ptr;
}
/// @cond
template <typename T, typename err>
T* com_raw_ptr(const wil::com_ptr_t<T, err>& ptr)
{
return ptr.get();
}
template <typename T>
T* com_raw_ptr(const Microsoft::WRL::ComPtr<T>& ptr)
{
return ptr.Get();
}
#ifdef __cplusplus_winrt
template <typename T>
inline IInspectable* com_raw_ptr(T^ ptr)
{
return reinterpret_cast<IInspectable*>(static_cast<::Platform::Object^>(ptr));
}
#endif
/// @endcond
//! @name Stand-alone query helpers
//! * Source pointer can be raw interface pointer, any wil com_ptr, or WRL ComPtr
//! * Retrieves the requested interface
//! * AV if the source pointer is null
//! * Produce an error if the requested interface is unsupported
//!
//! See @ref page_query for more information
//! @{
#ifdef WIL_ENABLE_EXCEPTIONS
//! Queries for the specified interface and returns an exception-based wil::com_ptr to that interface (exception if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr<U>` pointer to the given interface `U`. The returned pointer is guaranteed not null.
template <typename U, typename T>
inline com_ptr<U> com_query(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr<U>(raw, details::tag_com_query());
}
#endif
//! Queries for the specified interface and returns a fail-fast-based wil::com_ptr_failfast to that interface (fail-fast if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr<U>` pointer to the given interface `U`. The returned pointer is guaranteed not null.
template <typename U, typename T>
inline com_ptr_failfast<U> com_query_failfast(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr_failfast<U>(raw, details::tag_com_query());
}
#ifdef WIL_ENABLE_EXCEPTIONS
//! Queries for the interface specified by the type of the output parameter (throws an exception if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @param ptrResult Represents the output pointer to populate. The returned pointer is guaranteed not null.
template <typename U, typename T>
_Success_true_ void com_query_to(T&& ptrSource, _COM_Outptr_ U** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
THROW_IF_FAILED(details::query_policy_t<decltype(raw)>::query(raw, ptrResult));
__analysis_assume(*ptrResult != nullptr);
}
#endif
//! Queries for the interface specified by the type of the output parameter (fail-fast if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @param ptrResult Represents the output pointer to populate. The returned pointer is guaranteed not null.
template <typename U, typename T>
_Success_true_ void com_query_to_failfast(T&& ptrSource, _COM_Outptr_ U** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
FAIL_FAST_IF_FAILED(details::query_policy_t<decltype(raw)>::query(raw, ptrResult));
__analysis_assume(*ptrResult != nullptr);
}
//! Queries for the interface specified by the type of the output parameter (returns an error if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @param ptrResult Represents the output pointer to populate. The returned pointer will be null upon failure.
//! @return Returns an HRESULT representing whether the query succeeded.
template <typename U, typename T>
HRESULT com_query_to_nothrow(T&& ptrSource, _COM_Outptr_ U** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
auto hr = details::query_policy_t<decltype(raw)>::query(raw, ptrResult);
__analysis_assume(SUCCEEDED(hr) || (*ptrResult == nullptr));
RETURN_HR(hr);
}
#ifdef WIL_ENABLE_EXCEPTIONS
//! Queries for the interface specified by the given REFIID parameter (throws an exception if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @param riid The interface to query for
//! @param ptrResult Represents the output pointer to populate. The returned pointer is guaranteed not null.
template <typename T>
_Success_true_ void com_query_to(T&& ptrSource, REFIID riid, _COM_Outptr_ void** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
THROW_IF_FAILED(details::query_policy_t<decltype(raw)>::query(raw, riid, ptrResult));
__analysis_assume(*ptrResult != nullptr);
}
#endif
//! Queries for the interface specified by the given REFIID parameter (fail-fast if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @param riid The interface to query for
//! @param ptrResult Represents the output pointer to populate. The returned pointer is guaranteed not null.
template <typename T>
_Success_true_ void com_query_to_failfast(T&& ptrSource, REFIID riid, _COM_Outptr_ void** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
FAIL_FAST_IF_FAILED(details::query_policy_t<decltype(raw)>::query(raw, riid, ptrResult));
__analysis_assume(*ptrResult != nullptr);
}
//! Queries for the interface specified by the given REFIID parameter (returns an error if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @param riid The interface to query for
//! @param ptrResult Represents the output pointer to populate. The returned pointer will be null upon failure.
template <typename T>
HRESULT com_query_to_nothrow(T&& ptrSource, REFIID riid, _COM_Outptr_ void** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
auto hr = details::query_policy_t<decltype(raw)>::query(raw, riid, ptrResult);
__analysis_assume(SUCCEEDED(hr) || (*ptrResult == nullptr));
RETURN_HR(hr);
}
//! @}
//! @name Stand-alone try query helpers
//! * Source pointer can be raw interface pointer, any wil com_ptr, or WRL ComPtr
//! * Attempts to retrieves the requested interface
//! * AV if the source pointer is null
//! * Produce null if the requested interface is unsupported
//! * bool returns 'true' when query was successful (non-null return result)
//!
//! See @ref page_query for more information.
//! @{
#ifdef WIL_ENABLE_EXCEPTIONS
//! Attempts a query for the specified interface and returns an exception-based wil::com_ptr to that interface (returns null if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr<U>` pointer to the given interface `U`. The returned pointer is null if the requested interface was not supported.
template <class U, typename T>
inline com_ptr<U> try_com_query(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr<U>(raw, details::tag_try_com_query());
}
#endif
//! Attempts a query for the specified interface and returns an fail-fast wil::com_ptr_failfast to that interface (returns null if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr_failfast<U>` pointer to the given interface `U`. The returned pointer is null if the requested interface was not supported.
template <class U, typename T>
inline com_ptr_failfast<U> try_com_query_failfast(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr_failfast<U>(raw, details::tag_try_com_query());
}
//! Attempts a query for the specified interface and returns an error-code-based wil::com_ptr_nothrow to that interface (returns null if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr_nothrow<U>` pointer to the given interface `U`. The returned pointer is null if the requested interface was not supported.
template <class U, typename T>
inline com_ptr_nothrow<U> try_com_query_nothrow(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr_nothrow<U>(raw, details::tag_try_com_query());
}
//! Attempts a query for the interface specified by the type of the output parameter (returns `false` if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null.
//! @param ptrResult Represents the output pointer to populate. If the interface is unsupported, the returned pointer will be null.
//! @return A bool value representing whether the query was successful (non-null return result).
template <typename U, typename T>
_Success_return_ bool try_com_query_to(T&& ptrSource, _COM_Outptr_ U** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return (SUCCEEDED(details::query_policy_t<decltype(raw)>::query(raw, ptrResult)));
}
//! Attempts a query for the interface specified by the type of the output parameter (returns `false` if unsupported).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), should not be null.
//! @param riid The interface to query for
//! @param ptrResult Represents the output pointer to populate. If the interface is unsupported, the returned pointer will be null.
//! @return A bool value representing whether the query was successful (non-null return result).
template <typename T>
_Success_return_ bool try_com_query_to(T&& ptrSource, REFIID riid, _COM_Outptr_ void** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return (SUCCEEDED(details::query_policy_t<decltype(raw)>::query(raw, riid, ptrResult)));
}
//! @}
//! @name Stand-alone copy helpers
//! * Source pointer can be raw interface pointer, any wil com_ptr, or WRL ComPtr
//! * Retrieves the requested interface
//! * Succeeds with null if the source pointer is null
//! * Produce an error if the requested interface is unsupported
//!
//! See @ref page_query for more information
//! @{
#ifdef WIL_ENABLE_EXCEPTIONS
//! Queries for the specified interface and returns an exception-based wil::com_ptr to that interface (exception if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr<U>` pointer to the given interface `U`. The returned pointer will be null only if the source is null.
template <class U, typename T>
inline com_ptr<U> com_copy(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr<U>(raw, details::tag_com_copy());
}
#endif
//! Queries for the specified interface and returns a fail-fast-based wil::com_ptr_failfast to that interface (fail-fast if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr<U>` pointer to the given interface `U`. The returned pointer will be null only if the source is null.
template <class U, typename T>
inline com_ptr_failfast<U> com_copy_failfast(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr_failfast<U>(raw, details::tag_com_copy());
}
#ifdef WIL_ENABLE_EXCEPTIONS
//! Queries for the interface specified by the type of the output parameter (throws an exception if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @param ptrResult Represents the output pointer to populate. The returned pointer will be null only if the source is null.
template <typename U, typename T>
_Success_true_ void com_copy_to(T&& ptrSource, _COM_Outptr_result_maybenull_ U** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
THROW_IF_FAILED(details::query_policy_t<decltype(raw)>::query(raw, ptrResult));
return;
}
*ptrResult = nullptr;
}
#endif
//! Queries for the interface specified by the type of the output parameter (fail-fast if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @param ptrResult Represents the output pointer to populate. The returned pointer will be null only if the source is null.
template <typename U, typename T>
_Success_true_ void com_copy_to_failfast(T&& ptrSource, _COM_Outptr_result_maybenull_ U** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
FAIL_FAST_IF_FAILED(details::query_policy_t<decltype(raw)>::query(raw, ptrResult));
return;
}
*ptrResult = nullptr;
}
//! Queries for the interface specified by the type of the output parameter (returns an error if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @param ptrResult Represents the output pointer to populate. The returned pointer will be null upon failure or if the source is null.
//! @return Returns an HRESULT representing whether the query succeeded (returns S_OK if the source is null).
template <typename U, typename T>
HRESULT com_copy_to_nothrow(T&& ptrSource, _COM_Outptr_result_maybenull_ U** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
RETURN_HR(details::query_policy_t<decltype(raw)>::query(raw, ptrResult));
}
*ptrResult = nullptr;
return S_OK;
}
#ifdef WIL_ENABLE_EXCEPTIONS
//! Queries for the interface specified by the given REFIID parameter (throws an exception if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @param riid The interface to query for
//! @param ptrResult Represents the output pointer to populate. The returned pointer will be null only if the source is null.
template <typename T>
_Success_true_ void com_copy_to(T&& ptrSource, REFIID riid, _COM_Outptr_result_maybenull_ void** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
THROW_IF_FAILED(details::query_policy_t<decltype(raw)>::query(raw, riid, ptrResult));
return;
}
*ptrResult = nullptr;
}
#endif
//! Queries for the interface specified by the given REFIID parameter (fail-fast if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @param riid The interface to query for
//! @param ptrResult Represents the output pointer to populate. The returned pointer will be null only if the source is null.
template <typename T>
_Success_true_ void com_copy_to_failfast(T&& ptrSource, REFIID riid, _COM_Outptr_result_maybenull_ void** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
FAIL_FAST_IF_FAILED(details::query_policy_t<decltype(raw)>::query(raw, riid, ptrResult));
return;
}
*ptrResult = nullptr;
}
//! Queries for the interface specified by the given REFIID parameter (returns an error if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @param riid The interface to query for
//! @param ptrResult Represents the output pointer to populate. The returned pointer will be null upon failure or if the source is null.
//! @return Returns an HRESULT representing whether the query succeeded (returns S_OK if the source is null).
template <typename T>
HRESULT com_copy_to_nothrow(T&& ptrSource, REFIID riid, _COM_Outptr_result_maybenull_ void** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
RETURN_HR(details::query_policy_t<decltype(raw)>::query(raw, riid, ptrResult));
}
*ptrResult = nullptr;
return S_OK;
}
//! @}
//! @name Stand-alone try copy helpers
//! * Source pointer can be raw interface pointer, any wil com_ptr, or WRL ComPtr
//! * Attempts to retrieves the requested interface
//! * Succeeds with null if the source pointer is null
//! * Produce null if the requested interface is unsupported
//! * bool returns 'true' when query was successful (non-null return result)
//!
//! See @ref page_query for more information.
//! @{
#ifdef WIL_ENABLE_EXCEPTIONS
//! Attempts a query for the specified interface and returns an exception-based wil::com_ptr to that interface (returns null if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr<U>` pointer to the given interface `U`. The returned pointer is null if the requested interface was not supported.
template <class U, typename T>
inline com_ptr<U> try_com_copy(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr<U>(raw, details::tag_try_com_copy());
}
#endif
//! Attempts a query for the specified interface and returns an fail-fast wil::com_ptr_failfast to that interface (returns null if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr_failfast<U>` pointer to the given interface `U`. The returned pointer is null if the requested interface was not supported.
template <class U, typename T>
inline com_ptr_failfast<U> try_com_copy_failfast(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr_failfast<U>(raw, details::tag_try_com_copy());
}
//! Attempts a query for the specified interface and returns an error-code-based wil::com_ptr_nothrow to that interface (returns null if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null
//! @tparam U Represents the interface being queried
//! @return A `wil::com_ptr_nothrow<U>` pointer to the given interface `U`. The returned pointer is null if the requested interface was not supported.
template <class U, typename T>
inline com_ptr_nothrow<U> try_com_copy_nothrow(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
return com_ptr_nothrow<U>(raw, details::tag_try_com_copy());
}
//! Attempts a query for the interface specified by the type of the output parameter (returns `false` if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null.
//! @param ptrResult Represents the output pointer to populate. If the interface is unsupported, the returned pointer will be null.
//! @return A bool value representing whether the query was successful (non-null return result).
template <typename U, typename T>
_Success_return_ bool try_com_copy_to(T&& ptrSource, _COM_Outptr_result_maybenull_ U** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
return SUCCEEDED(details::query_policy_t<decltype(raw)>::query(raw, ptrResult));
}
*ptrResult = nullptr;
return false;
}
//! Attempts a query for the interface specified by the type of the output parameter (returns `false` if unsupported, preserves null).
//! See @ref page_query for more information.
//! @param ptrSource The pointer to query (may be a raw interface pointer, wil com_ptr, or WRL ComPtr), may be null.
//! @param riid The interface to query for
//! @param ptrResult Represents the output pointer to populate. If the interface is unsupported, the returned pointer will be null.
//! @return A bool value representing whether the query was successful (non-null return result).
template <typename T>
_Success_return_ bool try_com_copy_to(T&& ptrSource, REFIID riid, _COM_Outptr_result_maybenull_ void** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
return SUCCEEDED(details::query_policy_t<decltype(raw)>::query(raw, riid, ptrResult));
}
*ptrResult = nullptr;
return false;
}
//! @}
#ifdef __cplusplus_winrt
//! @name Stand-alone helpers to query for CX ref ("hat") types from ABI COM types.
//! * Source pointer can be raw interface pointer, any wil com_ptr, or WRL ComPtr
//! * Retrieves the requested C++/CX interface or ref class.
//! * Preserves null if the source pointer is null
//! * Produce an error if the requested interface is unsupported
//!
//! See @ref page_query for more information
//! @{
template <typename T>
::Platform::Object^ cx_object_from_abi(T&& ptr) WI_NOEXCEPT
{
IInspectable* const inspectable = com_raw_ptr(wistd::forward<T>(ptr));
return reinterpret_cast<::Platform::Object^>(inspectable);
}
template <typename U, typename T>
inline U^ cx_safe_cast(T&& ptrSource)
{
return safe_cast<U^>(cx_object_from_abi(wistd::forward<T>(ptrSource)));
}
template <typename U, typename T>
inline U^ cx_dynamic_cast(T&& ptrSource) WI_NOEXCEPT
{
return dynamic_cast<U^>(cx_object_from_abi(wistd::forward<T>(ptrSource)));
}
//! @}
#endif
//*****************************************************************************
// Agile References
//*****************************************************************************
#if (NTDDI_VERSION >= NTDDI_WINBLUE)
#ifdef WIL_ENABLE_EXCEPTIONS
//! Agile reference to a COM interface, errors throw exceptions (see @ref com_ptr_t and @ref com_agile_query for details)
using com_agile_ref = com_ptr<IAgileReference>;
#endif
//! Agile reference to a COM interface, errors return error codes (see @ref com_ptr_t and @ref com_agile_query_nothrow for details)
using com_agile_ref_nothrow = com_ptr_nothrow<IAgileReference>;
//! Agile reference to a COM interface, errors fail fast (see @ref com_ptr_t and @ref com_agile_query_failfast for details)
using com_agile_ref_failfast = com_ptr_failfast<IAgileReference>;
//! @name Create agile reference helpers
//! * Attempts to retrieve an agile reference to the requested interface (see [RoGetAgileReference](https://msdn.microsoft.com/en-us/library/dn269839.aspx))
//! * Source pointer can be raw interface pointer, any wil com_ptr, or WRL ComPtr
//! * `query` methods AV if the source pointer is null
//! * `copy` methods succeed with null if the source pointer is null
//! * Accept optional [AgileReferenceOptions](https://msdn.microsoft.com/en-us/library/dn269836.aspx)
//!
//! See @ref page_query for more information on resolving an agile ref
//! @{
#ifdef WIL_ENABLE_EXCEPTIONS
//! return a com_agile_ref representing the given source pointer (throws an exception on failure)
template <typename T>
com_agile_ref com_agile_query(T&& ptrSource, AgileReferenceOptions options = AGILEREFERENCE_DEFAULT)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
com_agile_ref agileRef;
THROW_IF_FAILED(::RoGetAgileReference(options, __uuidof(raw), raw, &agileRef));
return agileRef;
}
#endif
//! return a com_agile_ref_failfast representing the given source pointer (fail-fast on failure)
template <typename T>
com_agile_ref_failfast com_agile_query_failfast(T&& ptrSource, AgileReferenceOptions options = AGILEREFERENCE_DEFAULT)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
com_agile_ref_failfast agileRef;
FAIL_FAST_IF_FAILED(::RoGetAgileReference(options, __uuidof(raw), raw, &agileRef));
return agileRef;
}
//! return a com_agile_ref_nothrow representing the given source pointer (returns an HRESULT on failure)
template <typename T>
HRESULT com_agile_query_nothrow(T&& ptrSource, _COM_Outptr_ IAgileReference** ptrResult, AgileReferenceOptions options = AGILEREFERENCE_DEFAULT)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
auto hr = ::RoGetAgileReference(options, __uuidof(raw), raw, ptrResult);
__analysis_assume(SUCCEEDED(hr) || (*ptrResult == nullptr));
RETURN_HR(hr);
}
#ifdef WIL_ENABLE_EXCEPTIONS
//! return a com_agile_ref representing the given source pointer (throws an exception on failure, source maybe null)
template <typename T>
com_agile_ref com_agile_copy(T&& ptrSource, AgileReferenceOptions options = AGILEREFERENCE_DEFAULT)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
com_agile_ref agileRef;
if (raw)
{
THROW_IF_FAILED(::RoGetAgileReference(options, __uuidof(raw), raw, &agileRef));
}
return agileRef;
}
#endif
//! return a com_agile_ref_failfast representing the given source pointer (fail-fast on failure, source maybe null)
template <typename T>
com_agile_ref_failfast com_agile_copy_failfast(T&& ptrSource, AgileReferenceOptions options = AGILEREFERENCE_DEFAULT)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
com_agile_ref_failfast agileRef;
if (raw)
{
FAIL_FAST_IF_FAILED(::RoGetAgileReference(options, __uuidof(raw), raw, &agileRef));
}
return agileRef;
}
//! return an agile ref (com_agile_ref_XXX or other representation) representing the given source pointer (return error on failure, source maybe null)
template <typename T>
HRESULT com_agile_copy_nothrow(T&& ptrSource, _COM_Outptr_result_maybenull_ IAgileReference** ptrResult, AgileReferenceOptions options = AGILEREFERENCE_DEFAULT)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
RETURN_HR(::RoGetAgileReference(options, __uuidof(raw), raw, ptrResult));
}
*ptrResult = nullptr;
return S_OK;
}
//! @}
#endif
//*****************************************************************************
// Weak References
//*****************************************************************************
namespace details
{
template <typename T>
HRESULT GetWeakReference(T* ptr, _COM_Outptr_ IWeakReference** weakReference)
{
static_assert(!wistd::is_same<IWeakReference, T>::value, "Cannot get an IWeakReference to an IWeakReference");
*weakReference = nullptr;
com_ptr_nothrow<IWeakReferenceSource> source;
HRESULT hr = ptr->QueryInterface(IID_PPV_ARGS(&source));
if (SUCCEEDED(hr))
{
hr = source->GetWeakReference(weakReference);
}
return hr;
}
}
#ifdef WIL_ENABLE_EXCEPTIONS
//! Weak reference to a COM interface, errors throw exceptions (see @ref com_ptr_t and @ref com_weak_query for details)
using com_weak_ref = com_ptr<IWeakReference>;
#endif
//! Weak reference to a COM interface, errors return error codes (see @ref com_ptr_t and @ref com_weak_query_nothrow for details)
using com_weak_ref_nothrow = com_ptr_nothrow<IWeakReference>;
//! Weak reference to a COM interface, errors fail fast (see @ref com_ptr_t and @ref com_weak_query_failfast for details)
using com_weak_ref_failfast = com_ptr_failfast<IWeakReference>;
//! @name Create weak reference helpers
//! * Attempts to retrieve a weak reference to the requested interface (see WRL's similar [WeakRef](https://msdn.microsoft.com/en-us/library/br244853.aspx))
//! * Source pointer can be raw interface pointer, any wil com_ptr, or WRL ComPtr
//! * `query` methods AV if the source pointer is null
//! * `copy` methods succeed with null if the source pointer is null
//!
//! See @ref page_query for more information on resolving a weak ref
//! @{
#ifdef WIL_ENABLE_EXCEPTIONS
//! return a com_weak_ref representing the given source pointer (throws an exception on failure)
template <typename T>
com_weak_ref com_weak_query(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
com_weak_ref weakRef;
THROW_IF_FAILED(details::GetWeakReference(raw, &weakRef));
return weakRef;
}
#endif
//! return a com_weak_ref_failfast representing the given source pointer (fail-fast on failure)
template <typename T>
com_weak_ref_failfast com_weak_query_failfast(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
com_weak_ref_failfast weakRef;
FAIL_FAST_IF_FAILED(details::GetWeakReference(raw, &weakRef));
return weakRef;
}
//! return a com_weak_ref_nothrow representing the given source pointer (returns an HRESULT on failure)
template <typename T>
HRESULT com_weak_query_nothrow(T&& ptrSource, _COM_Outptr_ IWeakReference** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
auto hr = details::GetWeakReference(raw, ptrResult);
__analysis_assume(SUCCEEDED(hr) || (*ptrResult == nullptr));
RETURN_HR(hr);
}
#ifdef WIL_ENABLE_EXCEPTIONS
//! return a com_weak_ref representing the given source pointer (throws an exception on failure, source maybe null)
template <typename T>
com_weak_ref com_weak_copy(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
com_weak_ref weakRef;
if (raw)
{
THROW_IF_FAILED(details::GetWeakReference(raw, &weakRef));
}
return weakRef;
}
#endif
//! return a com_weak_ref_failfast representing the given source pointer (fail-fast on failure, source maybe null)
template <typename T>
com_weak_ref_failfast com_weak_copy_failfast(T&& ptrSource)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
com_weak_ref_failfast weakRef;
if (raw)
{
FAIL_FAST_IF_FAILED(details::GetWeakReference(raw, &weakRef));
}
return weakRef;
}
//! return a com_weak_ref_failfast representing the given source pointer (fail-fast on failure, source maybe null)
template <typename T>
HRESULT com_weak_copy_nothrow(T&& ptrSource, _COM_Outptr_result_maybenull_ IWeakReference** ptrResult)
{
auto raw = com_raw_ptr(wistd::forward<T>(ptrSource));
if (raw)
{
RETURN_HR(details::GetWeakReference(raw, ptrResult));
}
*ptrResult = nullptr;
return S_OK;
}
#pragma region COM Object Helpers
template <typename T>
inline bool is_agile(T&& ptrSource)
{
wil::com_ptr_nothrow<IAgileObject> agileObject;
return SUCCEEDED(com_raw_ptr(wistd::forward<T>(ptrSource))->QueryInterface(IID_PPV_ARGS(&agileObject)));
}
/** constructs a COM object using an CLSID on a specific interface or IUnknown.*/
template<typename Interface = IUnknown, typename error_policy = err_exception_policy>
wil::com_ptr_t<Interface, error_policy> CoCreateInstance(REFCLSID rclsid, DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
wil::com_ptr_t<Interface, error_policy> result;
error_policy::HResult(::CoCreateInstance(rclsid, nullptr, dwClsContext, IID_PPV_ARGS(&result)));
return result;
}
/** constructs a COM object using the class as the identifier (that has an associated CLSID) on a specific interface or IUnknown. */
template<typename Class, typename Interface = IUnknown, typename error_policy = err_exception_policy>
wil::com_ptr_t<Interface, error_policy> CoCreateInstance(DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoCreateInstance<Interface, error_policy>(__uuidof(Class), dwClsContext);
}
/** constructs a COM object using an CLSID on a specific interface or IUnknown. */
template<typename Interface = IUnknown>
wil::com_ptr_failfast<Interface> CoCreateInstanceFailFast(REFCLSID rclsid, DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoCreateInstance<Interface, err_failfast_policy>(rclsid, dwClsContext);
}
/** constructs a COM object using the class as the identifier (that has an associated CLSID) on a specific interface or IUnknown. */
template<typename Class, typename Interface = IUnknown>
wil::com_ptr_failfast<Interface> CoCreateInstanceFailFast(DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoCreateInstanceFailFast<Interface>(__uuidof(Class), dwClsContext);
}
/** constructs a COM object using an CLSID on a specific interface or IUnknown.
Note, failures are reported as a null result, the HRESULT is lost. */
template<typename Interface = IUnknown>
wil::com_ptr_nothrow<Interface> CoCreateInstanceNoThrow(REFCLSID rclsid, DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoCreateInstance<Interface, err_returncode_policy>(rclsid, dwClsContext);
}
/** constructs a COM object using the class as the identifier (that has an associated CLSID) on a specific interface or IUnknown.
Note, failures are reported as a null result, the HRESULT is lost. */
template<typename Class, typename Interface = IUnknown>
wil::com_ptr_nothrow<Interface> CoCreateInstanceNoThrow(DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoCreateInstanceNoThrow<Interface>(__uuidof(Class), dwClsContext);
}
/** constructs a COM object class factory using an CLSID on IClassFactory or a specific interface. */
template<typename Interface = IClassFactory, typename error_policy = err_exception_policy>
wil::com_ptr_t<Interface, error_policy> CoGetClassObject(REFCLSID rclsid, DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
wil::com_ptr_t<Interface, error_policy> result;
error_policy::HResult(CoGetClassObject(rclsid, dwClsContext, nullptr, IID_PPV_ARGS(&result)));
return result;
}
/** constructs a COM object class factory using the class as the identifier (that has an associated CLSID)
on IClassFactory or a specific interface. */
template<typename Class, typename Interface = IClassFactory, typename error_policy = err_exception_policy>
wil::com_ptr_t<Interface, error_policy> CoGetClassObject(DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoGetClassObject<Interface, error_policy>(__uuidof(Class), dwClsContext);
}
/** constructs a COM object class factory using an CLSID on IClassFactory or a specific interface. */
template<typename Interface = IClassFactory>
wil::com_ptr_failfast<Interface> CoGetClassObjectFailFast(REFCLSID rclsid, DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoGetClassObject<Interface, err_failfast_policy>(rclsid, dwClsContext);
}
/** constructs a COM object class factory using the class as the identifier (that has an associated CLSID)
on IClassFactory or a specific interface. */
template<typename Class, typename Interface = IClassFactory>
wil::com_ptr_failfast<Interface> CoGetClassObjectFailFast(DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoGetClassObjectFailFast<Interface>(__uuidof(Class), dwClsContext);
}
/** constructs a COM object class factory using an CLSID on IClassFactory or a specific interface.
Note, failures are reported as a null result, the HRESULT is lost. */
template<typename Interface = IClassFactory>
wil::com_ptr_nothrow<Interface> CoGetClassObjectNoThrow(REFCLSID rclsid, DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoGetClassObject<Interface, err_returncode_policy>(rclsid, dwClsContext);
}
/** constructs a COM object class factory using the class as the identifier (that has an associated CLSID)
on IClassFactory or a specific interface.
Note, failures are reported as a null result, the HRESULT is lost. */
template<typename Class, typename Interface = IClassFactory>
wil::com_ptr_nothrow<Interface> CoGetClassObjectNoThrow(DWORD dwClsContext = CLSCTX_INPROC_SERVER)
{
return CoGetClassObjectNoThrow<Interface>(__uuidof(Class), dwClsContext);
}
#pragma endregion
#pragma region Stream helpers
/** Read data from a stream into a buffer.
Reads up to a certain number of bytes into a buffer. Returns the amount of data written, which
may be less than the amount requested if the stream ran out.
~~~~
IStream* source = // ...
ULONG dataBlob = 0;
size_t read = 0;
RETURN_IF_FAILED(wil::stream_read_partial_nothrow(source, &dataBlob, sizeof(dataBlob), &read));
if (read != sizeof(dataBlob))
{
// end of stream, probably
}
else if (dataBlob == 0x8675309)
{
DoThing(dataBlob);
}
~~~~
@param stream The stream from which to read at most `size` bytes.
@param data A buffer into which up to `size` bytes will be read
@param size The size, in bytes, of the buffer pointed to by `data`
@param wrote The amount, in bytes, of data read from `stream` into `data`
*/
inline HRESULT stream_read_partial_nothrow(_In_ ISequentialStream* stream, _Out_writes_bytes_to_(size, *wrote) void* data, unsigned long size, unsigned long *wrote)
{
RETURN_HR(stream->Read(data, size, wrote));
}
/** Read an exact number of bytes from a stream into a buffer.
Fails if the stream didn't read all the bytes requested.
~~~~
IStream* source = // ...
ULONG dataBlob = 0;
RETURN_IF_FAILED(wil::stream_read_nothrow(source, &dataBlob, sizeof(dataBlob)));
if (dataBlob == 0x8675309)
{
DoThing(dataBlob);
}
~~~~
@param stream The stream from which to read at most `size` bytes.
@param data A buffer into which up to `size` bytes will be read
@param size The size, in bytes, of the buffer pointed to by `data`
@return The underlying stream read result, or HRESULT_FROM_WIN32(ERROR_INVALID_DATA) if the stream
did not read the complete buffer.
*/
inline HRESULT stream_read_nothrow(_In_ ISequentialStream* stream, _Out_writes_bytes_all_(size) void* data, unsigned long size)
{
unsigned long didRead;
RETURN_IF_FAILED(stream_read_partial_nothrow(stream, data, size, &didRead));
RETURN_HR_IF(HRESULT_FROM_WIN32(ERROR_INVALID_DATA), didRead != size);
return S_OK;
}
/** Read from a stream into a POD type.
Fails if the stream didn't have enough bytes.
~~~~
IStream* source = // ...
MY_HEADER header{};
RETURN_IF_FAILED(wil::stream_read_nothrow(source, &header));
if (header.Version == 0x8675309)
{
ConsumeOldHeader(stream, header);
}
~~~~
@param stream The stream from which to read at most `size` bytes.
@param pThing The POD data type to read from the stream.
@return The underlying stream read result, or HRESULT_FROM_WIN32(ERROR_INVALID_DATA) if the stream
did not read the complete buffer.
*/
template<typename T> HRESULT stream_read_nothrow(_In_ ISequentialStream* stream, _Out_ T* pThing)
{
static_assert(__is_pod(T), "Type must be POD.");
return stream_read_nothrow(stream, pThing, sizeof(T));
}
/** Write an exact number of bytes to a stream from a buffer.
Fails if the stream didn't read write the bytes requested.
~~~~
IStream* source = // ...
ULONG dataBlob = 0x8675309;
RETURN_IF_FAILED(wil::stream_write_nothrow(source, &dataBlob, sizeof(dataBlob)));
~~~~
@param stream The stream to which to write at most `size` bytes.
@param data A buffer from which up to `size` bytes will be read
@param size The size, in bytes, of the buffer pointed to by `data`
*/
inline HRESULT stream_write_nothrow(_In_ ISequentialStream* stream, _In_reads_bytes_(size) const void* data, unsigned long size)
{
unsigned long wrote;
RETURN_IF_FAILED(stream->Write(data, size, &wrote));
RETURN_HR_IF(HRESULT_FROM_WIN32(ERROR_INVALID_DATA), wrote != size);
return S_OK;
}
/** Write a POD type to a stream.
Fails if not all the bytes were written.
~~~~
IStream* source = // ...
MY_HEADER header { 0x8675309, HEADER_FLAG_1 | HEADER_FLAG_2 };
RETURN_IF_FAILED(wil::stream_write_nothrow(source, header));
ULONGLONG value = 16;
RETURN_IF_FAILED(wil::stream_write_nothrow(source, value));
~~~~
@param stream The stream to which to write `thing`
@param thing The POD data type to write to the stream.
*/
template<typename T> inline HRESULT stream_write_nothrow(_In_ ISequentialStream* stream, const T& thing)
{
return stream_write_nothrow(stream, wistd::addressof(thing), sizeof(thing));
}
/** Retrieve the size of this stream, in bytes
~~~~
IStream* source = // ...
ULONGLONG size;
RETURN_IF_FAILED(wil::stream_size_nothrow(source, &size));
RETURN_HR_IF(E_INVALIDARG, size > ULONG_MAX);
~~~~
@param stream The stream whose size is to be returned in `value`
@param value The size, in bytes, reported by `stream`
*/
inline HRESULT stream_size_nothrow(_In_ IStream* stream, _Out_ unsigned long long* value)
{
STATSTG st{};
RETURN_IF_FAILED(stream->Stat(&st, STATFLAG_NONAME));
*value = st.cbSize.QuadPart;
return S_OK;
}
/** Seek a stream to a relative offset or absolute position
~~~~
IStream* source = // ...
unsigned long long landed;
RETURN_IF_FAILED(wil::stream_seek_nothrow(source, 16, STREAM_SEEK_CUR, &landed));
RETURN_IF_FAILED(wil::stream_seek_nothrow(source, -5, STREAM_SEEK_END));
RETURN_IF_FAILED(wil::stream_seek_nothrow(source, LLONG_MAX, STREAM_SEEK_CUR));
~~~~
@param stream The stream to seek
@param offset The position, in bytes from the current position, to seek
@param from The starting point from which to seek, from the STREAM_SEEK_* set of values
@param value Optionally recieves the new absolute position from the stream
*/
inline HRESULT stream_seek_nothrow(_In_ IStream* stream, long long offset, unsigned long from, _Out_opt_ unsigned long long* value = nullptr)
{
LARGE_INTEGER amount;
ULARGE_INTEGER landed{};
amount.QuadPart = offset;
RETURN_IF_FAILED(stream->Seek(amount, from, value ? &landed : nullptr));
assign_to_opt_param(value, landed.QuadPart);
return S_OK;
}
/** Seek a stream to an absolute offset
~~~~
IStream* source = // ...
RETURN_HR(wil::stream_set_position_nothrow(source, 16));
~~~~
@param stream The stream whose size is to be returned in `value`
@param offset The position, in bytes from the start of the stream, to seek to
@param value Optionally recieves the new absolute position from the stream
*/
inline HRESULT stream_set_position_nothrow(_In_ IStream* stream, unsigned long long offset, _Out_opt_ unsigned long long* value = nullptr)
{
// IStream::Seek(..., _SET) interprets the first parameter as an unsigned value.
return stream_seek_nothrow(stream, static_cast<long long>(offset), STREAM_SEEK_SET, value);
}
/** Seek a relative amount in a stream
~~~~
IStream* source = // ...
RETURN_IF_FAILED(wil::stream_seek_from_current_position_nothrow(source, -16));
ULONGLONG newPosition;
RETURN_IF_FAILED(wil::stream_seek_from_current_position_nothrow(source, 16, &newPosition));
~~~~
@param stream The stream whose location is to be moved
@param amount The offset, in bytes, to seek the stream.
@param value Set to the new absolute steam position, in bytes
*/
inline HRESULT stream_seek_from_current_position_nothrow(_In_ IStream* stream, long long amount, _Out_opt_ unsigned long long* value = nullptr)
{
return stream_seek_nothrow(stream, amount, STREAM_SEEK_CUR, value);
}
/** Determine the current byte position in the stream
~~~~
IStream* source = // ...
ULONGLONG currentPos;
RETURN_IF_FAILED(wil::stream_get_position_nothrow(source, &currentPos));
~~~~
@param stream The stream whose location is to be moved
@param position Set to the current absolute steam position, in bytes
*/
inline HRESULT stream_get_position_nothrow(_In_ IStream* stream, _Out_ unsigned long long* position)
{
return stream_seek_from_current_position_nothrow(stream, 0, position);
}
/** Moves the stream to absolute position 0
~~~~
IStream* source = // ...
RETURN_IF_FAILED(wil::stream_reset_nothrow(source));
~~~~
@param stream The stream whose location is to be moved
*/
inline HRESULT stream_reset_nothrow(_In_ IStream* stream)
{
return stream_set_position_nothrow(stream, 0);
}
/** Copy data from one stream to another, returning the final amount copied.
~~~~
IStream* source = // ...
IStream* target = // ...
ULONGLONG copied;
RETURN_IF_FAILED(wil::stream_copy_bytes_nothrow(source, target, sizeof(MyType), &copied));
if (copied < sizeof(MyType))
{
DoSomethingAboutPartialCopy();
}
~~~~
@param source The stream from which to copy at most `amount` bytes
@param target The steam to which to copy at most `amount` bytes
@param amount The maximum number of bytes to copy from `source` to `target`
@param pCopied If non-null, set to the number of bytes copied between the two.
*/
inline HRESULT stream_copy_bytes_nothrow(_In_ IStream* source, _In_ IStream* target, unsigned long long amount, _Out_opt_ unsigned long long* pCopied = nullptr)
{
ULARGE_INTEGER toCopy;
ULARGE_INTEGER copied;
toCopy.QuadPart = amount;
RETURN_IF_FAILED(source->CopyTo(target, toCopy, nullptr, &copied));
assign_to_opt_param(pCopied, copied.QuadPart);
return S_OK;
}
/** Copy all data from one stream to another, returning the final amount copied.
~~~~
IStream* source = // ...
IStream* target = // ...
ULONGLONG copied;
RETURN_IF_FAILED(wil::stream_copy_all_nothrow(source, target, &copied));
if (copied < 8)
{
DoSomethingAboutPartialCopy();
}
~~~~
@param source The stream from which to copy all content
@param target The steam to which to copy all content
@param pCopied If non-null, set to the number of bytes copied between the two.
*/
inline HRESULT stream_copy_all_nothrow(_In_ IStream* source, _In_ IStream* target, _Out_opt_ unsigned long long* pCopied = nullptr)
{
return stream_copy_bytes_nothrow(source, target, ULLONG_MAX, pCopied);
}
/** Copies an exact amount of data from one stream to another, failing otherwise
~~~~
IStream* source = // ...
IStream* target = // ...
RETURN_IF_FAILED(wil::stream_copy_all_nothrow(source, target, 16));
~~~~
@param source The stream from which to copy at most `amount` bytes
@param target The steam to which to copy at most `amount` bytes
@param amount The number of bytes to copy from `source` to `target`
*/
inline HRESULT stream_copy_exact_nothrow(_In_ IStream* source, _In_ IStream* target, unsigned long long amount)
{
unsigned long long copied;
RETURN_IF_FAILED(stream_copy_bytes_nothrow(source, target, ULLONG_MAX, &copied));
RETURN_HR_IF(HRESULT_FROM_WIN32(ERROR_INVALID_DATA), copied != amount);
return S_OK;
}
//! Controls behavior when reading a zero-length string from a stream
enum class empty_string_options
{
//! Zero-length strings are returned as nullptr
returns_null,
//! Zero-length strings are allocated and returned with zero characters
returns_empty,
};
#ifdef __WIL_OBJBASE_H_
/** Read a string from a stream and returns an allocated copy
Deserializes strings in streams written by both IStream_WriteStr and wil::stream_write_string[_nothrow]. The format
is a single 16-bit quantity, followed by that many wchar_ts. The returned string is allocated with CoTaskMemAlloc.
Returns a zero-length (but non-null) string if the stream contained a zero-length string.
~~~~
IStream* source = // ...
wil::unique_cotaskmem_string content;
RETURN_IF_FAILED(wil::stream_read_string_nothrow(source, &content));
if (wcscmp(content.get(), L"waffles") == 0)
{
// Waffles!
}
~~~~
@param source The stream from which to read a string
@param value Set to point to the allocated result of reading a string from `source`
*/
inline HRESULT stream_read_string_nothrow(
_In_ ISequentialStream* source,
_When_(options == empty_string_options::returns_empty, _Outptr_result_z_) _When_(options == empty_string_options::returns_null, _Outptr_result_maybenull_z_) wchar_t** value,
empty_string_options options = empty_string_options::returns_empty)
{
unsigned short cch;
RETURN_IF_FAILED(stream_read_nothrow(source, &cch));
if ((cch == 0) && (options == empty_string_options::returns_null))
{
*value = nullptr;
}
else
{
auto allocated = make_unique_cotaskmem_nothrow<wchar_t[]>(static_cast<size_t>(cch) + 1);
RETURN_IF_NULL_ALLOC(allocated);
RETURN_IF_FAILED(stream_read_nothrow(source, allocated.get(), static_cast<unsigned long>(cch) * sizeof(wchar_t)));
allocated[cch] = 0;
*value = allocated.release();
}
return S_OK;
}
#endif // __WIL_OBJBASE_H
/** Write a string to a stream
Serializes a string into a stream by putting its length and then the wchar_ts in the string
into the stream. Zero-length strings have their length but no data written. This is the
form expected by IStream_ReadStr and wil::string_read_stream.
~~~~
IStream* target = // ...
RETURN_IF_FAILED(wil::stream_write_string_nothrow(target, L"Waffles", 3));
// Produces wchar_t[] { 0x3, L'W', L'a', L'f' };
~~~~
@param target The stream to which to write a string
@param source The string to write. Can be null if `writeLength` is zero
@param writeLength The number of characters to write from source into `target`
*/
inline HRESULT stream_write_string_nothrow(_In_ ISequentialStream* target, _In_reads_opt_(writeLength) const wchar_t* source, _In_ size_t writeLength)
{
FAIL_FAST_IF(writeLength > USHRT_MAX);
RETURN_IF_FAILED(stream_write_nothrow(target, static_cast<unsigned short>(writeLength)));
if (writeLength > 0)
{
RETURN_IF_FAILED(stream_write_nothrow(target, source, static_cast<unsigned short>(writeLength) * sizeof(wchar_t)));
}
return S_OK;
}
/** Write a string to a stream
Serializes a string into a stream by putting its length and then the wchar_ts in the string
into the stream. Zero-length strings have their length but no data written. This is the
form expected by IStream_ReadStr and wil::string_read_stream.
~~~~
IStream* target = // ...
RETURN_IF_FAILED(wil::stream_write_string_nothrow(target, L"Waffles"));
// Produces wchar_t[] { 0x3, L'W', L'a', L'f', L'f', L'l', L'e', L's' };
~~~~
@param target The stream to which to write a string
@param source The string to write. When nullptr, a zero-length string is written.
*/
inline HRESULT stream_write_string_nothrow(_In_ ISequentialStream* target, _In_opt_z_ const wchar_t* source)
{
return stream_write_string_nothrow(target, source, source ? wcslen(source) : 0);
}
#ifdef WIL_ENABLE_EXCEPTIONS
/** Read data from a stream into a buffer.
~~~~
IStream* source = // ...
ULONG dataBlob = 0;
auto read = wil::stream_read_partial(source, &dataBlob, sizeof(dataBlob));
if (read != sizeof(dataBlob))
{
// end of stream, probably
}
else if (dataBlob == 0x8675309)
{
DoThing(dataBlob);
}
~~~~
@param stream The stream from which to read at most `size` bytes.
@param data A buffer into which up to `size` bytes will be read
@param size The size, in bytes, of the buffer pointed to by `data`
@return The amount, in bytes, of data read from `stream` into `data`
*/
inline unsigned long stream_read_partial(_In_ ISequentialStream* stream, _Out_writes_bytes_to_(size, return) void* data, unsigned long size)
{
unsigned long didRead;
THROW_IF_FAILED(stream_read_partial_nothrow(stream, data, size, &didRead));
return didRead;
}
/** Read an exact number of bytes from a stream into a buffer.
Fails if the stream didn't read all the bytes requested by throwing HRESULT_FROM_WIN32(ERROR_INVALID_DATA).
~~~~
IStream* source = // ...
ULONG dataBlob = 0;
wil::stream_read(source, &dataBlob, sizeof(dataBlob));
if (dataBlob == 0x8675309)
{
DoThing(dataBlob);
}
~~~~
@param stream The stream from which to read at most `size` bytes.
@param data A buffer into which up to `size` bytes will be read
@param size The size, in bytes, of the buffer pointed to by `data`
*/
inline void stream_read(_In_ ISequentialStream* stream, _Out_writes_bytes_all_(size) void* data, unsigned long size)
{
THROW_HR_IF(HRESULT_FROM_WIN32(ERROR_INVALID_DATA), stream_read_partial(stream, data, size) != size);
}
/** Read from a stream into a POD type.
Fails if the stream didn't have enough bytes by throwing HRESULT_FROM_WIN32(ERROR_INVALID_DATA).
~~~~
IStream* source = // ...
MY_HEADER header = wil::stream_read<MY_HEADER>(source);
if (header.Version == 0x8675309)
{
ConsumeOldHeader(stream, header);
}
~~~~
@param stream The stream from which to read at most `sizeof(T)` bytes.
@return An instance of `T` read from the stream
*/
template<typename T> T stream_read(_In_ ISequentialStream* stream)
{
static_assert(__is_pod(T), "Read type must be POD");
T temp{};
stream_read(stream, &temp, sizeof(temp));
return temp;
}
/** Write an exact number of bytes to a stream from a buffer.
Fails if the stream didn't read write the bytes requested.
~~~~
IStream* source = // ...
ULONG dataBlob = 0;
wil::stream_write(source, dataBlob, sizeof(dataBlob));
~~~~
@param stream The stream to which to write at most `size` bytes.
@param data A buffer from which up to `size` bytes will be read
@param size The size, in bytes, of the buffer pointed to by `data`
*/
inline void stream_write(_In_ ISequentialStream* stream, _In_reads_bytes_(size) const void* data, unsigned long size)
{
THROW_IF_FAILED(stream_write_nothrow(stream, data, size));
}
/** Write a POD type to a stream.
Fails if the stream didn't accept the entire size.
~~~~
IStream* target = // ...
MY_HEADER header { 0x8675309, HEADER_FLAG_1 | HEADER_FLAG_2 };
wil::stream_write(target, header)
wil::stream_write<ULONGLONG>(target, 16);
~~~~
@param stream The stream to which to write `thing`
@param thing The POD data type to write to the stream.
*/
template<typename T> inline void stream_write(_In_ ISequentialStream* stream, const T& thing)
{
stream_write(stream, wistd::addressof(thing), sizeof(thing));
}
/** Retrieve the size of this stream, in bytes
~~~~
IStream* source = // ...
ULONGLONG size = wil::stream_size(source);
~~~~
@param stream The stream whose size is to be returned in `value`
@return The size, in bytes, reported by `stream`
*/
inline unsigned long long stream_size(_In_ IStream* stream)
{
unsigned long long size;
THROW_IF_FAILED(stream_size_nothrow(stream, &size));
return size;
}
/** Seek a stream to an absolute offset
~~~~
IStream* source = // ...
wil::stream_set_position(source, sizeof(HEADER));
~~~~
@param stream The stream whose size is to be returned in `value`
@param offset The offset, in bytes, to seek the stream.
@return The new absolute stream position, in bytes
*/
inline unsigned long long stream_set_position(_In_ IStream* stream, unsigned long long offset)
{
unsigned long long landed;
THROW_IF_FAILED(stream_set_position_nothrow(stream, offset, &landed));
return landed;
}
/** Seek a relative amount in a stream
~~~~
IStream* source = // ...
ULONGLONG newPosition = wil::stream_seek_from_current_position(source, 16);
~~~~
@param stream The stream whose location is to be moved
@param amount The offset, in bytes, to seek the stream.
@return The new absolute stream position, in bytes
*/
inline unsigned long long stream_seek_from_current_position(_In_ IStream* stream, long long amount)
{
unsigned long long landed;
THROW_IF_FAILED(stream_seek_from_current_position_nothrow(stream, amount, &landed));
return landed;
}
/** Determine the current byte position in the stream
~~~~
IStream* source = // ...
ULONGLONG currentPos = wil::stream_get_position(source);
~~~~
@param stream The stream whose location is to be moved
@return The current position reported by `stream`
*/
inline unsigned long long stream_get_position(_In_ IStream* stream)
{
return stream_seek_from_current_position(stream, 0);
}
/** Moves the stream to absolute position 0
~~~~
IStream* source = // ...
wil::stream_reset(source);
ASSERT(wil::stream_get_position(source) == 0);
~~~~
@param stream The stream whose location is to be moved
*/
inline void stream_reset(_In_ IStream* stream)
{
stream_set_position(stream, 0);
}
/** Copy data from one stream to another
~~~~
IStream* source = // ...
IStream* target = // ...
ULONGLONG copied = ;
if (wil::stream_copy_bytes(source, target, sizeof(Header)) < sizeof(Header))
{
DoSomethingAboutPartialCopy();
}
~~~~
@param source The stream from which to copy at most `amount` bytes
@param target The steam to which to copy at most `amount` bytes
@param amount The maximum number of bytes to copy from `source` to `target`
@return The number of bytes copied between the two streams
*/
inline unsigned long long stream_copy_bytes(_In_ IStream* source, _In_ IStream* target, unsigned long long amount)
{
unsigned long long copied;
THROW_IF_FAILED(stream_copy_bytes_nothrow(source, target, amount, &copied));
return copied;
}
/** Copy all data from one stream to another
~~~~
IStream* source = // ...
IStream* target = // ...
ULONGLONG copied = wil::stream_copy_all(source, target);
~~~~
@param source The stream from which to copy all content
@param target The steam to which to copy all content
@return The number of bytes copied between the two.
*/
inline unsigned long long stream_copy_all(_In_ IStream* source, _In_ IStream* target)
{
return stream_copy_bytes(source, target, ULLONG_MAX);
}
/** Copies an exact amount of data from one stream to another, failing otherwise
~~~~
IStream* source = // ...
IStream* target = // ...
wil::stream_copy_all_nothrow(source, target, sizeof(SOMETHING));
~~~~
@param source The stream from which to copy at most `amount` bytes
@param target The steam to which to copy at most `amount` bytes
@param amount The number of bytes to copy from `source` to `target`
*/
inline void stream_copy_exact(_In_ IStream* source, _In_ IStream* target, unsigned long long amount)
{
THROW_HR_IF(HRESULT_FROM_WIN32(ERROR_INVALID_DATA), stream_copy_bytes(source, target, amount) != amount);
}
#ifdef __WIL_OBJBASE_H_
/** Read a string from a stream and returns an allocated copy
Deserializes strings in streams written by both IStream_WriteStr and wil::stream_write_string[_nothrow]. The format
is a single 16-bit quantity, followed by that many wchar_ts. The returned string is allocated with CoTaskMemAlloc.
Returns a zero-length (but non-null) string if the stream contained a zero-length string.
~~~~
IStream* source = // ...
wil::unique_cotaskmem_string content = wil::stream_read_string(source);
if (wcscmp(content.get(), L"waffles") == 0)
{
// Waffles!
}
~~~~
@param source The stream from which to read a string
@return An non-null string (but possibly zero lengh) string read from `source`
*/
inline wil::unique_cotaskmem_string stream_read_string(_In_ ISequentialStream* source, empty_string_options options = empty_string_options::returns_empty)
{
wil::unique_cotaskmem_string result;
THROW_IF_FAILED(stream_read_string_nothrow(source, &result, options));
return result;
}
#endif // __WIL_OBJBASE_H
/** Write a string to a stream
Serializes a string into a stream by putting its length and then the wchar_ts in the string
into the stream. Zero-length strings have their length but no data written. This is the
form expected by IStream_ReadStr and wil::string_read_stream.
~~~~
IStream* target = // ...
wil::stream_write_string(target, L"Waffles", 3);
~~~~
@param target The stream to which to write a string
@param source The string to write. Can be null if `writeLength` is zero
@param writeLength The number of characters to write from source into `target`
*/
inline void stream_write_string(_In_ ISequentialStream* target, _In_reads_opt_(toWriteCch) const wchar_t* source, _In_ size_t toWriteCch)
{
THROW_IF_FAILED(stream_write_string_nothrow(target, source, toWriteCch));
}
/** Write a string to a stream
Serializes a string into a stream by putting its length and then the wchar_ts in the string
into the stream. Zero-length strings have their length but no data written.This is the
form expected by IStream_ReadStr and wil::string_read_stream.
~~~~
IStream* target = // ...
wil::stream_write_string(target, L"Waffles");
~~~~
@param target The stream to which to write a string
@param source The string to write. When nullptr, a zero-length string is written.
*/
inline void stream_write_string(_In_ ISequentialStream* target, _In_opt_z_ const wchar_t* source)
{
THROW_IF_FAILED(stream_write_string_nothrow(target, source, source ? wcslen(source) : 0));
}
/** Saves and restores the position of a stream
Useful for potentially reading data from a stream, or being able to read ahead, then reset
back to where one left off, such as conditionally reading content from a stream.
~~~~
void MaybeConsumeStream(IStream* stream)
{
// On error, reset the read position in the stream to where we left off
auto saver = wil::stream_position_saver(stream);
auto header = wil::stream_read<MY_HEADER>(stream);
for (ULONG i = 0; i < header.Count; ++i)
{
ProcessElement(wil::stream_read<MY_ELEMENT>(stream));
}
}
~~~~
*/
class stream_position_saver
{
public:
//! Constructs a saver from the current position of this stream
//! @param stream The stream instance whose position is to be saved.
explicit stream_position_saver(_In_opt_ IStream* stream) :
m_stream(stream),
m_position(stream ? stream_get_position(stream) : 0)
{
}
~stream_position_saver()
{
if (m_stream)
{
LOG_IF_FAILED(stream_set_position_nothrow(m_stream.get(), m_position));
}
}
/** Updates the current position in the stream
~~~~
// Read a size marker from the stream, then advance that much.
IStream* stream1 = // ...
auto saver = wil::stream_position_saver(stream1);
auto size = wil::stream_read<long>(stream1);
wil::stream_seek_from_current_position(stream, size);
saver.update();
~~~~
*/
void update()
{
m_position = stream_get_position(m_stream.get());
}
//! Returns the current position being saved for the stream
//! @returns The position, in bytes, being saved for the stream
unsigned long long position() const
{
return m_position;
}
/** Resets the position saver to manage a new stream
Reverts the position of any stream this saver is currently holding a place for.
~~~~
IStream* stream1 = // ...
IStream* stream2 = // ...
auto saver = wil::stream_position_saver(stream1);
if (wil::stream_read<MyType>(stream1).Flags != 0)
{
saver.reset(stream2); // position in stream1 is reverted, now holding stream2
}
~~~~
@param stream The stream whose position is to be saved
*/
void reset(_In_ IStream* stream)
{
reset();
m_stream = stream;
m_position = wil::stream_get_position(m_stream.get());
}
/** Resets the position of the stream
~~~~
IStream* stream1 = // ...
auto saver = wil::stream_position_saver(stream1);
MyType mt = wil::stream_read<MyType>(stream1);
if (mt.Flags & MyTypeFlags::Extended)
{
saver.reset();
ProcessExtended(stream1, wil::stream_read<MyTypeExtended>(stream1));
}
else
{
ProcessStandard(stream1, mt);
}
~~~~
*/
void reset()
{
if (m_stream)
{
wil::stream_set_position(m_stream.get(), m_position);
}
}
/** Stops saving the position of the stream
~~~~
// The stream has either a standard or extended header, followed by interesting content.
// Read either one, leaving the stream after the headers have been read off. On failure,
// the stream's position is restored.
std::pair<MyType, MyTypeExtended> get_headers(_In_ IStream* source)
{
auto saver = wil::stream_position_saver(stream1);
MyType mt = wil::stream_read<MyType>(stream1);
MyTypeExtended mte{};
if (mt.Flags & MyTypeFlags::Extended)
{
mte = wil::stream_read<MyTypeExtended>(stream1);
}
saver.dismiss();
return { mt, mte };
}
~~~~
*/
void dismiss()
{
m_stream.reset();
}
stream_position_saver(stream_position_saver&&) = default;
stream_position_saver& operator=(stream_position_saver&&) = default;
stream_position_saver(const stream_position_saver&) = delete;
void operator=(const stream_position_saver&) = delete;
private:
com_ptr<IStream> m_stream;
unsigned long long m_position;
};
#endif // WIL_ENABLE_EXCEPTIONS
#pragma endregion // stream helpers
#if defined(__IObjectWithSite_INTERFACE_DEFINED__)
/// @cond
namespace details
{
inline void __stdcall SetSiteNull(IObjectWithSite* objWithSite)
{
objWithSite->SetSite(nullptr); // break the cycle
}
} // details
/// @endcond
using unique_set_site_null_call = wil::unique_com_call<IObjectWithSite, decltype(details::SetSiteNull), details::SetSiteNull>;
/** RAII support for managing the site chain. This function sets the site pointer on an object and return an object
that resets it on destruction to break the cycle.
Note, this does not preserve the existing site if there is one (an uncommon case) so only use this when that is not required.
~~~
auto cleanup = wil::com_set_site(execCommand.get(), serviceProvider->GetAsSite());
~~~
Include ocidl.h before wil\com.h to use this.
*/
WI_NODISCARD inline unique_set_site_null_call com_set_site(_In_opt_ IUnknown* obj, _In_opt_ IUnknown* site)
{
wil::com_ptr_nothrow<IObjectWithSite> objWithSite;
if (site && wil::try_com_copy_to(obj, &objWithSite))
{
objWithSite->SetSite(site);
}
return unique_set_site_null_call(objWithSite.get());
}
/** Iterate over each object in a site chain. Useful for debugging site issues, here is sample use.
~~~
void OutputDebugSiteChainWatchWindowText(IUnknown* site)
{
OutputDebugStringW(L"Copy and paste these entries into the Visual Studio Watch Window\n");
wil::for_each_site(site, [](IUnknown* site)
{
wchar_t msg[64];
StringCchPrintfW(msg, ARRAYSIZE(msg), L"((IUnknown*)0x%p)->__vfptr[0]\n", site);
OutputDebugStringW(msg);
});
}
*/
template<typename TLambda>
void for_each_site(_In_opt_ IUnknown* siteInput, TLambda&& callback)
{
wil::com_ptr_nothrow<IUnknown> site(siteInput);
while (site)
{
callback(site.get());
auto objWithSite = site.try_query<IObjectWithSite>();
site.reset();
if (objWithSite)
{
objWithSite->GetSite(IID_PPV_ARGS(&site));
}
}
}
#endif // __IObjectWithSite_INTERFACE_DEFINED__
} // wil
#endif