runtime/probes/bench/bench_io1.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

#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_printf("dropped %d packets\n", atomic_read(&_stp_transport_failures));
  _stp_print_flush();
}

void cleanup_module(void)
{
  _stp_transport_close();
}

MODULE_LICENSE("GPL");