notdirty_write check for addr_write in snapshot case

unicorn-engine · Oct 22, 2024 · 2717b79 · 2717b79
1 parent 149f17c
commit 2717b79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qemu/accel/tcg/cputlb.c b/qemu/accel/tcg/cputlb.c
@@ -1234,7 +1234,7 @@ static void notdirty_write(CPUState *cpu, vaddr mem_vaddr, unsigned size,
     // - have memory hooks installed
     // - or doing snapshot
     // , then never clean the tlb
-    if (!(!mr || mr->priority < cpu->uc->snapshot_level) &&
+    if (!(!mr || (tlbe->addr_write != -1 && mr->priority < cpu->uc->snapshot_level)) &&
             !(tlbe->addr_code != -1) &&
             !uc_mem_hook_installed(cpu->uc, tlbe->paddr | (mem_vaddr & ~TARGET_PAGE_MASK))) {
         tlb_set_dirty(cpu, mem_vaddr);