1 /*
2 * Copyright (C) 2019 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 #include "src/trace_processor/importers/proto/track_event_module.h"
17
18 #include <cstdint>
19 #include <utility>
20
21 #include "perfetto/base/logging.h"
22 #include "perfetto/trace_processor/ref_counted.h"
23 #include "perfetto/trace_processor/trace_blob_view.h"
24 #include "src/trace_processor/importers/common/parser_types.h"
25 #include "src/trace_processor/importers/proto/android_track_event.descriptor.h"
26 #include "src/trace_processor/importers/proto/chrome_track_event.descriptor.h"
27 #include "src/trace_processor/importers/proto/packet_sequence_state_generation.h"
28 #include "src/trace_processor/importers/proto/proto_importer_module.h"
29 #include "src/trace_processor/importers/proto/track_event.descriptor.h"
30 #include "src/trace_processor/importers/proto/track_event_tracker.h"
31 #include "src/trace_processor/types/trace_processor_context.h"
32
33 #include "protos/perfetto/trace/trace_packet.pbzero.h"
34
35 namespace perfetto::trace_processor {
36
37 using perfetto::protos::pbzero::TracePacket;
38
TrackEventModule(TraceProcessorContext * context)39 TrackEventModule::TrackEventModule(TraceProcessorContext* context)
40 : track_event_tracker_(new TrackEventTracker(context)),
41 tokenizer_(context, track_event_tracker_.get()),
42 parser_(context, track_event_tracker_.get()) {
43 RegisterForField(TracePacket::kTrackEventRangeOfInterestFieldNumber, context);
44 RegisterForField(TracePacket::kTrackEventFieldNumber, context);
45 RegisterForField(TracePacket::kTrackDescriptorFieldNumber, context);
46 RegisterForField(TracePacket::kThreadDescriptorFieldNumber, context);
47 RegisterForField(TracePacket::kProcessDescriptorFieldNumber, context);
48
49 context->descriptor_pool_->AddFromFileDescriptorSet(
50 kTrackEventDescriptor.data(), kTrackEventDescriptor.size());
51 context->descriptor_pool_->AddFromFileDescriptorSet(
52 kChromeTrackEventDescriptor.data(), kChromeTrackEventDescriptor.size());
53 context->descriptor_pool_->AddFromFileDescriptorSet(
54 kAndroidTrackEventDescriptor.data(), kAndroidTrackEventDescriptor.size());
55 }
56
57 TrackEventModule::~TrackEventModule() = default;
58
TokenizePacket(const TracePacket::Decoder & decoder,TraceBlobView * packet,int64_t packet_timestamp,RefPtr<PacketSequenceStateGeneration> state,uint32_t field_id)59 ModuleResult TrackEventModule::TokenizePacket(
60 const TracePacket::Decoder& decoder,
61 TraceBlobView* packet,
62 int64_t packet_timestamp,
63 RefPtr<PacketSequenceStateGeneration> state,
64 uint32_t field_id) {
65 switch (field_id) {
66 case TracePacket::kTrackEventRangeOfInterestFieldNumber:
67 return tokenizer_.TokenizeRangeOfInterestPacket(std::move(state), decoder,
68 packet_timestamp);
69 case TracePacket::kTrackDescriptorFieldNumber:
70 return tokenizer_.TokenizeTrackDescriptorPacket(std::move(state), decoder,
71 packet_timestamp);
72 case TracePacket::kTrackEventFieldNumber:
73 return tokenizer_.TokenizeTrackEventPacket(std::move(state), decoder,
74 packet, packet_timestamp);
75 case TracePacket::kThreadDescriptorFieldNumber:
76 // TODO(eseckler): Remove once Chrome has switched to TrackDescriptors.
77 return tokenizer_.TokenizeThreadDescriptorPacket(std::move(state),
78 decoder);
79 }
80 return ModuleResult::Ignored();
81 }
82
ParseTracePacketData(const TracePacket::Decoder & decoder,int64_t ts,const TracePacketData &,uint32_t field_id)83 void TrackEventModule::ParseTracePacketData(const TracePacket::Decoder& decoder,
84 int64_t ts,
85 const TracePacketData&,
86 uint32_t field_id) {
87 switch (field_id) {
88 case TracePacket::kTrackDescriptorFieldNumber:
89 parser_.ParseTrackDescriptor(ts, decoder.track_descriptor(),
90 decoder.trusted_packet_sequence_id());
91 break;
92 case TracePacket::kProcessDescriptorFieldNumber:
93 // TODO(eseckler): Remove once Chrome has switched to TrackDescriptors.
94 parser_.ParseProcessDescriptor(ts, decoder.process_descriptor());
95 break;
96 case TracePacket::kThreadDescriptorFieldNumber:
97 // TODO(eseckler): Remove once Chrome has switched to TrackDescriptors.
98 parser_.ParseThreadDescriptor(decoder.thread_descriptor());
99 break;
100 case TracePacket::kTrackEventFieldNumber:
101 PERFETTO_DFATAL("Wrong TracePacket number");
102 }
103 }
104
OnIncrementalStateCleared(uint32_t packet_sequence_id)105 void TrackEventModule::OnIncrementalStateCleared(uint32_t packet_sequence_id) {
106 track_event_tracker_->OnIncrementalStateCleared(packet_sequence_id);
107 }
108
OnFirstPacketOnSequence(uint32_t packet_sequence_id)109 void TrackEventModule::OnFirstPacketOnSequence(uint32_t packet_sequence_id) {
110 track_event_tracker_->OnFirstPacketOnSequence(packet_sequence_id);
111 }
112
ParseTrackEventData(const TracePacket::Decoder & decoder,int64_t ts,const TrackEventData & data)113 void TrackEventModule::ParseTrackEventData(const TracePacket::Decoder& decoder,
114 int64_t ts,
115 const TrackEventData& data) {
116 parser_.ParseTrackEvent(ts, &data, decoder.track_event(),
117 decoder.trusted_packet_sequence_id());
118 }
119
NotifyEndOfFile()120 void TrackEventModule::NotifyEndOfFile() {
121 parser_.NotifyEndOfFile();
122 }
123
124 } // namespace perfetto::trace_processor
125