blob: 8ec7452ff31e94c04f3c88a80853b33b2f8207de [file] [log] [blame]
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ip_util.h"
#include <string.h>
#include "input.h"
#include <gtest/gtest.h>
namespace bssl {
TEST(IPUtilTest, IsValidNetmask) {
uint8_t kWrongSize[3] = {0xff, 0xff, 0xff};
EXPECT_FALSE(IsValidNetmask(der::Input(kWrongSize)));
// All zeros is a valid netmask.
uint8_t kZeroIPv4[4] = {0};
EXPECT_TRUE(IsValidNetmask(der::Input(kZeroIPv4)));
uint8_t kZeroIPv6[16] = {0};
EXPECT_TRUE(IsValidNetmask(der::Input(kZeroIPv6)));
// Test all valid non-zero IPv4 netmasks.
for (int i = 0; i < 4; i++) {
uint8_t addr[4] = {0};
memset(addr, 0xff, i);
for (int shift = 0; shift < 8; shift++) {
addr[i] = 0xff << shift;
EXPECT_TRUE(IsValidNetmask(der::Input(addr)));
}
}
// Test all valid non-zero IPv6 netmasks.
for (int i = 0; i < 16; i++) {
uint8_t addr[16] = {0};
memset(addr, 0xff, i);
for (int shift = 0; shift < 8; shift++) {
addr[i] = 0xff << shift;
EXPECT_TRUE(IsValidNetmask(der::Input(addr)));
}
}
// Error within a byte.
uint8_t kInvalidIPv4[4] = {0xff, 0xff, 0x81, 0x00};
EXPECT_FALSE(IsValidNetmask(der::Input(kInvalidIPv4)));
uint8_t kInvalidIPv6[16] = {0xff, 0xff, 0x81, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
EXPECT_FALSE(IsValidNetmask(der::Input(kInvalidIPv6)));
// Error at the end.
uint8_t kInvalidIPv4_2[4] = {0xff, 0xff, 0x80, 0x01};
EXPECT_FALSE(IsValidNetmask(der::Input(kInvalidIPv4_2)));
uint8_t kInvalidIPv6_2[16] = {0xff, 0xff, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01};
EXPECT_FALSE(IsValidNetmask(der::Input(kInvalidIPv6_2)));
// Leading zero.
uint8_t kInvalidIPv4_3[4] = {0x00, 0xff, 0x80, 0x00};
EXPECT_FALSE(IsValidNetmask(der::Input(kInvalidIPv4_3)));
uint8_t kInvalidIPv6_3[16] = {0x00, 0xff, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01};
EXPECT_FALSE(IsValidNetmask(der::Input(kInvalidIPv6_3)));
}
TEST(IPUtilTest, IPAddressMatchesWithNetmask) {
// Under a zero mask, any two addresses are equal.
{
uint8_t kMask[4] = {0};
uint8_t kAddr1[4] = {1, 2, 3, 4};
uint8_t kAddr2[4] = {255, 254, 253, 252};
EXPECT_TRUE(IPAddressMatchesWithNetmask(
der::Input(kAddr1), der::Input(kAddr1), der::Input(kMask)));
EXPECT_TRUE(IPAddressMatchesWithNetmask(
der::Input(kAddr1), der::Input(kAddr2), der::Input(kMask)));
}
// Under an all ones mask, all bits of the address are checked.
{
uint8_t kMask[4] = {0xff, 0xff, 0xff, 0xff};
uint8_t kAddr1[4] = {1, 2, 3, 4};
uint8_t kAddr2[4] = {255, 254, 253, 252};
uint8_t kAddr3[4] = {1, 2, 3, 5};
EXPECT_TRUE(IPAddressMatchesWithNetmask(
der::Input(kAddr1), der::Input(kAddr1), der::Input(kMask)));
EXPECT_FALSE(IPAddressMatchesWithNetmask(
der::Input(kAddr1), der::Input(kAddr2), der::Input(kMask)));
EXPECT_FALSE(IPAddressMatchesWithNetmask(
der::Input(kAddr1), der::Input(kAddr3), der::Input(kMask)));
}
// In general, only masked bits are considered.
{
uint8_t kMask[4] = {0xff, 0xff, 0x80, 0x00};
uint8_t kAddr1[4] = {1, 2, 3, 4};
uint8_t kAddr2[4] = {1, 2, 0x7f, 0xff};
uint8_t kAddr3[4] = {2, 2, 3, 4};
EXPECT_TRUE(IPAddressMatchesWithNetmask(
der::Input(kAddr1), der::Input(kAddr1), der::Input(kMask)));
EXPECT_TRUE(IPAddressMatchesWithNetmask(
der::Input(kAddr1), der::Input(kAddr2), der::Input(kMask)));
EXPECT_FALSE(IPAddressMatchesWithNetmask(
der::Input(kAddr1), der::Input(kAddr3), der::Input(kMask)));
}
}
} // namespace net