add elf32 elf32_get_sec_range_by_name

4ch12dy · Nov 21, 2019 · 1269a05 · 1269a05
1 parent 2758f1c
commit 1269a05
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/XB1nLib.c b/XB1nLib.c
@@ -109,6 +109,29 @@ XRange* macho64_get_sec_range_by_name(struct mach_header_64* mh, const char* seg
     return the_segment;
 }
 
+XRange* elf32_get_sec_range_by_name(Elf32_Ehdr* elfh, const char* sec_name){
+    Elf32_Off shoff = elfh->e_shoff;
+    Elf32_Half shnum = elfh->e_shnum;
+    Elf32_Half shstrndx = elfh->e_shstrndx;
+    Elf32_Shdr* shdr = (Elf32_Shdr*)((uint8_t *)elfh + shoff);
+    Elf32_Shdr* strsh = shdr + shstrndx;
+
+    char* strtab = (char*)((uint8_t *)elfh + strsh->sh_offset);
+    XRange* the_range = malloc(sizeof(XRange));
+
+    for (int i = 0; i < shnum; ++i) {
+        Elf32_Shdr* cur_shdr = shdr + i;
+        Elf32_Half cur_sec_name_idx = cur_shdr->sh_name;
+        char* cur_sec_name = strtab+cur_sec_name_idx;
+        if (!strcmp(cur_sec_name, sec_name)) {
+            the_range->start = (uint64_t)((uint8_t *)elfh + cur_shdr->sh_offset);
+            the_range->end = the_range->start + cur_shdr->sh_size;
+            return the_range;
+        }
+    }
+    return NULL;
+}
+
 XRange* elf64_get_sec_range_by_name(Elf64_Ehdr* elfh, const char* sec_name){
     Elf64_Off shoff = elfh->e_shoff;
     Elf64_Quarter shnum = elfh->e_shnum;

diff --git a/XB1nLib.h b/XB1nLib.h
@@ -15,6 +15,7 @@
 #include "xia0.h"
 #include "macho64.h"
 #include "arch/arm64.h"
+#include "arch/arm.h"
 #include "exec_elf.h"
 
 #pragma once
@@ -38,6 +39,8 @@ void* map_file_2_mem(const char* file_path);
 
 XRange* macho64_get_sec_range_by_name(struct mach_header_64* mh, const char* seg_name, const char* sec_name);
 
+XRange* elf32_get_sec_range_by_name(Elf32_Ehdr* elfh, const char* sec_name);
+
 XRange* elf64_get_sec_range_by_name(Elf64_Ehdr* elfh, const char* sec_name);
 
 #ifdef __cplusplus