xnu-7195.81.3.tar.gz

[apple/xnu.git] / libkern / firehose / chunk_private.h

diff --git a/libkern/firehose/chunk_private.h b/libkern/firehose/chunk_private.h
index dece91a3786bd5a21f2542501dd744cf3dfa3cd2..ac3fbe92efdf81560c39bbb7b8bcdabe39835852 100644 (file)
--- a/libkern/firehose/chunk_private.h
+++ b/libkern/firehose/chunk_private.h
@@ -21,12 +21,7 @@
 #ifndef __FIREHOSE_CHUNK_PRIVATE__
 #define __FIREHOSE_CHUNK_PRIVATE__
 
 #ifndef __FIREHOSE_CHUNK_PRIVATE__
 #define __FIREHOSE_CHUNK_PRIVATE__
 

-#if KERNEL
-#include <machine/atomic.h>
-#endif
-#include <stdatomic.h>

 #include <sys/param.h>
 #include <sys/param.h>

-#include <os/base.h>

 #include "firehose_types_private.h"
 #include "tracepoint_private.h"
 
 #include "firehose_types_private.h"
 #include "tracepoint_private.h"
 


@@ -42,7 +37,7 @@ __BEGIN_DECLS
                ((((pos).fcp_pos >> 48) & 0x1ff) == (uint16_t)stream)
 
 typedef union {
                ((((pos).fcp_pos >> 48) & 0x1ff) == (uint16_t)stream)
 
 typedef union {

-       _Atomic(uint64_t) fcp_atomic_pos;
+       os_atomic(uint64_t) fcp_atomic_pos;

        uint64_t fcp_pos;
        struct {
                uint16_t fcp_next_entry_offs;
        uint64_t fcp_pos;
        struct {
                uint16_t fcp_next_entry_offs;


@@ -73,14 +68,15 @@ typedef struct firehose_chunk_range_s {
 
 OS_ALWAYS_INLINE
 static inline bool
 
 OS_ALWAYS_INLINE
 static inline bool

-firehose_chunk_pos_fits(firehose_chunk_pos_u pos, uint16_t size)
+firehose_chunk_pos_fits(firehose_chunk_pos_u *pos, uint16_t size)

 {
 {

-       return pos.fcp_next_entry_offs + size <= pos.fcp_private_offs;
+       return pos->fcp_next_entry_offs + size <= pos->fcp_private_offs;

 }
 
 #define FIREHOSE_CHUNK_TRY_RESERVE_FAIL_ENQUEUE  (-1)
 #define FIREHOSE_CHUNK_TRY_RESERVE_FAIL          ( 0)
 
 }
 
 #define FIREHOSE_CHUNK_TRY_RESERVE_FAIL_ENQUEUE  (-1)
 #define FIREHOSE_CHUNK_TRY_RESERVE_FAIL          ( 0)
 

+#if OS_ATOMIC_HAS_STARVATION_FREE_RMW || !OS_ATOMIC_CONFIG_STARVATION_FREE_ONLY

 OS_ALWAYS_INLINE
 static inline long
 firehose_chunk_tracepoint_try_reserve(firehose_chunk_t fc, uint64_t stamp,
 OS_ALWAYS_INLINE
 static inline long
 firehose_chunk_tracepoint_try_reserve(firehose_chunk_t fc, uint64_t stamp,


@@ -109,8 +105,8 @@ firehose_chunk_tracepoint_try_reserve(firehose_chunk_t fc, uint64_t stamp,
                        // - read the chunk to find a very old thing
                        os_atomic_rmw_loop_give_up(return FIREHOSE_CHUNK_TRY_RESERVE_FAIL);
                }
                        // - read the chunk to find a very old thing
                        os_atomic_rmw_loop_give_up(return FIREHOSE_CHUNK_TRY_RESERVE_FAIL);
                }

-               pos = orig;
-               if (!firehose_chunk_pos_fits(orig,
+               pos.fcp_pos = orig.fcp_pos;
+               if (!firehose_chunk_pos_fits(&orig,

                ft_size + pubsize + privsize) || !stamp_delta_fits) {
                        pos.fcp_flag_full = true;
                        reservation_failed = true;
                ft_size + pubsize + privsize) || !stamp_delta_fits) {
                        pos.fcp_flag_full = true;
                        reservation_failed = true;


@@ -126,7 +122,7 @@ firehose_chunk_tracepoint_try_reserve(firehose_chunk_t fc, uint64_t stamp,
                        pos.fcp_pos -= privsize * FIREHOSE_CHUNK_POS_PRIVATE_OFFS_INC;
                        pos.fcp_pos += FIREHOSE_CHUNK_POS_REFCNT_INC;
                        const uint16_t minimum_payload_size = 16;
                        pos.fcp_pos -= privsize * FIREHOSE_CHUNK_POS_PRIVATE_OFFS_INC;
                        pos.fcp_pos += FIREHOSE_CHUNK_POS_REFCNT_INC;
                        const uint16_t minimum_payload_size = 16;

-                       if (!firehose_chunk_pos_fits(pos,
+                       if (!firehose_chunk_pos_fits(&pos,

                        roundup(ft_size + minimum_payload_size, 8))) {
                                // if we can't even have minimum_payload_size bytes of payload
                                // for the next tracepoint, just flush right away
                        roundup(ft_size + minimum_payload_size, 8))) {
                                // if we can't even have minimum_payload_size bytes of payload
                                // for the next tracepoint, just flush right away


@@ -162,8 +158,8 @@ firehose_chunk_tracepoint_begin(firehose_chunk_t fc, uint64_t stamp,
        stamp |= (uint64_t)pubsize << 48;
        // The compiler barrier is needed for userland process death handling, see
        // (tracepoint-begin) in libdispatch's firehose_buffer_stream_chunk_install.
        stamp |= (uint64_t)pubsize << 48;
        // The compiler barrier is needed for userland process death handling, see
        // (tracepoint-begin) in libdispatch's firehose_buffer_stream_chunk_install.

-       atomic_store_explicit(&ft->ft_atomic_stamp_and_length, stamp,
-           memory_order_relaxed);
+       os_atomic_std(atomic_store_explicit)(&ft->ft_atomic_stamp_and_length, stamp,
+           os_atomic_std(memory_order_relaxed));

        __asm__ __volatile__ ("" ::: "memory");
        ft->ft_thread = thread_id;
        return ft;
        __asm__ __volatile__ ("" ::: "memory");
        ft->ft_thread = thread_id;
        return ft;


@@ -176,12 +172,13 @@ firehose_chunk_tracepoint_end(firehose_chunk_t fc,
 {
        firehose_chunk_pos_u pos;
 
 {
        firehose_chunk_pos_u pos;
 

-       atomic_store_explicit(&ft->ft_id.ftid_atomic_value,
-           ftid.ftid_value, memory_order_release);
-       pos.fcp_pos = atomic_fetch_sub_explicit(&fc->fc_pos.fcp_atomic_pos,
-           FIREHOSE_CHUNK_POS_REFCNT_INC, memory_order_relaxed);
+       os_atomic_std(atomic_store_explicit)(&ft->ft_id.ftid_atomic_value,
+           ftid.ftid_value, os_atomic_std(memory_order_release));
+       pos.fcp_pos = os_atomic_std(atomic_fetch_sub_explicit)(&fc->fc_pos.fcp_atomic_pos,
+           FIREHOSE_CHUNK_POS_REFCNT_INC, os_atomic_std(memory_order_relaxed));

        return pos.fcp_refcnt == 1 && pos.fcp_flag_full;
 }
        return pos.fcp_refcnt == 1 && pos.fcp_flag_full;
 }

+#endif // OS_ATOMIC_HAS_STARVATION_FREE_RMW || !OS_ATOMIC_CONFIG_STARVATION_FREE_ONLY

 
 #endif // defined(KERNEL) || defined(OS_FIREHOSE_SPI)
 
 
 #endif // defined(KERNEL) || defined(OS_FIREHOSE_SPI)