Fixed #4202: missing debug location for pointer malloc primitive

Ary Borenszweig · Ary Borenszweig · commit 9be78408ccb8 · 2017-03-28T08:35:07.000-03:00
diff --git a/src/compiler/crystal/codegen/primitives.cr b/src/compiler/crystal/codegen/primitives.cr
@@ -423,11 +423,22 @@ class Crystal::CodeGenVisitor
   def codegen_primitive_pointer_malloc(node, target_def, call_args)
     type = node.type.as(PointerInstanceType)
     llvm_type = llvm_embedded_type(type.element_type)
+
+    old_debug_location = @current_debug_location
+    if @debug.line_numbers? && (location = node.location)
+      set_current_debug_location(node.location)
+    end
+
     if type.element_type.has_inner_pointers?
       last = array_malloc(llvm_type, call_args[1])
     else
       last = array_malloc_atomic(llvm_type, call_args[1])
     end
+
+    if @debug.line_numbers?
+      set_current_debug_location(old_debug_location)
+    end
+
     last
   end
 
diff --git a/src/compiler/crystal/semantic/top_level_visitor.cr b/src/compiler/crystal/semantic/top_level_visitor.cr
@@ -355,7 +355,10 @@ class Crystal::TopLevelVisitor < Crystal::SemanticVisitor
       node.raise "method marked as Primitive must have an empty body"
     end
 
-    node.body = Primitive.new(value)
+    primitive = Primitive.new(value)
+    primitive.location = node.location
+
+    node.body = primitive
   end
 
   def visit(node : Include)
