chromium/src/third_party/blink/renderer/platform/wtf/text/line_ending.cc

/*
 * Copyright (C) 2005, 2006, 2008 Apple Inc. All rights reserved.
 * Copyright (C) 2010 Google Inc. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 *     * Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 * copyright notice, this list of conditions and the following disclaimer
 * in the documentation and/or other materials provided with the
 * distribution.
 *     * Neither the name of Google Inc. nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "third_party/blink/renderer/platform/wtf/text/line_ending.h"

#include "build/build_config.h"
#include "third_party/blink/renderer/platform/wtf/allocator/allocator.h"
#include "third_party/blink/renderer/platform/wtf/text/string_buffer.h"
#include "third_party/blink/renderer/platform/wtf/text/wtf_string.h"

namespace WTF {
namespace {

template <typename CharType>
wtf_size_t RequiredSizeForCRLF(const CharType* data, wtf_size_t length) {
  wtf_size_t new_len = 0;
  const CharType* p = data;
  while (p < data + length) {
    CharType c = *p++;
    if (c == '\r') {
      if (p >= data + length || *p != '\n') {
        // Turn CR into CRLF.
        new_len += 2;
      } else {
        // We already have \r\n. We don't count this \r, and the
        // following \n will count 2.
      }
    } else if (c == '\n') {
      // Turn LF into CRLF.
      new_len += 2;
    } else {
      // Leave other characters alone.
      new_len += 1;
    }
  }
  return new_len;
}

template <typename CharType>
void NormalizeToCRLF(const CharType* src, wtf_size_t src_length, CharType* q) {
  const CharType* p = src;
  while (p < src + src_length) {
    CharType c = *p++;
    if (c == '\r') {
      if (p >= src + src_length || *p != '\n') {
        // Turn CR into CRLF.
        *q++ = '\r';
        *q++ = '\n';
      }
    } else if (c == '\n') {
      // Turn LF into CRLF.
      *q++ = '\r';
      *q++ = '\n';
    } else {
      // Leave other characters alone.
      *q++ = c;
    }
  }
}

#if defined(OS_WIN)
void InternalNormalizeLineEndingsToCRLF(const std::string& from,
                                        Vector<char>& buffer) {
  size_t new_len = RequiredSizeForCRLF(from.c_str(), from.length());
  if (new_len < from.length())
    return;

  if (new_len == from.length()) {
    buffer.Append(from.c_str(), from.length());
    return;
  }

  wtf_size_t old_buffer_size = buffer.size();
  buffer.Grow(old_buffer_size + new_len);
  char* write_position = buffer.data() + old_buffer_size;
  NormalizeToCRLF(from.c_str(), from.length(), write_position);
}
#endif  // defined(OS_WIN)

}  // namespace

void NormalizeLineEndingsToLF(const std::string& from, Vector<char>& result) {
  // Compute the new length.
  wtf_size_t new_len = 0;
  bool need_fix = false;
  const char* p = from.c_str();
  char from_ending_char = '\r';
  char to_ending_char = '\n';
  while (p < from.c_str() + from.length()) {
    char c = *p++;
    if (c == '\r' && *p == '\n') {
      // Turn CRLF into CR or LF.
      p++;
      need_fix = true;
    } else if (c == from_ending_char) {
      // Turn CR/LF into LF/CR.
      need_fix = true;
    }
    new_len += 1;
  }

  // Grow the result buffer.
  p = from.c_str();
  wtf_size_t old_result_size = result.size();
  result.Grow(old_result_size + new_len);
  char* q = result.data() + old_result_size;

  // If no need to fix the string, just copy the string over.
  if (!need_fix) {
    memcpy(q, p, from.length());
    return;
  }

  // Make a copy of the string.
  while (p < from.c_str() + from.length()) {
    char c = *p++;
    if (c == '\r' && *p == '\n') {
      // Turn CRLF or CR into CR or LF.
      p++;
      *q++ = to_ending_char;
    } else if (c == from_ending_char) {
      // Turn CR/LF into LF/CR.
      *q++ = to_ending_char;
    } else {
      // Leave other characters alone.
      *q++ = c;
    }
  }
}

String NormalizeLineEndingsToCRLF(const String& src) {
  wtf_size_t length = src.length();
  if (length == 0)
    return src;
  if (src.Is8Bit()) {
    wtf_size_t new_length = RequiredSizeForCRLF(src.Characters8(), length);
    if (new_length == length)
      return src;
    StringBuffer<LChar> buffer(new_length);
    NormalizeToCRLF(src.Characters8(), length, buffer.Characters());
    return String::Adopt(buffer);
  }
  wtf_size_t new_length = RequiredSizeForCRLF(src.Characters16(), length);
  if (new_length == length)
    return src;
  StringBuffer<UChar> buffer(new_length);
  NormalizeToCRLF(src.Characters16(), length, buffer.Characters());
  return String::Adopt(buffer);
}

void NormalizeLineEndingsToNative(const std::string& from,
                                  Vector<char>& result) {
#if defined(OS_WIN)
  InternalNormalizeLineEndingsToCRLF(from, result);
#else
  NormalizeLineEndingsToLF(from, result);
#endif
}

}  // namespace WTF