mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
a3a43d46c5
On macOS it is possible for one application to contain Swift modules compiled for different triples that are incompatible as far as the Swift compiler is concerned. Examples include an iOS simulator application hunning on a macOS host, or a macCatalyst application running on macOS. A debugger might see .swift_ast sections for all triples at the same time. This patch adds an interface to let the client provide a triple to filter Swift modules in an ASTSection. rdar://107869141
85 lines
3.0 KiB
C++
85 lines
3.0 KiB
C++
//===--- ASTSectionImporter.cpp - Import AST Section Modules --------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements support for loading modules serialized into a
|
|
// Mach-O AST section into Swift.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/ASTSectionImporter/ASTSectionImporter.h"
|
|
#include "../Serialization/ModuleFormat.h"
|
|
#include "swift/AST/ASTContext.h"
|
|
#include "swift/Basic/Dwarf.h"
|
|
#include "swift/Serialization/SerializedModuleLoader.h"
|
|
#include "swift/Serialization/Validation.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace swift;
|
|
|
|
bool swift::parseASTSection(MemoryBufferSerializedModuleLoader &Loader,
|
|
StringRef buf,
|
|
const llvm::Triple &filter,
|
|
SmallVectorImpl<std::string> &foundModules) {
|
|
if (!serialization::isSerializedAST(buf))
|
|
return false;
|
|
|
|
bool haveFilter = filter.getOS() != llvm::Triple::UnknownOS &&
|
|
filter.getArch() != llvm::Triple::UnknownArch;
|
|
// An AST section consists of one or more AST modules, optionally with
|
|
// headers. Iterate over all AST modules.
|
|
while (!buf.empty()) {
|
|
auto info = serialization::validateSerializedAST(
|
|
buf, Loader.isRequiredOSSAModules(), /*requiredSDK*/ StringRef(),
|
|
/*requiresRevisionMatch*/ false);
|
|
|
|
assert(info.name.size() < (2 << 10) && "name failed sanity check");
|
|
|
|
if (info.status == serialization::Status::Valid) {
|
|
assert(info.bytes != 0);
|
|
bool selected = true;
|
|
if (haveFilter) {
|
|
llvm::Triple moduleTriple(info.targetTriple);
|
|
selected = serialization::areCompatible(moduleTriple, filter);
|
|
}
|
|
if (!info.name.empty() && selected) {
|
|
StringRef moduleData = buf.substr(0, info.bytes);
|
|
std::unique_ptr<llvm::MemoryBuffer> bitstream(
|
|
llvm::MemoryBuffer::getMemBuffer(moduleData, info.name, false));
|
|
|
|
// Register the memory buffer.
|
|
Loader.registerMemoryBuffer(info.name, std::move(bitstream),
|
|
info.userModuleVersion);
|
|
foundModules.push_back(info.name.str());
|
|
}
|
|
} else {
|
|
llvm::dbgs() << "Unable to load module";
|
|
if (!info.name.empty())
|
|
llvm::dbgs() << " '" << info.name << '\'';
|
|
llvm::dbgs() << ".\n";
|
|
}
|
|
|
|
if (info.bytes == 0)
|
|
return false;
|
|
|
|
if (info.bytes > buf.size()) {
|
|
llvm::dbgs() << "AST section too small.\n";
|
|
return false;
|
|
}
|
|
|
|
buf = buf.substr(
|
|
llvm::alignTo(info.bytes, swift::serialization::SWIFTMODULE_ALIGNMENT));
|
|
}
|
|
|
|
return true;
|
|
}
|