From 22ccaa9985211ad13e8486bf2845114f409df55d Mon Sep 17 00:00:00 2001 From: hunt Date: Sat, 18 Jun 2005 07:06:38 +0000 Subject: 2005-06-18 Martin Hunt * build: Modified to use build_probe. * build_probe: New file. This does the work of building a single probe or set of probes in a directory. * Makefile.template: New file. Template used by build_probe to create Makefiles. * agg: New set of probes to test/demonstrate Counter and Stat aggregations. * all probes modified for latest changes and new build process. --- runtime/probes/agg/count2.c | 107 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 107 insertions(+) create mode 100644 runtime/probes/agg/count2.c (limited to 'runtime/probes/agg/count2.c') diff --git a/runtime/probes/agg/count2.c b/runtime/probes/agg/count2.c new file mode 100644 index 00000000..ab13f5af --- /dev/null +++ b/runtime/probes/agg/count2.c @@ -0,0 +1,107 @@ +#define STP_NETLINK_ONLY +#define STP_NUM_STRINGS 1 +#include "runtime.h" + +#include "counter.c" +#include "probes.c" + +MODULE_DESCRIPTION("SystemTap probe: count1"); +MODULE_AUTHOR("Martin Hunt "); + +Counter opens; +Counter reads; +Counter writes; +Counter read_bytes; +Counter write_bytes; + +asmlinkage long inst_sys_open (const char __user * filename, int flags, int mode) +{ + _stp_counter_add (opens, 1); + jprobe_return(); + return 0; +} + +asmlinkage ssize_t inst_sys_read (unsigned int fd, char __user * buf, size_t count) +{ + _stp_counter_add (reads, 1); + _stp_counter_add (read_bytes, count); + jprobe_return(); + return 0; +} + +asmlinkage ssize_t inst_sys_write (unsigned int fd, const char __user * buf, size_t count) +{ + _stp_counter_add (writes, 1); + _stp_counter_add (write_bytes, count); + jprobe_return(); + return 0; +} + +static struct jprobe stp_probes[] = { + { + .kp.addr = (kprobe_opcode_t *)"sys_open", + .entry = (kprobe_opcode_t *) inst_sys_open + }, + { + .kp.addr = (kprobe_opcode_t *)"sys_read", + .entry = (kprobe_opcode_t *) inst_sys_read + }, + { + .kp.addr = (kprobe_opcode_t *)"sys_write", + .entry = (kprobe_opcode_t *) inst_sys_write + }, +}; + +#define MAX_STP_ROUTINE (sizeof(stp_probes)/sizeof(struct jprobe)) + +static int pid; +module_param(pid, int, 0); +MODULE_PARM_DESC(pid, "daemon pid"); + +int init_module(void) +{ + int ret; + + if (!pid) { + printk("init: Can't start without daemon pid\n"); + return -1; + } + + if (_stp_transport_open(n_subbufs, subbuf_size, pid) < 0) { + printk("init: Couldn't open transport\n"); + return -1; + } + + opens = _stp_counter_init(); + reads = _stp_counter_init(); + writes = _stp_counter_init(); + read_bytes = _stp_counter_init(); + write_bytes = _stp_counter_init(); + + ret = _stp_register_jprobes (stp_probes, MAX_STP_ROUTINE); + + return ret; +} + +static void probe_exit (void) +{ + int i; + + _stp_unregister_jprobes (stp_probes, MAX_STP_ROUTINE); + + _stp_printf ("open calls: %lld\n", _stp_counter_get(opens, 0)); + _stp_printf ("read calls: %lld\n", _stp_counter_get(reads, 0)); + _stp_printf ("read bytes: %lld\n", _stp_counter_get(read_bytes, 0)); + _stp_printf ("write calls: %lld\n", _stp_counter_get(writes, 0)); + _stp_printf ("write bytes: %lld\n", _stp_counter_get(write_bytes, 0)); + + _stp_print_flush(); +} + +void cleanup_module(void) +{ + _stp_transport_close(); +} + +MODULE_LICENSE("GPL"); + -- cgit