File Manager

003 File Manager

Current Path: /usr/src/contrib/llvm-project/llvm/lib/Target/BPF

usr / src / contrib / llvm-project / llvm / lib / Target / BPF /

📁 ..
📁 AsmParser
📄 BPF.h(1.33 KB)
📄 BPF.td(1.63 KB)
📄 BPFAbstractMemberAccess.cpp(33.35 KB)
📄 BPFAsmPrinter.cpp(5.11 KB)
📄 BPFCORE.h(1.16 KB)
📄 BPFCallingConv.td(1.79 KB)
📄 BPFFrameLowering.cpp(1.49 KB)
📄 BPFFrameLowering.h(1.46 KB)
📄 BPFISelDAGToDAG.cpp(16.73 KB)
📄 BPFISelLowering.cpp(28.04 KB)
📄 BPFISelLowering.h(5.33 KB)
📄 BPFInstrFormats.td(2.81 KB)
📄 BPFInstrInfo.cpp(8.23 KB)
📄 BPFInstrInfo.h(2.43 KB)
📄 BPFInstrInfo.td(30.55 KB)
📄 BPFMCInstLower.cpp(2.62 KB)
📄 BPFMCInstLower.h(1.17 KB)
📄 BPFMIChecking.cpp(6.54 KB)
📄 BPFMIPeephole.cpp(15.45 KB)
📄 BPFMISimplifyPatchable.cpp(10.86 KB)
📄 BPFPreserveDIType.cpp(3.93 KB)
📄 BPFRegisterInfo.cpp(3.96 KB)
📄 BPFRegisterInfo.h(1.23 KB)
📄 BPFRegisterInfo.td(1.51 KB)
📄 BPFSelectionDAGInfo.cpp(1.63 KB)
📄 BPFSelectionDAGInfo.h(1.24 KB)
📄 BPFSubtarget.cpp(1.84 KB)
📄 BPFSubtarget.h(2.83 KB)
📄 BPFTargetMachine.cpp(4.43 KB)
📄 BPFTargetMachine.h(1.41 KB)
📄 BTF.def(1015 B)
📄 BTF.h(8.27 KB)
📄 BTFDebug.cpp(41.2 KB)
📄 BTFDebug.h(11.66 KB)
📁 Disassembler
📁 MCTargetDesc
📁 TargetInfo

Editing: BPFPreserveDIType.cpp

//===----------- BPFPreserveDIType.cpp - Preserve DebugInfo Types ---------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// Preserve Debuginfo types encoded in __builtin_btf_type_id() metadata.
//
//===----------------------------------------------------------------------===//

#include "BPF.h"
#include "BPFCORE.h"
#include "llvm/IR/DebugInfoMetadata.h"
#include "llvm/IR/GlobalVariable.h"
#include "llvm/IR/Instruction.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/Type.h"
#include "llvm/IR/User.h"
#include "llvm/IR/Value.h"
#include "llvm/Pass.h"
#include "llvm/Transforms/Utils/BasicBlockUtils.h"

#define DEBUG_TYPE "bpf-preserve-di-type"

namespace llvm {
constexpr StringRef BPFCoreSharedInfo::TypeIdAttr;
} // namespace llvm

using namespace llvm;

namespace {

class BPFPreserveDIType final : public ModulePass {
  StringRef getPassName() const override {
    return "BPF Preserve DebugInfo Type";
  }

bool runOnModule(Module &M) override;

public:
  static char ID;
  BPFPreserveDIType() : ModulePass(ID) {}

private:
  bool doTransformation(Module &M);
};
} // End anonymous namespace

char BPFPreserveDIType::ID = 0;
INITIALIZE_PASS(BPFPreserveDIType, DEBUG_TYPE, "preserve debuginfo type", false,
                false)

ModulePass *llvm::createBPFPreserveDIType() { return new BPFPreserveDIType(); }

bool BPFPreserveDIType::runOnModule(Module &M) {
  LLVM_DEBUG(dbgs() << "********** preserve debuginfo type **********\n");

// Bail out if no debug info.
  if (M.debug_compile_units().empty())
    return false;

return doTransformation(M);
}

bool BPFPreserveDIType::doTransformation(Module &M) {
  std::vector<CallInst *> PreserveDITypeCalls;

for (auto &F : M) {
    for (auto &BB : F) {
      for (auto &I : BB) {
        auto *Call = dyn_cast<CallInst>(&I);
        if (!Call)
          continue;

const auto *GV = dyn_cast<GlobalValue>(Call->getCalledOperand());
        if (!GV)
          continue;

if (GV->getName().startswith("llvm.bpf.btf.type.id")) {
          if (!Call->getMetadata(LLVMContext::MD_preserve_access_index))
            report_fatal_error(
                "Missing metadata for llvm.bpf.btf.type.id intrinsic");
          PreserveDITypeCalls.push_back(Call);
        }
      }
    }
  }

if (PreserveDITypeCalls.empty())
    return false;

std::string BaseName = "llvm.btf_type_id.";
  int Count = 0;
  for (auto Call : PreserveDITypeCalls) {
    const ConstantInt *Flag = dyn_cast<ConstantInt>(Call->getArgOperand(2));
    assert(Flag);
    uint64_t FlagValue = Flag->getValue().getZExtValue();

if (FlagValue >= BPFCoreSharedInfo::MAX_BTF_TYPE_ID_FLAG)
      report_fatal_error("Incorrect flag for llvm.bpf.btf.type.id intrinsic");

uint32_t Reloc;
    if (FlagValue == BPFCoreSharedInfo::BTF_TYPE_ID_LOCAL_RELOC)
      Reloc = BPFCoreSharedInfo::BTF_TYPE_ID_LOCAL;
    else
      Reloc = BPFCoreSharedInfo::BTF_TYPE_ID_REMOTE;

BasicBlock *BB = Call->getParent();
    IntegerType *VarType = Type::getInt32Ty(BB->getContext());
    std::string GVName = BaseName + std::to_string(Count) + "$" +
        std::to_string(Reloc);
    GlobalVariable *GV =
        new GlobalVariable(M, VarType, false, GlobalVariable::ExternalLinkage,
                           NULL, GVName);
    GV->addAttribute(BPFCoreSharedInfo::TypeIdAttr);
    MDNode *MD = Call->getMetadata(LLVMContext::MD_preserve_access_index);
    GV->setMetadata(LLVMContext::MD_preserve_access_index, MD);

// Load the global variable which represents the type info.
    auto *LDInst = new LoadInst(Type::getInt32Ty(BB->getContext()), GV, "",
                                Call);
    Call->replaceAllUsesWith(LDInst);
    Call->eraseFromParent();
    Count++;
  }

return true;
}

003 File Manager

Editing: BPFPreserveDIType.cpp

Upload File

Create Folder