diff --git a/lib/Option/OptTable.cpp b/lib/Option/OptTable.cpp index 47c5f80c7d7..5833d03069f 100644 --- a/lib/Option/OptTable.cpp +++ b/lib/Option/OptTable.cpp @@ -301,6 +301,15 @@ unsigned OptTable::findNearest(StringRef Option, std::string &NearestString, unsigned Distance = CandidateRef.edit_distance(NormalizedName, /*AllowReplacements=*/true, /*MaxEditDistance=*/BestDistance); + if (RHS.empty() && CandidateHasDelimiter) { + // The Candidate ends with a = or : delimiter, but the option passed in + // didn't contain the delimiter (or doesn't have anything after it). + // In that case, penalize the correction: `-nodefaultlibs` is more + // likely to be a spello for `-nodefaultlib` than `-nodefaultlib:` even + // though both have an unmodified editing distance of 1, since the + // latter would need an argument. + ++Distance; + } if (Distance < BestDistance) { BestDistance = Distance; NearestString = (Candidate + RHS).str(); diff --git a/unittests/Option/OptionParsingTest.cpp b/unittests/Option/OptionParsingTest.cpp index ee741c25d05..e1d7a473ee7 100644 --- a/unittests/Option/OptionParsingTest.cpp +++ b/unittests/Option/OptionParsingTest.cpp @@ -298,10 +298,19 @@ TEST(Option, FindNearest) { EXPECT_EQ(1U, T.findNearest("/framb:foo", Nearest)); EXPECT_EQ(Nearest, "/cramb:foo"); - // `--glormp` should have an editing distance of 1 to `--glormp=`. - EXPECT_EQ(1U, T.findNearest("--glormp", Nearest)); - EXPECT_EQ(Nearest, "--glormp="); - EXPECT_EQ(0U, T.findNearest("--glormp=foo", Nearest)); + // `--glormp` should have an editing distance > 0 from `--glormp=`. + EXPECT_GT(T.findNearest("--glorrmp", Nearest), 0U); + EXPECT_EQ(Nearest, "--glorrmp="); + EXPECT_EQ(0U, T.findNearest("--glorrmp=foo", Nearest)); + + // `--blurmps` should correct to `--blurmp`, not `--blurmp=`, even though + // both naively have an editing distance of 1. + EXPECT_EQ(1U, T.findNearest("--blurmps", Nearest)); + EXPECT_EQ(Nearest, "--blurmp"); + + // ...but `--blurmps=foo` should correct to `--blurmp=foo`. + EXPECT_EQ(1U, T.findNearest("--blurmps=foo", Nearest)); + EXPECT_EQ(Nearest, "--blurmp=foo"); // Flags should be included and excluded as specified. EXPECT_EQ(1U, T.findNearest("-doopf", Nearest, /*FlagsToInclude=*/OptFlag2)); diff --git a/unittests/Option/Opts.td b/unittests/Option/Opts.td index 231af4914d8..70920a6a76b 100644 --- a/unittests/Option/Opts.td +++ b/unittests/Option/Opts.td @@ -37,6 +37,9 @@ def Doopf2 : Flag<["-"], "doopf2">, HelpText<"The doopf2 option">, Flags<[OptFla def Ermgh : Joined<["--"], "ermgh">, HelpText<"The ermgh option">, MetaVarName<"ERMGH">, Flags<[OptFlag1]>; def Fjormp : Flag<["--"], "fjormp">, HelpText<"The fjormp option">, Flags<[OptFlag1]>; -def Glormp_eq : Flag<["--"], "glormp=">; +def Glorrmp_eq : Flag<["--"], "glorrmp=">; + +def Blurmpq : Flag<["--"], "blurmp">; +def Blurmpq_eq : Flag<["--"], "blurmp=">; def DashDash : Option<["--"], "", KIND_REMAINING_ARGS>;