This is xnu-12377.1.9. See this file in: xnu-12377.1.9 (Darwin 25) xnu-8792.81.2 (MacOS 18, Darwin 22) xnu-8019 (MacOS 17.0, Darwin 21) xnu-6153.11.26 (MacOS 15.0, Darwin 19) xnu-4903.221.2 (MacOS 14.0, Darwin 18) xnu-4570.41.2 (MacOS 13.3, Darwin 17) xnu-3789.70.16 (MacOS 12.6, Darwin 16) xnu-3248.20.55 (MacOS 11.2, Darwin 15) xnu-2782.40.9 (MacOS 10.10.5, Darwin 14) xnu-2422.115.4 (MacOS 10.9.5, Darwin 13) xnu-2050.24.15 (MacOS 10.8.4, Darwin 12) xnu-1699.32.7 (MacOS 10.7.5, Darwin 11) xnu-1504.15.3 (MacOS 10.6.8, Darwin 10) xnu-1228.15.4 (MacOS 10.5.8, Darwin 9)

// Copyright (c) 2024 Apple Inc.  All rights reserved.

#pragma once

#include <stdint.h>
#include <stdbool.h>

#include "sched_runqueue_harness.h"
#include "sched_migration_harness.h"

extern void                  impl_init_runqueue(void);
extern struct thread_group  *impl_create_tg(int interactivity_score);
extern test_thread_t         impl_create_thread(int th_sched_bucket, struct thread_group *tg, int pri);
extern void                  impl_set_thread_sched_mode(test_thread_t thread, int mode);
extern bool                  impl_get_thread_is_realtime(test_thread_t thread);
extern void                  impl_set_thread_processor_bound(test_thread_t thread, int cpu_id);
extern void                  impl_cpu_set_thread_current(int cpu_id, test_thread_t thread);
extern test_thread_t         impl_cpu_clear_thread_current(int cpu_id);
extern void                  impl_cpu_enqueue_thread(int cpu_id, test_thread_t thread);
extern test_thread_t         impl_cpu_dequeue_thread(int cpu_id);
extern test_thread_t         impl_cpu_dequeue_thread_compare_current(int cpu_id);
extern bool                  impl_processor_csw_check(int cpu_id);
extern void                  impl_pop_tracepoint(uint64_t *trace_code, uint64_t *arg1, uint64_t *arg2, uint64_t *arg3, uint64_t *arg4);
extern bool                  impl_thread_should_yield(int cpu_id);
extern void                  impl_pop_ipi(int *cpu_id, test_ipi_type_t *ipi_type);
extern void                  impl_send_ipi(int cpu_id, test_thread_t thread, test_ipi_event_t event);

/* Migration-specific functions */
extern void                  impl_init_migration_harness(test_hw_topology_t hw_topology);
extern void                  impl_set_tg_sched_bucket_preferred_pset(struct thread_group *tg, int sched_bucket, int cluster_id);
extern void                  impl_set_thread_cluster_bound(test_thread_t thread, int cluster_id);
extern int                   impl_choose_pset_for_thread(test_thread_t thread);
extern bool                  impl_thread_avoid_processor(test_thread_t thread, int cpu_id, bool quantum_expiry);
extern void                  impl_cpu_expire_quantum(int cpu_id);
extern test_thread_t         impl_steal_thread(int cpu_id);
extern bool                  impl_processor_balance(int cpu_id);
extern void                  impl_set_current_processor(int cpu_id);
extern void                  impl_set_pset_load_avg(int cluster_id, int QoS, uint64_t load_avg);
extern void                  impl_set_pset_derecommended(int cluster_id);
extern int                   impl_iterate_pset_search_order_rt(int src_pset_id, int offset);
extern void                  impl_set_pset_recommended(int cluster_id);
extern uint32_t              impl_qos_max_parallelism(int qos, uint64_t options);
extern int                  *impl_iterate_pset_search_order(int src_pset_id, uint64_t candidate_map, int sched_bucket);

/* Realtime */
extern void                  impl_set_thread_realtime(test_thread_t thread, uint32_t period, uint32_t computation, uint32_t constraint, bool preemptible, uint8_t priority_offset, uint64_t deadline);
extern test_thread_t         impl_rt_choose_thread(int cpu_id);