diff options
Diffstat (limited to 'src/analysis/decomp')
-rw-r--r-- | src/analysis/decomp/decompiler.c | 2 | ||||
-rw-r--r-- | src/analysis/decomp/reduce.c | 395 | ||||
-rw-r--r-- | src/analysis/decomp/reduce.h | 9 |
3 files changed, 392 insertions, 14 deletions
diff --git a/src/analysis/decomp/decompiler.c b/src/analysis/decomp/decompiler.c index a61e7c7..67583bb 100644 --- a/src/analysis/decomp/decompiler.c +++ b/src/analysis/decomp/decompiler.c @@ -33,6 +33,7 @@ #include "il.h" +#include "reduce.h" #include "../../decomp/output.h" #include "../../decomp/expr/block.h" #include "../../decomp/lang/java.h" /* FIXME : remme ! */ @@ -183,6 +184,7 @@ static void prepare_all_routines_for_decomp(const GLoadedBinary *binary, const c //instr = g_binary_format_decompile_routine(G_BIN_FORMAT(format), routines[i], context); + reduce_used_variables(dinstrs); g_binary_routine_set_decomp_instructions(routines[i], dinstrs); diff --git a/src/analysis/decomp/reduce.c b/src/analysis/decomp/reduce.c index 926650d..2d6fce3 100644 --- a/src/analysis/decomp/reduce.c +++ b/src/analysis/decomp/reduce.c @@ -2,7 +2,7 @@ /* OpenIDA - Outil d'analyse de fichiers binaires * reduce.c - réduction de l'usage des [pseudo]-registres * - * Copyright (C) 2012 Cyrille Bagard + * Copyright (C) 2012-2013 Cyrille Bagard * * This file is part of OpenIDA. * @@ -24,30 +24,399 @@ #include "reduce.h" -#if 0 +#include <malloc.h> +#include "../../decomp/expr/assign.h" +#include "../../decomp/expr/block.h" +#include "../../decomp/expr/immediate.h" +#include "../../decomp/expr/pseudo.h" -/* Mémorisation de l'usage de chaque variable */ -typedef struct _var_usage + + +/* ------------------------ DECOMPTE DES USAGES DE VARIABLES ------------------------ */ + + +/* Mémorisation de l'usage d'une variable */ +typedef struct _assigned_value +{ + GDecInstruction *value; /* Valeur de cette variable */ + + GDecInstruction *assign; /* Lieu d'assignation */ + GDecInstruction *parent; /* Zone à nettoyer au besoin */ + + size_t used_counter; /* Décompte des utilisations */ + size_t shared_counter; /* Partage entre tables */ + +} assigned_value; + + +/* Conserve tous les éléments concernant une assignation. */ +static assigned_value *memorize_assigned_value(GDecInstruction *, GDecInstruction *, GDecInstruction *); + +/* Met (éventuellement) fin à un suivi d'assignation. */ +static void forget_assigned_value(assigned_value *); + +/* Constitue un lieu de conservation d'associations pour bloc. */ +static GHashTable *create_tracking_table(GHashTable *); + +/* Met à jour les décomptes des différentes assignations. */ +static void merge_usages(GHashTable *, GHashTable *); + +/* Prend notre d'une assignation d'une valeur à une variable. */ +static void note_new_variable(GHashTable *, GDecInstruction *, assigned_value *); + + + +/* ---------------------- VISITES DES INSTRUCTIONS DECOMPILEES ---------------------- */ + + +/* Prend note de tous les accès aux différentes variables. */ +static void collect_used_variables(GDecInstruction *, GDecInstruction *, GHashTable *); + +/* Procède au remplacement de variables à rejeter. */ +static void replace_useless_variable(GDecInstruction *, GHashTable *); + +/* Visite un groupe d'instructions décompilées. */ +static bool visit_instructions_block(GDecInstruction *, GDecInstruction *, DecInstrVisitFlags, GHashTable **); + + + +/* ---------------------------------------------------------------------------------- */ +/* DECOMPTE DES USAGES DE VARIABLES */ +/* ---------------------------------------------------------------------------------- */ + + +/****************************************************************************** +* * +* Paramètres : value = nouvelle valeur assignée à suivre. * +* assign = instruction où la valeur est assignée. * +* parent = bloc d'instructions à soulager en cas d'effacement. * +* table = mémorisation des usages de variables. * +* * +* Description : Conserve tous les éléments concernant une assignation. * +* * +* Retour : Structure mise en place. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static assigned_value *memorize_assigned_value(GDecInstruction *value, GDecInstruction *assign, GDecInstruction *parent) +{ + assigned_value *result; /* Mémorisation à retourner */ + + result = (assigned_value *)calloc(1, sizeof(assigned_value)); + + g_object_ref(G_OBJECT(value)); + result->value = value; + + g_object_ref(G_OBJECT(assign)); + g_object_ref(G_OBJECT(parent)); + result->assign = assign; + result->parent = parent; + + result->used_counter = 0; + result->shared_counter = 0; + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : val = structure à libérer de la mémoire. * +* * +* Description : Met (éventuellement) fin à un suivi d'assignation. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void forget_assigned_value(assigned_value *val) +{ + if (--val->shared_counter == 0) + { + g_object_unref(G_OBJECT(val->value)); + + g_object_unref(G_OBJECT(val->assign)); + g_object_unref(G_OBJECT(val->parent)); + + free(val); + + } + +} + + +/****************************************************************************** +* * +* Paramètres : parent = éventuel bloc d'instructions parent ou NULL. * +* * +* Description : Constitue un lieu de conservation d'associations pour bloc. * +* * +* Retour : Table prête à emploi. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static GHashTable *create_tracking_table(GHashTable *parent) +{ + GHashTable *result; /* Création à retourner */ + GHashTableIter iter; /* Boucle de parcours */ + GDecInstruction *var; /* Variable assignée */ + assigned_value *val; /* Valeur assignée */ + + result = g_hash_table_new_full(NULL, NULL, g_object_unref, + (GDestroyNotify)forget_assigned_value); + + if (parent != NULL) + { + g_hash_table_iter_init(&iter, parent); + + while (g_hash_table_iter_next(&iter, (gpointer *)&var, (gpointer *)&val)) + note_new_variable(result, var, val); + + } + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : table = conservation des associations variable <-> valeur. * +* sub = conservation issue d'un sous bloc. * +* * +* Description : Met à jour les décomptes des différentes assignations. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void merge_usages(GHashTable *table, GHashTable *sub) +{ + GHashTableIter iter; /* Boucle de parcours */ + GDecInstruction *var; /* Variable assignée */ + assigned_value *val; /* Valeur assignée */ + assigned_value *update; /* Même valeur, mise à jour */ + + g_hash_table_iter_init(&iter, table); + + while (g_hash_table_iter_next(&iter, (gpointer *)&var, (gpointer *)&val)) + { + update = (assigned_value *)g_hash_table_lookup(sub, var); + + val->used_counter = update->used_counter; + + g_hash_table_remove(sub, var); + + } + +} + + +/****************************************************************************** +* * +* Paramètres : table = conservation des associations variable <-> valeur. * +* var = variable en place à suivre. * +* val = valeur associée. * +* * +* Description : Prend notre d'une assignation d'une valeur à une variable. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void note_new_variable(GHashTable *table, GDecInstruction *var, assigned_value *val) +{ + g_object_ref(G_OBJECT(var)); + val->shared_counter++; + + g_hash_table_insert(table, var, val); + +} + + + +/* ---------------------------------------------------------------------------------- */ +/* VISITES DES INSTRUCTIONS DECOMPILEES */ +/* ---------------------------------------------------------------------------------- */ + + +/****************************************************************************** +* * +* Paramètres : instr = instruction visitée. * +* parent = instruction parente. * +* table = mémorisation des usages de variables. * +* * +* Description : Prend note de tous les accès aux différentes variables. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void collect_used_variables(GDecInstruction *instr, GDecInstruction *parent, GHashTable *table) +{ + GAssignExpression *assign; /* Instruction d'assignation */ + GDecInstruction *dest; /* Variable de destination */ + bool valid; /* Destination valide ? */ + GDecInstruction *src; /* Valeur de source */ + assigned_value *value; /* Valeur et ses informations */ + + if (G_IS_ASSIGN_EXPRESSION(instr)) + { + assign = G_ASSIGN_EXPRESSION(instr); + + dest = G_DEC_INSTRUCTION(g_assign_expression_get_dest(assign)); + + if (G_IS_PSEUDO_REGISTER(dest)) + valid = (g_pseudo_register_get_usage(G_PSEUDO_REGISTER(dest)) == PRU_LOCAL); + else + valid = true; + + if (valid) + { + src = G_DEC_INSTRUCTION(g_assign_expression_get_src(assign)); + + value = memorize_assigned_value(src, instr, parent); + note_new_variable(table, dest, value); + + } + + } + else + { + value = (assigned_value *)g_hash_table_lookup(table, instr); + + if (value != NULL) + value->used_counter++; + + } + +} + + +/****************************************************************************** +* * +* Paramètres : block = bloc d'instructions décompilées à traiter. * +* table = mémorisation des usages de variables. * +* * +* Description : Procède au remplacement de variables à rejeter. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void replace_useless_variable(GDecInstruction *block, GHashTable *table) { - GDecInstruction *var; /* Variable manipulée */ + GHashTableIter iter; /* Boucle de parcours */ + GDecInstruction *var; /* Variable assignée */ + assigned_value *val; /* Valeur assignée */ + bool replace; /* Ordre de remplacement */ + + g_hash_table_iter_init(&iter, table); + + while (g_hash_table_iter_next(&iter, (gpointer *)&var, (gpointer *)&val)) + { + replace = G_IS_IMM_EXPRESSION(val->value); + replace |= (val->used_counter == 2); /* 1 assign. + 1 usage */ + + if (!replace) continue; - size_t ref_counter; /* Décompte des utilisations */ - GDecInstruction *creation; /* Emplacement de la création */ - GDecInstruction *usage; /* Lieu de dernère utilisation */ + g_expr_block_delete_item(G_EXPR_BLOCK(val->parent), val->assign); -} var_usage; + g_dec_instruction_replace(block, var, val->value); -/* Réduction des usages */ -typedef struct _usage_reduc + } + +} + + +/****************************************************************************** +* * +* Paramètres : instr = instruction visitée. * +* parent = instruction parente. * +* flags = moments des appels réalisés en retour. * +* table = mémorisation des usages de variables. * +* * +* Description : Visite un groupe d'instructions décompilées. * +* * +* Retour : true afin d'aller jusqu'au terme du parcours. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static bool visit_instructions_block(GDecInstruction *instr, GDecInstruction *parent, DecInstrVisitFlags flags, GHashTable **table) { + GHashTable *sub; /* Associations pour sous-bloc */ + + if (G_IS_EXPR_BLOCK(instr)) + { + /* Entrée dans un bloc : on crée un contexte propre au bloc ! */ + if (flags == DVF_ENTER) + { + g_object_set_data(G_OBJECT(instr), "var_collect", *table); + *table = create_tracking_table(*table); + } + + /* Sortie du bloc : actions locales + purge ! */ + else if (flags == DVF_EXIT) + { + sub = *table; + *table = (GHashTable *)g_object_get_data(G_OBJECT(instr), "var_collect"); + + if (*table != NULL) + merge_usages(*table, sub); + replace_useless_variable(instr, sub); + g_hash_table_unref(sub); + } -} usage_reduc; + } + else if (flags == DVF_ENTER) + collect_used_variables(instr, parent, *table); + + return true; + +} + + +/****************************************************************************** +* * +* Paramètres : instr = instructions à traiter. * +* * +* Description : Réduit le nombre de variables utilisées, via propagations. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +void reduce_used_variables(GDecInstruction *instr) +{ + GHashTable *table; -#endif + table = NULL; + g_dec_instruction_visit(instr, (dec_instr_visitor_cb)visit_instructions_block, + DVF_ENTER | DVF_EXIT, &table); +} diff --git a/src/analysis/decomp/reduce.h b/src/analysis/decomp/reduce.h index aa13d7f..1b62a06 100644 --- a/src/analysis/decomp/reduce.h +++ b/src/analysis/decomp/reduce.h @@ -2,7 +2,7 @@ /* OpenIDA - Outil d'analyse de fichiers binaires * reduce.h - prototypes pour la réduction de l'usage des [pseudo]-registres * - * Copyright (C) 2012 Cyrille Bagard + * Copyright (C) 2012-2013 Cyrille Bagard * * This file is part of OpenIDA. * @@ -25,6 +25,13 @@ #define _ANALYSIS_DECOMP_REDUCE_H +#include "../../decomp/instruction.h" + + + +/* Réduit le nombre de variables utilisées, via propagations. */ +void reduce_used_variables(GDecInstruction *instr); + #endif /* _ANALYSIS_DECOMP_REDUCE_H */ |