blob: 1d3846c472abe8a8b91a5539e0f1b0616073bbf1 [file] [log] [blame]
/*
* Copyright (C) 2018 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 "tools/trace_to_text/trace_to_profile.h"
#include <string>
#include <vector>
#include "perfetto/base/build_config.h"
#include "src/profiling/symbolizer/symbolize_database.h"
#if PERFETTO_BUILDFLAG(PERFETTO_LOCAL_SYMBOLIZER)
#include "src/profiling/symbolizer/local_symbolizer.h"
#endif
#include "tools/trace_to_text/utils.h"
#include "perfetto/base/logging.h"
#include "perfetto/ext/base/file_utils.h"
#include "perfetto/ext/base/temp_file.h"
#include "perfetto/ext/base/utils.h"
#include "perfetto/profiling/pprof_builder.h"
#include "src/profiling/symbolizer/symbolizer.h"
namespace {
constexpr const char* kDefaultTmp = "/tmp";
std::string GetTemp() {
const char* tmp = getenv("TMPDIR");
if (tmp == nullptr)
tmp = kDefaultTmp;
return tmp;
}
} // namespace
namespace perfetto {
namespace trace_to_text {
int TraceToProfile(std::istream* input,
std::ostream* output,
uint64_t pid,
std::vector<uint64_t> timestamps) {
std::unique_ptr<profiling::Symbolizer> symbolizer;
auto binary_path = profiling::GetPerfettoBinaryPath();
if (!binary_path.empty()) {
#if PERFETTO_BUILDFLAG(PERFETTO_LOCAL_SYMBOLIZER)
symbolizer.reset(new profiling::LocalSymbolizer(std::move(binary_path)));
#else
PERFETTO_ELOG(
"This build does not support local symbolization. "
"Continuing without symbolization.");
#endif
}
std::vector<SerializedProfile> profiles;
TraceToPprof(input, &profiles, symbolizer.get(), pid, timestamps);
if (profiles.empty()) {
return 0;
}
std::string temp_dir = GetTemp() + "/heap_profile-XXXXXXX";
PERFETTO_CHECK(mkdtemp(&temp_dir[0]));
size_t itr = 0;
for (const auto& profile : profiles) {
std::string filename = temp_dir + "/heap_dump." + std::to_string(++itr) +
"." + std::to_string(profile.pid) + ".pb";
base::ScopedFile fd(base::OpenFile(filename, O_CREAT | O_WRONLY, 0700));
if (!fd)
PERFETTO_FATAL("Failed to open %s", filename.c_str());
PERFETTO_CHECK(base::WriteAll(*fd, profile.serialized.c_str(),
profile.serialized.size()) ==
static_cast<ssize_t>(profile.serialized.size()));
}
*output << "Wrote profiles to " << temp_dir << std::endl;
return 0;
}
} // namespace trace_to_text
} // namespace perfetto