summaryrefslogtreecommitdiff
path: root/plugins
diff options
context:
space:
mode:
authorCyrille Bagard <nocbos@gmail.com>2020-09-07 22:11:49 (GMT)
committerCyrille Bagard <nocbos@gmail.com>2020-09-07 22:11:49 (GMT)
commitf663a08007095e58f60fcf9a815a8b3d31b87c83 (patch)
tree80da49384da78cb9f1b7f81e23e638e5d1dc054e /plugins
parent39ceed9ed4e4484919f9eecc35d0b51372e5a233 (diff)
Rewritten some code managing comments.
Diffstat (limited to 'plugins')
-rw-r--r--plugins/arm/v7/fetch.c25
-rw-r--r--plugins/dalvik/link.c2
-rw-r--r--plugins/fmtp/parser.c34
-rw-r--r--plugins/lnxsyscalls/writer.c48
-rw-r--r--plugins/pychrysalide/analysis/db/items/Makefile.am1
-rw-r--r--plugins/pychrysalide/analysis/db/items/comment.c550
-rw-r--r--plugins/pychrysalide/analysis/db/items/comment.h20
-rw-r--r--plugins/pychrysalide/analysis/db/items/constants.c129
-rw-r--r--plugins/pychrysalide/analysis/db/items/constants.h42
-rw-r--r--plugins/pychrysalide/analysis/db/items/module.c1
10 files changed, 760 insertions, 92 deletions
diff --git a/plugins/arm/v7/fetch.c b/plugins/arm/v7/fetch.c
index 876ad51..2d38cbf 100644
--- a/plugins/arm/v7/fetch.c
+++ b/plugins/arm/v7/fetch.c
@@ -33,6 +33,7 @@
#include <arch/instructions/raw.h>
#include <arch/operands/immediate.h>
#include <arch/operands/register.h>
+#include <common/extstr.h>
#include <format/known.h>
#include <format/format.h>
#include <format/preload.h>
@@ -389,6 +390,7 @@ void help_fetching_with_instruction_ldr_literal_with_orig(GArchInstruction *inst
GArchInstruction *loaded; /* Instruction de valeur */
bool inserted; /* Bilan d'une insertion */
char *desc; /* Description d'accompagnement*/
+ size_t index; /* Indice d'un existant ? */
GDbComment *comment; /* Définition de commentaire */
GArchOperand *new; /* Instruction de ciblage */
@@ -494,30 +496,35 @@ void help_fetching_with_instruction_ldr_literal_with_orig(GArchInstruction *inst
g_preload_info_lock_comments(info);
- comment = _g_preload_info_find_comment_at(info, &loaded_addr);
+ comment = g_preload_info_find_comment_at(info, &loaded_addr, &index);
- if (comment != NULL)
+ if (comment == NULL)
{
- g_db_comment_add_static_text(comment, "\n");
- g_db_comment_add_dynamic_text(comment, desc);
+ comment = g_db_comment_new(&loaded_addr, CET_INLINED, BLF_HAS_CODE, desc);
+ g_db_item_add_flag(G_DB_ITEM(comment), DIF_VOLATILE);
- g_object_unref(G_OBJECT(comment));
+ _g_preload_info_add_comment(info, comment);
}
else
{
- comment = g_db_comment_new_inlined(&loaded_addr, BLF_HAS_CODE, false);
- g_db_item_add_flag(G_DB_ITEM(comment), DIF_VOLATILE);
+ desc = strprep(desc, "\n");
+ desc = strprep(desc, g_db_comment_get_text(comment));
- g_db_comment_add_dynamic_text(comment, desc);
+ g_object_unref(G_OBJECT(comment));
- _g_preload_info_add_comment(info, comment);
+ comment = g_db_comment_new(&loaded_addr, CET_INLINED, BLF_HAS_CODE, desc);
+ g_db_item_add_flag(G_DB_ITEM(comment), DIF_VOLATILE);
+
+ g_preload_info_replace_comment_at(info, index, comment);
}
g_preload_info_unlock_comments(info);
+ free(desc);
+
}
/* Mise à jour de l'affichage et conclusion */
diff --git a/plugins/dalvik/link.c b/plugins/dalvik/link.c
index d412511..9fb1992 100644
--- a/plugins/dalvik/link.c
+++ b/plugins/dalvik/link.c
@@ -358,7 +358,7 @@ void handle_dalvik_packed_switch_links(GArchInstruction *instr, GArchProcessor *
}
- item = g_db_comment_new_area(&comment->handler, BLF_NONE, msg, true);
+ item = g_db_comment_new(&comment->handler, CET_BEFORE, BLF_NONE, msg);
g_db_item_add_flag(G_DB_ITEM(item), DIF_VOLATILE);
g_proc_context_add_db_item(context, G_DB_ITEM(item));
diff --git a/plugins/fmtp/parser.c b/plugins/fmtp/parser.c
index 4a8749d..84bb86f 100644
--- a/plugins/fmtp/parser.c
+++ b/plugins/fmtp/parser.c
@@ -25,10 +25,12 @@
#include <assert.h>
+#include <string.h>
#include <i18n.h>
#include <arch/instructions/raw.h>
+#include <common/extstr.h>
#include <format/known.h>
@@ -63,6 +65,7 @@ static bool parse_field_definition(const fmt_field_def *def, GBinFormat *format,
GArchInstruction *instr; /* Instruction décodée */
GImmOperand *imm; /* Opérande à transformer */
size_t i; /* Boucle de parcours */
+ char *text; /* Texte du commentaire complet*/
const vmpa2t *addr; /* Emplacement d'instruction */
GDbComment *comment; /* Définition de commentaire */
uint64_t raw; /* Valeur brute à étudier */
@@ -125,15 +128,12 @@ static bool parse_field_definition(const fmt_field_def *def, GBinFormat *format,
/* Commentaire */
- addr = get_mrange_addr(g_arch_instruction_get_range(instr));
-
- comment = g_db_comment_new_inlined(addr, BLF_HAS_CODE, false);
- g_db_item_add_flag(G_DB_ITEM(comment), DIF_VOLATILE);
+ text = NULL;
switch (def->ctype)
{
case FCT_PLAIN:
- g_db_comment_add_static_text(comment, _(def->comment.plain));
+ text = strdup(_(def->comment.plain));
break;
case FCT_SWITCH:
@@ -155,13 +155,16 @@ static bool parse_field_definition(const fmt_field_def *def, GBinFormat *format,
else if (raw != def->comment.choices[i].fixed)
continue;
- g_db_comment_add_static_text(comment, _(def->comment.choices[i].desc));
+ text = strnadd(text, _(def->comment.choices[i].desc), strlen(_(def->comment.choices[i].desc)));
break;
}
- if (i == def->comment.ccount)
- g_db_comment_add_static_text(comment, _(def->comment.def_choice));
+ if (text == NULL)
+ {
+ assert(i == def->comment.ccount);
+ text = strdup(_(def->comment.def_choice));
+ }
break;
@@ -174,16 +177,16 @@ static bool parse_field_definition(const fmt_field_def *def, GBinFormat *format,
if (part->is_static)
{
if (part->avoid_i18n)
- g_db_comment_add_static_text(comment, part->static_text);
+ text = strnadd(text, part->static_text, strlen(part->static_text));
else
- g_db_comment_add_static_text(comment, _(part->static_text));
+ text = strnadd(text, _(part->static_text), strlen(_(part->static_text)));
}
else
{
if (part->avoid_i18n)
- g_db_comment_add_dynamic_text(comment, part->dynamic_text);
+ text = strnadd(text, part->dynamic_text, strlen(part->dynamic_text));
else
- g_db_comment_add_dynamic_text(comment, _(part->dynamic_text));
+ text = strnadd(text, _(part->dynamic_text), strlen(_(part->dynamic_text)));
}
}
@@ -192,6 +195,13 @@ static bool parse_field_definition(const fmt_field_def *def, GBinFormat *format,
}
+ addr = get_mrange_addr(g_arch_instruction_get_range(instr));
+
+ comment = g_db_comment_new(addr, CET_INLINED, BLF_HAS_CODE, text);
+ g_db_item_add_flag(G_DB_ITEM(comment), DIF_VOLATILE);
+
+ free(text);
+
/* Insertions */
inserted = g_preload_info_add_instruction(info, instr);
diff --git a/plugins/lnxsyscalls/writer.c b/plugins/lnxsyscalls/writer.c
index 743e5c6..0b90d68 100644
--- a/plugins/lnxsyscalls/writer.c
+++ b/plugins/lnxsyscalls/writer.c
@@ -29,6 +29,7 @@
#include <analysis/db/items/comment.h>
+#include <common/extstr.h>
@@ -196,43 +197,60 @@ void write_all_comments(comment_writer *writer, GPreloadInfo *preload)
{
size_t i; /* Boucle de parcours #1 */
comment_data *target; /* Commentaire à éditer */
- GDbComment *comment; /* Commentaire final à intégrer*/
- bool new; /* Nature du commentaire */
+ char *text; /* Texte du commentaire complet*/
size_t k; /* Boucle de parcours #2 */
+ size_t index; /* Indice d'un existant ? */
+ GDbComment *comment; /* Commentaire final à intégrer*/
for (i = 0; i < writer->count; i++)
{
target = &writer->comments[i];
- g_preload_info_lock_comments(preload);
+ /* Construction de la nouveauté */
- comment = _g_preload_info_find_comment_at(preload, &target->addr);
+ text = NULL;
+
+ for (k = 0; k < target->count; k++)
+ {
+ if (k > 0)
+ text = stradd(text, " / ");
- new = (comment == NULL);
+ text = stradd(text, target->text[k]);
+
+ }
+
+ /* Inclusion de l'existant */
+
+ g_preload_info_lock_comments(preload);
+
+ comment = g_preload_info_find_comment_at(preload, &target->addr, &index);
if (comment == NULL)
{
- comment = g_db_comment_new_inlined(&target->addr, BLF_HAS_CODE, false);
+ comment = g_db_comment_new(&target->addr, CET_INLINED, BLF_HAS_CODE, text);
g_db_item_add_flag(G_DB_ITEM(comment), DIF_VOLATILE);
- g_object_ref(G_OBJECT(comment));
-
_g_preload_info_add_comment(preload, comment);
}
- g_preload_info_unlock_comments(preload);
-
- for (k = 0; k < target->count; k++)
+ else
{
- if (k > 0 || !new)
- g_db_comment_add_static_text(comment, " / ");
+ text = strprep(text, " / ");
+ text = strprep(text, g_db_comment_get_text(comment));
- g_db_comment_add_dynamic_text(comment, strdup(target->text[k]));
+ g_object_unref(G_OBJECT(comment));
+
+ comment = g_db_comment_new(&target->addr, CET_INLINED, BLF_HAS_CODE, text);
+ g_db_item_add_flag(G_DB_ITEM(comment), DIF_VOLATILE);
+
+ g_preload_info_replace_comment_at(preload, index, comment);
}
- g_object_unref(G_OBJECT(comment));
+ g_preload_info_unlock_comments(preload);
+
+ free(text);
}
diff --git a/plugins/pychrysalide/analysis/db/items/Makefile.am b/plugins/pychrysalide/analysis/db/items/Makefile.am
index 07593f8..35f4ee6 100644
--- a/plugins/pychrysalide/analysis/db/items/Makefile.am
+++ b/plugins/pychrysalide/analysis/db/items/Makefile.am
@@ -4,6 +4,7 @@ noinst_LTLIBRARIES = libpychrysaanalysisdbitems.la
libpychrysaanalysisdbitems_la_SOURCES = \
bookmark.h bookmark.c \
comment.h comment.c \
+ constants.h constants.c \
module.h module.c \
switcher.h switcher.c
diff --git a/plugins/pychrysalide/analysis/db/items/comment.c b/plugins/pychrysalide/analysis/db/items/comment.c
index 845a118..78d4902 100644
--- a/plugins/pychrysalide/analysis/db/items/comment.c
+++ b/plugins/pychrysalide/analysis/db/items/comment.c
@@ -29,28 +29,57 @@
#include <pygobject.h>
-#include <i18n.h>
#include <analysis/db/items/comment.h>
+#include <plugins/dt.h>
+#include "constants.h"
+#include "../collection.h"
#include "../item.h"
#include "../../../access.h"
#include "../../../helpers.h"
#include "../../../arch/vmpa.h"
+#include "../../../glibext/constants.h"
+#include "../../../glibext/bufferline.h"
+/* --------------------- ELABORATION D'UN ELEMENT DE COLLECTION --------------------- */
+
+
/* Crée un nouvel objet Python de type 'DbComment'. */
static PyObject *py_db_comment_new(PyTypeObject *, PyObject *, PyObject *);
-/* Associe un contenu statique supplémentaire à un commentaire. */
-static PyObject *py_db_comment_add_text(PyObject *, PyObject *);
+/* Initialise une instance sur la base du dérivé de GObject. */
+static int py_db_comment_init(PyObject *, PyObject *, PyObject *);
+
+/* Fournit l'adresse associée à un commentaire. */
+static PyObject *py_db_comment_get_address(PyObject *, void *);
+
+/* Indique le type d'incrustation prévue pour un commentaire. */
+static PyObject *py_db_comment_get_embedding_type(PyObject *, void *);
+
+/* Fournit les particularités d'accroche liées à un commentaire. */
+static PyObject *py_db_comment_get_flags(PyObject *, void *);
/* Fournit le commentaire associé à un commentaire. */
static PyObject *py_db_comment_get_text(PyObject *, void *);
+/* ---------------------- DEFINITION DE LA COLLECTION ASSOCIEE ---------------------- */
+
+
+/* Crée un nouvel objet Python de type 'CommentCollection'. */
+static PyObject *py_comment_collection_new(PyTypeObject *, PyObject *, PyObject *);
+
+
+
+/* ---------------------------------------------------------------------------------- */
+/* ELABORATION D'UN ELEMENT DE COLLECTION */
+/* ---------------------------------------------------------------------------------- */
+
+
/******************************************************************************
* *
* Paramètres : type = type de l'objet à instancier. *
@@ -67,58 +96,170 @@ static PyObject *py_db_comment_get_text(PyObject *, void *);
static PyObject *py_db_comment_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
- PyObject *result; /* Instance à retourner */
- int repeatable; /* Note une répétition demandée*/
- const char *text; /* Eventuel premier commentaire*/
- int before; /* Indication sur l'emplacement*/
+ 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_db_comment_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_COMMENT, 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 : self = objet à initialiser (théoriquement). *
+* args = arguments fournis à l'appel. *
+* kwds = arguments de type key=val fournis. *
+* *
+* Description : Initialise une instance sur la base du dérivé de GObject. *
+* *
+* Retour : 0 en cas de succès, -1 sinon. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+static int py_db_comment_init(PyObject *self, PyObject *args, PyObject *kwds)
+{
+ int result; /* Bilan à renvoyer */
vmpa2t *addr; /* Emplacement ciblé */
- unsigned long flags; /* Identifiants de ligne visée */
+ CommentEmbeddingType type; /* Type d'incrustation */
+ BufferLineFlags flags; /* Particularités de l'accroche*/
+ const char *text; /* Eventuel contenu textuel */
int ret; /* Bilan de lecture des args. */
- GDbComment *comment; /* Version GLib du commentaire */
-
- static char *kwlist[] = { "addr", "flags", "repeatable", "text", "before", NULL };
+ GDbComment *comment; /* Version GLib du signet */
+ bool status; /* Bilan de l'initialisation */
+
+#define DB_COMMENT_DOC \
+ "DbComment provides support for comments to embed into the disassembled" \
+ " code.\n" \
+ "\n" \
+ "Instances can be created using the following constructor:\n" \
+ "\n" \
+ " DbComment(addr, addr, type, flags, text=None)\n" \
+ "\n" \
+ "Where *addr* is a location of type pychrysalide.arch.vmpa, *type* defines" \
+ " the kind of embedding as a" \
+ " pychrysalide.analysis.db.items.DbComment.CommentEmbeddingType value," \
+ " *flags* states for the pychrysalide.glibext.BufferLine.BufferLineFlags" \
+ " property of the line to attach and *text* is an optional string for the" \
+ " comment.\n" \
+ "\n" \
+ "An empty comment is not enough to delete a comment for a given address;" \
+ " the *ERASER* flag from the pychrysalide.analysis.db.DbItem.DbItemFlags" \
+ " enumeration must be explicitly add to the item by a call to the" \
+ " pychrysalide.analysis.db.DbItem.add_flag() function."
+
+ result = -1;
/* Récupération des paramètres */
- repeatable = -1;
text = NULL;
- before = -1;
- ret = PyArg_ParseTupleAndKeywords(args, kwds, "O&k|$psp", kwlist,
- convert_any_to_vmpa, &addr, &flags, &repeatable, &text, &before);
- if (!ret) return NULL;
+ ret = PyArg_ParseTuple(args, "O&O&O&|s",
+ convert_any_to_vmpa, &addr,
+ convert_to_comment_embedding_type, &type,
+ convert_to_buffer_line_flags, &flags,
+ &text);
+ if (!ret) goto exit;
- /* Vérifications diverses */
+ /* Initialisation d'un objet GLib */
- if (flags > BLF_ALL)
+ ret = forward_pygobjet_init(self);
+ if (ret == -1)
{
clean_vmpa_arg(addr);
+ goto exit;
+ }
- PyErr_SetString(PyExc_ValueError, _("Invalid flag combination"));
- return NULL;
+ /* Eléments de base */
- }
+ comment = G_DB_COMMENT(pygobject_get(self));
- if ((repeatable == -1 && before == -1) || (repeatable != -1 && before != -1))
+ status = g_db_comment_fill(comment, addr, type, flags, text);
+ if (!status)
{
clean_vmpa_arg(addr);
+ goto exit;
+ }
- PyErr_SetString(PyExc_ValueError, _("repeatable or before has to be defined"));
- return NULL;
+ clean_vmpa_arg(addr);
- }
+ result = 0;
- /* Construction */
+ exit:
- if (repeatable)
- comment = g_db_comment_new_inlined(addr, flags, repeatable);
- else
- comment = g_db_comment_new_area(addr, flags, text, before);
+ return result;
- result = pygobject_new(G_OBJECT(comment));
- g_object_unref(comment);
+}
- clean_vmpa_arg(addr);
+
+/******************************************************************************
+* *
+* Paramètres : self = objet Python concerné par l'appel. *
+* closure = non utilisé ici. *
+* *
+* Description : Fournit l'adresse associée à un commentaire. *
+* *
+* Retour : Adresse mémoire. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+static PyObject *py_db_comment_get_address(PyObject *self, void *closure)
+{
+ PyObject *result; /* Résultat à retourner */
+ GDbComment *comment; /* Commentaire à consulter */
+ const vmpa2t *addr; /* Localisation du commentaire */
+
+#define DB_COMMENT_ADDRESS_ATTRIB PYTHON_GET_DEF_FULL \
+( \
+ address, py_db_comment, \
+ "Location of the comment, provided as a" \
+ " pychrysalide.arch.vmpa instance." \
+)
+
+ comment = G_DB_COMMENT(pygobject_get(self));
+
+ addr = g_db_comment_get_address(comment);
+
+ result = build_from_internal_vmpa(addr);
return result;
@@ -127,31 +268,74 @@ static PyObject *py_db_comment_new(PyTypeObject *type, PyObject *args, PyObject
/******************************************************************************
* *
-* Paramètres : self = classe représentant un commentaire. *
-* args = arguments fournis à l'appel. *
+* Paramètres : self = objet Python concerné par l'appel. *
+* closure = non utilisé ici. *
* *
-* Description : Associe un contenu statique supplémentaire à un commentaire. *
+* Description : Indique le type d'incrustation prévue pour un commentaire. *
* *
-* Retour : None. *
+* Retour : Incrustation associée au commentaire. *
* *
* Remarques : - *
* *
******************************************************************************/
-static PyObject *py_db_comment_add_text(PyObject *self, PyObject *args)
+static PyObject *py_db_comment_get_embedding_type(PyObject *self, void *closure)
{
- const char *text; /* Commentaire complémentaire */
- int ret; /* Bilan de lecture des args. */
+ PyObject *result; /* Résultat à retourner */
GDbComment *comment; /* Commentaire à consulter */
+ CommentEmbeddingType type; /* Type d'incrustation */
- ret = PyArg_ParseTuple(args, "s", &text);
- if (!ret) return NULL;
+#define DB_COMMENT_EMBEDDING_TYPE_ATTRIB PYTHON_GET_DEF_FULL \
+( \
+ embedding_type, py_db_comment, \
+ "Type of embedding required for the comment, as a" \
+ " pychrysalide.analysis.db.items.DbComment.CommentEmbeddingType value." \
+)
comment = G_DB_COMMENT(pygobject_get(self));
- g_db_comment_add_static_text(comment, text);
+ type = g_db_comment_get_embedding_type(comment);
+
+ result = cast_with_constants_group_from_type(get_python_db_comment_type(), "CommentEmbeddingType", type);
- Py_RETURN_NONE;
+ return result;
+
+}
+
+
+/******************************************************************************
+* *
+* Paramètres : self = objet Python concerné par l'appel. *
+* closure = non utilisé ici. *
+* *
+* Description : Fournit les particularités d'accroche liées à un commentaire.*
+* *
+* Retour : Particularités éventuelles pour l'accroche. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+static PyObject *py_db_comment_get_flags(PyObject *self, void *closure)
+{
+ PyObject *result; /* Résultat à retourner */
+ GDbComment *comment; /* Commentaire à consulter */
+ BufferLineFlags flags; /* Fanions à rechercher */
+
+#define DB_COMMENT_FLAGS_ATTRIB PYTHON_GET_DEF_FULL \
+( \
+ flags, py_db_comment, \
+ "Flags of the line where to attach the comment, as a" \
+ " pychrysalide.glibext.BufferLine.BufferLineFlags value." \
+)
+
+ comment = G_DB_COMMENT(pygobject_get(self));
+
+ flags = g_db_comment_get_flags(comment);
+
+ result = cast_with_constants_group_from_type(get_python_buffer_line_type(), "BufferLineFlags", flags);
+
+ return result;
}
@@ -175,6 +359,13 @@ static PyObject *py_db_comment_get_text(PyObject *self, void *closure)
GDbComment *comment; /* Commentaire à consulter */
char *text; /* Contenu textuel associé */
+#define DB_COMMENT_TEXT_ATTRIB PYTHON_GET_DEF_FULL \
+( \
+ text, py_db_comment, \
+ "Content of the comment, as a string which may contain several lines," \
+ " or None of no text is linked to the comment." \
+)
+
comment = G_DB_COMMENT(pygobject_get(self));
text = g_db_comment_get_text(comment);
@@ -209,19 +400,14 @@ static PyObject *py_db_comment_get_text(PyObject *self, void *closure)
PyTypeObject *get_python_db_comment_type(void)
{
static PyMethodDef py_db_comment_methods[] = {
- {
- "add_text", py_db_comment_add_text,
- METH_VARARGS,
- "add_text($self, text, /)\n--\n\nAppend extra text to a given comment."
- },
{ NULL }
};
static PyGetSetDef py_db_comment_getseters[] = {
- {
- "text", py_db_comment_get_text, NULL,
- "Give access to the content of a given comment.", NULL
- },
+ DB_COMMENT_ADDRESS_ATTRIB,
+ DB_COMMENT_EMBEDDING_TYPE_ATTRIB,
+ DB_COMMENT_FLAGS_ATTRIB,
+ DB_COMMENT_TEXT_ATTRIB,
{ NULL }
};
@@ -234,11 +420,13 @@ PyTypeObject *get_python_db_comment_type(void)
.tp_flags = Py_TPFLAGS_DEFAULT,
- .tp_doc = "PyChrysalide comment for edited binary",
+ .tp_doc = DB_COMMENT_DOC,
.tp_methods = py_db_comment_methods,
.tp_getset = py_db_comment_getseters,
- .tp_new = py_db_comment_new
+
+ .tp_init = py_db_comment_init,
+ .tp_new = py_db_comment_new,
};
@@ -279,8 +467,260 @@ bool ensure_python_db_comment_is_registered(void)
if (!register_class_for_pygobject(dict, G_TYPE_DB_COMMENT, type, get_python_db_item_type()))
return false;
+ if (!define_db_comment_constants(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 commentaire de base. *
+* *
+* Retour : Bilan de l'opération, voire indications supplémentaires. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+int convert_to_db_comment(PyObject *arg, void *dst)
+{
+ int result; /* Bilan à retourner */
+
+ result = PyObject_IsInstance(arg, (PyObject *)get_python_db_comment_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 database comment");
+ break;
+
+ case 1:
+ *((GDbComment **)dst) = G_DB_COMMENT(pygobject_get(arg));
+ break;
+
+ default:
+ assert(false);
+ break;
+
+ }
+
+ 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 'CommentCollection'. *
+* *
+* Retour : Instance Python mise en place. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+static PyObject *py_comment_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 COMMENT_COLLECTION_DOC \
+ "CommentCollection remembers all comment definitions.\n" \
+ "\n" \
+ "Instances can be created using the following constructor:\n" \
+ "\n" \
+ " CommentCollection()\n" \
+ "\n" \
+ "There should be no need for creating such instances manually."
+
+ /* Validations diverses */
+
+ base = get_python_db_comment_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_COMMENT, 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_comment_collection_type(void)
+{
+ static PyMethodDef py_comment_collection_methods[] = {
+ { NULL }
+ };
+
+ static PyGetSetDef py_comment_collection_getseters[] = {
+ { NULL }
+ };
+
+ static PyTypeObject py_comment_collection_type = {
+
+ PyVarObject_HEAD_INIT(NULL, 0)
+
+ .tp_name = "pychrysalide.analysis.db.items.CommentCollection",
+ .tp_basicsize = sizeof(PyGObject),
+
+ .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
+
+ .tp_doc = COMMENT_COLLECTION_DOC,
+
+ .tp_methods = py_comment_collection_methods,
+ .tp_getset = py_comment_collection_getseters,
+
+ .tp_new = py_comment_collection_new,
+
+ };
+
+ return &py_comment_collection_type;
+
+}
+
+
+/******************************************************************************
+* *
+* Paramètres : module = module dont la définition est à compléter. *
+* *
+* Description : Prend en charge l'objet 'pychrysalide....CommentCollection'.*
+* *
+* Retour : Bilan de l'opération. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+bool ensure_python_comment_collection_is_registered(void)
+{
+ PyTypeObject *type; /* Type Python 'DbComment' */
+ PyObject *module; /* Module à recompléter */
+ PyObject *dict; /* Dictionnaire du module */
+
+ type = get_python_comment_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_collection_is_registered())
+ return false;
+
+ if (!register_class_for_pygobject(dict, G_TYPE_COMMENT_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 commentaires. *
+* *
+* Retour : Bilan de l'opération, voire indications supplémentaires. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+int convert_to_comment_collection(PyObject *arg, void *dst)
+{
+ int result; /* Bilan à retourner */
+
+ result = PyObject_IsInstance(arg, (PyObject *)get_python_comment_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 comment collection");
+ break;
+
+ case 1:
+ *((GCommentCollection **)dst) = G_COMMENT_COLLECTION(pygobject_get(arg));
+ break;
+
+ default:
+ assert(false);
+ break;
+
+ }
+
+ return result;
+
+}
diff --git a/plugins/pychrysalide/analysis/db/items/comment.h b/plugins/pychrysalide/analysis/db/items/comment.h
index df02cc6..4e5af64 100644
--- a/plugins/pychrysalide/analysis/db/items/comment.h
+++ b/plugins/pychrysalide/analysis/db/items/comment.h
@@ -31,12 +31,32 @@
+/* --------------------- ELABORATION D'UN ELEMENT DE COLLECTION --------------------- */
+
+
/* Fournit un accès à une définition de type à diffuser. */
PyTypeObject *get_python_db_comment_type(void);
/* Prend en charge l'objet 'pychrysalide.analysis.db.items.DbComment'. */
bool ensure_python_db_comment_is_registered(void);
+/* Tente de convertir en commentaire de base. */
+int convert_to_db_comment(PyObject *, void *);
+
+
+
+/* ---------------------- DEFINITION DE LA COLLECTION ASSOCIEE ---------------------- */
+
+
+/* Fournit un accès à une définition de type à diffuser. */
+PyTypeObject *get_python_comment_collection_type(void);
+
+/* Prend en charge l'objet 'pychrysalide.analysis.db.items.CommentCollection'. */
+bool ensure_python_comment_collection_is_registered(void);
+
+/* Tente de convertir en collection de commentaires. */
+int convert_to_comment_collection(PyObject *, void *);
+
#endif /* _PLUGINS_PYCHRYSALIDE_ANALYSIS_DB_ITEMS_COMMENT_H */
diff --git a/plugins/pychrysalide/analysis/db/items/constants.c b/plugins/pychrysalide/analysis/db/items/constants.c
new file mode 100644
index 0000000..e3ef99e
--- /dev/null
+++ b/plugins/pychrysalide/analysis/db/items/constants.c
@@ -0,0 +1,129 @@
+
+/* Chrysalide - Outil d'analyse de fichiers binaires
+ * constants.c - ajout des constantes liées aux éléments de base
+ *
+ * Copyright (C) 2020 Cyrille Bagard
+ *
+ * This file is part of Chrysalide.
+ *
+ * Chrysalide is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Chrysalide is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+
+
+#include "constants.h"
+
+
+#include <i18n.h>
+#include <analysis/db/items/comment.h>
+
+
+#include "../../../helpers.h"
+
+
+
+/******************************************************************************
+* *
+* Paramètres : type = type dont le dictionnaire est à compléter. *
+* *
+* Description : Définit les constantes relatives aux commentaires de base. *
+* *
+* Retour : true en cas de succès de l'opération, false sinon. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+bool define_db_comment_constants(PyTypeObject *type)
+{
+ bool result; /* Bilan à retourner */
+ PyObject *values; /* Groupe de valeurs à établir */
+
+ values = PyDict_New();
+
+ result = add_const_to_group(values, "INLINED", CET_INLINED);
+ if (result) result = add_const_to_group(values, "REPEATED", CET_REPEATED);
+ if (result) result = add_const_to_group(values, "BEFORE", CET_BEFORE);
+ if (result) result = add_const_to_group(values, "AFTER", CET_AFTER);
+
+ if (!result)
+ {
+ Py_DECREF(values);
+ goto exit;
+ }
+
+ result = attach_constants_group_to_type(type, false, "CommentEmbeddingType", values,
+ "Kinds of insertion for comments.");
+
+ exit:
+
+ return result;
+
+}
+
+
+/******************************************************************************
+* *
+* 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 constante CommentEmbeddingType. *
+* *
+* Retour : Bilan de l'opération, voire indications supplémentaires. *
+* *
+* Remarques : - *
+* *
+******************************************************************************/
+
+int convert_to_comment_embedding_type(PyObject *arg, void *dst)
+{
+ int result; /* Bilan à retourner */
+ unsigned long value; /* Valeur récupérée */
+
+ result = PyObject_IsInstance(arg, (PyObject *)&PyLong_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 CommentEmbeddingType");
+ break;
+
+ case 1:
+ value = PyLong_AsUnsignedLong(arg);
+
+ if (value >= CET_COUNT)
+ {
+ PyErr_SetString(PyExc_TypeError, "invalid value to convert to CommentEmbeddingType");
+ result = 0;
+ }
+
+ else
+ *((CommentEmbeddingType *)dst) = value;
+
+ break;
+
+ default:
+ assert(false);
+ break;
+
+ }
+
+ return result;
+
+}
diff --git a/plugins/pychrysalide/analysis/db/items/constants.h b/plugins/pychrysalide/analysis/db/items/constants.h
new file mode 100644
index 0000000..80210d9
--- /dev/null
+++ b/plugins/pychrysalide/analysis/db/items/constants.h
@@ -0,0 +1,42 @@
+
+/* Chrysalide - Outil d'analyse de fichiers binaires
+ * constants.h - prototypes pour l'ajout des constantes liées aux éléments de base
+ *
+ * Copyright (C) 2020 Cyrille Bagard
+ *
+ * This file is part of Chrysalide.
+ *
+ * Chrysalide is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Chrysalide is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+
+
+#ifndef _PLUGINS_PYCHRYSALIDE_ANALYSIS_DB_ITEMS_CONSTANTS_H
+#define _PLUGINS_PYCHRYSALIDE_ANALYSIS_DB_ITEMS_CONSTANTS_H
+
+
+#include <Python.h>
+#include <stdbool.h>
+
+
+
+/* Définit les constantes relatives aux commentaires de base. */
+bool define_db_comment_constants(PyTypeObject *);
+
+/* Tente de convertir en constante CommentEmbeddingType. */
+int convert_to_comment_embedding_type(PyObject *, void *);
+
+
+
+#endif /* _PLUGINS_PYCHRYSALIDE_ANALYSIS_DB_ITEMS_CONSTANTS_H */
diff --git a/plugins/pychrysalide/analysis/db/items/module.c b/plugins/pychrysalide/analysis/db/items/module.c
index c963354..cbc367d 100644
--- a/plugins/pychrysalide/analysis/db/items/module.c
+++ b/plugins/pychrysalide/analysis/db/items/module.c
@@ -91,6 +91,7 @@ bool populate_analysis_db_items_module(void)
result = true;
if (result) result = ensure_python_bookmark_collection_is_registered();
+ if (result) result = ensure_python_comment_collection_is_registered();
if (result) result = ensure_python_switcher_collection_is_registered();
if (result) result = ensure_python_db_bookmark_is_registered();
if (result) result = ensure_python_db_comment_is_registered();