Kept all information about real addresses for routine symbols (ARM vs Thumb).

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

2 files changed, 59 insertions, 29 deletions

diff --git a/src/format/elf/helper_arm.c b/src/format/elf/helper_arm.c
index cd2d922..f47df5d 100644
--- a/src/format/elf/helper_arm.c
+++ b/src/format/elf/helper_arm.c
@@ -63,7 +63,9 @@ bool load_elf_arm_relocated_symbols(GElfFormat *format, const elf_shdr *relxxx,
 
 
 
-	virt_t virt;                            /* Adresse en mémoire virtuelle */
+    virt_t virt;                            /* Adresse en mémoire virtuelle*/
+    virt_t final_virt;                      /* Adresse virtuelle retenue   */
+    bool status;                            /* Bilan d'une opération       */
     vmpa2t addr;                            /* Localisation d'une routine  */
     mrange_t range;                         /* Couverture mémoire associée */
     GBinRoutine *routine;                   /* Nouvelle routine trouvée    */
@@ -100,14 +102,14 @@ bool load_elf_arm_relocated_symbols(GElfFormat *format, const elf_shdr *relxxx,
         {
             case R_ARM_JUMP_SLOT:
 
-
                 virt = ELF_SYM(format, sym, st_value);
-
-
                 if (virt == 0) continue;
 
+                final_virt = virt & ~0x1;
+
+                status = g_exe_format_translate_address_into_vmpa(G_EXE_FORMAT(format), final_virt, &addr);
+                if (!status) continue;
 
-                init_vmpa(&addr, VMPA_NO_PHYSICAL, virt);
                 init_mrange(&range, &addr, 0);
 
                 routine = try_to_demangle_routine(name);
@@ -118,6 +120,8 @@ bool load_elf_arm_relocated_symbols(GElfFormat *format, const elf_shdr *relxxx,
                 g_binary_symbol_attach_routine(symbol, routine);
                 g_binary_format_add_symbol(G_BIN_FORMAT(format), symbol);
 
+                /* Comptabilisation pour le désassemblage brut */
+                g_binary_format_register_code_point(G_BIN_FORMAT(format), virt, false);
 
 
                 /*
diff --git a/src/format/elf/symbols.c b/src/format/elf/symbols.c
index c3c2ecd..fa7cb8f 100644
--- a/src/format/elf/symbols.c
+++ b/src/format/elf/symbols.c
@@ -168,26 +168,25 @@ bool load_elf_symbols(GElfFormat *format)
 
 static void register_elf_entry_point(GElfFormat *format, virt_t vaddr, phys_t len, GBinRoutine *routine)
 {
-    GBinFormat *base;                   /* Version basique de l'instance   */
-    vmpa2t addr;                        /* Localisation d'une routine  */
-    mrange_t range;                     /* Couverture mémoire associée */
-    GBinSymbol *symbol;                 /* Nouveau symbole construit   */
-
-    base = G_BIN_FORMAT(format);
+    GBinFormat *base;                       /* Version basique du format   */
+    virt_t final_vaddr;                     /* Adresse virtuelle retenue   */
+    bool status;                            /* Bilan d'une opération       */
+    vmpa2t addr;                            /* Localisation d'une routine  */
+    mrange_t range;                         /* Couverture mémoire associée */
+    GBinSymbol *symbol;                     /* Nouveau symbole construit   */
 
-    /* Comptabilisation pour le désassemblage brut */
+    /* Localisation complète du symbole */
 
-    base->entry_points = (virt_t *)realloc(base->entry_points, ++base->ep_count * sizeof(virt_t));
+	if (ELF_HDR(format, format->header, e_machine) == EM_ARM)
+		final_vaddr = vaddr & ~0x1;
+    else
+        final_vaddr = vaddr;
 
-    base->entry_points[base->ep_count - 1] = vaddr;
+    status = g_exe_format_translate_address_into_vmpa(G_EXE_FORMAT(format), final_vaddr, &addr);
+    if (!status) return;
 
     /* Comptabilisation en tant que symbole */
 
-	if (ELF_HDR(format, format->header, e_machine) == EM_ARM)
-		vaddr &= ~0x1;
-
-	init_vmpa(&addr, VMPA_NO_PHYSICAL, vaddr);
-
     if (g_binary_format_find_symbol_at(G_BIN_FORMAT(format), &addr, &symbol))
     {
         g_object_unref(G_OBJECT(routine));
@@ -200,6 +199,8 @@ static void register_elf_entry_point(GElfFormat *format, virt_t vaddr, phys_t le
     }
     else
     {
+        base = G_BIN_FORMAT(format);
+
         init_mrange(&range, &addr, len);
 
         g_binary_routine_set_range(routine, &range);
@@ -208,6 +209,9 @@ static void register_elf_entry_point(GElfFormat *format, virt_t vaddr, phys_t le
         g_binary_symbol_attach_routine(symbol, routine);
         _g_binary_format_add_symbol(base, symbol, false);
 
+        /* Comptabilisation pour le désassemblage brut */
+        g_binary_format_register_code_point(base, vaddr, true);
+
     }
 
 }
@@ -513,6 +517,7 @@ static bool load_elf_internal_symbols(GElfFormat *format)
     /* Charge tous les symboles définis dans une section */
     bool add_all_symbols_from_section(GElfFormat *format, const elf_shdr *section, bool use_virt)
     {
+        GBinFormat *base;                   /* Version basique du format   */
         elf_shdr strtab;                    /* Section .strtab trouvée     */
         bool has_strtab;                    /* Présence de cette section   */
         phys_t start;                       /* Début de la zone à traiter  */
@@ -520,6 +525,7 @@ static bool load_elf_internal_symbols(GElfFormat *format)
         phys_t iter;                        /* Boucle de parcours          */
         elf_sym sym;                        /* Symbole aux infos visées    */
         virt_t virt;                        /* Adresse virtuelle           */
+        virt_t final_virt;                  /* Adresse virtuelle retenue   */
         vmpa2t addr;                        /* Localisation d'une routine  */
         mrange_t range;                     /* Couverture mémoire associée */
         const char *name;                   /* Nom du symbole trouvé       */
@@ -527,6 +533,8 @@ static bool load_elf_internal_symbols(GElfFormat *format)
         GBinRoutine *routine;               /* Nouvelle routine trouvée    */
         GBinSymbol *symbol;                 /* Nouveau symbole construit   */
 
+        base = G_BIN_FORMAT(format);
+
         has_strtab = find_elf_section_by_index(format, ELF_SHDR(format, *section, sh_link), &strtab);
 
         get_elf_section_content(format, section, &start, &size, NULL);
@@ -562,15 +570,6 @@ static bool load_elf_internal_symbols(GElfFormat *format)
 
             virt = ELF_SYM(format, sym, st_value);
 
-            if (ELF_HDR(format, format->header, e_machine) == EM_ARM)
-                virt &= ~0x1;
-
-            if (!g_exe_format_translate_address_into_vmpa(G_EXE_FORMAT(format), virt, &addr))
-                continue;
-
-
-
-            init_mrange(&range, &addr, ELF_SYM(format, sym, st_size));
             //init_mrange(&range, &addr, 0);
 
 
@@ -586,6 +585,14 @@ static bool load_elf_internal_symbols(GElfFormat *format)
             {
                 case STT_OBJECT:
 
+                    /* Ajustement de la position */
+
+                    if (!g_exe_format_translate_address_into_vmpa(G_EXE_FORMAT(format), virt, &addr))
+                    {
+                        symbol = NULL;
+                        break;
+                    }
+
                     /* Création d'un nom unique ? */
 
                     if (name != NULL)
@@ -609,6 +616,21 @@ static bool load_elf_internal_symbols(GElfFormat *format)
 
                 case STT_FUNC:
 
+                    /* Ajustement de la position */
+
+                    if (ELF_HDR(format, format->header, e_machine) == EM_ARM)
+                        final_virt = virt & ~0x1;
+                    else
+                        final_virt = virt;
+
+                    if (!g_exe_format_translate_address_into_vmpa(G_EXE_FORMAT(format), final_virt, &addr))
+                    {
+                        symbol = NULL;
+                        break;
+                    }
+
+                    init_mrange(&range, &addr, ELF_SYM(format, sym, st_size));
+
                     /* Création d'un nom unique ? */
 
                     if (name != NULL)
@@ -635,6 +657,10 @@ static bool load_elf_internal_symbols(GElfFormat *format)
                     symbol = g_binary_symbol_new(STP_ROUTINE);
                     g_binary_symbol_attach_routine(symbol, routine);
 
+                    /* Comptabilisation pour le désassemblage brut */
+
+                    g_binary_format_register_code_point(base, virt, false);
+
                     break;
 
                 default:
@@ -644,7 +670,7 @@ static bool load_elf_internal_symbols(GElfFormat *format)
             }
 
             if (symbol != NULL)
-                _g_binary_format_add_symbol(G_BIN_FORMAT(format), symbol, false);
+                _g_binary_format_add_symbol(base, symbol, false);
 
         }