aboutsummaryrefslogtreecommitdiff
path: root/src/trace_processor/importers/proto/translation_table_module.cc
blob: 8684df031e972e437fa1af14ec90f150f22ef6c6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*
 * Copyright (C) 2022 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include "src/trace_processor/importers/proto/translation_table_module.h"

#include "src/trace_processor/importers/common/args_translation_table.h"
#include "src/trace_processor/importers/common/slice_translation_table.h"
#include "src/trace_processor/importers/proto/packet_sequence_state_generation.h"

#include "protos/perfetto/trace/trace_packet.pbzero.h"
#include "protos/perfetto/trace/translation/translation_table.pbzero.h"

namespace perfetto {
namespace trace_processor {

using perfetto::protos::pbzero::TracePacket;

TranslationTableModule::TranslationTableModule(TraceProcessorContext* context)
    : context_(context) {
  RegisterForField(TracePacket::kTranslationTableFieldNumber, context);
}

TranslationTableModule::~TranslationTableModule() = default;

ModuleResult TranslationTableModule::TokenizePacket(
    const protos::pbzero::TracePacket_Decoder& decoder,
    TraceBlobView* /*packet*/,
    int64_t /*packet_timestamp*/,
    RefPtr<PacketSequenceStateGeneration> /*state*/,
    uint32_t field_id) {
  if (field_id != TracePacket::kTranslationTableFieldNumber) {
    return ModuleResult::Ignored();
  }
  const auto translation_table =
      protos::pbzero::TranslationTable::Decoder(decoder.translation_table());
  if (translation_table.has_chrome_histogram()) {
    ParseChromeHistogramRules(translation_table.chrome_histogram());
  } else if (translation_table.has_chrome_user_event()) {
    ParseChromeUserEventRules(translation_table.chrome_user_event());
  } else if (translation_table.has_chrome_performance_mark()) {
    ParseChromePerformanceMarkRules(
        translation_table.chrome_performance_mark());
  } else if (translation_table.has_slice_name()) {
    ParseSliceNameRules(translation_table.slice_name());
  }
  return ModuleResult::Handled();
}

void TranslationTableModule::ParseChromeHistogramRules(
    protozero::ConstBytes bytes) {
  const auto chrome_histogram =
      protos::pbzero::ChromeHistorgramTranslationTable::Decoder(bytes);
  for (auto it = chrome_histogram.hash_to_name(); it; ++it) {
    protos::pbzero::ChromeHistorgramTranslationTable::HashToNameEntry::Decoder
        entry(*it);
    context_->args_translation_table->AddChromeHistogramTranslationRule(
        entry.key(), entry.value());
  }
}

void TranslationTableModule::ParseChromeUserEventRules(
    protozero::ConstBytes bytes) {
  const auto chrome_user_event =
      protos::pbzero::ChromeUserEventTranslationTable::Decoder(bytes);
  for (auto it = chrome_user_event.action_hash_to_name(); it; ++it) {
    protos::pbzero::ChromeUserEventTranslationTable::ActionHashToNameEntry::
        Decoder entry(*it);
    context_->args_translation_table->AddChromeUserEventTranslationRule(
        entry.key(), entry.value());
  }
}

void TranslationTableModule::ParseChromePerformanceMarkRules(
    protozero::ConstBytes bytes) {
  const auto chrome_performance_mark =
      protos::pbzero::ChromePerformanceMarkTranslationTable::Decoder(bytes);
  for (auto it = chrome_performance_mark.site_hash_to_name(); it; ++it) {
    protos::pbzero::ChromePerformanceMarkTranslationTable::SiteHashToNameEntry::
        Decoder entry(*it);
    context_->args_translation_table
        ->AddChromePerformanceMarkSiteTranslationRule(entry.key(),
                                                      entry.value());
  }
  for (auto it = chrome_performance_mark.mark_hash_to_name(); it; ++it) {
    protos::pbzero::ChromePerformanceMarkTranslationTable::MarkHashToNameEntry::
        Decoder entry(*it);
    context_->args_translation_table
        ->AddChromePerformanceMarkMarkTranslationRule(entry.key(),
                                                      entry.value());
  }
}

void TranslationTableModule::ParseSliceNameRules(protozero::ConstBytes bytes) {
  const auto slice_name =
      protos::pbzero::SliceNameTranslationTable::Decoder(bytes);
  for (auto it = slice_name.raw_to_deobfuscated_name(); it; ++it) {
    protos::pbzero::SliceNameTranslationTable::RawToDeobfuscatedNameEntry::
        Decoder entry(*it);
    context_->slice_translation_table->AddNameTranslationRule(entry.key(),
                                                              entry.value());
  }
}

}  // namespace trace_processor
}  // namespace perfetto