LLVM:从C ++ API检索常量函数参数的值

时间:2015-05-18 23:32:14

标签: c++ llvm

我正在使用LLVM API编写一些代码。我正在使用llvm :: CallGraph对象来遍历父函数调用的所有子函数:

CallGraph cg( *mod );
for( auto i = cg.begin(); i != cg.end(); ++i )
{
    const Function *parent = i->first;
    if( !parent )
        continue;

    for( auto j = i->second->begin(); j != i->second->end(); ++j )
    {
        Function *child = j->second->getFunction();
        if( !child )
            continue;
        for( auto iarg = child->arg_begin(); iarg != child->arg_end(); ++iarg )
        {
            // Print values here!
        }
    }
    o.flush();

我感兴趣的IR中的实际函数调用如下所示:

call void @_ZN3abc3FooC1Ejjj(%"struct.abc::Foo"* %5, i32 4, i32 2, i32 0)

我要做的是检索最后三个常数整数值:4,2,0。如果我也可以检索%5,那么奖励积分,但它并不重要。我花了大约两个小时盯着http://llvm.org/docs/doxygen/,但我根本无法弄清楚我应该如何得到这三个价值。

1 个答案:

答案 0 :(得分:3)

在第二个循环中,当您遍历CallGraphNode时,会返回std::pair<WeakVH, CallGraphNode*>的实例。 WeakVH表示调用指令,而CallGraphNode*表示被调用的函数。您所拥有的代码的问题在于您正在查看被调用函数并在其定义中迭代形式参数而不是查看调用站点。你需要这样的东西(n.b.我还没有测试过这个,只是关闭了签名):

CallGraph cg( *mod );
for( auto i = cg.begin(); i != cg.end(); ++i )
{
    for( auto j = i->second->begin(); j != i->second->end(); ++j )
    {
        CallSite CS(*j->first);
        for (auto arg = CS.arg_begin(); arg != CS.arg_end(); ++arg) {
            // Print values here!
        }
    }
    o.flush();

从那里你可以得到代表参数的Value*指针,检查他们是否ConstantInt等等。