Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add a flag for edge TB so we can unlink it completely #53

Merged
merged 4 commits into from
Apr 26, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions accel/tcg/tb-maint.c
Original file line number Diff line number Diff line change
Expand Up @@ -866,6 +866,11 @@ static inline void tb_jmp_unlink(TranslationBlock *dest)

TB_FOR_EACH_JMP(dest, tb, n) {
tb_reset_jump(tb, n);
//// --- Begin LibAFL code ---
if (tb->cflags & CF_IS_EDGE) {
tb_phys_invalidate(tb, -1);
}
//// --- End LibAFL code ---
qatomic_and(&tb->jmp_dest[n], (uintptr_t)NULL | 1);
/* No need to clear the list entry; setting the dest ptr is enough */
}
Expand Down
2 changes: 1 addition & 1 deletion accel/tcg/translate-all.c
Original file line number Diff line number Diff line change
Expand Up @@ -403,7 +403,7 @@ TranslationBlock *libafl_gen_edge(CPUState *cpu, target_ulong src_block,
}
tb->cs_base = cs_base;
tb->flags = flags;
tb->cflags = cflags;
tb->cflags = cflags | CF_IS_EDGE;
//tb_set_page_addr0(tb, phys_pc);
//tb_set_page_addr1(tb, -1);
tcg_ctx->gen_tb = tb;
Expand Down
3 changes: 3 additions & 0 deletions include/exec/translation-block.h
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,9 @@ struct TranslationBlock {
#define CF_PARALLEL 0x00008000 /* Generate code for a parallel context */
#define CF_NOIRQ 0x00010000 /* Generate an uninterruptible TB */
#define CF_PCREL 0x00020000 /* Opcodes in TB are PC-relative */
//// --- Begin LibAFL code ---
#define CF_IS_EDGE 0x00800000 /* The current TB is an edge */
//// --- End LibAFL code ---
#define CF_CLUSTER_MASK 0xff000000 /* Top 8 bits are cluster ID */
#define CF_CLUSTER_SHIFT 24

Expand Down
Loading