webrtc/net/dcsctp/common/math_test.cc
Victor Boivie 78aa5cd359 dcsctp: Ensure packet size doesn't exceed MTU
Due to a previous refactoring, the SCTP packet header is only added when
the first chunk is written. This wasn't reflected in the
`bytes_remaining`, which made it add more than could fit within the MTU.

Additionally, the maximum packet size must be even divisible by four as
padding will be added to chunks that are not even divisble by four (up
to three bytes of padding). So compensate for that.

Bug: webrtc:12614
Change-Id: I6b57dfbf88d1fcfcbf443038915dd180e796191a
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/215145
Reviewed-by: Tommi <tommi@webrtc.org>
Reviewed-by: Florent Castelli <orphis@webrtc.org>
Commit-Queue: Victor Boivie <boivie@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#33760}
2021-04-16 14:42:44 +00:00

116 lines
5.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/common/math.h"
#include "test/gmock.h"
namespace dcsctp {
namespace {
TEST(MathUtilTest, CanRoundUpTo4) {
// Signed numbers
EXPECT_EQ(RoundUpTo4(static_cast<int>(-5)), -4);
EXPECT_EQ(RoundUpTo4(static_cast<int>(-4)), -4);
EXPECT_EQ(RoundUpTo4(static_cast<int>(-3)), 0);
EXPECT_EQ(RoundUpTo4(static_cast<int>(-2)), 0);
EXPECT_EQ(RoundUpTo4(static_cast<int>(-1)), 0);
EXPECT_EQ(RoundUpTo4(static_cast<int>(0)), 0);
EXPECT_EQ(RoundUpTo4(static_cast<int>(1)), 4);
EXPECT_EQ(RoundUpTo4(static_cast<int>(2)), 4);
EXPECT_EQ(RoundUpTo4(static_cast<int>(3)), 4);
EXPECT_EQ(RoundUpTo4(static_cast<int>(4)), 4);
EXPECT_EQ(RoundUpTo4(static_cast<int>(5)), 8);
EXPECT_EQ(RoundUpTo4(static_cast<int>(6)), 8);
EXPECT_EQ(RoundUpTo4(static_cast<int>(7)), 8);
EXPECT_EQ(RoundUpTo4(static_cast<int>(8)), 8);
EXPECT_EQ(RoundUpTo4(static_cast<int64_t>(10000000000)), 10000000000);
EXPECT_EQ(RoundUpTo4(static_cast<int64_t>(10000000001)), 10000000004);
// Unsigned numbers
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(0)), 0u);
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(1)), 4u);
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(2)), 4u);
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(3)), 4u);
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(4)), 4u);
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(5)), 8u);
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(6)), 8u);
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(7)), 8u);
EXPECT_EQ(RoundUpTo4(static_cast<unsigned int>(8)), 8u);
EXPECT_EQ(RoundUpTo4(static_cast<uint64_t>(10000000000)), 10000000000u);
EXPECT_EQ(RoundUpTo4(static_cast<uint64_t>(10000000001)), 10000000004u);
}
TEST(MathUtilTest, CanRoundDownTo4) {
// Signed numbers
EXPECT_EQ(RoundDownTo4(static_cast<int>(-5)), -8);
EXPECT_EQ(RoundDownTo4(static_cast<int>(-4)), -4);
EXPECT_EQ(RoundDownTo4(static_cast<int>(-3)), -4);
EXPECT_EQ(RoundDownTo4(static_cast<int>(-2)), -4);
EXPECT_EQ(RoundDownTo4(static_cast<int>(-1)), -4);
EXPECT_EQ(RoundDownTo4(static_cast<int>(0)), 0);
EXPECT_EQ(RoundDownTo4(static_cast<int>(1)), 0);
EXPECT_EQ(RoundDownTo4(static_cast<int>(2)), 0);
EXPECT_EQ(RoundDownTo4(static_cast<int>(3)), 0);
EXPECT_EQ(RoundDownTo4(static_cast<int>(4)), 4);
EXPECT_EQ(RoundDownTo4(static_cast<int>(5)), 4);
EXPECT_EQ(RoundDownTo4(static_cast<int>(6)), 4);
EXPECT_EQ(RoundDownTo4(static_cast<int>(7)), 4);
EXPECT_EQ(RoundDownTo4(static_cast<int>(8)), 8);
EXPECT_EQ(RoundDownTo4(static_cast<int64_t>(10000000000)), 10000000000);
EXPECT_EQ(RoundDownTo4(static_cast<int64_t>(10000000001)), 10000000000);
// Unsigned numbers
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(0)), 0u);
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(1)), 0u);
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(2)), 0u);
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(3)), 0u);
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(4)), 4u);
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(5)), 4u);
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(6)), 4u);
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(7)), 4u);
EXPECT_EQ(RoundDownTo4(static_cast<unsigned int>(8)), 8u);
EXPECT_EQ(RoundDownTo4(static_cast<uint64_t>(10000000000)), 10000000000u);
EXPECT_EQ(RoundDownTo4(static_cast<uint64_t>(10000000001)), 10000000000u);
}
TEST(MathUtilTest, IsDivisibleBy4) {
// Signed numbers
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(-4)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(-3)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(-2)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(-1)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(0)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(1)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(2)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(3)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(4)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(5)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(6)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(7)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<int>(8)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<int64_t>(10000000000)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<int64_t>(10000000001)), false);
// Unsigned numbers
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(0)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(1)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(2)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(3)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(4)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(5)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(6)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(7)), false);
EXPECT_EQ(IsDivisibleBy4(static_cast<unsigned int>(8)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<uint64_t>(10000000000)), true);
EXPECT_EQ(IsDivisibleBy4(static_cast<uint64_t>(10000000001)), false);
}
} // namespace
} // namespace dcsctp