利用llvm 源码提供的API,获取IR中函数调用实际参数的名字并打印

在LLVM Pass开发中,由于需要我们可能需要获取IR中函数调用的的实际参数名字,在C语言中函数调用是这样的func1("hello",3),如果在编译器的前端clang中我们想要获取第一个参数"hello",还是比较容易的,但是在当程序被转成IR就类似这样了:

@.str = private unnamed_addr constant [6 x i8] c"hello\00", align 1
//这里还有很多IR指令...
call void @func1(i8* getelementptr inbounds ([6 x i8], [6 x i8]* @.str, i32 0, i32 0), i32 %0)

这样给我们的感觉并不是一目了然,发现第一个参数“hello”变成了i8* getelementptr inbounds ([6 x i8], [6 x i8]* @.str, i32 0, i32 0),我们只能往上捋一下才能找到hello.但是怎么样在llvm pass中让程序找到呢?下面我们首先来分析一下这两行IR。

   对于第二行IR我们发现func1函数的第一个参数位置变成了i8* getelementptr inbounds ([6 x i8], [6 x i8]* @.str, i32 0, i32 0),函数调用在llvm中是CallInst,然后我们去llvm源码提供的文档中找这个类如下图所示:

利用llvm 源码提供的API,获取IR中函数调用实际参数的名字并打印_第1张图片

根据该类提供的getArgOperand(0)方法获得第一个参数,Value *op0 = CInst->getArgOperand(0);然后将op0转成ConstantExpr,

auto constExpr = dyn_cast(op0);

这些转换和调用的方法在llvm源码文档中都能查到。然后再次利用getOperand方法该方法的底层实现如下:

   Value *getOperand(unsigned i) const {
     assert(i < NumUserOperands && "getOperand() out of range!");
     return getOperandList()[i];
   }

auto var = dyn_cast(constExpr->getOperand(0));

最后我们转换成ConstantDataArray

auto a5 = dyn_cast(var->getInitializer());

然后获取String:

auto a6 = a5->getAsString();

完整的函数如下:

  std::string getCallArgName(CallInst *CInst){
    Value *op0 = CInst->getArgOperand(0);
                  if (isa(op0)) {
                 auto constExpr = dyn_cast(op0);
                 if (constExpr->isGEPWithNoNotionalOverIndexing() ){
                        if (constExpr->getOpcode() == Instruction::GetElementPtr){
                                if (isa(constExpr->getOperand(0))){
                                        auto var = dyn_cast(constExpr->getOperand(0));
                                        auto a5 = dyn_cast(var->getInitializer());
                                        if (a5) {
                                                auto a6 = a5->getAsString();
                                                 return a6;
                                               // errs() << "String: " << a6 << "\n";
                                        }
                                }
                        }
                }
              }
     return NULL;
}

 

你可能感兴趣的:(编译器,llvm/clang,llvm,pass,llvm,clang,编译器,编译器后端)