| #include "bolt/Profile/DataAggregator.h" |
| #include "bolt/Rewrite/RewriteInstance.h" |
| #include "bolt/Utils/CommandLineOpts.h" |
| #include "llvm/MC/TargetRegistry.h" |
| #include "llvm/Object/Binary.h" |
| #include "llvm/Support/CommandLine.h" |
| #include "llvm/Support/Errc.h" |
| #include "llvm/Support/Error.h" |
| #include "llvm/Support/Path.h" |
| #include "llvm/Support/TargetSelect.h" |
| |
| using namespace llvm; |
| using namespace bolt; |
| |
| namespace opts { |
| |
| static cl::OptionCategory *HeatmapCategories[] = {&HeatmapCategory, |
| &BoltOutputCategory}; |
| |
| static cl::opt<std::string> InputFilename(cl::Positional, |
| cl::desc("<executable>"), |
| cl::Required, |
| cl::cat(HeatmapCategory)); |
| |
| } // namespace opts |
| |
| static StringRef ToolName; |
| |
| static void report_error(StringRef Message, std::error_code EC) { |
| assert(EC); |
| errs() << ToolName << ": '" << Message << "': " << EC.message() << ".\n"; |
| exit(1); |
| } |
| |
| static void report_error(StringRef Message, Error E) { |
| assert(E); |
| errs() << ToolName << ": '" << Message << "': " << toString(std::move(E)) |
| << ".\n"; |
| exit(1); |
| } |
| |
| static std::string GetExecutablePath(const char *Argv0) { |
| SmallString<256> ExecutablePath(Argv0); |
| // Do a PATH lookup if Argv0 isn't a valid path. |
| if (!llvm::sys::fs::exists(ExecutablePath)) |
| if (llvm::ErrorOr<std::string> P = |
| llvm::sys::findProgramByName(ExecutablePath)) |
| ExecutablePath = *P; |
| return std::string(ExecutablePath.str()); |
| } |
| |
| int main(int argc, char **argv) { |
| cl::HideUnrelatedOptions(ArrayRef(opts::HeatmapCategories)); |
| cl::ParseCommandLineOptions(argc, argv, ""); |
| |
| if (opts::PerfData.empty()) { |
| errs() << ToolName << ": expected -perfdata=<filename> option.\n"; |
| exit(1); |
| } |
| |
| opts::HeatmapMode = true; |
| opts::AggregateOnly = true; |
| if (!sys::fs::exists(opts::InputFilename)) |
| report_error(opts::InputFilename, errc::no_such_file_or_directory); |
| |
| // Output to stdout by default |
| if (opts::OutputFilename.empty()) |
| opts::OutputFilename = "-"; |
| |
| // Initialize targets and assembly printers/parsers. |
| llvm::InitializeAllTargetInfos(); |
| llvm::InitializeAllTargetMCs(); |
| llvm::InitializeAllAsmParsers(); |
| llvm::InitializeAllDisassemblers(); |
| |
| llvm::InitializeAllTargets(); |
| llvm::InitializeAllAsmPrinters(); |
| |
| ToolName = argv[0]; |
| std::string ToolPath = GetExecutablePath(argv[0]); |
| Expected<OwningBinary<Binary>> BinaryOrErr = |
| createBinary(opts::InputFilename); |
| if (Error E = BinaryOrErr.takeError()) |
| report_error(opts::InputFilename, std::move(E)); |
| Binary &Binary = *BinaryOrErr.get().getBinary(); |
| |
| if (auto *e = dyn_cast<ELFObjectFileBase>(&Binary)) { |
| auto RIOrErr = |
| RewriteInstance::createRewriteInstance(e, argc, argv, ToolPath); |
| if (Error E = RIOrErr.takeError()) |
| report_error("RewriteInstance", std::move(E)); |
| |
| RewriteInstance &RI = *RIOrErr.get(); |
| if (Error E = RI.setProfile(opts::PerfData)) |
| report_error(opts::PerfData, std::move(E)); |
| |
| if (Error E = RI.run()) |
| report_error(opts::InputFilename, std::move(E)); |
| } else { |
| report_error(opts::InputFilename, object_error::invalid_file_type); |
| } |
| |
| return EXIT_SUCCESS; |
| } |