From 6026e1a6e4519762d57b034550cdf30087e40d68 Mon Sep 17 00:00:00 2001
From: art <art@openbsd.org>
Date: Fri, 3 Apr 2009 09:29:15 +0000
Subject: sched_peg_curproc_to_cpu() - function to force a proc to stay on a
 cpu forever.

---
 sys/kern/kern_sched.c | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

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

diff --git a/sys/kern/kern_sched.c b/sys/kern/kern_sched.c
index 1b921159985..29dd5793efe 100644
--- a/sys/kern/kern_sched.c
+++ b/sys/kern/kern_sched.c
@@ -1,4 +1,4 @@
-/*	$OpenBSD: kern_sched.c,v 1.9 2009/03/23 13:25:11 art Exp $	*/
+/*	$OpenBSD: kern_sched.c,v 1.10 2009/04/03 09:29:15 art Exp $	*/
 /*
  * Copyright (c) 2007, 2008 Artur Grabowski <art@openbsd.org>
  *
@@ -291,6 +291,12 @@ sched_choosecpu(struct proc *p)
 	struct cpu_info *ci;
 	struct cpuset set;
 
+	/*
+	 * If pegged to a cpu, don't allow it to move.
+	 */
+	if (p->p_flag & P_CPUPEG)
+		return;
+
 	sched_choose++;
 
 	/*
@@ -481,6 +487,26 @@ sched_proc_to_cpu_cost(struct cpu_info *ci, struct proc *p)
 	return (cost);
 }
 
+/*
+ * Peg a proc to a cpu.
+ */
+void
+sched_peg_curproc(struct cpu_info *ci)
+{
+	struct proc *p = curproc;
+	int s;
+
+	SCHED_LOCK(s);
+	p->p_priority = p->p_usrpri;
+	p->p_stat = SRUN;
+	p->p_cpu = ci;
+	atomic_setbits_int(&p->p_flag, P_CPUPEG);
+	setrunqueue(p);
+	p->p_stats->p_ru.ru_nvcsw++;
+	mi_switch();
+	SCHED_UNLOCK(s);
+}
+
 /*
  * Functions to manipulate cpu sets.
  */
-- 
cgit v1.2.3-59-g8ed1b