vm_mngr_py.c File Reference

#include <Python.h>
#include "structmember.h"
#include <stdint.h>
#include <inttypes.h>
#include <signal.h>
#include "queue.h"
#include "vm_mngr.h"
#include "vm_mngr_py.h"

Include dependency graph for vm_mngr_py.c:

Go to the source code of this file.

Macros
#define	MIN(a, b)   (((a)<(b))?(a):(b))
#define	MAX(a, b)   (((a)>(b))?(a):(b))
#define	RAISE(errtype, msg)   {PyObject* p; p = PyErr_Format( errtype, msg ); return p;}
#define	PyGetInt(item, value)

Functions
PyObject *	_vm_get_exception (unsigned int xcpt)
PyObject *	vm_is_mem_mapped (VmMngr *self, PyObject *item)
PyObject *	vm_get_mem_base_addr (VmMngr *self, PyObject *item)
static void	sig_alarm (int signo)
PyObject *	set_alarm (VmMngr *self)
PyObject *	vm_add_memory_page (VmMngr *self, PyObject *args)
PyObject *	vm_set_mem_access (VmMngr *self, PyObject *args)
PyObject *	vm_set_mem (VmMngr *self, PyObject *args)
PyObject *	vm_get_mem (VmMngr *self, PyObject *args)
PyObject *	vm_add_memory_breakpoint (VmMngr *self, PyObject *args)
PyObject *	vm_remove_memory_breakpoint (VmMngr *self, PyObject *args)
PyObject *	vm_set_exception (VmMngr *self, PyObject *args)
PyObject *	vm_get_exception (VmMngr *self, PyObject *args)
PyObject *	vm_init_memory_page_pool (VmMngr *self, PyObject *args)
PyObject *	vm_init_code_bloc_pool (VmMngr *self, PyObject *args)
PyObject *	vm_init_memory_breakpoint (VmMngr *self, PyObject *args)
PyObject *	vm_reset_memory_breakpoint (VmMngr *self, PyObject *args)
PyObject *	vm_dump (PyObject *self)
PyObject *	vm_dump_memory_breakpoint (VmMngr *self, PyObject *args)
PyObject *	vm_get_all_memory (VmMngr *self, PyObject *args)
PyObject *	vm_reset_memory_page_pool (VmMngr *self, PyObject *args)
PyObject *	vm_reset_code_bloc_pool (VmMngr *self, PyObject *args)
PyObject *	vm_add_code_bloc (VmMngr *self, PyObject *args)
PyObject *	vm_dump_code_bloc_pool (VmMngr *self)
PyObject *	vm_set_addr2obj (VmMngr *self, PyObject *args)
static PyObject *	vm_set_big_endian (VmMngr *self, PyObject *value, void *closure)
static PyObject *	vm_set_little_endian (VmMngr *self, PyObject *value, void *closure)
static void	VmMngr_dealloc (VmMngr *self)
static PyObject *	VmMngr_new (PyTypeObject *type, PyObject *args, PyObject *kwds)
static PyObject *	VmMngr_get_vmmngr (VmMngr *self, void *closure)
static int	VmMngr_set_vmmngr (VmMngr *self, PyObject *value, void *closure)
static int	VmMngr_init (VmMngr *self, PyObject *args, PyObject *kwds)
PyMODINIT_FUNC	initVmMngr (void)

Variables
struct memory_page_list_head	memory_page_pool
struct code_bloc_list_head	code_bloc_pool
VmMngr *	global_vmmngr
static PyMemberDef	VmMngr_members []
static PyMethodDef	VmMngr_methods []
static PyGetSetDef	VmMngr_getseters []
static PyTypeObject	VmMngrType
static PyMethodDef	VmMngr_Methods []
static PyObject *	Vm_Mngr_Error

Macro Definition Documentation

#define MAX	(		a,
			b
	)		(((a)>(b))?(a):(b))

Definition at line 28 of file vm_mngr_py.c.

#define MIN	(		a,
			b
	)		(((a)<(b))?(a):(b))

Definition at line 27 of file vm_mngr_py.c.

#define PyGetInt	(		item,
			value
	)

Value:

if (PyInt_Check(item)){                     \
        value = (uint64_t)PyInt_AsLong(item);           \
    }                               \
    else if (PyLong_Check(item)){                   \
        value = (uint64_t)PyLong_AsUnsignedLongLong(item);  \
    }                               \
    else{                               \
		RAISE(PyExc_TypeError,"arg must be int");     \
    }                               \

Definition at line 58 of file vm_mngr_py.c.

#define RAISE	(		errtype,
			msg
	)		{PyObject* p; p = PyErr_Format( errtype, msg ); return p;}

Definition at line 33 of file vm_mngr_py.c.

Function Documentation

PyObject* _vm_get_exception

(

unsigned int

xcpt

)

Definition at line 40 of file vm_mngr_py.c.

{
    PyObject*p;

    if (!xcpt)
        p = NULL;
    else if (xcpt & EXCEPT_CODE_AUTOMOD)
        p = PyErr_Format( PyExc_RuntimeError, "EXCEPT_CODE_AUTOMOD" );
    else if (xcpt & EXCEPT_UNK_EIP)
        p = PyErr_Format( PyExc_RuntimeError, "EXCEPT_UNK_EIP" );
    else if (xcpt & EXCEPT_UNK_MEM_AD)
        p = PyErr_Format( PyExc_RuntimeError, "EXCEPT_UNK_MEM_AD" );

    else  p = PyErr_Format( PyExc_RuntimeError, "EXCEPT_UNKNOWN" );
    return p;
}

PyMODINIT_FUNC initVmMngr

(

void

)

Definition at line 664 of file vm_mngr_py.c.

{
    PyObject *m;

    if (PyType_Ready(&VmMngrType) < 0)
    return;

    m = Py_InitModule("VmMngr", VmMngr_Methods);
    if (m == NULL)
        return;

    Vm_Mngr_Error = PyErr_NewException("VmMngr.error", NULL, NULL);
    Py_INCREF(Vm_Mngr_Error);
    PyModule_AddObject(m, "error", Vm_Mngr_Error);

    Py_INCREF(&VmMngrType);
    PyModule_AddObject(m, "Vm", (PyObject *)&VmMngrType);
}

PyObject* set_alarm

(

VmMngr *

self

)

Definition at line 113 of file vm_mngr_py.c.

{
    global_vmmngr = self;
    signal(SIGALRM, sig_alarm);
    return PyLong_FromUnsignedLongLong((uint64_t)0);
}

Here is the call graph for this function:

static void sig_alarm ( int  signo )

static

Definition at line 107 of file vm_mngr_py.c.

{
    global_vmmngr->vm_mngr.exception_flags |= BREAK_SIGALARM;
    return;
}

Here is the caller graph for this function:

PyObject* vm_add_code_bloc	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 427 of file vm_mngr_py.c.

{
    PyObject *item1;
    PyObject *item2;
    uint64_t ret = 0x1337beef;
    uint64_t ad_start, ad_stop, ad_code = 0;

    struct code_bloc_node * cbp;

    if (!PyArg_ParseTuple(args, "OO", &item1, &item2))
        return NULL;

    PyGetInt(item1, ad_start);
    PyGetInt(item2, ad_stop);

    cbp = create_code_bloc_node(ad_start, ad_stop);
    cbp->ad_start = ad_start;
    cbp->ad_stop = ad_stop;
    cbp->ad_code = ad_code;
    add_code_bloc(&self->vm_mngr, cbp);
    return PyLong_FromUnsignedLongLong((uint64_t)ret);
}

Here is the call graph for this function:

PyObject* vm_add_memory_breakpoint	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 260 of file vm_mngr_py.c.

{
    PyObject *ad;
    PyObject *size;
    PyObject *access;

    uint64_t b_ad;
    uint64_t b_size;
    uint64_t b_access;

    if (!PyArg_ParseTuple(args, "OOO", &ad, &size, &access))
        return NULL;

    PyGetInt(ad, b_ad);
    PyGetInt(size, b_size);
    PyGetInt(access, b_access);

    add_memory_breakpoint(&self->vm_mngr, b_ad, b_size, b_access);
    Py_INCREF(Py_None);
    return Py_None;
}

Here is the call graph for this function:

PyObject* vm_add_memory_page	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 122 of file vm_mngr_py.c.

{
    PyObject *addr;
    PyObject *access;
    PyObject *item_str;
    uint64_t buf_size;
    char* buf_data;
    Py_ssize_t length;
    uint64_t ret = 0x1337beef;
    uint64_t page_addr;
    uint64_t page_access;

    struct memory_page_node * mpn;

    if (!PyArg_ParseTuple(args, "OOO", &addr, &access, &item_str))
        return NULL;

    PyGetInt(addr, page_addr);
    PyGetInt(access, page_access);

    if(!PyString_Check(item_str))
        RAISE(PyExc_TypeError,"arg must be str");

    buf_size = PyString_Size(item_str);
    PyString_AsStringAndSize(item_str, &buf_data, &length);

    /*
    fprintf(stderr, "add page %"PRIX64" %"PRIX64" %"PRIX64"\n",
        page_addr, buf_size, page_access);
    */
    mpn = create_memory_page_node(page_addr, buf_size, page_access);
    if (mpn == NULL)
        RAISE(PyExc_TypeError,"cannot create page");
    if (is_mpn_in_tab(&self->vm_mngr, mpn)) {
        free(mpn->ad_hp);
        free(mpn);
        RAISE(PyExc_TypeError,"known page in memory");
    }

    memcpy(mpn->ad_hp, buf_data, buf_size);
    add_memory_page(&self->vm_mngr, mpn);

    return PyLong_FromUnsignedLongLong((uint64_t)ret);

}

Here is the call graph for this function:

PyObject* vm_dump

(

PyObject *

self

)

Definition at line 356 of file vm_mngr_py.c.

{
    char* buf_final;
    PyObject* ret_obj;

    buf_final = dump(&((VmMngr* )self)->vm_mngr);
    ret_obj = PyString_FromString(buf_final);
    free(buf_final);
    return ret_obj;
}

Here is the call graph for this function:

PyObject* vm_dump_code_bloc_pool

(

VmMngr *

self

)

Definition at line 450 of file vm_mngr_py.c.

{
    dump_code_bloc_pool(&self->vm_mngr);
    Py_INCREF(Py_None);
    return Py_None;

}

Here is the call graph for this function:

PyObject* vm_dump_memory_breakpoint	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 367 of file vm_mngr_py.c.

{
    dump_memory_breakpoint_pool(&self->vm_mngr);
    Py_INCREF(Py_None);
    return Py_None;
}

Here is the call graph for this function:

PyObject* vm_get_all_memory	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 375 of file vm_mngr_py.c.

{
    PyObject *o;
    struct memory_page_node * mpn;
    PyObject *dict;
    PyObject *dict2;


    dict =  PyDict_New();

    LIST_FOREACH(mpn, &self->vm_mngr.memory_page_pool, next){

        dict2 =  PyDict_New();

        o = PyString_FromStringAndSize(mpn->ad_hp, mpn->size);
        PyDict_SetItemString(dict2, "data", o);
        Py_DECREF(o);

        o = PyInt_FromLong((long)mpn->size);
        PyDict_SetItemString(dict2, "size", o);
        Py_DECREF(o);

        o = PyInt_FromLong((long)mpn->access);
        PyDict_SetItemString(dict2, "access", o);
        Py_DECREF(o);

        o = PyInt_FromLong((long)mpn->ad);
        PyDict_SetItem(dict, o, dict2);
        Py_DECREF(o);
        Py_DECREF(dict2);
    }
    return dict;
}

PyObject* vm_get_exception	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 317 of file vm_mngr_py.c.

{
    return PyLong_FromUnsignedLongLong((uint64_t)self->vm_mngr.exception_flags);
}

PyObject* vm_get_mem	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 231 of file vm_mngr_py.c.

{
       PyObject *py_addr;
       PyObject *py_len;

       uint64_t addr;
       uint64_t size;
       PyObject *obj_out;
       char * buf_out;
       int ret;

       if (!PyArg_ParseTuple(args, "OO", &py_addr, &py_len))
          return NULL;

       PyGetInt(py_addr, addr);
       PyGetInt(py_len, size);

       ret = vm_read_mem(&self->vm_mngr, addr, &buf_out, size);
       if (ret < 0) {
          PyErr_SetString(PyExc_RuntimeError, "cannot find address");
          return NULL;
       }

       obj_out = PyString_FromStringAndSize(buf_out, size);
       free(buf_out);
       return obj_out;
}

Here is the call graph for this function:

PyObject* vm_get_mem_base_addr	(	VmMngr *	self,
		PyObject *	item
	)

Definition at line 86 of file vm_mngr_py.c.

{
    PyObject *addr;

    uint64_t page_addr;
    uint64_t addr_base;
    unsigned int ret;

    if (!PyArg_ParseTuple(item, "O", &addr))
        return NULL;

    PyGetInt(addr, page_addr);

    ret = get_mem_base_addr(&self->vm_mngr, page_addr, &addr_base);
    if (ret == 0){
        Py_INCREF(Py_None);
        return Py_None;
    }
    return PyLong_FromUnsignedLongLong((uint64_t)addr_base);
}

Here is the call graph for this function:

PyObject* vm_init_code_bloc_pool	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 332 of file vm_mngr_py.c.

{
    init_code_bloc_pool(&self->vm_mngr);
    Py_INCREF(Py_None);
    return Py_None;

}

Here is the call graph for this function:

PyObject* vm_init_memory_breakpoint	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 340 of file vm_mngr_py.c.

{
    init_memory_breakpoint(&self->vm_mngr);
    Py_INCREF(Py_None);
    return Py_None;

}

Here is the call graph for this function:

PyObject* vm_init_memory_page_pool	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 325 of file vm_mngr_py.c.

{
    init_memory_page_pool(&self->vm_mngr);
    Py_INCREF(Py_None);
    return Py_None;
}

Here is the call graph for this function:

PyObject* vm_is_mem_mapped	(	VmMngr *	self,
		PyObject *	item
	)

Definition at line 70 of file vm_mngr_py.c.

{
    PyObject *addr;
    uint64_t page_addr;
    uint32_t ret;
    if (!PyArg_ParseTuple(item, "O", &addr))
        return NULL;

    PyGetInt(addr, page_addr);

    ret = is_mem_mapped(&self->vm_mngr, page_addr);
    return PyInt_FromLong((long)ret);
}

Here is the call graph for this function:

PyObject* vm_remove_memory_breakpoint	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 283 of file vm_mngr_py.c.

{
    PyObject *ad;
    PyObject *access;
    uint64_t b_ad;
    uint64_t b_access;

    if (!PyArg_ParseTuple(args, "OO", &ad, &access))
        return NULL;

    PyGetInt(ad, b_ad);
    PyGetInt(access, b_access);
    remove_memory_breakpoint(&self->vm_mngr, b_ad, b_access);

    Py_INCREF(Py_None);
    return Py_None;
}

Here is the call graph for this function:

PyObject* vm_reset_code_bloc_pool	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 418 of file vm_mngr_py.c.

{
    reset_code_bloc_pool(&self->vm_mngr);
    Py_INCREF(Py_None);
    return Py_None;

}

Here is the call graph for this function:

Here is the caller graph for this function:

PyObject* vm_reset_memory_breakpoint	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 348 of file vm_mngr_py.c.

{
    reset_memory_breakpoint(&self->vm_mngr);
    Py_INCREF(Py_None);
    return Py_None;

}

Here is the call graph for this function:

Here is the caller graph for this function:

PyObject* vm_reset_memory_page_pool	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 410 of file vm_mngr_py.c.

{
    reset_memory_page_pool(&self->vm_mngr);
    Py_INCREF(Py_None);
    return Py_None;

}

Here is the call graph for this function:

Here is the caller graph for this function:

PyObject* vm_set_addr2obj	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 459 of file vm_mngr_py.c.

{
    PyObject* addr2obj;

    if (!PyArg_ParseTuple(args, "O", &addr2obj))
        return NULL;

    if (self->vm_mngr.addr2obj != NULL){
        Py_DECREF(self->vm_mngr.addr2obj);
    }

    Py_INCREF(addr2obj);
    self->vm_mngr.addr2obj = addr2obj;
    Py_INCREF(Py_None);
    return Py_None;
}

static PyObject* vm_set_big_endian ( VmMngr *  self, PyObject *  value, void *  closure  )

static

Definition at line 478 of file vm_mngr_py.c.

{
    self->vm_mngr.sex   = __BIG_ENDIAN;
    Py_INCREF(Py_None);
    return Py_None;
}

PyObject* vm_set_exception	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 302 of file vm_mngr_py.c.

{
    PyObject *item1;
    uint64_t i;

    if (!PyArg_ParseTuple(args, "O", &item1))
        return NULL;

    PyGetInt(item1, i);

    self->vm_mngr.exception_flags = i;
    Py_INCREF(Py_None);
    return Py_None;
}

static PyObject* vm_set_little_endian ( VmMngr *  self, PyObject *  value, void *  closure  )

static

Definition at line 486 of file vm_mngr_py.c.

{
    self->vm_mngr.sex   = __LITTLE_ENDIAN;
    Py_INCREF(Py_None);
    return Py_None;
}

PyObject* vm_set_mem	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 196 of file vm_mngr_py.c.

{
       PyObject *py_addr;
       PyObject *py_buffer;
       Py_ssize_t py_length;

       char * buffer;
       uint64_t size;
       uint64_t addr;
       int ret = 0x1337;

       if (!PyArg_ParseTuple(args, "OO", &py_addr, &py_buffer))
          return NULL;

       PyGetInt(py_addr, addr);

       if(!PyString_Check(py_buffer))
          RAISE(PyExc_TypeError,"arg must be str");

       size = PyString_Size(py_buffer);
       PyString_AsStringAndSize(py_buffer, &buffer, &py_length);

       ret = vm_write_mem(&self->vm_mngr, addr, buffer, size);
       if (ret < 0)
          RAISE(PyExc_TypeError, "Error in set_mem");

       check_write_code_bloc(&self->vm_mngr, size*8, addr);

       Py_INCREF(Py_None);
       return Py_None;
}

Here is the call graph for this function:

PyObject* vm_set_mem_access	(	VmMngr *	self,
		PyObject *	args
	)

Definition at line 170 of file vm_mngr_py.c.

{
    PyObject *addr;
    PyObject *access;

    uint64_t ret = 0x1337beef;
    uint64_t page_addr;
    uint64_t page_access;
    struct memory_page_node * mpn;

    if (!PyArg_ParseTuple(args, "OO", &addr, &access))
        return NULL;

    PyGetInt(addr, page_addr);
    PyGetInt(access, page_access);

    mpn = get_memory_page_from_address(&self->vm_mngr, page_addr);
    if (!mpn){
        PyErr_SetString(PyExc_RuntimeError, "cannot find address");
        return 0;
    }

    mpn->access = page_access;
    return PyLong_FromUnsignedLongLong((uint64_t)ret);
}

Here is the call graph for this function:

static void VmMngr_dealloc ( VmMngr *  self )

static

Definition at line 513 of file vm_mngr_py.c.

{
    vm_reset_memory_page_pool(self, NULL);
    vm_reset_code_bloc_pool(self, NULL);
    vm_reset_memory_breakpoint(self, NULL);
    self->ob_type->tp_free((PyObject*)self);
}

Here is the call graph for this function:

static PyObject* VmMngr_get_vmmngr ( VmMngr *  self, void *  closure  )

static

Definition at line 532 of file vm_mngr_py.c.

{
    return PyLong_FromUnsignedLongLong((uint64_t)&(self->vm_mngr));
}

static int VmMngr_init ( VmMngr *  self, PyObject *  args, PyObject *  kwds  )

static

Definition at line 599 of file vm_mngr_py.c.

{
    memset(&(self->vm_mngr), 0, sizeof(self->vm_mngr));
    return 0;
}

static PyObject* VmMngr_new ( PyTypeObject *  type, PyObject *  args, PyObject *  kwds  )

static

Definition at line 523 of file vm_mngr_py.c.

{
    VmMngr *self;

    self = (VmMngr *)type->tp_alloc(type, 0);
    return (PyObject *)self;
}

static int VmMngr_set_vmmngr ( VmMngr *  self, PyObject *  value, void *  closure  )

static

Definition at line 538 of file vm_mngr_py.c.

{
    PyErr_SetString(PyExc_TypeError, "immutable vmmngr");
    return -1;
}

Variable Documentation

struct code_bloc_list_head code_bloc_pool

VmMngr* global_vmmngr

Definition at line 38 of file vm_mngr_py.c.

struct memory_page_list_head memory_page_pool

PyObject* Vm_Mngr_Error

static

Definition at line 661 of file vm_mngr_py.c.

PyGetSetDef VmMngr_getseters[]

static

Initial value:

= {
    {"vmmngr",
     (getter)VmMngr_get_vmmngr, (setter)VmMngr_set_vmmngr,
     "first name",
     NULL},
    {NULL}  
}

Definition at line 605 of file vm_mngr_py.c.

PyMemberDef VmMngr_members[]

static

Initial value:

= {
    {NULL}  
}

Definition at line 544 of file vm_mngr_py.c.

PyMethodDef VmMngr_methods[]

static

Definition at line 548 of file vm_mngr_py.c.

PyMethodDef VmMngr_Methods[]

static

Initial value:

= {
    {NULL, NULL, 0, NULL}        

}

Definition at line 656 of file vm_mngr_py.c.

PyTypeObject VmMngrType

static

Definition at line 613 of file vm_mngr_py.c.