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 #include <optional>
19
20 #include "src/trace_processor/forwarding_trace_parser.h"
21 #include "src/trace_processor/importers/common/args_tracker.h"
22 #include "src/trace_processor/importers/common/args_translation_table.h"
23 #include "src/trace_processor/importers/common/async_track_set_tracker.h"
24 #include "src/trace_processor/importers/common/chunked_trace_reader.h"
25 #include "src/trace_processor/importers/common/clock_converter.h"
26 #include "src/trace_processor/importers/common/clock_tracker.h"
27 #include "src/trace_processor/importers/common/cpu_tracker.h"
28 #include "src/trace_processor/importers/common/deobfuscation_mapping_table.h"
29 #include "src/trace_processor/importers/common/event_tracker.h"
30 #include "src/trace_processor/importers/common/flow_tracker.h"
31 #include "src/trace_processor/importers/common/global_args_tracker.h"
32 #include "src/trace_processor/importers/common/machine_tracker.h"
33 #include "src/trace_processor/importers/common/mapping_tracker.h"
34 #include "src/trace_processor/importers/common/metadata_tracker.h"
35 #include "src/trace_processor/importers/common/process_track_translation_table.h"
36 #include "src/trace_processor/importers/common/process_tracker.h"
37 #include "src/trace_processor/importers/common/sched_event_tracker.h"
38 #include "src/trace_processor/importers/common/slice_tracker.h"
39 #include "src/trace_processor/importers/common/slice_translation_table.h"
40 #include "src/trace_processor/importers/common/stack_profile_tracker.h"
41 #include "src/trace_processor/importers/common/track_tracker.h"
42 #include "src/trace_processor/importers/ftrace/ftrace_module.h"
43 #include "src/trace_processor/importers/proto/chrome_track_event.descriptor.h"
44 #include "src/trace_processor/importers/proto/multi_machine_trace_manager.h"
45 #include "src/trace_processor/importers/proto/perf_sample_tracker.h"
46 #include "src/trace_processor/importers/proto/proto_importer_module.h"
47 #include "src/trace_processor/importers/proto/track_event.descriptor.h"
48 #include "src/trace_processor/importers/proto/track_event_module.h"
49 #include "src/trace_processor/sorter/trace_sorter.h"
50 #include "src/trace_processor/trace_reader_registry.h"
51 #include "src/trace_processor/types/destructible.h"
52
53 namespace perfetto {
54 namespace 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
97 slice_tracker->SetOnSliceBeginCallback(
98 [this](TrackId track_id, SliceId slice_id) {
99 flow_tracker->ClosePendingEventsOnTrack(track_id, slice_id);
100 });
101 }
102
103 TraceProcessorContext::TraceProcessorContext() = default;
104 TraceProcessorContext::~TraceProcessorContext() = default;
105
machine_id() const106 std::optional<MachineId> TraceProcessorContext::machine_id() const {
107 if (!machine_tracker) {
108 // Doesn't require that |machine_tracker| is initialzed, e.g. in unit tests.
109 return std::nullopt;
110 }
111 return machine_tracker->machine_id();
112 }
113
114 } // namespace trace_processor
115 } // namespace perfetto
116