From 3c4013d36353d519dbfeab1a2cc9b3b61a9c8ffd Mon Sep 17 00:00:00 2001
From: Cristiano Calcagno <cristianoc@fb.com>
Date: Mon, 14 Mar 2016 06:08:45 -0700
Subject: [PATCH] In debug mode, write html files for headers too.]

Reviewed By: akotulski

Differential Revision: D3046212

fb-gh-sync-id: 261f2ea
shipit-source-id: 261f2ea
---
 infer/src/backend/printer.ml | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/infer/src/backend/printer.ml b/infer/src/backend/printer.ml
index 514579140..45151eb7d 100644
--- a/infer/src/backend/printer.ml
+++ b/infer/src/backend/printer.ml
@@ -560,4 +560,20 @@ let write_html_file linereader filename cfg =
 (** Create filename.ext.html for each file in the exe_env. *)
 let write_all_html_files linereader exe_env =
   if !Config.write_html then
-    Exe_env.iter_files (write_html_file linereader) exe_env
+    Exe_env.iter_files
+      (fun _ cfg ->
+         let source_files_in_cfg =
+           let files = ref DB.SourceFileSet.empty in
+           Cfg.iter_proc_desc cfg
+             (fun _ proc_desc ->
+                if Cfg.Procdesc.is_defined proc_desc
+                then
+                  let file = (Cfg.Procdesc.get_loc proc_desc).Location.file in
+                  files := DB.SourceFileSet.add file !files);
+           !files in
+         DB.SourceFileSet.iter
+           (fun file ->
+              DB.current_source := file;
+              write_html_file linereader file cfg)
+           source_files_in_cfg)
+      exe_env