bpf: Support atomic CAS

taiki-e · Feb 23, 2023 · 6785a9b · 6785a9b
1 parent fe84be8
commit 6785a9b
Show file tree

Hide file tree

Showing 5 changed files with 868 additions and 2 deletions.
diff --git a/src/imp/core_atomic.rs b/src/imp/core_atomic.rs
@@ -5,10 +5,12 @@
 
 use core::sync::atomic::Ordering;
 
+#[cfg(not(target_arch = "bpf"))]
 #[repr(transparent)]
 pub(crate) struct AtomicBool {
     inner: core::sync::atomic::AtomicBool,
 }
+#[cfg(not(target_arch = "bpf"))]
 impl AtomicBool {
     #[inline]
     pub(crate) const fn new(v: bool) -> Self {
@@ -43,9 +45,11 @@ impl AtomicBool {
         self.inner.store(val, order);
     }
 }
+#[cfg(not(target_arch = "bpf"))]
 #[cfg_attr(portable_atomic_no_cfg_target_has_atomic, cfg(not(portable_atomic_no_atomic_cas)))]
 #[cfg_attr(not(portable_atomic_no_cfg_target_has_atomic), cfg(target_has_atomic = "ptr"))]
 no_fetch_ops_impl!(AtomicBool, bool);
+#[cfg(not(target_arch = "bpf"))]
 #[cfg_attr(portable_atomic_no_cfg_target_has_atomic, cfg(not(portable_atomic_no_atomic_cas)))]
 #[cfg_attr(not(portable_atomic_no_cfg_target_has_atomic), cfg(target_has_atomic = "ptr"))]
 impl AtomicBool {
@@ -78,6 +82,7 @@ impl AtomicBool {
         self.inner.compare_exchange_weak(current, new, success, failure)
     }
 }
+#[cfg(not(target_arch = "bpf"))]
 impl core::ops::Deref for AtomicBool {
     type Target = core::sync::atomic::AtomicBool;
     #[inline]
@@ -411,12 +416,18 @@ macro_rules! atomic_int {
 
 atomic_int!(int, AtomicIsize, isize);
 atomic_int!(uint, AtomicUsize, usize);
+#[cfg(not(target_arch = "bpf"))]
 atomic_int!(int, AtomicI8, i8);
+#[cfg(not(target_arch = "bpf"))]
 atomic_int!(uint, AtomicU8, u8);
+#[cfg(not(target_arch = "bpf"))]
 atomic_int!(int, AtomicI16, i16);
+#[cfg(not(target_arch = "bpf"))]
 atomic_int!(uint, AtomicU16, u16);
+#[cfg(not(target_arch = "bpf"))]
 #[cfg(not(target_pointer_width = "16"))] // cfg(target_has_atomic_load_store = "32")
 atomic_int!(int, AtomicI32, i32);
+#[cfg(not(target_arch = "bpf"))]
 #[cfg(not(target_pointer_width = "16"))] // cfg(target_has_atomic_load_store = "32")
 atomic_int!(uint, AtomicU32, u32);
 #[cfg_attr(portable_atomic_no_cfg_target_has_atomic, cfg(not(portable_atomic_no_atomic_64)))]