如何使用libclang找出成员函数是const还是volatile?

时间:2012-08-19 12:41:21

标签: c++ c clang libclang

我有一个CXCursorCXCursor_CXXMethod的实例。我想知道函数是const还是volatile,例如:

class Foo {
public:
    void bar() const;
    void baz() volatile;
    void qux() const volatile;
};

我在libclang的文档中找不到任何有用的东西。我尝试了clang_isConstQualifiedTypeclang_isVolatileQualifiedType,但这些似乎总是在C ++成员函数类型上返回0

1 个答案:

答案 0 :(得分:18)

我可以想到两种方法:

使用libclang词法分析器

this SO answer中显示的代码对我有用;它使用libclang标记生成器将方法声明分开,然后在方法括号外部记录任何关键字。

访问代码的AST,据我所知,根本不涉及解析器。如果您确定您调查的代码是正确的C ++,我相信这种方法是安全的。

缺点:此解决方案似乎没有考虑预处理指令,因此必须首先处理代码(例如,通过cpp传递)。

示例代码(要解析的文件必须是程序的第一个参数,例如./a.out bla.cpp):

#include "clang-c/Index.h"
#include <string>
#include <set>
#include <iostream>

std::string GetClangString(CXString str)
{
  const char* tmp = clang_getCString(str);
  if (tmp == NULL) {
    return "";
  } else {
    std::string translated = std::string(tmp);
    clang_disposeString(str);
    return translated;
  }
}

void GetMethodQualifiers(CXTranslationUnit translationUnit,
                         std::set<std::string>& qualifiers,
                         CXCursor cursor) {
  qualifiers.clear();

  CXSourceRange range = clang_getCursorExtent(cursor);
  CXToken* tokens;
  unsigned int numTokens;
  clang_tokenize(translationUnit, range, &tokens, &numTokens);

  bool insideBrackets = false;
  for (unsigned int i = 0; i < numTokens; i++) {
    std::string token = GetClangString(clang_getTokenSpelling(translationUnit, tokens[i]));
    if (token == "(") {
      insideBrackets = true;
    } else if (token == "{" || token == ";") {
      break;
    } else if (token == ")") {
      insideBrackets = false;
    } else if (clang_getTokenKind(tokens[i]) == CXToken_Keyword && 
             !insideBrackets) {
      qualifiers.insert(token);
    }
  }

  clang_disposeTokens(translationUnit, tokens, numTokens);
}

int main(int argc, char *argv[]) {
  CXIndex Index = clang_createIndex(0, 0);
  CXTranslationUnit TU = clang_parseTranslationUnit(Index, 0, 
          argv, argc, 0, 0, CXTranslationUnit_None);

  // Set the file you're interested in, and the code location:
  CXFile file = clang_getFile(TU, argv[1]);
  int line = 5;
  int column = 6;
  CXSourceLocation location = clang_getLocation(TU, file, line, column);
  CXCursor cursor = clang_getCursor(TU, location);

  std::set<std::string> qualifiers;
  GetMethodQualifiers(TU, qualifiers, cursor);

  for (std::set<std::string>::const_iterator i = qualifiers.begin(); i != qualifiers.end(); ++i) {
    std::cout << *i << std::endl;
  }

  clang_disposeTranslationUnit(TU);
  clang_disposeIndex(Index);
  return 0;
}

使用libclang的统一符号解析(USR)

这种方法涉及使用解析器本身,并从AST中提取限定符信息。

优点:似乎适用于具有预处理程序指令的代码,至少对于简单情况而言。

缺点:我的解决方案解析了未记录的USR,并且可能在将来发生变化。尽管如此,编写单元测试以防止这种情况仍然很容易。

查看$(CLANG_SRC)/tools/libclang/CIndexUSRs.cpp,它包含生成USR的代码,因此包含解析USR字符串所需的信息。具体而言,第523-529行(在LLVM 3.1的源代码中从www.llvm.org下载)用于限定符部分。

在某处添加以下功能:

void parseUsrString(const std::string& usrString, bool* isVolatile, bool* isConst, bool *isRestrict) {
  size_t bangLocation = usrString.find("#");
  if (bangLocation == std::string::npos || bangLocation == usrString.length() - 1) {
    *isVolatile = *isConst = *isRestrict = false;
    return;
  }
  bangLocation++;
  int x = usrString[bangLocation];

  *isConst = x & 0x1;
  *isVolatile = x & 0x4;
  *isRestrict = x & 0x2;
}

main()

CXString usr = clang_getCursorUSR(cursor);
const char *usr_string = clang_getCString(usr);
std::cout << usr_string << "\n";
bool isVolatile, isConst, isRestrict;
parseUsrString(usr_string, &isVolatile, &isConst, &isRestrict);
printf("restrict, volatile, const: %d %d %d\n", isRestrict, isVolatile, isConst);
clang_disposeString(usr);

从{/ 1>}

运行
Foo::qux()

产生

的预期结果
#define BLA const

class Foo {
public:
    void bar() const;
    void baz() volatile;
    void qux() BLA volatile;
};

警告:您可能已经注意到c:@C@Foo@F@qux#5 restrict, volatile, const: 0 1 1 的源代码我的代码应该是libclang而不是isVolatile = x & 0x2,所以情况可能就是这样应该用0x2替换0x4。我的实现(OS X)可能会替换它们。