Mirror of BoringSSL (grpc依赖)
https://boringssl.googlesource.com/boringssl
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
3.8 KiB
107 lines
3.8 KiB
// 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
|
|
|