1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "net/dns/https_record_rdata.h"
#include <fuzzer/FuzzedDataProvider.h>
#include <stdint.h>
#include <memory>
#include <string>
#include <vector>
#include "base/check.h"
#include "base/containers/contains.h"
#include "net/base/ip_address.h"
#include "net/dns/public/dns_protocol.h"
namespace net {
namespace {
void ParseAndExercise(FuzzedDataProvider& data_provider) {
const size_t size = data_provider.ConsumeIntegralInRange<size_t>(
0u, data_provider.remaining_bytes());
const std::vector<uint8_t> data1 = data_provider.ConsumeBytes<uint8_t>(size);
std::unique_ptr<HttpsRecordRdata> parsed = HttpsRecordRdata::Parse(data1);
std::unique_ptr<HttpsRecordRdata> parsed2 = HttpsRecordRdata::Parse(data1);
std::unique_ptr<HttpsRecordRdata> parsed3 =
HttpsRecordRdata::Parse(data_provider.ConsumeRemainingBytes<uint8_t>());
CHECK_EQ(!!parsed, !!parsed2);
if (!parsed)
return;
// `parsed` and `parsed2` parsed from the same data, so they should always be
// equal.
CHECK(parsed->IsEqual(parsed.get()));
CHECK(parsed->IsEqual(parsed2.get()));
CHECK(parsed2->IsEqual(parsed.get()));
// Attempt comparison with an rdata parsed from separate data. IsEqual() will
// probably return false most of the time, but easily could be true if the
// input data is similar enough.
if (parsed3)
CHECK_EQ(parsed->IsEqual(parsed3.get()), parsed3->IsEqual(parsed.get()));
CHECK_EQ(parsed->Type(), dns_protocol::kTypeHttps);
if (parsed->IsAlias()) {
AliasFormHttpsRecordRdata* alias = parsed->AsAliasForm();
alias->alias_name();
} else {
ServiceFormHttpsRecordRdata* service = parsed->AsServiceForm();
CHECK_GT(service->priority(), 0);
service->service_name();
service->alpn_ids();
service->default_alpn();
service->port();
service->ech_config();
service->trust_anchor_ids();
service->IsCompatible();
CHECK(!base::Contains(service->mandatory_keys(),
dns_protocol::kHttpsServiceParamKeyMandatory));
std::vector<IPAddress> ipv4_hint = service->ipv4_hint();
for (const IPAddress& address : ipv4_hint) {
CHECK(address.IsIPv4());
}
std::vector<IPAddress> ipv6_hint = service->ipv6_hint();
for (const IPAddress& address : ipv6_hint) {
CHECK(address.IsIPv6());
}
}
}
} // namespace
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
FuzzedDataProvider data_provider(data, size);
ParseAndExercise(data_provider);
return 0;
}
} // namespace net
|