Question

我正在使用LLVM api来解析bitcode文件。我有以下代码片段，我正在使用此命令生成bitcode $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;
}

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

为了完整，解析bitcode我使用这段代码：

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

我读过这篇文章Why does Clang coerce struct parameters to ints关于clang对结构进行的翻译，我很确定这是同样的问题。

Answer 1

由于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 do\n";
      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();
          }
        }
      }
    }
  }
}

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

LLVM的bitcode错误检测了函数的参数

1 个答案: