#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"

struct syshook_ret {
    target_ulong retval;
    bool skip_syscall;
};

struct libafl_pre_syscall_hook {
    // functions
    struct syshook_ret (*callback)(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);

    // data
    uint64_t data;
    size_t num;

    // next
    struct libafl_pre_syscall_hook* next;
};

struct libafl_post_syscall_hook {
    // functions
    target_ulong (*callback)(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);

    // data
    uint64_t data;
    size_t num;

    // next
    struct libafl_post_syscall_hook* next;
};

size_t libafl_add_pre_syscall_hook(
    struct syshook_ret (*callback)(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),
    uint64_t data);
size_t libafl_add_post_syscall_hook(
    target_ulong (*callback)(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),
    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);