Branch data     Line data    Source code

       1                 :             : // Copyright (c) 2021-2022 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 <netbase.h>
       7                 :             : #include <test/fuzz/FuzzedDataProvider.h>
       8                 :             : #include <test/fuzz/fuzz.h>
       9                 :             : #include <test/fuzz/util/net.h>
      10                 :             : 
      11                 :             : #include <cstdint>
      12                 :             : #include <string>
      13                 :             : #include <vector>
      14                 :             : 
      15   [ +  -  +  - ]:         667 : FUZZ_TARGET(netbase_dns_lookup)
      16                 :             : {
      17                 :         664 :     FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
      18                 :         664 :     const std::string name = fuzzed_data_provider.ConsumeRandomLengthString(512);
      19         [ +  - ]:         664 :     const unsigned int max_results = fuzzed_data_provider.ConsumeIntegral<unsigned int>();
      20         [ +  - ]:         664 :     const bool allow_lookup = fuzzed_data_provider.ConsumeBool();
      21         [ +  - ]:         664 :     const uint16_t default_port = fuzzed_data_provider.ConsumeIntegral<uint16_t>();
      22                 :             : 
      23                 :        3870 :     auto fuzzed_dns_lookup_function = [&](const std::string&, bool) {
      24                 :        3206 :         std::vector<CNetAddr> resolved_addresses;
      25   [ +  -  +  +  :      539893 :         LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
                   +  + ]
      26         [ +  - ]:      536687 :             resolved_addresses.push_back(ConsumeNetAddr(fuzzed_data_provider));
      27                 :      536687 :         }
      28                 :        3206 :         return resolved_addresses;
      29         [ +  - ]:        3206 :     };
      30                 :             : 
      31                 :             :     {
      32         [ +  - ]:         664 :         const std::vector<CNetAddr> resolved_addresses{LookupHost(name, max_results, allow_lookup, fuzzed_dns_lookup_function)};
      33         [ +  + ]:      175349 :         for (const CNetAddr& resolved_address : resolved_addresses) {
      34   [ +  -  +  - ]:      174685 :             assert(!resolved_address.IsInternal());
      35                 :      174685 :         }
      36   [ +  +  +  - ]:         664 :         assert(resolved_addresses.size() <= max_results || max_results == 0);
      37                 :         664 :     }
      38                 :             :     {
      39         [ +  - ]:         664 :         const std::optional<CNetAddr> resolved_address{LookupHost(name, allow_lookup, fuzzed_dns_lookup_function)};
      40         [ +  + ]:         664 :         if (resolved_address.has_value()) {
      41   [ +  -  +  -  :         214 :             assert(!resolved_address.value().IsInternal());
                   +  - ]
      42                 :         214 :         }
      43                 :         664 :     }
      44                 :             :     {
      45         [ +  - ]:         664 :         const std::vector<CService> resolved_services{Lookup(name, default_port, allow_lookup, max_results, fuzzed_dns_lookup_function)};
      46         [ +  + ]:      137673 :         for (const CNetAddr& resolved_service : resolved_services) {
      47   [ +  -  +  - ]:      137009 :             assert(!resolved_service.IsInternal());
      48                 :      137009 :         }
      49   [ +  +  +  - ]:         664 :         assert(resolved_services.size() <= max_results || max_results == 0);
      50                 :         664 :     }
      51                 :             :     {
      52         [ +  - ]:         664 :         const std::optional<CService> resolved_service{Lookup(name, default_port, allow_lookup, fuzzed_dns_lookup_function)};
      53         [ +  + ]:         664 :         if (resolved_service.has_value()) {
      54   [ +  -  +  -  :         154 :             assert(!resolved_service.value().IsInternal());
                   +  - ]
      55                 :         154 :         }
      56                 :         664 :     }
      57                 :             :     {
      58         [ +  - ]:         664 :         CService resolved_service = LookupNumeric(name, default_port, fuzzed_dns_lookup_function);
      59   [ +  -  +  - ]:         664 :         assert(!resolved_service.IsInternal());
      60                 :         664 :     }
      61                 :             :     {
      62         [ +  - ]:         664 :         (void)LookupSubNet(name);
      63                 :             :     }
      64                 :         664 : }