mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2025-09-05 22:32:55 +02:00
73 lines
1.1 KiB
C++
73 lines
1.1 KiB
C++
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*
|
|
* (c) ZeroTier, Inc.
|
|
* https://www.zerotier.com/
|
|
*/
|
|
|
|
#ifndef ZT_ATOMICCOUNTER_HPP
|
|
#define ZT_ATOMICCOUNTER_HPP
|
|
|
|
#ifndef __GNUC__
|
|
#include <atomic>
|
|
#endif
|
|
|
|
namespace ZeroTier {
|
|
|
|
/**
|
|
* Simple atomic counter supporting increment and decrement
|
|
*/
|
|
class AtomicCounter {
|
|
public:
|
|
AtomicCounter()
|
|
{
|
|
_v = 0;
|
|
}
|
|
|
|
inline int load() const
|
|
{
|
|
#ifdef __GNUC__
|
|
return __sync_or_and_fetch(const_cast<int*>(&_v), 0);
|
|
#else
|
|
return _v.load();
|
|
#endif
|
|
}
|
|
|
|
inline int operator++()
|
|
{
|
|
#ifdef __GNUC__
|
|
return __sync_add_and_fetch(&_v, 1);
|
|
#else
|
|
return ++_v;
|
|
#endif
|
|
}
|
|
|
|
inline int operator--()
|
|
{
|
|
#ifdef __GNUC__
|
|
return __sync_sub_and_fetch(&_v, 1);
|
|
#else
|
|
return --_v;
|
|
#endif
|
|
}
|
|
|
|
private:
|
|
AtomicCounter(const AtomicCounter&)
|
|
{
|
|
}
|
|
const AtomicCounter& operator=(const AtomicCounter&)
|
|
{
|
|
return *this;
|
|
}
|
|
|
|
#ifdef __GNUC__
|
|
int _v;
|
|
#else
|
|
std::atomic_int _v;
|
|
#endif
|
|
};
|
|
|
|
} // namespace ZeroTier
|
|
|
|
#endif
|