+/*
+ * Copyright 2014 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
+
+/*
+ * This class provides a few spin lock implementations, depending on the
+ * platform. folly/SpinLock.h will select one of these as the folly::SpinLock
+ * implementation.
+ *
+ * The main reason we keep these separated out here is so that we can run unit
+ * tests for all supported spin lock implementations, even though only one will
+ * be selected as the actual folly::SpinLock implemenatation for any given
+ * platform.
+ */
+
+#include <boost/noncopyable.hpp>
+#include <folly/Portability.h>
+
+#if __x86_64__
+#include <folly/SmallLocks.h>
+
+namespace folly {
+
+class SpinLockMslImpl {
+ public:
+ FOLLY_ALWAYS_INLINE SpinLockMslImpl() {
+ lock_.init();
+ }
+ FOLLY_ALWAYS_INLINE void lock() const {
+ lock_.lock();
+ }
+ FOLLY_ALWAYS_INLINE void unlock() const {
+ lock_.unlock();
+ }
+ FOLLY_ALWAYS_INLINE bool trylock() const {
+ return lock_.try_lock();
+ }
+ private:
+ mutable folly::MicroSpinLock lock_;
+};
+
+}
+
+#endif // __x86_64__
+
+#if __APPLE__
+#include <libkern/OSAtomic.h>
+
+namespace folly {
+
+class SpinLockAppleImpl {
+ public:
+ FOLLY_ALWAYS_INLINE SpinLockAppleImpl() : lock_(0) {}
+ FOLLY_ALWAYS_INLINE void lock() const {
+ OSSpinLockLock(&lock_);
+ }
+ FOLLY_ALWAYS_INLINE void unlock() const {
+ OSSpinLockUnlock(&lock_);
+ }
+ FOLLY_ALWAYS_INLINE bool trylock() const {
+ return OSSpinLockTry(&lock_);
+ }
+ private:
+ mutable OSSpinLock lock_;
+};
+
+}
+
+#endif // __APPLE__
+
+#include <pthread.h>
+#include <folly/Exception.h>
+
+#if !__ANDROID__ && !__APPLE__
+
+// Apple and Android systems don't have pthread_spinlock_t, so we can't support
+// this version on those platforms.
+namespace folly {
+
+class SpinLockPthreadImpl {
+ public:
+ FOLLY_ALWAYS_INLINE SpinLockPthreadImpl() {
+ int rc = pthread_spin_init(&lock_, PTHREAD_PROCESS_PRIVATE);
+ checkPosixError(rc, "failed to initialize spinlock");
+ }
+ FOLLY_ALWAYS_INLINE ~SpinLockPthreadImpl() {
+ pthread_spin_destroy(&lock_);
+ }
+ void lock() const {
+ int rc = pthread_spin_lock(&lock_);
+ checkPosixError(rc, "error locking spinlock");
+ }
+ FOLLY_ALWAYS_INLINE void unlock() const {
+ int rc = pthread_spin_unlock(&lock_);
+ checkPosixError(rc, "error unlocking spinlock");
+ }
+ FOLLY_ALWAYS_INLINE bool trylock() const {
+ int rc = pthread_spin_trylock(&lock_);
+ if (rc == 0) {
+ return true;
+ } else if (rc == EBUSY) {
+ return false;
+ }
+ throwSystemErrorExplicit(rc, "spinlock trylock error");
+ }
+ private:
+ mutable pthread_spinlock_t lock_;
+};
+
+}
+
+#endif // !__ANDROID__ && !__APPLE__
+
+namespace folly {
+
+class SpinLockPthreadMutexImpl {
+ public:
+ FOLLY_ALWAYS_INLINE SpinLockPthreadMutexImpl() {
+ int rc = pthread_mutex_init(&lock_, nullptr);
+ checkPosixError(rc, "failed to initialize mutex");
+ }
+ FOLLY_ALWAYS_INLINE ~SpinLockPthreadMutexImpl() {
+ pthread_mutex_destroy(&lock_);
+ }
+ void lock() const {
+ int rc = pthread_mutex_lock(&lock_);
+ checkPosixError(rc, "error locking mutex");
+ }
+ FOLLY_ALWAYS_INLINE void unlock() const {
+ int rc = pthread_mutex_unlock(&lock_);
+ checkPosixError(rc, "error unlocking mutex");
+ }
+ FOLLY_ALWAYS_INLINE bool trylock() const {
+ int rc = pthread_mutex_trylock(&lock_);
+ if (rc == 0) {
+ return true;
+ } else if (rc == EBUSY) {
+ return false;
+ }
+ throwSystemErrorExplicit(rc, "mutex trylock error");
+ }
+ private:
+ mutable pthread_mutex_t lock_;
+};
+
+}