cppcheck/tools/clang-ast.cpp

102 lines
3.4 KiB
C++
Raw Normal View History

// Dump the AST for a file.
//
// Compile with:
// g++ `llvm-config-3.8 --cxxflags --ldflags` -lclang -o clang-ast clang-ast.cpp
#include <iostream>
#include <cstdlib>
#include <clang-c/Index.h>
std::ostream& operator<<(std::ostream& stream, const CXString& str)
{
stream << clang_getCString(str);
clang_disposeString(str);
return stream;
}
int main(int argc, char **argv)
{
if (argc == 1) {
std::cerr << "No source file\n";
return EXIT_FAILURE;
}
const char *argv1[] = { argv[0], "-std=c++11" };
CXIndex index = clang_createIndex(0, 0);
CXTranslationUnit unit = clang_parseTranslationUnit(
index,
argv[1], argv1, sizeof(argv1) / sizeof(argv1[0]),
nullptr, 0,
CXTranslationUnit_None);
if (unit == nullptr) {
std::cerr << "Unable to parse translation unit. Quitting." << std::endl;
return EXIT_FAILURE;
}
std::cout << "<?xml version=\"1.0\"?>\n"
<< "<clang-ast>\n";
CXCursor cursor = clang_getTranslationUnitCursor(unit);
clang_visitChildren(
cursor,
[](CXCursor c, CXCursor parent, CXClientData client_data) {
switch (clang_getCursorKind(c)) {
case CXCursor_FunctionDecl:
case CXCursor_Constructor:
case CXCursor_Destructor:
case CXCursor_CXXMethod: {
CXSourceLocation location = clang_getCursorLocation(c);
CXString filename;
unsigned int line, column;
clang_getPresumedLocation(location, &filename, &line, &column);
std::cout << "<Function" /* << clang_getCursorKindSpelling(clang_getCursorKind(c)) */
<< " name=\"" << clang_getCursorSpelling(c) << '\"'
<< " filename=\"" << filename << '\"'
<< " line=\"" << line << '\"'
// << " column=\"" << column << '\"'
<< "/>\n";
break;
}
case CXCursor_CallExpr: {
CXSourceLocation location = clang_getCursorLocation(c);
CXString filename;
unsigned int line, column;
clang_getPresumedLocation(location, &filename, &line, &column);
CXCursor ref = clang_getCursorReferenced(c);
CXSourceLocation locationRef = clang_getCursorLocation(ref);
CXString filenameRef;
unsigned int lineRef, columnRef;
clang_getPresumedLocation(locationRef, &filenameRef, &lineRef, &columnRef);
std::cout << "<CallExpr"
<< " name=\"" << clang_getCursorSpelling(c) << '\"'
<< " filename=\"" << filename << '\"'
<< " line=\"" << line << '\"'
// << " column=\"" << column << '\"'
<< " filenameRef=\"" << filenameRef << '\"'
<< " lineRef=\"" << lineRef << '\"'
<< "/>\n";
break;
}
break;
default:
//std::cout << "<" << clang_getCursorKindSpelling(clang_getCursorKind(c)) << " kind=\"" << clang_getCursorKind(c) << "\"/>\n";
break;
};
return CXChildVisit_Recurse;
},
nullptr);
std::cout << "</clang-ast>\n";
clang_disposeTranslationUnit(unit);
clang_disposeIndex(index);
return EXIT_SUCCESS;
}