Jitter: fix automod reset

2 files changed, 12 insertions, 0 deletions

diff --git a/miasm2/jitter/jitcore.py b/miasm2/jitter/jitcore.py
index 2e3411c0..42e6d7a9 100644
--- a/miasm2/jitter/jitcore.py
+++ b/miasm2/jitter/jitcore.py
@@ -269,3 +269,4 @@ class JitCore(object):
             self.del_bloc_in_range(addr_start, addr_stop)
         self.__updt_jitcode_mem_range(vm)
         self.addr_mod = interval()
+        vm.reset_memory_access()
diff --git a/miasm2/jitter/vm_mngr_py.c b/miasm2/jitter/vm_mngr_py.c
index 3e0c820c..1e751ba4 100644
--- a/miasm2/jitter/vm_mngr_py.c
+++ b/miasm2/jitter/vm_mngr_py.c
@@ -321,6 +321,15 @@ PyObject* vm_reset_memory_breakpoint(VmMngr* self, PyObject* args)
 
 }
 
+PyObject* vm_reset_memory_access(VmMngr* self, PyObject* args)
+{
+    reset_memory_access(&self->vm_mngr);
+    Py_INCREF(Py_None);
+    return Py_None;
+
+}
+
+
 PyObject *vm_dump(PyObject* self)
 {
 	char* buf_final;
@@ -583,6 +592,8 @@ static PyMethodDef VmMngr_methods[] = {
 	 "X"},
 	{"get_memory_write",(PyCFunction)vm_get_memory_write, METH_VARARGS,
 	 "X"},
+	{"reset_memory_access",(PyCFunction)vm_reset_memory_access, METH_VARARGS,
+	 "X"},
 
 	{NULL}  /* Sentinel */
 };