memory leaks are all you need

Rust-GCC · Oct 10, 2023 · eae29a9 · eae29a9
1 parent 2ad8c35
commit eae29a9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/gcc/rust/ast/rust-macro.h b/gcc/rust/ast/rust-macro.h
@@ -576,6 +576,11 @@ class MacroRulesDefinition : public VisItem, public LocatedImpl
 
   MacroKind get_kind () const { return kind; }
 
+  std::unique_ptr<MacroRulesDefinition> clone_macro_rules_def () const
+  {
+    return std::unique_ptr<MacroRulesDefinition> (clone_item_impl ());
+  }
+
 protected:
   /* Use covariance to implement clone function as returning this object rather
    * than base */

diff --git a/gcc/rust/util/rust-hir-map.cc b/gcc/rust/util/rust-hir-map.cc
@@ -917,7 +917,8 @@ Mappings::insert_macro_invocation (AST::MacroInvocation &invoc,
   auto it = macroInvocations.find (invoc.get_macro_node_id ());
   rust_assert (it == macroInvocations.end ());
 
-  macroInvocations[invoc.get_macro_node_id ()] = def;
+  // TODO: remove hack that converts use-after-free into memory leak
+  macroInvocations[invoc.get_macro_node_id ()] = def->clone_macro_rules_def ().release ();
 }
 
 bool