blob: c158aa5591457468bdd78d071d4bfa2fce5ae264 [file] [log] [blame]
/*
* Copyright (C) 2017 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/traced/probes/ftrace/proto_translation_table.h"
#include <regex.h>
#include <algorithm>
#include "perfetto/base/string_utils.h"
#include "src/traced/probes/ftrace/event_info.h"
#include "src/traced/probes/ftrace/ftrace_procfs.h"
#include "perfetto/trace/ftrace/ftrace_event.pbzero.h"
#include "perfetto/trace/ftrace/ftrace_event_bundle.pbzero.h"
#include "perfetto/trace/ftrace/generic.pbzero.h"
namespace perfetto {
namespace {
using protos::pbzero::GenericFtraceEvent;
ProtoTranslationTable::FtracePageHeaderSpec MakeFtracePageHeaderSpec(
const std::vector<FtraceEvent::Field>& fields) {
ProtoTranslationTable::FtracePageHeaderSpec spec;
for (const FtraceEvent::Field& field : fields) {
std::string name = GetNameFromTypeAndName(field.type_and_name);
if (name == "timestamp")
spec.timestamp = field;
else if (name == "commit")
spec.size = field;
else if (name == "overwrite")
spec.overwrite = field;
else if (name != "data")
PERFETTO_DFATAL("Invalid field in header spec: %s", name.c_str());
}
return spec;
}
const std::vector<Event> BuildEventsVector(const std::vector<Event>& events) {
size_t largest_id = 0;
for (const Event& event : events) {
if (event.ftrace_event_id > largest_id)
largest_id = event.ftrace_event_id;
}
std::vector<Event> events_by_id;
events_by_id.resize(largest_id + 1);
for (const Event& event : events) {
events_by_id[event.ftrace_event_id] = event;
}
events_by_id.shrink_to_fit();
return events_by_id;
}
// Merge the information from |ftrace_field| into |field| (mutating it).
// We should set the following fields: offset, size, ftrace field type and
// translation strategy.
bool MergeFieldInfo(const FtraceEvent::Field& ftrace_field,
Field* field,
const char* event_name_for_debug) {
PERFETTO_DCHECK(field->ftrace_name);
PERFETTO_DCHECK(field->proto_field_id);
PERFETTO_DCHECK(field->proto_field_type);
PERFETTO_DCHECK(!field->ftrace_offset);
PERFETTO_DCHECK(!field->ftrace_size);
PERFETTO_DCHECK(!field->ftrace_type);
if (!InferFtraceType(ftrace_field.type_and_name, ftrace_field.size,
ftrace_field.is_signed, &field->ftrace_type)) {
PERFETTO_FATAL(
"Failed to infer ftrace field type for \"%s.%s\" (type:\"%s\" "
"size:%d "
"signed:%d)",
event_name_for_debug, field->ftrace_name,
ftrace_field.type_and_name.c_str(), ftrace_field.size,
ftrace_field.is_signed);
return false;
}
field->ftrace_offset = ftrace_field.offset;
field->ftrace_size = ftrace_field.size;
if (!SetTranslationStrategy(field->ftrace_type, field->proto_field_type,
&field->strategy)) {
PERFETTO_DLOG(
"Failed to find translation strategy for ftrace field \"%s.%s\" (%s -> "
"%s)",
event_name_for_debug, field->ftrace_name, ToString(field->ftrace_type),
ToString(field->proto_field_type));
// TODO(hjd): Uncomment DCHECK when proto generation is fixed.
// PERFETTO_DCHECK(false);
return false;
}
return true;
}
// For each field in |fields| find the matching field from |ftrace_fields| (by
// comparing ftrace_name) and copy the information from the FtraceEvent::Field
// into the Field (mutating it). If there is no matching field in
// |ftrace_fields| remove the Field from |fields|. Return the maximum observed
// 'field end' (offset + size).
uint16_t MergeFields(const std::vector<FtraceEvent::Field>& ftrace_fields,
std::vector<Field>* fields,
const char* event_name_for_debug) {
uint16_t fields_end = 0;
// Loop over each Field in |fields| modifiying it with information from the
// matching |ftrace_fields| field or removing it.
auto field = fields->begin();
while (field != fields->end()) {
bool success = false;
for (const FtraceEvent::Field& ftrace_field : ftrace_fields) {
if (GetNameFromTypeAndName(ftrace_field.type_and_name) !=
field->ftrace_name)
continue;
success = MergeFieldInfo(ftrace_field, &*field, event_name_for_debug);
uint16_t field_end = field->ftrace_offset + field->ftrace_size;
fields_end = std::max<uint16_t>(fields_end, field_end);
break;
}
if (success) {
++field;
} else {
field = fields->erase(field);
}
}
return fields_end;
}
bool Contains(const std::string& haystack, const std::string& needle) {
return haystack.find(needle) != std::string::npos;
}
std::string RegexError(int errcode, const regex_t* preg) {
char buf[64];
regerror(errcode, preg, buf, sizeof(buf));
return {buf, sizeof(buf)};
}
bool Match(const char* string, const char* pattern) {
regex_t re;
int ret = regcomp(&re, pattern, REG_EXTENDED | REG_NOSUB);
if (ret != 0) {
PERFETTO_FATAL("regcomp: %s", RegexError(ret, &re).c_str());
}
ret = regexec(&re, string, 0, nullptr, 0);
regfree(&re);
return ret != REG_NOMATCH;
}
// Set proto field type and id based on the ftrace type.
void SetProtoType(FtraceFieldType ftrace_type,
ProtoFieldType* proto_type,
uint32_t* proto_field_id) {
switch (ftrace_type) {
case kFtraceCString:
case kFtraceFixedCString:
case kFtraceStringPtr:
case kFtraceDataLoc:
*proto_type = kProtoString;
*proto_field_id = GenericFtraceEvent::Field::kStrValueFieldNumber;
break;
case kFtraceInt8:
case kFtraceInt16:
case kFtraceInt32:
case kFtracePid32:
case kFtraceCommonPid32:
case kFtraceInt64:
*proto_type = kProtoInt64;
*proto_field_id = GenericFtraceEvent::Field::kIntValueFieldNumber;
break;
case kFtraceUint8:
case kFtraceUint16:
case kFtraceUint32:
case kFtraceBool:
case kFtraceDevId32:
case kFtraceDevId64:
case kFtraceUint64:
case kFtraceInode32:
case kFtraceInode64:
*proto_type = kProtoUint64;
*proto_field_id = GenericFtraceEvent::Field::kUintValueFieldNumber;
break;
}
}
} // namespace
// This is similar but different from InferProtoType (see format_parser.cc).
// TODO(hjd): Fold FtraceEvent(::Field) into Event.
bool InferFtraceType(const std::string& type_and_name,
size_t size,
bool is_signed,
FtraceFieldType* out) {
// Fixed length strings: e.g. "char foo[16]" we don't care about the number
// since we get the size as it's own field. Somewhat awkwardly these fields
// are both fixed size and null terminated meaning that we can't just drop
// them directly into the protobuf (since if the string is shorter than 15
// characters we want only the bit up to the null terminator).
if (Match(type_and_name.c_str(), R"(char [a-zA-Z_]+\[[0-9]+\])")) {
*out = kFtraceFixedCString;
return true;
}
// String pointers: "__data_loc char[] foo" (as in
// 'cpufreq_interactive_boost').
// TODO(fmayer): Handle u32[], u8[], __u8[] as well.
if (Contains(type_and_name, "__data_loc char[] ")) {
if (size != 4) {
PERFETTO_ELOG("__data_loc with incorrect size: %s (%zd)",
type_and_name.c_str(), size);
return false;
}
*out = kFtraceDataLoc;
return true;
}
if (Contains(type_and_name, "char[] ")) {
*out = kFtraceStringPtr;
return true;
}
if (Contains(type_and_name, "char * ")) {
*out = kFtraceStringPtr;
return true;
}
// Variable length strings: "char foo" + size: 0 (as in 'print').
if (base::StartsWith(type_and_name, "char ") && size == 0) {
*out = kFtraceCString;
return true;
}
if (base::StartsWith(type_and_name, "bool ")) {
*out = kFtraceBool;
return true;
}
if (base::StartsWith(type_and_name, "ino_t ") ||
base::StartsWith(type_and_name, "i_ino ")) {
if (size == 4) {
*out = kFtraceInode32;
return true;
} else if (size == 8) {
*out = kFtraceInode64;
return true;
}
}
if (base::StartsWith(type_and_name, "dev_t ")) {
if (size == 4) {
*out = kFtraceDevId32;
return true;
} else if (size == 8) {
*out = kFtraceDevId64;
return true;
}
}
// Pids (as in 'sched_switch').
if (base::StartsWith(type_and_name, "pid_t ") && size == 4) {
*out = kFtracePid32;
return true;
}
if (Contains(type_and_name, "common_pid") && size == 4) {
*out = kFtraceCommonPid32;
return true;
}
// Ints of various sizes:
if (size == 1 && is_signed) {
*out = kFtraceInt8;
return true;
} else if (size == 1 && !is_signed) {
*out = kFtraceUint8;
return true;
} else if (size == 2 && is_signed) {
*out = kFtraceInt16;
return true;
} else if (size == 2 && !is_signed) {
*out = kFtraceUint16;
return true;
} else if (size == 4 && is_signed) {
*out = kFtraceInt32;
return true;
} else if (size == 4 && !is_signed) {
*out = kFtraceUint32;
return true;
} else if (size == 8 && is_signed) {
*out = kFtraceInt64;
return true;
} else if (size == 8 && !is_signed) {
*out = kFtraceUint64;
return true;
}
PERFETTO_DLOG("Could not infer ftrace type for '%s'", type_and_name.c_str());
return false;
}
// static
ProtoTranslationTable::FtracePageHeaderSpec
ProtoTranslationTable::DefaultPageHeaderSpecForTesting() {
std::string page_header =
R"( field: u64 timestamp; offset:0; size:8; signed:0;
field: local_t commit; offset:8; size:8; signed:1;
field: int overwrite; offset:8; size:1; signed:1;
field: char data; offset:16; size:4080; signed:0;)";
std::vector<FtraceEvent::Field> page_header_fields;
PERFETTO_CHECK(ParseFtraceEventBody(std::move(page_header), nullptr,
&page_header_fields));
return MakeFtracePageHeaderSpec(page_header_fields);
}
// static
std::unique_ptr<ProtoTranslationTable> ProtoTranslationTable::Create(
const FtraceProcfs* ftrace_procfs,
std::vector<Event> events,
std::vector<Field> common_fields) {
bool common_fields_processed = false;
uint16_t common_fields_end = 0;
std::vector<FtraceEvent::Field> page_header_fields;
std::string page_header = ftrace_procfs->ReadPageHeaderFormat();
PERFETTO_CHECK(!page_header.empty());
PERFETTO_CHECK(ParseFtraceEventBody(std::move(page_header), nullptr,
&page_header_fields));
for (Event& event : events) {
if (event.proto_field_id ==
protos::pbzero::FtraceEvent::kGenericFieldNumber) {
continue;
}
PERFETTO_DCHECK(event.name);
PERFETTO_DCHECK(event.group);
PERFETTO_DCHECK(event.proto_field_id);
PERFETTO_DCHECK(!event.ftrace_event_id);
std::string contents =
ftrace_procfs->ReadEventFormat(event.group, event.name);
FtraceEvent ftrace_event;
if (contents.empty() || !ParseFtraceEvent(contents, &ftrace_event)) {
continue;
}
event.ftrace_event_id = ftrace_event.id;
if (!common_fields_processed) {
common_fields_end =
MergeFields(ftrace_event.common_fields, &common_fields, event.name);
common_fields_processed = true;
}
uint16_t fields_end =
MergeFields(ftrace_event.fields, &event.fields, event.name);
event.size = std::max<uint16_t>(fields_end, common_fields_end);
}
events.erase(std::remove_if(events.begin(), events.end(),
[](const Event& event) {
return event.proto_field_id == 0 ||
event.ftrace_event_id == 0;
}),
events.end());
auto table = std::unique_ptr<ProtoTranslationTable>(
new ProtoTranslationTable(ftrace_procfs, events, std::move(common_fields),
MakeFtracePageHeaderSpec(page_header_fields)));
return table;
}
ProtoTranslationTable::ProtoTranslationTable(
const FtraceProcfs* ftrace_procfs,
const std::vector<Event>& events,
std::vector<Field> common_fields,
FtracePageHeaderSpec ftrace_page_header_spec)
: ftrace_procfs_(ftrace_procfs),
events_(BuildEventsVector(events)),
largest_id_(events_.size() - 1),
common_fields_(std::move(common_fields)),
ftrace_page_header_spec_(ftrace_page_header_spec) {
for (const Event& event : events) {
group_and_name_to_event_[GroupAndName(event.group, event.name)] =
&events_.at(event.ftrace_event_id);
name_to_events_[event.name].push_back(&events_.at(event.ftrace_event_id));
group_to_events_[event.group].push_back(&events_.at(event.ftrace_event_id));
}
}
const Event* ProtoTranslationTable::GetOrCreateEvent(
const GroupAndName& group_and_name) {
const Event* event = GetEvent(group_and_name);
if (event)
return event;
// The ftrace event does not already exist so a new one will be created
// by parsing the format file.
std::string contents = ftrace_procfs_->ReadEventFormat(group_and_name.group(),
group_and_name.name());
if (contents.empty())
return nullptr;
FtraceEvent ftrace_event = {};
ParseFtraceEvent(contents, &ftrace_event);
// Ensure events vector is large enough
if (ftrace_event.id > largest_id_) {
events_.resize(ftrace_event.id + 1);
largest_id_ = ftrace_event.id;
}
// Set known event variables
Event* e = &events_.at(ftrace_event.id);
e->ftrace_event_id = ftrace_event.id;
e->proto_field_id = protos::pbzero::FtraceEvent::kGenericFieldNumber;
e->name = InternString(group_and_name.name());
e->group = InternString(group_and_name.group());
// Calculate size of common fields.
for (const FtraceEvent::Field& ftrace_field : ftrace_event.common_fields) {
uint16_t field_end = ftrace_field.offset + ftrace_field.size;
e->size = std::max(field_end, e->size);
}
// For every field in the ftrace event, make a field in the generic event.
for (const FtraceEvent::Field& ftrace_field : ftrace_event.fields)
e->size = std::max(CreateGenericEventField(ftrace_field, *e), e->size);
group_and_name_to_event_[group_and_name] = &events_.at(e->ftrace_event_id);
name_to_events_[e->name].push_back(&events_.at(e->ftrace_event_id));
group_to_events_[e->group].push_back(&events_.at(e->ftrace_event_id));
return e;
};
const char* ProtoTranslationTable::InternString(const std::string& str) {
auto it_and_inserted = interned_strings_.insert(str);
return it_and_inserted.first->c_str();
};
uint16_t ProtoTranslationTable::CreateGenericEventField(
const FtraceEvent::Field& ftrace_field,
Event& event) {
uint16_t field_end = ftrace_field.offset + ftrace_field.size;
std::string field_name = GetNameFromTypeAndName(ftrace_field.type_and_name);
if (field_name.empty()) {
PERFETTO_DLOG("Field: %s could not be added to the generic event.",
ftrace_field.type_and_name.c_str());
return field_end;
}
event.fields.emplace_back();
Field* field = &event.fields.back();
field->ftrace_name = InternString(field_name);
if (!InferFtraceType(ftrace_field.type_and_name, ftrace_field.size,
ftrace_field.is_signed, &field->ftrace_type)) {
PERFETTO_DLOG(
"Failed to infer ftrace field type for \"%s.%s\" (type:\"%s\" "
"size:%d "
"signed:%d)",
event.name, field->ftrace_name, ftrace_field.type_and_name.c_str(),
ftrace_field.size, ftrace_field.is_signed);
event.fields.pop_back();
return field_end;
}
SetProtoType(field->ftrace_type, &field->proto_field_type,
&field->proto_field_id);
field->ftrace_offset = ftrace_field.offset;
field->ftrace_size = ftrace_field.size;
// Proto type is set based on ftrace type so all fields should have a
// translation strategy.
PERFETTO_DCHECK(SetTranslationStrategy(
field->ftrace_type, field->proto_field_type, &field->strategy));
return field_end;
}
EventFilter::EventFilter() = default;
EventFilter::~EventFilter() = default;
void EventFilter::AddEnabledEvent(size_t ftrace_event_id) {
if (ftrace_event_id >= enabled_ids_.size())
enabled_ids_.resize(ftrace_event_id + 1);
enabled_ids_[ftrace_event_id] = true;
}
void EventFilter::DisableEvent(size_t ftrace_event_id) {
if (ftrace_event_id >= enabled_ids_.size())
return;
enabled_ids_[ftrace_event_id] = false;
}
bool EventFilter::IsEventEnabled(size_t ftrace_event_id) const {
if (ftrace_event_id == 0 || ftrace_event_id > enabled_ids_.size())
return false;
return enabled_ids_[ftrace_event_id];
}
std::set<size_t> EventFilter::GetEnabledEvents() const {
std::set<size_t> enabled;
for (size_t i = 0; i < enabled_ids_.size(); i++) {
if (enabled_ids_[i]) {
enabled.insert(i);
}
}
return enabled;
}
void EventFilter::EnableEventsFrom(const EventFilter& other) {
size_t max_length = std::max(enabled_ids_.size(), other.enabled_ids_.size());
enabled_ids_.resize(max_length);
for (size_t i = 0; i < other.enabled_ids_.size(); i++) {
if (other.enabled_ids_[i])
enabled_ids_[i] = true;
}
}
ProtoTranslationTable::~ProtoTranslationTable() = default;
} // namespace perfetto