From 2c9fe3973a28504e664c5d7c9d990bc2f16c0348 Mon Sep 17 00:00:00 2001 From: Vito Caputo Date: Tue, 7 Mar 2017 18:27:41 -0800 Subject: libvmon: add "sys-wide" arg for sample callbacks We need to eventually plumb an vwm_overlays_t reference back to sample_cb, for now we'll just obviate the need for the vwm_ptr global by plumbing the vwm_t through. --- src/libvmon/vmon.c | 24 ++++++++++++------------ src/libvmon/vmon.h | 9 +++++---- 2 files changed, 17 insertions(+), 16 deletions(-) (limited to 'src/libvmon') diff --git a/src/libvmon/vmon.c b/src/libvmon/vmon.c index 5279ab7..1995228 100644 --- a/src/libvmon/vmon.c +++ b/src/libvmon/vmon.c @@ -1123,7 +1123,7 @@ static int find_proc_in_array(vmon_t *vmon, vmon_proc_t *proc, int hint) /* simple helper for installing callbacks on the callback lists, currently only used for the per-process sample callbacks */ /* will not install the same callback function & arg combination more than once, and will not install NULL-functioned callbacks at all! */ -static int maybe_install_proc_callback(vmon_t *vmon, list_head_t *callbacks, void (*func)(vmon_t *, vmon_proc_t *, void *), void *arg) +static int maybe_install_proc_callback(vmon_t *vmon, list_head_t *callbacks, void (*func)(vmon_t *, void *, vmon_proc_t *, void *), void *arg) { if(func) { vmon_proc_callback_t *cb; @@ -1152,7 +1152,7 @@ static int maybe_install_proc_callback(vmon_t *vmon, list_head_t *callbacks, voi /* monitor a process under a given vmon instance, the public interface. * XXX note it's impossible to say "none" for wants per-process, just "inherit", if vmon_init() was told a proc_wants of "inherit" then it's like having "none" * proc_wants for all proceses, perhaps improve this if there's a pressure to support this use case */ -vmon_proc_t * vmon_proc_monitor(vmon_t *vmon, vmon_proc_t *parent, int pid, vmon_proc_wants_t wants, void (*sample_cb)(vmon_t *, vmon_proc_t *, void *), void *sample_cb_data) +vmon_proc_t * vmon_proc_monitor(vmon_t *vmon, vmon_proc_t *parent, int pid, vmon_proc_wants_t wants, void (*sample_cb)(vmon_t *, void *, vmon_proc_t *, void *), void *sample_cb_arg) { vmon_proc_t *proc; int hash = (pid % VMON_HTAB_SIZE), i; @@ -1165,7 +1165,7 @@ vmon_proc_t * vmon_proc_monitor(vmon_t *vmon, vmon_proc_t *parent, int pid, vmon list_for_each_entry(proc, &vmon->htab[hash], bucket) { /* search for the process to see if it's already monitored, we allow threads to exist with the same pid hence the additional is_thread comparison */ if(proc->pid == pid && proc->is_thread == is_thread) { - if(!maybe_install_proc_callback(vmon, &proc->sample_callbacks, sample_cb, sample_cb_data)) { + if(!maybe_install_proc_callback(vmon, &proc->sample_callbacks, sample_cb, sample_cb_arg)) { return NULL; } @@ -1215,7 +1215,7 @@ vmon_proc_t * vmon_proc_monitor(vmon_t *vmon, vmon_proc_t *parent, int pid, vmon INIT_LIST_HEAD(&proc->siblings); INIT_LIST_HEAD(&proc->threads); - if(!maybe_install_proc_callback(vmon, &proc->sample_callbacks, sample_cb, sample_cb_data)) { + if(!maybe_install_proc_callback(vmon, &proc->sample_callbacks, sample_cb, sample_cb_arg)) { free(proc); return NULL; } @@ -1256,8 +1256,8 @@ vmon_proc_t * vmon_proc_monitor(vmon_t *vmon, vmon_proc_t *parent, int pid, vmon } -/* stop monitoring a process under a given vmon instance, a caller who supplied a sample_cb & sample_cb_data pair @ monitor() must also supply it @ unmonitor! */ -void vmon_proc_unmonitor(vmon_t *vmon, vmon_proc_t *proc, void (*sample_cb)(vmon_t *, vmon_proc_t *, void *), void *sample_cb_data) +/* stop monitoring a process under a given vmon instance, a caller who supplied a sample_cb & sample_cb_arg pair @ monitor() must also supply it @ unmonitor! */ +void vmon_proc_unmonitor(vmon_t *vmon, vmon_proc_t *proc, void (*sample_cb)(vmon_t *, void *, vmon_proc_t *, void *), void *sample_cb_arg) { vmon_proc_t *child, *_child; int i; @@ -1266,7 +1266,7 @@ void vmon_proc_unmonitor(vmon_t *vmon, vmon_proc_t *proc, void (*sample_cb)(vmon vmon_proc_callback_t *cb, *_cb; list_for_each_entry_safe(cb, _cb, &proc->sample_callbacks, callbacks) { - if(cb->func == sample_cb && cb->arg == sample_cb_data) { + if(cb->func == sample_cb && cb->arg == sample_cb_arg) { list_del(&cb->callbacks); free(cb); break; @@ -1374,7 +1374,7 @@ static int sample_threads(vmon_t *vmon, list_head_t *threads) vmon_proc_callback_t *cb; list_for_each_entry(cb, &proc->sample_callbacks, callbacks) { - cb->func(vmon, proc, cb->arg); + cb->func(vmon, vmon->sample_cb_arg, proc, cb->arg); } #endif } @@ -1414,7 +1414,7 @@ static int sample_siblings(vmon_t *vmon, list_head_t *siblings) * being monitored, handy when automatically following children, an immediately relevant use case (vwm) */ list_for_each_entry(cb, &proc->sample_callbacks, callbacks) { - cb->func(vmon, proc, cb->arg); + cb->func(vmon, vmon->sample_cb_arg, proc, cb->arg); } /* transition new to non-new processes where we're responsible, this is a slight problem */ @@ -1482,7 +1482,7 @@ static int sample_siblings_pass2(vmon_t *vmon, list_head_t *siblings) sample_siblings_pass2(vmon, &proc->children); /* recurse into children, we invoke callbacks as encountered on nodes from the leaves up */ list_for_each_entry(cb, &proc->sample_callbacks, callbacks) { - cb->func(vmon, proc, cb->arg); + cb->func(vmon, vmon->sample_cb_arg, proc, cb->arg); } if(!proc->parent && proc->is_new) { /* top-level processes aren't managed by a follower/sampler, so we need to clear their is_new flag, this approach is slightly deviant from the managed case, @@ -1581,7 +1581,7 @@ int vmon_sample(vmon_t *vmon) } if(vmon->sample_cb) { - vmon->sample_cb(vmon); + vmon->sample_cb(vmon, vmon->sample_cb_arg); } /* then the per-process samplers */ @@ -1607,7 +1607,7 @@ int vmon_sample(vmon_t *vmon) sample(vmon, proc); list_for_each_entry(cb, &proc->sample_callbacks, callbacks) { - cb->func(vmon, proc, cb->arg); + cb->func(vmon, vmon->sample_cb_arg, proc, cb->arg); } /* age process, we use the presence of a parent as a flag indicating if the process is managed ala follow children/threads diff --git a/src/libvmon/vmon.h b/src/libvmon/vmon.h index 06a062b..56c7b56 100644 --- a/src/libvmon/vmon.h +++ b/src/libvmon/vmon.h @@ -217,7 +217,7 @@ struct _vmon_t; /* list of callbacks is maintained for the per-process callbacks, it's convenient to do things like update multiple dynamic contexts associated with a given process monitor (think multiple windows) */ typedef struct _vmon_proc_callback_t { - void (*func)(struct _vmon_t *, struct _vmon_proc_t *, void *); + void (*func)(struct _vmon_t *, void *, struct _vmon_proc_t *, void *); void *arg; list_head_t callbacks; } vmon_proc_callback_t; @@ -283,7 +283,8 @@ typedef struct _vmon_t { vmon_sys_wants_t activity; /* bits updated when there's activity on the respective wants (stores have changes) */ void *stores[VMON_STORE_SYS_NR]; /* stores for the sys-wide wants */ - void (*sample_cb)(struct _vmon_t *); /* callback invoked after executing the selected sys wants (once per vmon_sample() call)) */ + void (*sample_cb)(struct _vmon_t *, void *); /* callback invoked after executing the selected sys wants (once per vmon_sample() call)) */ + void *sample_cb_arg; /* user pointer for sample_cb */ char buf[8192]; /* scratch buffer for private use XXX: it may make sense to dynamically size these... */ char buf_bis[8192]; /* secondary scratch buffer for private use */ @@ -301,8 +302,8 @@ typedef struct _vmon_t { int vmon_init(vmon_t *, vmon_flags_t, vmon_sys_wants_t, vmon_proc_wants_t); void vmon_destroy(vmon_t *); -vmon_proc_t * vmon_proc_monitor(vmon_t *, vmon_proc_t *, int, vmon_proc_wants_t, void (*)(vmon_t *, vmon_proc_t *, void *), void *); -void vmon_proc_unmonitor(vmon_t *, vmon_proc_t *, void (*)(vmon_t *, vmon_proc_t *, void *), void *); +vmon_proc_t * vmon_proc_monitor(vmon_t *, vmon_proc_t *, int, vmon_proc_wants_t, void (*)(vmon_t *, void *, vmon_proc_t *, void *), void *); +void vmon_proc_unmonitor(vmon_t *, vmon_proc_t *, void (*)(vmon_t *, void *, vmon_proc_t *, void *), void *); int vmon_sample(vmon_t *); #endif -- cgit v1.2.3