Skip to content

Commit a088c6f

Browse files
committed
fix: individual function matching
1 parent c254e4d commit a088c6f

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

reai_toolkit/features/match_current_function/match_current_function.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,9 @@ def parse_confidence(item):
116116
line = {
117117
"icon_path": f"{os.path.dirname(__file__)}/../../images/failed.png",
118118
"icon_text": "Failed",
119+
"source_function_id": function_by_distance.function_id,
119120
"matched_function_name": function_by_distance.matched_functions[0].function_name,
121+
"matched_mangled_name": function_by_distance.matched_functions[0].mangled_name,
120122
"signature": "N/A",
121123
"matched_hash": function_by_distance.matched_functions[0].sha_256_hash,
122124
"matched_binary_name": function_by_distance.matched_functions[0].binary_name,
@@ -191,7 +193,7 @@ def _process_rename_batch(self, chunk: List[Dict], bv: BinaryView, deci: Decompi
191193
return 0, 0
192194

193195
addr = int(result['function_address'])
194-
if rename_function_util(bv, addr, result["matched_function_name"]):
196+
if rename_function_util(self.config, bv, addr, result["matched_function_name"], result["matched_mangled_name"], result["source_function_id"]):
195197
renamed_count += 1
196198

197199
if result.get('signature_data', None) is not None:

0 commit comments

Comments
 (0)