2
* Copyright (C) 2020 Apple Inc. All rights reserved.
4
* Redistribution and use in source and binary forms, with or without
5
* modification, are permitted provided that the following conditions
7
* 1. Redistributions of source code must retain the above copyright
8
* notice, this list of conditions and the following disclaimer.
9
* 2. Redistributions in binary form must reproduce the above copyright
10
* notice, this list of conditions and the following disclaimer in the
11
* documentation and/or other materials provided with the distribution.
13
* THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
17
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28
#include <wtf/DumbPtrTraits.h>
29
#include <wtf/FastMalloc.h>
33
// The purpose of this class is to ensure that the wrapped pointer will never be
34
// used uninitialized.
36
template <typename T> class NakedRef {
37
WTF_MAKE_FAST_ALLOCATED;
39
ALWAYS_INLINE NakedRef(T& ref) : m_ptr(&ref) { }
40
ALWAYS_INLINE NakedRef(const NakedRef&) = delete;
41
template<typename U> NakedRef(const NakedRef<U>&) = delete;
43
ALWAYS_INLINE NakedRef(NakedRef&& other)
44
: m_ptr(&other.leakRef())
50
NakedRef(NakedRef<U>&& other)
51
: m_ptr(&other.leakRef())
56
ALWAYS_INLINE T* operator->() const { ASSERT(m_ptr); return m_ptr; }
57
T* ptr() const RETURNS_NONNULL { ASSERT(m_ptr); return m_ptr; }
58
T& get() const { return *m_ptr; }
59
operator T&() const { ASSERT(m_ptr); return *m_ptr; }
60
bool operator!() const { ASSERT(m_ptr); return !*m_ptr; }
62
NakedRef copyRef() && = delete;
63
NakedRef copyRef() const & WARN_UNUSED_RETURN { return NakedRef(*m_ptr); }
65
NakedRef& operator=(T&);
66
NakedRef& operator=(NakedRef&&);
67
template<typename U> NakedRef& operator=(NakedRef<U>&&);
69
// Use copyRef() and the move assignment operators instead.
70
NakedRef& operator=(const NakedRef&) = delete;
71
template<typename X> NakedRef& operator=(const NakedRef<X>&) = delete;
73
template<typename U> void swap(NakedRef<U>&);
75
T& leakRef() WARN_UNUSED_RETURN
78
T& result = *DumbPtrTraits<T>::exchange(m_ptr, nullptr);
86
template<typename T> inline NakedRef<T>& NakedRef<T>::operator=(NakedRef&& reference)
88
NakedRef movedReference = WTFMove(reference);
93
template<typename T> inline NakedRef<T>& NakedRef<T>::operator=(T& ref)
95
NakedRef copiedReference = ref;
96
swap(copiedReference);
100
template<typename T> template<typename U> inline NakedRef<T>& NakedRef<T>::operator=(NakedRef<U>&& other)
102
NakedRef ref = WTFMove(other);
109
inline void NakedRef<T>::swap(NakedRef<U>& other)
111
std::swap(m_ptr, other.m_ptr);
114
template<class T, class U> inline void swap(NakedRef<T>& a, NakedRef<U>& b)