84 lines
3.1 KiB
C
84 lines
3.1 KiB
C
// Protocol Buffers - Google's data interchange format
|
|
// Copyright 2023 Google LLC. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#ifndef UPB_PORT_ATOMIC_H_
|
|
#define UPB_PORT_ATOMIC_H_
|
|
|
|
#include "upb/port/def.inc"
|
|
|
|
#ifdef UPB_USE_C11_ATOMICS
|
|
|
|
// IWYU pragma: begin_exports
|
|
#include <stdatomic.h>
|
|
#include <stdbool.h>
|
|
// IWYU pragma: end_exports
|
|
|
|
#define upb_Atomic_Init(addr, val) atomic_init(addr, val)
|
|
#define upb_Atomic_Load(addr, order) atomic_load_explicit(addr, order)
|
|
#define upb_Atomic_Store(addr, val, order) \
|
|
atomic_store_explicit(addr, val, order)
|
|
#define upb_Atomic_Add(addr, val, order) \
|
|
atomic_fetch_add_explicit(addr, val, order)
|
|
#define upb_Atomic_Sub(addr, val, order) \
|
|
atomic_fetch_sub_explicit(addr, val, order)
|
|
#define upb_Atomic_Exchange(addr, val, order) \
|
|
atomic_exchange_explicit(addr, val, order)
|
|
#define upb_Atomic_CompareExchangeStrong(addr, expected, desired, \
|
|
success_order, failure_order) \
|
|
atomic_compare_exchange_strong_explicit(addr, expected, desired, \
|
|
success_order, failure_order)
|
|
#define upb_Atomic_CompareExchangeWeak(addr, expected, desired, success_order, \
|
|
failure_order) \
|
|
atomic_compare_exchange_weak_explicit(addr, expected, desired, \
|
|
success_order, failure_order)
|
|
|
|
#else // !UPB_USE_C11_ATOMICS
|
|
|
|
#include <string.h>
|
|
|
|
#define upb_Atomic_Init(addr, val) (*addr = val)
|
|
#define upb_Atomic_Load(addr, order) (*addr)
|
|
#define upb_Atomic_Store(addr, val, order) (*(addr) = val)
|
|
#define upb_Atomic_Add(addr, val, order) (*(addr) += val)
|
|
#define upb_Atomic_Sub(addr, val, order) (*(addr) -= val)
|
|
|
|
UPB_INLINE void* _upb_NonAtomic_Exchange(void* addr, void* value) {
|
|
void* old;
|
|
memcpy(&old, addr, sizeof(value));
|
|
memcpy(addr, &value, sizeof(value));
|
|
return old;
|
|
}
|
|
|
|
#define upb_Atomic_Exchange(addr, val, order) _upb_NonAtomic_Exchange(addr, val)
|
|
|
|
// `addr` and `expected` are logically double pointers.
|
|
UPB_INLINE bool _upb_NonAtomic_CompareExchangeStrongP(void* addr,
|
|
void* expected,
|
|
void* desired) {
|
|
if (memcmp(addr, expected, sizeof(desired)) == 0) {
|
|
memcpy(addr, &desired, sizeof(desired));
|
|
return true;
|
|
} else {
|
|
memcpy(expected, addr, sizeof(desired));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
#define upb_Atomic_CompareExchangeStrong(addr, expected, desired, \
|
|
success_order, failure_order) \
|
|
_upb_NonAtomic_CompareExchangeStrongP((void*)addr, (void*)expected, \
|
|
(void*)desired)
|
|
#define upb_Atomic_CompareExchangeWeak(addr, expected, desired, success_order, \
|
|
failure_order) \
|
|
upb_Atomic_CompareExchangeStrong(addr, expected, desired, 0, 0)
|
|
|
|
#endif
|
|
|
|
#include "upb/port/undef.inc"
|
|
|
|
#endif // UPB_PORT_ATOMIC_H_
|