LLVM 的位码错误检测函数参数

LLVM's bitcode wrong detections of function's parameters

本文关键字：检测 函数 参数 错误 LLVM      更新时间：2023-10-16

我正在使用LLVM api来解析位码文件。我有以下代码片段，我正在使用此命令生成位码$CC -emit-llvm -c -g source.c其中 CC 设置为clang路径。

#include <stdio.h>
struct Point {
    int a;
    int b;
};
int func_0(struct Point p, int x) {
    return 0;
}

类型ID 应该有一个基于参数类型的数值。但是，对于整数x和结构Point我都获得了10的值，这称为TokenTyID。因此，我决定分别使用函数isIntegerTy()和isStructTy()，看看至少在这种情况下，我是否获得了正确的结果。此解决方案适用于整数参数 x ，但不适用于结构。如何正确识别结构并读取其字段？

为了完整起见，为了解析位码，我使用以下代码：

using namespace llvm;
int main(int argc, char** argv) {
    LLVMContext context;
    OwningPtr<MemoryBuffer> mb;
    MemoryBuffer::getFile(FileName, mb);
    Module *m = ParseBitcodeFile(mb.get(), context);
    for (Module::const_iterator i = m->getFunctionList().begin(), e = m->getFunctionList().end(); i != e; ++i) {
        if (i->isDeclaration() || i->getName().str() == "main")
            continue;
        std::cout << i->getName().str() << std::endl;
        Type* ret_type = i->getReturnType();
        std::cout << "t(ret) " << ret_type->getTypeID() << std::endl;
        Function::const_arg_iterator ai;
        Function::const_arg_iterator ae;
        for (ai = i->arg_begin(), ae = i->arg_end(); ai != ae; ++ai) {
            Type* t = ai->getType();
            std::cout << "t" << ai->getName().str() << " " << t->getTypeID()
                      << "(" << t->getFunctionNumParams() << ")"
                      << " is struct? " << (t->isStructTy() ? "Y" : "N")
                      << " is int? " << (t->isIntegerTy() ? "Y" : "N")
                      << "n";
        }
    }
    return 0;
}

我读了这篇文章 为什么 Clang 将结构参数强制为 ints 关于 clang 与结构体执行的翻译，我很确定这是我的相同问题。

由于 clang 更改了 IR 中的函数签名，因此您必须使用调试信息获取该信息。下面是一些粗略的代码： 

DITypeIdentifierMap TypeIdentifierMap;
DIType* getLowestDINode(DIType* Ty) {
  if (Ty->getTag() == dwarf::DW_TAG_pointer_type ||
      Ty->getTag() == dwarf::DW_TAG_member) {
    DIType *baseTy =
      dyn_cast<DIDerivedType>(Ty)->getBaseType().resolve(TypeIdentifierMap);
    if (!baseTy) {
      errs() << "Type : NULL - Nothing more to don";
      return NULL;
    }
    //Skip all the DINodes with DW_TAG_typedef tag
    while ((baseTy->getTag() == dwarf::DW_TAG_typedef || baseTy->getTag() == dwarf::DW_TAG_const_type
          || baseTy->getTag() == dwarf::DW_TAG_pointer_type)) {
      if (DITypeRef temp = dyn_cast<DIDerivedType>(baseTy)->getBaseType())
        baseTy = temp.resolve(TypeIdentifierMap);
      else
        break;
    } 
    return baseTy;
  }
  return Ty;
}
int main(int argc, char** argv) {
  LLVMContext context;
  OwningPtr<MemoryBuffer> mb;
  MemoryBuffer::getFile(FileName, mb);
  Module *m = ParseBitcodeFile(mb.get(), context);
  if (NamedMDNode *CU_Nodes = m.getNamedMetadata("llvm.dbg.cu")) {
    TypeIdentifierMap = generateDITypeIdentifierMap(CU_Nodes);
  }
  SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
  F.getAllMetadata(MDs);
  for (auto &MD : MDs) {
    if (MDNode *N = MD.second) {
      if (auto *subRoutine = dyn_cast<DISubprogram>(N)->getType()) {
        if (!subRoutine->getTypeArray()[0]) {
          errs() << "return type "void" for Function : " << F.getName().str()
            << "n";
        }
        const auto &TypeRef = subRoutine->getTypeArray();
        for (int i=0; i<TypeRef.size(); i++) {
          // Resolve the type
          DIType *Ty = ArgTypeRef.resolve(TypeIdentifierMap);
          DIType* baseTy = getLowestDINode(Ty);
          if (!baseTy)
            return;
          // If that pointer is a struct
          if (baseTy->getTag() == dwarf::DW_TAG_structure_type) {
            std::cout << "structure type name: " << baseTy->getName().str() << std::endl();
          }
        }
      }
    }
  }
}

我知道它看起来很丑，但使用调试信息并不容易。

相关文章：