Improved the loop detection.

1 files changed, 46 insertions, 9 deletions

diff --git a/src/analysis/disass/loop.c b/src/analysis/disass/loop.c
index b001992..c35bab7 100644
--- a/src/analysis/disass/loop.c
+++ b/src/analysis/disass/loop.c
@@ -78,6 +78,9 @@ static void tag_loop_head(bblock_info_t *, bblock_info_t *);
 static bblock_info_t *traverse_basic_blocks_dfs(bblock_info_t *, GBlockList *, bblock_info_t *, unsigned int);
 
 /* Indique si une boucle doit être définie. */
+static bool _should_be_natural_loop_link(bblock_info_t *, bblock_info_t *);
+
+/* Indique si une boucle doit être définie. */
 static bool should_be_natural_loop_link(bblock_info_t *, bblock_info_t *);
 
 /* Définit les boucles entre un ensemble de blocs basiques. */
@@ -356,7 +359,7 @@ static bblock_info_t *traverse_basic_blocks_dfs(bblock_info_t *root, GBlockList
 *                                                                             *
 ******************************************************************************/
 
-static bool should_be_natural_loop_link(bblock_info_t *dest, bblock_info_t *header)
+static bool _should_be_natural_loop_link(bblock_info_t *dest, bblock_info_t *header)
 {
     bool result;                            /* Conclusion à retourner      */
 
@@ -372,6 +375,33 @@ static bool should_be_natural_loop_link(bblock_info_t *dest, bblock_info_t *head
 
 /******************************************************************************
 *                                                                             *
+*  Paramètres  : dest   = informations du bloc de destination.                *
+*                header = informations de l'entête de boucle.                 *
+*                                                                             *
+*  Description : Indique si une boucle doit être définie.                     *
+*                                                                             *
+*  Retour      : true si une boucle naturelle est bien présente.              *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+static bool should_be_natural_loop_link(bblock_info_t *dest, bblock_info_t *header)
+{
+    bool result;                            /* Conclusion à retourner      */
+
+    result = _should_be_natural_loop_link(dest, header);
+
+    if (!result && dest->iloop_header != NULL)
+        result = should_be_natural_loop_link(dest->iloop_header, header);
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
 *  Paramètres  : list = liste de blocs de code à consulter.                   *
 *                info = informations complémentaires quant aux blocs.         *
 *                                                                             *
@@ -445,7 +475,7 @@ static void define_basic_blocks_loops(GBlockList *list, bblock_info_t *info)
             links = get_block_successors(block, info, &count);
 
             for (k = 0; k < count; k++)
-                if (should_be_natural_loop_link(links[k].info, iter->iloop_header)
+                if (_should_be_natural_loop_link(links[k].info, iter->iloop_header)
                     /**
                      * Il se peut qu'un bloc fasse référence à lui même !
                      *
@@ -504,16 +534,23 @@ void detect_loops_in_basic_blocks(GBlockList *list)
 
     count = g_block_list_count_blocks(list);
 
-    if (count > 1)
-    {
-        info = calloc(count, sizeof(bblock_info_t));
+    /**
+     * Un premier jet consistait à filtrer sur le nombre de blocs : s'il n'y
+     * en avait qu'un, il n'y avait à priori pas de raison de rechercher des
+     * boucles !
+     *
+     * Mais c'était sans compter une routine se résumant à une boucle infinie...
+     *
+     * C'est par exemple le cas avec la fonction operator new[] (_ZnajRKSt9nothrow_t)
+     * de l'échantillon b6990fc6913d839809c72d1d482cb2c295c4840fc6a1f40f38923464e958ffae.
+     */
 
-        traverse_basic_blocks_dfs(&info[0], list, info, 1);
+    info = calloc(count, sizeof(bblock_info_t));
 
-        define_basic_blocks_loops(list, info);
+    traverse_basic_blocks_dfs(&info[0], list, info, 1);
 
-        free(info);
+    define_basic_blocks_loops(list, info);
 
-    }
+    free(info);
 
 }