/* * Set initial default state on a thread as stored in the MACHINE_TASK data. * Note: currently only debug state is supported. */ 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; if (task_has_64BitAddr(parent_task)) { flavor = x86_DEBUG_STATE64; count = x86_DEBUG_STATE64_COUNT; } else { flavor = x86_DEBUG_STATE32; count = x86_DEBUG_STATE32_COUNT; } return machine_thread_set_state(thread, flavor, parent_task->task_debug, count); } return KERN_SUCCESS; }
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 x86_DEBUG_STATE32: { x86_debug_state32_t *tstate = (x86_debug_state32_t*) state; if ((task_has_64BitAddr(task)) || (state_count != x86_DEBUG_STATE32_COUNT) || (!debug_state_is_valid32(tstate))) { return KERN_INVALID_ARGUMENT; } if (task->task_debug == NULL) { task->task_debug = zalloc(ids_zone); } copy_debug_state32(tstate, (x86_debug_state32_t*) task->task_debug, FALSE); return KERN_SUCCESS; break; } case x86_DEBUG_STATE64: { x86_debug_state64_t *tstate = (x86_debug_state64_t*) state; if ((!task_has_64BitAddr(task)) || (state_count != x86_DEBUG_STATE64_COUNT) || (!debug_state_is_valid64(tstate))) { return KERN_INVALID_ARGUMENT; } if (task->task_debug == NULL) { task->task_debug = zalloc(ids_zone); } copy_debug_state64(tstate, (x86_debug_state64_t*) task->task_debug, FALSE); return KERN_SUCCESS; break; } case x86_DEBUG_STATE: { x86_debug_state_t *tstate = (x86_debug_state_t*) state; if (state_count != x86_DEBUG_STATE_COUNT) { return KERN_INVALID_ARGUMENT; } if ((tstate->dsh.flavor == x86_DEBUG_STATE32) && (tstate->dsh.count == x86_DEBUG_STATE32_COUNT) && (!task_has_64BitAddr(task)) && debug_state_is_valid32(&tstate->uds.ds32)) { if (task->task_debug == NULL) { task->task_debug = zalloc(ids_zone); } copy_debug_state32(&tstate->uds.ds32, (x86_debug_state32_t*) task->task_debug, FALSE); return KERN_SUCCESS; } else if ((tstate->dsh.flavor == x86_DEBUG_STATE64) && (tstate->dsh.count == x86_DEBUG_STATE64_COUNT) && task_has_64BitAddr(task) && debug_state_is_valid64(&tstate->uds.ds64)) { if (task->task_debug == NULL) { task->task_debug = zalloc(ids_zone); } copy_debug_state64(&tstate->uds.ds64, (x86_debug_state64_t*) task->task_debug, FALSE); return KERN_SUCCESS; } else { return KERN_INVALID_ARGUMENT; } break; } default: { return KERN_INVALID_ARGUMENT; break; } } }
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 x86_DEBUG_STATE32: { x86_debug_state32_t *tstate = (x86_debug_state32_t*) state; if ((task_has_64BitAddr(task)) || (*state_count != x86_DEBUG_STATE32_COUNT)) { return KERN_INVALID_ARGUMENT; } if (task->task_debug == NULL) { bzero(state, sizeof(*tstate)); } else { copy_debug_state32((x86_debug_state32_t*) task->task_debug, tstate, TRUE); } return KERN_SUCCESS; break; } case x86_DEBUG_STATE64: { x86_debug_state64_t *tstate = (x86_debug_state64_t*) state; if ((!task_has_64BitAddr(task)) || (*state_count != x86_DEBUG_STATE64_COUNT)) { return KERN_INVALID_ARGUMENT; } if (task->task_debug == NULL) { bzero(state, sizeof(*tstate)); } else { copy_debug_state64((x86_debug_state64_t*) task->task_debug, tstate, TRUE); } return KERN_SUCCESS; break; } case x86_DEBUG_STATE: { x86_debug_state_t *tstate = (x86_debug_state_t*)state; if (*state_count != x86_DEBUG_STATE_COUNT) return(KERN_INVALID_ARGUMENT); if (task_has_64BitAddr(task)) { tstate->dsh.flavor = x86_DEBUG_STATE64; tstate->dsh.count = x86_DEBUG_STATE64_COUNT; if (task->task_debug == NULL) { bzero(&tstate->uds.ds64, sizeof(tstate->uds.ds64)); } else { copy_debug_state64((x86_debug_state64_t*)task->task_debug, &tstate->uds.ds64, TRUE); } } else { tstate->dsh.flavor = x86_DEBUG_STATE32; tstate->dsh.count = x86_DEBUG_STATE32_COUNT; if (task->task_debug == NULL) { bzero(&tstate->uds.ds32, sizeof(tstate->uds.ds32)); } else { copy_debug_state32((x86_debug_state32_t*)task->task_debug, &tstate->uds.ds32, TRUE); } } return KERN_SUCCESS; break; } default: { return KERN_INVALID_ARGUMENT; break; } } }
__private_extern__ boolean_t chudxnu_is_64bit_task(task_t task) { return (task_has_64BitAddr(task)); }