#include <mach/mach_types.h>
#include <mach/mach_traps.h>
#include <mach/mach_port_server.h>
#include <mach/mk_timer.h>
#include <ipc/ipc_space.h>
#include <kern/lock_group.h>
#include <kern/mk_timer.h>
#include <kern/thread_call.h>
#include <ipc/ipc_kmsg.h>
static zone_t mk_timer_zone;
static mach_port_qos_t mk_timer_qos = {
.name = FALSE,
.prealloc = TRUE,
.len = sizeof(mk_timer_expire_msg_t),
};
static void mk_timer_expire(
void *p0,
void *p1);
mach_port_name_t
mk_timer_create_trap(
__unused struct mk_timer_create_trap_args *args)
{
mk_timer_t timer;
ipc_space_t myspace = current_space();
mach_port_name_t name = MACH_PORT_NULL;
ipc_port_init_flags_t init_flags;
ipc_port_t port;
kern_return_t result;
timer = (mk_timer_t)zalloc(mk_timer_zone);
if (timer == NULL) {
return MACH_PORT_NULL;
}
simple_lock_init(&timer->lock, 0);
thread_call_setup(&timer->call_entry, mk_timer_expire, timer);
timer->is_armed = timer->is_dead = FALSE;
timer->active = 0;
ipc_kmsg_t kmsg;
kmsg = ipc_kmsg_prealloc(mk_timer_qos.len + MAX_TRAILER_SIZE);
if (kmsg == IKM_NULL) {
zfree(mk_timer_zone, timer);
return MACH_PORT_NULL;
}
init_flags = IPC_PORT_INIT_MESSAGE_QUEUE;
result = ipc_port_alloc(myspace, init_flags, &name, &port);
if (result != KERN_SUCCESS) {
zfree(mk_timer_zone, timer);
ipc_kmsg_free(kmsg);
return MACH_PORT_NULL;
}
ipc_kmsg_set_prealloc(kmsg, port);
ipc_kobject_set_atomically(port, (ipc_kobject_t)timer, IKOT_TIMER);
timer->port = ipc_port_make_send_locked(port);
ip_unlock(port);
return name;
}
void
mk_timer_port_destroy(
ipc_port_t port)
{
mk_timer_t timer = NULL;
ip_lock(port);
if (ip_kotype(port) == IKOT_TIMER) {
timer = (mk_timer_t) ip_get_kobject(port);
assert(timer != NULL);
ipc_kobject_set_atomically(port, IKO_NULL, IKOT_NONE);
simple_lock(&timer->lock, LCK_GRP_NULL);
assert(timer->port == port);
}
ip_unlock(port);
if (timer != NULL) {
if (thread_call_cancel(&timer->call_entry)) {
timer->active--;
}
timer->is_armed = FALSE;
timer->is_dead = TRUE;
if (timer->active == 0) {
simple_unlock(&timer->lock);
zfree(mk_timer_zone, timer);
ipc_port_release_send(port);
return;
}
simple_unlock(&timer->lock);
}
}
void
mk_timer_init(void)
{
int s = sizeof(mk_timer_data_t);
assert(!(mk_timer_zone != NULL));
mk_timer_zone = zinit(s, (4096 * s), (16 * s), "mk_timer");
zone_change(mk_timer_zone, Z_NOENCRYPT, TRUE);
}
static void
mk_timer_expire(
void *p0,
__unused void *p1)
{
mk_timer_t timer = p0;
ipc_port_t port;
simple_lock(&timer->lock, LCK_GRP_NULL);
if (timer->active > 1) {
timer->active--;
simple_unlock(&timer->lock);
return;
}
port = timer->port;
assert(port != IP_NULL);
assert(timer->active == 1);
while (timer->is_armed && timer->active == 1) {
mk_timer_expire_msg_t msg;
timer->is_armed = FALSE;
simple_unlock(&timer->lock);
msg.header.msgh_bits =
MACH_MSGH_BITS_SET(MACH_MSG_TYPE_COPY_SEND, 0, 0, 0);
msg.header.msgh_remote_port = port;
msg.header.msgh_local_port = MACH_PORT_NULL;
msg.header.msgh_voucher_port = MACH_PORT_NULL;
msg.header.msgh_id = 0;
msg.unused[0] = msg.unused[1] = msg.unused[2] = 0;
(void) mach_msg_send_from_kernel_proper(&msg.header, sizeof(msg));
simple_lock(&timer->lock, LCK_GRP_NULL);
}
if (--timer->active == 0 && timer->is_dead) {
simple_unlock(&timer->lock);
zfree(mk_timer_zone, timer);
ipc_port_release_send(port);
return;
}
simple_unlock(&timer->lock);
}
kern_return_t
mk_timer_destroy_trap(
struct mk_timer_destroy_trap_args *args)
{
mach_port_name_t name = args->name;
ipc_space_t myspace = current_space();
ipc_port_t port;
kern_return_t result;
result = ipc_port_translate_receive(myspace, name, &port);
if (result != KERN_SUCCESS) {
return result;
}
if (ip_kotype(port) == IKOT_TIMER) {
ip_unlock(port);
result = mach_port_destroy(myspace, name);
} else {
ip_unlock(port);
result = KERN_INVALID_ARGUMENT;
}
return result;
}
static kern_return_t
mk_timer_arm_trap_internal(mach_port_name_t name, uint64_t expire_time, uint64_t mk_leeway, uint64_t mk_timer_flags)
{
mk_timer_t timer;
ipc_space_t myspace = current_space();
ipc_port_t port;
kern_return_t result;
result = ipc_port_translate_receive(myspace, name, &port);
if (result != KERN_SUCCESS) {
return result;
}
if (ip_kotype(port) == IKOT_TIMER) {
timer = (mk_timer_t) ip_get_kobject(port);
assert(timer != NULL);
simple_lock(&timer->lock, LCK_GRP_NULL);
assert(timer->port == port);
ip_unlock(port);
if (!timer->is_dead) {
timer->is_armed = TRUE;
if (expire_time > mach_absolute_time()) {
uint32_t tcflags = THREAD_CALL_DELAY_USER_NORMAL;
if (mk_timer_flags & MK_TIMER_CRITICAL) {
tcflags = THREAD_CALL_DELAY_USER_CRITICAL;
}
if (mk_leeway != 0) {
tcflags |= THREAD_CALL_DELAY_LEEWAY;
}
if (!thread_call_enter_delayed_with_leeway(
&timer->call_entry, NULL,
expire_time, mk_leeway, tcflags)) {
timer->active++;
}
} else {
if (!thread_call_enter1(&timer->call_entry, NULL)) {
timer->active++;
}
}
}
simple_unlock(&timer->lock);
} else {
ip_unlock(port);
result = KERN_INVALID_ARGUMENT;
}
return result;
}
kern_return_t
mk_timer_arm_trap(struct mk_timer_arm_trap_args *args)
{
return mk_timer_arm_trap_internal(args->name, args->expire_time, 0, MK_TIMER_NORMAL);
}
kern_return_t
mk_timer_arm_leeway_trap(struct mk_timer_arm_leeway_trap_args *args)
{
return mk_timer_arm_trap_internal(args->name, args->expire_time, args->mk_leeway, args->mk_timer_flags);
}
kern_return_t
mk_timer_cancel_trap(
struct mk_timer_cancel_trap_args *args)
{
mach_port_name_t name = args->name;
mach_vm_address_t result_time_addr = args->result_time;
uint64_t armed_time = 0;
mk_timer_t timer;
ipc_space_t myspace = current_space();
ipc_port_t port;
kern_return_t result;
result = ipc_port_translate_receive(myspace, name, &port);
if (result != KERN_SUCCESS) {
return result;
}
if (ip_kotype(port) == IKOT_TIMER) {
timer = (mk_timer_t) ip_get_kobject(port);
assert(timer != NULL);
simple_lock(&timer->lock, LCK_GRP_NULL);
assert(timer->port == port);
ip_unlock(port);
if (timer->is_armed) {
armed_time = timer->call_entry.tc_call.deadline;
if (thread_call_cancel(&timer->call_entry)) {
timer->active--;
}
timer->is_armed = FALSE;
}
simple_unlock(&timer->lock);
} else {
ip_unlock(port);
result = KERN_INVALID_ARGUMENT;
}
if (result == KERN_SUCCESS) {
if (result_time_addr != 0 &&
copyout((void *)&armed_time, result_time_addr,
sizeof(armed_time)) != 0) {
result = KERN_FAILURE;
}
}
return result;
}