/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

#include <climits>
#include <memory>
#include "nssb64.h"

#include "gtest/gtest.h"
#include "scoped_ptrs_util.h"

namespace nss_test {

class B64EncodeDecodeTest : public ::testing::Test {
 public:
  void TestDecodeStr(const std::string &str) {
    ScopedSECItem tmp(
        NSSBase64_DecodeBuffer(nullptr, nullptr, str.c_str(), str.size()));
    ASSERT_TRUE(tmp);
    char *out = NSSBase64_EncodeItem(nullptr, nullptr, 0, tmp.get());
    ASSERT_TRUE(out);
    ASSERT_EQ(std::string(out), str);
    PORT_Free(out);
  }
  bool TestEncodeItem(SECItem *item) {
    bool rv = true;
    char *out = NSSBase64_EncodeItem(nullptr, nullptr, 0, item);
    rv = !!out;
    if (out) {
      ScopedSECItem tmp(
          NSSBase64_DecodeBuffer(nullptr, nullptr, out, strlen(out)));
      EXPECT_TRUE(tmp);
      EXPECT_EQ(SECEqual, SECITEM_CompareItem(item, tmp.get()));
      PORT_Free(out);
    }
    return rv;
  }
  bool TestFakeDecode(size_t str_len) {
    std::string str(str_len, 'A');
    ScopedSECItem tmp(
        NSSBase64_DecodeBuffer(nullptr, nullptr, str.c_str(), str.size()));
    return !!tmp;
  }
  bool TestFakeEncode(size_t len) {
    std::vector<uint8_t> data(len, 0x30);
    SECItem tmp = {siBuffer, data.data(),
                   static_cast<unsigned int>(data.size())};
    return TestEncodeItem(&tmp);
  }

 protected:
};

TEST_F(B64EncodeDecodeTest, DecEncTest) { TestDecodeStr("VGhpcyBpcyBOU1Mh"); }

TEST_F(B64EncodeDecodeTest, EncDecTest) {
  uint8_t data[] = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09};
  SECItem tmp = {siBuffer, data, sizeof(data)};
  TestEncodeItem(&tmp);
}

TEST_F(B64EncodeDecodeTest, IncompleteData) {
  NSSBase64Decoder *context = NSSBase64Decoder_Create(
      [](void *, const unsigned char *, PRInt32) { return 0; }, nullptr);
  EXPECT_TRUE(!!context);
  char data = 'A';
  EXPECT_EQ(SECSuccess, NSSBase64Decoder_Update(context, &data, 1));
  EXPECT_EQ(SECFailure, NSSBase64Decoder_Destroy(context, false));
}

TEST_F(B64EncodeDecodeTest, FakeDecTest) { EXPECT_TRUE(TestFakeDecode(100)); }

TEST_F(B64EncodeDecodeTest, FakeEncDecTest) {
  EXPECT_TRUE(TestFakeEncode(100));
}

// These takes a while ...
TEST_F(B64EncodeDecodeTest, DISABLED_LongFakeDecTest1) {
  EXPECT_TRUE(TestFakeDecode(0x66666666));
}
TEST_F(B64EncodeDecodeTest, DISABLED_LongFakeEncDecTest1) {
  TestFakeEncode(0x3fffffff);
}
TEST_F(B64EncodeDecodeTest, DISABLED_LongFakeEncDecTest2) {
  EXPECT_FALSE(TestFakeEncode(0x40000000));
}

// Regression test for integer overflow in PL_UpdateBase64Decoder: when
// size > PR_UINT32_MAX - data->token_size, size + token_size (see
// nssb64d.c:464) wraps to a small value, PL_Base64MaxDecodedLength returns too
// small a value, and pl_base64_decode_4to3 writes out of bounds.
TEST_F(B64EncodeDecodeTest, OverflowingUpdateSizeIsRejected) {
  NSSBase64Decoder *ctx = NSSBase64Decoder_Create(
      [](void *, const unsigned char *, PRInt32) { return 0; }, nullptr);
  ASSERT_TRUE(ctx);

  // First update: 3 chars leave token_size = 3.
  const char first[] = "AAA";
  EXPECT_EQ(SECSuccess, NSSBase64Decoder_Update(ctx, first, 3));

  // Second update: size + token_size (3) wraps to 2.
  const char second[] = "A";
  EXPECT_EQ(SECFailure, NSSBase64Decoder_Update(ctx, second, PR_UINT32_MAX));

  NSSBase64Decoder_Destroy(ctx, false);
}

}  // namespace nss_test
