/*
 * Copyright 2020 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "hci/uuid.h"

#include <gmock/gmock.h>
#include <gtest/gtest.h>

namespace testing {

using bluetooth::hci::Uuid;

static const Uuid ONES =
        Uuid::From128BitBE(Uuid::UUID128Bit{{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
                                             0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}});

static const Uuid SEQUENTIAL =
        Uuid::From128BitBE(Uuid::UUID128Bit{{0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0xab,
                                             0xcd, 0xef, 0x01, 0x23, 0x45, 0x67, 0x89}});

static const Uuid kBase =
        Uuid::From128BitBE(Uuid::UUID128Bit{{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x00, 0x80,
                                             0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb}});

static const Uuid kBaseLe =
        Uuid::From128BitLE(Uuid::UUID128Bit{{0xfb, 0x34, 0x9b, 0x5f, 0x80, 0x00, 0x00, 0x80, 0x00,
                                             0x10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}});

TEST(UuidTest, IsEmpty) {
  ASSERT_TRUE(Uuid::kEmpty.IsEmpty());
  ASSERT_FALSE(kBase.IsEmpty());
}

TEST(UuidTest, GetShortestRepresentationSize) {
  ASSERT_EQ(Uuid::kNumBytes16, kBase.GetShortestRepresentationSize());
  ASSERT_EQ(Uuid::kNumBytes32, Uuid::From32Bit(0x01234567).GetShortestRepresentationSize());
  ASSERT_EQ(Uuid::kNumBytes128, Uuid::kEmpty.GetShortestRepresentationSize());
}

TEST(UuidTest, As16Bit) {
  // Even though this is is not 16bit UUID, we should be able to get proper bits
  ASSERT_EQ((uint16_t)0x1111, ONES.As16Bit());
  ASSERT_EQ((uint16_t)0x4567, SEQUENTIAL.As16Bit());
  ASSERT_EQ((uint16_t)0x0000, kBase.As16Bit());
}

TEST(UuidTest, As32Bit) {
  // Even though this is is not 32bit UUID, we should be able to get proper bits
  ASSERT_EQ((uint32_t)0x11111111, ONES.As32Bit());
  ASSERT_EQ((uint32_t)0x01234567, SEQUENTIAL.As32Bit());
  ASSERT_EQ((uint32_t)0x00000000, kBase.As32Bit());
  ASSERT_EQ((uint32_t)0x12345678, Uuid::From32Bit(0x12345678).As32Bit());
}

TEST(UuidTest, Is16Bit) {
  ASSERT_FALSE(ONES.Is16Bit());
  ASSERT_FALSE(SEQUENTIAL.Is16Bit());
  ASSERT_TRUE(kBase.Is16Bit());
  auto uuid = Uuid::FromString("1ae8");
  ASSERT_TRUE(uuid);
  ASSERT_TRUE(uuid->Is16Bit());
}

TEST(UuidTest, From16Bit) {
  ASSERT_EQ(Uuid::From16Bit(0x0000), kBase);

  const uint8_t u2[] = {0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x10, 0x00,
                        0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb};
  Uuid uuid = Uuid::From16Bit(0x0001);
  ASSERT_EQ(0, memcmp(uuid.data(), u2, sizeof(u2)));

  const uint8_t u3[] = {0x00, 0x00, 0x55, 0x3e, 0x00, 0x00, 0x10, 0x00,
                        0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb};
  uuid = Uuid::From16Bit(0x553e);
  ASSERT_EQ(0, memcmp(uuid.data(), u3, sizeof(u3)));

  const uint8_t u4[] = {0x00, 0x00, 0xff, 0xff, 0x00, 0x00, 0x10, 0x00,
                        0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb};
  uuid = Uuid::From16Bit(0xffff);
  ASSERT_EQ(0, memcmp(uuid.data(), u4, sizeof(u4)));
}

TEST(UuidTest, From32Bit) {
  ASSERT_EQ(Uuid::From32Bit(0x00000000), kBase);

  const uint8_t u2[] = {0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x10, 0x00,
                        0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb};
  Uuid uuid = Uuid::From32Bit(0x00000001);
  ASSERT_EQ(0, memcmp(uuid.data(), u2, sizeof(u2)));

  const uint8_t u3[] = {0x33, 0x44, 0x55, 0x3e, 0x00, 0x00, 0x10, 0x00,
                        0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb};
  uuid = Uuid::From32Bit(0x3344553e);
  ASSERT_EQ(0, memcmp(uuid.data(), u3, sizeof(u3)));

  const uint8_t u4[] = {0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x10, 0x00,
                        0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb};
  uuid = Uuid::From32Bit(0xffffffff);
  ASSERT_EQ(0, memcmp(uuid.data(), u4, sizeof(u4)));
}

TEST(UuidTest, ToString) {
  const std::string UUID_BASE_STR = "00000000-0000-1000-8000-00805f9b34fb";
  const std::string UUID_EMP_STR = "00000000-0000-0000-0000-000000000000";
  const std::string UUID_ONES_STR = "11111111-1111-1111-1111-111111111111";
  const std::string UUID_SEQ_STR = "01234567-89ab-cdef-abcd-ef0123456789";

  ASSERT_EQ(UUID_BASE_STR, kBase.ToString());
  ASSERT_EQ(UUID_EMP_STR, Uuid::kEmpty.ToString());
  ASSERT_EQ(UUID_ONES_STR, ONES.ToString());
  ASSERT_EQ(UUID_SEQ_STR, SEQUENTIAL.ToString());

  Uuid uuid = Uuid::From32Bit(0x12345678);
  ASSERT_EQ("12345678-0000-1000-8000-00805f9b34fb", uuid.ToString());
}

TEST(UuidTest, test_string_to_uuid) {
  const uint8_t u1[] = {0xe3, 0x9c, 0x62, 0x85, 0x86, 0x7f, 0x4b, 0x1d,
                        0x9d, 0xb0, 0x35, 0xfb, 0xd9, 0xae, 0xbf, 0x22};
  auto uuid = Uuid::FromString("e39c6285-867f-4b1d-9db0-35fbd9aebf22");
  ASSERT_TRUE(uuid);
  ASSERT_EQ(0, memcmp(uuid->data(), u1, sizeof(u1)));

  const uint8_t u2[] = {0x00, 0x00, 0x1a, 0xe8, 0x00, 0x00, 0x10, 0x00,
                        0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb};
  uuid = Uuid::FromString("1Ae8");
  ASSERT_TRUE(uuid);
  ASSERT_EQ(0, memcmp(uuid->data(), u2, sizeof(u2)));

  const uint8_t u3[] = {0x12, 0x34, 0x11, 0x28, 0x00, 0x00, 0x10, 0x00,
                        0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb};
  uuid = Uuid::FromString("12341128");
  ASSERT_TRUE(uuid);
  ASSERT_EQ(0, memcmp(uuid->data(), u3, sizeof(u3)));
}

TEST(UuidTest, legacy) {
  const std::string UUID_BASE_STR = "e39c6285-867f-4b1d-9db0-35fbd9aebf22";
  auto uuid = Uuid::FromString("e39c6285-867f-4b1d-9db0-35fbd9aebf22");
  ASSERT_EQ(UUID_BASE_STR, uuid->ToLegacyConfigString());
  ASSERT_EQ(uuid, Uuid::FromLegacyConfigString(UUID_BASE_STR));
}

TEST(UuidTest, inequalities) {
  auto uuid1 = Uuid::kEmpty;
  auto uuid2 = Uuid::FromString("11111111-1111-1111-1111-111111111111");

  ASSERT_TRUE(uuid1 < uuid2);
  ASSERT_TRUE(uuid1 != uuid2);
}

TEST(UuidTest, endianness) {
  Uuid uuidBe = kBase;
  Uuid uuidLe = kBaseLe;
  ASSERT_EQ(kBase, kBaseLe);
}

TEST(UuidTest, test_string_to_uuid_invalid) {
  ASSERT_FALSE(Uuid::FromString("This is not a UUID"));
  ASSERT_FALSE(Uuid::FromString("11212"));
  ASSERT_FALSE(Uuid::FromString("1121 "));
  ASSERT_FALSE(Uuid::FromString("AGFE"));
  ASSERT_FALSE(Uuid::FromString("ABFG"));
  ASSERT_FALSE(Uuid::FromString("e39c6285867f14b1d9db035fbd9aebf22"));
  ASSERT_FALSE(Uuid::FromString("12234567-89ab-cdef-abcd-ef01234567ZZ"));
  ASSERT_FALSE(Uuid::FromString(std::string()));
  ASSERT_FALSE(Uuid::FromString("e39c6285 867f 4b1d 9db0 35fbd9aebf22"));
  ASSERT_FALSE(Uuid::FromString("ab34xx78"));
}

}  // namespace testing