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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
//===--- FrontendUtil.cpp - Driver Utilities for Frontend -----------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#include "swift/Driver/FrontendUtil.h"
#include "swift/AST/DiagnosticsDriver.h"
#include "swift/Driver/Action.h"
#include "swift/Driver/Compilation.h"
#include "swift/Driver/Driver.h"
#include "swift/Driver/Job.h"
#include "swift/Driver/ToolChain.h"
#include "llvm/ADT/DenseSet.h"
#include "llvm/Option/ArgList.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/StringSaver.h"
#include "llvm/TargetParser/Host.h"
using namespace swift;
using namespace swift::driver;
void swift::driver::ExpandResponseFilesWithRetry(llvm::StringSaver &Saver,
llvm::SmallVectorImpl<const char *> &Args) {
const unsigned MAX_COUNT = 30;
for (unsigned I = 0; I != MAX_COUNT; ++I) {
if (llvm::cl::ExpandResponseFiles(Saver,
llvm::Triple(llvm::sys::getProcessTriple()).isOSWindows()
? llvm::cl::TokenizeWindowsCommandLine
: llvm::cl::TokenizeGNUCommandLine,
Args)) {
return;
}
}
}
static void removeSupplementaryOutputs(llvm::opt::ArgList &ArgList) {
llvm::DenseSet<unsigned> OptSpecifiersToRemove;
for (llvm::opt::Arg *Arg : ArgList.getArgs()) {
if (!Arg)
continue;
const llvm::opt::Option &Opt = Arg->getOption();
if (Opt.hasFlag(options::SupplementaryOutput))
OptSpecifiersToRemove.insert(Opt.getID());
}
for (unsigned Specifier : OptSpecifiersToRemove) {
ArgList.eraseArg(Specifier);
}
}
bool swift::driver::getSingleFrontendInvocationFromDriverArguments(
StringRef DriverPath, ArrayRef<const char *> Argv, DiagnosticEngine &Diags,
llvm::function_ref<bool(ArrayRef<const char *> FrontendArgs)> Action,
bool ForceNoOutputs) {
SmallVector<const char *, 16> Args;
Args.push_back("<swiftc>"); // FIXME: Remove dummy argument.
Args.insert(Args.end(), Argv.begin(), Argv.end());
// When creating a CompilerInvocation, ensure that the driver creates a single
// frontend command.
Args.push_back("-whole-module-optimization");
// Explicitly disable batch mode to avoid a spurious warning when combining
// -enable-batch-mode with -whole-module-optimization. This is an
// implementation detail.
Args.push_back("-disable-batch-mode");
// Avoid using filelists
std::string neverThreshold =
std::to_string(Compilation::NEVER_USE_FILELIST);
Args.push_back("-driver-filelist-threshold");
Args.push_back(neverThreshold.c_str());
// Expand any file list args.
llvm::BumpPtrAllocator Allocator;
llvm::StringSaver Saver(Allocator);
ExpandResponseFilesWithRetry(Saver, Args);
// Force the driver into batch mode by specifying "swiftc" as the name.
Driver TheDriver(DriverPath, "swiftc", Args, Diags);
// Don't check for the existence of input files, since the user of the
// CompilerInvocation may wish to remap inputs to source buffers.
TheDriver.setCheckInputFilesExist(false);
TheDriver.setIsDummyDriverForFrontendInvocation(true);
std::unique_ptr<llvm::opt::InputArgList> ArgList =
TheDriver.parseArgStrings(ArrayRef<const char *>(Args).slice(1));
if (Diags.hadAnyError())
return true;
if (ForceNoOutputs) {
// Clear existing output modes and supplementary outputs.
ArgList->eraseArg(options::OPT_modes_Group);
removeSupplementaryOutputs(*ArgList);
unsigned index = ArgList->MakeIndex("-typecheck");
// Takes ownership of the Arg.
ArgList->append(new llvm::opt::Arg(
TheDriver.getOpts().getOption(options::OPT_typecheck),
ArgList->getArgString(index), index));
}
std::unique_ptr<ToolChain> TC = TheDriver.buildToolChain(*ArgList);
if (Diags.hadAnyError())
return true;
std::unique_ptr<Compilation> C =
TheDriver.buildCompilation(*TC, std::move(ArgList), /*AllowErrors=*/true);
if (!C || C->getJobs().empty())
return true; // Don't emit an error; one should already have been emitted
SmallPtrSet<const Job *, 4> CompileCommands;
for (const Job *Cmd : C->getJobs()) {
if (isa<CompileJobAction>(Cmd->getSource())) {
CompileCommands.insert(Cmd);
}
}
if (CompileCommands.size() != 1) {
// TODO: include Jobs in the diagnostic.
Diags.diagnose(SourceLoc(), diag::error_expected_one_frontend_job);
return true;
}
const Job *Cmd = *CompileCommands.begin();
if (StringRef("-frontend") != Cmd->getArguments().front()) {
Diags.diagnose(SourceLoc(), diag::error_expected_frontend_command);
return true;
}
const llvm::opt::ArgStringList &BaseFrontendArgs = Cmd->getArguments();
return Action(llvm::ArrayRef(BaseFrontendArgs).drop_front());
}
|
