1 //===--- CompileCommands.cpp ----------------------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "CompileCommands.h" 10 #include "Config.h" 11 #include "support/Logger.h" 12 #include "support/Trace.h" 13 #include "clang/Driver/Driver.h" 14 #include "clang/Driver/Options.h" 15 #include "clang/Driver/ToolChain.h" 16 #include "clang/Frontend/CompilerInvocation.h" 17 #include "clang/Tooling/ArgumentsAdjusters.h" 18 #include "llvm/ADT/ArrayRef.h" 19 #include "llvm/ADT/STLExtras.h" 20 #include "llvm/ADT/SmallVector.h" 21 #include "llvm/ADT/StringRef.h" 22 #include "llvm/Option/ArgList.h" 23 #include "llvm/Option/Option.h" 24 #include "llvm/Support/Allocator.h" 25 #include "llvm/Support/Debug.h" 26 #include "llvm/Support/FileSystem.h" 27 #include "llvm/Support/FileUtilities.h" 28 #include "llvm/Support/MemoryBuffer.h" 29 #include "llvm/Support/Path.h" 30 #include "llvm/Support/Program.h" 31 #include <iterator> 32 #include <string> 33 #include <vector> 34 35 namespace clang { 36 namespace clangd { 37 namespace { 38 39 // Query apple's `xcrun` launcher, which is the source of truth for "how should" 40 // clang be invoked on this system. 41 llvm::Optional<std::string> queryXcrun(llvm::ArrayRef<llvm::StringRef> Argv) { 42 auto Xcrun = llvm::sys::findProgramByName("xcrun"); 43 if (!Xcrun) { 44 log("Couldn't find xcrun. Hopefully you have a non-apple toolchain..."); 45 return llvm::None; 46 } 47 llvm::SmallString<64> OutFile; 48 llvm::sys::fs::createTemporaryFile("clangd-xcrun", "", OutFile); 49 llvm::FileRemover OutRemover(OutFile); 50 llvm::Optional<llvm::StringRef> Redirects[3] = { 51 /*stdin=*/{""}, /*stdout=*/{OutFile}, /*stderr=*/{""}}; 52 vlog("Invoking {0} to find clang installation", *Xcrun); 53 int Ret = llvm::sys::ExecuteAndWait(*Xcrun, Argv, 54 /*Env=*/llvm::None, Redirects, 55 /*SecondsToWait=*/10); 56 if (Ret != 0) { 57 log("xcrun exists but failed with code {0}. " 58 "If you have a non-apple toolchain, this is OK. " 59 "Otherwise, try xcode-select --install.", 60 Ret); 61 return llvm::None; 62 } 63 64 auto Buf = llvm::MemoryBuffer::getFile(OutFile); 65 if (!Buf) { 66 log("Can't read xcrun output: {0}", Buf.getError().message()); 67 return llvm::None; 68 } 69 StringRef Path = Buf->get()->getBuffer().trim(); 70 if (Path.empty()) { 71 log("xcrun produced no output"); 72 return llvm::None; 73 } 74 return Path.str(); 75 } 76 77 // Resolve symlinks if possible. 78 std::string resolve(std::string Path) { 79 llvm::SmallString<128> Resolved; 80 if (llvm::sys::fs::real_path(Path, Resolved)) { 81 log("Failed to resolve possible symlink {0}", Path); 82 return Path; 83 } 84 return std::string(Resolved.str()); 85 } 86 87 // Get a plausible full `clang` path. 88 // This is used in the fallback compile command, or when the CDB returns a 89 // generic driver with no path. 90 std::string detectClangPath() { 91 // The driver and/or cc1 sometimes depend on the binary name to compute 92 // useful things like the standard library location. 93 // We need to emulate what clang on this system is likely to see. 94 // cc1 in particular looks at the "real path" of the running process, and 95 // so if /usr/bin/clang is a symlink, it sees the resolved path. 96 // clangd doesn't have that luxury, so we resolve symlinks ourselves. 97 98 // On Mac, `which clang` is /usr/bin/clang. It runs `xcrun clang`, which knows 99 // where the real clang is kept. We need to do the same thing, 100 // because cc1 (not the driver!) will find libc++ relative to argv[0]. 101 #ifdef __APPLE__ 102 if (auto MacClang = queryXcrun({"xcrun", "--find", "clang"})) 103 return resolve(std::move(*MacClang)); 104 #endif 105 // On other platforms, just look for compilers on the PATH. 106 for (const char *Name : {"clang", "gcc", "cc"}) 107 if (auto PathCC = llvm::sys::findProgramByName(Name)) 108 return resolve(std::move(*PathCC)); 109 // Fallback: a nonexistent 'clang' binary next to clangd. 110 static int StaticForMainAddr; 111 std::string ClangdExecutable = 112 llvm::sys::fs::getMainExecutable("clangd", (void *)&StaticForMainAddr); 113 SmallString<128> ClangPath; 114 ClangPath = llvm::sys::path::parent_path(ClangdExecutable); 115 llvm::sys::path::append(ClangPath, "clang"); 116 return std::string(ClangPath.str()); 117 } 118 119 // On mac, /usr/bin/clang sets SDKROOT and then invokes the real clang. 120 // The effect of this is to set -isysroot correctly. We do the same. 121 const llvm::Optional<std::string> detectSysroot() { 122 #ifndef __APPLE__ 123 return llvm::None; 124 #endif 125 126 // SDKROOT overridden in environment, respect it. Driver will set isysroot. 127 if (::getenv("SDKROOT")) 128 return llvm::None; 129 return queryXcrun({"xcrun", "--show-sdk-path"}); 130 } 131 132 std::string detectStandardResourceDir() { 133 static int StaticForMainAddr; // Just an address in this process. 134 return CompilerInvocation::GetResourcesPath("clangd", 135 (void *)&StaticForMainAddr); 136 } 137 138 // The path passed to argv[0] is important: 139 // - its parent directory is Driver::Dir, used for library discovery 140 // - its basename affects CLI parsing (clang-cl) and other settings 141 // Where possible it should be an absolute path with sensible directory, but 142 // with the original basename. 143 static std::string resolveDriver(llvm::StringRef Driver, bool FollowSymlink, 144 llvm::Optional<std::string> ClangPath) { 145 auto SiblingOf = [&](llvm::StringRef AbsPath) { 146 llvm::SmallString<128> Result = llvm::sys::path::parent_path(AbsPath); 147 llvm::sys::path::append(Result, llvm::sys::path::filename(Driver)); 148 return Result.str().str(); 149 }; 150 151 // First, eliminate relative paths. 152 std::string Storage; 153 if (!llvm::sys::path::is_absolute(Driver)) { 154 // If it's working-dir relative like bin/clang, we can't resolve it. 155 // FIXME: we could if we had the working directory here. 156 // Let's hope it's not a symlink. 157 if (llvm::any_of(Driver, 158 [](char C) { return llvm::sys::path::is_separator(C); })) 159 return Driver.str(); 160 // If the driver is a generic like "g++" with no path, add clang dir. 161 if (ClangPath && 162 (Driver == "clang" || Driver == "clang++" || Driver == "gcc" || 163 Driver == "g++" || Driver == "cc" || Driver == "c++")) { 164 return SiblingOf(*ClangPath); 165 } 166 // Otherwise try to look it up on PATH. This won't change basename. 167 auto Absolute = llvm::sys::findProgramByName(Driver); 168 if (Absolute && llvm::sys::path::is_absolute(*Absolute)) 169 Driver = Storage = std::move(*Absolute); 170 else if (ClangPath) // If we don't find it, use clang dir again. 171 return SiblingOf(*ClangPath); 172 else // Nothing to do: can't find the command and no detected dir. 173 return Driver.str(); 174 } 175 176 // Now we have an absolute path, but it may be a symlink. 177 assert(llvm::sys::path::is_absolute(Driver)); 178 if (FollowSymlink) { 179 llvm::SmallString<256> Resolved; 180 if (!llvm::sys::fs::real_path(Driver, Resolved)) 181 return SiblingOf(Resolved); 182 } 183 return Driver.str(); 184 } 185 186 } // namespace 187 188 CommandMangler CommandMangler::detect() { 189 CommandMangler Result; 190 Result.ClangPath = detectClangPath(); 191 Result.ResourceDir = detectStandardResourceDir(); 192 Result.Sysroot = detectSysroot(); 193 return Result; 194 } 195 196 CommandMangler CommandMangler::forTests() { return CommandMangler(); } 197 198 void CommandMangler::adjust(std::vector<std::string> &Cmd, 199 llvm::StringRef File) const { 200 trace::Span S("AdjustCompileFlags"); 201 // Most of the modifications below assumes the Cmd starts with a driver name. 202 // We might consider injecting a generic driver name like "cc" or "c++", but 203 // a Cmd missing the driver is probably rare enough in practice and errnous. 204 if (Cmd.empty()) 205 return; 206 auto &OptTable = clang::driver::getDriverOptTable(); 207 // OriginalArgs needs to outlive ArgList. 208 llvm::SmallVector<const char *, 16> OriginalArgs; 209 OriginalArgs.reserve(Cmd.size()); 210 for (const auto &S : Cmd) 211 OriginalArgs.push_back(S.c_str()); 212 bool IsCLMode = driver::IsClangCL(driver::getDriverMode( 213 OriginalArgs[0], llvm::makeArrayRef(OriginalArgs).slice(1))); 214 // ParseArgs propagates missig arg/opt counts on error, but preserves 215 // everything it could parse in ArgList. So we just ignore those counts. 216 unsigned IgnoredCount; 217 // Drop the executable name, as ParseArgs doesn't expect it. This means 218 // indices are actually of by one between ArgList and OriginalArgs. 219 llvm::opt::InputArgList ArgList; 220 ArgList = OptTable.ParseArgs( 221 llvm::makeArrayRef(OriginalArgs).drop_front(), IgnoredCount, IgnoredCount, 222 /*FlagsToInclude=*/ 223 IsCLMode ? (driver::options::CLOption | driver::options::CoreOption) 224 : /*everything*/ 0, 225 /*FlagsToExclude=*/driver::options::NoDriverOption | 226 (IsCLMode ? 0 : driver::options::CLOption)); 227 228 llvm::SmallVector<unsigned, 1> IndicesToDrop; 229 // Having multiple architecture options (e.g. when building fat binaries) 230 // results in multiple compiler jobs, which clangd cannot handle. In such 231 // cases strip all the `-arch` options and fallback to default architecture. 232 // As there are no signals to figure out which one user actually wants. They 233 // can explicitly specify one through `CompileFlags.Add` if need be. 234 unsigned ArchOptCount = 0; 235 for (auto *Input : ArgList.filtered(driver::options::OPT_arch)) { 236 ++ArchOptCount; 237 for (auto I = 0U; I <= Input->getNumValues(); ++I) 238 IndicesToDrop.push_back(Input->getIndex() + I); 239 } 240 // If there is a single `-arch` option, keep it. 241 if (ArchOptCount < 2) 242 IndicesToDrop.clear(); 243 244 // Strip all the inputs and `--`. We'll put the input for the requested file 245 // explicitly at the end of the flags. This ensures modifications done in the 246 // following steps apply in more cases (like setting -x, which only affects 247 // inputs that come after it). 248 for (auto *Input : ArgList.filtered(driver::options::OPT_INPUT)) 249 IndicesToDrop.push_back(Input->getIndex()); 250 // Anything after `--` is also treated as input, drop them as well. 251 if (auto *DashDash = 252 ArgList.getLastArgNoClaim(driver::options::OPT__DASH_DASH)) { 253 Cmd.resize(DashDash->getIndex() + 1); // +1 to account for Cmd[0]. 254 } 255 llvm::sort(IndicesToDrop); 256 llvm::for_each(llvm::reverse(IndicesToDrop), 257 // +1 to account for the executable name in Cmd[0] that 258 // doesn't exist in ArgList. 259 [&Cmd](unsigned Idx) { Cmd.erase(Cmd.begin() + Idx + 1); }); 260 // All the inputs are stripped, append the name for the requested file. Rest 261 // of the modifications should respect `--`. 262 Cmd.push_back("--"); 263 Cmd.push_back(File.str()); 264 265 for (auto &Edit : Config::current().CompileFlags.Edits) 266 Edit(Cmd); 267 268 // Check whether the flag exists, either as -flag or -flag=* 269 auto Has = [&](llvm::StringRef Flag) { 270 for (llvm::StringRef Arg : Cmd) { 271 if (Arg.consume_front(Flag) && (Arg.empty() || Arg[0] == '=')) 272 return true; 273 } 274 return false; 275 }; 276 277 llvm::erase_if(Cmd, [](llvm::StringRef Elem) { 278 return Elem.startswith("--save-temps") || Elem.startswith("-save-temps"); 279 }); 280 281 std::vector<std::string> ToAppend; 282 if (ResourceDir && !Has("-resource-dir")) 283 ToAppend.push_back(("-resource-dir=" + *ResourceDir)); 284 285 // Don't set `-isysroot` if it is already set or if `--sysroot` is set. 286 // `--sysroot` is a superset of the `-isysroot` argument. 287 if (Sysroot && !Has("-isysroot") && !Has("--sysroot")) { 288 ToAppend.push_back("-isysroot"); 289 ToAppend.push_back(*Sysroot); 290 } 291 292 if (!ToAppend.empty()) { 293 Cmd.insert(llvm::find(Cmd, "--"), std::make_move_iterator(ToAppend.begin()), 294 std::make_move_iterator(ToAppend.end())); 295 } 296 297 if (!Cmd.empty()) { 298 bool FollowSymlink = !Has("-no-canonical-prefixes"); 299 Cmd.front() = 300 (FollowSymlink ? ResolvedDrivers : ResolvedDriversNoFollow) 301 .get(Cmd.front(), [&, this] { 302 return resolveDriver(Cmd.front(), FollowSymlink, ClangPath); 303 }); 304 } 305 } 306 307 CommandMangler::operator clang::tooling::ArgumentsAdjuster() && { 308 // ArgumentsAdjuster is a std::function and so must be copyable. 309 return [Mangler = std::make_shared<CommandMangler>(std::move(*this))]( 310 const std::vector<std::string> &Args, llvm::StringRef File) { 311 auto Result = Args; 312 Mangler->adjust(Result, File); 313 return Result; 314 }; 315 } 316 317 // ArgStripper implementation 318 namespace { 319 320 // Determine total number of args consumed by this option. 321 // Return answers for {Exact, Prefix} match. 0 means not allowed. 322 std::pair<unsigned, unsigned> getArgCount(const llvm::opt::Option &Opt) { 323 constexpr static unsigned Rest = 10000; // Should be all the rest! 324 // Reference is llvm::opt::Option::acceptInternal() 325 using llvm::opt::Option; 326 switch (Opt.getKind()) { 327 case Option::FlagClass: 328 return {1, 0}; 329 case Option::JoinedClass: 330 case Option::CommaJoinedClass: 331 return {1, 1}; 332 case Option::GroupClass: 333 case Option::InputClass: 334 case Option::UnknownClass: 335 case Option::ValuesClass: 336 return {1, 0}; 337 case Option::JoinedAndSeparateClass: 338 return {2, 2}; 339 case Option::SeparateClass: 340 return {2, 0}; 341 case Option::MultiArgClass: 342 return {1 + Opt.getNumArgs(), 0}; 343 case Option::JoinedOrSeparateClass: 344 return {2, 1}; 345 case Option::RemainingArgsClass: 346 return {Rest, 0}; 347 case Option::RemainingArgsJoinedClass: 348 return {Rest, Rest}; 349 } 350 llvm_unreachable("Unhandled option kind"); 351 } 352 353 // Flag-parsing mode, which affects which flags are available. 354 enum DriverMode : unsigned char { 355 DM_None = 0, 356 DM_GCC = 1, // Default mode e.g. when invoked as 'clang' 357 DM_CL = 2, // MS CL.exe compatible mode e.g. when invoked as 'clang-cl' 358 DM_CC1 = 4, // When invoked as 'clang -cc1' or after '-Xclang' 359 DM_All = 7 360 }; 361 362 // Examine args list to determine if we're in GCC, CL-compatible, or cc1 mode. 363 DriverMode getDriverMode(const std::vector<std::string> &Args) { 364 DriverMode Mode = DM_GCC; 365 llvm::StringRef Argv0 = Args.front(); 366 if (Argv0.endswith_insensitive(".exe")) 367 Argv0 = Argv0.drop_back(strlen(".exe")); 368 if (Argv0.endswith_insensitive("cl")) 369 Mode = DM_CL; 370 for (const llvm::StringRef Arg : Args) { 371 if (Arg == "--driver-mode=cl") { 372 Mode = DM_CL; 373 break; 374 } 375 if (Arg == "-cc1") { 376 Mode = DM_CC1; 377 break; 378 } 379 } 380 return Mode; 381 } 382 383 // Returns the set of DriverModes where an option may be used. 384 unsigned char getModes(const llvm::opt::Option &Opt) { 385 // Why is this so complicated?! 386 // Reference is clang::driver::Driver::getIncludeExcludeOptionFlagMasks() 387 unsigned char Result = DM_None; 388 if (Opt.hasFlag(driver::options::CC1Option)) 389 Result |= DM_CC1; 390 if (!Opt.hasFlag(driver::options::NoDriverOption)) { 391 if (Opt.hasFlag(driver::options::CLOption)) { 392 Result |= DM_CL; 393 } else { 394 Result |= DM_GCC; 395 if (Opt.hasFlag(driver::options::CoreOption)) { 396 Result |= DM_CL; 397 } 398 } 399 } 400 return Result; 401 } 402 403 } // namespace 404 405 llvm::ArrayRef<ArgStripper::Rule> ArgStripper::rulesFor(llvm::StringRef Arg) { 406 // All the hard work is done once in a static initializer. 407 // We compute a table containing strings to look for and #args to skip. 408 // e.g. "-x" => {-x 2 args, -x* 1 arg, --language 2 args, --language=* 1 arg} 409 using TableTy = 410 llvm::StringMap<llvm::SmallVector<Rule, 4>, llvm::BumpPtrAllocator>; 411 static TableTy *Table = [] { 412 auto &DriverTable = driver::getDriverOptTable(); 413 using DriverID = clang::driver::options::ID; 414 415 // Collect sets of aliases, so we can treat -foo and -foo= as synonyms. 416 // Conceptually a double-linked list: PrevAlias[I] -> I -> NextAlias[I]. 417 // If PrevAlias[I] is INVALID, then I is canonical. 418 DriverID PrevAlias[DriverID::LastOption] = {DriverID::OPT_INVALID}; 419 DriverID NextAlias[DriverID::LastOption] = {DriverID::OPT_INVALID}; 420 auto AddAlias = [&](DriverID Self, DriverID T) { 421 if (NextAlias[T]) { 422 PrevAlias[NextAlias[T]] = Self; 423 NextAlias[Self] = NextAlias[T]; 424 } 425 PrevAlias[Self] = T; 426 NextAlias[T] = Self; 427 }; 428 // Also grab prefixes for each option, these are not fully exposed. 429 const char *const *Prefixes[DriverID::LastOption] = {nullptr}; 430 #define PREFIX(NAME, VALUE) static const char *const NAME[] = VALUE; 431 #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \ 432 HELP, METAVAR, VALUES) \ 433 if (DriverID::OPT_##ALIAS != DriverID::OPT_INVALID && ALIASARGS == nullptr) \ 434 AddAlias(DriverID::OPT_##ID, DriverID::OPT_##ALIAS); \ 435 Prefixes[DriverID::OPT_##ID] = PREFIX; 436 #include "clang/Driver/Options.inc" 437 #undef OPTION 438 #undef PREFIX 439 440 auto Result = std::make_unique<TableTy>(); 441 // Iterate over distinct options (represented by the canonical alias). 442 // Every spelling of this option will get the same set of rules. 443 for (unsigned ID = 1 /*Skip INVALID */; ID < DriverID::LastOption; ++ID) { 444 if (PrevAlias[ID] || ID == DriverID::OPT_Xclang) 445 continue; // Not canonical, or specially handled. 446 llvm::SmallVector<Rule> Rules; 447 // Iterate over each alias, to add rules for parsing it. 448 for (unsigned A = ID; A != DriverID::OPT_INVALID; A = NextAlias[A]) { 449 if (Prefixes[A] == nullptr) // option groups. 450 continue; 451 auto Opt = DriverTable.getOption(A); 452 // Exclude - and -foo pseudo-options. 453 if (Opt.getName().empty()) 454 continue; 455 auto Modes = getModes(Opt); 456 std::pair<unsigned, unsigned> ArgCount = getArgCount(Opt); 457 // Iterate over each spelling of the alias, e.g. -foo vs --foo. 458 for (auto *Prefix = Prefixes[A]; *Prefix != nullptr; ++Prefix) { 459 llvm::SmallString<64> Buf(*Prefix); 460 Buf.append(Opt.getName()); 461 llvm::StringRef Spelling = Result->try_emplace(Buf).first->getKey(); 462 Rules.emplace_back(); 463 Rule &R = Rules.back(); 464 R.Text = Spelling; 465 R.Modes = Modes; 466 R.ExactArgs = ArgCount.first; 467 R.PrefixArgs = ArgCount.second; 468 // Concrete priority is the index into the option table. 469 // Effectively, earlier entries take priority over later ones. 470 assert(ID < std::numeric_limits<decltype(R.Priority)>::max() && 471 "Rules::Priority overflowed by options table"); 472 R.Priority = ID; 473 } 474 } 475 // Register the set of rules under each possible name. 476 for (const auto &R : Rules) 477 Result->find(R.Text)->second.append(Rules.begin(), Rules.end()); 478 } 479 #ifndef NDEBUG 480 // Dump the table and various measures of its size. 481 unsigned RuleCount = 0; 482 dlog("ArgStripper Option spelling table"); 483 for (const auto &Entry : *Result) { 484 dlog("{0}", Entry.first()); 485 RuleCount += Entry.second.size(); 486 for (const auto &R : Entry.second) 487 dlog(" {0} #={1} *={2} Mode={3}", R.Text, R.ExactArgs, R.PrefixArgs, 488 int(R.Modes)); 489 } 490 dlog("Table spellings={0} rules={1} string-bytes={2}", Result->size(), 491 RuleCount, Result->getAllocator().getBytesAllocated()); 492 #endif 493 // The static table will never be destroyed. 494 return Result.release(); 495 }(); 496 497 auto It = Table->find(Arg); 498 return (It == Table->end()) ? llvm::ArrayRef<Rule>() : It->second; 499 } 500 501 void ArgStripper::strip(llvm::StringRef Arg) { 502 auto OptionRules = rulesFor(Arg); 503 if (OptionRules.empty()) { 504 // Not a recognized flag. Strip it literally. 505 Storage.emplace_back(Arg); 506 Rules.emplace_back(); 507 Rules.back().Text = Storage.back(); 508 Rules.back().ExactArgs = 1; 509 if (Rules.back().Text.consume_back("*")) 510 Rules.back().PrefixArgs = 1; 511 Rules.back().Modes = DM_All; 512 Rules.back().Priority = -1; // Max unsigned = lowest priority. 513 } else { 514 Rules.append(OptionRules.begin(), OptionRules.end()); 515 } 516 } 517 518 const ArgStripper::Rule *ArgStripper::matchingRule(llvm::StringRef Arg, 519 unsigned Mode, 520 unsigned &ArgCount) const { 521 const ArgStripper::Rule *BestRule = nullptr; 522 for (const Rule &R : Rules) { 523 // Rule can fail to match if... 524 if (!(R.Modes & Mode)) 525 continue; // not applicable to current driver mode 526 if (BestRule && BestRule->Priority < R.Priority) 527 continue; // lower-priority than best candidate. 528 if (!Arg.startswith(R.Text)) 529 continue; // current arg doesn't match the prefix string 530 bool PrefixMatch = Arg.size() > R.Text.size(); 531 // Can rule apply as an exact/prefix match? 532 if (unsigned Count = PrefixMatch ? R.PrefixArgs : R.ExactArgs) { 533 BestRule = &R; 534 ArgCount = Count; 535 } 536 // Continue in case we find a higher-priority rule. 537 } 538 return BestRule; 539 } 540 541 void ArgStripper::process(std::vector<std::string> &Args) const { 542 if (Args.empty()) 543 return; 544 545 // We're parsing the args list in some mode (e.g. gcc-compatible) but may 546 // temporarily switch to another mode with the -Xclang flag. 547 DriverMode MainMode = getDriverMode(Args); 548 DriverMode CurrentMode = MainMode; 549 550 // Read and write heads for in-place deletion. 551 unsigned Read = 0, Write = 0; 552 bool WasXclang = false; 553 while (Read < Args.size()) { 554 unsigned ArgCount = 0; 555 if (matchingRule(Args[Read], CurrentMode, ArgCount)) { 556 // Delete it and its args. 557 if (WasXclang) { 558 assert(Write > 0); 559 --Write; // Drop previous -Xclang arg 560 CurrentMode = MainMode; 561 WasXclang = false; 562 } 563 // Advance to last arg. An arg may be foo or -Xclang foo. 564 for (unsigned I = 1; Read < Args.size() && I < ArgCount; ++I) { 565 ++Read; 566 if (Read < Args.size() && Args[Read] == "-Xclang") 567 ++Read; 568 } 569 } else { 570 // No match, just copy the arg through. 571 WasXclang = Args[Read] == "-Xclang"; 572 CurrentMode = WasXclang ? DM_CC1 : MainMode; 573 if (Write != Read) 574 Args[Write] = std::move(Args[Read]); 575 ++Write; 576 } 577 ++Read; 578 } 579 Args.resize(Write); 580 } 581 582 std::string printArgv(llvm::ArrayRef<llvm::StringRef> Args) { 583 std::string Buf; 584 llvm::raw_string_ostream OS(Buf); 585 bool Sep = false; 586 for (llvm::StringRef Arg : Args) { 587 if (Sep) 588 OS << ' '; 589 Sep = true; 590 if (llvm::all_of(Arg, llvm::isPrint) && 591 Arg.find_first_of(" \t\n\"\\") == llvm::StringRef::npos) { 592 OS << Arg; 593 continue; 594 } 595 OS << '"'; 596 OS.write_escaped(Arg, /*UseHexEscapes=*/true); 597 OS << '"'; 598 } 599 return std::move(OS.str()); 600 } 601 602 std::string printArgv(llvm::ArrayRef<std::string> Args) { 603 std::vector<llvm::StringRef> Refs(Args.size()); 604 llvm::copy(Args, Refs.begin()); 605 return printArgv(Refs); 606 } 607 608 } // namespace clangd 609 } // namespace clang 610