ee43af7f80
* Run thread hooks for KVM mode * Unify qemu init function symbol for systemmode and usermode * get tid from caller instead of callee
48 lines
1.2 KiB
C
48 lines
1.2 KiB
C
#include "libafl/hooks/thread.h"
|
|
#include "libafl/cpu.h"
|
|
|
|
#include <linux/unistd.h>
|
|
|
|
struct libafl_new_thread_hook* libafl_new_thread_hooks;
|
|
size_t libafl_new_thread_hooks_num = 0;
|
|
|
|
GEN_REMOVE_HOOK1(new_thread)
|
|
|
|
size_t libafl_add_new_thread_hook(bool (*callback)(uint64_t data,
|
|
CPUArchState* env,
|
|
uint32_t tid),
|
|
uint64_t data)
|
|
{
|
|
struct libafl_new_thread_hook* hook =
|
|
calloc(sizeof(struct libafl_new_thread_hook), 1);
|
|
hook->callback = callback;
|
|
hook->data = data;
|
|
hook->num = libafl_new_thread_hooks_num++;
|
|
hook->next = libafl_new_thread_hooks;
|
|
libafl_new_thread_hooks = hook;
|
|
|
|
return hook->num;
|
|
}
|
|
|
|
bool libafl_hook_new_thread_run(CPUArchState* env, uint32_t tid)
|
|
{
|
|
#ifdef CONFIG_USER_ONLY
|
|
libafl_set_qemu_env(env);
|
|
#endif
|
|
|
|
if (libafl_new_thread_hooks) {
|
|
bool continue_execution = true;
|
|
|
|
struct libafl_new_thread_hook* h = libafl_new_thread_hooks;
|
|
while (h) {
|
|
continue_execution =
|
|
h->callback(h->data, env, tid) && continue_execution;
|
|
h = h->next;
|
|
}
|
|
|
|
return continue_execution;
|
|
}
|
|
|
|
return true;
|
|
}
|