1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
//===--- 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/Serialization/SerializedModuleLoader.h"
#include "swift/Serialization/Validation.h"
#include "llvm/Support/Debug.h"
#include "llvm/Support/raw_ostream.h"
using namespace swift;
std::string ASTSectionParseError::toString() const {
std::string S;
llvm::raw_string_ostream SS(S);
SS << serialization::StatusToString(Error);
if (!ErrorMessage.empty())
SS << ": " << ErrorMessage;
return SS.str();
}
void ASTSectionParseError::log(raw_ostream &OS) const { OS << toString(); }
std::error_code ASTSectionParseError::convertToErrorCode() const {
llvm_unreachable("Function not implemented.");
}
char ASTSectionParseError::ID;
llvm::Expected<SmallVector<std::string, 4>>
swift::parseASTSection(MemoryBufferSerializedModuleLoader &Loader,
StringRef buf,
const llvm::Triple &filter) {
if (!serialization::isSerializedAST(buf))
return llvm::make_error<ASTSectionParseError>(
serialization::Status::Malformed);
SmallVector<std::string, 4> foundModules;
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());
assert(info.name.size() < (2 << 10) && "name failed sanity check");
std::string error;
llvm::raw_string_ostream errs(error);
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 {
errs << "Unable to load module";
if (!info.name.empty())
errs << " '" << info.name << '\'';
errs << ".";
}
if (info.bytes == 0)
return llvm::make_error<ASTSectionParseError>(info.status, errs.str());
if (info.bytes > buf.size()) {
errs << "AST section too small.";
return llvm::make_error<ASTSectionParseError>(
serialization::Status::Malformed, errs.str());
}
buf = buf.substr(
llvm::alignTo(info.bytes, swift::serialization::SWIFTMODULE_ALIGNMENT));
}
return foundModules;
}
bool swift::parseASTSection(MemoryBufferSerializedModuleLoader &Loader,
StringRef buf,
const llvm::Triple &filter,
SmallVectorImpl<std::string> &foundModules) {
auto Result = parseASTSection(Loader, buf, filter);
if (auto E = Result.takeError()) {
llvm::dbgs() << toString(std::move(E));
return false;
}
for (auto m : *Result)
foundModules.push_back(m);
return true;
}
|
