From 36af98697d92141dcd4f4cca56da38a426ccb089 Mon Sep 17 00:00:00 2001
From: Edward Tomasz Napierala <trasz@FreeBSD.org>
Date: Fri, 26 Oct 2012 16:01:08 +0000
Subject: Add CPU percentage limit enforcement to RCTL.  The resouce name is
 "pcpu". It was implemented by Rudolf Tomori during Google Summer of Code
 2012.

---
 sys/kern/subr_trap.c | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'sys/kern/subr_trap.c')

diff --git a/sys/kern/subr_trap.c b/sys/kern/subr_trap.c
index 66e32018e04f..70df8a10895c 100644
--- a/sys/kern/subr_trap.c
+++ b/sys/kern/subr_trap.c
@@ -100,6 +100,9 @@ void
 userret(struct thread *td, struct trapframe *frame)
 {
 	struct proc *p = td->td_proc;
+#ifdef	RACCT
+	int sig;
+#endif
 
 	CTR3(KTR_SYSC, "userret: thread %p (pid %d, %s)", td, p->p_pid,
             td->td_name);
@@ -163,6 +166,16 @@ userret(struct thread *td, struct trapframe *frame)
 	    __func__, td, p->p_pid, td->td_name, curvnet,
 	    (td->td_vnet_lpush != NULL) ? td->td_vnet_lpush : "N/A"));
 #endif
+#ifdef	RACCT
+	PROC_LOCK(p);
+	while (p->p_throttled == 1) {
+		sig = msleep(p->p_racct, &p->p_mtx, PCATCH | PBDRY, "racct",
+		    hz);
+		if ((sig == EINTR) || (sig == ERESTART))
+			break;
+	}
+	PROC_UNLOCK(p);
+#endif
 }
 
 /*
-- 
cgit v1.2.3