Merge pull request #41066 from apple/ABI-Add-predicate-to-identify-sections-containing-reflection-data

[ABI] Add predicate to identify sections containing reflection data

(cherry picked from commit 56a614a)

Author: kastiglione

include/swift/ABI/ObjectFile.h

@@ -28,6 +28,8 @@ class SwiftObjectFileFormat {
 public:
   virtual ~SwiftObjectFileFormat() {}
   virtual llvm::StringRef getSectionName(ReflectionSectionKind section) = 0;
+  /// Predicate to identify if the named section can contain reflection data.
+  virtual bool sectionContainsReflectionData(llvm::StringRef sectionName) = 0;
 };
 
 /// Responsible for providing the Mach-O reflection section identifiers.
@@ -50,6 +52,15 @@ class SwiftObjectFileFormatMachO : public SwiftObjectFileFormat {
     }
     llvm_unreachable("Section type not found.");
   }
+
+  bool sectionContainsReflectionData(llvm::StringRef sectionName) override {
+    // For Mach-O, the caller must call this function twice, once with just the
+    // section name (ex `__swift5_fieldmd`), and again with the segment
+    // qualified section name (ex `__DATA,__const`).
+    return sectionName.startswith("__swift5_") ||
+           sectionName == "__DATA_CONST,__const" ||
+           sectionName == "__DATA,__const";
+  }
 };
 
 /// Responsible for providing the ELF reflection section identifiers.
@@ -72,6 +83,10 @@ class SwiftObjectFileFormatELF : public SwiftObjectFileFormat {
     }
     llvm_unreachable("Section type not found.");
   }
+
+  bool sectionContainsReflectionData(llvm::StringRef sectionName) override {
+    return sectionName.startswith("swift5_");
+  }
 };
 
 /// Responsible for providing the COFF reflection section identifiers
@@ -94,6 +109,10 @@ class SwiftObjectFileFormatCOFF : public SwiftObjectFileFormat {
     }
     llvm_unreachable("Section  not found.");
   }
+
+  bool sectionContainsReflectionData(llvm::StringRef sectionName) override {
+    return sectionName.startswith(".sw5");
+  }
 };
 } // namespace swift
 #endif // SWIFT_ABI_OBJECTFILE_H