使用clang API处理C++函数以插入代码

#include "clang/AST/AST.h"
#include "clang/AST/ASTConsumer.h"
#include "clang/ASTMatchers/ASTMatchers.h"
#include "clang/ASTMatchers/ASTMatchFinder.h"
#include "clang/Frontend/FrontendPluginRegistry.h"

using namespace clang;
using namespace ast_matchers;

class MyASTConsumer : public ASTConsumer {
public:
  MyASTConsumer(ASTContext &Context)
      : Visitor(Context) {}

  void HandleTranslationUnit(ASTContext &Context) override {
    Visitor.TraverseDecl(Context.getTranslationUnitDecl());
  }

private:
  class FindFunctionVisitor : public RecursiveASTVisitor<FindFunctionVisitor> {
  public:
    FindFunctionVisitor(ASTContext &Context) : Context(Context) {}

    bool VisitFunctionDecl(FunctionDecl *Decl) {
      // 在这里处理函数
      return true;
    }

  private:
    ASTContext &Context;
  };

  FindFunctionVisitor Visitor;
};

class MyFrontendAction : public ASTFrontendAction {
public:
  MyFrontendAction() {}

  std::unique_ptr<ASTConsumer> CreateASTConsumer(CompilerInstance &CI, StringRef file) override {
    return std::make_unique<MyASTConsumer>(CI.getASTContext());
  }
};

static FrontendPluginRegistry::Add<MyFrontendAction>
X("my-plugin", "My frontend plugin");

clang++ -std=c++11 -shared -fPIC my-plugin.cpp -o libmy-plugin.so

clang++ -Xclang -load -Xclang libmy-plugin.so my-code.cpp