X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/5ba3f43ea354af8ad55bea84372a2bc834d8757c..a991bd8d3e7fe02dbca0644054bab73c5b75324a:/libsyscall/wrappers/work_interval.c?ds=sidebyside

diff --git a/libsyscall/wrappers/work_interval.c b/libsyscall/wrappers/work_interval.c
index 96ca9ef83..89d2d0cce 100644
--- a/libsyscall/wrappers/work_interval.c
+++ b/libsyscall/wrappers/work_interval.c
@@ -30,6 +30,7 @@
 
 #include <sys/errno.h>
 #include <stdlib.h>
+#include <strings.h>
 
 struct work_interval {
 	uint64_t thread_id;
@@ -57,7 +58,7 @@ work_interval_create(work_interval_t *interval_handle, uint32_t create_flags)
 	};
 
 	ret = __work_interval_ctl(WORK_INTERVAL_OPERATION_CREATE2, 0,
-	                          &create_params, sizeof(create_params));
+	    &create_params, sizeof(create_params));
 	if (ret == -1) {
 		return ret;
 	}
@@ -77,10 +78,30 @@ work_interval_create(work_interval_t *interval_handle, uint32_t create_flags)
 	return 0;
 }
 
+int
+work_interval_get_flags_from_port(mach_port_t port, uint32_t *flags)
+{
+	if (!MACH_PORT_VALID(port) || flags == NULL) {
+		errno = EINVAL;
+		return -1;
+	}
+
+	struct work_interval_create_params create_params = { 0 };
+
+	int ret = __work_interval_ctl(WORK_INTERVAL_OPERATION_GET_FLAGS, port,
+	    &create_params, sizeof(create_params));
+	if (ret == -1) {
+		return ret;
+	}
+
+	*flags = create_params.wicp_create_flags;
+	return 0;
+}
+
 int
 work_interval_notify(work_interval_t interval_handle, uint64_t start,
-                     uint64_t finish, uint64_t deadline, uint64_t next_start,
-                     uint32_t notify_flags)
+    uint64_t finish, uint64_t deadline, uint64_t next_start,
+    uint32_t notify_flags)
 {
 	int ret;
 	uint64_t work_interval_id;
@@ -97,22 +118,27 @@ work_interval_notify(work_interval_t interval_handle, uint64_t start,
 		return -1;
 	}
 
+	if (interval_handle->create_flags & WORK_INTERVAL_FLAG_IGNORED) {
+		return 0;
+	}
+
 	notification.create_flags = interval_handle->create_flags;
 	work_interval_id = interval_handle->work_interval_id;
 
 	ret = __work_interval_ctl(WORK_INTERVAL_OPERATION_NOTIFY, work_interval_id,
-	                          &notification, sizeof(notification));
+	    &notification, sizeof(notification));
 	return ret;
 }
 
 int
 work_interval_notify_simple(work_interval_t interval_handle, uint64_t start,
-                            uint64_t deadline, uint64_t next_start)
+    uint64_t deadline, uint64_t next_start)
 {
 	return work_interval_notify(interval_handle, start, mach_absolute_time(),
-	                            deadline, next_start, 0);
+	           deadline, next_start, 0);
 }
 
+
 int
 work_interval_destroy(work_interval_t interval_handle)
 {
@@ -158,7 +184,7 @@ work_interval_destroy(work_interval_t interval_handle)
 		uint64_t work_interval_id = interval_handle->work_interval_id;
 
 		int ret = __work_interval_ctl(WORK_INTERVAL_OPERATION_DESTROY,
-		                              work_interval_id, NULL, 0);
+		    work_interval_id, NULL, 0);
 
 		interval_handle->work_interval_id = 0;
 
@@ -201,14 +227,14 @@ work_interval_join_port(mach_port_t port)
 	}
 
 	return __work_interval_ctl(WORK_INTERVAL_OPERATION_JOIN,
-	                           (uint64_t)port, NULL, 0);
+	           (uint64_t)port, NULL, 0);
 }
 
 int
 work_interval_leave(void)
 {
 	return __work_interval_ctl(WORK_INTERVAL_OPERATION_JOIN,
-	                           (uint64_t)MACH_PORT_NULL, NULL, 0);
+	           (uint64_t)MACH_PORT_NULL, NULL, 0);
 }
 
 int
@@ -234,7 +260,7 @@ work_interval_copy_port(work_interval_t interval_handle, mach_port_t *port)
 	mach_port_t wi_port = interval_handle->wi_port;
 
 	kern_return_t kr = mach_port_mod_refs(mach_task_self(), wi_port,
-	                                      MACH_PORT_RIGHT_SEND, 1);
+	    MACH_PORT_RIGHT_SEND, 1);
 
 	if (kr != KERN_SUCCESS) {
 		*port = MACH_PORT_NULL;
@@ -246,7 +272,3 @@ work_interval_copy_port(work_interval_t interval_handle, mach_port_t *port)
 
 	return 0;
 }
-
-
-
-