Restored some parts of the decompilation process using the new basic blocks.

git-svn-id: svn://svn.gna.org/svn/chrysalide/trunk@320 abbe820e-26c8-41b2-8c08-b7b2b41f8b0a

4 files changed, 122 insertions, 2 deletions

diff --git a/src/analysis/blocks/flow.c b/src/analysis/blocks/flow.c
index 85d98ee..6fb5849 100644
--- a/src/analysis/blocks/flow.c
+++ b/src/analysis/blocks/flow.c
@@ -65,6 +65,9 @@ static void g_flow_block_dispose(GFlowBlock *);
 /* Procède à la libération totale de la mémoire. */
 static void g_flow_block_finalize(GFlowBlock *);
 
+/* Recherche le bloc contenant une adresse donnée. */
+static GInstrBlock *g_flow_block_find_by_addr(const GFlowBlock *, vmpa_t);
+
 /* Parcours le bloc d'instructions dans un ordre donné. */
 static bool g_flow_block_visit(GFlowBlock *, instr_block_visitor_cb, void *);
 
@@ -128,6 +131,7 @@ static void g_flow_block_init(GFlowBlock *block)
 
     parent = G_INSTR_BLOCK(block);
 
+    parent->find_by_addr = (find_by_addr_fc)g_flow_block_find_by_addr;
     parent->visit_blocks = (visit_all_blocks_fc)g_flow_block_visit;
     parent->list_blocks = (list_all_blocks_fc)g_flow_block_list_all_blocks;
     parent->list_regs = (list_regs_accesses_fc)g_flow_block_list_regs_accesses;
@@ -236,6 +240,38 @@ GInstrBlock *g_flow_block_new(GArchInstruction *instrs, GArchInstruction *first,
 
 /******************************************************************************
 *                                                                             *
+*  Paramètres  : block = bloc de départ des recherches.                       *
+*                addr  = ensemble de blocs à parcourir.                       *
+*                                                                             *
+*  Description : Recherche le bloc contenant une adresse donnée.              *
+*                                                                             *
+*  Retour      : bloc basique trouvé ou NULL en cas d'échec.                  *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static GInstrBlock *g_flow_block_find_by_addr(const GFlowBlock *block, vmpa_t addr)
+{
+    GInstrBlock *result;                    /* Resultat à retourner        */
+    vmpa_t start;                           /* Adresse de début du bloc    */
+    vmpa_t end;                             /* Adresse de fin du bloc      */
+
+    g_flow_block_get_boundary_addresses(block, &start, &end);
+
+    if (start <= addr && addr <= end)
+        result = G_INSTR_BLOCK(block);
+
+    else
+        result = NULL;
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
 *  Paramètres  : block    = bloc d'instructions concerné par la visite.       *
 *                callback = ensemble de blocs à parcourir.                    *
 *                data     = donnée utilisateur à associer au parcours.        *
@@ -419,6 +455,25 @@ static const reg_access *g_flow_block_list_regs_accesses(const GFlowBlock *block
 /******************************************************************************
 *                                                                             *
 *  Paramètres  : block = bloc d'instructions à consulter.                     *
+*                                                                             *
+*  Description : Fournit la liste d'appartenance des instructions du bloc.    *
+*                                                                             *
+*  Retour      : Liste de l'ensemble des instructions.                        *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+GArchInstruction *g_flow_block_get_all_instructions_list(const GFlowBlock *block)
+{
+    return block->instrs;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : block = bloc d'instructions à consulter.                     *
 *                first = instruction de départ du bloc. [OUT]                 *
 *                last  = dernière instruction du bloc. [OUT]                  *
 *                                                                             *
@@ -432,8 +487,11 @@ static const reg_access *g_flow_block_list_regs_accesses(const GFlowBlock *block
 
 void g_flow_block_get_boundary(const GFlowBlock *block, GArchInstruction **first, GArchInstruction **last)
 {
-    *first = block->first;
-    *last = block->last;
+    if (first != NULL)
+        *first = block->first;
+
+    if (last != NULL)
+        *last = block->last;
 
 }
 
diff --git a/src/analysis/blocks/flow.h b/src/analysis/blocks/flow.h
index 8bd7257..28fb5a9 100644
--- a/src/analysis/blocks/flow.h
+++ b/src/analysis/blocks/flow.h
@@ -55,6 +55,9 @@ GType g_flow_block_get_type(void);
 /* Crée un bloc d'exécution d'instructions. */
 GInstrBlock *g_flow_block_new(GArchInstruction *, GArchInstruction *, GArchInstruction *);
 
+/* Fournit la liste d'appartenance des instructions du bloc. */
+GArchInstruction *g_flow_block_get_all_instructions_list(const GFlowBlock *);
+
 /* Fournit les instructions limites d'un bloc d'exécution. */
 void g_flow_block_get_boundary(const GFlowBlock *, GArchInstruction **, GArchInstruction **);
 
diff --git a/src/analysis/blocks/virtual.c b/src/analysis/blocks/virtual.c
index 90bccca..71a6d06 100644
--- a/src/analysis/blocks/virtual.c
+++ b/src/analysis/blocks/virtual.c
@@ -69,6 +69,9 @@ static void g_virtual_block_dispose(GVirtualBlock *);
 /* Procède à la libération totale de la mémoire. */
 static void g_virtual_block_finalize(GVirtualBlock *);
 
+/* Recherche le bloc contenant une adresse donnée. */
+static GInstrBlock *g_virtual_block_find_by_addr(const GVirtualBlock *, vmpa_t);
+
 /* Parcours le bloc d'instructions dans un ordre donné. */
 static bool g_virtual_block_visit(GVirtualBlock *, instr_block_visitor_cb, void *);
 
@@ -126,6 +129,7 @@ static void g_virtual_block_init(GVirtualBlock *block)
 
     parent = G_INSTR_BLOCK(block);
 
+    parent->find_by_addr = (find_by_addr_fc)g_virtual_block_find_by_addr;
     parent->visit_blocks = (visit_all_blocks_fc)g_virtual_block_visit;
     parent->list_blocks = (list_all_blocks_fc)g_virtual_block_list_all_blocks;
     parent->list_regs = (list_regs_accesses_fc)g_virtual_block_list_regs_accesses;
@@ -207,6 +211,34 @@ GInstrBlock *g_virtual_block_new(void)
 
 /******************************************************************************
 *                                                                             *
+*  Paramètres  : block = bloc de départ des recherches.                       *
+*                addr  = ensemble de blocs à parcourir.                       *
+*                                                                             *
+*  Description : Recherche le bloc contenant une adresse donnée.              *
+*                                                                             *
+*  Retour      : bloc basique trouvé ou NULL en cas d'échec.                  *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static GInstrBlock *g_virtual_block_find_by_addr(const GVirtualBlock *block, vmpa_t addr)
+{
+    GInstrBlock *result;                    /* Resultat à retourner        */
+    size_t i;                               /* Boucle de parcours          */
+
+    result = NULL;
+
+    for (i = 0; i < block->children_count && result == NULL; i++)
+        result = g_instr_block_find_by_addr(block->children[i], addr);
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
 *  Paramètres  : block    = bloc d'instructions concerné par la visite.       *
 *                callback = ensemble de blocs à parcourir.                    *
 *                data     = donnée utilisateur à associer au parcours.        *
@@ -324,3 +356,27 @@ size_t g_virtual_block_count_children(GVirtualBlock *block)
     return block->children_count;
 
 }
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : block = bloc d'instructions à consulter.                     *
+*                index = indice du sous-bloc recherché.                       *
+*                                                                             *
+*  Description : Renvoie un des blocs contenus dans le groupe courant.        *
+*                                                                             *
+*  Retour      : Un des blocs du groupe.                                      *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+GInstrBlock *g_virtual_block_get_child(GVirtualBlock *block, size_t index)
+{
+    if (index >= block->children_count)
+        return NULL;
+
+    else
+        return block->children[index];
+
+}
diff --git a/src/analysis/blocks/virtual.h b/src/analysis/blocks/virtual.h
index 5e8ddcd..b254a27 100644
--- a/src/analysis/blocks/virtual.h
+++ b/src/analysis/blocks/virtual.h
@@ -61,6 +61,9 @@ void g_virtual_block_add_child(GVirtualBlock *, GInstrBlock *);
 /* Compte le nombre de blocs contenus dans le groupe courant. */
 size_t g_virtual_block_count_children(GVirtualBlock *);
 
+/* Renvoie un des blocs contenus dans le groupe courant. */
+GInstrBlock *g_virtual_block_get_child(GVirtualBlock *, size_t);
+
 
 
 #endif  /* _ANALYSIS_BLOCKS_VIRTUAL_H */