AST访问者函数调用表达式无法正确识别函数调用

时间:2016-07-08 15:02:00

标签: c++ abstract-syntax-tree llvm-clang

你好我已经实现了一个AST访问器,它工作得很好,它可以在控制台中打印我想要的信息,如变量声明,函数声明和函数调用。今天,在我进行实验时,我遇到了一个函数调用,它不被识别为函数调用。 Syntacticaly与函数调用相同。这是代码:

void  
TIFFError(const char* module, const char* fmt, ...)  
{  
   va_list ap;
   va_start(ap, fmt);    <------------------------------ THIS IS THE FUNCTION CALL
   if (_TIFFerrorHandler)
      (*_TIFFerrorHandler)(module, fmt, ap);
   if (_TIFFerrorHandlerExt)
      (*_TIFFerrorHandlerExt)(0, module, fmt, ap);
   va_end(ap);            <--------------------------------AND THIS ONE
}

我的ASTvisitor代码是:

bool VisitStmt(Stmt *st)
{
    FullSourceLoc FullLocation = astContext->getFullLoc(st->getLocStart());
    FileID fileID = FullLocation.getFileID();
    unsigned int thisFileID = fileID.getHashValue();
    if(thisFileID == 1) //checks if the node is in the main = input file.
    {
        if (CallExpr *call = dyn_cast<CallExpr>(st))
        {
            numFuncCalls++;
            //call->dump(); //prints the corresponding line of the AST.
            FunctionDecl *func_decl;
            if(call->getDirectCallee())
            {
                func_decl = call ->getDirectCallee();
                string funcCall = func_decl->getNameInfo().getName().getAsString();
                cout << "Function call: " << funcCall << " with arguments ";
                APIs << funcCall << ",";
                for(int i=0, j = call->getNumArgs(); i<j; i++)
                {
                    //For each argument it prints its type. The function must be declared otherwise it will return int-for unknown argument type.
                    APIs << call->getArg(i)->getType().getAsString()<< ",";
                    cout << call->getArg(i)->getType().getAsString() << ", ";
                }
                cout << "\n";
            }
            else
            {
                Expr *expr = call->getCallee();
                string exprCall = expr->getStmtClassName();
                cout << "Expression call: " << exprCall << " with arguments ";
                APIs << exprCall << ",";
                for(int i=0, j = call->getNumArgs(); i<j; i++)
                {
                    //For each argument it prints its type. The function must be declared otherwise it will return int-for unknown argument type.
                    APIs << call->getArg(i)->getType().getAsString()<< ",";
                    cout << call->getArg(i)->getType().getAsString() << ", ";
                }
                cout << "\n";
            }
        }
    }
    return true;
}

表达式if(call-&gt; getDirectCallee())对于这些调用不正确。

如何提取“函数名称”及其参数,就像我正在使用“普通”函数调用一样? 或者甚至有人告诉我为什么AST递归访问者不能将这些调用识别为普通函数调用。

谢谢。

1 个答案:

答案 0 :(得分:0)

它没有作为一个函数显示给你的原因是因为它实际上不是一个函数。如果你查看<cstdarg>头文件,你会发现该函数是一个MACRO

#define va_start(ap, param) __builtin_va_start(ap, param)

现在,它被称为LinkageSpecDecl,它将链接到它指向的实际functiondecl。

您可以采取的措施来解决此类问题,即查看您要解析的代码的原始ASTDump,因为它会告诉您预期的结果。

例如,我将您的功能更改为此。

#include <cstdarg>

void TIFFError(const char* module, const char* fmt, ...)  
{  
    va_list ap;
    va_start(ap, fmt);    <------------------------------ THIS IS THE FUNCTION CALL
    if (_TIFFerrorHandler)
        (*_TIFFerrorHandler)(module, fmt, ap);
    if (_TIFFerrorHandlerExt)
        (*_TIFFerrorHandlerExt)(0, module, fmt, ap);
    va_end(ap);            <--------------------------------AND THIS ONE
}

然后使用生成它的ast转储(假设上面的代码存储在temp.cpp中):

clang -Xclang -ast-dump -fsyntax-only temp.cpp 

另一个指针将代替visitStmt并检查functiondecl,你可以做的是实现visitFuncDecl,visitDeclRef和其他,因为它会单独访问它们。你不需要做任何演员和维护案件,因为一切都会以适当的功能来到你身边。您可以阅读有关访问者模式如何工作的更多信息。我没有很好的联系,否则我也会给你。