// Copyright 2019 The Fuchsia 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 "frame_headers.h"

#include <gtest/gtest.h>

#include "src/connectivity/bluetooth/core/bt-host/common/byte_buffer.h"

namespace bt::l2cap::internal {
namespace {

TEST(FrameHeadersEnhancedControlFieldTest, IdentifiesInformationFrame) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_TRUE(
      StaticByteBuffer(0b0000'0000, 0).To<EnhancedControlField>().designates_information_frame());
}

TEST(FrameHeadersEnhancedControlFieldTest, IdentifiesNonInformationFrame) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_FALSE(
      StaticByteBuffer(0b0000'0001, 0).To<EnhancedControlField>().designates_information_frame());
}

TEST(FrameHeadersEnhancedControlFieldTest, IdentifiesSupervisoryFrame) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_TRUE(
      StaticByteBuffer(0b00000001, 0).To<EnhancedControlField>().designates_supervisory_frame());
}

TEST(FrameHeadersEnhancedControlFieldTest, IdentifiesNonSupervisoryFrame) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_FALSE(
      StaticByteBuffer(0b00000000, 1).To<EnhancedControlField>().designates_supervisory_frame());
}

TEST(FrameHeadersEnhancedControlFieldTest, IdentifiesStartOfSegmentedSdu) {
  // See Core Spec, v5, Vol 3, Part A, Tables 3.2 and 3.4.
  EXPECT_TRUE(StaticByteBuffer(0, 0b01000000)
                  .To<EnhancedControlField>()
                  .designates_start_of_segmented_sdu());
}

TEST(FrameHeadersEnhancedControlFieldTest, IdentifiesNonStartOfSegmentedSdu) {
  // See Core Spec, v5, Vol 3, Part A, Tables 3.2 and 3.4.
  EXPECT_FALSE(StaticByteBuffer(0, 0b00000000)
                   .To<EnhancedControlField>()
                   .designates_start_of_segmented_sdu());
  EXPECT_FALSE(StaticByteBuffer(0, 0b10000000)
                   .To<EnhancedControlField>()
                   .designates_start_of_segmented_sdu());
  EXPECT_FALSE(StaticByteBuffer(0, 0b11000000)
                   .To<EnhancedControlField>()
                   .designates_start_of_segmented_sdu());
  EXPECT_FALSE(StaticByteBuffer(1, 0b01000000)
                   .To<EnhancedControlField>()
                   .designates_start_of_segmented_sdu());
}

TEST(FrameHeadersEnhancedControlFieldTest, IdentifiesPartOfSegmentedSdu) {
  // See Core Spec, v5, Vol 3, Part A, Tables 3.2 and 3.4.
  EXPECT_TRUE(StaticByteBuffer(0, 0b01000000)
                  .To<EnhancedControlField>()
                  .designates_part_of_segmented_sdu());
  EXPECT_TRUE(StaticByteBuffer(0, 0b10000000)
                  .To<EnhancedControlField>()
                  .designates_part_of_segmented_sdu());
  EXPECT_TRUE(StaticByteBuffer(0, 0b11000000)
                  .To<EnhancedControlField>()
                  .designates_part_of_segmented_sdu());
}

TEST(FrameHeadersEnhancedControlFieldTest, IdentifiesNotPartOfSegmentedSdu) {
  // See Core Spec, v5, Vol 3, Part A, Tables 3.2 and 3.4.
  EXPECT_FALSE(StaticByteBuffer(0, 0b00000000)
                   .To<EnhancedControlField>()
                   .designates_part_of_segmented_sdu());
  EXPECT_FALSE(StaticByteBuffer(1, 0b01000000)
                   .To<EnhancedControlField>()
                   .designates_part_of_segmented_sdu());
  EXPECT_FALSE(StaticByteBuffer(1, 0b10000000)
                   .To<EnhancedControlField>()
                   .designates_part_of_segmented_sdu());
  EXPECT_FALSE(StaticByteBuffer(1, 0b11000000)
                   .To<EnhancedControlField>()
                   .designates_part_of_segmented_sdu());
}

TEST(FrameHeadersEnhancedControlFieldTest, ReadsRequestSequenceNumber) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2, and Core Spec v5, Vol 3, Part
  // A, Sec 8.3.
  for (uint8_t seq_num = 0; seq_num < 64; ++seq_num) {
    EXPECT_EQ(seq_num, StaticByteBuffer(0, seq_num).To<EnhancedControlField>().receive_seq_num());
  }
}

TEST(FrameHeadersEnhancedControlFieldTest, IsConstructedProperly) {
  EnhancedControlField ecf;
  EXPECT_EQ(StaticByteBuffer(0, 0), BufferView(&ecf, sizeof(ecf)));
}

TEST(FrameHeadersEnhancedControlFieldTest, SetSupervisoryFrameSetsBitCorrectly) {
  EnhancedControlField ecf;
  ecf.set_supervisory_frame();
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_EQ(StaticByteBuffer(0b1, 0), BufferView(&ecf, sizeof(ecf)));
}

TEST(FrameHeadersEnhancedControlFieldTest, SetRequestSeqNumSetsBitsCorrectly) {
  for (uint8_t seq_num = 0; seq_num < 64; ++seq_num) {
    EnhancedControlField ecf;
    ecf.set_receive_seq_num(seq_num);
    // See Core Spec, v5, Vol 3, Part A, Table 3.2.
    EXPECT_EQ(StaticByteBuffer(0, seq_num), BufferView(&ecf, sizeof(ecf)));
  }
}

TEST(FrameHeadersEnhancedControlFieldTest, SetSegmentationStatusWorksCorrectlyOnFreshFrame) {
  // See Core Spec, v5, Vol 3, Part A, Tables 3.2 and 3.4.

  {
    EnhancedControlField ecf;
    ecf.set_segmentation_status(SegmentationStatus::Unsegmented);
    EXPECT_EQ(StaticByteBuffer(0, 0), BufferView(&ecf, sizeof(ecf)));
  }

  {
    EnhancedControlField ecf;
    ecf.set_segmentation_status(SegmentationStatus::FirstSegment);
    EXPECT_EQ(StaticByteBuffer(0, 0b0100'0000), BufferView(&ecf, sizeof(ecf)));
  }

  {
    EnhancedControlField ecf;
    ecf.set_segmentation_status(SegmentationStatus::LastSegment);
    EXPECT_EQ(StaticByteBuffer(0, 0b1000'0000), BufferView(&ecf, sizeof(ecf)));
  }

  {
    EnhancedControlField ecf;
    ecf.set_segmentation_status(SegmentationStatus::MiddleSegment);
    EXPECT_EQ(StaticByteBuffer(0, 0b1100'0000), BufferView(&ecf, sizeof(ecf)));
  }
}

TEST(FrameHeadersEnhancedControlFieldTest, SetSegmentationStatusWorksCorrectlyOnRecycledFrame) {
  // See Core Spec, v5, Vol 3, Part A, Tables 3.2 and 3.4.

  {
    EnhancedControlField ecf;
    ecf.set_segmentation_status(SegmentationStatus::MiddleSegment);
    ecf.set_segmentation_status(SegmentationStatus::Unsegmented);
    EXPECT_EQ(StaticByteBuffer(0, 0), BufferView(&ecf, sizeof(ecf)));
  }

  {
    EnhancedControlField ecf;
    ecf.set_segmentation_status(SegmentationStatus::MiddleSegment);
    ecf.set_segmentation_status(SegmentationStatus::FirstSegment);
    EXPECT_EQ(StaticByteBuffer(0, 0b0100'0000), BufferView(&ecf, sizeof(ecf)));
  }

  {
    EnhancedControlField ecf;
    ecf.set_segmentation_status(SegmentationStatus::MiddleSegment);
    ecf.set_segmentation_status(SegmentationStatus::LastSegment);
    EXPECT_EQ(StaticByteBuffer(0, 0b1000'0000), BufferView(&ecf, sizeof(ecf)));
  }

  {
    EnhancedControlField ecf;
    ecf.set_segmentation_status(SegmentationStatus::MiddleSegment);
    ecf.set_segmentation_status(SegmentationStatus::MiddleSegment);
    EXPECT_EQ(StaticByteBuffer(0, 0b1100'0000), BufferView(&ecf, sizeof(ecf)));
  }
}

TEST(FrameHeadersEnhancedControlFieldTest, SetSegmentationStatusPreservesRequestSeqNum) {
  EnhancedControlField ecf;
  ecf.set_receive_seq_num(EnhancedControlField::kMaxSeqNum);
  ecf.set_segmentation_status(SegmentationStatus::Unsegmented);
  EXPECT_EQ(EnhancedControlField::kMaxSeqNum, ecf.receive_seq_num());
}

TEST(FrameHeadersSimpleInformationFrameHeaderTest, ReadsTxSequenceNumber) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2, and Core Spec v5, Vol 3, Part
  // A, Sec 8.3.
  for (uint8_t seq_num = 0; seq_num < 64; ++seq_num) {
    EXPECT_EQ(seq_num,
              StaticByteBuffer(seq_num << 1, 0).To<SimpleInformationFrameHeader>().tx_seq());
  }
}

TEST(FrameHeadersSimpleInformationFrameHeaderTest, IsConstructedProperly) {
  constexpr uint8_t kTxSeq = 63;
  SimpleInformationFrameHeader frame(kTxSeq);
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_EQ(StaticByteBuffer(0b111'1110, 0), BufferView(&frame, sizeof(frame)));
}

TEST(FrameHeadersSimpleInformationFrameHeaderTest, SetSegmentationStatusPreservesTxSeq) {
  constexpr uint8_t kTxSeq = 63;
  SimpleInformationFrameHeader frame(kTxSeq);
  frame.set_segmentation_status(SegmentationStatus::Unsegmented);
  EXPECT_EQ(kTxSeq, frame.tx_seq());
}

TEST(FrameHeadersSimpleStartOfSduFrameHeaderTest, IsConstructedProperly) {
  constexpr uint8_t kTxSeq = 63;
  SimpleStartOfSduFrameHeader frame(kTxSeq);
  // See Core Spec, v5, Vol 3, Part A, Table 3.2, and Figure 3.3.
  EXPECT_EQ(StaticByteBuffer(0b111'1110, 0b0100'0000, 0, 0), BufferView(&frame, sizeof(frame)));
}

TEST(FrameHeadersSimpleSupervisoryFrameTest, IsConstructedProperly) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.

  {
    SimpleSupervisoryFrame frame(SupervisoryFunction::ReceiverReady);
    EXPECT_EQ(StaticByteBuffer(0b0001, 0), BufferView(&frame, sizeof(frame)));
  }

  {
    SimpleSupervisoryFrame frame(SupervisoryFunction::Reject);
    EXPECT_EQ(StaticByteBuffer(0b0101, 0), BufferView(&frame, sizeof(frame)));
  }

  {
    SimpleSupervisoryFrame frame(SupervisoryFunction::ReceiverNotReady);
    EXPECT_EQ(StaticByteBuffer(0b1001, 0), BufferView(&frame, sizeof(frame)));
  }

  {
    SimpleSupervisoryFrame frame(SupervisoryFunction::SelectiveReject);
    EXPECT_EQ(StaticByteBuffer(0b1101, 0), BufferView(&frame, sizeof(frame)));
  }
}

TEST(FrameHeadersSimpleSupervisoryFrameTest, IdentifiesPollRequest) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_FALSE(StaticByteBuffer(0b0'0001, 0).To<SimpleSupervisoryFrame>().is_poll_request());
  EXPECT_TRUE(StaticByteBuffer(0b1'0001, 0).To<SimpleSupervisoryFrame>().is_poll_request());
}

TEST(FrameHeadersSimpleSupervisoryFrameTest, IdentifiesPollResponse) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_FALSE(StaticByteBuffer(0b0000'0001, 0).To<SimpleSupervisoryFrame>().is_poll_response());
  EXPECT_TRUE(StaticByteBuffer(0b1000'0001, 0).To<SimpleSupervisoryFrame>().is_poll_response());
}

TEST(FrameHeadersSimpleSupervisoryFrameTest, FunctionReadsSupervisoryFunction) {
  // See Core Spec, v5, Vol 3, Part A, Table 3.2 and Table 3.5.
  EXPECT_EQ(SupervisoryFunction::ReceiverReady,
            StaticByteBuffer(0b0001, 0).To<SimpleSupervisoryFrame>().function());
  EXPECT_EQ(SupervisoryFunction::Reject,
            StaticByteBuffer(0b0101, 0).To<SimpleSupervisoryFrame>().function());
  EXPECT_EQ(SupervisoryFunction::ReceiverNotReady,
            StaticByteBuffer(0b1001, 0).To<SimpleSupervisoryFrame>().function());
  EXPECT_EQ(SupervisoryFunction::SelectiveReject,
            StaticByteBuffer(0b1101, 0).To<SimpleSupervisoryFrame>().function());
}

TEST(FrameHeadersSimpleSupervisoryFrameTest, SetIsPollRequestSetsCorrectBit) {
  SimpleSupervisoryFrame sframe(SupervisoryFunction::ReceiverReady);
  sframe.set_is_poll_request();
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_EQ(StaticByteBuffer(0b1'0001, 0), BufferView(&sframe, sizeof(sframe)));
}

TEST(FrameHeadersSimpleSupervisoryFrameTest, SetIsPollResponseSetsCorrectBit) {
  SimpleSupervisoryFrame sframe(SupervisoryFunction::ReceiverReady);
  sframe.set_is_poll_response();
  // See Core Spec, v5, Vol 3, Part A, Table 3.2.
  EXPECT_EQ(StaticByteBuffer(0b1000'0001, 0), BufferView(&sframe, sizeof(sframe)));
}

TEST(FrameHeadersSimpleReceiverReadyFrameTest, IsConstructedProperly) {
  SimpleReceiverReadyFrame frame;
  // See Core Spec, v5, Vol 3, Part A, Table 3.2 and Table 3.5.
  EXPECT_EQ(StaticByteBuffer(0b0001, 0), BufferView(&frame, sizeof(frame)));
}

}  // namespace
}  // namespace bt::l2cap::internal