Extended the Python API.

3 files changed, 253 insertions, 60 deletions

diff --git a/plugins/pychrysalide/arch/instruction.c b/plugins/pychrysalide/arch/instruction.c
index 881c6ff..fb2df5b 100644
--- a/plugins/pychrysalide/arch/instruction.c
+++ b/plugins/pychrysalide/arch/instruction.c
@@ -35,6 +35,7 @@
 #include <plugins/dt.h>
 
 
+#include "operand.h"
 #include "vmpa.h"
 #include "../access.h"
 #include "../helpers.h"
@@ -105,6 +106,23 @@ static int py_arch_instruction_init(PyObject *, PyObject *, PyObject *);
 
 
 
+/* --------------------------- MANIPULATION DES OPERANDES --------------------------- */
+
+
+/* Attache un opérande supplémentaire à une instruction. */
+static PyObject *py_arch_instruction_attach_extra_operand(PyObject *, PyObject *);
+
+/* Fournit tous les opérandes d'une instruction. */
+static PyObject *py_arch_instruction_get_operands(PyObject *, void *);
+
+/* Remplace un opérande d'une instruction par un autre. */
+static PyObject *py_arch_instruction_replace_operand(PyObject *, PyObject *);
+
+/* Détache un opérande liée d'une instruction. */
+static PyObject *py_arch_instruction_detach_operand(PyObject *, PyObject *);
+
+
+
 /* ------------------- DEFINITION DES LIAISONS ENTRE INSTRUCTIONS ------------------- */
 
 
@@ -131,9 +149,6 @@ static int py_arch_instruction_set_range(PyObject *, PyObject *, void *);
 /* Fournit le nom humain de l'instruction manipulée. */
 static PyObject *py_arch_instruction_get_keyword(PyObject *, void *);
 
-/* Fournit tous les opérandes d'une instruction. */
-static PyObject *py_arch_instruction_get_operands(PyObject *, void *);
-
 /* Définit les constantes pour les instructions. */
 static bool py_arch_instruction_define_constants(PyTypeObject *);
 
@@ -401,6 +416,178 @@ static int py_arch_instruction_init(PyObject *self, PyObject *args, PyObject *kw
 
 
 /* ---------------------------------------------------------------------------------- */
+/*                             MANIPULATION DES OPERANDES                             */
+/* ---------------------------------------------------------------------------------- */
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : self = architecture concernée par la procédure.              *
+*                args = instruction représentant le point de départ.          *
+*                                                                             *
+*  Description : Attache un opérande supplémentaire à une instruction.        *
+*                                                                             *
+*  Retour      : None.                                                        *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static PyObject *py_arch_instruction_attach_extra_operand(PyObject *self, PyObject *args)
+{
+    GArchOperand *op;                       /* Opérande concerné à ajouter */
+    int ret;                                /* Bilan de lecture des args.  */
+    GArchInstruction *instr;                /* Instruction manipulée       */
+
+    ret = PyArg_ParseTuple(args, "O&", convert_to_arch_operand, &op);
+    if (!ret) return NULL;
+
+    instr = G_ARCH_INSTRUCTION(pygobject_get(self));
+
+    g_object_ref(G_OBJECT(op));
+
+    g_arch_instruction_attach_extra_operand(instr, op);
+
+    Py_RETURN_NONE;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : self   = objet représentant une instruction.                 *
+*                unused = adresse non utilisée ici.                           *
+*                                                                             *
+*  Description : Fournit tous les opérandes d'une instruction.                *
+*                                                                             *
+*  Retour      : Valeur associée à la propriété consultée.                    *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static PyObject *py_arch_instruction_get_operands(PyObject *self, void *unused)
+{
+    PyObject *result;                       /* Instance à retourner        */
+    GArchInstruction *instr;                /* Version native              */
+    size_t count;                           /* Nombre d'opérandes présents */
+    size_t i;                               /* Boucle de parcours          */
+    GArchOperand *operand;                  /* Opérande à manipuler        */
+    PyObject *opobj;                        /* Version Python              */
+#ifndef NDEBUG
+    int ret;                                /* Bilan d'une écriture d'arg. */
+#endif
+
+    instr = G_ARCH_INSTRUCTION(pygobject_get(self));
+
+    g_arch_instruction_lock_operands(instr);
+
+    count = _g_arch_instruction_count_operands(instr);
+
+    result = PyTuple_New(count);
+
+    for (i = 0; i < count; i++)
+    {
+        operand = _g_arch_instruction_get_operand(instr, i);
+
+        opobj = pygobject_new(G_OBJECT(operand));
+
+#ifndef NDEBUG
+        ret = PyTuple_SetItem(result, i, Py_BuildValue("O", opobj));
+        assert(ret == 0);
+#else
+        PyTuple_SetItem(result, i, Py_BuildValue("O", opobj));
+#endif
+
+        g_object_unref(G_OBJECT(operand));
+
+    }
+
+    g_arch_instruction_unlock_operands(instr);
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : self = architecture concernée par la procédure.              *
+*                args = instruction représentant le point de départ.          *
+*                                                                             *
+*  Description : Remplace un opérande d'une instruction par un autre.         *
+*                                                                             *
+*  Retour      : Bilan de l'opération.                                        *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static PyObject *py_arch_instruction_replace_operand(PyObject *self, PyObject *args)
+{
+    PyObject *result;                       /* Bilan à retourner           */
+    GArchOperand *old;                      /* Ancien opérande à remplacer */
+    GArchOperand *new;                      /* Nouvel opérande à intégrer  */
+    int ret;                                /* Bilan de lecture des args.  */
+    GArchInstruction *instr;                /* Instruction manipulée       */
+    bool status;                            /* Bilan de l'opération        */
+
+    ret = PyArg_ParseTuple(args, "O&O&", convert_to_arch_operand, &old, convert_to_arch_operand, &new);
+    if (!ret) return NULL;
+
+    instr = G_ARCH_INSTRUCTION(pygobject_get(self));
+
+    status = g_arch_instruction_replace_operand(instr, old, new);
+
+    if (status)
+        g_object_ref(G_OBJECT(new));
+
+    result = status ? Py_True : Py_False;
+    Py_INCREF(result);
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : self = architecture concernée par la procédure.              *
+*                args = instruction représentant le point de départ.          *
+*                                                                             *
+*  Description : Détache un opérande liée d'une instruction.                  *
+*                                                                             *
+*  Retour      : Bilan de l'opération.                                        *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static PyObject *py_arch_instruction_detach_operand(PyObject *self, PyObject *args)
+{
+    PyObject *result;                       /* Bilan à retourner           */
+    GArchOperand *target;                   /* Opérande ciblé par l'action */
+    int ret;                                /* Bilan de lecture des args.  */
+    GArchInstruction *instr;                /* Instruction manipulée       */
+    bool status;                            /* Bilan de l'opération        */
+
+    ret = PyArg_ParseTuple(args, "O&", convert_to_arch_operand, &target);
+    if (!ret) return NULL;
+
+    instr = G_ARCH_INSTRUCTION(pygobject_get(self));
+
+    status = g_arch_instruction_detach_operand(instr, target);
+
+    result = status ? Py_True : Py_False;
+    Py_INCREF(result);
+
+    return result;
+
+}
+
+
+
+/* ---------------------------------------------------------------------------------- */
 /*                     DEFINITION DES LIAISONS ENTRE INSTRUCTIONS                     */
 /* ---------------------------------------------------------------------------------- */
 
@@ -652,63 +839,6 @@ static PyObject *py_arch_instruction_get_keyword(PyObject *self, void *unused)
 
 /******************************************************************************
 *                                                                             *
-*  Paramètres  : self   = objet représentant une instruction.                 *
-*                unused = adresse non utilisée ici.                           *
-*                                                                             *
-*  Description : Fournit tous les opérandes d'une instruction.                *
-*                                                                             *
-*  Retour      : Valeur associée à la propriété consultée.                    *
-*                                                                             *
-*  Remarques   : -                                                            *
-*                                                                             *
-******************************************************************************/
-
-static PyObject *py_arch_instruction_get_operands(PyObject *self, void *unused)
-{
-    PyObject *result;                       /* Instance à retourner        */
-    GArchInstruction *instr;                /* Version native              */
-    size_t count;                           /* Nombre d'opérandes présents */
-    size_t i;                               /* Boucle de parcours          */
-    GArchOperand *operand;                  /* Opérande à manipuler        */
-    PyObject *opobj;                        /* Version Python              */
-#ifndef NDEBUG
-    int ret;                                /* Bilan d'une écriture d'arg. */
-#endif
-
-    instr = G_ARCH_INSTRUCTION(pygobject_get(self));
-
-    g_arch_instruction_lock_operands(instr);
-
-    count = _g_arch_instruction_count_operands(instr);
-
-    result = PyTuple_New(count);
-
-    for (i = 0; i < count; i++)
-    {
-        operand = _g_arch_instruction_get_operand(instr, i);
-
-        opobj = pygobject_new(G_OBJECT(operand));
-
-#ifndef NDEBUG
-        ret = PyTuple_SetItem(result, i, Py_BuildValue("O", opobj));
-        assert(ret == 0);
-#else
-        PyTuple_SetItem(result, i, Py_BuildValue("O", opobj));
-#endif
-
-        g_object_unref(G_OBJECT(operand));
-
-    }
-
-    g_arch_instruction_unlock_operands(instr);
-
-    return result;
-
-}
-
-
-/******************************************************************************
-*                                                                             *
 *  Paramètres  : obj_type = type dont le dictionnaire est à compléter.        *
 *                                                                             *
 *  Description : Définit les constantes pour les instructions.                *
@@ -755,6 +885,21 @@ static bool py_arch_instruction_define_constants(PyTypeObject *obj_type)
 PyTypeObject *get_python_arch_instruction_type(void)
 {
     static PyMethodDef py_arch_instruction_methods[] = {
+        {
+            "attach_operand", py_arch_instruction_attach_extra_operand,
+            METH_VARARGS,
+            "attach_operand($self, op, /)\n--\n\nAdd a new operand to the instruction."
+        },
+        {
+            "replace_operand", py_arch_instruction_replace_operand,
+            METH_VARARGS,
+            "replace_operand($self, old, new, /)\n--\n\nReplace an old instruction operand by a another one."
+        },
+        {
+            "detach_operand", py_arch_instruction_detach_operand,
+            METH_VARARGS,
+            "detach_operand($self, target, /)\n--\n\nRemove an operand from the instruction."
+        },
         { NULL }
     };
 
diff --git a/plugins/pychrysalide/arch/operand.c b/plugins/pychrysalide/arch/operand.c
index 4630b3a..6780e10 100644
--- a/plugins/pychrysalide/arch/operand.c
+++ b/plugins/pychrysalide/arch/operand.c
@@ -223,3 +223,48 @@ bool ensure_python_arch_operand_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 opérande d'architecture.               *
+*                                                                             *
+*  Retour      : Bilan de l'opération, voire indications supplémentaires.     *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+int convert_to_arch_operand(PyObject *arg, void *dst)
+{
+    int result;                             /* Bilan à retourner           */
+
+    result = PyObject_IsInstance(arg, (PyObject *)get_python_arch_operand_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 arch operand");
+            break;
+
+        case 1:
+            *((GArchOperand **)dst) = G_ARCH_OPERAND(pygobject_get(arg));
+            break;
+
+        default:
+            assert(false);
+            break;
+
+    }
+
+    return result;
+
+}
diff --git a/plugins/pychrysalide/arch/operand.h b/plugins/pychrysalide/arch/operand.h
index a718d8f..595a4fc 100644
--- a/plugins/pychrysalide/arch/operand.h
+++ b/plugins/pychrysalide/arch/operand.h
@@ -37,6 +37,9 @@ PyTypeObject *get_python_arch_operand_type(void);
 /* Prend en charge l'objet 'pychrysalide.arch.ArchOperand'. */
 bool ensure_python_arch_operand_is_registered(void);
 
+/* Tente de convertir en opérande d'architecture. */
+int convert_to_arch_operand(PyObject *, void *);
+
 
 
 #endif  /* _PLUGINS_PYCHRYSALIDE_ARCH_OPERAND_H */