xref: /aosp_15_r20/external/perfetto/src/trace_processor/importers/proto/system_probes_module.cc (revision 6dbdd20afdafa5e3ca9b8809fa73465d530080dc)
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 
17 #include "src/trace_processor/importers/proto/system_probes_module.h"
18 #include "perfetto/base/build_config.h"
19 #include "src/trace_processor/importers/proto/packet_sequence_state_generation.h"
20 #include "src/trace_processor/importers/proto/system_probes_parser.h"
21 
22 #include "protos/perfetto/trace/trace_packet.pbzero.h"
23 
24 namespace perfetto {
25 namespace trace_processor {
26 
27 using perfetto::protos::pbzero::TracePacket;
28 
SystemProbesModule(TraceProcessorContext * context)29 SystemProbesModule::SystemProbesModule(TraceProcessorContext* context)
30     : parser_(context) {
31   RegisterForField(TracePacket::kProcessTreeFieldNumber, context);
32   RegisterForField(TracePacket::kProcessStatsFieldNumber, context);
33   RegisterForField(TracePacket::kSysStatsFieldNumber, context);
34   RegisterForField(TracePacket::kSystemInfoFieldNumber, context);
35   RegisterForField(TracePacket::kCpuInfoFieldNumber, context);
36 }
37 
TokenizePacket(const protos::pbzero::TracePacket::Decoder & decoder,TraceBlobView *,int64_t,RefPtr<PacketSequenceStateGeneration>,uint32_t field_id)38 ModuleResult SystemProbesModule::TokenizePacket(
39     const protos::pbzero::TracePacket::Decoder& decoder,
40     TraceBlobView*,
41     int64_t,
42     RefPtr<PacketSequenceStateGeneration>,
43     uint32_t field_id) {
44   switch (field_id) {
45     case TracePacket::kSystemInfoFieldNumber:
46       parser_.ParseSystemInfo(decoder.system_info());
47       return ModuleResult::Handled();
48     case TracePacket::kCpuInfoFieldNumber:
49       parser_.ParseCpuInfo(decoder.cpu_info());
50       return ModuleResult::Handled();
51   }
52   return ModuleResult::Ignored();
53 }
54 
ParseTracePacketData(const TracePacket::Decoder & decoder,int64_t ts,const TracePacketData &,uint32_t field_id)55 void SystemProbesModule::ParseTracePacketData(
56     const TracePacket::Decoder& decoder,
57     int64_t ts,
58     const TracePacketData&,
59     uint32_t field_id) {
60   switch (field_id) {
61     case TracePacket::kProcessTreeFieldNumber:
62       parser_.ParseProcessTree(decoder.process_tree());
63       return;
64     case TracePacket::kProcessStatsFieldNumber:
65       parser_.ParseProcessStats(ts, decoder.process_stats());
66       return;
67     case TracePacket::kSysStatsFieldNumber:
68       parser_.ParseSysStats(ts, decoder.sys_stats());
69       return;
70   }
71 }
72 
73 }  // namespace trace_processor
74 }  // namespace perfetto
75