Reviewed By: jvillard Differential Revision: D4705978 fbshipit-source-id: a635fb8master
parent
707d514386
commit
f65ac4f094
@ -0,0 +1,51 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2017 - present Facebook, Inc.
|
||||||
|
* All rights reserved.
|
||||||
|
*
|
||||||
|
* This source code is licensed under the BSD style license found in the
|
||||||
|
* LICENSE file in the root directory of this source tree. An additional grant
|
||||||
|
* of patent rights can be found in the PATENTS file in the same directory.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <infer_model/infer_traits.h>
|
||||||
|
#include <infer_model/portability.h>
|
||||||
|
|
||||||
|
INFER_NAMESPACE_STD_BEGIN
|
||||||
|
|
||||||
|
int __INFER_UNLOCKED = 0;
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
void __infer_mutex_deref_first_arg(T* ptr) INFER_MODEL_AS_DEREF_FIRST_ARG;
|
||||||
|
|
||||||
|
class mutex {
|
||||||
|
private:
|
||||||
|
int* null_if_locked = &__INFER_UNLOCKED;
|
||||||
|
|
||||||
|
public:
|
||||||
|
constexpr mutex() noexcept;
|
||||||
|
~mutex();
|
||||||
|
mutex(const mutex&) = delete;
|
||||||
|
mutex& operator=(const mutex&) = delete;
|
||||||
|
|
||||||
|
void lock() {
|
||||||
|
__infer_mutex_deref_first_arg(null_if_locked); // HACK
|
||||||
|
null_if_locked = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool try_lock() {
|
||||||
|
if (null_if_locked) {
|
||||||
|
null_if_locked = nullptr;
|
||||||
|
return true;
|
||||||
|
} else {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void unlock() {
|
||||||
|
null_if_locked = &__INFER_UNLOCKED;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
INFER_NAMESPACE_STD_END
|
@ -0,0 +1,74 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2017 - present Facebook, Inc.
|
||||||
|
* All rights reserved.
|
||||||
|
*
|
||||||
|
* This source code is licensed under the BSD style license found in the
|
||||||
|
* LICENSE file in the root directory of this source tree. An additional grant
|
||||||
|
* of patent rights can be found in the PATENTS file in the same directory.
|
||||||
|
*/
|
||||||
|
#include <mutex>
|
||||||
|
|
||||||
|
void alarm1(std::mutex& m) {
|
||||||
|
m.lock();
|
||||||
|
m.lock();
|
||||||
|
}
|
||||||
|
|
||||||
|
void alarm2(std::mutex& m) {
|
||||||
|
m.try_lock();
|
||||||
|
m.lock();
|
||||||
|
}
|
||||||
|
|
||||||
|
void no_alarm1(std::mutex& m) {
|
||||||
|
for (int i = 0; i < 999999; i++) {
|
||||||
|
m.try_lock();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void ensure_locked(std::mutex& m) {
|
||||||
|
m.try_lock();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ensure_unlocked(std::mutex& m) {
|
||||||
|
ensure_locked(m);
|
||||||
|
m.unlock();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ends_locked(std::mutex& m) {
|
||||||
|
for (int i = 0; i < 999999; i++) {
|
||||||
|
ensure_unlocked(m);
|
||||||
|
ensure_locked(m);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void starts_with_lock(std::mutex& m) {
|
||||||
|
for (int i = 0; i < 999999; i++) {
|
||||||
|
m.lock();
|
||||||
|
ensure_unlocked(m);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void alarm3() {
|
||||||
|
std::mutex m;
|
||||||
|
ends_locked(m);
|
||||||
|
starts_with_lock(m);
|
||||||
|
}
|
||||||
|
|
||||||
|
void no_alarm3(bool b) {
|
||||||
|
std::mutex m;
|
||||||
|
if (b) {
|
||||||
|
m.lock();
|
||||||
|
}
|
||||||
|
if (b) {
|
||||||
|
m.unlock();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void alarm3(bool b) {
|
||||||
|
std::mutex m;
|
||||||
|
if (b) {
|
||||||
|
m.lock();
|
||||||
|
}
|
||||||
|
if (b) {
|
||||||
|
m.lock();
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,88 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2017 - present Facebook, Inc.
|
||||||
|
* All rights reserved.
|
||||||
|
*
|
||||||
|
* This source code is licensed under the BSD style license found in the
|
||||||
|
* LICENSE file in the root directory of this source tree. An additional grant
|
||||||
|
* of patent rights can be found in the PATENTS file in the same directory.
|
||||||
|
*/
|
||||||
|
#include <mutex>
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
inline int lock_impl(std::mutex* lock) {
|
||||||
|
lock->lock();
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline int try_lock_impl(std::mutex* lock) {
|
||||||
|
if (lock->try_lock()) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return EBUSY;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
int lp_lock(T* lock) {
|
||||||
|
int rv = detail::try_lock_impl(lock);
|
||||||
|
|
||||||
|
switch (rv) {
|
||||||
|
case 0:
|
||||||
|
return 0;
|
||||||
|
case EBUSY: {
|
||||||
|
rv = detail::lock_impl(lock);
|
||||||
|
if (rv == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
/* fallthrough */
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
return rv;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void good_usage() {
|
||||||
|
std::mutex m;
|
||||||
|
lp_lock(&m);
|
||||||
|
}
|
||||||
|
|
||||||
|
void bad_usage1() {
|
||||||
|
std::mutex m;
|
||||||
|
lp_lock(&m);
|
||||||
|
lp_lock(&m);
|
||||||
|
}
|
||||||
|
|
||||||
|
void bad_usage2() {
|
||||||
|
std::mutex m;
|
||||||
|
m.lock();
|
||||||
|
lp_lock(&m);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
class LpLockGuard {
|
||||||
|
public:
|
||||||
|
LpLockGuard(T& lock)
|
||||||
|
: lock_(lock) {
|
||||||
|
lp_lock(&lock_);
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
T& lock_;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct LockMapBucket {
|
||||||
|
public:
|
||||||
|
void good_usage2() {
|
||||||
|
LpLockGuard<std::mutex> lock(bucketLock);
|
||||||
|
}
|
||||||
|
|
||||||
|
void bad_usage3() {
|
||||||
|
LpLockGuard<std::mutex> lock(bucketLock);
|
||||||
|
this->good_usage2();
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::mutex bucketLock; // Lock protecting against concurrent access
|
||||||
|
};
|
Loading…
Reference in new issue