Properly add perfetto_build_flags.h to cc_binary targets

R=ssid@google.com

Change-Id: Ief2fb0dc295a3be9894c59b094cb1fd1fc1a358f
diff --git a/BUILD b/BUILD
index 2fec784..d74fa36 100644
--- a/BUILD
+++ b/BUILD
@@ -562,6 +562,7 @@
     name = "trace_processor_shell",
     srcs = [
         "include/perfetto/base/build_config.h",
+        "include/perfetto/base/build_configs/bazel/perfetto_build_flags.h",
         "include/perfetto/base/compiler.h",
         "include/perfetto/base/copyable_ptr.h",
         "include/perfetto/base/export.h",
@@ -844,6 +845,7 @@
     name = "trace_to_text",
     srcs = [
         "include/perfetto/base/build_config.h",
+        "include/perfetto/base/build_configs/bazel/perfetto_build_flags.h",
         "include/perfetto/base/compiler.h",
         "include/perfetto/base/copyable_ptr.h",
         "include/perfetto/base/export.h",
diff --git a/tools/gen_bazel b/tools/gen_bazel
index baabfc3..282bf84 100755
--- a/tools/gen_bazel
+++ b/tools/gen_bazel
@@ -385,10 +385,14 @@
       else:
         target.srcs.add(label)
 
-    if target.type == 'cc_library' and ('//include/perfetto/base/build_config.h'
-        in module_desc.get('sources', [])):
-      target.hdrs.add(Label(os.path.join(buildflags_dir,
-          'perfetto_build_flags.h')))
+    if '//include/perfetto/base/build_config.h' in module_desc.get(
+        'sources', []):
+      label = Label(os.path.join(buildflags_dir, 'perfetto_build_flags.h'))
+      if target.type == 'cc_library':
+          target.hdrs.add(label)
+      elif target.type == 'cc_binary':
+        target.srcs.add(label)
+
 
   def apply_module_dependency(self, target, dep_name):
     """