Gecko [ dataharbor.africa ]

Name	Size	Permission
?;dyninst	[ DIR ]	drwxr-xr-x
?;linux	[ DIR ]	drwxr-xr-x
?;transport	[ DIR ]	drwxr-xr-x
?;unwind	[ DIR ]	drwxr-xr-x
addr-map.c	523 B	-rw-r--r--
addr-map.h	723 B	-rw-r--r--
alloc.c	498 B	-rw-r--r--
arith.c	483 B	-rw-r--r--
common_probe_context.h	6.08 KB	-rw-r--r--
common_session_state.h	415 B	-rw-r--r--
compatdefs.h	3.02 KB	-rw-r--r--
copy.c	585 B	-rw-r--r--
debug.h	485 B	-rw-r--r--
io.c	508 B	-rw-r--r--
k_syms.h	182 B	-rw-r--r--
loc2c-runtime.h	8.96 KB	-rw-r--r--
map-gen.c	25.29 KB	-rw-r--r--
map-stat.c	3.38 KB	-rw-r--r--
map.c	14.1 KB	-rw-r--r--
map.h	5.78 KB	-rw-r--r--
mempool.c	2.58 KB	-rw-r--r--
namespaces.h	487 B	-rw-r--r--
pmap-gen.c	8.17 KB	-rw-r--r--
print.c	486 B	-rw-r--r--
print.h	680 B	-rw-r--r--
print_flush.c	4.29 KB	-rw-r--r--
probe_lock.h	498 B	-rw-r--r--
proc_fs_compatibility.h	2.22 KB	-rw-r--r--
procfs-probes.c	4.21 KB	-rw-r--r--
procfs-probes.h	674 B	-rw-r--r--
procfs.c	7.3 KB	-rw-r--r--
ptrace_compatibility.h	1.48 KB	-rw-r--r--
regs.c	4.18 KB	-rw-r--r--
regs.h	2.12 KB	-rw-r--r--
runtime.h	928 B	-rw-r--r--
runtime_context.h	669 B	-rw-r--r--
runtime_defines.h	3.5 KB	-rw-r--r--
stack-arm.c	1.65 KB	-rw-r--r--
stack-ia64.c	1.06 KB	-rw-r--r--
stack-mips.c	125 B	-rw-r--r--
stack-s390.c	2.25 KB	-rw-r--r--
stack.c	18.79 KB	-rw-r--r--
stat-common.c	8.29 KB	-rw-r--r--
stat.c	7.25 KB	-rw-r--r--
stat.h	1.92 KB	-rw-r--r--
stp_helper_lock.h	2.98 KB	-rw-r--r--
stp_string.c	8.15 KB	-rw-r--r--
stp_string.h	598 B	-rw-r--r--
stp_task_work.c	2.75 KB	-rw-r--r--
stp_utrace.c	94.13 KB	-rw-r--r--
stp_utrace.h	13.68 KB	-rw-r--r--
sym.c	34.44 KB	-rw-r--r--
sym.h	5.83 KB	-rw-r--r--
syscall.h	15.61 KB	-rw-r--r--
task_finder_vma.c	12.81 KB	-rw-r--r--
time.c	13.31 KB	-rw-r--r--
timer.c	481 B	-rw-r--r--
timer.h	482 B	-rw-r--r--
uidgid_compatibility.h	603 B	-rw-r--r--
unwind.c	46.51 KB	-rw-r--r--
utrace_compatibility.h	3.07 KB	-rw-r--r--
vma.c	8.83 KB	-rw-r--r--
vsprintf.c	22.98 KB	-rw-r--r--
vsprintf.h	1.47 KB	-rw-r--r--

Code Editor : stp_task_work.c

#ifndef _STP_TASK_WORK_C
#define _STP_TASK_WORK_C

#include "linux/task_work_compatibility.h"

#if !defined(STAPCONF_TASK_WORK_ADD_EXPORTED)
// First typedef from the original decls, then #define as typecasted calls.
typedef typeof(&task_work_add) task_work_add_fn;
#define task_work_add (* (task_work_add_fn)kallsyms_task_work_add)
typedef typeof(&task_work_cancel) task_work_cancel_fn;
#define task_work_cancel (* (task_work_cancel_fn)kallsyms_task_work_cancel)
#endif

/* To avoid a crash when a task_work callback gets called after the
 * module is unloaded, keep track of the number of current callbacks. */
static atomic_t stp_task_work_callbacks = ATOMIC_INIT(0);

/*
 * stp_task_work_init() should be called before any other
 * stp_task_work_* functions are called to do setup.
 */
static int
stp_task_work_init(void)
{
#if !defined(STAPCONF_TASK_WORK_ADD_EXPORTED)
	/* The task_work_add()/task_work_cancel() functions aren't
	 * exported. Look up those function addresses. */
        kallsyms_task_work_add = (void *)kallsyms_lookup_name("task_work_add");
        if (kallsyms_task_work_add == NULL) {
		_stp_error("Can't resolve task_work_add!");
		return -ENOENT;
        }
        kallsyms_task_work_cancel = (void *)kallsyms_lookup_name("task_work_cancel");
        if (kallsyms_task_work_cancel == NULL) {
		_stp_error("Can't resolve task_work_cancel!");
		return -ENOENT;
        }
#endif
	return 0;
}

/*
 * stap_task_work_exit() should be called when no more
 * stp_task_work_* functions will be called (before module exit).
 *
 * This function makes sure that all the callbacks are finished before
 * letting the module unload.  If the module unloads before a callback
 * is called, the kernel will try to make a function call to an
 * invalid address.
 */
static void
stp_task_work_exit(void)
{
	while (atomic_read(&stp_task_work_callbacks))
		schedule_timeout_uninterruptible(1);
	return;
}

/*
 * Our task_work_add() wrapper that remembers that we've got a pending
 * callback.
 */
static int
stp_task_work_add(struct task_struct *task, struct task_work *twork)
{
	int rc;

rc = task_work_add(task, twork, true);
	if (rc == 0)
		atomic_inc(&stp_task_work_callbacks);	
	return rc;
}

/*
 * Our task_work_cancel() wrapper that remembers that a callback has
 * been cancelled.
 */
static struct task_work *
stp_task_work_cancel(struct task_struct *task, task_work_func_t func)
{
	struct task_work *twork;

twork = task_work_cancel(task, func);
	if (twork != NULL)
		atomic_dec(&stp_task_work_callbacks);
	return twork;
}

/*
 * stp_task_work_func_done() should be called at the very end of a
 * task_work callback function so that we can keep up with callback
 * accounting.
 */
static void
stp_task_work_func_done(void)
{
	atomic_dec(&stp_task_work_callbacks);
}

#endif /* _STP_TASK_WORK_C */

${this.title}

Code Editor : stp_task_work.c