bolt/lib/Passes/FixRISCVCallsPass.cpp

   1 #include "bolt/Passes/FixRISCVCallsPass.h"
   2 #include "bolt/Core/ParallelUtilities.h"
   3
   4 #include <iterator>
   5
   6 using namespace llvm;
   7
   8 namespace llvm {
   9 namespace bolt {
  10
  11 void FixRISCVCallsPass::runOnFunction(BinaryFunction &BF) {
  12   auto &BC = BF.getBinaryContext();
  13   auto &MIB = BC.MIB;
  14   auto *Ctx = BC.Ctx.get();
  15
  16   for (auto &BB : BF) {
  17     for (auto II = BB.begin(); II != BB.end();) {
  18       if (MIB->isCall(*II) && !MIB->isIndirectCall(*II)) {
  19         auto *Target = MIB->getTargetSymbol(*II);
  20         assert(Target && "Cannot find call target");
  21
  22         MCInst OldCall = *II;
  23         auto L = BC.scopeLock();
  24
  25         if (MIB->isTailCall(*II))
  26           MIB->createTailCall(*II, Target, Ctx);
  27         else
  28           MIB->createCall(*II, Target, Ctx);
  29
  30         MIB->moveAnnotations(std::move(OldCall), *II);
  31         ++II;
  32         continue;
  33       }
  34
  35       auto NextII = std::next(II);
  36
  37       if (NextII == BB.end())
  38         break;
  39
  40       if (MIB->isRISCVCall(*II, *NextII)) {
  41         auto *Target = MIB->getTargetSymbol(*II);
  42         assert(Target && "Cannot find call target");
  43
  44         MCInst OldCall = *NextII;
  45         auto L = BC.scopeLock();
  46
  47         if (MIB->isTailCall(*NextII))
  48           MIB->createTailCall(*II, Target, Ctx);
  49         else
  50           MIB->createCall(*II, Target, Ctx);
  51
  52         MIB->moveAnnotations(std::move(OldCall), *II);
  53
  54         // The original offset was set on the jalr of the auipc+jalr pair. Since
  55         // the whole pair is replaced by a call, adjust the offset by -4 (the
  56         // size of a auipc).
  57         if (std::optional<uint32_t> Offset = MIB->getOffset(*II)) {
  58           assert(*Offset >= 4 && "Illegal jalr offset");
  59           MIB->setOffset(*II, *Offset - 4);
  60         }
  61
  62         II = BB.eraseInstruction(NextII);
  63         continue;
  64       }
  65
  66       ++II;
  67     }
  68   }
  69 }
  70
  71 void FixRISCVCallsPass::runOnFunctions(BinaryContext &BC) {
  72   if (!BC.isRISCV() || !BC.HasRelocations)
  73     return;
  74
  75   ParallelUtilities::WorkFuncTy WorkFun = [&](BinaryFunction &BF) {
  76     runOnFunction(BF);
  77   };
  78
  79   ParallelUtilities::runOnEachFunction(
  80       BC, ParallelUtilities::SchedulingPolicy::SP_INST_LINEAR, WorkFun, nullptr,
  81       "FixRISCVCalls");
  82 }
  83
  84 } // namespace bolt
  85 } // namespace llvm