1 /*
2 * Copyright (C) 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "src/trace_processor/types/trace_processor_context.h"
18
19 #include <memory>
20 #include <optional>
21
22 #include "perfetto/base/logging.h"
23 #include "src/trace_processor/forwarding_trace_parser.h"
24 #include "src/trace_processor/importers/common/args_tracker.h"
25 #include "src/trace_processor/importers/common/args_translation_table.h"
26 #include "src/trace_processor/importers/common/async_track_set_tracker.h"
27 #include "src/trace_processor/importers/common/clock_converter.h"
28 #include "src/trace_processor/importers/common/clock_tracker.h"
29 #include "src/trace_processor/importers/common/cpu_tracker.h"
30 #include "src/trace_processor/importers/common/event_tracker.h"
31 #include "src/trace_processor/importers/common/flow_tracker.h"
32 #include "src/trace_processor/importers/common/global_args_tracker.h"
33 #include "src/trace_processor/importers/common/legacy_v8_cpu_profile_tracker.h"
34 #include "src/trace_processor/importers/common/machine_tracker.h"
35 #include "src/trace_processor/importers/common/mapping_tracker.h"
36 #include "src/trace_processor/importers/common/metadata_tracker.h"
37 #include "src/trace_processor/importers/common/process_track_translation_table.h"
38 #include "src/trace_processor/importers/common/process_tracker.h"
39 #include "src/trace_processor/importers/common/sched_event_tracker.h"
40 #include "src/trace_processor/importers/common/slice_tracker.h"
41 #include "src/trace_processor/importers/common/slice_translation_table.h"
42 #include "src/trace_processor/importers/common/stack_profile_tracker.h"
43 #include "src/trace_processor/importers/common/trace_file_tracker.h"
44 #include "src/trace_processor/importers/common/track_tracker.h"
45 #include "src/trace_processor/importers/proto/android_track_event.descriptor.h"
46 #include "src/trace_processor/importers/proto/chrome_track_event.descriptor.h"
47 #include "src/trace_processor/importers/proto/multi_machine_trace_manager.h"
48 #include "src/trace_processor/importers/proto/perf_sample_tracker.h"
49 #include "src/trace_processor/importers/proto/proto_importer_module.h"
50 #include "src/trace_processor/importers/proto/track_event.descriptor.h"
51 #include "src/trace_processor/storage/trace_storage.h"
52 #include "src/trace_processor/trace_reader_registry.h"
53
54 namespace perfetto::trace_processor {
55
TraceProcessorContext(const InitArgs & args)56 TraceProcessorContext::TraceProcessorContext(const InitArgs& args)
57 : config(args.config), storage(args.storage) {
58 reader_registry = std::make_unique<TraceReaderRegistry>(this);
59 // Init the trackers.
60 machine_tracker.reset(new MachineTracker(this, args.raw_machine_id));
61 if (!machine_id()) {
62 multi_machine_trace_manager.reset(new MultiMachineTraceManager(this));
63 }
64 track_tracker.reset(new TrackTracker(this));
65 async_track_set_tracker.reset(new AsyncTrackSetTracker(this));
66 args_tracker.reset(new ArgsTracker(this));
67 args_translation_table.reset(new ArgsTranslationTable(storage.get()));
68 slice_tracker.reset(new SliceTracker(this));
69 slice_translation_table.reset(new SliceTranslationTable(storage.get()));
70 flow_tracker.reset(new FlowTracker(this));
71 event_tracker.reset(new EventTracker(this));
72 sched_event_tracker.reset(new SchedEventTracker(this));
73 process_tracker.reset(new ProcessTracker(this));
74 process_track_translation_table.reset(
75 new ProcessTrackTranslationTable(storage.get()));
76 clock_tracker.reset(new ClockTracker(this));
77 clock_converter.reset(new ClockConverter(this));
78 mapping_tracker.reset(new MappingTracker(this));
79 perf_sample_tracker.reset(new PerfSampleTracker(this));
80 stack_profile_tracker.reset(new StackProfileTracker(this));
81 metadata_tracker.reset(new MetadataTracker(storage.get()));
82 cpu_tracker.reset(new CpuTracker(this));
83 global_args_tracker.reset(new GlobalArgsTracker(storage.get()));
84 {
85 descriptor_pool_.reset(new DescriptorPool());
86 auto status = descriptor_pool_->AddFromFileDescriptorSet(
87 kTrackEventDescriptor.data(), kTrackEventDescriptor.size());
88
89 PERFETTO_DCHECK(status.ok());
90
91 status = descriptor_pool_->AddFromFileDescriptorSet(
92 kChromeTrackEventDescriptor.data(), kChromeTrackEventDescriptor.size());
93
94 PERFETTO_DCHECK(status.ok());
95
96 status = descriptor_pool_->AddFromFileDescriptorSet(
97 kAndroidTrackEventDescriptor.data(),
98 kAndroidTrackEventDescriptor.size());
99
100 PERFETTO_DCHECK(status.ok());
101 }
102
103 slice_tracker->SetOnSliceBeginCallback(
104 [this](TrackId track_id, SliceId slice_id) {
105 flow_tracker->ClosePendingEventsOnTrack(track_id, slice_id);
106 });
107
108 trace_file_tracker = std::make_unique<TraceFileTracker>(this);
109 legacy_v8_cpu_profile_tracker =
110 std::make_unique<LegacyV8CpuProfileTracker>(this);
111 }
112
113 TraceProcessorContext::TraceProcessorContext() = default;
114 TraceProcessorContext::~TraceProcessorContext() = default;
115
116 TraceProcessorContext::TraceProcessorContext(TraceProcessorContext&&) = default;
117 TraceProcessorContext& TraceProcessorContext::operator=(
118 TraceProcessorContext&&) = default;
119
machine_id() const120 std::optional<MachineId> TraceProcessorContext::machine_id() const {
121 if (!machine_tracker) {
122 // Doesn't require that |machine_tracker| is initialzed, e.g. in unit tests.
123 return std::nullopt;
124 }
125 return machine_tracker->machine_id();
126 }
127
128 } // namespace perfetto::trace_processor
129