1 // Copyright (c) 2020 The Bitcoin Core developers
2 // Distributed under the MIT software license, see the accompanying
3 // file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5 #include <netaddress.h>
6 #include <test/fuzz/fuzz.h>
7
8 #include <cstdint>
9 #include <vector>
10
11 //! asmap code that consumes nothing
12 static const std::vector<bool> IPV6_PREFIX_ASMAP = {};
13
14 //! asmap code that consumes the 96 prefix bits of ::ffff:0/96 (IPv4-in-IPv6 map)
15 static const std::vector<bool> IPV4_PREFIX_ASMAP = {
16 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
17 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
18 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
19 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
20 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
21 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
22 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
23 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
24 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
25 true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
26 true, true, false, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true, // Match 0xFF
27 true, true, false, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true // Match 0xFF
28 };
29
test_one_input(const std::vector<uint8_t> & buffer)30 void test_one_input(const std::vector<uint8_t>& buffer)
31 {
32 // Encoding: [7 bits: asmap size] [1 bit: ipv6?] [3-130 bytes: asmap] [4 or 16 bytes: addr]
33 if (buffer.size() < 1 + 3 + 4) return;
34 int asmap_size = 3 + (buffer[0] & 127);
35 bool ipv6 = buffer[0] & 128;
36 const size_t addr_size = ipv6 ? ADDR_IPV6_SIZE : ADDR_IPV4_SIZE;
37 if (buffer.size() < size_t(1 + asmap_size + addr_size)) return;
38 std::vector<bool> asmap = ipv6 ? IPV6_PREFIX_ASMAP : IPV4_PREFIX_ASMAP;
39 asmap.reserve(asmap.size() + 8 * asmap_size);
40 for (int i = 0; i < asmap_size; ++i) {
41 for (int j = 0; j < 8; ++j) {
42 asmap.push_back((buffer[1 + i] >> j) & 1);
43 }
44 }
45 if (!SanityCheckASMap(asmap)) return;
46
47 const uint8_t* addr_data = buffer.data() + 1 + asmap_size;
48 CNetAddr net_addr;
49 if (ipv6) {
50 assert(addr_size == ADDR_IPV6_SIZE);
51 net_addr.SetLegacyIPv6(Span<const uint8_t>(addr_data, addr_size));
52 } else {
53 assert(addr_size == ADDR_IPV4_SIZE);
54 in_addr ipv4;
55 memcpy(&ipv4, addr_data, addr_size);
56 net_addr.SetIP(CNetAddr{ipv4});
57 }
58 (void)net_addr.GetMappedAS(asmap);
59 }
60