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