feat: Define char array of correct length when using code-based heuri…

…stic
cxiao · Oct 2, 2023 · 05cab7e · 05cab7e
1 parent 6d5338e
commit 05cab7e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/binja_plugin/actions.py b/binja_plugin/actions.py
@@ -275,6 +275,14 @@ def recover_string_slices_from_code(bv: BinaryView) -> Optional[List[RustStringS
                                     f"Candidate string: {candidate_string_slice_data[:candidate_string_slice_length]}"
                                 )
 
+                                bv.define_user_data_var(
+                                    addr=data_var.address,
+                                    var_type=Type.array(type=Type.char(), count=candidate_string_slice_length),
+                                )
+                                logger.log_info(
+                                    f"Defined string: {candidate_string_slice_data[:candidate_string_slice_length]}"
+                                )
+
 
 def action_recover_string_slices_from_code(bv: BinaryView):
     RustStringSlice.create_binary_ninja_type(bv)