/* OpenIDA - Outil d'analyse de fichiers binaires
 * links.c - résolution des liens entre différentes instructions
 *
 * Copyright (C) 2010 Cyrille Bagard
 *
 *  This file is part of OpenIDA.
 *
 *  OpenIDA 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.
 *
 *  OpenIDA 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 Foobar.  If not, see <http://www.gnu.org/licenses/>.
 */


#include "links.h"



/******************************************************************************
*                                                                             *
*  Paramètres  : list      = ensemble d'instructions à relier.                *
*                routines  = prototypes existants à insérer.                  *
*                count     = quantité de ces prototypes.                      *
*                statusbar = barre de statut avec progression à mettre à jour.*
*                id        = identifiant du message affiché à l'utilisateur.  *
*                                                                             *
*  Description : Etablit les liens entres les différentes lignes de code.     *
*                                                                             *
*  Retour      : -                                                            *
*                                                                             *
*  Remarques   : -                                                            *
*                                                                             *
******************************************************************************/

void establish_links_between_lines(GArchInstruction *list, GBinRoutine **routines, size_t count, GtkExtStatusBar *statusbar, guint id)
{
    size_t i;                               /* Boucle de parcours          */
    vmpa_t start;                           /* Adresse de départ           */
    vmpa_t end;                             /* Adresse de fin              */
    GArchInstruction *iter;                 /* Boucle de parcours          */
    vmpa_t addr;                            /* Adresse référencée          */
    InstructionLinkType type;               /* Type de référence           */
    GArchInstruction *target;               /* Ligne visée par la référence*/
    GArchInstruction *prev;                 /* Instruction précédente      */

    for (i = 0; i < count; i++)
    {
        start = g_binary_routine_get_address(routines[i]);
        end = start + g_binary_routine_get_size(routines[i]);

        /* Définition de toutes les destinations */
        for (iter = g_arch_instruction_find_by_address(list, start, true);
             iter != NULL;
             iter = g_arch_instruction_get_next_iter(list, iter, end))
        {
            type = g_arch_instruction_get_link(iter, &addr);

            switch (type)
            {
                case ILT_NONE:
                    break;

                case ILT_JUMP:

                    target = g_arch_instruction_find_by_address(list, addr, true);

                    if (target != NULL)
                        g_arch_instruction_link_with(iter, target, type);

                    break;

                case ILT_JUMP_IF_FALSE:
                    break;

                case ILT_JUMP_IF_TRUE:

                    target = g_arch_instruction_find_by_address(list, addr, true);

                    if (target != NULL)
                    {
                        g_arch_instruction_link_with(iter, target, type);

                        target = g_arch_instruction_get_next_iter(list, iter, end);
                        if (target != NULL)
                            g_arch_instruction_link_with(iter, target, ILT_JUMP_IF_FALSE);

                    }

                    break;

                case ILT_CALL:

                    target = g_arch_instruction_find_by_address(list, addr, true);

                    if (target != NULL)
                        g_arch_instruction_link_with(iter, target, type);

                    break;

                default:
                    /**
                     * Note pour GCC : à ce stade du désassemblage, ILT_CASE_JUMP et
                     * ILT_CATCH_EXCEPTION ne peuvent être présentes, car ne provenant
                     * que de greffons. Pour ILT_EXEC_FLOW, sa seule insertion est ici, plus bas.
                     */
                    break;

            }

        }

        /* Rattachement de deux blocs selon le flux normal */

        iter = g_arch_instruction_find_by_address(list, start, true);

        if (iter == NULL)
            printf("no match for 0x%08llx\n", start);

        if (iter != NULL)

        for (iter = g_arch_instruction_get_next_iter(list, iter, end);
             iter != NULL;
             iter = g_arch_instruction_get_next_iter(list, iter, end))
        {
            if (!g_arch_instruction_has_sources(iter))
                continue;

            prev = g_arch_instruction_get_prev_iter(list, iter);

            if (g_arch_instruction_is_return(prev))
                continue;

            if (!g_arch_instruction_has_destinations(prev))
                g_arch_instruction_link_with(prev, iter, ILT_EXEC_FLOW);

        }

        gtk_extended_status_bar_update_activity(statusbar, id, (i + 1) * 1.0 / count);

    }

}