如何获取LLVM中指令引用的所有全局变量

How to get all Global Variables referenced by instruction in LLVM?

本文关键字：引用 指令 全局变量 LLVM 何获取 获取      更新时间：2023-10-16

在示例中，我正在尝试提取每种说明所引用的所有全局变量，但是我一直缺少其中的一些。

... // inside a BB
for (Instruction * I : BB) {
  for (Use &U : I->operands()) {
    if(GlobalVariable * GV = dyn_cast<GlobalVariable >(U)){
      // Do something with GV
      GV->dump();
    }
  }
}

，但是当我定位以下全局值时：

@end = global i8 0, align 1
@.str = private unnamed_addr constant [4 x i8] c"YES0", align 1
@.str.2 = private unnamed_addr constant [3 x i8] c"NO0", align 1
define void @TempF() {
entry:
  %tmp8 = load i8, i8* @end, align 1
  %tmp9 = trunc i8 %tmp8 to i1
  %tmp10 = select i1 %tmp9, i8* getelementptr inbounds ([4 x i8], [4 x i8]* @.str, i32 0, i32 0), i8* getelementptr inbounds ([3 x i8], [3 x i8]* @.str.2, i32 0, i32 0)
  ret void
}

在此IR上运行我的通行证时，只有@end = global i8 0, align 1被打印。但不是... 

@.str = private unnamed_addr constant [4 x i8] c"YES0", align 1
@.str.2 = private unnamed_addr constant [3 x i8] c"NO0", align 1

我知道@.str不像@end那样属于操作数的一部分。@.str是操作数之一的参考。

我必须实现哪些修改才能在操作数中获取全局变量？

在您的情况下，选择指令具有getElementPtr运算符。您将需要扩展代码以检测此情况，然后通过参数（指针和）迭代运算符。我建议下面您的for loop扩展，目前打印出其他这些情况。

for (Use &U : (&*I)->operands()) {
    if(GlobalVariable * GV = dyn_cast<GlobalVariable>(U)){
      // Do something with GV
      GV->dump();
    }
    else if (GEPOperator* gepo = dyn_cast<GEPOperator>(&U))
    {
        errs() << "GEPO - " << *gepo << "n";
        if (GlobalVariable* gv = dyn_cast<GlobalVariable>(gepo->getPointerOperand()))
        {
            errs() << "GV - " << *gv << "n";
        }
        for (auto it = gepo->idx_begin(), et = gepo->idx_end(); it != et; ++it)
        {
            if (GlobalVariable* gv = dyn_cast<GlobalVariable>(*it))
            {
                errs() << "GVi - " << *gv <<  "n";
            }
        }
    }
}
相关文章：