// Copyright 2020 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "fpdfsdk/cpdfsdk_helpers.h"

#include "testing/gtest/include/gtest/gtest.h"

TEST(CPDFSDK_HelpersTest, NulTerminateMaybeCopyAndReturnLength) {
  {
    const ByteString to_be_copied("toBeCopied");
    constexpr size_t kExpectedToBeCopiedLen = 10;
    ASSERT_EQ(kExpectedToBeCopiedLen, to_be_copied.GetLength());

    EXPECT_EQ(kExpectedToBeCopiedLen + 1,
              NulTerminateMaybeCopyAndReturnLength(to_be_copied, nullptr, 0));

    // Buffer should not change if declared length is too short.
    char buf[kExpectedToBeCopiedLen + 1];
    memset(buf, 0x42, kExpectedToBeCopiedLen + 1);
    ASSERT_EQ(kExpectedToBeCopiedLen + 1,
              NulTerminateMaybeCopyAndReturnLength(to_be_copied, buf,
                                                   kExpectedToBeCopiedLen));
    for (char c : buf)
      EXPECT_EQ(0x42, c);

    // Buffer should copy over if long enough.
    ASSERT_EQ(kExpectedToBeCopiedLen + 1,
              NulTerminateMaybeCopyAndReturnLength(to_be_copied, buf,
                                                   kExpectedToBeCopiedLen + 1));
    EXPECT_EQ(to_be_copied, ByteString(buf));
  }
  {
    // Empty ByteString should still copy NUL terminator.
    const ByteString empty;
    char buf[1];
    ASSERT_EQ(1u, NulTerminateMaybeCopyAndReturnLength(empty, buf, 1));
    EXPECT_EQ(empty, ByteString(buf));
  }
}

TEST(CPDFSDK_HelpersTest, ParsePageRangeString) {
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("", 1));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString(" ", 1));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("clams", 1));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("0", 0));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1", 0));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString(",1", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1,", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1,clams", 1));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("clams,1", 1));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("0-1", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-0", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-5", 4));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-11,", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString(",1-1", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-,", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("-2,", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-clams", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("clams-1,", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-2clams", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("0,1", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1,0", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-2,,,,3-4", 10));
  EXPECT_EQ(std::vector<uint32_t>({}), ParsePageRangeString("1-2-", 10));
  EXPECT_EQ(std::vector<uint32_t>({1}), ParsePageRangeString("1-1", 10));
  EXPECT_EQ(std::vector<uint32_t>{1}, ParsePageRangeString("1", 1));
  EXPECT_EQ(std::vector<uint32_t>({1, 2, 3, 4}),
            ParsePageRangeString("1-4", 4));
  EXPECT_EQ(std::vector<uint32_t>({1, 2, 3, 4}),
            ParsePageRangeString("1- 4", 4));
  EXPECT_EQ(std::vector<uint32_t>({1, 2, 3, 4}),
            ParsePageRangeString("1 -4", 4));
  EXPECT_EQ(std::vector<uint32_t>({1, 2}), ParsePageRangeString("1,2", 10));
  EXPECT_EQ(std::vector<uint32_t>({2, 1}), ParsePageRangeString("2,1", 10));
  EXPECT_EQ(std::vector<uint32_t>({1, 50, 2}),
            ParsePageRangeString("1,50,2", 100));
  EXPECT_EQ(std::vector<uint32_t>({1, 2, 3, 4, 50}),
            ParsePageRangeString("1-4,50", 100));
  EXPECT_EQ(std::vector<uint32_t>({50, 1, 2}),
            ParsePageRangeString("50,1-2", 100));
  EXPECT_EQ(std::vector<uint32_t>({50, 1, 2}),
            ParsePageRangeString("5  0, 1-2 ", 100));  // ???
  EXPECT_EQ(std::vector<uint32_t>({1, 2, 3, 4, 5, 6}),
            ParsePageRangeString("1-3,4-6", 10));
  EXPECT_EQ(std::vector<uint32_t>({1, 2, 3, 4, 3, 4, 5, 6}),
            ParsePageRangeString("1-4,3-6", 10));
}