123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348 |
- // Copyright 2014 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #ifndef BASE_SCOPED_GENERIC_H_
- #define BASE_SCOPED_GENERIC_H_
- #include <stdlib.h>
- #include <ostream>
- #include <algorithm>
- #include <utility>
- #include "base/check.h"
- // TODO(crbug.com/1010217) Remove once no #includers are getting base/macros.h
- // by including this header.
- #include "base/macros.h"
- namespace base {
- // This class acts like unique_ptr with a custom deleter (although is slightly
- // less fancy in some of the more escoteric respects) except that it keeps a
- // copy of the object rather than a pointer, and we require that the contained
- // object has some kind of "invalid" value.
- //
- // Defining a scoper based on this class allows you to get a scoper for
- // non-pointer types without having to write custom code for set, reset, and
- // move, etc. and get almost identical semantics that people are used to from
- // unique_ptr.
- //
- // It is intended that you will typedef this class with an appropriate deleter
- // to implement clean up tasks for objects that act like pointers from a
- // resource management standpoint but aren't, such as file descriptors and
- // various types of operating system handles. Using unique_ptr for these
- // things requires that you keep a pointer to the handle valid for the lifetime
- // of the scoper (which is easy to mess up).
- //
- // For an object to be able to be put into a ScopedGeneric, it must support
- // standard copyable semantics and have a specific "invalid" value. The traits
- // must define a free function and also the invalid value to assign for
- // default-constructed and released objects.
- //
- // struct FooScopedTraits {
- // // It's assumed that this is a fast inline function with little-to-no
- // // penalty for duplicate calls. This must be a static function even
- // // for stateful traits.
- // static int InvalidValue() {
- // return 0;
- // }
- //
- // // This free function will not be called if f == InvalidValue()!
- // static void Free(int f) {
- // ::FreeFoo(f);
- // }
- // };
- //
- // typedef ScopedGeneric<int, FooScopedTraits> ScopedFoo;
- //
- // A Traits type may choose to track ownership of objects in parallel with
- // ScopedGeneric. To do so, it must implement the Acquire and Release methods,
- // which will be called by ScopedGeneric during ownership transfers and extend
- // the ScopedGenericOwnershipTracking tag type.
- //
- // struct BarScopedTraits : public ScopedGenericOwnershipTracking {
- // using ScopedGenericType = ScopedGeneric<int, BarScopedTraits>;
- // static int InvalidValue() {
- // return 0;
- // }
- //
- // static void Free(int b) {
- // ::FreeBar(b);
- // }
- //
- // static void Acquire(const ScopedGenericType& owner, int b) {
- // ::TrackAcquisition(b, owner);
- // }
- //
- // static void Release(const ScopedGenericType& owner, int b) {
- // ::TrackRelease(b, owner);
- // }
- // };
- //
- // typedef ScopedGeneric<int, BarScopedTraits> ScopedBar;
- struct ScopedGenericOwnershipTracking {};
- template<typename T, typename Traits>
- class ScopedGeneric {
- private:
- // This must be first since it's used inline below.
- //
- // Use the empty base class optimization to allow us to have a D
- // member, while avoiding any space overhead for it when D is an
- // empty class. See e.g. http://www.cantrip.org/emptyopt.html for a good
- // discussion of this technique.
- struct Data : public Traits {
- explicit Data(const T& in) : generic(in) {}
- Data(const T& in, const Traits& other) : Traits(other), generic(in) {}
- T generic;
- };
- public:
- typedef T element_type;
- typedef Traits traits_type;
- ScopedGeneric() : data_(traits_type::InvalidValue()) {}
- // Constructor. Takes responsibility for freeing the resource associated with
- // the object T.
- explicit ScopedGeneric(const element_type& value) : data_(value) {
- TrackAcquire(data_.generic);
- }
- // Constructor. Allows initialization of a stateful traits object.
- ScopedGeneric(const element_type& value, const traits_type& traits)
- : data_(value, traits) {
- TrackAcquire(data_.generic);
- }
- // Move constructor. Allows initialization from a ScopedGeneric rvalue.
- ScopedGeneric(ScopedGeneric<T, Traits>&& rvalue)
- : data_(rvalue.release(), rvalue.get_traits()) {
- TrackAcquire(data_.generic);
- }
- ScopedGeneric(const ScopedGeneric&) = delete;
- ScopedGeneric& operator=(const ScopedGeneric&) = delete;
- virtual ~ScopedGeneric() {
- CHECK(!receiving_) << "ScopedGeneric destroyed with active receiver";
- FreeIfNecessary();
- }
- // operator=. Allows assignment from a ScopedGeneric rvalue.
- ScopedGeneric& operator=(ScopedGeneric<T, Traits>&& rvalue) {
- reset(rvalue.release());
- return *this;
- }
- // Frees the currently owned object, if any. Then takes ownership of a new
- // object, if given. Self-resets are not allowd as on unique_ptr. See
- // http://crbug.com/162971
- void reset(const element_type& value = traits_type::InvalidValue()) {
- if (data_.generic != traits_type::InvalidValue() && data_.generic == value)
- abort();
- FreeIfNecessary();
- data_.generic = value;
- TrackAcquire(value);
- }
- void swap(ScopedGeneric& other) {
- if (&other == this) {
- return;
- }
- TrackRelease(data_.generic);
- other.TrackRelease(other.data_.generic);
- // Standard swap idiom: 'using std::swap' ensures that std::swap is
- // present in the overload set, but we call swap unqualified so that
- // any more-specific overloads can be used, if available.
- using std::swap;
- swap(static_cast<Traits&>(data_), static_cast<Traits&>(other.data_));
- swap(data_.generic, other.data_.generic);
- TrackAcquire(data_.generic);
- other.TrackAcquire(other.data_.generic);
- }
- // Release the object. The return value is the current object held by this
- // object. After this operation, this object will hold a null value, and
- // will not own the object any more.
- element_type release() WARN_UNUSED_RESULT {
- element_type old_generic = data_.generic;
- data_.generic = traits_type::InvalidValue();
- TrackRelease(old_generic);
- return old_generic;
- }
- // A helper class that provides a T* that can be used to take ownership of
- // a value returned from a function via out-parameter. When the Receiver is
- // destructed (which should usually be at the end of the statement in which
- // receive is called), ScopedGeneric::reset() will be called with the
- // Receiver's value.
- //
- // In the simple case of a function that assigns the value before it returns,
- // C++'s lifetime extension can be used as follows:
- //
- // ScopedFoo foo;
- // bool result = GetFoo(ScopedFoo::Receiver(foo).get());
- //
- // Note that the lifetime of the Receiver is extended until the semicolon,
- // and ScopedGeneric is assigned the value upon destruction of the Receiver,
- // so the following code would not work:
- //
- // // BROKEN!
- // ScopedFoo foo;
- // UseFoo(&foo, GetFoo(ScopedFoo::Receiver(foo).get()));
- //
- // In more complicated scenarios, you may need to provide an explicit scope
- // for the Receiver, as in the following:
- //
- // std::vector<ScopedFoo> foos(64);
- //
- // {
- // std::vector<ScopedFoo::Receiver> foo_receivers;
- // for (auto foo : foos) {
- // foo_receivers_.emplace_back(foo);
- // }
- // for (auto receiver : foo_receivers) {
- // SubmitGetFooRequest(receiver.get());
- // }
- // WaitForFooRequests();
- // }
- // UseFoos(foos);
- class Receiver {
- public:
- explicit Receiver(ScopedGeneric& parent) : scoped_generic_(&parent) {
- CHECK(!scoped_generic_->receiving_)
- << "attempted to construct Receiver for ScopedGeneric with existing "
- "Receiver";
- scoped_generic_->receiving_ = true;
- }
- Receiver(const Receiver&) = delete;
- Receiver& operator=(const Receiver&) = delete;
- Receiver(Receiver&& move) {
- CHECK(!used_) << "moving into already-used Receiver";
- CHECK(!move.used_) << "moving from already-used Receiver";
- scoped_generic_ = move.scoped_generic_;
- move.scoped_generic_ = nullptr;
- }
- Receiver& operator=(Receiver&& move) {
- CHECK(!used_) << "moving into already-used Receiver";
- CHECK(!move.used_) << "moving from already-used Receiver";
- scoped_generic_ = move.scoped_generic_;
- move.scoped_generic_ = nullptr;
- }
- ~Receiver() {
- if (scoped_generic_) {
- CHECK(scoped_generic_->receiving_);
- scoped_generic_->reset(value_);
- scoped_generic_->receiving_ = false;
- }
- }
- // We hand out a pointer to a field in Receiver instead of directly to
- // ScopedGeneric's internal storage in order to make it so that users can't
- // accidentally silently break ScopedGeneric's invariants. This way, an
- // incorrect use-after-scope-exit is more detectable by ASan or static
- // analysis tools, as the pointer is only valid for the lifetime of the
- // Receiver, not the ScopedGeneric.
- T* get() {
- used_ = true;
- return &value_;
- }
- private:
- T value_ = Traits::InvalidValue();
- ScopedGeneric* scoped_generic_;
- bool used_ = false;
- };
- const element_type& get() const { return data_.generic; }
- // Returns true if this object doesn't hold the special null value for the
- // associated data type.
- bool is_valid() const { return data_.generic != traits_type::InvalidValue(); }
- bool operator==(const element_type& value) const {
- return data_.generic == value;
- }
- bool operator!=(const element_type& value) const {
- return data_.generic != value;
- }
- Traits& get_traits() { return data_; }
- const Traits& get_traits() const { return data_; }
- private:
- void FreeIfNecessary() {
- if (data_.generic != traits_type::InvalidValue()) {
- TrackRelease(data_.generic);
- data_.Free(data_.generic);
- data_.generic = traits_type::InvalidValue();
- }
- }
- template <typename Void = void>
- typename std::enable_if_t<
- std::is_base_of<ScopedGenericOwnershipTracking, Traits>::value,
- Void>
- TrackAcquire(const T& value) {
- if (value != traits_type::InvalidValue()) {
- data_.Acquire(static_cast<const ScopedGeneric&>(*this), value);
- }
- }
- template <typename Void = void>
- typename std::enable_if_t<
- !std::is_base_of<ScopedGenericOwnershipTracking, Traits>::value,
- Void>
- TrackAcquire(const T& value) {}
- template <typename Void = void>
- typename std::enable_if_t<
- std::is_base_of<ScopedGenericOwnershipTracking, Traits>::value,
- Void>
- TrackRelease(const T& value) {
- if (value != traits_type::InvalidValue()) {
- data_.Release(static_cast<const ScopedGeneric&>(*this), value);
- }
- }
- template <typename Void = void>
- typename std::enable_if_t<
- !std::is_base_of<ScopedGenericOwnershipTracking, Traits>::value,
- Void>
- TrackRelease(const T& value) {}
- // Forbid comparison. If U != T, it totally doesn't make sense, and if U ==
- // T, it still doesn't make sense because you should never have the same
- // object owned by two different ScopedGenerics.
- template <typename T2, typename Traits2> bool operator==(
- const ScopedGeneric<T2, Traits2>& p2) const;
- template <typename T2, typename Traits2> bool operator!=(
- const ScopedGeneric<T2, Traits2>& p2) const;
- Data data_;
- bool receiving_ = false;
- };
- template<class T, class Traits>
- void swap(const ScopedGeneric<T, Traits>& a,
- const ScopedGeneric<T, Traits>& b) {
- a.swap(b);
- }
- template<class T, class Traits>
- bool operator==(const T& value, const ScopedGeneric<T, Traits>& scoped) {
- return value == scoped.get();
- }
- template<class T, class Traits>
- bool operator!=(const T& value, const ScopedGeneric<T, Traits>& scoped) {
- return value != scoped.get();
- }
- } // namespace base
- #endif // BASE_SCOPED_GENERIC_H_
|