Start refactoring symbols management

1 files changed, 45 insertions, 0 deletions

diff --git a/src/module.c b/src/module.c
index ffff0ca..9148aa8 100644
--- a/src/module.c
+++ b/src/module.c
@@ -1,8 +1,21 @@
 struct module {
     void* base_addr;
+    u8**    functions_name;
+    u32*    functions_start;
+    u32*    functions_end;
+    u32*    functions_unwind;
+    u32     functions_count;
+    u32     functions_capacity;
+    // u32     functions_name_capacity;
+    // u32     function_start_capacity;
+    // u32     function_end_capacity;
+    // u32     function_unwind_capacity;
+    
     IMAGE_DOS_HEADER dos_header;
     IMAGE_NT_HEADERS64 nt_header;
+#ifdef HASHMAP
     struct hashmap symbols;
+#endif
     u8 name[256U];
 };
 
@@ -18,7 +31,9 @@ u8 add_module() {
     _BitScanForward64((unsigned long*)&module_index, modules_list);
     modules_list ^= (u64)(0x1 << module_index);
 
+#ifdef HASHMAP
     modules[module_index].symbols = create_unordered_flat();
+#endif
 
     return (u8)module_index;
 };
@@ -26,3 +41,33 @@ u8 add_module() {
 void remove_module(u8 index) {
     modules_list |= (u64)(0x1 << (((i8)index) - 1U));
 }
+
+#ifdef HASHMAP
+struct element* find_function(struct module* module, u32 rip_offset) {
+    for (u32 sym_index = 0U; sym_index < module->symbols.capacity; sym_index++) {
+        struct element* cur_sym = &module->symbols.arr[sym_index];
+        if (cur_sym->start != 0U && cur_sym->start <= rip_offset && rip_offset < cur_sym->end) {
+            return cur_sym;
+        }
+    }
+    return NULL;
+}
+#else
+u32 find_function(struct module* module, u64 rip_offset) {
+    if (module->functions_count == 0U) {
+        return (u32)-1;
+    }
+    u32 min = 0U, max = module->functions_count - 1U;
+
+    while(min != max) {
+        u32 mid = (min + max) / 2U;
+        if (module->functions_start[mid] < rip_offset) {
+            min = mid + 1U;
+        } else {
+            max = mid - 1U;
+        }
+    }
+
+    return min;
+}
+#endif