summaryrefslogtreecommitdiff
path: root/plugins/pychrysalide
diff options
context:
space:
mode:
authorCyrille Bagard <nocbos@gmail.com>2019-01-28 00:24:40 (GMT)
committerCyrille Bagard <nocbos@gmail.com>2019-01-28 00:24:40 (GMT)
commit94cb8acac1027a4deee933c84d7918f4a5ea4983 (patch)
treee6f5a1f372cdceb942d4bd659668c20dc34e649b /plugins/pychrysalide
parentfa99f722ed6b06ea7f6b56a8816c61ecd5053289 (diff)
Allowed to filter contents before running analysis.
Diffstat (limited to 'plugins/pychrysalide')
-rw-r--r--plugins/pychrysalide/analysis/project.c80
-rw-r--r--plugins/pychrysalide/helpers.c45
-rw-r--r--plugins/pychrysalide/helpers.h3
3 files changed, 123 insertions, 5 deletions
diff --git a/plugins/pychrysalide/analysis/project.c b/plugins/pychrysalide/analysis/project.c
index 1a85f71..137efbf 100644
--- a/plugins/pychrysalide/analysis/project.c
+++ b/plugins/pychrysalide/analysis/project.c
@@ -45,6 +45,9 @@ static PyObject *py_study_project_new(PyTypeObject *, PyObject *, PyObject *);
/* Procède à l'enregistrement d'un projet donné. */
static PyObject *py_study_project_save(PyObject *, PyObject *);
+/* Détermine si un contenu doit être écarté ou conservé. */
+static bool filter_loadable_content_with_python(GLoadedContent *, PyObject *);
+
/* Assure l'intégration de contenus binaires dans un projet. */
static PyObject *py_study_project_discover_binary_content(PyObject *, PyObject *);
@@ -135,6 +138,57 @@ static PyObject *py_study_project_save(PyObject *self, PyObject *args)
/******************************************************************************
* *
+* Paramètres : content = contenu chargeable à étudier ou NULL à la fin. *
+* callable = procédure de filtre en Python. *
+* *
+* Description : Détermine si un contenu doit être écarté ou conservé. *
+* *
+* Retour : true si le contenu doit être conservé, false sinon. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+static bool filter_loadable_content_with_python(GLoadedContent *content, PyObject *callable)
+{
+ bool result; /* Bilan à retourner */
+ PyGILState_STATE gstate; /* Sauvegarde d'environnement */
+ PyObject *arg; /* Argument à fournir au filtre*/
+ PyObject *status; /* Bilan de l'analyse */
+
+ gstate = PyGILState_Ensure();
+
+ if (content == NULL)
+ {
+ Py_DECREF(callable);
+ result = false;
+ }
+
+ else
+ {
+ arg = pygobject_new(G_OBJECT(content));
+
+ status = PyObject_CallFunctionObjArgs(callable, arg, NULL);
+
+ if (PyErr_Occurred())
+ PyErr_Print();
+
+ result = status == NULL || status == Py_False || status == Py_None ? false : true;
+ Py_XDECREF(status);
+
+ Py_DECREF(arg);
+
+ }
+
+ PyGILState_Release(gstate);
+
+ return result;
+
+}
+
+
+/******************************************************************************
+* *
* Paramètres : self = projet d'étude à manipuler. *
* args = arguments accompagnant l'appel. *
* *
@@ -148,16 +202,32 @@ static PyObject *py_study_project_save(PyObject *self, PyObject *args)
static PyObject *py_study_project_discover_binary_content(PyObject *self, PyObject *args)
{
- GStudyProject *project; /* Version GLib du format */
- int ret; /* Bilan de lecture des args. */
+ PyObject *callable; /* Filtre de contenus éventuel */
GBinContent *content; /* Instance de contenu binaire */
+ int ret; /* Bilan de lecture des args. */
+ GStudyProject *project; /* Version GLib du format */
- project = G_STUDY_PROJECT(pygobject_get(self));
+ callable = NULL;
- ret = PyArg_ParseTuple(args, "O&", convert_to_binary_content, &content);
+ ret = PyArg_ParseTuple(args, "O&|O&",
+ convert_to_binary_content, &content,
+ convert_to_callable, &callable);
if (!ret) return NULL;
- g_study_project_discover_binary_content(project, content);
+ project = G_STUDY_PROJECT(pygobject_get(self));
+
+ if (callable != NULL)
+ {
+ Py_INCREF(callable);
+
+ g_study_project_discover_binary_content(project, content,
+ (filter_loadable_cb)filter_loadable_content_with_python,
+ callable);
+
+ }
+
+ else
+ g_study_project_discover_binary_content(project, content, NULL, NULL);
Py_RETURN_NONE;
diff --git a/plugins/pychrysalide/helpers.c b/plugins/pychrysalide/helpers.c
index 7e31fa7..2b62e57 100644
--- a/plugins/pychrysalide/helpers.c
+++ b/plugins/pychrysalide/helpers.c
@@ -109,6 +109,51 @@ PyObject *status_to_rich_cmp_state(int status, int op)
/******************************************************************************
* *
+* 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 élément appelable. *
+* *
+* Retour : Bilan de l'opération, voire indications supplémentaires. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+int convert_to_callable(PyObject *arg, void *dst)
+{
+ int result; /* Bilan à retourner */
+
+ result = PyCallable_Check(arg);
+
+ 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 a callable object");
+ break;
+
+ case 1:
+ *((PyObject **)dst) = arg;
+ break;
+
+ default:
+ assert(false);
+ break;
+
+ }
+
+ return result;
+
+}
+
+
+/******************************************************************************
+* *
* Paramètres : target = propriétaire de la routine visée. *
* method = désignation de la fonction à appeler. *
* *
diff --git a/plugins/pychrysalide/helpers.h b/plugins/pychrysalide/helpers.h
index ce5840b..f7ebdc3 100644
--- a/plugins/pychrysalide/helpers.h
+++ b/plugins/pychrysalide/helpers.h
@@ -38,6 +38,9 @@
/* Traduit pour Python le bilan d'une comparaison riche. */
PyObject *status_to_rich_cmp_state(int, int);
+/* Tente de convertir en élément appelable. */
+int convert_to_callable(PyObject *, void *);
+
/* Indique si une routine Python existe ou non. */
bool has_python_method(PyObject *, const char *);