[llvm] r236359 - Re-land "[WinEH] Add an EH registration and state insertion pass for 32-bit x86"

Mehdi Amini mehdi.amini at apple.com
Wed May 6 08:17:14 PDT 2015


> On May 5, 2015, at 7:42 PM, Ahmed Bougacha <ahmed.bougacha at gmail.com> wrote:
> 
> On Tue, May 5, 2015 at 3:56 PM, Mehdi Amini <mehdi.amini at apple.com> wrote:
>> Hi Reid,
>> 
>> You relanded this in r236508 (for some reason I can’t find the email for it?) and it breaks our ASAN bot:
> 
> That looks like the one I had, did r236561 fix it for you?

Yes, thanks Ahmed.

— 
Mehdi



> 
> -Ahmed
> 
>> =================================================================
>> ==8942==ERROR: AddressSanitizer: heap-use-after-free on address 0x60b00000a358 at pc 0x0001179973f2 bp 0x7fff564a1f70 sp 0x7fff564a1f68
>> READ of size 8 at 0x60b00000a358 thread T0
>>    #0 0x1179973f1 in llvm::Value::use_begin() const Value.h:264
>>    #1 0x1153a51bf in llvm::Value::hasNUses(unsigned int) const Value.cpp:97
>>    #2 0x11425508a in (anonymous namespace)::WinEHPrepare::prepareExceptionHandlers(llvm::Function&, llvm::SmallVectorImpl<llvm::LandingPadInst*>&) WinEHPrepare.cpp:924
>>    #3 0x1142474fe in (anonymous namespace)::WinEHPrepare::runOnFunction(llvm::Function&) WinEHPrepare.cpp:370
>>    #4 0x114e575f2 in llvm::FPPassManager::runOnFunction(llvm::Function&) LegacyPassManager.cpp:1537
>>    #5 0x114e58c85 in llvm::FPPassManager::runOnModule(llvm::Module&) LegacyPassManager.cpp:1557
>>    #6 0x114e5d2c0 in (anonymous namespace)::MPPassManager::runOnModule(llvm::Module&) LegacyPassManager.cpp:1615
>>    #7 0x114e5a0a7 in llvm::legacy::PassManagerImpl::run(llvm::Module&) LegacyPassManager.cpp:1722
>>    #8 0x114e6012e in llvm::legacy::PassManager::run(llvm::Module&) LegacyPassManager.cpp:1755
>>    #9 0x109761abc in compileModule(char**, llvm::LLVMContext&) llc.cpp:377
>>    #10 0x109754760 in main llc.cpp:201
>> 
>> 0x60b00000a358 is located 40 bytes inside of 112-byte region [0x60b00000a330,0x60b00000a3a0)
>> freed by thread T0 here:
>>    #0 0x11f1c8879 in wrap__ZdlPv
>>    #1 0x1153a11c0 in llvm::User::operator delete(void*) User.cpp:76
>>    #2 0x114cd892e in llvm::AllocaInst::~AllocaInst() Instructions.cpp:886
>>    #3 0x1142096ac in llvm::ilist_node_traits<llvm::Instruction>::deleteNode(llvm::Instruction*) ilist.h:113
>>    #4 0x11420944c in llvm::iplist<llvm::Instruction, llvm::ilist_traits<llvm::Instruction> >::erase(llvm::ilist_iterator<llvm::Instruction>) ilist.h:466
>>    #5 0x114ca27e4 in llvm::Instruction::eraseFromParent() Instruction.cpp:66
>>    #6 0x118c64cee in rewriteSingleStoreAlloca(llvm::AllocaInst*, (anonymous namespace)::AllocaInfo&, (anonymous namespace)::LargeBlockInfo&, llvm::DominatorTree&, llvm::AliasSetTracker*) PromoteMemoryToRegister.cpp:417
>>    #7 0x118c5af72 in (anonymous namespace)::PromoteMem2Reg::run() PromoteMemoryToRegister.cpp:550
>>    #8 0x118c58b7e in llvm::PromoteMemToReg(llvm::ArrayRef<llvm::AllocaInst*>, llvm::DominatorTree&, llvm::AliasSetTracker*, llvm::AssumptionCache*) PromoteMemoryToRegister.cpp:987
>>    #9 0x114255366 in (anonymous namespace)::WinEHPrepare::prepareExceptionHandlers(llvm::Function&, llvm::SmallVectorImpl<llvm::LandingPadInst*>&) WinEHPrepare.cpp:927
>>    #10 0x1142474fe in (anonymous namespace)::WinEHPrepare::runOnFunction(llvm::Function&) WinEHPrepare.cpp:370
>>    #11 0x114e575f2 in llvm::FPPassManager::runOnFunction(llvm::Function&) LegacyPassManager.cpp:1537
>>    #12 0x114e58c85 in llvm::FPPassManager::runOnModule(llvm::Module&) LegacyPassManager.cpp:1557
>>    #13 0x114e5d2c0 in (anonymous namespace)::MPPassManager::runOnModule(llvm::Module&) LegacyPassManager.cpp:1615
>>    #14 0x114e5a0a7 in llvm::legacy::PassManagerImpl::run(llvm::Module&) LegacyPassManager.cpp:1722
>>    #15 0x114e6012e in llvm::legacy::PassManager::run(llvm::Module&) LegacyPassManager.cpp:1755
>>    #16 0x109761abc in compileModule(char**, llvm::LLVMContext&) llc.cpp:377
>>    #17 0x109754760 in main llc.cpp:201
>> 
>> previously allocated by thread T0 here:
>>    #0 0x11f1c82f9 in wrap__Znwm
>>    #1 0x1153a0c61 in llvm::User::operator new(unsigned long, unsigned int) User.cpp:57
>>    #2 0x1147777e2 in llvm::UnaryInstruction::operator new(unsigned long) InstrTypes.h:101
>>    #3 0x11424a60f in (anonymous namespace)::WinEHPrepare::prepareExceptionHandlers(llvm::Function&, llvm::SmallVectorImpl<llvm::LandingPadInst*>&) WinEHPrepare.cpp:698
>>    #4 0x1142474fe in (anonymous namespace)::WinEHPrepare::runOnFunction(llvm::Function&) WinEHPrepare.cpp:370
>>    #5 0x114e575f2 in llvm::FPPassManager::runOnFunction(llvm::Function&) LegacyPassManager.cpp:1537
>>    #6 0x114e58c85 in llvm::FPPassManager::runOnModule(llvm::Module&) LegacyPassManager.cpp:1557
>>    #7 0x114e5d2c0 in (anonymous namespace)::MPPassManager::runOnModule(llvm::Module&) LegacyPassManager.cpp:1615
>>    #8 0x114e5a0a7 in llvm::legacy::PassManagerImpl::run(llvm::Module&) LegacyPassManager.cpp:1722
>>    #9 0x114e6012e in llvm::legacy::PassManager::run(llvm::Module&) LegacyPassManager.cpp:1755
>>    #10 0x109761abc in compileModule(char**, llvm::LLVMContext&) llc.cpp:377
>>    #11 0x109754760 in main llc.cpp:201
>> 
>> 
>> Thanks,
>> 
>> Mehdi
>> 
>> 
>> 
>> 
>>> On May 1, 2015, at 3:40 PM, Reid Kleckner <reid at kleckner.net> wrote:
>>> 
>>> Author: rnk
>>> Date: Fri May  1 17:40:25 2015
>>> New Revision: 236359
>>> 
>>> URL: http://llvm.org/viewvc/llvm-project?rev=236359&view=rev
>>> Log:
>>> Re-land "[WinEH] Add an EH registration and state insertion pass for 32-bit x86"
>>> 
>>> This reverts commit r236340.
>>> 
>>> Added:
>>>   llvm/trunk/lib/Target/X86/X86WinEHState.cpp
>>>   llvm/trunk/test/CodeGen/X86/win32-eh.ll
>>> Modified:
>>>   llvm/trunk/include/llvm/MC/MCAsmInfo.h
>>>   llvm/trunk/lib/CodeGen/AsmPrinter/AsmPrinter.cpp
>>>   llvm/trunk/lib/CodeGen/WinEHPrepare.cpp
>>>   llvm/trunk/lib/Target/X86/CMakeLists.txt
>>>   llvm/trunk/lib/Target/X86/MCTargetDesc/X86MCAsmInfo.cpp
>>>   llvm/trunk/lib/Target/X86/X86.h
>>>   llvm/trunk/lib/Target/X86/X86TargetMachine.cpp
>>>   llvm/trunk/test/CodeGen/X86/inalloca-invoke.ll
>>> 
>>> Modified: llvm/trunk/include/llvm/MC/MCAsmInfo.h
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/include/llvm/MC/MCAsmInfo.h?rev=236359&r1=236358&r2=236359&view=diff
>>> ==============================================================================
>>> --- llvm/trunk/include/llvm/MC/MCAsmInfo.h (original)
>>> +++ llvm/trunk/include/llvm/MC/MCAsmInfo.h Fri May  1 17:40:25 2015
>>> @@ -37,6 +37,7 @@ enum class EncodingType {
>>>  ARM,     /// Windows NT (Windows on ARM)
>>>  CE,      /// Windows CE ARM, PowerPC, SH3, SH4
>>>  Itanium, /// Windows x64, Windows Itanium (IA-64)
>>> +  X86,     /// Windows x86, uses no CFI, just EH tables
>>>  MIPS = Alpha,
>>> };
>>> }
>>> @@ -506,12 +507,13 @@ public:
>>>  /// frame information to unwind.
>>>  bool usesCFIForEH() const {
>>>    return (ExceptionsType == ExceptionHandling::DwarfCFI ||
>>> -            ExceptionsType == ExceptionHandling::ARM ||
>>> -            ExceptionsType == ExceptionHandling::WinEH);
>>> +            ExceptionsType == ExceptionHandling::ARM || usesWindowsCFI());
>>>  }
>>> 
>>>  bool usesWindowsCFI() const {
>>> -    return ExceptionsType == ExceptionHandling::WinEH;
>>> +    return ExceptionsType == ExceptionHandling::WinEH &&
>>> +           (WinEHEncodingType != WinEH::EncodingType::Invalid &&
>>> +            WinEHEncodingType != WinEH::EncodingType::X86);
>>>  }
>>> 
>>>  bool doesDwarfUseRelocationsAcrossSections() const {
>>> 
>>> Modified: llvm/trunk/lib/CodeGen/AsmPrinter/AsmPrinter.cpp
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/CodeGen/AsmPrinter/AsmPrinter.cpp?rev=236359&r1=236358&r2=236359&view=diff
>>> ==============================================================================
>>> --- llvm/trunk/lib/CodeGen/AsmPrinter/AsmPrinter.cpp (original)
>>> +++ llvm/trunk/lib/CodeGen/AsmPrinter/AsmPrinter.cpp Fri May  1 17:40:25 2015
>>> @@ -266,6 +266,8 @@ bool AsmPrinter::doInitialization(Module
>>>  case ExceptionHandling::WinEH:
>>>    switch (MAI->getWinEHEncodingType()) {
>>>    default: llvm_unreachable("unsupported unwinding information encoding");
>>> +    case WinEH::EncodingType::Invalid:
>>> +      break;
>>>    case WinEH::EncodingType::Itanium:
>>>      ES = new Win64Exception(this);
>>>      break;
>>> 
>>> Modified: llvm/trunk/lib/CodeGen/WinEHPrepare.cpp
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/CodeGen/WinEHPrepare.cpp?rev=236359&r1=236358&r2=236359&view=diff
>>> ==============================================================================
>>> --- llvm/trunk/lib/CodeGen/WinEHPrepare.cpp (original)
>>> +++ llvm/trunk/lib/CodeGen/WinEHPrepare.cpp Fri May  1 17:40:25 2015
>>> @@ -8,9 +8,11 @@
>>> //===----------------------------------------------------------------------===//
>>> //
>>> // This pass lowers LLVM IR exception handling into something closer to what the
>>> -// backend wants. It snifs the personality function to see which kind of
>>> -// preparation is necessary. If the personality function uses the Itanium LSDA,
>>> -// this pass delegates to the DWARF EH preparation pass.
>>> +// backend wants for functions using a personality function from a runtime
>>> +// provided by MSVC. Functions with other personality functions are left alone
>>> +// and may be prepared by other passes. In particular, all supported MSVC
>>> +// personality functions require cleanup code to be outlined, and the C++
>>> +// personality requires catch handler code to be outlined.
>>> //
>>> //===----------------------------------------------------------------------===//
>>> 
>>> @@ -31,7 +33,6 @@
>>> #include "llvm/IR/Module.h"
>>> #include "llvm/IR/PatternMatch.h"
>>> #include "llvm/Pass.h"
>>> -#include "llvm/Support/CommandLine.h"
>>> #include "llvm/Support/Debug.h"
>>> #include "llvm/Support/raw_ostream.h"
>>> #include "llvm/Transforms/Utils/BasicBlockUtils.h"
>>> @@ -922,7 +923,7 @@ bool WinEHPrepare::prepareExceptionHandl
>>>  if (SEHExceptionCodeSlot) {
>>>    if (SEHExceptionCodeSlot->hasNUses(0))
>>>      SEHExceptionCodeSlot->eraseFromParent();
>>> -    else
>>> +    else if (isAllocaPromotable(SEHExceptionCodeSlot))
>>>      PromoteMemToReg(SEHExceptionCodeSlot, *DT);
>>>  }
>>> 
>>> 
>>> Modified: llvm/trunk/lib/Target/X86/CMakeLists.txt
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/CMakeLists.txt?rev=236359&r1=236358&r2=236359&view=diff
>>> ==============================================================================
>>> --- llvm/trunk/lib/Target/X86/CMakeLists.txt (original)
>>> +++ llvm/trunk/lib/Target/X86/CMakeLists.txt Fri May  1 17:40:25 2015
>>> @@ -32,6 +32,7 @@ set(sources
>>>  X86TargetTransformInfo.cpp
>>>  X86VZeroUpper.cpp
>>>  X86FixupLEAs.cpp
>>> +  X86WinEHState.cpp
>>>  )
>>> 
>>> if( CMAKE_CL_64 )
>>> 
>>> Modified: llvm/trunk/lib/Target/X86/MCTargetDesc/X86MCAsmInfo.cpp
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/MCTargetDesc/X86MCAsmInfo.cpp?rev=236359&r1=236358&r2=236359&view=diff
>>> ==============================================================================
>>> --- llvm/trunk/lib/Target/X86/MCTargetDesc/X86MCAsmInfo.cpp (original)
>>> +++ llvm/trunk/lib/Target/X86/MCTargetDesc/X86MCAsmInfo.cpp Fri May  1 17:40:25 2015
>>> @@ -132,9 +132,10 @@ X86MCAsmInfoMicrosoft::X86MCAsmInfoMicro
>>>    PrivateLabelPrefix = ".L";
>>>    PointerSize = 8;
>>>    WinEHEncodingType = WinEH::EncodingType::Itanium;
>>> -    ExceptionsType = ExceptionHandling::WinEH;
>>>  }
>>> 
>>> +  ExceptionsType = ExceptionHandling::WinEH;
>>> +
>>>  AssemblerDialect = AsmWriterFlavor;
>>> 
>>>  TextAlignFillValue = 0x90;
>>> 
>>> Modified: llvm/trunk/lib/Target/X86/X86.h
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/X86.h?rev=236359&r1=236358&r2=236359&view=diff
>>> ==============================================================================
>>> --- llvm/trunk/lib/Target/X86/X86.h (original)
>>> +++ llvm/trunk/lib/Target/X86/X86.h Fri May  1 17:40:25 2015
>>> @@ -69,6 +69,12 @@ FunctionPass *createX86FixupLEAs();
>>> /// esp-relative movs with pushes.
>>> FunctionPass *createX86CallFrameOptimization();
>>> 
>>> +/// createX86WinEHStatePass - Return an IR pass that inserts EH registration
>>> +/// stack objects and explicit EH state updates. This pass must run after EH
>>> +/// preparation, which does Windows-specific but architecture-neutral
>>> +/// preparation.
>>> +FunctionPass *createX86WinEHStatePass();
>>> +
>>> } // End llvm namespace
>>> 
>>> #endif
>>> 
>>> Modified: llvm/trunk/lib/Target/X86/X86TargetMachine.cpp
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/X86TargetMachine.cpp?rev=236359&r1=236358&r2=236359&view=diff
>>> ==============================================================================
>>> --- llvm/trunk/lib/Target/X86/X86TargetMachine.cpp (original)
>>> +++ llvm/trunk/lib/Target/X86/X86TargetMachine.cpp Fri May  1 17:40:25 2015
>>> @@ -185,6 +185,7 @@ public:
>>>  void addIRPasses() override;
>>>  bool addInstSelector() override;
>>>  bool addILPOpts() override;
>>> +  bool addPreISel() override;
>>>  void addPreRegAlloc() override;
>>>  void addPostRegAlloc() override;
>>>  void addPreEmitPass() override;
>>> @@ -220,6 +221,14 @@ bool X86PassConfig::addILPOpts() {
>>>  return true;
>>> }
>>> 
>>> +bool X86PassConfig::addPreISel() {
>>> +  // Only add this pass for 32-bit x86.
>>> +  Triple TT(TM->getTargetTriple());
>>> +  if (TT.getArch() == Triple::x86)
>>> +    addPass(createX86WinEHStatePass());
>>> +  return true;
>>> +}
>>> +
>>> void X86PassConfig::addPreRegAlloc() {
>>>  addPass(createX86CallFrameOptimization());
>>> }
>>> 
>>> Added: llvm/trunk/lib/Target/X86/X86WinEHState.cpp
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/X86WinEHState.cpp?rev=236359&view=auto
>>> ==============================================================================
>>> --- llvm/trunk/lib/Target/X86/X86WinEHState.cpp (added)
>>> +++ llvm/trunk/lib/Target/X86/X86WinEHState.cpp Fri May  1 17:40:25 2015
>>> @@ -0,0 +1,317 @@
>>> +//===-- X86WinEHState - Insert EH state updates for win32 exceptions ------===//
>>> +//
>>> +//                     The LLVM Compiler Infrastructure
>>> +//
>>> +// This file is distributed under the University of Illinois Open Source
>>> +// License. See LICENSE.TXT for details.
>>> +//
>>> +//===----------------------------------------------------------------------===//
>>> +//
>>> +// All functions using an MSVC EH personality use an explicitly updated state
>>> +// number stored in an exception registration stack object. The registration
>>> +// object is linked into a thread-local chain of registrations stored at fs:00.
>>> +// This pass adds the registration object and EH state updates.
>>> +//
>>> +//===----------------------------------------------------------------------===//
>>> +
>>> +#include "X86.h"
>>> +#include "llvm/Analysis/LibCallSemantics.h"
>>> +#include "llvm/CodeGen/Passes.h"
>>> +#include "llvm/CodeGen/WinEHFuncInfo.h"
>>> +#include "llvm/IR/Dominators.h"
>>> +#include "llvm/IR/Function.h"
>>> +#include "llvm/IR/IRBuilder.h"
>>> +#include "llvm/IR/Instructions.h"
>>> +#include "llvm/IR/IntrinsicInst.h"
>>> +#include "llvm/IR/Module.h"
>>> +#include "llvm/IR/PatternMatch.h"
>>> +#include "llvm/Pass.h"
>>> +#include "llvm/Support/Debug.h"
>>> +#include "llvm/Support/raw_ostream.h"
>>> +#include "llvm/Transforms/Utils/BasicBlockUtils.h"
>>> +#include "llvm/Transforms/Utils/Cloning.h"
>>> +#include "llvm/Transforms/Utils/Local.h"
>>> +
>>> +using namespace llvm;
>>> +using namespace llvm::PatternMatch;
>>> +
>>> +#define DEBUG_TYPE "winehstate"
>>> +
>>> +namespace {
>>> +class WinEHStatePass : public FunctionPass {
>>> +public:
>>> +  static char ID; // Pass identification, replacement for typeid.
>>> +
>>> +  WinEHStatePass() : FunctionPass(ID) {}
>>> +
>>> +  bool runOnFunction(Function &Fn) override;
>>> +
>>> +  bool doInitialization(Module &M) override;
>>> +
>>> +  bool doFinalization(Module &M) override;
>>> +
>>> +  void getAnalysisUsage(AnalysisUsage &AU) const override;
>>> +
>>> +  const char *getPassName() const override {
>>> +    return "Windows 32-bit x86 EH state insertion";
>>> +  }
>>> +
>>> +private:
>>> +  void emitExceptionRegistrationRecord(Function *F);
>>> +
>>> +  void linkExceptionRegistration(IRBuilder<> &Builder, Value *RegNode,
>>> +                                 Value *Handler);
>>> +  void unlinkExceptionRegistration(IRBuilder<> &Builder, Value *RegNode);
>>> +
>>> +  // Module-level type getters.
>>> +  Type *getEHRegistrationType();
>>> +  Type *getSEH3RegistrationType();
>>> +  Type *getSEH4RegistrationType();
>>> +  Type *getCXXEH3RegistrationType();
>>> +
>>> +  // Per-module data.
>>> +  Module *TheModule = nullptr;
>>> +  StructType *EHRegistrationTy = nullptr;
>>> +  StructType *CXXEH3RegistrationTy = nullptr;
>>> +  StructType *SEH3RegistrationTy = nullptr;
>>> +  StructType *SEH4RegistrationTy = nullptr;
>>> +
>>> +  // Per-function state
>>> +  EHPersonality Personality = EHPersonality::Unknown;
>>> +  Function *PersonalityFn = nullptr;
>>> +};
>>> +}
>>> +
>>> +FunctionPass *llvm::createX86WinEHStatePass() { return new WinEHStatePass(); }
>>> +
>>> +char WinEHStatePass::ID = 0;
>>> +
>>> +bool WinEHStatePass::doInitialization(Module &M) {
>>> +  TheModule = &M;
>>> +  return false;
>>> +}
>>> +
>>> +bool WinEHStatePass::doFinalization(Module &M) {
>>> +  assert(TheModule == &M);
>>> +  TheModule = nullptr;
>>> +  EHRegistrationTy = nullptr;
>>> +  CXXEH3RegistrationTy = nullptr;
>>> +  SEH3RegistrationTy = nullptr;
>>> +  SEH4RegistrationTy = nullptr;
>>> +  return false;
>>> +}
>>> +
>>> +void WinEHStatePass::getAnalysisUsage(AnalysisUsage &AU) const {
>>> +  // This pass should only insert a stack allocation, memory accesses, and
>>> +  // framerecovers.
>>> +  AU.setPreservesCFG();
>>> +}
>>> +
>>> +bool WinEHStatePass::runOnFunction(Function &F) {
>>> +  // Check the personality. Do nothing if this is not an MSVC personality.
>>> +  LandingPadInst *LP = nullptr;
>>> +  for (BasicBlock &BB : F) {
>>> +    LP = BB.getLandingPadInst();
>>> +    if (LP)
>>> +      break;
>>> +  }
>>> +  if (!LP)
>>> +    return false;
>>> +  PersonalityFn =
>>> +      dyn_cast<Function>(LP->getPersonalityFn()->stripPointerCasts());
>>> +  if (!PersonalityFn)
>>> +    return false;
>>> +  Personality = classifyEHPersonality(PersonalityFn);
>>> +  if (!isMSVCEHPersonality(Personality))
>>> +    return false;
>>> +
>>> +  emitExceptionRegistrationRecord(&F);
>>> +  // FIXME: State insertion.
>>> +
>>> +  // Reset per-function state.
>>> +  PersonalityFn = nullptr;
>>> +  Personality = EHPersonality::Unknown;
>>> +  return true;
>>> +}
>>> +
>>> +/// Get the common EH registration subobject:
>>> +///   struct EHRegistrationNode {
>>> +///     EHRegistrationNode *Next;
>>> +///     EXCEPTION_DISPOSITION (*Handler)(...);
>>> +///   };
>>> +Type *WinEHStatePass::getEHRegistrationType() {
>>> +  if (EHRegistrationTy)
>>> +    return EHRegistrationTy;
>>> +  LLVMContext &Context = TheModule->getContext();
>>> +  EHRegistrationTy = StructType::create(Context, "EHRegistrationNode");
>>> +  Type *FieldTys[] = {
>>> +      EHRegistrationTy->getPointerTo(0), // EHRegistrationNode *Next
>>> +      Type::getInt8PtrTy(Context) // EXCEPTION_DISPOSITION (*Handler)(...)
>>> +  };
>>> +  EHRegistrationTy->setBody(FieldTys, false);
>>> +  return EHRegistrationTy;
>>> +}
>>> +
>>> +/// The __CxxFrameHandler3 registration node:
>>> +///   struct CXXExceptionRegistration {
>>> +///     void *SavedESP;
>>> +///     EHRegistrationNode SubRecord;
>>> +///     int32_t TryLevel;
>>> +///   };
>>> +Type *WinEHStatePass::getCXXEH3RegistrationType() {
>>> +  if (CXXEH3RegistrationTy)
>>> +    return CXXEH3RegistrationTy;
>>> +  LLVMContext &Context = TheModule->getContext();
>>> +  Type *FieldTys[] = {
>>> +      Type::getInt8PtrTy(Context), // void *SavedESP
>>> +      getEHRegistrationType(),     // EHRegistrationNode SubRecord
>>> +      Type::getInt32Ty(Context)    // int32_t TryLevel
>>> +  };
>>> +  CXXEH3RegistrationTy =
>>> +      StructType::create(FieldTys, "CXXExceptionRegistration");
>>> +  return CXXEH3RegistrationTy;
>>> +}
>>> +
>>> +/// The _except_handler3 registration node:
>>> +///   struct EH3ExceptionRegistration {
>>> +///     EHRegistrationNode SubRecord;
>>> +///     void *ScopeTable;
>>> +///     int32_t TryLevel;
>>> +///   };
>>> +Type *WinEHStatePass::getSEH3RegistrationType() {
>>> +  if (SEH3RegistrationTy)
>>> +    return SEH3RegistrationTy;
>>> +  LLVMContext &Context = TheModule->getContext();
>>> +  Type *FieldTys[] = {
>>> +      getEHRegistrationType(),     // EHRegistrationNode SubRecord
>>> +      Type::getInt8PtrTy(Context), // void *ScopeTable
>>> +      Type::getInt32Ty(Context)    // int32_t TryLevel
>>> +  };
>>> +  SEH3RegistrationTy = StructType::create(FieldTys, "EH3ExceptionRegistration");
>>> +  return SEH3RegistrationTy;
>>> +}
>>> +
>>> +/// The _except_handler4 registration node:
>>> +///   struct EH4ExceptionRegistration {
>>> +///     void *SavedESP;
>>> +///     _EXCEPTION_POINTERS *ExceptionPointers;
>>> +///     EHRegistrationNode SubRecord;
>>> +///     int32_t EncodedScopeTable;
>>> +///     int32_t TryLevel;
>>> +///   };
>>> +Type *WinEHStatePass::getSEH4RegistrationType() {
>>> +  if (SEH4RegistrationTy)
>>> +    return SEH4RegistrationTy;
>>> +  LLVMContext &Context = TheModule->getContext();
>>> +  Type *FieldTys[] = {
>>> +      Type::getInt8PtrTy(Context), // void *SavedESP
>>> +      Type::getInt8PtrTy(Context), // void *ExceptionPointers
>>> +      getEHRegistrationType(),     // EHRegistrationNode SubRecord
>>> +      Type::getInt32Ty(Context),   // int32_t EncodedScopeTable
>>> +      Type::getInt32Ty(Context)    // int32_t TryLevel
>>> +  };
>>> +  SEH4RegistrationTy = StructType::create(FieldTys, "EH4ExceptionRegistration");
>>> +  return SEH4RegistrationTy;
>>> +}
>>> +
>>> +// Emit an exception registration record. These are stack allocations with the
>>> +// common subobject of two pointers: the previous registration record (the old
>>> +// fs:00) and the personality function for the current frame. The data before
>>> +// and after that is personality function specific.
>>> +void WinEHStatePass::emitExceptionRegistrationRecord(Function *F) {
>>> +  assert(Personality == EHPersonality::MSVC_CXX ||
>>> +         Personality == EHPersonality::MSVC_X86SEH);
>>> +
>>> +  StringRef PersonalityName = PersonalityFn->getName();
>>> +  IRBuilder<> Builder(&F->getEntryBlock(), F->getEntryBlock().begin());
>>> +  Type *Int8PtrType = Builder.getInt8PtrTy();
>>> +  Value *SubRecord = nullptr;
>>> +  if (PersonalityName == "__CxxFrameHandler3") {
>>> +    Type *RegNodeTy = getCXXEH3RegistrationType();
>>> +    Value *RegNode = Builder.CreateAlloca(RegNodeTy);
>>> +    // FIXME: We can skip this in -GS- mode, when we figure that out.
>>> +    // SavedESP = llvm.stacksave()
>>> +    Value *SP = Builder.CreateCall(
>>> +        Intrinsic::getDeclaration(TheModule, Intrinsic::stacksave));
>>> +    Builder.CreateStore(SP, Builder.CreateStructGEP(RegNodeTy, RegNode, 0));
>>> +    // TryLevel = -1
>>> +    Builder.CreateStore(Builder.getInt32(-1),
>>> +                        Builder.CreateStructGEP(RegNodeTy, RegNode, 2));
>>> +    // FIXME: 'Personality' is incorrect here. We need to generate a trampoline
>>> +    // that effectively gets the LSDA.
>>> +    SubRecord = Builder.CreateStructGEP(RegNodeTy, RegNode, 1);
>>> +    linkExceptionRegistration(Builder, SubRecord, PersonalityFn);
>>> +  } else if (PersonalityName == "_except_handler3") {
>>> +    Type *RegNodeTy = getSEH3RegistrationType();
>>> +    Value *RegNode = Builder.CreateAlloca(RegNodeTy);
>>> +    // TryLevel = -1
>>> +    Builder.CreateStore(Builder.getInt32(-1),
>>> +                        Builder.CreateStructGEP(RegNodeTy, RegNode, 2));
>>> +    // FIXME: Generalize llvm.eh.sjljl.lsda for this.
>>> +    // ScopeTable = nullptr
>>> +    Builder.CreateStore(Constant::getNullValue(Int8PtrType),
>>> +                        Builder.CreateStructGEP(RegNodeTy, RegNode, 1));
>>> +    SubRecord = Builder.CreateStructGEP(RegNodeTy, RegNode, 0);
>>> +    linkExceptionRegistration(Builder, SubRecord, PersonalityFn);
>>> +  } else if (PersonalityName == "_except_handler4") {
>>> +    Type *RegNodeTy = getSEH4RegistrationType();
>>> +    Value *RegNode = Builder.CreateAlloca(RegNodeTy);
>>> +    // SavedESP = llvm.stacksave()
>>> +    Value *SP = Builder.CreateCall(
>>> +        Intrinsic::getDeclaration(TheModule, Intrinsic::stacksave));
>>> +    Builder.CreateStore(SP, Builder.CreateStructGEP(RegNodeTy, RegNode, 0));
>>> +    // TryLevel = -2
>>> +    Builder.CreateStore(Builder.getInt32(-2),
>>> +                        Builder.CreateStructGEP(RegNodeTy, RegNode, 4));
>>> +    // FIXME: Generalize llvm.eh.sjljl.lsda for this, and then do the stack
>>> +    // cookie xor.
>>> +    // ScopeTable = nullptr
>>> +    Builder.CreateStore(Builder.getInt32(0),
>>> +                        Builder.CreateStructGEP(RegNodeTy, RegNode, 3));
>>> +    SubRecord = Builder.CreateStructGEP(RegNodeTy, RegNode, 2);
>>> +    linkExceptionRegistration(Builder, SubRecord, PersonalityFn);
>>> +  } else {
>>> +    llvm_unreachable("unexpected personality function");
>>> +  }
>>> +
>>> +  // FIXME: Insert an unlink before all returns.
>>> +  for (BasicBlock &BB : *F) {
>>> +    TerminatorInst *T = BB.getTerminator();
>>> +    if (!isa<ReturnInst>(T))
>>> +      continue;
>>> +    Builder.SetInsertPoint(T);
>>> +    unlinkExceptionRegistration(Builder, SubRecord);
>>> +  }
>>> +}
>>> +
>>> +void WinEHStatePass::linkExceptionRegistration(IRBuilder<> &Builder,
>>> +                                               Value *RegNode, Value *Handler) {
>>> +  Type *RegNodeTy = getEHRegistrationType();
>>> +  // Handler = Handler
>>> +  Handler = Builder.CreateBitCast(Handler, Builder.getInt8PtrTy());
>>> +  Builder.CreateStore(Handler, Builder.CreateStructGEP(RegNodeTy, RegNode, 1));
>>> +  // Next = [fs:00]
>>> +  Constant *FSZero =
>>> +      Constant::getNullValue(RegNodeTy->getPointerTo()->getPointerTo(257));
>>> +  Value *Next = Builder.CreateLoad(FSZero);
>>> +  Builder.CreateStore(Next, Builder.CreateStructGEP(RegNodeTy, RegNode, 0));
>>> +  // [fs:00] = RegNode
>>> +  Builder.CreateStore(RegNode, FSZero);
>>> +}
>>> +
>>> +void WinEHStatePass::unlinkExceptionRegistration(IRBuilder<> &Builder,
>>> +                                                 Value *RegNode) {
>>> +  // Clone RegNode into the current BB for better address mode folding.
>>> +  if (auto *GEP = dyn_cast<GetElementPtrInst>(RegNode)) {
>>> +    GEP = cast<GetElementPtrInst>(GEP->clone());
>>> +    Builder.Insert(GEP);
>>> +    RegNode = GEP;
>>> +  }
>>> +  Type *RegNodeTy = getEHRegistrationType();
>>> +  // [fs:00] = RegNode->Next
>>> +  Value *Next =
>>> +      Builder.CreateLoad(Builder.CreateStructGEP(RegNodeTy, RegNode, 0));
>>> +  Constant *FSZero =
>>> +      Constant::getNullValue(RegNodeTy->getPointerTo()->getPointerTo(257));
>>> +  Builder.CreateStore(Next, FSZero);
>>> +}
>>> 
>>> Modified: llvm/trunk/test/CodeGen/X86/inalloca-invoke.ll
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/X86/inalloca-invoke.ll?rev=236359&r1=236358&r2=236359&view=diff
>>> ==============================================================================
>>> --- llvm/trunk/test/CodeGen/X86/inalloca-invoke.ll (original)
>>> +++ llvm/trunk/test/CodeGen/X86/inalloca-invoke.ll Fri May  1 17:40:25 2015
>>> @@ -4,6 +4,7 @@
>>> 
>>> %frame.reverse = type { %Iter, %Iter }
>>> 
>>> +declare i32 @pers(...)
>>> declare void @llvm.stackrestore(i8*)
>>> declare i8* @llvm.stacksave()
>>> declare void @begin(%Iter* sret)
>>> @@ -22,8 +23,7 @@ blah:
>>> 
>>> ; CHECK:  calll   __chkstk
>>> ; CHECK:  movl %esp, %[[beg:[^ ]*]]
>>> -; CHECK:  movl %esp, %[[end:[^ ]*]]
>>> -; CHECK:  addl $12, %[[end]]
>>> +; CHECK:  leal 12(%[[beg]]), %[[end:[^ ]*]]
>>> 
>>>  call void @begin(%Iter* sret %temp.lvalue)
>>> ; CHECK:  calll _begin
>>> @@ -49,7 +49,7 @@ invoke.cont5:
>>>  ret i32 0
>>> 
>>> lpad:                                             ; preds = %invoke.cont, %entry
>>> -  %lp = landingpad { i8*, i32 } personality i8* null
>>> +  %lp = landingpad { i8*, i32 } personality i32 (...)* @pers
>>>          cleanup
>>>  unreachable
>>> }
>>> 
>>> Added: llvm/trunk/test/CodeGen/X86/win32-eh.ll
>>> URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/X86/win32-eh.ll?rev=236359&view=auto
>>> ==============================================================================
>>> --- llvm/trunk/test/CodeGen/X86/win32-eh.ll (added)
>>> +++ llvm/trunk/test/CodeGen/X86/win32-eh.ll Fri May  1 17:40:25 2015
>>> @@ -0,0 +1,77 @@
>>> +; RUN: llc -mtriple=i686-pc-windows-msvc < %s | FileCheck %s
>>> +
>>> +declare void @may_throw_or_crash()
>>> +declare i32 @_except_handler3(...)
>>> +declare i32 @_except_handler4(...)
>>> +declare i32 @__CxxFrameHandler3(...)
>>> +declare void @llvm.eh.begincatch(i8*, i8*)
>>> +declare void @llvm.eh.endcatch()
>>> +
>>> +define void @use_except_handler3() {
>>> +  invoke void @may_throw_or_crash()
>>> +      to label %cont unwind label %catchall
>>> +cont:
>>> +  ret void
>>> +catchall:
>>> +  landingpad { i8*, i32 } personality i32 (...)* @_except_handler3
>>> +      catch i8* null
>>> +  br label %cont
>>> +}
>>> +
>>> +; CHECK-LABEL: _use_except_handler3:
>>> +; CHECK: subl ${{[0-9]+}}, %esp
>>> +; CHECK: movl %fs:0, %[[next:[^ ,]*]]
>>> +; CHECK: movl %[[next]], (%esp)
>>> +; CHECK: leal (%esp), %[[node:[^ ,]*]]
>>> +; CHECK: movl %[[node]], %fs:0
>>> +; CHECK: calll _may_throw_or_crash
>>> +; CHECK: movl (%esp), %[[next:[^ ,]*]]
>>> +; CHECK: movl %[[next]], %fs:0
>>> +; CHECK: retl
>>> +
>>> +define void @use_except_handler4() {
>>> +  invoke void @may_throw_or_crash()
>>> +      to label %cont unwind label %catchall
>>> +cont:
>>> +  ret void
>>> +catchall:
>>> +  landingpad { i8*, i32 } personality i32 (...)* @_except_handler4
>>> +      catch i8* null
>>> +  br label %cont
>>> +}
>>> +
>>> +; CHECK-LABEL: _use_except_handler4:
>>> +; CHECK: subl ${{[0-9]+}}, %esp
>>> +; CHECK: leal 8(%esp), %[[node:[^ ,]*]]
>>> +; CHECK: movl %fs:0, %[[next:[^ ,]*]]
>>> +; CHECK: movl %[[next]], 8(%esp)
>>> +; CHECK: movl %[[node]], %fs:0
>>> +; CHECK: calll _may_throw_or_crash
>>> +; CHECK: movl 8(%esp), %[[next:[^ ,]*]]
>>> +; CHECK: movl %[[next]], %fs:0
>>> +; CHECK: retl
>>> +
>>> +define void @use_CxxFrameHandler3() {
>>> +  invoke void @may_throw_or_crash()
>>> +      to label %cont unwind label %catchall
>>> +cont:
>>> +  ret void
>>> +catchall:
>>> +  %ehvals = landingpad { i8*, i32 } personality i32 (...)* @__CxxFrameHandler3
>>> +      catch i8* null
>>> +  %ehptr = extractvalue { i8*, i32 } %ehvals, 0
>>> +  call void @llvm.eh.begincatch(i8* %ehptr, i8* null)
>>> +  call void @llvm.eh.endcatch()
>>> +  br label %cont
>>> +}
>>> +
>>> +; CHECK-LABEL: _use_CxxFrameHandler3:
>>> +; CHECK: subl ${{[0-9]+}}, %esp
>>> +; CHECK: leal 4(%esp), %[[node:[^ ,]*]]
>>> +; CHECK: movl %fs:0, %[[next:[^ ,]*]]
>>> +; CHECK: movl %[[next]], 4(%esp)
>>> +; CHECK: movl %[[node]], %fs:0
>>> +; CHECK: calll _may_throw_or_crash
>>> +; CHECK: movl 4(%esp), %[[next:[^ ,]*]]
>>> +; CHECK: movl %[[next]], %fs:0
>>> +; CHECK: retl
>>> 
>>> 
>>> _______________________________________________
>>> llvm-commits mailing list
>>> llvm-commits at cs.uiuc.edu
>>> http://lists.cs.uiuc.edu/mailman/listinfo/llvm-commits
>> 
>> 
>> _______________________________________________
>> llvm-commits mailing list
>> llvm-commits at cs.uiuc.edu
>> http://lists.cs.uiuc.edu/mailman/listinfo/llvm-commits





More information about the llvm-commits mailing list