#include <mach/mach_types.h>
#include <kern/thread.h>
#include <kern/machine.h>
#include <kern/kalloc.h>
#include <sys/errno.h>
#include <kperf/filter.h>
#include <kperf/sample.h>
#include <kperf/kperfbsd.h>
#include <kperf/pet.h>
#include <kperf/action.h>
#include <kperf/kperf.h>
#include <kperf/timetrigger.h>
#include <chud/chud_xnu.h>
static struct kperf_sample *intr_samplev = NULL;
static unsigned intr_samplec = 0;
static unsigned sampling_status = KPERF_SAMPLING_OFF;
static unsigned kperf_initted = 0;
extern void (*chudxnu_thread_ast_handler)(thread_t);
struct kperf_sample*
kperf_intr_sample_buffer(void)
{
unsigned ncpu = chudxnu_cpu_number();
if( ncpu >= intr_samplec )
return NULL;
return &intr_samplev[ncpu];
}
int
kperf_init(void)
{
unsigned ncpus = 0;
if( kperf_initted )
return 0;
ncpus = machine_info.logical_cpu_max;
intr_samplev = kalloc( ncpus * sizeof(*intr_samplev));
if( intr_samplev == NULL )
return ENOMEM;
bzero( intr_samplev, ncpus * sizeof(*intr_samplev) );
chudxnu_thread_ast_handler = kperf_thread_ast_handler;
intr_samplec = ncpus;
kperf_initted = 1;
return 0;
}
kern_return_t kperf_start (kmod_info_t * ki, void * d);
kern_return_t
kperf_start (kmod_info_t * ki, void * d)
{
(void) ki;
(void) d;
printf( "aprof: kext starting\n" );
kperf_filter_init();
kperf_pet_init();
return KERN_SUCCESS;
}
uint32_t
kperf_get_thread_bits( thread_t thread )
{
return thread->t_chud;
}
void
kperf_set_thread_bits( thread_t thread, uint32_t bits )
{
thread->t_chud = bits;
}
void
kperf_set_thread_ast( thread_t thread )
{
if( thread != current_thread() )
panic( "unsafe AST set" );
act_set_kperf(thread);
}
unsigned
kperf_sampling_status(void)
{
return sampling_status;
}
int
kperf_sampling_enable(void)
{
if( sampling_status == KPERF_SAMPLING_ON )
return 0;
if ( sampling_status != KPERF_SAMPLING_OFF )
panic( "kperf: sampling wasn't off" );
if( !kperf_initted
|| (kperf_action_get_count() == 0) )
return ECANCELED;
sampling_status = KPERF_SAMPLING_ON;
kperf_timer_go();
return 0;
}
int
kperf_sampling_disable(void)
{
if( sampling_status != KPERF_SAMPLING_ON )
return 0;
sampling_status = KPERF_SAMPLING_SHUTDOWN;
kperf_timer_stop();
sampling_status = KPERF_SAMPLING_OFF;
return 0;
}