summaryrefslogtreecommitdiffstats
path: root/runtime/probes/tasklet/stp_tasklet.c
blob: c3581eb2c2025c2b6458b2217b99bc53ab862f0c (plain)
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
/* Framework for putting a jprobe in a tasklet. */
/* Useful for testing probes in interrupt context. */
/* Doesn't do anything useful as is.  Put test code in the inst func */

#define HASH_TABLE_BITS 8
#define HASH_TABLE_SIZE (1<<HASH_TABLE_BITS)
#define BUCKETS 16 /* largest histogram width */

#define STP_NETLINK_ONLY
#define STP_NUM_STRINGS 1

#include "runtime.h"
#include "probes.c"

MODULE_DESCRIPTION("test jprobes of tasklets");
MODULE_AUTHOR("Martin Hunt <hunt@redhat.com>");

void inst__rcu_process_callbacks(struct rcu_ctrlblk *rcp,
				 struct rcu_state *rsp, struct rcu_data *rdp)
{
  _stp_log ("interrupt=%d\n", in_interrupt());
  jprobe_return();
}

static struct jprobe stp_probes[] = {
  {
    .kp.addr =  (kprobe_opcode_t *)"__rcu_process_callbacks",
    .entry = (kprobe_opcode_t *) inst__rcu_process_callbacks
  },
};
#define MAX_STP_PROBES (sizeof(stp_probes)/sizeof(struct jprobe))

static unsigned n_subbufs = 4;
module_param(n_subbufs, uint, 0);
MODULE_PARM_DESC(n_subbufs, "number of sub-buffers per per-cpu buffer");

static unsigned subbuf_size = 65536;
module_param(subbuf_size, uint, 0);
MODULE_PARM_DESC(subbuf_size, "size of each per-cpu sub-buffers");

static int pid;
module_param(pid, int, 0);
MODULE_PARM_DESC(pid, "daemon pid");

static int init_stp(void)
{
	int ret;

	if (!pid) {
		printk("init_dtr: Can't start without daemon pid\n");		
		return -1;
	}

	if (_stp_transport_open(n_subbufs, subbuf_size, pid) < 0) {
		printk("init_dtr: Couldn't open transport\n");		
		return -1;
	}

	ret = _stp_register_jprobes (stp_probes, MAX_STP_PROBES);
	printk("instrumentation is enabled...\n");
	return ret;
}

static int exited; /* FIXME: this is a stopgap - if we don't do this
		    * and are manually removed, bad things happen */

static void probe_exit (void)
{
	exited = 1;

	_stp_unregister_jprobes (stp_probes, MAX_STP_PROBES);
	_stp_log ("EXIT\n");

}
static void cleanup_stp(void)
{
	if (!exited)
		probe_exit();
	
	_stp_transport_close();
}

module_init(init_stp);
module_exit(cleanup_stp);
MODULE_LICENSE("GPL");