1 files changed, 74 insertions, 0 deletions
diff --git a/rust/examples/dwarf/dwarf_import/src/functions.rs b/rust/examples/dwarf/dwarf_import/src/functions.rs
new file mode 100644
index 00000000..20858917
--- /dev/null
+++ b/rust/examples/dwarf/dwarf_import/src/functions.rs
@@ -0,0 +1,74 @@
+// Copyright 2021-2023 Vector 35 Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+use crate::dwarfdebuginfo::{DebugInfoBuilder, TypeUID};
+use crate::helpers::*;
+use crate::types::get_type;
+
+use gimli::{constants, DebuggingInformationEntry, Dwarf, Reader, Unit};
+
+use std::ffi::CString;
+
+fn get_parameters<R: Reader<Offset = usize>>(
+    dwarf: &Dwarf<R>,
+    unit: &Unit<R>,
+    entry: &DebuggingInformationEntry<R>,
+    debug_info_builder: &mut DebugInfoBuilder,
+) -> Vec<Option<(CString, TypeUID)>> {
+    if !entry.has_children() {
+        vec![]
+    } else {
+        // We make a new tree from the current entry to iterate over its children
+        // TODO : We could instead pass the `entries` object down from parse_dwarf to avoid parsing the same object multiple times
+        let mut sub_die_tree = unit.entries_tree(Some(entry.offset())).unwrap();
+        let root = sub_die_tree.root().unwrap();
+
+        let mut result = vec![];
+        let mut children = root.children();
+        while let Some(child) = children.next().unwrap() {
+            match child.entry().tag() {
+                constants::DW_TAG_formal_parameter => {
+                    let name = get_name(dwarf, unit, child.entry());
+                    let type_ = get_type(dwarf, unit, child.entry(), debug_info_builder);
+                    if let (Some(parameter_name), Some(parameter_type)) = (name, type_) {
+                        result.push(Some((parameter_name, parameter_type)));
+                    } else {
+                        result.push(None)
+                    }
+                }
+                constants::DW_TAG_unspecified_parameters => (),
+                _ => (),
+            }
+        }
+        result
+    }
+}
+
+pub fn parse_function_entry<R: Reader<Offset = usize>>(
+    dwarf: &Dwarf<R>,
+    unit: &Unit<R>,
+    entry: &DebuggingInformationEntry<R>,
+    debug_info_builder: &mut DebugInfoBuilder,
+) {
+    // TODO : Handle OOT, stubs/trampolines
+
+    // Collect function properties (if they exist in this DIE)
+    let full_name = debug_info_builder.get_name(unit, entry);
+    let raw_name = get_raw_name(dwarf, unit, entry);
+    let return_type = get_type(dwarf, unit, entry, debug_info_builder);
+    let address = get_start_address(dwarf, unit, entry);
+    let parameters = get_parameters(dwarf, unit, entry, debug_info_builder);
+
+    debug_info_builder.insert_function(full_name, raw_name, return_type, address, parameters);
+}