mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
44f8e195ef
If the compiler arguments have errors in them (e.g. because a file with the same name is used twice), we can often still fulfill SourceKit requests because the compiler argument errors are only relevant for later stages of the compilation process. Instead of bailing out early, do a best effor retrieving the compiler arguments that are valid and ignoring the errors. Fixes rdar://77618144
127 lines
4.5 KiB
C++
127 lines
4.5 KiB
C++
//===--- 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/Option/ArgList.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Host.h"
|
|
#include "llvm/Support/StringSaver.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;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool swift::driver::getSingleFrontendInvocationFromDriverArguments(
|
|
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");
|
|
|
|
// Explictly 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("swiftc", "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);
|
|
ArgList->eraseArgIf([](const llvm::opt::Arg *A) {
|
|
return A && A->getOption().hasFlag(options::SupplementaryOutput);
|
|
});
|
|
|
|
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);
|
|
}
|
|
|
|
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::makeArrayRef(BaseFrontendArgs).drop_front());
|
|
}
|