2
0
mirror of https://gitlab.isc.org/isc-projects/kea synced 2025-08-31 14:05:33 +00:00

[#1415] Improved offsetAddres

According to the review comments, the conversion of the offset to a vector
was simplified. In addition, when the offset address exceeds the maximum
value for IPv4 address the max value is returned.
This commit is contained in:
Marcin Siodelski
2020-09-16 12:26:34 +02:00
parent e8c3f1afde
commit 5cfc8f929a
3 changed files with 15 additions and 10 deletions

View File

@@ -381,21 +381,23 @@ IOAddress offsetAddress(const IOAddress& addr, uint64_t offset) {
// If this is IPv4 addrss we utilize the conversion to uint32_t.
if (addr.isV4()) {
return (IOAddress(addr.toUint32() + offset));
auto addr_uint32 = static_cast<uint64_t>(addr.toUint32());
// If the result would exceed the maximum possible IPv4 address, let's return
// the maximum IPv4 address.
if (static_cast<uint64_t>(std::numeric_limits<uint32_t>::max() - addr_uint32) < offset) {
return (IOAddress(std::numeric_limits<uint32_t>::max()));
}
return (IOAddress(static_cast<uint32_t>(addr_uint32 + offset)));
}
// This is IPv6 address. Let's first convert the offset value to network
// byte order and store within the vector.
std::vector<uint8_t> offset_bytes(8);
int offset_idx = 0;
offset_bytes[offset_idx++] = static_cast<uint8_t>((offset & 0xff00000000000000) >> 56);
offset_bytes[offset_idx++] = static_cast<uint8_t>((offset & 0x00ff000000000000) >> 48);
offset_bytes[offset_idx++] = static_cast<uint8_t>((offset & 0x0000ff0000000000) >> 40);
offset_bytes[offset_idx++] = static_cast<uint8_t>((offset & 0x000000ff00000000) >> 32);
offset_bytes[offset_idx++] = static_cast<uint8_t>((offset & 0x00000000ff000000) >> 24);
offset_bytes[offset_idx++] = static_cast<uint8_t>((offset & 0x0000000000ff0000) >> 16);
offset_bytes[offset_idx++] = static_cast<uint8_t>((offset & 0x000000000000ff00) >> 8);
offset_bytes[offset_idx++] = static_cast<uint8_t>(offset & 0x00000000000000ff);
for (int offset_idx = offset_bytes.size() - 1; offset_idx >= 0; --offset_idx) {
offset_bytes[offset_idx] = static_cast<uint8_t>(offset & 0x00000000000000ff);
offset = offset >> 8;
}
// Convert the IPv6 address to vector.
auto addr_bytes = addr.toBytes();

View File

@@ -84,7 +84,8 @@ uint64_t prefixesInRange(const uint8_t pool_len, const uint8_t delegated_len);
///
/// Adds offset to the IPv4 or iPv6 address and finds the resulting address.
/// Note that the current limitation is the maximum value of the offset,
/// i.e. max uint64_t.
/// i.e. max uint64_t. If the sum of the IPv4 address and the offset exceeds
/// the maximum value of uint32_t type, the 255.255.255.255 is returned.
///
/// @param addr input address
/// @param offset distance of the returned address from the input address.

View File

@@ -371,6 +371,8 @@ TEST(AddrUtilitiesTest, prefixesInRange) {
TEST(AddrUtilitiesTest, offsetIPv4Address) {
EXPECT_EQ("10.1.2.46", offsetAddress(IOAddress("10.1.1.45"), 257).toText());
EXPECT_EQ("10.1.7.9", offsetAddress(IOAddress("10.1.1.45"), 1500).toText());
// Using very large offset. The maximum IPv4 address should be returned.
EXPECT_EQ("255.255.255.255", offsetAddress(IOAddress("255.255.254.254"), 0xFFFFFFFFFFFFFFFA).toText());
}
// Checks the function which finds an IPv6 address from input address and offset.