8037fc6ffa
Bug: webrtc:342905193 No-Try: True Change-Id: Icc968be43b8830038ea9a1f5f604307220457807 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/361021 Auto-Submit: Florent Castelli <orphis@webrtc.org> Reviewed-by: Harald Alvestrand <hta@webrtc.org> Commit-Queue: Florent Castelli <orphis@webrtc.org> Cr-Commit-Position: refs/heads/main@{#42911}
134 lines
4.4 KiB
C++
134 lines
4.4 KiB
C++
/*
|
|
* Copyright (c) 2021 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
#include "net/dcsctp/packet/tlv_trait.h"
|
|
|
|
#include <vector>
|
|
|
|
#include "api/array_view.h"
|
|
#include "rtc_base/buffer.h"
|
|
#include "rtc_base/checks.h"
|
|
#include "rtc_base/gunit.h"
|
|
#include "test/gmock.h"
|
|
|
|
namespace dcsctp {
|
|
namespace {
|
|
using ::testing::ElementsAre;
|
|
using ::testing::SizeIs;
|
|
|
|
struct OneByteTypeConfig {
|
|
static constexpr int kTypeSizeInBytes = 1;
|
|
static constexpr int kType = 0x49;
|
|
static constexpr size_t kHeaderSize = 12;
|
|
static constexpr int kVariableLengthAlignment = 4;
|
|
};
|
|
|
|
class OneByteChunk : public TLVTrait<OneByteTypeConfig> {
|
|
public:
|
|
static constexpr size_t kVariableSize = 4;
|
|
|
|
void SerializeTo(std::vector<uint8_t>& out) {
|
|
BoundedByteWriter<OneByteTypeConfig::kHeaderSize> writer =
|
|
AllocateTLV(out, kVariableSize);
|
|
writer.Store32<4>(0x01020304);
|
|
writer.Store16<8>(0x0506);
|
|
writer.Store16<10>(0x0708);
|
|
|
|
uint8_t variable_data[kVariableSize] = {0xDE, 0xAD, 0xBE, 0xEF};
|
|
writer.CopyToVariableData(rtc::ArrayView<const uint8_t>(variable_data));
|
|
}
|
|
|
|
static std::optional<BoundedByteReader<OneByteTypeConfig::kHeaderSize>> Parse(
|
|
rtc::ArrayView<const uint8_t> data) {
|
|
return ParseTLV(data);
|
|
}
|
|
};
|
|
|
|
TEST(TlvDataTest, CanWriteOneByteTypeTlvs) {
|
|
std::vector<uint8_t> out;
|
|
OneByteChunk().SerializeTo(out);
|
|
|
|
EXPECT_THAT(out, SizeIs(OneByteTypeConfig::kHeaderSize +
|
|
OneByteChunk::kVariableSize));
|
|
EXPECT_THAT(out, ElementsAre(0x49, 0x00, 0x00, 0x10, 0x01, 0x02, 0x03, 0x04,
|
|
0x05, 0x06, 0x07, 0x08, 0xDE, 0xAD, 0xBE, 0xEF));
|
|
}
|
|
|
|
TEST(TlvDataTest, CanReadOneByteTypeTlvs) {
|
|
uint8_t data[] = {0x49, 0x00, 0x00, 0x10, 0x01, 0x02, 0x03, 0x04,
|
|
0x05, 0x06, 0x07, 0x08, 0xDE, 0xAD, 0xBE, 0xEF};
|
|
|
|
std::optional<BoundedByteReader<OneByteTypeConfig::kHeaderSize>> reader =
|
|
OneByteChunk::Parse(data);
|
|
ASSERT_TRUE(reader.has_value());
|
|
EXPECT_EQ(reader->Load32<4>(), 0x01020304U);
|
|
EXPECT_EQ(reader->Load16<8>(), 0x0506U);
|
|
EXPECT_EQ(reader->Load16<10>(), 0x0708U);
|
|
EXPECT_THAT(reader->variable_data(), ElementsAre(0xDE, 0xAD, 0xBE, 0xEF));
|
|
}
|
|
|
|
struct TwoByteTypeConfig {
|
|
static constexpr int kTypeSizeInBytes = 2;
|
|
static constexpr int kType = 31337;
|
|
static constexpr size_t kHeaderSize = 8;
|
|
static constexpr int kVariableLengthAlignment = 2;
|
|
};
|
|
|
|
class TwoByteChunk : public TLVTrait<TwoByteTypeConfig> {
|
|
public:
|
|
static constexpr size_t kVariableSize = 8;
|
|
|
|
void SerializeTo(std::vector<uint8_t>& out) {
|
|
BoundedByteWriter<TwoByteTypeConfig::kHeaderSize> writer =
|
|
AllocateTLV(out, kVariableSize);
|
|
writer.Store32<4>(0x01020304U);
|
|
|
|
uint8_t variable_data[] = {0x05, 0x06, 0x07, 0x08, 0xDE, 0xAD, 0xBE, 0xEF};
|
|
writer.CopyToVariableData(rtc::ArrayView<const uint8_t>(variable_data));
|
|
}
|
|
|
|
static std::optional<BoundedByteReader<TwoByteTypeConfig::kHeaderSize>> Parse(
|
|
rtc::ArrayView<const uint8_t> data) {
|
|
return ParseTLV(data);
|
|
}
|
|
};
|
|
|
|
TEST(TlvDataTest, CanWriteTwoByteTypeTlvs) {
|
|
std::vector<uint8_t> out;
|
|
|
|
TwoByteChunk().SerializeTo(out);
|
|
|
|
EXPECT_THAT(out, SizeIs(TwoByteTypeConfig::kHeaderSize +
|
|
TwoByteChunk::kVariableSize));
|
|
EXPECT_THAT(out, ElementsAre(0x7A, 0x69, 0x00, 0x10, 0x01, 0x02, 0x03, 0x04,
|
|
0x05, 0x06, 0x07, 0x08, 0xDE, 0xAD, 0xBE, 0xEF));
|
|
}
|
|
|
|
TEST(TlvDataTest, CanReadTwoByteTypeTlvs) {
|
|
uint8_t data[] = {0x7A, 0x69, 0x00, 0x10, 0x01, 0x02, 0x03, 0x04,
|
|
0x05, 0x06, 0x07, 0x08, 0xDE, 0xAD, 0xBE, 0xEF};
|
|
|
|
std::optional<BoundedByteReader<TwoByteTypeConfig::kHeaderSize>> reader =
|
|
TwoByteChunk::Parse(data);
|
|
EXPECT_TRUE(reader.has_value());
|
|
EXPECT_EQ(reader->Load32<4>(), 0x01020304U);
|
|
EXPECT_THAT(reader->variable_data(),
|
|
ElementsAre(0x05, 0x06, 0x07, 0x08, 0xDE, 0xAD, 0xBE, 0xEF));
|
|
}
|
|
|
|
TEST(TlvDataTest, CanHandleInvalidLengthSmallerThanFixedSize) {
|
|
// Has 'length=6', which is below the kHeaderSize of 8.
|
|
uint8_t data[] = {0x7A, 0x69, 0x00, 0x06, 0x01, 0x02, 0x03, 0x04};
|
|
|
|
EXPECT_FALSE(TwoByteChunk::Parse(data).has_value());
|
|
}
|
|
|
|
} // namespace
|
|
} // namespace dcsctp
|