Add a new api findUnreferenceableStorage to check if struct or its fields have unreferenceable storage

Author: meg-gupta

include/swift/SILOptimizer/Utils/InstOptUtils.h

@@ -604,6 +604,10 @@ bool specializeAppliesInFunction(SILFunction &F,
 /// types aggregated together at each level.
 SILValue createEmptyAndUndefValue(SILType ty, SILInstruction *insertionPoint,
                                   SILBuilderContext &ctx, bool noUndef = false);
+
+/// Check if a struct or its fields can have unreferenceable storage.
+bool findUnreferenceableStorage(StructDecl *decl, SILType structType,
+                                SILFunction *func);
 } // end namespace swift
 
 #endif // SWIFT_SILOPTIMIZER_UTILS_INSTOPTUTILS_H

lib/SILOptimizer/Utils/InstOptUtils.cpp

@@ -2009,3 +2009,21 @@ SILValue swift::createEmptyAndUndefValue(SILType ty,
   assert(!noUndef);
   return SILUndef::get(insertionPoint->getFunction(), ty);
 }
+
+bool swift::findUnreferenceableStorage(StructDecl *decl, SILType structType,
+                                       SILFunction *func) {
+  if (decl->hasUnreferenceableStorage()) {
+    return true;
+  }
+  // Check if any fields have unreferenceable stoage
+  for (auto *field : decl->getStoredProperties()) {
+    TypeExpansionContext tec = *func;
+    auto fieldTy = structType.getFieldType(field, func->getModule(), tec);
+    if (auto *fieldStructDecl = fieldTy.getStructOrBoundGenericStruct()) {
+      if (findUnreferenceableStorage(fieldStructDecl, fieldTy, func)) {
+        return true;
+      }
+    }
+  }
+  return false;
+}