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

Question

在下面的示例中，我试图提取每条指令引用的所有全局变量，但我始终遗漏了其中一些。

... // 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"YES[=11=]", align 1
@.str.2 = private unnamed_addr constant [3 x i8] c"NO[=11=]", 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"YES[=12=]", align 1
@.str.2 = private unnamed_addr constant [3 x i8] c"NO[=12=]", align 1

我知道 @.str 不像 @end 那样是操作数的一部分。 @.str 是操作数之一的引用。

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

Answer 1

在您的例子中，select 指令具有 GetElementPtr operators。您需要扩展代码以检测这种情况，然后遍历运算符的参数（指针和）。我在下面建议对您的 for 循环进行扩展，目前正在打印这些其他案例。

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";
            }
        }
    }
}

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

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

c++

compiler-construction

llvm