149 lines
5.4 KiB
C++
149 lines
5.4 KiB
C++
// SPDX-FileCopyrightText: Copyright 2020 yuzu Emulator Project
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#pragma once
|
|
|
|
#include "common/common_types.h"
|
|
|
|
#if _MSC_VER
|
|
#include <intrin.h>
|
|
#else
|
|
#include <cstring>
|
|
#endif
|
|
|
|
namespace Common {
|
|
|
|
#if _MSC_VER
|
|
|
|
template <typename T>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap(T* pointer, T value, T expected);
|
|
template <typename T>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap(T* pointer, T value, T expected, T& actual);
|
|
|
|
template <>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap<u8>(u8* pointer, u8 value, u8 expected) {
|
|
const u8 result =
|
|
_InterlockedCompareExchange8(reinterpret_cast<volatile char*>(pointer), value, expected);
|
|
return result == expected;
|
|
}
|
|
|
|
template <>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap<u16>(u16* pointer, u16 value, u16 expected) {
|
|
const u16 result =
|
|
_InterlockedCompareExchange16(reinterpret_cast<volatile short*>(pointer), value, expected);
|
|
return result == expected;
|
|
}
|
|
|
|
template <>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap<u32>(u32* pointer, u32 value, u32 expected) {
|
|
const u32 result =
|
|
_InterlockedCompareExchange(reinterpret_cast<volatile long*>(pointer), value, expected);
|
|
return result == expected;
|
|
}
|
|
|
|
template <>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap<u64>(u64* pointer, u64 value, u64 expected) {
|
|
const u64 result = _InterlockedCompareExchange64(reinterpret_cast<volatile __int64*>(pointer),
|
|
value, expected);
|
|
return result == expected;
|
|
}
|
|
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap(volatile u64* pointer, u128 value, u128 expected) {
|
|
return _InterlockedCompareExchange128(reinterpret_cast<volatile __int64*>(pointer), value[1],
|
|
value[0],
|
|
reinterpret_cast<__int64*>(expected.data())) != 0;
|
|
}
|
|
|
|
template <>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap<u8>(u8* pointer, u8 value, u8 expected, u8& actual) {
|
|
actual =
|
|
_InterlockedCompareExchange8(reinterpret_cast<volatile char*>(pointer), value, expected);
|
|
return actual == expected;
|
|
}
|
|
|
|
template <>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap<u16>(u16* pointer, u16 value, u16 expected,
|
|
u16& actual) {
|
|
actual =
|
|
_InterlockedCompareExchange16(reinterpret_cast<volatile short*>(pointer), value, expected);
|
|
return actual == expected;
|
|
}
|
|
|
|
template <>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap<u32>(u32* pointer, u32 value, u32 expected,
|
|
u32& actual) {
|
|
actual =
|
|
_InterlockedCompareExchange(reinterpret_cast<volatile long*>(pointer), value, expected);
|
|
return actual == expected;
|
|
}
|
|
|
|
template <>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap<u64>(u64* pointer, u64 value, u64 expected,
|
|
u64& actual) {
|
|
actual = _InterlockedCompareExchange64(reinterpret_cast<volatile __int64*>(pointer), value,
|
|
expected);
|
|
return actual == expected;
|
|
}
|
|
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap(volatile u64* pointer, u128 value, u128 expected,
|
|
u128& actual) {
|
|
const bool result =
|
|
_InterlockedCompareExchange128(reinterpret_cast<volatile __int64*>(pointer), value[1],
|
|
value[0], reinterpret_cast<__int64*>(expected.data())) != 0;
|
|
actual = expected;
|
|
return result;
|
|
}
|
|
|
|
[[nodiscard]] inline u128 AtomicLoad128(volatile u64* pointer) {
|
|
u128 result{};
|
|
_InterlockedCompareExchange128(reinterpret_cast<volatile __int64*>(pointer), result[1],
|
|
result[0], reinterpret_cast<__int64*>(result.data()));
|
|
return result;
|
|
}
|
|
|
|
#else
|
|
|
|
template <typename T>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap(T* pointer, T value, T expected) {
|
|
return __sync_bool_compare_and_swap(pointer, expected, value);
|
|
}
|
|
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap(u64* pointer, u128 value, u128 expected) {
|
|
unsigned __int128 value_a;
|
|
unsigned __int128 expected_a;
|
|
std::memcpy(&value_a, value.data(), sizeof(u128));
|
|
std::memcpy(&expected_a, expected.data(), sizeof(u128));
|
|
return __sync_bool_compare_and_swap((unsigned __int128*)pointer, expected_a, value_a);
|
|
}
|
|
|
|
template <typename T>
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap(T* pointer, T value, T expected, T& actual) {
|
|
actual = __sync_val_compare_and_swap(pointer, expected, value);
|
|
return actual == expected;
|
|
}
|
|
|
|
[[nodiscard]] inline bool AtomicCompareAndSwap(u64* pointer, u128 value, u128 expected,
|
|
u128& actual) {
|
|
unsigned __int128 value_a;
|
|
unsigned __int128 expected_a;
|
|
unsigned __int128 actual_a;
|
|
std::memcpy(&value_a, value.data(), sizeof(u128));
|
|
std::memcpy(&expected_a, expected.data(), sizeof(u128));
|
|
actual_a = __sync_val_compare_and_swap((unsigned __int128*)pointer, expected_a, value_a);
|
|
std::memcpy(actual.data(), &actual_a, sizeof(u128));
|
|
return actual_a == expected_a;
|
|
}
|
|
|
|
[[nodiscard]] inline u128 AtomicLoad128(u64* pointer) {
|
|
unsigned __int128 zeros_a = 0;
|
|
unsigned __int128 result_a =
|
|
__sync_val_compare_and_swap((unsigned __int128*)pointer, zeros_a, zeros_a);
|
|
|
|
u128 result;
|
|
std::memcpy(result.data(), &result_a, sizeof(u128));
|
|
return result;
|
|
}
|
|
|
|
#endif
|
|
|
|
} // namespace Common
|