如何获取 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\00", align 1
@.str.2 = private unnamed_addr constant [3 x i8] c"NO\00", 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\00", align 1
@.str.2 = private unnamed_addr constant [3 x i8] c"NO\00", align 1

我知道@.str 不像@end 那样是操作数的一部分。 @.str 是被其中一个操作数引用的东西。

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

Answer 1

在您的情况下，选择指令具有 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";
            }
        }
    }
}