86d38fbfa7
* Add cpu_run pre/post exec hooks, to hook the cpu just before / after it runs target code. * Refactor hooks to separate them in different files * Remove most extern variables * Reduce the amount of extra code in QEMU * Add clang-format script
42 lines
1.0 KiB
C
42 lines
1.0 KiB
C
#include "libafl/hooks/cpu_run.h"
|
|
|
|
static struct libafl_cpu_run_hook* libafl_cpu_run_hooks = NULL;
|
|
static size_t libafl_cpu_run_hooks_num = 0;
|
|
|
|
GEN_REMOVE_HOOK1(cpu_run)
|
|
|
|
size_t libafl_hook_cpu_run_add(libafl_cpu_run_fn pre_cpu_run,
|
|
libafl_cpu_run_fn post_cpu_run, uint64_t data)
|
|
{
|
|
struct libafl_cpu_run_hook* hook =
|
|
calloc(sizeof(struct libafl_cpu_run_hook), 1);
|
|
|
|
hook->pre_cpu_run = pre_cpu_run;
|
|
hook->post_cpu_run = post_cpu_run;
|
|
|
|
hook->data = data;
|
|
hook->num = libafl_cpu_run_hooks_num++;
|
|
hook->next = libafl_cpu_run_hooks;
|
|
libafl_cpu_run_hooks = hook;
|
|
|
|
return hook->num;
|
|
}
|
|
|
|
void libafl_hook_cpu_run_pre_exec(CPUState* cpu)
|
|
{
|
|
struct libafl_cpu_run_hook* h = libafl_cpu_run_hooks;
|
|
while (h) {
|
|
h->pre_cpu_run(h->data, cpu);
|
|
h = h->next;
|
|
}
|
|
}
|
|
|
|
void libafl_hook_cpu_run_post_exec(CPUState* cpu)
|
|
{
|
|
struct libafl_cpu_run_hook* h = libafl_cpu_run_hooks;
|
|
while (h) {
|
|
h->post_cpu_run(h->data, cpu);
|
|
h = h->next;
|
|
}
|
|
}
|