File Manager

003 File Manager

Current Path: /usr/src/contrib/llvm-project/llvm/lib/IR

usr / src / contrib / llvm-project / llvm / lib / IR /

📁 ..
📄 AbstractCallSite.cpp(5.14 KB)
📄 AsmWriter.cpp(152.8 KB)
📄 AttributeImpl.h(10.49 KB)
📄 Attributes.cpp(63.53 KB)
📄 AutoUpgrade.cpp(188.79 KB)
📄 BasicBlock.cpp(16.46 KB)
📄 Comdat.cpp(2.33 KB)
📄 ConstantFold.cpp(104.22 KB)
📄 ConstantFold.h(2.55 KB)
📄 ConstantRange.cpp(54.53 KB)
📄 Constants.cpp(117.17 KB)
📄 ConstantsContext.h(26.51 KB)
📄 Core.cpp(143.39 KB)
📄 DIBuilder.cpp(43.31 KB)
📄 DataLayout.cpp(31.11 KB)
📄 DebugInfo.cpp(54.5 KB)
📄 DebugInfoMetadata.cpp(52.5 KB)
📄 DebugLoc.cpp(4 KB)
📄 DiagnosticHandler.cpp(3.58 KB)
📄 DiagnosticInfo.cpp(13.72 KB)
📄 DiagnosticPrinter.cpp(2.93 KB)
📄 Dominators.cpp(14.38 KB)
📄 FPEnv.cpp(2.63 KB)
📄 Function.cpp(58.45 KB)
📄 GVMaterializer.cpp(640 B)
📄 Globals.cpp(19.65 KB)
📄 IRBuilder.cpp(44.38 KB)
📄 IRPrintingPasses.cpp(4.37 KB)
📄 InlineAsm.cpp(9.63 KB)
📄 Instruction.cpp(26.63 KB)
📄 Instructions.cpp(165.63 KB)
📄 IntrinsicInst.cpp(11.05 KB)
📄 LLVMContext.cpp(11.43 KB)
📄 LLVMContextImpl.cpp(7.66 KB)
📄 LLVMContextImpl.h(51.94 KB)
📄 LLVMRemarkStreamer.cpp(6.35 KB)
📄 LegacyPassManager.cpp(60.13 KB)
📄 MDBuilder.cpp(11.87 KB)
📄 Mangler.cpp(7.32 KB)
📄 Metadata.cpp(47.29 KB)
📄 MetadataImpl.h(1.43 KB)
📄 Module.cpp(24.01 KB)
📄 ModuleSummaryIndex.cpp(21.49 KB)
📄 Operator.cpp(4.97 KB)
📄 OptBisect.cpp(1.95 KB)
📄 Pass.cpp(8.33 KB)
📄 PassInstrumentation.cpp(701 B)
📄 PassManager.cpp(3.69 KB)
📄 PassRegistry.cpp(4.47 KB)
📄 PassTimingInfo.cpp(9.08 KB)
📄 ProfileSummary.cpp(10.8 KB)
📄 SafepointIRVerifier.cpp(34.4 KB)
📄 Statepoint.cpp(1.51 KB)
📄 SymbolTableListTraitsImpl.h(4.21 KB)
📄 Type.cpp(24.49 KB)
📄 TypeFinder.cpp(5.02 KB)
📄 Use.cpp(1.05 KB)
📄 User.cpp(7.17 KB)
📄 Value.cpp(36.33 KB)
📄 ValueSymbolTable.cpp(4.36 KB)
📄 Verifier.cpp(224.01 KB)

Editing: AbstractCallSite.cpp

//===-- AbstractCallSite.cpp - Implementation of abstract call sites ------===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// This file implements abstract call sites which unify the interface for
// direct, indirect, and callback call sites.
//
// For more information see:
// https://llvm.org/devmtg/2018-10/talk-abstracts.html#talk20
//
//===----------------------------------------------------------------------===//

#include "llvm/IR/AbstractCallSite.h"
#include "llvm/ADT/Statistic.h"
#include "llvm/Support/Debug.h"

using namespace llvm;

#define DEBUG_TYPE "abstract-call-sites"

STATISTIC(NumCallbackCallSites, "Number of callback call sites created");
STATISTIC(NumDirectAbstractCallSites,
          "Number of direct abstract call sites created");
STATISTIC(NumInvalidAbstractCallSitesUnknownUse,
          "Number of invalid abstract call sites created (unknown use)");
STATISTIC(NumInvalidAbstractCallSitesUnknownCallee,
          "Number of invalid abstract call sites created (unknown callee)");
STATISTIC(NumInvalidAbstractCallSitesNoCallback,
          "Number of invalid abstract call sites created (no callback)");

void AbstractCallSite::getCallbackUses(
    const CallBase &CB, SmallVectorImpl<const Use *> &CallbackUses) {
  const Function *Callee = CB.getCalledFunction();
  if (!Callee)
    return;

MDNode *CallbackMD = Callee->getMetadata(LLVMContext::MD_callback);
  if (!CallbackMD)
    return;

for (const MDOperand &Op : CallbackMD->operands()) {
    MDNode *OpMD = cast<MDNode>(Op.get());
    auto *CBCalleeIdxAsCM = cast<ConstantAsMetadata>(OpMD->getOperand(0));
    uint64_t CBCalleeIdx =
        cast<ConstantInt>(CBCalleeIdxAsCM->getValue())->getZExtValue();
    if (CBCalleeIdx < CB.arg_size())
      CallbackUses.push_back(CB.arg_begin() + CBCalleeIdx);
  }
}

/// Create an abstract call site from a use.
AbstractCallSite::AbstractCallSite(const Use *U)
    : CB(dyn_cast<CallBase>(U->getUser())) {

// First handle unknown users.
  if (!CB) {

// If the use is actually in a constant cast expression which itself
    // has only one use, we look through the constant cast expression.
    // This happens by updating the use @p U to the use of the constant
    // cast expression and afterwards re-initializing CB accordingly.
    if (ConstantExpr *CE = dyn_cast<ConstantExpr>(U->getUser()))
      if (CE->hasOneUse() && CE->isCast()) {
        U = &*CE->use_begin();
        CB = dyn_cast<CallBase>(U->getUser());
      }

if (!CB) {
      NumInvalidAbstractCallSitesUnknownUse++;
      return;
    }
  }

// Then handle direct or indirect calls. Thus, if U is the callee of the
  // call site CB it is not a callback and we are done.
  if (CB->isCallee(U)) {
    NumDirectAbstractCallSites++;
    return;
  }

// If we cannot identify the broker function we cannot create a callback and
  // invalidate the abstract call site.
  Function *Callee = CB->getCalledFunction();
  if (!Callee) {
    NumInvalidAbstractCallSitesUnknownCallee++;
    CB = nullptr;
    return;
  }

MDNode *CallbackMD = Callee->getMetadata(LLVMContext::MD_callback);
  if (!CallbackMD) {
    NumInvalidAbstractCallSitesNoCallback++;
    CB = nullptr;
    return;
  }

unsigned UseIdx = CB->getArgOperandNo(U);
  MDNode *CallbackEncMD = nullptr;
  for (const MDOperand &Op : CallbackMD->operands()) {
    MDNode *OpMD = cast<MDNode>(Op.get());
    auto *CBCalleeIdxAsCM = cast<ConstantAsMetadata>(OpMD->getOperand(0));
    uint64_t CBCalleeIdx =
        cast<ConstantInt>(CBCalleeIdxAsCM->getValue())->getZExtValue();
    if (CBCalleeIdx != UseIdx)
      continue;
    CallbackEncMD = OpMD;
    break;
  }

if (!CallbackEncMD) {
    NumInvalidAbstractCallSitesNoCallback++;
    CB = nullptr;
    return;
  }

NumCallbackCallSites++;

assert(CallbackEncMD->getNumOperands() >= 2 && "Incomplete !callback metadata");

unsigned NumCallOperands = CB->getNumArgOperands();
  // Skip the var-arg flag at the end when reading the metadata.
  for (unsigned u = 0, e = CallbackEncMD->getNumOperands() - 1; u < e; u++) {
    Metadata *OpAsM = CallbackEncMD->getOperand(u).get();
    auto *OpAsCM = cast<ConstantAsMetadata>(OpAsM);
    assert(OpAsCM->getType()->isIntegerTy(64) &&
           "Malformed !callback metadata");

int64_t Idx = cast<ConstantInt>(OpAsCM->getValue())->getSExtValue();
    assert(-1 <= Idx && Idx <= NumCallOperands &&
           "Out-of-bounds !callback metadata index");

CI.ParameterEncoding.push_back(Idx);
  }

if (!Callee->isVarArg())
    return;

Metadata *VarArgFlagAsM =
      CallbackEncMD->getOperand(CallbackEncMD->getNumOperands() - 1).get();
  auto *VarArgFlagAsCM = cast<ConstantAsMetadata>(VarArgFlagAsM);
  assert(VarArgFlagAsCM->getType()->isIntegerTy(1) &&
         "Malformed !callback metadata var-arg flag");

if (VarArgFlagAsCM->getValue()->isNullValue())
    return;

// Add all variadic arguments at the end.
  for (unsigned u = Callee->arg_size(); u < NumCallOperands; u++)
    CI.ParameterEncoding.push_back(u);
}

003 File Manager

Editing: AbstractCallSite.cpp

Upload File

Create Folder