1 //===-- WinEHPrepare - Prepare exception handling for code generation ---===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This pass lowers LLVM IR exception handling into something closer to what the
11 // backend wants. It snifs the personality function to see which kind of
12 // preparation is necessary. If the personality function uses the Itanium LSDA,
13 // this pass delegates to the DWARF EH preparation pass.
15 //===----------------------------------------------------------------------===//
17 #include "llvm/CodeGen/Passes.h"
18 #include "llvm/Analysis/LibCallSemantics.h"
19 #include "llvm/IR/Function.h"
20 #include "llvm/IR/IRBuilder.h"
21 #include "llvm/IR/Instructions.h"
22 #include "llvm/Pass.h"
25 #define DEBUG_TYPE "winehprepare"
28 class WinEHPrepare : public FunctionPass {
29 FunctionPass *DwarfPrepare;
32 static char ID; // Pass identification, replacement for typeid.
33 WinEHPrepare(const TargetMachine *TM = nullptr)
34 : FunctionPass(ID), DwarfPrepare(createDwarfEHPass(TM)) {}
36 bool runOnFunction(Function &Fn) override;
38 bool doFinalization(Module &M) override;
40 void getAnalysisUsage(AnalysisUsage &AU) const override;
42 const char *getPassName() const override {
43 return "Windows exception handling preparation";
46 } // end anonymous namespace
48 char WinEHPrepare::ID = 0;
49 INITIALIZE_TM_PASS(WinEHPrepare, "winehprepare",
50 "Prepare Windows exceptions", false, false)
52 FunctionPass *llvm::createWinEHPass(const TargetMachine *TM) {
53 return new WinEHPrepare(TM);
56 static bool isMSVCPersonality(EHPersonality Pers) {
57 return Pers == EHPersonality::MSVC_Win64SEH ||
58 Pers == EHPersonality::MSVC_CXX;
61 bool WinEHPrepare::runOnFunction(Function &Fn) {
62 SmallVector<LandingPadInst *, 4> LPads;
63 SmallVector<ResumeInst *, 4> Resumes;
64 for (BasicBlock &BB : Fn) {
65 if (auto *LP = BB.getLandingPadInst())
67 if (auto *Resume = dyn_cast<ResumeInst>(BB.getTerminator()))
68 Resumes.push_back(Resume);
71 // No need to prepare functions that lack landing pads.
75 // Classify the personality to see what kind of preparation we need.
76 EHPersonality Pers = ClassifyEHPersonality(LPads.back()->getPersonalityFn());
78 // Delegate through to the DWARF pass if this is unrecognized.
79 if (!isMSVCPersonality(Pers))
80 return DwarfPrepare->runOnFunction(Fn);
82 // FIXME: Cleanups are unimplemented. Replace them with calls to @llvm.trap.
86 for (ResumeInst *Resume : Resumes) {
87 IRBuilder<>(Resume).CreateUnreachable();
88 Resume->eraseFromParent();
94 bool WinEHPrepare::doFinalization(Module &M) {
95 return DwarfPrepare->doFinalization(M);
98 void WinEHPrepare::getAnalysisUsage(AnalysisUsage &AU) const {
99 DwarfPrepare->getAnalysisUsage(AU);