diff --git a/clang/include/clang/Tooling/ReplacementsYaml.h b/clang/include/clang/Tooling/ReplacementsYaml.h --- a/clang/include/clang/Tooling/ReplacementsYaml.h +++ b/clang/include/clang/Tooling/ReplacementsYaml.h @@ -35,13 +35,7 @@ NormalizedReplacement(const IO &, const clang::tooling::Replacement &R) : FilePath(R.getFilePath()), Offset(R.getOffset()), - Length(R.getLength()), ReplacementText(R.getReplacementText()) { - size_t lineBreakPos = ReplacementText.find('\n'); - while (lineBreakPos != std::string::npos) { - ReplacementText.replace(lineBreakPos, 1, "\n\n"); - lineBreakPos = ReplacementText.find('\n', lineBreakPos + 2); - } - } + Length(R.getLength()), ReplacementText(R.getReplacementText()) {} clang::tooling::Replacement denormalize(const IO &) { return clang::tooling::Replacement(FilePath, Offset, Length, diff --git a/llvm/lib/Support/YAMLTraits.cpp b/llvm/lib/Support/YAMLTraits.cpp --- a/llvm/lib/Support/YAMLTraits.cpp +++ b/llvm/lib/Support/YAMLTraits.cpp @@ -876,13 +876,35 @@ } void ScalarTraits::output(const std::string &Val, void *, - raw_ostream &Out) { - Out << Val; + raw_ostream &Out) { + SmallVector Lines; + StringRef(Val).split(Lines, '\n'); + bool First = true; + for (StringRef Line : Lines) { + if (First) + First = false; + else + Out << "\n\n"; + Out << Line; + } } StringRef ScalarTraits::input(StringRef Scalar, void *, - std::string &Val) { - Val = Scalar.str(); + std::string &Val) { + Val.clear(); + SmallVector Lines; + Scalar.split(Lines, '\n'); + size_t C = Lines.size(); + for (size_t I = 0; I < C; ++I) { + Val += Lines[I]; + // Next empty line means that it was '\n\n' that should convert to '\n'. + // Single '\n' should be skipped. +2 here for the right handling + // single '\n' at the end of line. + if (I + 2 < C && Lines[I + 1].empty()) { + Val += '\n'; + ++I; + } + } return StringRef(); } diff --git a/llvm/unittests/Support/YAMLIOTest.cpp b/llvm/unittests/Support/YAMLIOTest.cpp --- a/llvm/unittests/Support/YAMLIOTest.cpp +++ b/llvm/unittests/Support/YAMLIOTest.cpp @@ -274,8 +274,8 @@ TEST(YAMLIO, MultilineStrings) { WithStringField Original; - Original.str1 = "a multiline string\nfoobarbaz"; - Original.str2 = "another one\rfoobarbaz"; + Original.str1 = "a\n\nmultiline\nstring\nfoobarbaz"; + Original.str2 = "another one\rfoobarbaz\n"; Original.str3 = "a one-line string"; std::string Serialized; @@ -285,10 +285,10 @@ YOut << Original; } auto Expected = "---\n" - "str1: 'a multiline string\n" + "str1: 'a\n\n\n\nmultiline\n\nstring\n\n" "foobarbaz'\n" "str2: 'another one\r" - "foobarbaz'\n" + "foobarbaz\n\n'\n" "str3: a one-line string\n" "...\n"; ASSERT_EQ(Serialized, Expected); @@ -305,6 +305,25 @@ EXPECT_EQ(Original.str1, Deserialized.str1); EXPECT_EQ(Original.str2, Deserialized.str2); EXPECT_EQ(Original.str3, Deserialized.str3); + + // Check deserialization of single '\n' that should be ignored. + Serialized = "---\n" + "str1: '\na\n\n\n\nmultiline\n\nstring\n\n" + "foobarbaz\n'\n" + "str2: 'another\n one\r" + "foobarbaz\n\n'\n" + "str3: a one-line string\n" + "...\n"; + { + Input YIn(Serialized); + YIn >> Deserialized; + ASSERT_FALSE(YIn.error()) + << "Parsing error occurred during deserialization. Serialized string:\n" + << Serialized; + } + EXPECT_EQ(Original.str1, Deserialized.str1); + EXPECT_EQ(Original.str2, Deserialized.str2); + EXPECT_EQ(Original.str3, Deserialized.str3); } TEST(YAMLIO, NoQuotesForTab) {