Changeset View
Changeset View
Standalone View
Standalone View
src/test/fuzz/asmap.cpp
Show First 20 Lines • Show All 43 Lines • ▼ Show 20 Lines | |||||
void test_one_input(const std::vector<uint8_t> &buffer) { | void test_one_input(const std::vector<uint8_t> &buffer) { | ||||
// Encoding: [7 bits: asmap size] [1 bit: ipv6?] [3-130 bytes: asmap] [4 or | // Encoding: [7 bits: asmap size] [1 bit: ipv6?] [3-130 bytes: asmap] [4 or | ||||
// 16 bytes: addr] | // 16 bytes: addr] | ||||
if (buffer.size() < 1 + 3 + 4) { | if (buffer.size() < 1 + 3 + 4) { | ||||
return; | return; | ||||
} | } | ||||
int asmap_size = 3 + (buffer[0] & 127); | int asmap_size = 3 + (buffer[0] & 127); | ||||
bool ipv6 = buffer[0] & 128; | bool ipv6 = buffer[0] & 128; | ||||
int addr_size = ipv6 ? 16 : 4; | const size_t addr_size = ipv6 ? ADDR_IPV6_SIZE : ADDR_IPV4_SIZE; | ||||
if (buffer.size() < size_t(1 + asmap_size + addr_size)) { | if (buffer.size() < size_t(1 + asmap_size + addr_size)) { | ||||
return; | return; | ||||
} | } | ||||
std::vector<bool> asmap = ipv6 ? IPV6_PREFIX_ASMAP : IPV4_PREFIX_ASMAP; | std::vector<bool> asmap = ipv6 ? IPV6_PREFIX_ASMAP : IPV4_PREFIX_ASMAP; | ||||
asmap.reserve(asmap.size() + 8 * asmap_size); | asmap.reserve(asmap.size() + 8 * asmap_size); | ||||
for (int i = 0; i < asmap_size; ++i) { | for (int i = 0; i < asmap_size; ++i) { | ||||
for (int j = 0; j < 8; ++j) { | for (int j = 0; j < 8; ++j) { | ||||
asmap.push_back((buffer[1 + i] >> j) & 1); | asmap.push_back((buffer[1 + i] >> j) & 1); | ||||
} | } | ||||
} | } | ||||
if (!SanityCheckASMap(asmap)) { | if (!SanityCheckASMap(asmap)) { | ||||
return; | return; | ||||
} | } | ||||
const uint8_t *addr_data = buffer.data() + 1 + asmap_size; | |||||
CNetAddr net_addr; | CNetAddr net_addr; | ||||
net_addr.SetRaw(ipv6 ? NET_IPV6 : NET_IPV4, buffer.data() + 1 + asmap_size); | if (ipv6) { | ||||
assert(addr_size == ADDR_IPV6_SIZE); | |||||
net_addr.SetLegacyIPv6(Span<const uint8_t>(addr_data, addr_size)); | |||||
} else { | |||||
assert(addr_size == ADDR_IPV4_SIZE); | |||||
in_addr ipv4; | |||||
memcpy(&ipv4, addr_data, addr_size); | |||||
net_addr.SetIP(CNetAddr{ipv4}); | |||||
} | |||||
(void)net_addr.GetMappedAS(asmap); | (void)net_addr.GetMappedAS(asmap); | ||||
} | } |