diff options
Diffstat (limited to 'plugins')
| -rw-r--r-- | plugins/pychrysalide/analysis/binary.c | 109 | ||||
| -rw-r--r-- | plugins/pychrysalide/analysis/db/collection.c | 63 | ||||
| -rw-r--r-- | plugins/pychrysalide/analysis/db/constants.c | 39 | ||||
| -rw-r--r-- | plugins/pychrysalide/analysis/db/item.c | 3 | ||||
| -rw-r--r-- | plugins/pychrysalide/analysis/db/items/bookmark.c | 223 | ||||
| -rw-r--r-- | plugins/pychrysalide/analysis/db/items/bookmark.h | 17 | ||||
| -rw-r--r-- | plugins/pychrysalide/analysis/db/items/module.c | 1 |
7 files changed, 452 insertions, 3 deletions
diff --git a/plugins/pychrysalide/analysis/binary.c b/plugins/pychrysalide/analysis/binary.c index 9d57952..01160a5 100644 --- a/plugins/pychrysalide/analysis/binary.c +++ b/plugins/pychrysalide/analysis/binary.c @@ -25,6 +25,7 @@ #include "binary.h" +#include <malloc.h> #include <pygobject.h> @@ -48,12 +49,18 @@ static PyObject *py_loaded_binary_new(PyTypeObject *, PyObject *, PyObject *); /* Fournit un client assurant la liaison avec un serveur. */ static PyObject *py_loaded_binary_get_client(PyObject *, PyObject *); +/* Trouve une collection assurant une fonctionnalité donnée. */ +static PyObject *py_loaded_binary_find_collection(PyObject *, PyObject *); + /* Demande l'intégration d'une modification dans une collection. */ static PyObject *py_loaded_binary_add_to_collection(PyObject *, PyObject *); /* Fournit le nom associé à l'élément binaire. */ static PyObject *py_loaded_binary_get_name(PyObject *, void *); +/* Fournit l'ensemble des collections utilisées par un binaire. */ +static PyObject *py_loaded_binary_get_collections(PyObject *, void *); + /* Fournit le format de fichier reconnu dans le contenu binaire. */ static PyObject *py_loaded_binary_get_format(PyObject *, void *); @@ -158,6 +165,59 @@ static PyObject *py_loaded_binary_get_client(PyObject *self, PyObject *args) * Paramètres : self = objet représentant un binaire chargé. * * args = arguments fournis pour l'opération. * * * +* Description : Trouve une collection assurant une fonctionnalité donnée. * +* * +* Retour : Collection trouvée ou None. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static PyObject *py_loaded_binary_find_collection(PyObject *self, PyObject *args) +{ + PyObject *result; /* Bilan à retourner */ + unsigned int feature; /* Fonctionnalité recherchée */ + int ret; /* Bilan de lecture des args. */ + GLoadedBinary *binary; /* Binaire en cours d'analyse */ + GDbCollection *found; /* Collection trouvée */ + +#define LOADED_BINARY_FIND_COLLECTION_METHOD PYTHON_METHOD_DEF \ +( \ + find_collection, "$self, feature, /", \ + METH_VARARGS, py_loaded_binary, \ + "Provide the collection managing a given database feature." \ + "\n" \ + "The feature is a value of type pychrysalide.analysis.db.DbItem.DbItemFlags." \ +) + + ret = PyArg_ParseTuple(args, "I", &feature); + if (!ret) return NULL; + + binary = G_LOADED_BINARY(pygobject_get(self)); + + found = g_loaded_binary_find_collection(binary, feature); + + if (found != NULL) + { + result = pygobject_new(G_OBJECT(found)); + g_object_unref(G_OBJECT(found)); + } + else + { + result = Py_None; + Py_INCREF(result); + } + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : self = objet représentant un binaire chargé. * +* args = arguments fournis pour l'opération. * +* * * Description : Demande l'intégration d'une modification dans une collection.* * * * Retour : Bilan partiel de l'opération demandée. * @@ -241,6 +301,53 @@ static PyObject *py_loaded_binary_get_name(PyObject *self, void *closure) * Paramètres : self = objet Python concerné par l'appel. * * closure = non utilisé ici. * * * +* Description : Fournit l'ensemble des collections utilisées par un binaire. * +* * +* Retour : Liste de collections en place. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static PyObject *py_loaded_binary_get_collections(PyObject *self, void *closure) +{ + PyObject *result; /* Trouvailles à retourner */ + GLoadedBinary *binary; /* Version native */ + size_t count; /* Quantité de collections */ + GDbCollection **collections; /* Ensemble de collections */ + size_t i; /* Boucle de parcours */ + +#define LOADED_BINARY_COLLECTIONS_ATTRIB PYTHON_GET_DEF_FULL \ +( \ + collections, py_loaded_binary, \ + "List of all collections of database items linked to the binary." \ +) + + binary = G_LOADED_BINARY(pygobject_get(self)); + + collections = g_loaded_binary_get_collections(binary, &count); + + result = PyTuple_New(count); + + for (i = 0; i < count; i++) + { + PyTuple_SetItem(result, i, pygobject_new(G_OBJECT(collections[i]))); + g_object_unref(G_OBJECT(collections[i])); + } + + if (collections != NULL) + free(collections); + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : self = objet Python concerné par l'appel. * +* closure = non utilisé ici. * +* * * Description : Fournit le format de fichier reconnu dans le contenu binaire.* * * * Retour : Instance du format reconnu. * @@ -359,6 +466,7 @@ PyTypeObject *get_python_loaded_binary_type(void) { static PyMethodDef py_loaded_binary_methods[] = { LOADED_BINARY_GET_CLIENT_METHOD, + LOADED_BINARY_FIND_COLLECTION_METHOD, LOADED_BINARY_ADD_TO_COLLECTION_METHOD, { NULL } }; @@ -368,6 +476,7 @@ PyTypeObject *get_python_loaded_binary_type(void) "name", py_loaded_binary_get_name, NULL, "Name of the loaded binary.", NULL }, + LOADED_BINARY_COLLECTIONS_ATTRIB, { "format", py_loaded_binary_get_format, NULL, "File format recognized in the binary content.", NULL diff --git a/plugins/pychrysalide/analysis/db/collection.c b/plugins/pychrysalide/analysis/db/collection.c index b7e3e0b..e62c17e 100644 --- a/plugins/pychrysalide/analysis/db/collection.c +++ b/plugins/pychrysalide/analysis/db/collection.c @@ -37,6 +37,66 @@ +/* Renvoie la liste des éléments rassemblés. */ +static PyObject *py_db_collection_get_items(PyObject *, void *); + + + +/****************************************************************************** +* * +* Paramètres : self = objet Python concerné par l'appel. * +* closure = non utilisé ici. * +* * +* Description : Renvoie la liste des éléments rassemblés. * +* * +* Retour : Liste d'éléments à parcourir. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static PyObject *py_db_collection_get_items(PyObject *self, void *closure) +{ + PyObject *result; /* Trouvailles à retourner */ + GDbCollection *collec; /* Version native */ + size_t counter; /* Décompte des éléments */ + GList *items; /* Eléments déjà en place */ + GList *iter; /* Boucle de parcours */ + int ret; /* Bilan d'une extension */ + +#define DB_COLLECTION_ITEMS_ATTRIB PYTHON_GET_DEF_FULL \ +( \ + items, py_db_collection, \ + "List of all items contained in the collection." \ + "\n" \ + "These items can currently be applied or not." \ +) + + collec = G_DB_COLLECTION(pygobject_get(self)); + + counter = 0; + result = PyTuple_New(counter); + + g_db_collection_rlock(collec); + + items = g_db_collection_get_items(G_DB_COLLECTION(collec)); + + for (iter = g_list_first(items); iter != NULL; iter = g_list_next(iter)) + { + ret = _PyTuple_Resize(&result, ++counter); + if (ret == -1) break; + + PyTuple_SetItem(result, counter - 1, pygobject_new(G_OBJECT(iter->data))); + + } + + g_db_collection_runlock(collec); + + return result; + +} + + /****************************************************************************** * * * Paramètres : - * @@ -56,9 +116,8 @@ PyTypeObject *get_python_db_collection_type(void) }; static PyGetSetDef py_db_collection_getseters[] = { - + DB_COLLECTION_ITEMS_ATTRIB, { NULL } - }; static PyTypeObject py_db_collection_type = { diff --git a/plugins/pychrysalide/analysis/db/constants.c b/plugins/pychrysalide/analysis/db/constants.c index 0c03cfc..9c628f9 100644 --- a/plugins/pychrysalide/analysis/db/constants.c +++ b/plugins/pychrysalide/analysis/db/constants.c @@ -37,6 +37,45 @@ * * * Paramètres : type = type dont le dictionnaire est à compléter. * * * +* Description : Définit les constantes relatives au protocole. * +* * +* Retour : true en cas de succès de l'opération, false sinon. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool define_db_protocol_constants(PyTypeObject *type) +{ + bool result; /* Bilan à retourner */ + PyObject *values; /* Groupe de valeurs à établir */ + + values = PyDict_New(); + + result = add_const_to_group(values, "BOOKMARKS", DBF_BOOKMARKS); + if (result) result = add_const_to_group(values, "COMMENTS", DBF_COMMENTS); + if (result) result = add_const_to_group(values, "MOVES", DBF_MOVES); + if (result) result = add_const_to_group(values, "DISPLAY_SWITCHERS", DBF_DISPLAY_SWITCHERS); + + if (!result) + { + Py_DECREF(values); + goto exit; + } + + result = attach_constants_group(type, false, "DBFeatures", values, "Features provided by database items."); + + exit: + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : type = type dont le dictionnaire est à compléter. * +* * * Description : Définit les constantes pour les éléments de base de données. * * * * Retour : true en cas de succès de l'opération, false sinon. * diff --git a/plugins/pychrysalide/analysis/db/item.c b/plugins/pychrysalide/analysis/db/item.c index 9505838..0923268 100644 --- a/plugins/pychrysalide/analysis/db/item.c +++ b/plugins/pychrysalide/analysis/db/item.c @@ -286,6 +286,9 @@ bool ensure_python_db_item_is_registered(void) if (!register_class_for_pygobject(dict, G_TYPE_DB_ITEM, type, &PyGObject_Type)) return false; + if (!define_db_protocol_constants(type)) + return false; + if (!define_db_item_constants(type)) return false; diff --git a/plugins/pychrysalide/analysis/db/items/bookmark.c b/plugins/pychrysalide/analysis/db/items/bookmark.c index 783efa6..468f38c 100644 --- a/plugins/pychrysalide/analysis/db/items/bookmark.c +++ b/plugins/pychrysalide/analysis/db/items/bookmark.c @@ -33,6 +33,7 @@ #include <plugins/dt.h> +#include "../collection.h" #include "../item.h" #include "../../../access.h" #include "../../../helpers.h" @@ -40,6 +41,9 @@ +/* --------------------- ELABORATION D'UN ELEMENT DE COLLECTION --------------------- */ + + /* Crée un nouvel objet Python de type 'DbBookmark'. */ static PyObject *py_db_bookmark_new(PyTypeObject *, PyObject *, PyObject *); @@ -54,6 +58,19 @@ static PyObject *py_db_bookmark_get_comment(PyObject *, void *); +/* ---------------------- DEFINITION DE LA COLLECTION ASSOCIEE ---------------------- */ + + +/* Crée un nouvel objet Python de type 'BookmarkCollection'. */ +static PyObject *py_bookmark_collection_new(PyTypeObject *, PyObject *, PyObject *); + + + +/* ---------------------------------------------------------------------------------- */ +/* ELABORATION D'UN ELEMENT DE COLLECTION */ +/* ---------------------------------------------------------------------------------- */ + + /****************************************************************************** * * * Paramètres : type = type de l'objet à instancier. * @@ -144,7 +161,7 @@ static int py_db_bookmark_init(PyObject *self, PyObject *args, PyObject *kwds) "\n" \ "Instances can be created using the following constructor:\n" \ "\n" \ - " DbBookmark(addr, comment=None)" \ + " DbBookmark(addr, comment=None)\n" \ "\n" \ "Where addr is a location of type pychrysalide.arch.vmpa and" \ " comment is a string or None.\n" \ @@ -400,3 +417,207 @@ int convert_to_db_bookmark(PyObject *arg, void *dst) return result; } + + + +/* ---------------------------------------------------------------------------------- */ +/* DEFINITION DE LA COLLECTION ASSOCIEE */ +/* ---------------------------------------------------------------------------------- */ + + +/****************************************************************************** +* * +* Paramètres : type = type de l'objet à instancier. * +* args = arguments fournis à l'appel. * +* kwds = arguments de type key=val fournis. * +* * +* Description : Crée un nouvel objet Python de type 'BookmarkCollection'. * +* * +* Retour : Instance Python mise en place. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static PyObject *py_bookmark_collection_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 */ + +#define BOOKMARK_COLLECTION_DOC \ + "BookmarkCollection remembers all bookmark definitions.\n" \ + "\n" \ + "Instances can be created using the following constructor:\n" \ + "\n" \ + " DbBookmark()\n" \ + "\n" \ + "There should be no need for creating such instances manually." + + /* Validations diverses */ + + base = get_python_db_bookmark_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_DB_BOOKMARK, type->tp_name, NULL, NULL, NULL); + + if (first_time) + { + status = register_class_for_dynamic_pygobject(gtype, type, base); + + 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 : - * +* * +* Description : Fournit un accès à une définition de type à diffuser. * +* * +* Retour : Définition d'objet pour Python. * +* * +* Remarques : - * +* * +******************************************************************************/ + +PyTypeObject *get_python_bookmark_collection_type(void) +{ + static PyMethodDef py_bookmark_collection_methods[] = { + { NULL } + }; + + static PyGetSetDef py_bookmark_collection_getseters[] = { + { NULL } + }; + + static PyTypeObject py_bookmark_collection_type = { + + PyVarObject_HEAD_INIT(NULL, 0) + + .tp_name = "pychrysalide.analysis.db.items.BookmarkCollection", + .tp_basicsize = sizeof(PyGObject), + + .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, + + .tp_doc = BOOKMARK_COLLECTION_DOC, + + .tp_methods = py_bookmark_collection_methods, + .tp_getset = py_bookmark_collection_getseters, + + .tp_new = py_bookmark_collection_new, + + }; + + return &py_bookmark_collection_type; + +} + + +/****************************************************************************** +* * +* Paramètres : module = module dont la définition est à compléter. * +* * +* Description : Prend en charge l'objet 'pychrysalide....BookmarkCollection'.* +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool ensure_python_bookmark_collection_is_registered(void) +{ + PyTypeObject *type; /* Type Python 'DbBookmark' */ + PyObject *module; /* Module à recompléter */ + PyObject *dict; /* Dictionnaire du module */ + + type = get_python_bookmark_collection_type(); + + if (!PyType_HasFeature(type, Py_TPFLAGS_READY)) + { + module = get_access_to_python_module("pychrysalide.analysis.db.items"); + + dict = PyModule_GetDict(module); + + if (!ensure_python_db_item_is_registered()) + return false; + + if (!register_class_for_pygobject(dict, G_TYPE_BM_COLLECTION, type, get_python_db_collection_type())) + return false; + + } + + 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 collection de signets. * +* * +* Retour : Bilan de l'opération, voire indications supplémentaires. * +* * +* Remarques : - * +* * +******************************************************************************/ + +int convert_to_bookmark_collection(PyObject *arg, void *dst) +{ + int result; /* Bilan à retourner */ + + result = PyObject_IsInstance(arg, (PyObject *)get_python_bookmark_collection_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 bookmark collection"); + break; + + case 1: + *((GBookmarkCollection **)dst) = G_BM_COLLECTION(pygobject_get(arg)); + break; + + default: + assert(false); + break; + + } + + return result; + +} diff --git a/plugins/pychrysalide/analysis/db/items/bookmark.h b/plugins/pychrysalide/analysis/db/items/bookmark.h index bcf1c11..504795f 100644 --- a/plugins/pychrysalide/analysis/db/items/bookmark.h +++ b/plugins/pychrysalide/analysis/db/items/bookmark.h @@ -31,6 +31,9 @@ +/* --------------------- ELABORATION D'UN ELEMENT DE COLLECTION --------------------- */ + + /* Fournit un accès à une définition de type à diffuser. */ PyTypeObject *get_python_db_bookmark_type(void); @@ -42,4 +45,18 @@ int convert_to_db_bookmark(PyObject *, void *); +/* ---------------------- DEFINITION DE LA COLLECTION ASSOCIEE ---------------------- */ + + +/* Fournit un accès à une définition de type à diffuser. */ +PyTypeObject *get_python_bookmark_collection_type(void); + +/* Prend en charge l'objet 'pychrysalide.analysis.db.items.BookmarkCollection'. */ +bool ensure_python_bookmark_collection_is_registered(void); + +/* Tente de convertir en collection de signets. */ +int convert_to_bookmark_collection(PyObject *, void *); + + + #endif /* _PLUGINS_PYCHRYSALIDE_ANALYSIS_DB_ITEMS_BOOKMARK_H */ diff --git a/plugins/pychrysalide/analysis/db/items/module.c b/plugins/pychrysalide/analysis/db/items/module.c index fcf636f..04349b5 100644 --- a/plugins/pychrysalide/analysis/db/items/module.c +++ b/plugins/pychrysalide/analysis/db/items/module.c @@ -89,6 +89,7 @@ bool populate_analysis_db_items_module(void) result = true; + if (result) result = ensure_python_bookmark_collection_is_registered(); if (result) result = ensure_python_db_bookmark_is_registered(); if (result) result = ensure_python_db_comment_is_registered(); |