2005-06-28 Martin Hunt <hunt@redhat.com>

	* bench: New probe to do benchmarks.

1 files changed, 60 insertions, 0 deletions

diff --git a/runtime/probes/bench/bench_io2.c b/runtime/probes/bench/bench_io2.c
new file mode 100644
index 00000000..d606fa64
--- /dev/null
+++ b/runtime/probes/bench/bench_io2.c
@@ -0,0 +1,60 @@
+//#define STP_NETLINK_ONLY
+#define STP_NUM_STRINGS 1
+
+#include "runtime.h"
+#include "probes.c"
+
+MODULE_DESCRIPTION("SystemTap probe: bench_io");
+MODULE_AUTHOR("Martin Hunt");
+
+static int inst_sys_read (struct kprobe *p, struct pt_regs *regs)
+{
+  /* print 100 chars */
+  _stp_printf ("123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789\n");
+  _stp_print_flush();
+  return 0;
+}
+
+static int inst_sys_write (struct kprobe *p, struct pt_regs *regs)
+{
+  /* print 100 chars */
+  _stp_print_cstr ("123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789\n");
+  _stp_print_flush();
+  return 0;
+}
+
+static struct kprobe kp[] = {
+  {
+    .addr = "sys_read",
+    .pre_handler = inst_sys_read
+  },
+  {
+    .addr = "sys_write",
+    .pre_handler = inst_sys_write
+  }
+};
+
+#define NUM_KPROBES (sizeof(kp)/sizeof(struct kprobe))
+
+int init_module(void)
+{
+  int ret;
+
+  TRANSPORT_OPEN;
+  
+  ret = _stp_register_kprobes (kp, NUM_KPROBES);
+  return ret;
+}
+
+static void probe_exit (void)
+{
+  _stp_unregister_kprobes (kp, NUM_KPROBES); 
+  _stp_print_flush();
+}
+
+void cleanup_module(void)
+{
+  _stp_transport_close();
+}
+
+MODULE_LICENSE("GPL");