Updated the interface for flat formats.

2 files changed, 65 insertions, 47 deletions

diff --git a/plugins/pychrysalide/format/flat.c b/plugins/pychrysalide/format/flat.c
index 39ef1f6..2c8e9fd 100644
--- a/plugins/pychrysalide/format/flat.c
+++ b/plugins/pychrysalide/format/flat.c
@@ -41,9 +41,6 @@
 /* Crée un nouvel objet Python de type 'FlatFormat'. */
 static PyObject *py_flat_format_new(PyTypeObject *, PyObject *, PyObject *);
 
-/* Indique le type d'architecture visée par le format. */
-static PyObject *py_flat_format_set_target_machine(PyObject *, PyObject *);
-
 
 
 /******************************************************************************
@@ -64,13 +61,27 @@ static PyObject *py_flat_format_new(PyTypeObject *type, PyObject *args, PyObject
 {
     PyObject *result;                       /* Instance à retourner        */
     GBinContent *content;                   /* Instance GLib du contenu    */
+    const char *machine;                    /* Identifiant d'architecture  */
+    unsigned int endian;                    /* Boutisme de l'architecture  */
     int ret;                                /* Bilan de lecture des args.  */
     GExeFormat *format;                     /* Création GLib à transmettre */
 
-    ret = PyArg_ParseTuple(args, "O&", convert_to_binary_content, &content);
+#define FLAT_FORMAT_DOC                                                     \
+    "FlatFormat is suitable for all executable contents without a proper"   \
+    " file format, such as shellcodes ou eBPF programs.\n"                  \
+    "\n"                                                                    \
+    "Instances can be created using the following constructor:\n"           \
+    "\n"                                                                    \
+    "    FlatFormat(content, machine, endian)"                              \
+    "\n"                                                                    \
+    "Where content is a pychrysalide.analysis.BinContent object, machine"   \
+    " defines the target architecture and endian provides the right"        \
+    " endianness of the data."
+
+    ret = PyArg_ParseTuple(args, "O&sI", convert_to_binary_content, &content, &machine, &endian);
     if (!ret) return NULL;
 
-    format = g_flat_format_new(content);
+    format = g_flat_format_new(content, machine, endian);
 
     if (format == NULL)
     {
@@ -91,42 +102,6 @@ static PyObject *py_flat_format_new(PyTypeObject *type, PyObject *args, PyObject
 
 /******************************************************************************
 *                                                                             *
-*  Paramètres  : self = description de l'exécutable à consulter.              *
-*                args = arguments accompagnant l'appel.                       *
-*                                                                             *
-*  Description : Indique le type d'architecture visée par le format.          *
-*                                                                             *
-*  Retour      : Rien, donc None en Python.                                   *
-*                                                                             *
-*  Remarques   : -                                                            *
-*                                                                             *
-******************************************************************************/
-
-static PyObject *py_flat_format_set_target_machine(PyObject *self, PyObject *args)
-{
-    PyObject *result;                       /* Instance à retourner        */
-    GFlatFormat *format;                    /* Version GLib du format      */
-    const char *machine;                    /* Identifiant d'architecture  */
-    int ret;                                /* Bilan de lecture des args.  */
-
-    format = G_FLAT_FORMAT(pygobject_get(self));
-    assert(format != NULL);
-
-    ret = PyArg_ParseTuple(args, "s", &machine);
-    if (!ret) return NULL;
-
-    g_flat_format_set_target_machine(format, machine);
-
-    result = Py_None;
-    Py_INCREF(result);
-
-    return result;
-
-}
-
-
-/******************************************************************************
-*                                                                             *
 *  Paramètres  : -                                                            *
 *                                                                             *
 *  Description : Fournit un accès à une définition de type à diffuser.        *
@@ -140,11 +115,6 @@ static PyObject *py_flat_format_set_target_machine(PyObject *self, PyObject *arg
 PyTypeObject *get_python_flat_format_type(void)
 {
     static PyMethodDef py_flat_format_methods[] = {
-        {
-            "set_machine", py_flat_format_set_target_machine,
-            METH_VARARGS,
-            "set_machine($self, machine, /)\n--\n\nSet the architecture linked to the flat format."
-        },
         { NULL }
     };
 
@@ -161,7 +131,7 @@ PyTypeObject *get_python_flat_format_type(void)
 
         .tp_flags       = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
 
-        .tp_doc         = "PyChrysalide flat format",
+        .tp_doc         = FLAT_FORMAT_DOC,
 
         .tp_methods     = py_flat_format_methods,
         .tp_getset      = py_flat_format_getseters,
@@ -211,3 +181,48 @@ bool ensure_python_flat_format_is_registered(void)
     return true;
 
 }
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : arg = argument quelconque à tenter de convertir.             *
+*                dst = destination des valeurs récupérées en cas de succès.   *
+*                                                                             *
+*  Description : Tente de convertir en format à plat.                         *
+*                                                                             *
+*  Retour      : Bilan de l'opération, voire indications supplémentaires.     *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+int convert_to_flat_format(PyObject *arg, void *dst)
+{
+    int result;                             /* Bilan à retourner           */
+
+    result = PyObject_IsInstance(arg, (PyObject *)get_python_flat_format_type());
+
+    switch (result)
+    {
+        case -1:
+            /* L'exception est déjà fixée par Python */
+            result = 0;
+            break;
+
+        case 0:
+            PyErr_SetString(PyExc_TypeError, "unable to convert the provided argument to flat format");
+            break;
+
+        case 1:
+            *((GFlatFormat **)dst) = G_FLAT_FORMAT(pygobject_get(arg));
+            break;
+
+        default:
+            assert(false);
+            break;
+
+    }
+
+    return result;
+
+}
diff --git a/plugins/pychrysalide/format/flat.h b/plugins/pychrysalide/format/flat.h
index d7447dd..e3fd65a 100644
--- a/plugins/pychrysalide/format/flat.h
+++ b/plugins/pychrysalide/format/flat.h
@@ -37,6 +37,9 @@ PyTypeObject *get_python_flat_format_type(void);
 /* Prend en charge l'objet 'pychrysalide.format.FlatFormat'. */
 bool ensure_python_flat_format_is_registered(void);
 
+/* Tente de convertir en format à plat. */
+int convert_to_flat_format(PyObject *, void *);
+
 
 
 #endif  /* _PLUGINS_PYCHRYSALIDE_FORMAT_FLAT_H */