commit 12f0f8e837f09162edd31a8a5e86964f1a7bf9ac
author Florian Mayer <fmayer@google.com> Tue Apr 02 12:31:13 2019 +0100
committer Florian Mayer <fmayer@google.com> Tue Apr 02 12:31:13 2019 +0100
tree c7ed7b87d67176f739baf1cfe0ead469d3af46c9
parent 2c56916c8c6a1d43e789bc20f46c0cf5b5c16bca

Add data source parsing packages.list.

Bug: 123186697

Change-Id: Ie446b8e96fddc77d8087605c27b4134a1ced59f5

src/traced/probes/probes_producer.cc

diff --git a/src/traced/probes/probes_producer.cc b/src/traced/probes/probes_producer.cc
index 1832fd5..9128ba7 100644
--- a/src/traced/probes/probes_producer.cc
+++ b/src/traced/probes/probes_producer.cc
@@ -36,6 +36,7 @@
 #include "src/traced/probes/android_log/android_log_data_source.h"
 #include "src/traced/probes/filesystem/inode_file_data_source.h"
 #include "src/traced/probes/ftrace/ftrace_data_source.h"
+#include "src/traced/probes/packages_list/packages_list_data_source.h"
 #include "src/traced/probes/power/android_power_data_source.h"
 #include "src/traced/probes/probes_data_source.h"
 #include "src/traced/probes/ps/process_stats_data_source.h"
@@ -61,6 +62,7 @@
 constexpr char kSysStatsSourceName[] = "linux.sys_stats";
 constexpr char kAndroidPowerSourceName[] = "android.power";
 constexpr char kAndroidLogSourceName[] = "android.log";
+constexpr char kPackagesListSourceName[] = "android.packages_list";
 
 }  // namespace.
 
@@ -120,6 +122,12 @@
     desc.set_name(kAndroidLogSourceName);
     endpoint_->RegisterDataSource(desc);
   }
+
+  {
+    DataSourceDescriptor desc;
+    desc.set_name(kPackagesListSourceName);
+    endpoint_->RegisterDataSource(desc);
+  }
 }
 
 void ProbesProducer::OnDisconnect() {
@@ -172,6 +180,8 @@
     data_source = CreateAndroidPowerDataSource(session_id, config);
   } else if (config.name() == kAndroidLogSourceName) {
     data_source = CreateAndroidLogDataSource(session_id, config);
+  } else if (config.name() == kPackagesListSourceName) {
+    data_source = CreatePackagesListDataSource(session_id, config);
   }
 
   if (!data_source) {
@@ -282,6 +292,14 @@
                                endpoint_->CreateTraceWriter(buffer_id)));
 }
 
+std::unique_ptr<ProbesDataSource> ProbesProducer::CreatePackagesListDataSource(
+    TracingSessionID session_id,
+    const DataSourceConfig& config) {
+  auto buffer_id = static_cast<BufferID>(config.target_buffer());
+  return std::unique_ptr<ProbesDataSource>(new PackagesListDataSource(
+      session_id, endpoint_->CreateTraceWriter(buffer_id)));
+}
+
 std::unique_ptr<ProbesDataSource> ProbesProducer::CreateSysStatsDataSource(
     TracingSessionID session_id,
     const DataSourceConfig& config) {