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