1 // Copyright 2020 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #ifndef SERVICES_RESOURCE_COORDINATOR_PUBLIC_CPP_MEMORY_INSTRUMENTATION_TRACING_OBSERVER_PROTO_H_
6 #define SERVICES_RESOURCE_COORDINATOR_PUBLIC_CPP_MEMORY_INSTRUMENTATION_TRACING_OBSERVER_PROTO_H_
7
8 #include "base/component_export.h"
9 #include "base/macros.h"
10 #include "base/trace_event/memory_dump_manager.h"
11 #include "base/trace_event/trace_event.h"
12 #include "services/resource_coordinator/public/cpp/memory_instrumentation/tracing_observer.h"
13 #include "services/resource_coordinator/public/mojom/memory_instrumentation/memory_instrumentation.mojom.h"
14 #include "services/tracing/public/cpp/perfetto/perfetto_traced_process.h"
15 #include "third_party/perfetto/include/perfetto/ext/tracing/core/trace_writer.h"
16
17 namespace perfetto {
18 namespace protos {
19 namespace pbzero {
20 class SmapsPacket;
21 }
22 } // namespace protos
23 } // namespace perfetto
24
25 namespace memory_instrumentation {
26
27 // Version of TracingObserver that serializes the dump into a proto TracePacket.
COMPONENT_EXPORT(RESOURCE_COORDINATOR_PUBLIC_MEMORY_INSTRUMENTATION)28 class COMPONENT_EXPORT(RESOURCE_COORDINATOR_PUBLIC_MEMORY_INSTRUMENTATION)
29 TracingObserverProto
30 : public TracingObserver,
31 public tracing::PerfettoTracedProcess::DataSourceBase {
32 public:
33 TracingObserverProto(base::trace_event::TraceLog*,
34 base::trace_event::MemoryDumpManager*);
35 ~TracingObserverProto() override;
36
37 bool AddChromeDumpToTraceIfEnabled(
38 const base::trace_event::MemoryDumpRequestArgs&,
39 const base::ProcessId pid,
40 const base::trace_event::ProcessMemoryDump*,
41 const base::TimeTicks& timestamp) override;
42 bool AddOsDumpToTraceIfEnabled(
43 const base::trace_event::MemoryDumpRequestArgs&,
44 const base::ProcessId,
45 const mojom::OSMemDump&,
46 const std::vector<mojom::VmRegionPtr>&,
47 const base::TimeTicks& timestamp) override;
48
49 void StartTracing(
50 tracing::PerfettoProducer* producer,
51 const perfetto::DataSourceConfig& data_source_config) override;
52
53 void StopTracing(
54 base::OnceClosure stop_complete_callback = base::OnceClosure()) override;
55
56 void Flush(base::RepeatingClosure flush_complete_callback) override;
57
58 static void MemoryMapsAsProtoInto(
59 const std::vector<mojom::VmRegionPtr>& memory_maps,
60 perfetto::protos::pbzero::SmapsPacket* smaps,
61 bool is_argument_filtering_enabled);
62
63 private:
64 std::unique_ptr<perfetto::TraceWriter> trace_writer_;
65
66 base::Lock producer_lock_;
67
68 DISALLOW_COPY_AND_ASSIGN(TracingObserverProto);
69 };
70
71 } // namespace memory_instrumentation
72
73 #endif // SERVICES_RESOURCE_COORDINATOR_PUBLIC_CPP_MEMORY_INSTRUMENTATION_TRACING_OBSERVER_PROTO_H_
74