1 | //===- LowerInvoke.cpp - Eliminate Invoke instructions --------------------===// |
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 | // This transformation is designed for use by code generators which do not yet |
10 | // support stack unwinding. This pass converts 'invoke' instructions to 'call' |
11 | // instructions, so that any exception-handling 'landingpad' blocks become dead |
12 | // code (which can be removed by running the '-simplifycfg' pass afterwards). |
13 | // |
14 | //===----------------------------------------------------------------------===// |
15 | |
16 | #include "llvm/Transforms/Utils/LowerInvoke.h" |
17 | #include "llvm/ADT/SmallVector.h" |
18 | #include "llvm/ADT/Statistic.h" |
19 | #include "llvm/IR/Instructions.h" |
20 | #include "llvm/InitializePasses.h" |
21 | #include "llvm/Pass.h" |
22 | #include "llvm/Transforms/Utils.h" |
23 | using namespace llvm; |
24 | |
25 | #define DEBUG_TYPE "lower-invoke" |
26 | |
27 | STATISTIC(NumInvokes, "Number of invokes replaced" ); |
28 | |
29 | namespace { |
30 | class LowerInvokeLegacyPass : public FunctionPass { |
31 | public: |
32 | static char ID; // Pass identification, replacement for typeid |
33 | explicit LowerInvokeLegacyPass() : FunctionPass(ID) { |
34 | initializeLowerInvokeLegacyPassPass(*PassRegistry::getPassRegistry()); |
35 | } |
36 | bool runOnFunction(Function &F) override; |
37 | }; |
38 | } |
39 | |
40 | char LowerInvokeLegacyPass::ID = 0; |
41 | INITIALIZE_PASS(LowerInvokeLegacyPass, "lowerinvoke" , |
42 | "Lower invoke and unwind, for unwindless code generators" , |
43 | false, false) |
44 | |
45 | static bool runImpl(Function &F) { |
46 | bool Changed = false; |
47 | for (BasicBlock &BB : F) |
48 | if (InvokeInst *II = dyn_cast<InvokeInst>(Val: BB.getTerminator())) { |
49 | SmallVector<Value *, 16> CallArgs(II->args()); |
50 | SmallVector<OperandBundleDef, 1> OpBundles; |
51 | II->getOperandBundlesAsDefs(Defs&: OpBundles); |
52 | // Insert a normal call instruction... |
53 | CallInst *NewCall = |
54 | CallInst::Create(Ty: II->getFunctionType(), Func: II->getCalledOperand(), |
55 | Args: CallArgs, Bundles: OpBundles, NameStr: "" , InsertBefore: II->getIterator()); |
56 | NewCall->takeName(V: II); |
57 | NewCall->setCallingConv(II->getCallingConv()); |
58 | NewCall->setAttributes(II->getAttributes()); |
59 | NewCall->setDebugLoc(II->getDebugLoc()); |
60 | II->replaceAllUsesWith(V: NewCall); |
61 | |
62 | // Insert an unconditional branch to the normal destination. |
63 | BranchInst::Create(IfTrue: II->getNormalDest(), InsertBefore: II->getIterator()); |
64 | |
65 | // Remove any PHI node entries from the exception destination. |
66 | II->getUnwindDest()->removePredecessor(Pred: &BB); |
67 | |
68 | // Remove the invoke instruction now. |
69 | II->eraseFromParent(); |
70 | |
71 | ++NumInvokes; |
72 | Changed = true; |
73 | } |
74 | return Changed; |
75 | } |
76 | |
77 | bool LowerInvokeLegacyPass::runOnFunction(Function &F) { |
78 | return runImpl(F); |
79 | } |
80 | |
81 | namespace llvm { |
82 | char &LowerInvokePassID = LowerInvokeLegacyPass::ID; |
83 | |
84 | // Public Interface To the LowerInvoke pass. |
85 | FunctionPass *createLowerInvokePass() { return new LowerInvokeLegacyPass(); } |
86 | |
87 | PreservedAnalyses LowerInvokePass::run(Function &F, |
88 | FunctionAnalysisManager &AM) { |
89 | bool Changed = runImpl(F); |
90 | if (!Changed) |
91 | return PreservedAnalyses::all(); |
92 | |
93 | return PreservedAnalyses::none(); |
94 | } |
95 | } |
96 | |