dasm: sneak a ref to original di during flatten_graph

jjyg · Oct 20, 2022 · 06adbc4 · 06adbc4
1 parent def1dd4
commit 06adbc4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/metasm/disassemble_api.rb b/metasm/disassemble_api.rb
@@ -835,6 +835,8 @@ def flatten_graph(entry, include_subfunc=true)
 			end
 
 			b = @decoded[addr].block
+			# keep a ref to the di, for postprocessing (rip-relative addressing...)
+			b.list.each { |di| di.instruction.instance_variable_set("@di", di) }
 			ret.concat b.list.map { |di| di.instruction }
 
 			b.each_to_otherfunc(self) { |to|