This optional patch adds a fork/exit rate measurement facility using task
watchers. It is intended to be a tool for measuring the impact of task watchers
on fork and exit-heavy workloads.

Signed-off-by: Matt Helsley <matthltc@us.ibm.com>
---
 kernel/Makefile   |    1 
 kernel/twbench.c  |  103 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 lib/Kconfig.debug |   12 ++++++
 3 files changed, 116 insertions(+)

Index: linux-2.6.18-mm1/kernel/twbench.c
===================================================================
--- /dev/null
+++ linux-2.6.18-mm1/kernel/twbench.c
@@ -0,0 +1,103 @@
+#include <linux/kernel.h>
+#include <linux/module.h>
+#include <linux/task_watchers.h>
+
+#include <linux/debugfs.h>
+#include <linux/time.h>
+#include <linux/preempt.h>
+
+#include <asm/atomic.h>
+#include <asm/div64.h>
+
+struct twb_counter {
+	atomic_t count;
+	u64 reset_time_ns;
+};
+
+static u64 read_reset_rate(void *p)
+{
+	u64 rate, ts_ns;
+	struct timespec ts;
+	struct twb_counter *ctr = p;
+
+	/* begin pseudo-atomic region */
+	preempt_disable();
+	rate = atomic_xchg(((atomic_t*)&ctr->count), 0);
+	ktime_get_ts(&ts);
+	preempt_enable();
+	/* end pseudo-atomic region */
+
+	ts_ns = timespec_to_ns(&ts);
+	rate *= NSEC_PER_SEC;
+	do_div(rate, (u32)(ts_ns - ctr->reset_time_ns));
+	ctr->reset_time_ns = ts_ns;
+	return rate;
+}
+
+/* Counter bits */
+static struct twb_counter num_clones;
+static struct twb_counter num_exits;
+
+#ifndef MODULE
+static int inc_clone(unsigned long val, struct task_struct *task)
+{
+	atomic_inc(&num_clones.count);
+	return 0;
+}
+task_watcher_func(clone, inc_clone);
+
+static int inc_exit(unsigned long val, struct task_struct *task)
+{
+	atomic_inc(&num_exits.count);
+	return 0;
+}
+task_watcher_func(exit, inc_exit);
+#endif
+
+/* Debugfs bits */
+static struct dentry *twb_root, *twb_clone_file, *twb_exit_file;
+
+/*
+ * NOTE: Because open doesn't reset the count nor the reset time,
+ *       userspace must discard the first values read before starting
+ *       to monitor the rates.
+ */
+DEFINE_SIMPLE_ATTRIBUTE(rrr, read_reset_rate, NULL, "%llu\n");
+
+static int __init twb_debugfs_init(void)
+{
+	twb_root = debugfs_create_dir("twbench", NULL);
+	if (!twb_root)
+		return -ENOMEM;
+	twb_clone_file = debugfs_create_file("clones", 0744, twb_root,
+					     &num_clones, &rrr);
+	twb_exit_file = debugfs_create_file("exits", 0744, twb_root,
+					    &num_exits, &rrr);
+	return 0;
+}
+#ifndef MODULE
+__initcall(twb_debugfs_init);
+#endif
+
+static void __exit twb_debugfs_exit(void)
+{
+	debugfs_remove(twb_clone_file);
+	debugfs_remove(twb_exit_file);
+	debugfs_remove(twb_root);
+}
+
+static int __init twb_mod_init(void)
+{
+	int ret;
+
+	ret = twb_debugfs_init();
+	return ret;
+}
+
+static void __exit twb_mod_exit(void)
+{
+	twb_debugfs_exit();
+}
+
+module_init(twb_mod_init);
+module_exit(twb_mod_exit);
Index: linux-2.6.18-mm1/lib/Kconfig.debug
===================================================================
--- linux-2.6.18-mm1.orig/lib/Kconfig.debug
+++ linux-2.6.18-mm1/lib/Kconfig.debug
@@ -443,10 +443,22 @@ config RCU_TORTURE_TEST
 	  Say Y here if you want RCU torture tests to start automatically
 	  at boot time (you probably don't).
 	  Say M if you want the RCU torture tests to build as a module.
 	  Say N if you are unsure.
 
+config TWBENCH
+	bool "Output fork/clone and exit rates"
+	depends on DEBUG_KERNEL && DEBUG_FS
+	default n
+	help
+	   Print out the rate at which the system is fork/cloning new
+	   processes to <debugfs>/twbench/clone
+	   Print out the rate at which the system is exitting existing
+	   processes to <debugfs>/twbench/exit
+
+	   If unsure, say N.
+
 config LKDTM
 	tristate "Linux Kernel Dump Test Tool Module"
 	depends on KPROBES
 	default n
 	help
Index: linux-2.6.18-mm1/kernel/Makefile
===================================================================
--- linux-2.6.18-mm1.orig/kernel/Makefile
+++ linux-2.6.18-mm1/kernel/Makefile
@@ -45,10 +45,11 @@ obj-$(CONFIG_KPROBES) += kprobes.o
 obj-$(CONFIG_SYSFS) += ksysfs.o
 obj-$(CONFIG_DETECT_SOFTLOCKUP) += softlockup.o
 obj-$(CONFIG_GENERIC_HARDIRQS) += irq/
 obj-$(CONFIG_SECCOMP) += seccomp.o
 obj-$(CONFIG_RCU_TORTURE_TEST) += rcutorture.o
+obj-$(CONFIG_TWBENCH) += twbench.o
 obj-$(CONFIG_DEBUG_SYNCHRO_TEST) += synchro-test.o
 obj-$(CONFIG_RELAY) += relay.o
 obj-$(CONFIG_UTS_NS) += utsname.o
 obj-$(CONFIG_TASK_DELAY_ACCT) += delayacct.o
 obj-$(CONFIG_TASKSTATS) += taskstats.o tsacct.o

--