#include <kern/task.h>
#include <kern/thread.h>
#include <arm/misc_protos.h>
extern zone_t ads_zone;
kern_return_t
machine_task_set_state(
task_t task,
int flavor,
thread_state_t state,
mach_msg_type_number_t state_count)
{
switch (flavor) {
case ARM_DEBUG_STATE:
{
arm_legacy_debug_state_t *tstate = (arm_legacy_debug_state_t *) state;
if (task_has_64Bit_data(task) ||
(state_count != ARM_LEGACY_DEBUG_STATE_COUNT) ||
(!debug_legacy_state_is_valid(tstate))) {
return KERN_INVALID_ARGUMENT;
}
if (task->task_debug == NULL) {
task->task_debug = zalloc(ads_zone);
if (task->task_debug == NULL) {
return KERN_FAILURE;
}
}
copy_legacy_debug_state(tstate, (arm_legacy_debug_state_t *) task->task_debug, FALSE);
return KERN_SUCCESS;
}
case ARM_DEBUG_STATE32:
{
arm_debug_state32_t *tstate = (arm_debug_state32_t *) state;
if (task_has_64Bit_data(task) ||
(state_count != ARM_DEBUG_STATE32_COUNT) ||
(!debug_state_is_valid32(tstate))) {
return KERN_INVALID_ARGUMENT;
}
if (task->task_debug == NULL) {
task->task_debug = zalloc(ads_zone);
if (task->task_debug == NULL) {
return KERN_FAILURE;
}
}
copy_debug_state32(tstate, (arm_debug_state32_t *) task->task_debug, FALSE);
return KERN_SUCCESS;
}
case ARM_DEBUG_STATE64:
{
arm_debug_state64_t *tstate = (arm_debug_state64_t *) state;
if ((!task_has_64Bit_data(task)) ||
(state_count != ARM_DEBUG_STATE64_COUNT) ||
(!debug_state_is_valid64(tstate))) {
return KERN_INVALID_ARGUMENT;
}
if (task->task_debug == NULL) {
task->task_debug = zalloc(ads_zone);
if (task->task_debug == NULL) {
return KERN_FAILURE;
}
}
copy_debug_state64(tstate, (arm_debug_state64_t *) task->task_debug, FALSE);
return KERN_SUCCESS;
}
case THREAD_STATE_NONE:
{
if (task->task_debug != NULL) {
zfree(ads_zone, task->task_debug);
task->task_debug = NULL;
return KERN_SUCCESS;
}
return KERN_FAILURE;
}
default:
{
return KERN_INVALID_ARGUMENT;
}
}
return KERN_FAILURE;
}
kern_return_t
machine_task_get_state(task_t task,
int flavor,
thread_state_t state,
mach_msg_type_number_t *state_count)
{
switch (flavor) {
case ARM_DEBUG_STATE:
{
arm_legacy_debug_state_t *tstate = (arm_legacy_debug_state_t *) state;
if (task_has_64Bit_data(task) || (*state_count != ARM_LEGACY_DEBUG_STATE_COUNT)) {
return KERN_INVALID_ARGUMENT;
}
if (task->task_debug == NULL) {
bzero(state, sizeof(*tstate));
} else {
copy_legacy_debug_state((arm_legacy_debug_state_t*) task->task_debug, tstate, FALSE);
}
return KERN_SUCCESS;
}
case ARM_DEBUG_STATE32:
{
arm_debug_state32_t *tstate = (arm_debug_state32_t *) state;
if (task_has_64Bit_data(task) || (*state_count != ARM_DEBUG_STATE32_COUNT)) {
return KERN_INVALID_ARGUMENT;
}
if (task->task_debug == NULL) {
bzero(state, sizeof(*tstate));
} else {
copy_debug_state32((arm_debug_state32_t*) task->task_debug, tstate, FALSE);
}
return KERN_SUCCESS;
}
case ARM_DEBUG_STATE64:
{
arm_debug_state64_t *tstate = (arm_debug_state64_t *) state;
if ((!task_has_64Bit_data(task)) || (*state_count != ARM_DEBUG_STATE64_COUNT)) {
return KERN_INVALID_ARGUMENT;
}
if (task->task_debug == NULL) {
bzero(state, sizeof(*tstate));
} else {
copy_debug_state64((arm_debug_state64_t*) task->task_debug, tstate, FALSE);
}
return KERN_SUCCESS;
}
default:
{
return KERN_INVALID_ARGUMENT;
}
}
return KERN_FAILURE;
}
void
machine_task_terminate(task_t task)
{
if (task) {
void *task_debug;
task_debug = task->task_debug;
if (task_debug != NULL) {
task->task_debug = NULL;
zfree(ads_zone, task_debug);
}
}
}
kern_return_t
machine_thread_inherit_taskwide(
thread_t thread,
task_t parent_task)
{
if (parent_task->task_debug) {
int flavor;
mach_msg_type_number_t count;
flavor = task_has_64Bit_data(parent_task) ? ARM_DEBUG_STATE64 : ARM_DEBUG_STATE32;
count = task_has_64Bit_data(parent_task) ? ARM_DEBUG_STATE64_COUNT : ARM_DEBUG_STATE32_COUNT;
return machine_thread_set_state(thread, flavor, parent_task->task_debug, count);
}
return KERN_SUCCESS;
}
void
machine_task_init(__unused task_t new_task,
__unused task_t parent_task,
__unused boolean_t memory_inherit)
{
}