chud_osfmk_callback.c [plain text]
#include <stdint.h>
#include <mach/boolean.h>
#include <mach/mach_types.h>
#include <kern/kern_types.h>
#include <kern/processor.h>
#include <kern/timer_call.h>
#include <kern/thread_call.h>
#include <kern/kalloc.h>
#include <kern/thread.h>
#include <libkern/OSAtomic.h>
#include <machine/machine_routines.h>
#include <machine/cpu_data.h>
#include <chud/chud_xnu.h>
#include <chud/chud_xnu_private.h>
#pragma mark **** timer ****
__private_extern__ chud_timer_t
chudxnu_timer_alloc(chudxnu_timer_callback_func_t func, uint32_t param0)
{
return (chud_timer_t)thread_call_allocate((thread_call_func_t)func, (thread_call_param_t)param0);
}
__private_extern__ kern_return_t
chudxnu_timer_callback_enter(
chud_timer_t timer,
uint32_t param1,
uint32_t time,
uint32_t units)
{
uint64_t t_delay;
clock_interval_to_deadline(time, units, &t_delay);
thread_call_enter1_delayed((thread_call_t)timer, (thread_call_param_t)param1, t_delay);
return KERN_SUCCESS;
}
__private_extern__ kern_return_t
chudxnu_timer_callback_cancel(chud_timer_t timer)
{
thread_call_cancel((thread_call_t)timer);
return KERN_SUCCESS;
}
__private_extern__ kern_return_t
chudxnu_timer_free(chud_timer_t timer)
{
thread_call_cancel((thread_call_t)timer);
thread_call_free((thread_call_t)timer);
return KERN_SUCCESS;
}
static chudxnu_dtrace_callback_t
dtrace_callback = (chudxnu_dtrace_callback_t) NULL;
kern_return_t
chudxnu_dtrace_callback(uint64_t selector, uint64_t *args, uint32_t count)
{
kern_return_t ret = KERN_SUCCESS;
chudxnu_dtrace_callback_t fn = dtrace_callback;
if(fn) {
ret = fn(selector, args, count);
}
return ret;
}
__private_extern__ void
chudxnu_dtrace_callback_enter(chudxnu_dtrace_callback_t fn)
{
chudxnu_dtrace_callback_t old_fn = dtrace_callback;
while(!OSCompareAndSwap((UInt32)old_fn, (UInt32)fn,
(volatile UInt32 *) &dtrace_callback)) {
old_fn = dtrace_callback;
}
}
__private_extern__ void
chudxnu_dtrace_callback_cancel(void)
{
chudxnu_dtrace_callback_enter(NULL);
}