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 @@ -887,13 +887,32 @@ } void ScalarTraits::output(const std::string &Val, void *, - raw_ostream &Out) { - Out << Val; + raw_ostream &Out) { + size_t currentPos = 0; + size_t lineBreakPos = Val.find('\n'); + while (lineBreakPos != std::string::npos) { + Out << StringRef(&Val[currentPos], lineBreakPos - currentPos + 1); + Out << '\n'; + currentPos = lineBreakPos + 1; + lineBreakPos = Val.find('\n', currentPos); + } + Out << StringRef(&Val[currentPos], Val.size() - currentPos); } StringRef ScalarTraits::input(StringRef Scalar, void *, - std::string &Val) { - Val = Scalar.str(); + std::string &Val) { + Val.clear(); + size_t currentPos = 0; + size_t lineBreakPos = Scalar.find('\n'); + while (lineBreakPos != std::string::npos) { + // '\n\n' convert to '\n' and don't copy single '\n'. + if (currentPos + 1 < Scalar.size() && Scalar[lineBreakPos + 1] == '\n') + ++lineBreakPos; + Val += Scalar.substr(currentPos, lineBreakPos); + currentPos = lineBreakPos + 1; + lineBreakPos = Scalar.find('\n', currentPos); + } + Val += Scalar.substr(currentPos, Scalar.size() - currentPos); 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 @@ -285,7 +285,7 @@ YOut << Original; } auto Expected = "---\n" - "str1: 'a multiline string\n" + "str1: 'a multiline string\n\n" "foobarbaz'\n" "str2: 'another one\r" "foobarbaz'\n"