// Copyright 2020 Google LLC
//
// 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
//
//     https://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 "connections/implementation/mediums/ble_v2/ble_advertisement.h"

#include <cstddef>
#include <cstdint>
#include <optional>
#include <string>
#include <utility>

#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/strings/str_cat.h"
#include "connections/implementation/mediums/ble_v2/ble_advertisement_header.h"
#include "internal/platform/byte_array.h"
#include "internal/platform/logging.h"
#include "internal/platform/stream_reader.h"
#include "internal/platform/stream_writer.h"

namespace nearby {
namespace connections {
namespace mediums {

namespace {

constexpr uint8_t kPsmBitmask = 0x01;
constexpr uint8_t kRxInstantConnectionAdvBitmask = 0x02;

bool HasField(uint8_t field_mask, uint8_t psm_bit) {
  return (field_mask & psm_bit) > 0;
}

}  // namespace

BleAdvertisement::BleAdvertisement(Version version,
                                   SocketVersion socket_version,
                                   const ByteArray &service_id_hash,
                                   const ByteArray &data,
                                   const ByteArray &device_token, int psm) {
  DoInitialize(/*fast_advertisement=*/service_id_hash.Empty(), version,
               socket_version, service_id_hash, data, device_token, psm);
}

void BleAdvertisement::DoInitialize(bool fast_advertisement, Version version,
                                    SocketVersion socket_version,
                                    const ByteArray &service_id_hash,
                                    const ByteArray &data,
                                    const ByteArray &device_token, int psm) {
  // Check that the given input is valid.
  fast_advertisement_ = fast_advertisement;
  if (!fast_advertisement_) {
    if (service_id_hash.size() != kServiceIdHashLength) return;
  }
  if (!IsSupportedVersion(version) ||
      !IsSupportedSocketVersion(socket_version) ||
      (!device_token.Empty() && device_token.size() != kDeviceTokenLength)) {
    return;
  }

  int advertisement_Length = ComputeAdvertisementLength(
      data.size(), device_token.size(), fast_advertisement_);
  int max_advertisement_length = fast_advertisement
                                     ? kMaxFastAdvertisementLength
                                     : kMaxAdvertisementLength;
  if (advertisement_Length > max_advertisement_length) {
    return;
  }

  version_ = version;
  socket_version_ = socket_version;
  if (!fast_advertisement_) service_id_hash_ = service_id_hash;
  data_ = data;
  device_token_ = device_token;
  psm_ = psm;
}

absl::StatusOr<BleAdvertisement> BleAdvertisement::CreateBleAdvertisement(
    const ByteArray &ble_advertisement_bytes) {
  if (ble_advertisement_bytes.Empty()) {
    return absl::InvalidArgumentError(
        "Cannot deserialize BleAdvertisement: null bytes passed in.");
  }

  if (ble_advertisement_bytes.size() < kVersionLength) {
    return absl::InvalidArgumentError(absl::StrCat(
        "Cannot deserialize BleAdvertisement: expecting min ", kVersionLength,
        " bytes, got ", ble_advertisement_bytes.size()));
  }

  ByteArray advertisement_bytes(ble_advertisement_bytes);
  StreamReader stream_reader(&advertisement_bytes);
  // The first 1 byte is supposed to be the version, socket version and the fast
  // advertisement flag.
  auto version_byte = stream_reader.ReadUint8();
  if (!version_byte.has_value()) {
    return absl::InvalidArgumentError(
        "Cannot deserialize BleAdvertisement: version.");
  }

  Version version =
      static_cast<Version>((*version_byte & kVersionBitmask) >> 5);
  if (!IsSupportedVersion(version)) {
    return absl::InvalidArgumentError(absl::StrCat(
        "Cannot deserialize BleAdvertisement: unsupported Version ", version));
  }

  SocketVersion socket_version =
      static_cast<SocketVersion>((*version_byte & kSocketVersionBitmask) >> 2);
  if (!IsSupportedSocketVersion(socket_version)) {
    return absl::InvalidArgumentError(absl::StrCat(
        "Cannot deserialize BleAdvertisement: unsupported SocketVersion ",
        socket_version));
  }

  bool fast_advertisement =
      static_cast<bool>((*version_byte & kFastAdvertisementFlagBitmask) >> 1);

  bool is_second_profile =
      static_cast<bool>(*version_byte & kSecondProfileBitmask);

  // The next 3 bytes are supposed to be the service_id_hash if not fast
  // advertisement.
  ByteArray service_id_hash;
  if (!fast_advertisement) {
    auto service_id_hash_bytes = stream_reader.ReadBytes(kServiceIdHashLength);
    if (!service_id_hash_bytes.has_value()) {
      return absl::InvalidArgumentError(
          "Cannot deserialize BleAdvertisement: service_id_hash.");
    }
    service_id_hash = *service_id_hash_bytes;
  }

  // Data length.
  uint32_t expected_data_size;
  if (fast_advertisement) {
    auto fast_data_size_bytes = stream_reader.ReadBytes(kFastDataSizeLength);
    if (!fast_data_size_bytes.has_value()) {
      return absl::InvalidArgumentError(
          "Cannot deserialize BleAdvertisement: fast_data_size.");
    }
    expected_data_size = static_cast<uint32_t>(fast_data_size_bytes->data()[0]);
  } else {
    auto data_size_bytes = stream_reader.ReadUint32();
    if (!data_size_bytes.has_value()) {
      return absl::InvalidArgumentError(
          "Cannot deserialize BleAdvertisement: data_size.");
    }
    expected_data_size = *data_size_bytes;
  }

  // Data.
  // Check that the stated data size is the same as what we received.
  ByteArray data;
  if (expected_data_size > 0) {
    auto data_bytes = stream_reader.ReadBytes(expected_data_size);
    if (!data_bytes.has_value()) {
      return absl::InvalidArgumentError(
          "Cannot deserialize BleAdvertisement: data.");
    }
    data = *data_bytes;
  }

  BleAdvertisement ble_advertisement;
  ble_advertisement.version_ = version;
  ble_advertisement.socket_version_ = socket_version;
  ble_advertisement.fast_advertisement_ = fast_advertisement;
  ble_advertisement.is_second_profile_ = is_second_profile;
  ble_advertisement.service_id_hash_ = service_id_hash;
  ble_advertisement.data_ = data;

  // Device token. If the number of remaining bytes are valid for device token,
  // then read it.
  if (stream_reader.IsAvailable(kDeviceTokenLength)) {
    auto device_token_bytes = stream_reader.ReadBytes(kDeviceTokenLength);
    if (!device_token_bytes.has_value()) {
      return absl::InvalidArgumentError(
          "Cannot deserialize BleAdvertisement: device_token.");
    }
    ble_advertisement.device_token_ = *device_token_bytes;
  }

  // Extra fields, for backward compatible reason, put this field in the end of
  // this advertisement. That means it must support device token if there's any
  // extra field. E.g. If iOS or other platform wants to use extra fields, need
  // to put a random or empty device token in the advertisement.
  if (stream_reader.IsAvailable(kExtraFieldsMaskLength)) {
    uint8_t extra_fields_mask = stream_reader.ReadUint8().value_or(0);
    if (extra_fields_mask & kPsmBitmask) {
      auto psm_value = stream_reader.ReadUint16();
      if (!psm_value.has_value()) {
        return absl::InvalidArgumentError(
            "Cannot deserialize BleAdvertisement: psm_value.");
      }
      ble_advertisement.psm_ = *psm_value;
    }

    if (extra_fields_mask & kRxInstantConnectionAdvBitmask) {
      auto rx_instant_connection_adv_byte = stream_reader.ReadUint8();
      if (!rx_instant_connection_adv_byte.has_value()) {
        return absl::InvalidArgumentError(
            "Cannot deserialize BleAdvertisement: "
            "rx_instant_connection_adv_byte.");
      }
      auto rx_instant_connection_adv_data =
          stream_reader.ReadBytes(*rx_instant_connection_adv_byte);
      if (!rx_instant_connection_adv_data.has_value()) {
        return absl::InvalidArgumentError(
            "Cannot deserialize BleAdvertisement: "
            "rx_instant_connection_adv_data.");
      }
      ble_advertisement.rx_instant_connection_adv_ =
          *rx_instant_connection_adv_data;
    }
  }
  return ble_advertisement;
}

BleAdvertisement::operator ByteArray() const {
  if (!IsValid()) {
    return ByteArray{};
  }

  StreamWriter stream_writer;

  // The first 3 bits are the Version.
  char version_byte = (static_cast<char>(version_) << 5) & kVersionBitmask;
  // The next 3 bits are the Socket version. 2 bits left are reserved.
  version_byte |=
      (static_cast<char>(socket_version_) << 2) & kSocketVersionBitmask;
  // The next 1 bit is the fast advertisement flag. 1 bit left is reserved.
  version_byte |= (static_cast<char>(fast_advertisement_ ? 1 : 0) << 1) &
                  kFastAdvertisementFlagBitmask;
  version_byte |=
      (static_cast<char>(is_second_profile_ ? 1 : 0)) & kSecondProfileBitmask;

  stream_writer.WriteUint8(version_byte);

  // Serialize Data size bytes
  if (fast_advertisement_) {
    stream_writer.WriteUint8(data_.size());
    stream_writer.WriteBytes(std::string(data_));
    stream_writer.WriteBytes(std::string(device_token_));
  } else {
    stream_writer.WriteBytes(std::string(service_id_hash_));
    stream_writer.WriteUint32(data_.size());
    stream_writer.WriteBytes(std::string(data_));
    stream_writer.WriteBytes(std::string(device_token_));
  }

  // For Extra fields, there's no space for legacy fast advertisement, use
  // ByteArrayWithExtraField() to get the new advertisement bytes for extended
  // advertising.

  return ByteArray{stream_writer.GetData()};
}

ByteArray BleAdvertisement::ByteArrayWithExtraField() const {
  ByteArray advertisement_bytes = ByteArray(*this);
  ByteArray extra_fields_bytes =
      ByteArray(BleExtraFields(psm_, rx_instant_connection_adv_));

  std::string advertisement_with_extra_fields_bytes = absl::StrCat(
      std::string(advertisement_bytes), std::string(extra_fields_bytes));

  return ByteArray(std::move(advertisement_with_extra_fields_bytes));
}

bool BleAdvertisement::operator==(const BleAdvertisement &rhs) const {
  return this->GetVersion() == rhs.GetVersion() &&
         this->GetSocketVersion() == rhs.GetSocketVersion() &&
         this->IsFastAdvertisement() == rhs.IsFastAdvertisement() &&
         this->GetServiceIdHash() == rhs.GetServiceIdHash() &&
         this->GetData() == rhs.GetData() &&
         this->GetDeviceToken() == rhs.GetDeviceToken() &&
         this->GetPsm() == rhs.GetPsm() &&
         this->GetRxInstantConnectionAdv() == rhs.GetRxInstantConnectionAdv();
}

bool BleAdvertisement::IsSupportedVersion(Version version) {
  return version >= Version::kV1 && version <= Version::kV2;
}

bool BleAdvertisement::IsSupportedSocketVersion(SocketVersion socket_version) {
  return socket_version >= SocketVersion::kV1 &&
         socket_version <= SocketVersion::kV2;
}

BleAdvertisement::BleExtraFields::BleExtraFields(
    int psm, const ByteArray &rx_instant_connection_adv)
    : psm_(psm), rx_instant_connection_adv_(rx_instant_connection_adv) {}

BleAdvertisement::BleExtraFields::BleExtraFields(
    const ByteArray &ble_extra_fields_bytes) {
  if (ble_extra_fields_bytes.Empty()) {
    return;
  }

  ByteArray mutated_extra_fields_bytes = {ble_extra_fields_bytes};
  StreamReader stream_reader{&mutated_extra_fields_bytes};
  // The first 1 byte is field mask.
  auto mask_byte = stream_reader.ReadUint8().value_or(0);
  if (!mask_byte) {
    return;
  }

  // The next 2 bytes are supposed to be the psm value.
  if (HasField(mask_byte, kPsmBitmask) &&
      stream_reader.IsAvailable(BleAdvertisementHeader::kPsmValueByteLength)) {
    psm_ = stream_reader.ReadUint16().value_or(0);
  }

  if (HasField(mask_byte, kRxInstantConnectionAdvBitmask) &&
      stream_reader.IsAvailable(kRxInstantConnectionAdvSizeLength)) {
    auto rx_instant_connection_adv_byte = stream_reader.ReadUint8();
    auto rx_instant_connection_adv_data =
        stream_reader.ReadBytes(*rx_instant_connection_adv_byte);
    if (rx_instant_connection_adv_data.has_value()) {
      rx_instant_connection_adv_ = *rx_instant_connection_adv_data;
    }
  }
}

BleAdvertisement::BleExtraFields::operator ByteArray() const {
  if (psm_ == BleAdvertisementHeader::kDefaultPsmValue &&
      rx_instant_connection_adv_.Empty()) {
    return ByteArray{};
  }

  StreamWriter stream_writer;
  stream_writer.WriteUint8(
      (psm_ != BleAdvertisementHeader::kDefaultPsmValue ? kPsmBitmask : 0) |
      (rx_instant_connection_adv_.Empty() ? 0
                                          : kRxInstantConnectionAdvBitmask));
  if (psm_ != BleAdvertisementHeader::kDefaultPsmValue) {
    stream_writer.WriteUint16(psm_);
  }

  if (!rx_instant_connection_adv_.Empty()) {
    stream_writer.WriteUint8(rx_instant_connection_adv_.size());
    stream_writer.WriteBytes(std::string(rx_instant_connection_adv_));
  }

  return ByteArray{stream_writer.GetData()};
}

}  // namespace mediums
}  // namespace connections
}  // namespace nearby