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