1 files changed, 7 insertions, 9 deletions
diff --git a/runtime/probes/bench/bench_io1.c b/runtime/probes/bench/bench_io1.c
index a0a44d34..89be7f06 100644
--- a/runtime/probes/bench/bench_io1.c
+++ b/runtime/probes/bench/bench_io1.c
@@ -4,10 +4,10 @@
 #include "runtime.h"
 #include "probes.c"
 
-MODULE_DESCRIPTION("SystemTap probe: bench_io");
+MODULE_DESCRIPTION("SystemTap probe: bench_io1");
 MODULE_AUTHOR("Martin Hunt");
 
-static int inst_sys_read (struct kprobe *p, struct pt_regs *regs)
+static int inst_sys_getuid (struct kprobe *p, struct pt_regs *regs)
 {
   /* print 100 chars */
   _stp_printf ("123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789\n");
@@ -15,7 +15,7 @@ static int inst_sys_read (struct kprobe *p, struct pt_regs *regs)
   return 0;
 }
 
-static int inst_sys_write (struct kprobe *p, struct pt_regs *regs)
+static int inst_sys_getgid (struct kprobe *p, struct pt_regs *regs)
 {
   /* print 100 chars */
   _stp_print_cstr ("123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789\n");
@@ -25,12 +25,12 @@ static int inst_sys_write (struct kprobe *p, struct pt_regs *regs)
 
 static struct kprobe kp[] = {
   {
-    .addr = "sys_read",
-    .pre_handler = inst_sys_read
+    .addr = "sys_getuid",
+    .pre_handler = inst_sys_getuid
   },
   {
-    .addr = "sys_write",
-    .pre_handler = inst_sys_write
+    .addr = "sys_getgid",
+    .pre_handler = inst_sys_getgid
   }
 };
 
@@ -44,6 +44,4 @@ int probe_start(void)
 static void probe_exit (void)
 {
   _stp_unregister_kprobes (kp, NUM_KPROBES); 
-  _stp_printf("dropped %d packets\n", atomic_read(&_stp_transport_failures));
-  _stp_print_flush();
 }