79 lines
2.3 KiB
C
79 lines
2.3 KiB
C
#pragma once
|
|
|
|
#include "qemu/osdep.h"
|
|
|
|
#include "qapi/error.h"
|
|
|
|
#include "exec/exec-all.h"
|
|
#include "exec/tb-flush.h"
|
|
|
|
#include "libafl/exit.h"
|
|
#include "libafl/hook.h"
|
|
|
|
enum libafl_syshook_ret_tag {
|
|
LIBAFL_SYSHOOK_RUN,
|
|
LIBAFL_SYSHOOK_SKIP,
|
|
};
|
|
|
|
// Representation of a pre-syscall hook result.
|
|
// It is associated with the LibAFL enum `SyscallHookResult`.
|
|
// Any change made here should be also propagated to the Rust enum.
|
|
struct libafl_syshook_ret {
|
|
enum libafl_syshook_ret_tag tag;
|
|
union {
|
|
target_ulong syshook_skip_retval;
|
|
};
|
|
};
|
|
|
|
typedef struct libafl_syshook_ret (*libafl_pre_syscall_cb)(
|
|
uint64_t data, int sys_num, target_ulong arg0, target_ulong arg1,
|
|
target_ulong arg2, target_ulong arg3, target_ulong arg4, target_ulong arg5,
|
|
target_ulong arg6, target_ulong arg7);
|
|
|
|
typedef target_ulong (*libafl_post_syscall_cb)(
|
|
uint64_t data, target_ulong ret, int sys_num, target_ulong arg0,
|
|
target_ulong arg1, target_ulong arg2, target_ulong arg3, target_ulong arg4,
|
|
target_ulong arg5, target_ulong arg6, target_ulong arg7);
|
|
|
|
struct libafl_pre_syscall_hook {
|
|
// functions
|
|
libafl_pre_syscall_cb callback;
|
|
|
|
// data
|
|
uint64_t data;
|
|
size_t num;
|
|
|
|
// next
|
|
struct libafl_pre_syscall_hook* next;
|
|
};
|
|
|
|
struct libafl_post_syscall_hook {
|
|
// functions
|
|
libafl_post_syscall_cb callback;
|
|
|
|
// data
|
|
uint64_t data;
|
|
size_t num;
|
|
|
|
// next
|
|
struct libafl_post_syscall_hook* next;
|
|
};
|
|
|
|
size_t libafl_add_pre_syscall_hook(libafl_pre_syscall_cb callback,
|
|
uint64_t data);
|
|
size_t libafl_add_post_syscall_hook(libafl_post_syscall_cb callback,
|
|
uint64_t data);
|
|
|
|
int libafl_qemu_remove_pre_syscall_hook(size_t num);
|
|
int libafl_qemu_remove_post_syscall_hook(size_t num);
|
|
|
|
bool libafl_hook_syscall_pre_run(CPUArchState* env, int num, abi_long arg1,
|
|
abi_long arg2, abi_long arg3, abi_long arg4,
|
|
abi_long arg5, abi_long arg6, abi_long arg7,
|
|
abi_long arg8, abi_long* ret);
|
|
|
|
void libafl_hook_syscall_post_run(int num, abi_long arg1, abi_long arg2,
|
|
abi_long arg3, abi_long arg4, abi_long arg5,
|
|
abi_long arg6, abi_long arg7, abi_long arg8,
|
|
abi_long* ret);
|