Provided access to Elf structures from Python.

4 files changed, 208 insertions, 1 deletions

diff --git a/plugins/pychrysa/Makefile.am b/plugins/pychrysa/Makefile.am
index e8c2bed..0648bb3 100644
--- a/plugins/pychrysa/Makefile.am
+++ b/plugins/pychrysa/Makefile.am
@@ -8,6 +8,7 @@ pychrysalide_la_SOURCES =				\
 	helpers.h helpers.c					\
 	plugin.h plugin.c					\
 	pychrysa.h pychrysa.c				\
+	struct.h struct.c					\
 	weak.h weak.c
 
 pychrysalide_la_LIBADD =				\
diff --git a/plugins/pychrysa/pychrysa.c b/plugins/pychrysa/pychrysa.c
index 7629132..35d14f4 100644
--- a/plugins/pychrysa/pychrysa.c
+++ b/plugins/pychrysa/pychrysa.c
@@ -45,6 +45,7 @@
 
 #include "helpers.h"
 #include "plugin.h"
+#include "struct.h"
 #include "analysis/module.h"
 #include "arch/module.h"
 #include "common/module.h"
@@ -386,8 +387,10 @@ PyMODINIT_FUNC PyInit_pychrysalide(void)
 
     result = PyModule_Create(&py_chrysalide_module);
 
+    status = register_python_py_struct(result);
+
     /* Interface 'LineGenerator' en premier... */
-    status = add_glibext_module_to_python_module(result);
+    if (status) status = add_glibext_module_to_python_module(result);
 
     /* BinRoutine hérite de BinSymbol... */
     if (status) status = add_format_module_to_python_module(result);
diff --git a/plugins/pychrysa/struct.c b/plugins/pychrysa/struct.c
new file mode 100644
index 0000000..33b58b8
--- /dev/null
+++ b/plugins/pychrysa/struct.c
@@ -0,0 +1,161 @@
+
+/* Chrysalide - Outil d'analyse de fichiers binaires
+ * struct.c - prototypes pour la conversion de structures C en équivalent Python
+ *
+ * Copyright (C) 2017 Cyrille Bagard
+ *
+ *  This file is part of Chrysalide.
+ *
+ *  Chrysalide is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  Chrysalide is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+
+#include "struct.h"
+
+
+
+/* Objet à vocation abstraite */
+typedef struct _PyStructObject
+{
+    PyDictObject base;                      /* A laisser en premier        */
+
+} PyStructObject;
+
+
+/* Assure l'encadrement des accès aux champs d'une structure. */
+static PyObject *py_struct_getattr(PyObject *, char *);
+
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : self = structure C convertie en Python.                      *
+*                name = nom du champ auquel un accès est demandé.             *
+*                                                                             *
+*  Description : Assure l'encadrement des accès aux champs d'une structure.   *
+*                                                                             *
+*  Retour      : Valeur du champ demandé.                                     *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static PyObject *py_struct_getattr(PyObject *self, char *name)
+{
+    PyObject *result;                       /* Elément à retourner         */
+    PyObject *w;                            /* Conversion du nom de champ  */
+    PyTypeObject *tp;                       /* Type de l'objet manipulé    */
+
+    result = PyDict_GetItemString(self, name);
+
+    if (result != NULL)
+        Py_INCREF(result);
+
+    else
+    {
+        w = PyUnicode_InternFromString(name);
+        if (w == NULL) return NULL;
+
+        tp = Py_TYPE(self);
+
+        if (tp->tp_base->tp_getattro != NULL)
+            result = tp->tp_base->tp_getattro(self, w);
+
+        Py_DECREF(w);
+
+    }
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : -                                                            *
+*                                                                             *
+*  Description : Fournit un accès à une définition de type à diffuser.        *
+*                                                                             *
+*  Retour      : Définition d'objet pour Python.                              *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+PyTypeObject *get_python_py_struct_type(void)
+{
+    static PyMethodDef py_struct_methods[] = {
+        { NULL }
+    };
+
+    static PyGetSetDef py_struct_getseters[] = {
+        { NULL }
+    };
+
+    static PyTypeObject py_struct_type = {
+
+        PyVarObject_HEAD_INIT(NULL, 0)
+
+        .tp_name        = "pychrysalide.PyStructObject",
+        .tp_basicsize   = sizeof(PyStructObject),
+
+        .tp_getattr     = py_struct_getattr,
+
+        .tp_flags       = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
+
+        .tp_doc         = "PyChrysalide structure",
+
+        .tp_methods     = py_struct_methods,
+        .tp_getset      = py_struct_getseters,
+        .tp_base        = &PyDict_Type,
+
+    };
+
+    return &py_struct_type;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : module = module dont la définition est à compléter.          *
+*                                                                             *
+*  Description : Prend en charge l'objet 'pychrysalide.PyStructObject'.       *
+*                                                                             *
+*  Retour      : Bilan de l'opération.                                        *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+bool register_python_py_struct(PyObject *module)
+{
+    PyTypeObject *py_struct_type;           /* Type Python 'PyStructObject'*/
+    int ret;                                /* Bilan des préparatifs       */
+
+    py_struct_type = get_python_py_struct_type();
+
+    ret = PyType_Ready(py_struct_type);
+
+    if (ret != 0)
+        return false;
+
+    Py_INCREF(py_struct_type);
+
+    PyModule_AddObject(module, "PyStructObject", (PyObject *)py_struct_type);;
+
+    return true;
+
+}
diff --git a/plugins/pychrysa/struct.h b/plugins/pychrysa/struct.h
new file mode 100644
index 0000000..cca00b8
--- /dev/null
+++ b/plugins/pychrysa/struct.h
@@ -0,0 +1,42 @@
+
+/* Chrysalide - Outil d'analyse de fichiers binaires
+ * struct.h - prototypes pour la conversion de structures C en équivalent Python
+ *
+ * Copyright (C) 2017 Cyrille Bagard
+ *
+ *  This file is part of Chrysalide.
+ *
+ *  Chrysalide is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  Chrysalide is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+
+#ifndef _PLUGINS_PYCHRYSA_STRUCT_H
+#define _PLUGINS_PYCHRYSA_STRUCT_H
+
+
+#include <Python.h>
+#include <stdbool.h>
+
+
+
+/* Fournit un accès à une définition de type à diffuser. */
+PyTypeObject *get_python_py_struct_type(void);
+
+/* Prend en charge l'objet 'pychrysalide.PyStructObject'. */
+bool register_python_py_struct(PyObject *);
+
+
+
+#endif  /* _PLUGINS_PYCHRYSA_STRUCT_H */