c++ - 将尾调用 void (i32, ...) bitcast 转换为 llvm::Function 以获取 FnAttribute

Question

我有以下 LLVM IR 代码:

 //Code snippet #1
; <label>:30:                                     ; preds = %30, %18
  tail call void (i32, ...) bitcast (void (...)* @delay to void (i32, ...)*)(i32 5) #3
  %31 = tail call zeroext i1 (...) @testFunc() #3
  br i1 %31, label %30, label %32

首先是对 Delay 函数的函数调用。这是一个无效函数，但我向其添加了 FnAttributes。我对 testFunc 做了同样的事情，但这个返回一个 1 位整数( bool 值)。

像这样遍历代码时:

//Code snippet #2
for (llvm::Instruction& inst : BB)
{
   llvm::CallInst * callInst = llvm::dyn_cast<llvm::CallInst>(&inst);

    if (callInst == nullptr)
        continue;

    if (llvm::Function *calledFunction = callInst->getCalledFunction()) {
        if (calledFunction->hasFnAttribute("MyAttr")) {
            llvm::outs() << calledFunction.getFnAttribute("MyAttr").getValueAsString().str() << "\n";
        } 
    }

在上面的代码中获取属性(并将其打印到控制台)适用于 testFunc 但不适用于 void 函数。我需要能够将 void 函数转换为 llvm::Function 才能获得 FnAttribute。

使用类似的东西

callInst->getCalledValue()

给我值，但我不能将它转换为 llvm::Function。

如果我使用以下代码完成所有功能:

//Code snippet #3
for (llvm::Function& F : M)
{
    if (F.hasFnAttribute("MyAttr"))
        {
            llvm::outs() << F.getFnAttribute("MyAttr").getValueAsString().str() << "\n";
        }
}

我得到了延迟函数的 FnAttribute 就好了。

我的问题是:如何将此代码中的“延迟”函数转换为 llvm::Function，以便在代码片段 #2 中获取 FnAttribute？

非常感谢!

Answer 1

经过一番搜索，我找到了解决我自己问题的方法。

对于那些在这里寻找答案的人来说。我还找到了一种获取类型的方法。它包含在下面的代码中。

for (llvm::Instruction& inst : BB)
{
    llvm::CallInst * callInst = llvm::dyn_cast<llvm::CallInst>(&inst);

    if (callInst == nullptr)
           continue;

    //this prints out "normal functions"
    if (llvm::Function *calledFunction = callInst->getCalledFunction()) {
                    llvm::outs() << "Call to => " << calledFunction->getName().str() << "\n";
    }
    //this prints out the void bitcasted functions
    else if (llvm::Function *voidFunc = llvm::dyn_cast<llvm::Function>(callInst->getCalledValue()->stripPointerCasts())) {
                     llvm::outs() << "Call to => " << voidFunc ->getName().str() << "\n";
    }

    llvm::Type* t = callInst->getCalledValue()->getType();
    if (llvm::FunctionType* ft = llvm::cast<llvm::FunctionType>(llvm::cast<llvm::PointerType>(t)->getElementType()))
    { 
    //here is your function type
    }

}

解释:

1. 首先我们将指令转换为 Callinstruction (Callinst)。
2. 如果它是 nullptr 则继续下一次迭代。
3. 尝试将其转换为函数指针。
4. 如果它是位转换指针，请尝试转换它。 (这解决了我的问题)。
5. 要获取类型，您可以使用 getCalledValue()->getType() 函数并将其转换为 FunctionType