Fixed some issue on tracking when protecting memory with PROT=0

ptitSeb · Jul 13, 2024 · dcc0355 · dcc0355
1 parent 666b5a6
commit dcc0355
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/custommem.c b/src/custommem.c
@@ -957,6 +957,7 @@ void setProtection_mmap(uintptr_t addr, size_t size, uint32_t prot)
     else {
         mutex_lock(&mutex_prot);
         rb_set(mapallmem, addr, addr+size, 1);
+        rb_unset(memprot, addr, addr+size);
         mutex_unlock(&mutex_prot);
     }
 }
@@ -970,6 +971,7 @@ void setProtection_elf(uintptr_t addr, size_t size, uint32_t prot)
     else {
         mutex_lock(&mutex_prot);
         rb_set(mapallmem, addr, addr+size, 1);
+        rb_unset(memprot, addr, addr+size);
         mutex_unlock(&mutex_prot);
     }
 }