xref: /llvm-project/llvm/include/llvm/Transforms/Instrumentation/LowerAllowCheckPass.h (revision fa9ac62d02fd6b5028f301ee398c3d3a1c0eacae)
1 //===- LowerAllowCheckPass.h ------------------------------------*- 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 /// \file
9 /// This file provides the interface for the pass responsible for removing
10 /// expensive ubsan checks.
11 ///
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_TRANSFORMS_INSTRUMENTATION_LOWERALLOWCHECKPASS_H
15 #define LLVM_TRANSFORMS_INSTRUMENTATION_LOWERALLOWCHECKPASS_H
16 
17 #include "llvm/IR/Function.h"
18 #include "llvm/IR/PassManager.h"
19 #include "llvm/Pass.h"
20 
21 namespace llvm {
22 
23 // This pass is responsible for removing optional traps, like llvm.ubsantrap
24 // from the hot code.
25 class LowerAllowCheckPass : public PassInfoMixin<LowerAllowCheckPass> {
26 public:
27   struct Options {
28     std::vector<unsigned int> cutoffs;
29   };
30 
31   explicit LowerAllowCheckPass(LowerAllowCheckPass::Options Opts)
32       : Opts(std::move(Opts)) {};
33   PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM);
34 
35   static bool IsRequested();
36   void printPipeline(raw_ostream &OS,
37                      function_ref<StringRef(StringRef)> MapClassName2PassName);
38 
39 private:
40   LowerAllowCheckPass::Options Opts;
41 };
42 
43 } // namespace llvm
44 
45 #endif
46