fix 16 bit bug

1 files changed, 42 insertions, 0 deletions

diff --git a/miasm/tools/emul_lib/libcodenat_interface.c b/miasm/tools/emul_lib/libcodenat_interface.c
index d43401dd..7348ff00 100644
--- a/miasm/tools/emul_lib/libcodenat_interface.c
+++ b/miasm/tools/emul_lib/libcodenat_interface.c
@@ -582,6 +582,26 @@ PyObject* _vm_pop_uint32_t(void)
     return PyInt_FromLong((long)val);;
 }
 
+PyObject* _vm_push_uint16_t(int val)
+{
+    vmcpu.esp-=2;
+    MEM_WRITE(16, vmcpu.esp, val);
+
+    return PyInt_FromLong((long)vmcpu.esp);
+}
+
+
+PyObject* _vm_pop_uint16_t(void)
+{
+    unsigned int val;
+
+    val = MEM_LOOKUP(16, vmcpu.esp);
+    vmcpu.esp+=2;
+
+    return PyInt_FromLong((long)val);;
+}
+
+
 PyObject* _vm_set_mem(PyObject *addr, PyObject *item_str)
 {
     unsigned int buf_size;
@@ -770,6 +790,24 @@ PyObject* vm_pop_uint32_t(PyObject* self, PyObject* args)
     return p;
 }
 
+PyObject* vm_push_uint16_t(PyObject* self, PyObject *args)
+{
+    PyObject* p;
+    int item;
+    if (!PyArg_ParseTuple(args, "I", &item))
+	    return NULL;
+    p = _vm_push_uint16_t(item);
+    return p;
+}
+
+
+PyObject* vm_pop_uint16_t(PyObject* self, PyObject* args)
+{
+    PyObject* p;
+    p = _vm_pop_uint16_t();
+    return p;
+}
+
 PyObject* vm_set_mem(PyObject* self, PyObject* args)
 {
     PyObject* p;
@@ -1248,6 +1286,10 @@ static PyMethodDef CodenatMethods[] = {
      "x"},
     {"vm_pop_uint32_t",vm_pop_uint32_t, METH_VARARGS,
      "X"},
+    {"vm_push_uint16_t", vm_push_uint16_t, METH_VARARGS,
+     "x"},
+    {"vm_pop_uint16_t",vm_pop_uint16_t, METH_VARARGS,
+     "X"},
     {"vm_get_gpreg", vm_get_gpreg, METH_VARARGS,
      "X"},
     {"vm_set_gpreg",vm_set_gpreg, METH_VARARGS,