webrtc/net/dcsctp/packet/bounded_byte_reader_test.cc
Victor Boivie 7d3c49a171 dcsctp: Add bounded byte reader and writer
Packets, chunks, parameters and error causes - the SCTP entities
that are sent on the wire - are buffers with fields that are stored
in big endian and that generally consist of a fixed header size, and
a variable sized part, that can e.g. be encoded sub-fields or
serialized strings.

The BoundedByteReader and BoundedByteWriter utilities make it easy
to read those fields with as much aid from the compiler as possible,
by having compile-time assertions that fields are not accessed
outside the buffer's span.

There are some byte reading functionality already in modules/rtp_rtcp,
but that module would be a bit unfortunate to depend on, and doesn't
have the compile time bounds checking that is the biggest feature of
this abstraction of an rtc::ArrayView.

Bug: webrtc:12614
Change-Id: I9fc641aff22221018dda9add4e2c44853c0f64f0
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/212967
Commit-Queue: Victor Boivie <boivie@webrtc.org>
Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org>
Reviewed-by: Tommi <tommi@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#33597}
2021-03-31 08:27:37 +00:00

43 lines
1.3 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/bounded_byte_reader.h"
#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;
TEST(BoundedByteReaderTest, CanLoadData) {
uint8_t data[14] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4};
BoundedByteReader<8> reader(data);
EXPECT_EQ(reader.variable_data_size(), 6U);
EXPECT_EQ(reader.Load32<0>(), 0x01020304U);
EXPECT_EQ(reader.Load32<4>(), 0x05060708U);
EXPECT_EQ(reader.Load16<4>(), 0x0506U);
EXPECT_EQ(reader.Load8<4>(), 0x05U);
EXPECT_EQ(reader.Load8<5>(), 0x06U);
BoundedByteReader<6> sub = reader.sub_reader<6>(0);
EXPECT_EQ(sub.Load16<0>(), 0x0900U);
EXPECT_EQ(sub.Load32<0>(), 0x09000102U);
EXPECT_EQ(sub.Load16<4>(), 0x0304U);
EXPECT_THAT(reader.variable_data(), ElementsAre(9, 0, 1, 2, 3, 4));
}
} // namespace
} // namespace dcsctp