Home | History | Annotate | Line # | Download | only in Basic 
      1 //===--- XRayLists.h - XRay automatic attribution ---------------*- C++ -*-===//
      2 //
      3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
      4 // See https://llvm.org/LICENSE.txt for license information.
      5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
      6 //
      7 //===----------------------------------------------------------------------===//
      8 //
      9 // User-provided filters for always/never XRay instrumenting certain functions.
     10 //
     11 //===----------------------------------------------------------------------===//
     12 #ifndef LLVM_CLANG_BASIC_XRAYLISTS_H
     13 #define LLVM_CLANG_BASIC_XRAYLISTS_H
     14 
     15 #include "clang/Basic/LLVM.h"
     16 #include "clang/Basic/SourceLocation.h"
     17 #include "llvm/ADT/ArrayRef.h"
     18 #include "llvm/ADT/StringRef.h"
     19 #include <memory>
     20 
     21 namespace llvm {
     22 class SpecialCaseList;
     23 }
     24 
     25 namespace clang {
     26 
     27 class SourceManager;
     28 
     29 class XRayFunctionFilter {
     30   std::unique_ptr<llvm::SpecialCaseList> AlwaysInstrument;
     31   std::unique_ptr<llvm::SpecialCaseList> NeverInstrument;
     32   std::unique_ptr<llvm::SpecialCaseList> AttrList;
     33   SourceManager &SM;
     34 
     35 public:
     36   XRayFunctionFilter(ArrayRef<std::string> AlwaysInstrumentPaths,
     37                      ArrayRef<std::string> NeverInstrumentPaths,
     38                      ArrayRef<std::string> AttrListPaths, SourceManager &SM);
     39   ~XRayFunctionFilter();
     40 
     41   enum class ImbueAttribute {
     42     NONE,
     43     ALWAYS,
     44     NEVER,
     45     ALWAYS_ARG1,
     46   };
     47 
     48   ImbueAttribute shouldImbueFunction(StringRef FunctionName) const;
     49 
     50   ImbueAttribute
     51   shouldImbueFunctionsInFile(StringRef Filename,
     52                              StringRef Category = StringRef()) const;
     53 
     54   ImbueAttribute shouldImbueLocation(SourceLocation Loc,
     55                                      StringRef Category = StringRef()) const;
     56 };
     57 
     58 } // namespace clang
     59 
     60 #endif
     61