// Copyright 2016 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 <limits>
#include <string>

#include "core/fpdfapi/parser/cpdf_object.h"
#include "core/fpdfapi/parser/cpdf_parser.h"
#include "core/fpdfapi/parser/cpdf_syntax_parser.h"
#include "core/fxcrt/cfx_memorystream.h"
#include "core/fxcrt/fx_extension.h"
#include "core/fxcrt/fx_stream.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/utils/path_service.h"

TEST(cpdf_syntax_parser, ReadHexString) {
  {
    // Empty string.
    uint8_t data[] = "";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 0, false), 0);
    EXPECT_EQ("", parser.ReadHexString());
    EXPECT_EQ(0, parser.SavePos());
  }

  {
    // Blank string.
    uint8_t data[] = "  ";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 2, false), 0);
    EXPECT_EQ("", parser.ReadHexString());
    EXPECT_EQ(2, parser.SavePos());
  }

  {
    // Skips unknown characters.
    uint8_t data[] = "z12b";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 4, false), 0);
    EXPECT_EQ("\x12\xb0", parser.ReadHexString());
    EXPECT_EQ(4, parser.SavePos());
  }

  {
    // Skips unknown characters.
    uint8_t data[] = "*<&*#$^&@1";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 10, false), 0);
    EXPECT_EQ("\x10", parser.ReadHexString());
    EXPECT_EQ(10, parser.SavePos());
  }

  {
    // Skips unknown characters.
    uint8_t data[] = "\x80zab";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 4, false), 0);
    EXPECT_EQ("\xab", parser.ReadHexString());
    EXPECT_EQ(4, parser.SavePos());
  }

  {
    // Skips unknown characters.
    uint8_t data[] = "\xffzab";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 4, false), 0);
    EXPECT_EQ("\xab", parser.ReadHexString());
    EXPECT_EQ(4, parser.SavePos());
  }

  {
    // Regular conversion.
    uint8_t data[] = "1A2b>abcd";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 9, false), 0);
    EXPECT_EQ("\x1a\x2b", parser.ReadHexString());
    EXPECT_EQ(5, parser.SavePos());
  }

  {
    // Position out of bounds.
    uint8_t data[] = "12ab>";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 5, false), 0);
    parser.RestorePos(5);
    EXPECT_EQ("", parser.ReadHexString());

    parser.RestorePos(6);
    EXPECT_EQ("", parser.ReadHexString());

    parser.RestorePos(-1);
    EXPECT_EQ("", parser.ReadHexString());

    parser.RestorePos(std::numeric_limits<FX_FILESIZE>::max());
    EXPECT_EQ("", parser.ReadHexString());

    // Check string still parses when set to 0.
    parser.RestorePos(0);
    EXPECT_EQ("\x12\xab", parser.ReadHexString());
  }

  {
    // Missing ending >.
    uint8_t data[] = "1A2b";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 4, false), 0);
    EXPECT_EQ("\x1a\x2b", parser.ReadHexString());
    EXPECT_EQ(4, parser.SavePos());
  }

  {
    // Missing ending >.
    uint8_t data[] = "12abz";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 5, false), 0);
    EXPECT_EQ("\x12\xab", parser.ReadHexString());
    EXPECT_EQ(5, parser.SavePos());
  }

  {
    // Uneven number of bytes.
    uint8_t data[] = "1A2>asdf";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 8, false), 0);
    EXPECT_EQ("\x1a\x20", parser.ReadHexString());
    EXPECT_EQ(4, parser.SavePos());
  }

  {
    // Uneven number of bytes.
    uint8_t data[] = "1A2zasdf";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 8, false), 0);
    EXPECT_EQ("\x1a\x2a\xdf", parser.ReadHexString());
    EXPECT_EQ(8, parser.SavePos());
  }

  {
    // Just ending character.
    uint8_t data[] = ">";
    CPDF_SyntaxParser parser;
    parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 1, false), 0);
    EXPECT_EQ("", parser.ReadHexString());
    EXPECT_EQ(1, parser.SavePos());
  }
}

TEST(cpdf_syntax_parser, GetInvalidReference) {
  CPDF_SyntaxParser parser;
  // Data with a reference with number CPDF_Object::kInvalidObjNum
  uint8_t data[] = "4294967295 0 R";
  parser.InitParser(pdfium::MakeRetain<CFX_MemoryStream>(data, 14, false), 0);
  std::unique_ptr<CPDF_Object> ref =
      parser.GetObject(nullptr, CPDF_Object::kInvalidObjNum, 0, false);
  EXPECT_FALSE(ref);
}