Changeset View
Changeset View
Standalone View
Standalone View
src/test/fuzz/asmap.cpp
// Copyright (c) 2020 The Bitcoin Core developers | // Copyright (c) 2020 The Bitcoin Core developers | ||||
// Distributed under the MIT software license, see the accompanying | // Distributed under the MIT software license, see the accompanying | ||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php. | // file COPYING or http://www.opensource.org/licenses/mit-license.php. | ||||
#include <netaddress.h> | #include <netaddress.h> | ||||
#include <test/fuzz/FuzzedDataProvider.h> | |||||
#include <test/fuzz/fuzz.h> | #include <test/fuzz/fuzz.h> | ||||
#include <cstdint> | #include <cstdint> | ||||
#include <vector> | #include <vector> | ||||
//! asmap code that consumes nothing | |||||
static const std::vector<bool> IPV6_PREFIX_ASMAP = {}; | |||||
//! asmap code that consumes the 96 prefix bits of ::ffff:0/96 (IPv4-in-IPv6 | |||||
//! map) | |||||
static const std::vector<bool> IPV4_PREFIX_ASMAP = { | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, false, false, false, false, false, false, false, false, // Match 0x00 | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, true, true, true, true, true, true, true, true, // Match 0xFF | |||||
true, true, false, true, true, true, true, true, true, | |||||
true, true, true, true, true, true, true, true, true // Match 0xFF | |||||
}; | |||||
void test_one_input(const std::vector<uint8_t> &buffer) { | void test_one_input(const std::vector<uint8_t> &buffer) { | ||||
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size()); | // Encoding: [7 bits: asmap size] [1 bit: ipv6?] [3-130 bytes: asmap] [4 or | ||||
const Network network = | // 16 bytes: addr] | ||||
fuzzed_data_provider.PickValueInArray({NET_IPV4, NET_IPV6}); | if (buffer.size() < 1 + 3 + 4) { | ||||
if (fuzzed_data_provider.remaining_bytes() < 16) { | |||||
return; | return; | ||||
} | } | ||||
CNetAddr net_addr; | int asmap_size = 3 + (buffer[0] & 127); | ||||
net_addr.SetRaw(network, | bool ipv6 = buffer[0] & 128; | ||||
fuzzed_data_provider.ConsumeBytes<uint8_t>(16).data()); | int addr_size = ipv6 ? 16 : 4; | ||||
std::vector<bool> asmap; | if (buffer.size() < size_t(1 + asmap_size + addr_size)) { | ||||
for (const char cur_byte : | return; | ||||
fuzzed_data_provider.ConsumeRemainingBytes<char>()) { | |||||
for (int bit = 0; bit < 8; ++bit) { | |||||
asmap.push_back((cur_byte >> bit) & 1); | |||||
} | } | ||||
std::vector<bool> asmap = ipv6 ? IPV6_PREFIX_ASMAP : IPV4_PREFIX_ASMAP; | |||||
asmap.reserve(asmap.size() + 8 * asmap_size); | |||||
for (int i = 0; i < asmap_size; ++i) { | |||||
for (int j = 0; j < 8; ++j) { | |||||
asmap.push_back((buffer[1 + i] >> j) & 1); | |||||
} | } | ||||
} | |||||
if (!SanityCheckASMap(asmap)) { | |||||
return; | |||||
} | |||||
CNetAddr net_addr; | |||||
net_addr.SetRaw(ipv6 ? NET_IPV6 : NET_IPV4, buffer.data() + 1 + asmap_size); | |||||
(void)net_addr.GetMappedAS(asmap); | (void)net_addr.GetMappedAS(asmap); | ||||
} | } |