#pragma once
-#include <atomic>
-#include <type_traits>
#include <folly/Traits.h>
-#include <string.h>
+#include <folly/detail/AtomicUtils.h>
#include <stdint.h>
+#include <string.h>
+#include <atomic>
+#include <type_traits>
namespace folly {
}
bool compare_exchange_strong(
- T& v0, T v1,
- std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ T& v0,
+ T v1,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_strong(
+ v0, v1, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_strong(
+ T& v0,
+ T v1,
+ std::memory_order success,
+ std::memory_order failure) noexcept {
Raw d0 = encode(v0);
- bool rv = data.compare_exchange_strong(d0, encode(v1), mo);
+ bool rv = data.compare_exchange_strong(d0, encode(v1), success, failure);
if (!rv) {
v0 = decode(d0);
}
}
bool compare_exchange_weak(
- T& v0, T v1,
- std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ T& v0,
+ T v1,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_weak(
+ v0, v1, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_weak(
+ T& v0,
+ T v1,
+ std::memory_order success,
+ std::memory_order failure) noexcept {
Raw d0 = encode(v0);
- bool rv = data.compare_exchange_weak(d0, encode(v1), mo);
+ bool rv = data.compare_exchange_weak(d0, encode(v1), success, failure);
if (!rv) {
v0 = decode(d0);
}
CPortability.h \
detail/AtomicHashUtils.h \
detail/AtomicUnorderedMapUtils.h \
+ detail/AtomicUtils.h \
detail/BitIteratorDetail.h \
detail/BitsDetail.h \
detail/CacheLocality.h \
Executor.h \
Expected.h \
experimental/AsymmetricMemoryBarrier.h \
+ experimental/AtomicSharedPtr.h \
+ experimental/detail/AtomicSharedPtr-detail.h \
experimental/AutoTimer.h \
experimental/Bits.h \
experimental/BitVectorCoding.h \
"PackedSyncPtr should be only 8 bytes---something is "
"messed up");
+template <typename T>
+std::ostream& operator<<(std::ostream& os, const PackedSyncPtr<T>& ptr) {
+ os << "PackedSyncPtr(" << ptr.get() << ", " << ptr.extra() << ")";
+ return os;
+}
}
--- /dev/null
+/*
+ * Copyright 2017 Facebook, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#pragma once
+
+#include <atomic>
+
+namespace folly {
+namespace detail {
+
+inline std::memory_order default_failure_memory_order(
+ std::memory_order successMode) {
+ switch (successMode) {
+ case std::memory_order_acq_rel:
+ return std::memory_order_acquire;
+ case std::memory_order_release:
+ return std::memory_order_relaxed;
+ default:
+ return successMode;
+ }
+}
+}
+} // namespace
--- /dev/null
+/*
+ * Copyright 2017-present Facebook, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#pragma once
+
+#include <folly/AtomicStruct.h>
+#include <folly/PackedSyncPtr.h>
+#include <folly/detail/AtomicUtils.h>
+#include <folly/experimental/detail/AtomicSharedPtr-detail.h>
+#include <atomic>
+#include <thread>
+
+/*
+ * This is an implementation of the std::atomic_shared_ptr TS
+ * http://en.cppreference.com/w/cpp/experimental/atomic_shared_ptr
+ * https://isocpp.org/files/papers/N4162.pdf
+ *
+ * AFAIK, the only other implementation is Anthony Williams from
+ * Just::thread library:
+ *
+ * https://bitbucket.org/anthonyw/atomic_shared_ptr
+ *
+ * implementation details:
+ *
+ * Basically, three things need to be atomically exchanged to make this work:
+ * * the local count
+ * * the pointer to the control block
+ * * the aliased pointer, if any.
+ *
+ * The Williams version does it with DWcas: 32 bits for local count, 64
+ * bits for control block ptr, and he changes the shared_ptr
+ * implementation to also store the aliased pointers using a linked list
+ * like structure, and provides 32-bit index accessors to them (like
+ * IndexedMemPool trick).
+ *
+ * This version instead stores the 48 bits of address, plus 16 bits of
+ * local count in a single 8byte pointer. This avoids 'lock cmpxchg16b',
+ * which is much slower than 'lock xchg' in the normal 'store' case. In
+ * the less-common aliased pointer scenaro, we just allocate it in a new
+ * block, and store a pointer to that instead.
+ *
+ * Note that even if we only want to use the 3-bits of pointer alignment,
+ * this trick should still work - Any more than 4 concurrent accesses
+ * will have to go to an external map count instead (slower, but lots of
+ * concurrent access will be slow anyway due to bouncing cachelines).
+ *
+ * As a perf optimization, we currently batch up local count and only
+ * move it global every once in a while. This means load() is usually
+ * only a single atomic operation, instead of 3. For this trick to work,
+ * we probably need at least 8 bits to make batching worth it.
+ */
+
+// A note on noexcept: If the pointer is an aliased pointer,
+// store() will allocate. Otherwise is noexcept.
+namespace folly {
+
+template <
+ typename T,
+ template <typename> class Atom = std::atomic,
+ typename CountedDetail = detail::shared_ptr_internals>
+class atomic_shared_ptr {
+ using SharedPtr = typename CountedDetail::template CountedPtr<T>;
+ using BasePtr = typename CountedDetail::counted_base;
+ using PackedPtr = folly::PackedSyncPtr<BasePtr>;
+
+ public:
+ atomic_shared_ptr() noexcept {
+ init();
+ }
+ explicit atomic_shared_ptr(SharedPtr foo) /* noexcept */
+ : atomic_shared_ptr() {
+ store(foo);
+ }
+ atomic_shared_ptr(const atomic_shared_ptr<T>&) = delete;
+
+ ~atomic_shared_ptr() {
+ store(SharedPtr(nullptr));
+ }
+ void operator=(SharedPtr desired) /* noexcept */ {
+ store(desired);
+ }
+ void operator=(const atomic_shared_ptr<T>&) = delete;
+
+ bool is_lock_free() const noexcept {
+ // lock free unless more than EXTERNAL_OFFSET threads are
+ // contending and they all get unlucky and scheduled out during
+ // load().
+ //
+ // TODO: Could use a lock-free external map to fix this
+ // corner case.
+ return true;
+ }
+
+ SharedPtr load(std::memory_order order = std::memory_order_seq_cst) const
+ noexcept {
+ auto local = takeOwnedBase(order);
+
+ auto res = get_shared_ptr(local, false);
+
+ return std::move(res);
+ }
+ /* implicit */ operator SharedPtr() const {
+ return load();
+ }
+
+ void store(
+ SharedPtr n,
+ std::memory_order order = std::memory_order_seq_cst) /* noexcept */ {
+ auto newptr = get_newptr(std::move(n));
+ auto old = ptr_.exchange(newptr, order);
+ release_external(old);
+ }
+
+ SharedPtr exchange(
+ SharedPtr n,
+ std::memory_order order = std::memory_order_seq_cst) /* noexcept */ {
+ auto newptr = get_newptr(std::move(n));
+ auto old = ptr_.exchange(newptr, order);
+
+ SharedPtr old_ptr;
+
+ if (old.get()) {
+ old_ptr = get_shared_ptr(old);
+ release_external(old);
+ }
+
+ return old_ptr;
+ }
+
+ bool compare_exchange_weak(
+ SharedPtr& expected,
+ const SharedPtr& n,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_weak(
+ expected, n, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_weak(
+ SharedPtr& expected,
+ const SharedPtr& n,
+ std::memory_order success,
+ std::memory_order failure) /* noexcept */ {
+ auto newptr = get_newptr(n);
+ PackedPtr oldptr, expectedptr;
+
+ oldptr = takeOwnedBase(success);
+ if (!owners_eq(oldptr, CountedDetail::get_counted_base(expected))) {
+ expected = get_shared_ptr(oldptr, false);
+ release_external(newptr);
+ return false;
+ }
+ expectedptr = oldptr; // Need oldptr to release if failed
+ if (ptr_.compare_exchange_weak(expectedptr, newptr, success, failure)) {
+ if (oldptr.get()) {
+ release_external(oldptr, -1);
+ }
+ return true;
+ } else {
+ if (oldptr.get()) {
+ expected = get_shared_ptr(oldptr, false);
+ } else {
+ expected = SharedPtr(nullptr);
+ }
+ release_external(newptr);
+ return false;
+ }
+ }
+ bool compare_exchange_weak(
+ SharedPtr& expected,
+ SharedPtr&& desired,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_weak(
+ expected, desired, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_weak(
+ SharedPtr& expected,
+ SharedPtr&& desired,
+ std::memory_order success,
+ std::memory_order failure) /* noexcept */ {
+ return compare_exchange_weak(expected, desired, success, failure);
+ }
+ bool compare_exchange_strong(
+ SharedPtr& expected,
+ const SharedPtr& n,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_strong(
+ expected, n, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_strong(
+ SharedPtr& expected,
+ const SharedPtr& n,
+ std::memory_order success,
+ std::memory_order failure) /* noexcept */ {
+ auto local_expected = expected;
+ do {
+ if (compare_exchange_weak(expected, n, success, failure)) {
+ return true;
+ }
+ } while (local_expected == expected);
+
+ return false;
+ }
+ bool compare_exchange_strong(
+ SharedPtr& expected,
+ SharedPtr&& desired,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_strong(
+ expected, desired, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_strong(
+ SharedPtr& expected,
+ SharedPtr&& desired,
+ std::memory_order success,
+ std::memory_order failure) /* noexcept */ {
+ return compare_exchange_strong(expected, desired, success, failure);
+ }
+
+ private:
+ // Matches packed_sync_pointer. Must be > max number of local
+ // counts. This is the max number of threads that can access this
+ // atomic_shared_ptr at once before we start blocking.
+ static constexpr unsigned EXTERNAL_OFFSET{0x2000};
+ // Bit signifying aliased constructor
+ static constexpr unsigned ALIASED_PTR{0x4000};
+
+ mutable AtomicStruct<PackedPtr, Atom> ptr_;
+
+ void add_external(BasePtr* res, int64_t c = 0) const {
+ assert(res);
+ CountedDetail::inc_shared_count(res, EXTERNAL_OFFSET + c);
+ }
+ void release_external(PackedPtr& res, int64_t c = 0) const {
+ if (!res.get()) {
+ return;
+ }
+ int64_t count = get_local_count(res) + c;
+ int64_t diff = EXTERNAL_OFFSET - count;
+ assert(diff >= 0);
+ CountedDetail::template release_shared<T>(res.get(), diff);
+ }
+ PackedPtr get_newptr(const SharedPtr& n) const {
+ BasePtr* newval;
+ unsigned count = 0;
+ if (!n) {
+ newval = nullptr;
+ } else {
+ newval = CountedDetail::get_counted_base(n);
+ if (n.get() != CountedDetail::template get_shared_ptr<T>(newval)) {
+ // This is an aliased sharedptr. Make an un-aliased one
+ // by wrapping in *another* shared_ptr.
+ auto data = CountedDetail::template make_ptr<SharedPtr>(n);
+ newval = CountedDetail::get_counted_base(data);
+ count = ALIASED_PTR;
+ // (add external must happen before data goes out of scope)
+ add_external(newval);
+ } else {
+ add_external(newval);
+ }
+ }
+
+ PackedPtr newptr;
+ newptr.init(newval, count);
+
+ return newptr;
+ }
+ PackedPtr get_newptr(SharedPtr&& n) const {
+ BasePtr* newval;
+ unsigned count = 0;
+ if (!n) {
+ newval = nullptr;
+ } else {
+ newval = CountedDetail::get_counted_base(n);
+ if (n.get() != CountedDetail::template get_shared_ptr<T>(newval)) {
+ // This is an aliased sharedptr. Make an un-aliased one
+ // by wrapping in *another* shared_ptr.
+ auto data = CountedDetail::template make_ptr<SharedPtr>(std::move(n));
+ newval = CountedDetail::get_counted_base(data);
+ count = ALIASED_PTR;
+ CountedDetail::release_ptr(data);
+ add_external(newval, -1);
+ } else {
+ CountedDetail::release_ptr(n);
+ add_external(newval, -1);
+ }
+ }
+
+ PackedPtr newptr;
+ newptr.init(newval, count);
+
+ return newptr;
+ }
+ void init() {
+ PackedPtr data;
+ data.init();
+ ptr_.store(data);
+ }
+
+ unsigned int get_local_count(const PackedPtr& p) const {
+ return p.extra() & ~ALIASED_PTR;
+ }
+
+ // Check pointer equality considering wrapped aliased pointers.
+ bool owners_eq(PackedPtr& p1, BasePtr* p2) {
+ bool aliased1 = p1.extra() & ALIASED_PTR;
+ if (aliased1) {
+ auto p1a = CountedDetail::template get_shared_ptr_from_counted_base<T>(
+ p1.get(), false);
+ return CountedDetail::get_counted_base(p1a) == p2;
+ }
+ return p1.get() == p2;
+ }
+
+ SharedPtr get_shared_ptr(const PackedPtr& p, bool inc = true) const {
+ bool aliased = p.extra() & ALIASED_PTR;
+
+ auto res = CountedDetail::template get_shared_ptr_from_counted_base<T>(
+ p.get(), inc);
+ if (aliased) {
+ auto aliasedp =
+ CountedDetail::template get_shared_ptr_from_counted_base<SharedPtr>(
+ p.get());
+ res = *aliasedp;
+ }
+ return std::move(res);
+ }
+
+ /* Get a reference to the pointer, either from the local batch or
+ * from the global count.
+ *
+ * return is the base ptr, and the previous local count, if it is
+ * needed for compare_and_swap later.
+ */
+ PackedPtr takeOwnedBase(std::memory_order order) const noexcept {
+ PackedPtr local, newlocal;
+ local = ptr_.load(std::memory_order_acquire);
+ while (true) {
+ if (!local.get()) {
+ return local;
+ }
+ newlocal = local;
+ if (get_local_count(newlocal) + 1 > EXTERNAL_OFFSET) {
+ // spinlock in the rare case we have more than
+ // EXTERNAL_OFFSET threads trying to access at once.
+ std::this_thread::yield();
+ // Force DeterministicSchedule to choose a different thread
+ local = ptr_.load(std::memory_order_acquire);
+ } else {
+ newlocal.setExtra(newlocal.extra() + 1);
+ assert(get_local_count(newlocal) > 0);
+ if (ptr_.compare_exchange_weak(local, newlocal, order)) {
+ break;
+ }
+ }
+ }
+
+ // Check if we need to push a batch from local -> global
+ auto batchcount = EXTERNAL_OFFSET / 2;
+ if (get_local_count(newlocal) > batchcount) {
+ CountedDetail::inc_shared_count(newlocal.get(), batchcount);
+ putOwnedBase(newlocal.get(), batchcount, order);
+ }
+
+ return newlocal;
+ }
+
+ void putOwnedBase(BasePtr* p, unsigned int count, std::memory_order mo) const
+ noexcept {
+ PackedPtr local = ptr_.load(std::memory_order_acquire);
+ while (true) {
+ if (local.get() != p) {
+ break;
+ }
+ auto newlocal = local;
+ if (get_local_count(local) > count) {
+ newlocal.setExtra(local.extra() - count);
+ } else {
+ // Otherwise it may be the same pointer, but someone else won
+ // the compare_exchange below, local count was already made
+ // global. We decrement the global count directly instead of
+ // the local one.
+ break;
+ }
+ if (ptr_.compare_exchange_weak(local, newlocal, mo)) {
+ return;
+ }
+ }
+
+ CountedDetail::template release_shared<T>(p, count);
+ }
+};
+
+} // namespace folly
--- /dev/null
+/*
+ * Copyright 2017-present Facebook, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#pragma once
+#include <atomic>
+#include <memory>
+
+namespace folly {
+namespace detail {
+
+class shared_ptr_internals {
+ public:
+ template <typename T, typename... Args>
+ static std::shared_ptr<T> make_ptr(Args&&... args) {
+ return std::make_shared<T>(std::forward<Args...>(args...));
+ }
+ typedef std::__shared_count<std::_S_atomic> shared_count;
+ typedef std::_Sp_counted_base<std::_S_atomic> counted_base;
+ template <typename T>
+ using CountedPtr = std::shared_ptr<T>;
+ template <typename T>
+ static counted_base* get_counted_base(const std::shared_ptr<T>& bar) {
+ // reinterpret_pointer_cast<const void>
+ // Not quite C++ legal, but explicit template instantiation access to
+ // private members requires full type name (i.e. shared_ptr<const void>, not
+ // shared_ptr<T>)
+ const std::shared_ptr<const void>& ptr(
+ reinterpret_cast<const std::shared_ptr<const void>&>(bar));
+ return (ptr.*fieldPtr(access_shared_ptr{})).*fieldPtr(access_base{});
+ }
+
+ static void inc_shared_count(counted_base* base, long count) {
+ __gnu_cxx::__atomic_add_dispatch(
+ &(base->*fieldPtr(access_use_count{})), count);
+ }
+
+ template <typename T>
+ static void release_shared(counted_base* base, long count) {
+ // If count == 1, this is equivalent to base->_M_release()
+ if (__gnu_cxx::__exchange_and_add_dispatch(
+ &(base->*fieldPtr(access_use_count{})), -count) == count) {
+ base->_M_dispose();
+
+ if (__gnu_cxx::__exchange_and_add_dispatch(
+ &(base->*fieldPtr(access_weak_count{})), -1) == 1) {
+ base->_M_destroy();
+ }
+ }
+ }
+
+ template <typename T>
+ static T* get_shared_ptr(counted_base* base) {
+ // See if this was a make_shared allocation
+ auto inplace = base->_M_get_deleter(typeid(std::_Sp_make_shared_tag));
+ if (inplace) {
+ return (T*)inplace;
+ }
+ // Could also be a _Sp_counted_deleter, but the layout is the same
+ auto ptr =
+ static_cast<std::_Sp_counted_ptr<const void*, std::_S_atomic>*>(base);
+ return (T*)(ptr->*fieldPtr(access_counted_ptr_ptr{}));
+ }
+
+ template <typename T>
+ static T* release_ptr(std::shared_ptr<T>& p) {
+ auto res = p.get();
+ std::shared_ptr<const void>& ptr(
+ reinterpret_cast<std::shared_ptr<const void>&>(p));
+ ptr.*fieldPtr(access_shared_ptr_ptr{}) = nullptr;
+ (ptr.*fieldPtr(access_refcount{})).*fieldPtr(access_base{}) = nullptr;
+ return res;
+ }
+
+ template <typename T>
+ static std::shared_ptr<T> get_shared_ptr_from_counted_base(
+ counted_base* base,
+ bool inc = true) {
+ if (!base) {
+ return nullptr;
+ }
+ std::shared_ptr<const void> newp;
+ if (inc) {
+ inc_shared_count(base, 1);
+ }
+ newp.*fieldPtr(access_shared_ptr_ptr{}) =
+ get_shared_ptr<const void>(base); // _M_ptr
+ (newp.*fieldPtr(access_refcount{})).*fieldPtr(access_base{}) = base;
+ // reinterpret_pointer_cast<T>
+ auto res = reinterpret_cast<std::shared_ptr<T>*>(&newp);
+ return std::move(*res);
+ }
+
+ private:
+ /* Accessors for private members using explicit template instantiation */
+ struct access_shared_ptr {
+ typedef shared_count std::__shared_ptr<const void, std::_S_atomic>::*type;
+ friend type fieldPtr(access_shared_ptr);
+ };
+
+ struct access_base {
+ typedef counted_base* shared_count::*type;
+ friend type fieldPtr(access_base);
+ };
+
+ struct access_use_count {
+ typedef _Atomic_word counted_base::*type;
+ friend type fieldPtr(access_use_count);
+ };
+
+ struct access_weak_count {
+ typedef _Atomic_word counted_base::*type;
+ friend type fieldPtr(access_weak_count);
+ };
+
+ struct access_counted_ptr_ptr {
+ typedef const void* std::_Sp_counted_ptr<const void*, std::_S_atomic>::*
+ type;
+ friend type fieldPtr(access_counted_ptr_ptr);
+ };
+
+ struct access_shared_ptr_ptr {
+ typedef const void* std::__shared_ptr<const void, std::_S_atomic>::*type;
+ friend type fieldPtr(access_shared_ptr_ptr);
+ };
+
+ struct access_refcount {
+ typedef shared_count std::__shared_ptr<const void, std::_S_atomic>::*type;
+ friend type fieldPtr(access_refcount);
+ };
+
+ template <typename Tag, typename Tag::type M>
+ struct Rob {
+ friend typename Tag::type fieldPtr(Tag) {
+ return M;
+ }
+ };
+};
+
+template struct shared_ptr_internals::Rob<
+ shared_ptr_internals::access_shared_ptr,
+ &std::__shared_ptr<const void, std::_S_atomic>::_M_refcount>;
+template struct shared_ptr_internals::Rob<
+ shared_ptr_internals::access_base,
+ &shared_ptr_internals::shared_count::_M_pi>;
+template struct shared_ptr_internals::Rob<
+ shared_ptr_internals::access_use_count,
+ &shared_ptr_internals::counted_base::_M_use_count>;
+template struct shared_ptr_internals::Rob<
+ shared_ptr_internals::access_weak_count,
+ &shared_ptr_internals::counted_base::_M_weak_count>;
+template struct shared_ptr_internals::Rob<
+ shared_ptr_internals::access_counted_ptr_ptr,
+ &std::_Sp_counted_ptr<const void*, std::_S_atomic>::_M_ptr>;
+template struct shared_ptr_internals::Rob<
+ shared_ptr_internals::access_shared_ptr_ptr,
+ &std::__shared_ptr<const void, std::_S_atomic>::_M_ptr>;
+template struct shared_ptr_internals::Rob<
+ shared_ptr_internals::access_refcount,
+ &std::__shared_ptr<const void, std::_S_atomic>::_M_refcount>;
+
+} // namespace detail
+} // namespace folly
--- /dev/null
+/*
+ * Copyright 2017-present Facebook, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#pragma once
+
+struct counted_shared_tag {};
+template <template <typename> class Atom = std::atomic>
+struct intrusive_shared_count {
+ intrusive_shared_count() {
+ counts.store(0);
+ }
+ void add_ref(uint64_t count = 1) {
+ counts.fetch_add(count);
+ }
+
+ uint64_t release_ref(uint64_t count = 1) {
+ return counts.fetch_sub(count);
+ }
+ Atom<uint64_t> counts;
+};
+
+template <template <typename> class Atom = std::atomic>
+struct counted_ptr_base {
+ protected:
+ static intrusive_shared_count<Atom>* getRef(void* pt) {
+ char* p = (char*)pt;
+ p -= sizeof(intrusive_shared_count<Atom>);
+ return (intrusive_shared_count<Atom>*)p;
+ }
+};
+
+// basically shared_ptr, but only supports make_counted, and provides
+// access to add_ref / release_ref with a count. Alias not supported.
+template <typename T, template <typename> class Atom = std::atomic>
+class counted_ptr : public counted_ptr_base<Atom> {
+ public:
+ T* p_;
+ counted_ptr() : p_(nullptr) {}
+ counted_ptr(counted_shared_tag, T* p) : p_(p) {
+ if (p_)
+ counted_ptr_base<Atom>::getRef(p_)->add_ref();
+ }
+
+ counted_ptr(const counted_ptr& o) : p_(o.p_) {
+ if (p_)
+ counted_ptr_base<Atom>::getRef(p_)->add_ref();
+ }
+ counted_ptr& operator=(const counted_ptr& o) {
+ if (p_ && counted_ptr_base<Atom>::getRef(p_)->release_ref() == 1) {
+ p_->~T();
+ free(counted_ptr_base<Atom>::getRef(p_));
+ }
+ p_ = o.p_;
+ if (p_)
+ counted_ptr_base<Atom>::getRef(p_)->add_ref();
+ return *this;
+ }
+ explicit counted_ptr(T* p) : p_(p) {
+ CHECK(!p);
+ }
+ ~counted_ptr() {
+ if (p_ && counted_ptr_base<Atom>::getRef(p_)->release_ref() == 1) {
+ p_->~T();
+ free(counted_ptr_base<Atom>::getRef(p_));
+ }
+ }
+ typename std::add_lvalue_reference<T>::type operator*() const {
+ return *p_;
+ }
+
+ T* get() const {
+ return p_;
+ }
+ T* operator->() const {
+ return p_;
+ }
+ explicit operator bool() const {
+ return p_ == nullptr ? false : true;
+ }
+ bool operator==(const counted_ptr<T, Atom>& p) const {
+ return get() == p.get();
+ }
+};
+
+template <
+ template <typename> class Atom = std::atomic,
+ typename T,
+ typename... Args>
+counted_ptr<T, Atom> make_counted(Args&&... args) {
+ char* mem = (char*)malloc(sizeof(T) + sizeof(intrusive_shared_count<Atom>));
+ if (!mem) {
+ throw std::bad_alloc();
+ }
+ new (mem) intrusive_shared_count<Atom>();
+ T* ptr = (T*)(mem + sizeof(intrusive_shared_count<Atom>));
+ new (ptr) T(std::forward<Args>(args)...);
+ return counted_ptr<T, Atom>(counted_shared_tag(), ptr);
+}
+
+template <template <typename> class Atom = std::atomic>
+class counted_ptr_internals : public counted_ptr_base<Atom> {
+ public:
+ template <typename T, typename... Args>
+ static counted_ptr<T, Atom> make_ptr(Args&&... args) {
+ return make_counted<Atom, T>(std::forward<Args...>(args...));
+ }
+ template <typename T>
+ using CountedPtr = counted_ptr<T, Atom>;
+ typedef void counted_base;
+
+ template <typename T>
+ static counted_base* get_counted_base(const counted_ptr<T, Atom>& bar) {
+ return bar.p_;
+ }
+
+ template <typename T>
+ static T* get_shared_ptr(counted_base* base) {
+ return (T*)base;
+ }
+
+ template <typename T>
+ static T* release_ptr(counted_ptr<T, Atom>& p) {
+ auto res = p.p_;
+ p.p_ = nullptr;
+ return res;
+ }
+
+ template <typename T>
+ static counted_ptr<T, Atom> get_shared_ptr_from_counted_base(
+ counted_base* base,
+ bool inc = true) {
+ auto res = counted_ptr<T, Atom>(counted_shared_tag(), (T*)(base));
+ if (!inc) {
+ release_shared<T>(base, 1);
+ }
+ return res;
+ }
+
+ static void inc_shared_count(counted_base* base, int64_t count) {
+ counted_ptr_base<Atom>::getRef(base)->add_ref(count);
+ }
+
+ template <typename T>
+ static void release_shared(counted_base* base, uint64_t count) {
+ if (count == counted_ptr_base<Atom>::getRef(base)->release_ref(count)) {
+ ((T*)base)->~T();
+ free(counted_ptr_base<Atom>::getRef(base));
+ }
+ }
+};
--- /dev/null
+/*
+ * Copyright 2017-present Facebook, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#include <atomic>
+#include <memory>
+#include <thread>
+
+#include <folly/experimental/AtomicSharedPtr.h>
+#include <folly/experimental/test/AtomicSharedPtrCounted.h>
+#include <folly/portability/GTest.h>
+
+#include <folly/test/DeterministicSchedule.h>
+
+using namespace folly;
+using namespace folly::test;
+using namespace std;
+static int c_count{0};
+static int d_count{0};
+
+using DSched = DeterministicSchedule;
+
+DEFINE_int64(seed, 0, "Seed for random number generators");
+DEFINE_int32(num_threads, 32, "Number of threads");
+
+struct foo {
+ foo() {
+ c_count++;
+ }
+ ~foo() {
+ d_count++;
+ }
+};
+
+TEST(AtomicSharedPtr, operators) {
+ atomic_shared_ptr<int> fooptr;
+ EXPECT_TRUE(fooptr.is_lock_free());
+ auto i = new int(5);
+ std::shared_ptr<int> s(i);
+ fooptr.store(s);
+ shared_ptr<int> bar(fooptr);
+ EXPECT_TRUE(fooptr.compare_exchange_strong(s, nullptr));
+ s.reset();
+ bar.reset();
+}
+
+TEST(AtomicSharedPtr, exchange) {
+ atomic_shared_ptr<int> fooptr;
+ auto a = make_shared<int>(1);
+ fooptr.store(std::move(a));
+ auto b = fooptr.exchange(make_shared<int>());
+ EXPECT_EQ(*b, 1);
+}
+
+TEST(AtomicSharedPtr, foo) {
+ c_count = 0;
+ d_count = 0;
+ {
+ atomic_shared_ptr<foo> fooptr;
+ fooptr.store(make_shared<foo>());
+ EXPECT_EQ(1, c_count);
+ EXPECT_EQ(0, d_count);
+ {
+ auto res = fooptr.load();
+ EXPECT_EQ(1, c_count);
+ EXPECT_EQ(0, d_count);
+ }
+ EXPECT_EQ(1, c_count);
+ EXPECT_EQ(0, d_count);
+ }
+ EXPECT_EQ(1, c_count);
+ EXPECT_EQ(1, d_count);
+}
+
+TEST(AtomicSharedPtr, counted) {
+ c_count = 0;
+ d_count = 0;
+ {
+ atomic_shared_ptr<foo, std::atomic, counted_ptr_internals<std::atomic>>
+ fooptr;
+ fooptr.store(make_counted<std::atomic, foo>());
+ EXPECT_EQ(1, c_count);
+ EXPECT_EQ(0, d_count);
+ {
+ auto res = fooptr.load();
+ EXPECT_EQ(1, c_count);
+ EXPECT_EQ(0, d_count);
+ }
+ EXPECT_EQ(1, c_count);
+ EXPECT_EQ(0, d_count);
+ }
+ EXPECT_EQ(1, c_count);
+ EXPECT_EQ(1, d_count);
+}
+
+TEST(AtomicSharedPtr, counted2) {
+ auto foo = make_counted<std::atomic, bool>();
+ atomic_shared_ptr<bool, std::atomic, counted_ptr_internals<std::atomic>>
+ fooptr(foo);
+ fooptr.store(foo);
+ fooptr.load();
+}
+
+TEST(AtomicSharedPtr, ConstTest) {
+ const auto a(std::make_shared<foo>());
+ atomic_shared_ptr<foo> atom;
+ atom.store(a);
+
+ atomic_shared_ptr<const foo> catom;
+}
+TEST(AtomicSharedPtr, AliasingConstructorTest) {
+ c_count = 0;
+ d_count = 0;
+ auto a = std::make_shared<foo>();
+ auto b = new foo;
+ auto alias = std::shared_ptr<foo>(a, b);
+
+ atomic_shared_ptr<foo> asp;
+ asp.store(alias);
+ a.reset();
+ alias.reset();
+ auto res1 = asp.load();
+ auto res2 = asp.exchange(nullptr);
+ EXPECT_EQ(b, res1.get());
+ EXPECT_EQ(b, res2.get());
+ EXPECT_EQ(2, c_count);
+ EXPECT_EQ(0, d_count);
+ res1.reset();
+ res2.reset();
+ EXPECT_EQ(2, c_count);
+ EXPECT_EQ(1, d_count);
+ delete b;
+ EXPECT_EQ(2, c_count);
+ EXPECT_EQ(2, d_count);
+}
+
+TEST(AtomicSharedPtr, DeterministicTest) {
+ DSched sched(DSched::uniform(FLAGS_seed));
+
+ auto foo = make_counted<DeterministicAtomic, bool>();
+ atomic_shared_ptr<
+ bool,
+ DeterministicAtomic,
+ counted_ptr_internals<DeterministicAtomic>>
+ fooptr(foo);
+ std::vector<std::thread> threads(FLAGS_num_threads);
+ for (int tid = 0; tid < FLAGS_num_threads; ++tid) {
+ threads[tid] = DSched::thread([&, tid]() {
+ for (int i = 0; i < 1000; i++) {
+ auto l = fooptr.load();
+ EXPECT_TRUE(l.get() != nullptr);
+ fooptr.compare_exchange_strong(l, l);
+ fooptr.store(make_counted<DeterministicAtomic, bool>());
+ EXPECT_FALSE(fooptr.compare_exchange_strong(
+ l, make_counted<DeterministicAtomic, bool>()));
+ }
+ });
+ }
+ for (auto& t : threads) {
+ DSched::join(t);
+ }
+}
#include <vector>
#include <folly/ScopeGuard.h>
+#include <folly/detail/AtomicUtils.h>
#include <folly/detail/CacheLocality.h>
#include <folly/detail/Futex.h>
bool is_lock_free() const noexcept { return data.is_lock_free(); }
bool compare_exchange_strong(
- T& v0, T v1, std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ T& v0,
+ T v1,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_strong(
+ v0, v1, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_strong(
+ T& v0,
+ T v1,
+ std::memory_order success,
+ std::memory_order failure) noexcept {
DeterministicSchedule::beforeSharedAccess();
auto orig = v0;
- bool rv = data.compare_exchange_strong(v0, v1, mo);
+ bool rv = data.compare_exchange_strong(v0, v1, success, failure);
FOLLY_TEST_DSCHED_VLOG(this << ".compare_exchange_strong(" << std::hex
<< orig << ", " << std::hex << v1 << ") -> "
<< rv << "," << std::hex << v0);
}
bool compare_exchange_weak(
- T& v0, T v1, std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ T& v0,
+ T v1,
+ std::memory_order mo = std::memory_order_seq_cst) noexcept {
+ return compare_exchange_weak(
+ v0, v1, mo, detail::default_failure_memory_order(mo));
+ }
+ bool compare_exchange_weak(
+ T& v0,
+ T v1,
+ std::memory_order success,
+ std::memory_order failure) noexcept {
DeterministicSchedule::beforeSharedAccess();
auto orig = v0;
- bool rv = data.compare_exchange_weak(v0, v1, mo);
+ bool rv = data.compare_exchange_weak(v0, v1, success, failure);
FOLLY_TEST_DSCHED_VLOG(this << ".compare_exchange_weak(" << std::hex << orig
<< ", " << std::hex << v1 << ") -> " << rv
<< "," << std::hex << v0);