[Dependency Scanning] Record whether discovered binary Swift modules are frameworks

Part of rdar://102824777

Author: artemcm

include/swift-c/DependencyScan/DependencyScan.h

@@ -164,6 +164,10 @@ SWIFTSCAN_PUBLIC swiftscan_string_ref_t
 swiftscan_swift_binary_detail_get_module_source_info_path(
     swiftscan_module_details_t details);
 
+SWIFTSCAN_PUBLIC bool
+swiftscan_swift_binary_detail_get_is_framework(
+    swiftscan_module_details_t details);
+
 //=== Swift Placeholder Module Details query APIs -------------------------===//
 
 SWIFTSCAN_PUBLIC swiftscan_string_ref_t

include/swift/DependencyScan/DependencyScanImpl.h

@@ -104,6 +104,9 @@ typedef struct {
 
   /// The path to the .swiftSourceInfo file.
   swiftscan_string_ref_t module_source_info_path;
+
+  /// A flag to indicate whether or not this module is a framework.
+  bool is_framework;
 } swiftscan_swift_binary_details_t;
 
 /// Swift placeholder modules carry additional details that specify their

include/swift/Serialization/SerializedModuleLoader.h

@@ -137,7 +137,7 @@ class SerializedModuleLoaderBase : public ModuleLoader {
   }
 
   /// Scan the given serialized module file to determine dependencies.
-  llvm::ErrorOr<ModuleDependencies> scanModuleFile(Twine modulePath);
+  llvm::ErrorOr<ModuleDependencies> scanModuleFile(Twine modulePath, bool isFramework);
 
   /// Load the module file into a buffer and also collect its module name.
   static std::unique_ptr<llvm::MemoryBuffer>

lib/DependencyScan/ModuleDependencyCacheSerialization.cpp

@@ -838,7 +838,8 @@ void ModuleDependenciesCacheSerializer::writeModuleInfo(ModuleDependencyID modul
         Out, ScratchRecord, AbbrCodes[SwiftBinaryModuleDetailsLayout::Code],
         getIdentifier(swiftBinDeps->compiledModulePath),
         getIdentifier(swiftBinDeps->moduleDocPath),
-        getIdentifier(swiftBinDeps->sourceInfoPath), swiftBinDeps->isFramework);
+        getIdentifier(swiftBinDeps->sourceInfoPath),
+        swiftBinDeps->isFramework);
 
     break;
   }

lib/DependencyScan/ScanDependencies.cpp

@@ -728,7 +728,9 @@ static void writeJSON(llvm::raw_ostream &out,
         writeJSONSingleField(out, "moduleSourceInfoPath",
                              swiftBinaryDeps->module_source_info_path,
                              /*indentLevel=*/5,
-                             /*trailingComma=*/false);
+                             /*trailingComma=*/true);
+      writeJSONSingleField(out, "isFramework", swiftBinaryDeps->is_framework,
+                           5, /*trailingComma=*/false);
     } else {
       out << "\"clang\": {\n";
 
@@ -893,7 +895,8 @@ generateFullDependencyGraph(CompilerInstance &instance,
         details->swift_binary_details = {
             create_clone(swiftBinaryDeps->compiledModulePath.c_str()),
             create_clone(swiftBinaryDeps->moduleDocPath.c_str()),
-            create_clone(swiftBinaryDeps->sourceInfoPath.c_str())};
+            create_clone(swiftBinaryDeps->sourceInfoPath.c_str()),
+            swiftBinaryDeps->isFramework};
       } else {
         // Clang module details
         details->kind = SWIFTSCAN_DEPENDENCY_INFO_CLANG;

lib/Frontend/ModuleInterfaceLoader.cpp

@@ -1993,6 +1993,24 @@ bool ExplicitSwiftModuleLoader::canImportModule(
   if (it == Impl.ExplicitModuleMap.end()) {
     return false;
   }
+
+  // If the caller doesn't want version info we're done.
+  if (!versionInfo)
+    return true;
+
+  // Open .swiftmodule file and read out the version
+  auto &fs = *Ctx.SourceMgr.getFileSystem();
+  auto moduleBuf = fs.getBufferForFile(it->second.modulePath);
+  if (!moduleBuf) {
+    Ctx.Diags.diagnose(SourceLoc(), diag::error_opening_explicit_module_file,
+                       it->second.modulePath);
+    return false;
+  }
+  auto metaData = serialization::validateSerializedAST(
+      (*moduleBuf)->getBuffer(), Ctx.SILOpts.EnableOSSAModules,
+      Ctx.LangOpts.SDKName, !Ctx.LangOpts.DebuggerSupport);
+  versionInfo->setVersion(metaData.userModuleVersion,
+                          ModuleVersionSourceKind::SwiftBinaryModule);
   return true;
 }
 

lib/Serialization/ModuleDependencyScanner.cpp

@@ -43,7 +43,7 @@ std::error_code ModuleDependencyScanner::findModuleFilesInDirectory(
   if (LoadMode == ModuleLoadingMode::OnlySerialized || !fs.exists(InPath)) {
     if (fs.exists(ModPath)) {
       // The module file will be loaded directly.
-      auto dependencies = scanModuleFile(ModPath);
+      auto dependencies = scanModuleFile(ModPath, IsFramework);
       if (dependencies) {
         this->dependencies = std::move(dependencies.get());
         return std::error_code();

lib/Serialization/SerializedModuleLoader.cpp

@@ -392,7 +392,7 @@ std::error_code SerializedModuleLoaderBase::openModuleFile(
 }
 
 llvm::ErrorOr<ModuleDependencies> SerializedModuleLoaderBase::scanModuleFile(
-    Twine modulePath) {
+    Twine modulePath, bool isFramework) {
   // Open the module file
   auto &fs = *Ctx.SourceMgr.getFileSystem();
   auto moduleBuf = fs.getBufferForFile(modulePath);
@@ -401,7 +401,6 @@ llvm::ErrorOr<ModuleDependencies> SerializedModuleLoaderBase::scanModuleFile(
 
   // Load the module file without validation.
   std::shared_ptr<const ModuleFileSharedCore> loadedModuleFile;
-  bool isFramework = false;
   serialization::ValidationInfo loadInfo = ModuleFileSharedCore::load(
       modulePath.str(), std::move(moduleBuf.get()), nullptr, nullptr,
       isFramework, isRequiredOSSAModules(), Ctx.LangOpts.SDKName,

tools/libSwiftScan/libSwiftScan.cpp

@@ -311,6 +311,11 @@ swiftscan_swift_binary_detail_get_module_source_info_path(
   return details->swift_binary_details.module_source_info_path;
 }
 
+bool swiftscan_swift_binary_detail_get_is_framework(
+    swiftscan_module_details_t details) {
+  return details->swift_binary_details.is_framework;
+}
+
 //=== Swift Placeholder Module Details query APIs -------------------------===//
 
 swiftscan_string_ref_t

tools/libSwiftScan/libSwiftScan.exports

@@ -18,6 +18,7 @@ swiftscan_swift_textual_detail_get_is_framework
 swiftscan_swift_binary_detail_get_compiled_module_path
 swiftscan_swift_binary_detail_get_module_doc_path
 swiftscan_swift_binary_detail_get_module_source_info_path
+swiftscan_swift_binary_detail_get_is_framework
 swiftscan_swift_placeholder_detail_get_compiled_module_path
 swiftscan_swift_placeholder_detail_get_module_doc_path
 swiftscan_swift_placeholder_detail_get_module_source_info_path