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