1 files changed, 8 insertions, 5 deletions

diff --git a/miasm/jitter/arch/JitCore_mep.c b/miasm/jitter/arch/JitCore_mep.c
index a572f3e9..3c98f2dc 100644
--- a/miasm/jitter/arch/JitCore_mep.c
+++ b/miasm/jitter/arch/JitCore_mep.c
@@ -248,7 +248,7 @@ PyObject* cpu_get_exception(JitCpu* self, PyObject* args)
     return PyLong_FromUnsignedLongLong((uint64_t)(((struct vm_cpu*)self->cpu)->exception_flags));
 }
 
-void check_automod(JitCpu* jitcpu, uint64_t addr, uint64_t size)
+void check_automod(JitCpu* jitcpu, uint64_t addr, size_t size)
 {
     PyObject *result;
 
@@ -291,7 +291,7 @@ PyObject* vm_set_mem(JitCpu *self, PyObject* args)
        Py_ssize_t py_length;
 
        char * buffer;
-       uint64_t size;
+       Py_ssize_t pysize;
        uint64_t addr;
        int ret = 0x1337;
 
@@ -303,13 +303,16 @@ PyObject* vm_set_mem(JitCpu *self, PyObject* args)
        if(!PyBytes_Check(py_buffer))
 	   RAISE(PyExc_TypeError,"arg must be bytes");
 
-       size = PyBytes_Size(py_buffer);
+       pysize = PyBytes_Size(py_buffer);
+       if (pysize < 0) {
+	       RAISE(PyExc_TypeError,"Python error");
+       }
        PyBytes_AsStringAndSize(py_buffer, &buffer, &py_length);
 
-       ret = vm_write_mem(&(((VmMngr*)self->pyvm)->vm_mngr), addr, buffer, size);
+       ret = vm_write_mem(&(((VmMngr*)self->pyvm)->vm_mngr), addr, buffer, pysize);
        if (ret < 0)
 	   RAISE(PyExc_TypeError,"arg must be str");
-       check_automod(self, addr, size*8);
+       check_automod(self, addr, (size_t)pysize);
 
        Py_INCREF(Py_None);
        return Py_None;