Merge pull request #34357 from gottesmm/pr-5b70b29de84d902644bc21dc530816c78e919469

[silgen] Ensure that cleanup cloner clones formal access cleanups to formal access cleanups.

Author: gottesmm

lib/SILGen/Cleanup.cpp

@@ -359,14 +359,16 @@ void CleanupStateRestorationScope::pop() && { popImpl(); }
 //===----------------------------------------------------------------------===//
 
 CleanupCloner::CleanupCloner(SILGenFunction &SGF, const ManagedValue &mv)
-    : SGF(SGF), hasCleanup(mv.hasCleanup()), isLValue(mv.isLValue()),
-      writebackBuffer(None) {
+    : SGF(SGF), writebackBuffer(None), hasCleanup(mv.hasCleanup()),
+      isLValue(mv.isLValue()), isFormalAccess(false) {
   if (hasCleanup) {
     auto handle = mv.getCleanup();
     auto state = SGF.Cleanups.getFlagsAndWritebackBuffer(handle);
-    if (SILValue value = std::get<1>(state).getValueOr(SILValue())) {
+    using RawTy = std::underlying_type<Cleanup::Flags>::type;
+    if (RawTy(std::get<0>(state)) & RawTy(Cleanup::Flags::FormalAccessCleanup))
+      isFormalAccess = true;
+    if (SILValue value = std::get<1>(state).getValueOr(SILValue()))
       writebackBuffer = value;
-    }
   }
 }
 
@@ -405,8 +407,16 @@ ManagedValue CleanupCloner::clone(SILValue value) const {
   }
 
   if (value->getType().isAddress()) {
+    if (isFormalAccess) {
+      auto loc = RegularLocation::getAutoGeneratedLocation();
+      return SGF.emitFormalAccessManagedBufferWithCleanup(loc, value);
+    }
     return SGF.emitManagedBufferWithCleanup(value);
   }
 
+  if (isFormalAccess) {
+    auto loc = RegularLocation::getAutoGeneratedLocation();
+    return SGF.emitFormalAccessManagedRValueWithCleanup(loc, value);
+  }
   return SGF.emitManagedRValueWithCleanup(value);
 }

lib/SILGen/Cleanup.h

@@ -84,9 +84,9 @@ class LLVM_LIBRARY_VISIBILITY Cleanup {
   //
   // Example: Distinguishing in between @owned cleanups with a writeback buffer
   // (ExclusiveBorrowCleanup) or ones that involve formal access cleanups.
-  enum class Flags : uint8_t {
+  enum Flags : uint8_t {
     None = 0,
-    ExclusiveBorrowCleanup = 1,
+    FormalAccessCleanup = 1,
   };
 
 private:
@@ -95,7 +95,7 @@ class LLVM_LIBRARY_VISIBILITY Cleanup {
   Flags flags : 8;
 
 protected:
-  Cleanup() {}
+  Cleanup() : flags(Flags::None) {}
   virtual ~Cleanup() {}
 
 public:
@@ -123,6 +123,14 @@ class LLVM_LIBRARY_VISIBILITY Cleanup {
   virtual bool getWritebackBuffer(function_ref<void(SILValue)> func) {
     return false;
   }
+
+  bool isFormalAccess() const {
+    return getFlags() & Flags::FormalAccessCleanup;
+  }
+
+  void setIsFormalAccess() {
+    flags = Flags(flags | Flags::FormalAccessCleanup);
+  }
 };
 
 /// A cleanup depth is generally used to denote the set of cleanups
@@ -310,9 +318,10 @@ class CleanupStateRestorationScope {
 /// writeback buffers.
 class CleanupCloner {
   SILGenFunction &SGF;
+  Optional<SILValue> writebackBuffer;
   bool hasCleanup;
   bool isLValue;
-  Optional<SILValue> writebackBuffer;
+  bool isFormalAccess;
 
 public:
   CleanupCloner(SILGenFunction &SGF, const ManagedValue &mv);

lib/SILGen/SILGenDecl.cpp

@@ -1431,10 +1431,12 @@ SILGenFunction::enterDormantTemporaryCleanup(SILValue addr,
 
 namespace {
 
-struct FormalAccessReleaseValueCleanup : Cleanup {
+struct FormalAccessReleaseValueCleanup final : Cleanup {
   FormalEvaluationContext::stable_iterator Depth;
 
-  FormalAccessReleaseValueCleanup() : Depth() {}
+  FormalAccessReleaseValueCleanup() : Cleanup(), Depth() {
+    setIsFormalAccess();
+  }
 
   void setState(SILGenFunction &SGF, CleanupState newState) override {
     if (newState == CleanupState::Dead) {