Update AtomicExpandPass.cpp

#115

Author: hzqst

llvm/lib/CodeGen/AtomicExpandPass.cpp

@@ -1847,6 +1847,12 @@ bool AtomicExpand::expandAtomicOpToLibcall(
     Args.push_back(ConstantInt::get(DL.getIntPtrType(Ctx), Size));
   }
 
+  bool bShouldDisableLifetimeMarker = false;
+
+  if (I->getParent() && I->getParent()->getParent() && I->getParent()->getParent()->hasSEHOrCXXSEH()) {
+    bShouldDisableLifetimeMarker = true;
+  }
+
   // 'ptr' argument.
   // note: This assumes all address spaces share a common libfunc
   // implementation and that addresses are convertable.  For systems without
@@ -1860,9 +1866,11 @@ bool AtomicExpand::expandAtomicOpToLibcall(
   if (CASExpected) {
     AllocaCASExpected = AllocaBuilder.CreateAlloca(CASExpected->getType());
     AllocaCASExpected->setAlignment(AllocaAlignment);
-#ifndef _WIN32
-    Builder.CreateLifetimeStart(AllocaCASExpected, SizeVal64);
-#endif
+    if (!bShouldDisableLifetimeMarker) {
+
+      Builder.CreateLifetimeStart(AllocaCASExpected, SizeVal64);
+    }
+
     Builder.CreateAlignedStore(CASExpected, AllocaCASExpected, AllocaAlignment);
     Args.push_back(AllocaCASExpected);
   }
@@ -1876,9 +1884,10 @@ bool AtomicExpand::expandAtomicOpToLibcall(
     } else {
       AllocaValue = AllocaBuilder.CreateAlloca(ValueOperand->getType());
       AllocaValue->setAlignment(AllocaAlignment);
-#ifndef _WIN32
-      Builder.CreateLifetimeStart(AllocaValue, SizeVal64);
-#endif
+      if (!bShouldDisableLifetimeMarker) {
+        Builder.CreateLifetimeStart(AllocaValue, SizeVal64);
+      }
+
       Builder.CreateAlignedStore(ValueOperand, AllocaValue, AllocaAlignment);
       Args.push_back(AllocaValue);
     }
@@ -1888,9 +1897,10 @@ bool AtomicExpand::expandAtomicOpToLibcall(
   if (!CASExpected && HasResult && !UseSizedLibcall) {
     AllocaResult = AllocaBuilder.CreateAlloca(I->getType());
     AllocaResult->setAlignment(AllocaAlignment);
-#ifndef _WIN32
-    Builder.CreateLifetimeStart(AllocaResult, SizeVal64);
-#endif
+    if (!bShouldDisableLifetimeMarker) {
+      Builder.CreateLifetimeStart(AllocaResult, SizeVal64);
+    }
+
     Args.push_back(AllocaResult);
   }
 
@@ -1920,21 +1930,21 @@ bool AtomicExpand::expandAtomicOpToLibcall(
   CallInst *Call = Builder.CreateCall(LibcallFn, Args);
   Call->setAttributes(Attr);
   Value *Result = Call;
-#ifndef _WIN32
-  // And then, extract the results...
-  if (ValueOperand && !UseSizedLibcall)
-    Builder.CreateLifetimeEnd(AllocaValue, SizeVal64);
-#endif
+  if (!bShouldDisableLifetimeMarker) {
+    // And then, extract the results...
+    if (ValueOperand && !UseSizedLibcall)
+      Builder.CreateLifetimeEnd(AllocaValue, SizeVal64);
+  }
   if (CASExpected) {
     // The final result from the CAS is {load of 'expected' alloca, bool result
     // from call}
     Type *FinalResultTy = I->getType();
     Value *V = PoisonValue::get(FinalResultTy);
     Value *ExpectedOut = Builder.CreateAlignedLoad(
         CASExpected->getType(), AllocaCASExpected, AllocaAlignment);
-#ifndef _WIN32
-    Builder.CreateLifetimeEnd(AllocaCASExpected, SizeVal64);
-#endif
+    if (!bShouldDisableLifetimeMarker) {
+      Builder.CreateLifetimeEnd(AllocaCASExpected, SizeVal64);
+    }
     V = Builder.CreateInsertValue(V, ExpectedOut, 0);
     V = Builder.CreateInsertValue(V, Result, 1);
     I->replaceAllUsesWith(V);
@@ -1945,9 +1955,9 @@ bool AtomicExpand::expandAtomicOpToLibcall(
     else {
       V = Builder.CreateAlignedLoad(I->getType(), AllocaResult,
                                     AllocaAlignment);
-#ifndef _WIN32
-      Builder.CreateLifetimeEnd(AllocaResult, SizeVal64);
-#endif
+      if (!bShouldDisableLifetimeMarker) {
+        Builder.CreateLifetimeEnd(AllocaResult, SizeVal64);
+      }
     }
     I->replaceAllUsesWith(V);
   }