[PATCH] kern: new task notifications

[Justus Winter]

* Makefrag.am: Add task_notify.defs.
* include/mach/gnumach.defs: Add register_new_task_notification.
* include/mach/task_notify.defs: New file.
* kern/task.c (new_task_notification): New variable.
(task_create): Send new task notifications.
(register_new_task_notification): Add server function.
* kern/task_notify.cli: New file.
---
 Makefrag.am                   |    8 ++++++++
 include/mach/gnumach.defs     |    8 ++++++++
 include/mach/task_notify.defs |   19 +++++++++++++++++++
 kern/task.c                   |   26 ++++++++++++++++++++++++++
 kern/task_notify.cli          |    7 +++++++
 5 files changed, 68 insertions(+)
 create mode 100644 include/mach/task_notify.defs
 create mode 100644 kern/task_notify.cli

diff --git a/Makefrag.am b/Makefrag.am
index bb08972..746d1ae 100644
--- a/Makefrag.am
+++ b/Makefrag.am
@@ -348,6 +348,7 @@ include_mach_HEADERS = \
        include/mach/mach.defs \
        include/mach/mach4.defs \
        include/mach/gnumach.defs \
+       include/mach/task_notify.defs \
        include/mach/mach_host.defs \
        include/mach/mach_port.defs \
        include/mach/mach_types.defs \
@@ -463,6 +464,13 @@ nodist_libkernel_a_SOURCES += \
 #      device/device_reply.user.defs
 #      device/memory_object_reply.user.defs
 
+nodist_lib_dep_tr_for_defs_a_SOURCES += \
+       kern/task_notify.user.defs.c
+nodist_libkernel_a_SOURCES += \
+       kern/task_notify.user.h \
+       kern/task_notify.user.c \
+       kern/task_notify.user.msgids
+
 # Server stubs.
 nodist_lib_dep_tr_for_defs_a_SOURCES += \
        device/device.server.defs.c \
diff --git a/include/mach/gnumach.defs b/include/mach/gnumach.defs
index 7331334..848ba2a 100644
--- a/include/mach/gnumach.defs
+++ b/include/mach/gnumach.defs
@@ -37,3 +37,11 @@ type vm_cache_statistics_data_t = struct[11] of integer_t;
 routine vm_cache_statistics(
                target_task     : vm_task_t;
        out     vm_cache_stats  : vm_cache_statistics_data_t);
+
+/*
+ * Register a port to which a notification about newly created tasks
+ * are sent.
+ */
+routine register_new_task_notification(
+               host_priv       : host_priv_t;
+               notification    : mach_port_send_t);
diff --git a/include/mach/task_notify.defs b/include/mach/task_notify.defs
new file mode 100644
index 0000000..0775d00
--- /dev/null
+++ b/include/mach/task_notify.defs
@@ -0,0 +1,19 @@
+/* XXX */
+
+subsystem
+#if    KERNEL_SERVER
+         KernelServer
+#endif /* KERNEL_SERVER */
+#if    KERNEL_USER
+         KernelUser
+#endif /* KERNEL_USER */
+                      task_notify 4400;
+
+#include <mach/std_types.defs>
+#include <mach/mach_types.defs>
+
+/* XXX */
+simpleroutine mach_notify_new_task(
+               notify  : mach_port_t;
+               task    : task_t;
+               parent  : task_t);
diff --git a/kern/task.c b/kern/task.c
index 114dd31..4d56273 100644
--- a/kern/task.c
+++ b/kern/task.c
@@ -47,11 +47,13 @@
 #include <kern/processor.h>
 #include <kern/sched_prim.h>   /* for thread_wakeup */
 #include <kern/ipc_tt.h>
+#include <kern/task_notify.user.h>
 #include <vm/vm_kern.h>                /* for kernel_map, ipc_kernel_map */
 #include <machine/machspl.h>   /* for splsched */
 
 task_t kernel_task = TASK_NULL;
 struct kmem_cache task_cache;
+ipc_port_t new_task_notification = NULL;
 
 extern void eml_init(void);
 extern void eml_task_reference(task_t, task_t);
@@ -167,6 +169,14 @@ kern_return_t task_create(
        }
 #endif /* FAST_TAS */
 
+       if (new_task_notification != NULL) {
+               task_reference (new_task);
+               task_reference (parent_task);
+               mach_notify_new_task(new_task_notification,
+                                    convert_task_to_port (new_task),
+                                    convert_task_to_port (parent_task));
+       }
+
        ipc_task_enable(new_task);
 
        *child_task = new_task;
@@ -1212,3 +1222,19 @@ task_ras_control(
 #endif
     return ret;
 }
+
+/* XXX */
+kern_return_t
+register_new_task_notification(
+       host_t host,
+       ipc_port_t notification)
+{
+       if (host == HOST_NULL)
+               return KERN_INVALID_ARGUMENT;
+
+       if (new_task_notification != NULL)
+               return KERN_NO_ACCESS;
+
+       new_task_notification = notification;
+       return KERN_SUCCESS;
+}
diff --git a/kern/task_notify.cli b/kern/task_notify.cli
new file mode 100644
index 0000000..c6c85d9
--- /dev/null
+++ b/kern/task_notify.cli
@@ -0,0 +1,7 @@
+/* XXX */
+
+/* This is a client presentation file.  */
+
+#define KERNEL_USER 1
+
+#include <mach/task_notify.defs>
-- 
1.7.10.4