Line data    Source code

       1             : // Copyright (c) 2020-2021 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 <netgroup.h>
       7             : #include <test/fuzz/fuzz.h>
       8             : #include <util/asmap.h>
       9             : 
      10             : #include <cstdint>
      11             : #include <vector>
      12             : 
      13             : //! asmap code that consumes nothing
      14           2 : static const std::vector<bool> IPV6_PREFIX_ASMAP = {};
      15             : 
      16             : //! asmap code that consumes the 96 prefix bits of ::ffff:0/96 (IPv4-in-IPv6 map)
      17           2 : static const std::vector<bool> IPV4_PREFIX_ASMAP = {
      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           2 :     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, false, false, false, false, false, false, false, false, // Match 0x00
      27             :     true, true, false, true, true, true, true, true, true, true, false, false, false, false, false, false, false, false, // Match 0x00
      28             :     true, true, false, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true,         // Match 0xFF
      29             :     true, true, false, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true          // Match 0xFF
      30             : };
      31             : 
      32           4 : FUZZ_TARGET(asmap)
      33             : {
      34             :     // Encoding: [7 bits: asmap size] [1 bit: ipv6?] [3-130 bytes: asmap] [4 or 16 bytes: addr]
      35           0 :     if (buffer.size() < 1 + 3 + 4) return;
      36           0 :     int asmap_size = 3 + (buffer[0] & 127);
      37           0 :     bool ipv6 = buffer[0] & 128;
      38           0 :     const size_t addr_size = ipv6 ? ADDR_IPV6_SIZE : ADDR_IPV4_SIZE;
      39           0 :     if (buffer.size() < size_t(1 + asmap_size + addr_size)) return;
      40           0 :     std::vector<bool> asmap = ipv6 ? IPV6_PREFIX_ASMAP : IPV4_PREFIX_ASMAP;
      41           0 :     asmap.reserve(asmap.size() + 8 * asmap_size);
      42           0 :     for (int i = 0; i < asmap_size; ++i) {
      43           0 :         for (int j = 0; j < 8; ++j) {
      44           0 :             asmap.push_back((buffer[1 + i] >> j) & 1);
      45           0 :         }
      46           0 :     }
      47           0 :     if (!SanityCheckASMap(asmap, 128)) return;
      48             : 
      49           0 :     const uint8_t* addr_data = buffer.data() + 1 + asmap_size;
      50           0 :     CNetAddr net_addr;
      51           0 :     if (ipv6) {
      52           0 :         assert(addr_size == ADDR_IPV6_SIZE);
      53           0 :         net_addr.SetLegacyIPv6({addr_data, addr_size});
      54           0 :     } else {
      55           0 :         assert(addr_size == ADDR_IPV4_SIZE);
      56             :         in_addr ipv4;
      57           0 :         memcpy(&ipv4, addr_data, addr_size);
      58           0 :         net_addr.SetIP(CNetAddr{ipv4});
      59             :     }
      60           0 :     NetGroupManager netgroupman{asmap};
      61           0 :     (void)netgroupman.GetMappedAS(net_addr);
      62           0 : }