Restore and improve the support for PE files.

1 files changed, 133 insertions, 71 deletions

diff --git a/plugins/pe/python/format.c b/plugins/pe/python/format.c
index 4bbb99a..6c97c7d 100644
--- a/plugins/pe/python/format.c
+++ b/plugins/pe/python/format.c
@@ -28,25 +28,23 @@
 #include <pygobject.h>
 
 
-#include <format/known.h>
 #include <plugins/dt.h>
 #include <plugins/pychrysalide/helpers.h>
 #include <plugins/pychrysalide/analysis/content.h>
 #include <plugins/pychrysalide/format/executable.h>
 
 
-#include "constants.h"
+//#include "constants.h"
 #include "translate.h"
-#include "../format.h"
-#include "../rich.h"
+#include "../pe-int.h"
+//#include "../rich.h"
 
 
 
 /* ------------------------ GLUE POUR CREATION DEPUIS PYTHON ------------------------ */
 
 
-/* Accompagne la création d'une instance dérivée en Python. */
-static PyObject *py_pe_format_new(PyTypeObject *, PyObject *, PyObject *);
+CREATE_DYN_CONSTRUCTOR(pe_format, G_TYPE_PE_FORMAT);
 
 /* Initialise une instance sur la base du dérivé de GObject. */
 static int py_pe_format_init(PyObject *, PyObject *, PyObject *);
@@ -59,9 +57,15 @@ static int py_pe_format_init(PyObject *, PyObject *, PyObject *);
 /* Présente l'en-tête MS-DOS du format chargé. */
 static PyObject *py_pe_format_get_dos_header(PyObject *, void *);
 
+/* Présente l'en-tête NT du format chargé. */
+static PyObject *py_pe_format_get_nt_headers(PyObject *, void *);
+
 /* Offre un raccourci vers les répertoires du format PE. */
 static PyObject *py_pe_format_get_directories(PyObject *, void *);
 
+/* Offre un raccourci vers les sections du format PE. */
+static PyObject *py_pe_format_get_sections(PyObject *, void *);
+
 /* Présente l'en-tête enrichi du format chargé. */
 static PyObject *py_pe_format_get_rich_header(PyObject *, void *);
 
@@ -80,66 +84,6 @@ static PyObject *py_pe_format_get_comp_ids(PyObject *, void *);
 
 /******************************************************************************
 *                                                                             *
-*  Paramètres  : type = type du nouvel objet à mettre en place.               *
-*                args = éventuelle liste d'arguments.                         *
-*                kwds = éventuel dictionnaire de valeurs mises à disposition. *
-*                                                                             *
-*  Description : Accompagne la création d'une instance dérivée en Python.     *
-*                                                                             *
-*  Retour      : Nouvel objet Python mis en place ou NULL en cas d'échec.     *
-*                                                                             *
-*  Remarques   : -                                                            *
-*                                                                             *
-******************************************************************************/
-
-static PyObject *py_pe_format_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
-{
-    PyObject *result;                       /* Objet à retourner           */
-    PyTypeObject *base;                     /* Type de base à dériver      */
-    bool first_time;                        /* Evite les multiples passages*/
-    GType gtype;                            /* Nouveau type de processeur  */
-    bool status;                            /* Bilan d'un enregistrement   */
-
-    /* Validations diverses */
-
-    base = get_python_pe_format_type();
-
-    if (type == base)
-        goto simple_way;
-
-    /* Mise en place d'un type dédié */
-
-    first_time = (g_type_from_name(type->tp_name) == 0);
-
-    gtype = build_dynamic_type(G_TYPE_PE_FORMAT, type->tp_name, NULL, NULL, NULL);
-
-    if (first_time)
-    {
-        status = register_class_for_dynamic_pygobject(gtype, type);
-
-        if (!status)
-        {
-            result = NULL;
-            goto exit;
-        }
-
-    }
-
-    /* On crée, et on laisse ensuite la main à PyGObject_Type.tp_init() */
-
- simple_way:
-
-    result = PyType_GenericNew(type, args, kwds);
-
- exit:
-
-    return result;
-
-}
-
-
-/******************************************************************************
-*                                                                             *
 *  Paramètres  : self = objet à initialiser (théoriquement).                  *
 *                args = arguments fournis à l'appel.                          *
 *                kwds = arguments de type key=val fournis.                    *
@@ -183,7 +127,8 @@ static int py_pe_format_init(PyObject *self, PyObject *args, PyObject *kwds)
 
     format = G_PE_FORMAT(pygobject_get(self));
 
-    g_known_format_set_content(G_KNOWN_FORMAT(format), content);
+    if (!g_pe_format_create(format, content))
+        return -1;
 
     return 0;
 
@@ -223,7 +168,6 @@ static PyObject *py_pe_format_get_dos_header(PyObject *self, void *closure)
     "\n"                                                        \
     "The provided information is composed of the following"     \
     " properties :\n"                                           \
-    "\n"                                                        \
     "* e_magic;\n"                                              \
     "* e_cblp;\n"                                               \
     "* e_cp;\n"                                                 \
@@ -259,6 +203,52 @@ static PyObject *py_pe_format_get_dos_header(PyObject *self, void *closure)
 *  Paramètres  : self    = format en place à consulter.                       *
 *                closure = non utilisé ici.                                   *
 *                                                                             *
+*  Description : Présente l'en-tête NT du format chargé.                      *
+*                                                                             *
+*  Retour      : Structure Python créée pour l'occasion.                      *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static PyObject *py_pe_format_get_nt_headers(PyObject *self, void *closure)
+{
+    PyObject *result;                       /* Trouvaille à retourner      */
+    GPeFormat *format;                      /* Version GLib du format      */
+
+#define PE_FORMAT_NT_HEADERS_ATTRIB PYTHON_GET_DEF_FULL         \
+(                                                               \
+    nt_headers, py_pe_format,                                   \
+    "NT headers of the file format.\n"                          \
+    "\n"                                                        \
+    "This property is a pychrysalide.StructObject instance."    \
+    "\n"                                                        \
+    "The provided information is composed of the following"     \
+    " properties :\n"                                           \
+    "* signature;\n"                                            \
+    "* file_header;\n"                                          \
+    "* optional_header.\n"                                      \
+    "\n"                                                        \
+    "The last two fields are pychrysalide.StructObject"         \
+    " which contain more fields. These fields can be"           \
+    " enumerated with the keys() method (for instance:"         \
+    " *mype.nt_headers.file_header.keys()*).\n"                 \
+)
+
+    format = G_PE_FORMAT(pygobject_get(self));
+
+    result = translate_pe_nt_headers_to_python(format, g_pe_format_get_nt_headers(format));
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : self    = format en place à consulter.                       *
+*                closure = non utilisé ici.                                   *
+*                                                                             *
 *  Description : Offre un raccourci vers les répertoires du format PE.        *
 *                                                                             *
 *  Retour      : Structure Python créée pour l'occasion.                      *
@@ -272,7 +262,7 @@ static PyObject *py_pe_format_get_directories(PyObject *self, void *closure)
     PyObject *result;                       /* Trouvaille à retourner      */
     GPeFormat *format;                      /* Version GLib du format      */
     size_t count;                           /* Quantité de répertoires     */
-    const image_data_directory *directories; /* Répertoires à exporter     */
+    const image_data_directory_t *directories; /* Répertoires à exporter   */
     size_t i;                               /* Boucle de parcours          */
     PyObject *item;                         /* Elément de tableau          */
     int ret;                                /* Bilan d'une mise en place   */
@@ -323,6 +313,76 @@ static PyObject *py_pe_format_get_directories(PyObject *self, void *closure)
 *  Paramètres  : self    = format en place à consulter.                       *
 *                closure = non utilisé ici.                                   *
 *                                                                             *
+*  Description : Offre un raccourci vers les sections du format PE.           *
+*                                                                             *
+*  Retour      : Structure Python créée pour l'occasion.                      *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static PyObject *py_pe_format_get_sections(PyObject *self, void *closure)
+{
+    PyObject *result;                       /* Trouvaille à retourner      */
+    GPeFormat *format;                      /* Version GLib du format      */
+    size_t count;                           /* Quantité de répertoires     */
+    const image_section_header_t *sections; /* Sections à exporter         */
+    size_t i;                               /* Boucle de parcours          */
+    PyObject *item;                         /* Elément de tableau          */
+    int ret;                                /* Bilan d'une mise en place   */
+
+#define PE_FORMAT_SECTIONS_ATTRIB PYTHON_GET_DEF_FULL               \
+(                                                                   \
+    sections, py_pe_format,                                         \
+    "Shortcut to the definitions of all PE format sections.\n"      \
+    "\n"                                                            \
+    "This property is a pychrysalide.StructObject instance.\n"      \
+    "\n"                                                            \
+    "Each returned item is composed of the following properties :\n"\
+    "\n"                                                            \
+    "* name;\n"                                                     \
+    "* misc.virtual_size;\n"                                        \
+    "* virtual_address;\n"			                                \
+    "* size_of_raw_data;\n"			                                \
+    "* pointer_to_raw_data;\n"			                            \
+    "* pointer_to_relocations;\n"			                        \
+    "* pointer_to_line_numbers;\n"			                        \
+    "* number_of_relocations;\n"			                        \
+    "* number_of_line_numbers;\n"			                        \
+    "* characteristics."			                                \
+)
+
+    format = G_PE_FORMAT(pygobject_get(self));
+
+    sections = g_pe_format_get_sections(format, &count);
+
+    result = PyTuple_New(count);
+
+    for (i = 0; i < count; i++)
+    {
+        item = translate_pe_section_header_to_python(format, sections + i);
+
+        ret = PyTuple_SetItem(result, i, item);
+
+        if (ret != 0)
+        {
+            Py_DECREF(result);
+            result = NULL;
+            break;
+        }
+
+    }
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : self    = format en place à consulter.                       *
+*                closure = non utilisé ici.                                   *
+*                                                                             *
 *  Description : Présente l'en-tête enrichi du format chargé.                 *
 *                                                                             *
 *  Retour      : Tableau de valeurs brutes d'information.                     *
@@ -487,7 +547,9 @@ PyTypeObject *get_python_pe_format_type(void)
 
     static PyGetSetDef py_pe_format_getseters[] = {
         PE_FORMAT_DOS_HEADER_ATTRIB,
+        PE_FORMAT_NT_HEADERS_ATTRIB,
         PE_FORMAT_DIRECTORIES_ATTRIB,
+        PE_FORMAT_SECTIONS_ATTRIB,
         PE_FORMAT_RICH_HEADER_ATTRIB,
         PE_FORMAT_RICH_HEADER_CHECKSUM_ATTRIB,
         PE_FORMAT_COMP_IDS_ATTRIB,
@@ -545,8 +607,8 @@ bool register_python_pe_format(PyObject *module)
     if (!register_class_for_pygobject(dict, G_TYPE_PE_FORMAT, type))
         return false;
 
-    if (!define_python_pe_format_constants(type))
-        return false;
+    //if (!define_python_pe_format_constants(type))
+    //    return false;
 
     return true;