Fix another bug introduced in r225641, which caused rctl to access certain

fields in 'struct proc' before they got initialized in do_fork().

MFC after:	3 days

1 files changed, 23 insertions, 0 deletions

diff --git a/sys/kern/kern_racct.c b/sys/kern/kern_racct.c
index 525866c..8414cda 100644
--- a/sys/kern/kern_racct.c
+++ b/sys/kern/kern_racct.c
@@ -585,6 +585,24 @@ out:
 	return (error);
 }
 
+/*
+ * Called at the end of fork1(), to handle rules that require the process
+ * to be fully initialized.
+ */
+void
+racct_proc_fork_done(struct proc *child)
+{
+
+#ifdef RCTL
+	PROC_LOCK(child);
+	mtx_lock(&racct_lock);
+	rctl_enforce(child, RACCT_NPROC, 0);
+	rctl_enforce(child, RACCT_NTHR, 0);
+	mtx_unlock(&racct_lock);
+	PROC_UNLOCK(child);
+#endif
+}
+
 void
 racct_proc_exit(struct proc *p)
 {
@@ -810,6 +828,11 @@ racct_proc_fork(struct proc *parent, struct proc *child)
 }
 
 void
+racct_proc_fork_done(struct proc *child)
+{
+}
+
+void
 racct_proc_exit(struct proc *p)
 {
 }