#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 libafl_edge_hook {
    // functions
    uint64_t (*gen)(uint64_t data, target_ulong src, target_ulong dst);
    size_t (*jit)(uint64_t data, uint64_t id); // optional opt

    // data
    uint64_t data;
    size_t num;
    uint64_t cur_id;

    // helpers
    TCGHelperInfo helper_info;

    // next
    struct libafl_edge_hook* next;
};

TranslationBlock* libafl_gen_edge(CPUState* cpu, target_ulong src_block,
                                  target_ulong dst_block, int exit_n,
                                  target_ulong cs_base, uint32_t flags,
                                  int cflags);

size_t libafl_add_edge_hook(uint64_t (*gen)(uint64_t data, target_ulong src,
                                            target_ulong dst),
                            void (*exec)(uint64_t data, uint64_t id),
                            uint64_t data);

bool libafl_qemu_edge_hook_set_jit(
    size_t num,
    size_t (*jit)(uint64_t,
                  uint64_t)); // no param names to avoid to be marked as safe

int libafl_qemu_remove_edge_hook(size_t num, int invalidate);

bool libafl_qemu_hook_edge_gen(target_ulong src_block, target_ulong dst_block);
void libafl_qemu_hook_edge_run(void);