// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "pdf/pdfium/pdfium_permissions.h"

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

namespace chrome_pdf {

namespace {

constexpr uint32_t GeneratePermissions2(uint32_t permissions) {
  constexpr uint32_t kBasePermissions = 0xffffffc0;
  return kBasePermissions | permissions;
}

constexpr uint32_t GeneratePermissions3(uint32_t permissions) {
  constexpr uint32_t kBasePermissions = 0xfffff0c0;
  return kBasePermissions | permissions;
}

// Sanity check the permission constants are correct.
static_assert(kPDFPermissionBit03PrintMask == 0x4, "Wrong permission");
static_assert(kPDFPermissionBit05CopyMask == 0x10, "Wrong permission");
static_assert(kPDFPermissionBit10CopyAccessibleMask == 0x200,
              "Wrong permission");
static_assert(kPDFPermissionBit12PrintHighQualityMask == 0x800,
              "Wrong permission");

// Sanity check the permission generation functions above do the right thing.
static_assert(GeneratePermissions2(0) == 0xffffffc0, "Wrong permission");
static_assert(GeneratePermissions2(kPDFPermissionBit03PrintMask |
                                   kPDFPermissionBit05CopyMask) == 0xffffffd4,
              "Wrong permission");
static_assert(GeneratePermissions3(0) == 0xfffff0c0, "Wrong permission");
static_assert(GeneratePermissions3(kPDFPermissionBit03PrintMask |
                                   kPDFPermissionBit05CopyMask |
                                   kPDFPermissionBit10CopyAccessibleMask |
                                   kPDFPermissionBit12PrintHighQualityMask) ==
                  0xfffffad4,
              "Wrong permission");

TEST(PDFiumPermissionTest, InvalidSecurityHandler) {
  constexpr int kPDFiumUnknownRevision = -1;
  constexpr uint32_t kNoPermissions = 0;
  auto unknown_perms = PDFiumPermissions::CreateForTesting(
      kPDFiumUnknownRevision, kNoPermissions);
  EXPECT_TRUE(unknown_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_TRUE(unknown_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_TRUE(
      unknown_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_TRUE(
      unknown_perms.HasPermission(DocumentPermission::kPrintHighQuality));

  constexpr int kInvalidRevision = 1;
  auto obsolete_perms =
      PDFiumPermissions::CreateForTesting(kInvalidRevision, kNoPermissions);
  EXPECT_TRUE(obsolete_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_TRUE(
      obsolete_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_TRUE(
      obsolete_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_TRUE(
      obsolete_perms.HasPermission(DocumentPermission::kPrintHighQuality));
}

TEST(PDFiumPermissionTest, Revision2SecurityHandlerNone) {
  uint32_t permissions = GeneratePermissions2(0);
  auto no_perms = PDFiumPermissions::CreateForTesting(2, permissions);
  EXPECT_FALSE(no_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_FALSE(no_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_FALSE(no_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_FALSE(no_perms.HasPermission(DocumentPermission::kPrintHighQuality));
}

TEST(PDFiumPermissionTest, Revision2SecurityHandlerAll) {
  uint32_t permissions = GeneratePermissions2(kPDFPermissionBit03PrintMask |
                                              kPDFPermissionBit05CopyMask);
  auto all_known_perms = PDFiumPermissions::CreateForTesting(2, permissions);
  EXPECT_TRUE(all_known_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_TRUE(
      all_known_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_TRUE(
      all_known_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_TRUE(
      all_known_perms.HasPermission(DocumentPermission::kPrintHighQuality));
}

TEST(PDFiumPermissionTest, Revision2SecurityHandlerCopyPrint) {
  uint32_t permissions = GeneratePermissions2(kPDFPermissionBit05CopyMask);
  auto no_print_perms = PDFiumPermissions::CreateForTesting(2, permissions);
  EXPECT_TRUE(no_print_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_TRUE(
      no_print_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_FALSE(
      no_print_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_FALSE(
      no_print_perms.HasPermission(DocumentPermission::kPrintHighQuality));

  permissions = GeneratePermissions2(kPDFPermissionBit03PrintMask);
  auto no_copy_perms = PDFiumPermissions::CreateForTesting(2, permissions);
  EXPECT_FALSE(no_copy_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_FALSE(
      no_copy_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_TRUE(
      no_copy_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_TRUE(
      no_copy_perms.HasPermission(DocumentPermission::kPrintHighQuality));
}

TEST(PDFiumPermissionTest, Revision3SecurityHandlerNone) {
  uint32_t permissions = GeneratePermissions3(0);
  auto no_perms = PDFiumPermissions::CreateForTesting(3, permissions);
  EXPECT_FALSE(no_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_FALSE(no_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_FALSE(no_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_FALSE(no_perms.HasPermission(DocumentPermission::kPrintHighQuality));
}

TEST(PDFiumPermissionTest, Revision3SecurityHandlerAll) {
  uint32_t permissions = GeneratePermissions3(
      kPDFPermissionBit03PrintMask | kPDFPermissionBit05CopyMask |
      kPDFPermissionBit10CopyAccessibleMask |
      kPDFPermissionBit12PrintHighQualityMask);
  auto all_known_perms = PDFiumPermissions::CreateForTesting(3, permissions);
  EXPECT_TRUE(all_known_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_TRUE(
      all_known_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_TRUE(
      all_known_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_TRUE(
      all_known_perms.HasPermission(DocumentPermission::kPrintHighQuality));
}

TEST(PDFiumPermissionTest, Revision3SecurityHandlerCopyPrint) {
  uint32_t permissions = GeneratePermissions3(
      kPDFPermissionBit05CopyMask | kPDFPermissionBit10CopyAccessibleMask);
  auto copy_no_print_perms =
      PDFiumPermissions::CreateForTesting(3, permissions);
  EXPECT_TRUE(copy_no_print_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_TRUE(
      copy_no_print_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_FALSE(
      copy_no_print_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_FALSE(
      copy_no_print_perms.HasPermission(DocumentPermission::kPrintHighQuality));

  permissions = GeneratePermissions3(kPDFPermissionBit03PrintMask |
                                     kPDFPermissionBit05CopyMask |
                                     kPDFPermissionBit10CopyAccessibleMask);
  auto copy_low_print_perms =
      PDFiumPermissions::CreateForTesting(3, permissions);
  EXPECT_TRUE(copy_low_print_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_TRUE(
      copy_low_print_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_TRUE(
      copy_low_print_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_FALSE(copy_low_print_perms.HasPermission(
      DocumentPermission::kPrintHighQuality));

  permissions = GeneratePermissions3(kPDFPermissionBit03PrintMask |
                                     kPDFPermissionBit12PrintHighQualityMask);
  auto print_no_copy_perms =
      PDFiumPermissions::CreateForTesting(3, permissions);
  EXPECT_FALSE(print_no_copy_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_FALSE(
      print_no_copy_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_TRUE(
      print_no_copy_perms.HasPermission(DocumentPermission::kPrintLowQuality));
  EXPECT_TRUE(
      print_no_copy_perms.HasPermission(DocumentPermission::kPrintHighQuality));

  permissions = GeneratePermissions3(kPDFPermissionBit03PrintMask |
                                     kPDFPermissionBit10CopyAccessibleMask |
                                     kPDFPermissionBit12PrintHighQualityMask);
  auto print_a11y_copy_perms =
      PDFiumPermissions::CreateForTesting(3, permissions);
  EXPECT_FALSE(print_a11y_copy_perms.HasPermission(DocumentPermission::kCopy));
  EXPECT_TRUE(
      print_a11y_copy_perms.HasPermission(DocumentPermission::kCopyAccessible));
  EXPECT_TRUE(print_a11y_copy_perms.HasPermission(
      DocumentPermission::kPrintLowQuality));
  EXPECT_TRUE(print_a11y_copy_perms.HasPermission(
      DocumentPermission::kPrintHighQuality));
}

}  // namespace

}  // namespace chrome_pdf